2 files changed, 59 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();
+	}
+}
diff --git a/qt-models/divelocationmodel.h b/qt-models/divelocationmodel.h
new file mode 100644
index 000000000..1bf7cedfd
--- /dev/null
+++ b/qt-models/divelocationmodel.h
@@ -0,0 +1,17 @@
+#ifndef DIVELOCATIONMODEL_H
+#define DIVELOCATIONMODEL_H
+
+#include <QAbstractListModel>
+
+class LocationInformationModel : public QAbstractListModel {
+Q_OBJECT
+public:
+	LocationInformationModel(QObject *obj = 0);
+	int rowCount(const QModelIndex &parent = QModelIndex()) const;
+	QVariant data(const QModelIndex &index = QModelIndex(), int role = Qt::DisplayRole) const;
+	void update();
+private:
+	int internalRowCount;
+};
+
+#endif