Dive d/l selection UI: Add the setData method

Signed-off-by: Tomaz Canabrava <tomaz.canabrava@intel.com>
Signed-off-by: Dirk Hohndel <dirk@hohndel.org>

1 files changed, 15 insertions, 0 deletions

diff --git a/qt-ui/downloadfromdivecomputer.cpp b/qt-ui/downloadfromdivecomputer.cpp
index 91e38ae36..a7363c4ed 100644
--- a/qt-ui/downloadfromdivecomputer.cpp
+++ b/qt-ui/downloadfromdivecomputer.cpp
@@ -539,6 +539,21 @@ QVariant DiveImportedModel::data(const QModelIndex& index, int role) const
 	return QVariant();
 }
 
+bool DiveImportedModel::setData(const QModelIndex &index, const QVariant &value, int role)
+{
+	if (!index.isValid())
+		return false;
+
+	if (index.row() + firstIndex > lastIndex)
+		return false;
+
+	if (role != Qt::CheckStateRole)
+		return false;
+
+	checkStates[index.row() + firstIndex ] = value.toBool();
+	dataChanged(index, index, QVector<int>() << Qt::CheckStateRole);
+}
+
 void DiveImportedModel::setImportedDivesIndexes(int first, int last)
 {
 	beginRemoveRows(QModelIndex(), 0, lastIndex - firstIndex);