2 files changed, 41 insertions, 7 deletions
diff --git a/qt-models/diveimportedmodel.cpp b/qt-models/diveimportedmodel.cpp
index 08c36e7e6..781cd72c6 100644
--- a/qt-models/diveimportedmodel.cpp
+++ b/qt-models/diveimportedmodel.cpp
@@ -7,6 +7,8 @@ DiveImportedModel::DiveImportedModel(QObject *o) : QAbstractTableModel(o),
 	checkStates(nullptr),
 	diveTable(nullptr)
 {
+	// Defaults to downloadTable, can be changed later.
+	diveTable = &downloadTable;
 }
 
 int DiveImportedModel::columnCount(const QModelIndex &model) const
@@ -25,8 +27,16 @@ QVariant DiveImportedModel::headerData(int section, Qt::Orientation orientation,
 {
 	if (orientation == Qt::Vertical)
 		return QVariant();
+
+	// widgets access the model via index.column(), qml via role.
+	int column = section;
+	if (role == DateTime || role == Duration || role == Depth) {
+		column = role - DateTime;
+		role = Qt::DisplayRole;
+	}
+
 	if (role == Qt::DisplayRole) {
-		switch (section) {
+		switch (column) {
 		case 0:
 			return QVariant(tr("Date/time"));
 		case 1:
@@ -40,7 +50,7 @@ QVariant DiveImportedModel::headerData(int section, Qt::Orientation orientation,
 
 void DiveImportedModel::setDiveTable(struct dive_table* table)
 {
-    diveTable = table;
+	diveTable = table;
 }
 
 QVariant DiveImportedModel::data(const QModelIndex &index, int role) const
@@ -54,8 +64,16 @@ QVariant DiveImportedModel::data(const QModelIndex &index, int role) const
 	struct dive *d = get_dive_from_table(index.row() + firstIndex, diveTable);
 	if (!d)
 		return QVariant();
+
+	// widgets access the model via index.column(), qml via role.
+	int column = index.column();
+	if (role == DateTime || role == Duration || role == Depth) {
+		column = role - DateTime;
+		role = Qt::DisplayRole;
+	}
+
 	if (role == Qt::DisplayRole) {
-		switch (index.column()) {
+		switch (column) {
 		case 0:
 			return QVariant(get_short_dive_date_string(d->when));
 		case 1:
@@ -119,3 +137,16 @@ void DiveImportedModel::setImportedDivesIndexes(int first, int last)
 	memset(checkStates, true, last - first + 1);
 	endInsertRows();
 }
+
+void DiveImportedModel::repopulate()
+{
+	setImportedDivesIndexes(0, diveTable->nr-1);
+}
+
+QHash<int, QByteArray> DiveImportedModel::roleNames() const {
+	static QHash<int, QByteArray> roles = {
+		{ DateTime, "datetime"},
+		{ Depth, "depth"},
+		{ Duration, "duration"}};
+	return roles;
+}
diff --git a/qt-models/diveimportedmodel.h b/qt-models/diveimportedmodel.h
index 1127fc97e..e68fec0f8 100644
--- a/qt-models/diveimportedmodel.h
+++ b/qt-models/diveimportedmodel.h
@@ -8,8 +8,10 @@ class DiveImportedModel : public QAbstractTableModel
 {
 	Q_OBJECT
 public:
-	DiveImportedModel(QObject *o);
-    void setDiveTable(struct dive_table *table);
+	enum roleTypes { DateTime = Qt::UserRole + 1, Duration, Depth};
+
+	DiveImportedModel(QObject *parent = 0);
+	void setDiveTable(struct dive_table *table);
 	int columnCount(const QModelIndex& index = QModelIndex()) const;
 	int rowCount(const QModelIndex& index = QModelIndex()) const;
 	QVariant data(const QModelIndex& index, int role) const;
@@ -17,7 +19,8 @@ public:
 	void setImportedDivesIndexes(int first, int last);
 	Qt::ItemFlags flags(const QModelIndex &index) const;
 	void clearTable();
-
+	QHash<int, QByteArray> roleNames() const;
+	Q_INVOKABLE void repopulate();
 public
 slots:
 	void changeSelected(QModelIndex clickedIndex);
@@ -28,7 +31,7 @@ private:
 	int firstIndex;
 	int lastIndex;
 	bool *checkStates;
-    struct dive_table *diveTable;
+	struct dive_table *diveTable;
 };
 
 #endif