1 files changed, 42 insertions, 0 deletions
diff --git a/qt-models/divelocationmodel.cpp b/qt-models/divelocationmodel.cpp
new file mode 100644
index 000000000..a5303da49
--- /dev/null
+++ b/qt-models/divelocationmodel.cpp
@@ -0,0 +1,42 @@
+#include "divelocationmodel.h"
+#include "dive.h"
+
+LocationInformationModel::LocationInformationModel(QObject *obj) : QAbstractListModel(obj), internalRowCount(0)
+{
+}
+
+int LocationInformationModel::rowCount(const QModelIndex &parent) const
+{
+	Q_UNUSED(parent);
+	return internalRowCount;
+}
+
+QVariant LocationInformationModel::data(const QModelIndex &index, int role) const
+{
+	if (!index.isValid())
+		return QVariant();
+	struct dive_site *ds = get_dive_site(index.row());
+
+	switch(role) {
+		case Qt::DisplayRole : return qPrintable(ds->name);
+	}
+
+	return QVariant();
+}
+
+void LocationInformationModel::update()
+{
+	int i;
+	struct dive_site *ds;
+	for_each_dive_site (i, ds);
+
+	if (rowCount()) {
+		beginRemoveRows(QModelIndex(), 0, rowCount()-1);
+		endRemoveRows();
+	}
+	if (i) {
+		beginInsertRows(QModelIndex(), 0, i);
+		internalRowCount = i;
+		endInsertRows();
+	}
+}