1 files changed, 125 insertions, 0 deletions
diff --git a/desktop-widgets/downloadfromdivecomputer.h b/desktop-widgets/downloadfromdivecomputer.h
new file mode 100644
index 000000000..7acd49e95
--- /dev/null
+++ b/desktop-widgets/downloadfromdivecomputer.h
@@ -0,0 +1,125 @@
+#ifndef DOWNLOADFROMDIVECOMPUTER_H
+#define DOWNLOADFROMDIVECOMPUTER_H
+
+#include <QDialog>
+#include <QThread>
+#include <QHash>
+#include <QMap>
+#include <QAbstractTableModel>
+
+#include "libdivecomputer.h"
+#include "configuredivecomputerdialog.h"
+#include "ui_downloadfromdivecomputer.h"
+
+#if defined(BT_SUPPORT)
+#include "btdeviceselectiondialog.h"
+#endif
+
+class QStringListModel;
+
+class DownloadThread : public QThread {
+	Q_OBJECT
+public:
+	DownloadThread(QObject *parent, device_data_t *data);
+	virtual void run();
+
+	QString error;
+
+private:
+	device_data_t *data;
+};
+
+class DiveImportedModel : public QAbstractTableModel
+{
+	Q_OBJECT
+public:
+	DiveImportedModel(QObject *o);
+	int columnCount(const QModelIndex& index = QModelIndex()) const;
+	int rowCount(const QModelIndex& index = QModelIndex()) const;
+	QVariant data(const QModelIndex& index, int role) const;
+	QVariant headerData(int section, Qt::Orientation orientation, int role) const;
+	void setImportedDivesIndexes(int first, int last);
+	Qt::ItemFlags flags(const QModelIndex &index) const;
+	void clearTable();
+
+public
+slots:
+	void changeSelected(QModelIndex clickedIndex);
+	void selectAll();
+	void selectNone();
+
+private:
+	int firstIndex;
+	int lastIndex;
+	bool *checkStates;
+};
+
+class DownloadFromDCWidget : public QDialog {
+	Q_OBJECT
+public:
+	explicit DownloadFromDCWidget(QWidget *parent = 0, Qt::WindowFlags f = 0);
+	void reject();
+
+	enum states {
+		INITIAL,
+		DOWNLOADING,
+		CANCELLING,
+		ERROR,
+		DONE,
+	};
+
+public
+slots:
+	void on_downloadCancelRetryButton_clicked();
+	void on_ok_clicked();
+	void on_cancel_clicked();
+	void on_search_clicked();
+	void on_vendor_currentIndexChanged(const QString &vendor);
+	void on_product_currentIndexChanged(const QString &product);
+
+	void onDownloadThreadFinished();
+	void updateProgressBar();
+	void checkLogFile(int state);
+	void checkDumpFile(int state);
+	void pickDumpFile();
+	void pickLogFile();
+#if defined(BT_SUPPORT)
+	void enableBluetoothMode(int state);
+	void selectRemoteBluetoothDevice();
+	void bluetoothSelectionDialogIsFinished(int result);
+#endif
+private:
+	void markChildrenAsDisabled();
+	void markChildrenAsEnabled();
+
+	Ui::DownloadFromDiveComputer ui;
+	DownloadThread *thread;
+	bool downloading;
+
+	QStringList vendorList;
+	QHash<QString, QStringList> productList;
+	QMap<QString, dc_descriptor_t *> descriptorLookup;
+	device_data_t data;
+	int previousLast;
+
+	QStringListModel *vendorModel;
+	QStringListModel *productModel;
+	void fill_computer_list();
+	void fill_device_list(int dc_type);
+	QString logFile;
+	QString dumpFile;
+	QTimer *timer;
+	bool dumpWarningShown;
+	OstcFirmwareCheck *ostcFirmwareCheck;
+	DiveImportedModel *diveImportedModel;
+#if defined(BT_SUPPORT)
+	BtDeviceSelectionDialog *btDeviceSelectionDialog;
+#endif
+
+public:
+	bool preferDownloaded();
+	void updateState(states state);
+	states currentState;
+};
+
+#endif // DOWNLOADFROMDIVECOMPUTER_H