1 files changed, 6 insertions, 3 deletions
diff --git a/qt-models/divelocationmodel.h b/qt-models/divelocationmodel.h
index 47164d2d4..8dbc8f3e6 100644
--- a/qt-models/divelocationmodel.h
+++ b/qt-models/divelocationmodel.h
@@ -8,18 +8,21 @@
 #include "core/units.h"
 #include "ssrfsortfilterproxymodel.h"
 
-class QLineEdit;
-
 #define RECENTLY_ADDED_DIVESITE 1
 
 bool filter_same_gps_cb (QAbstractItemModel *m, int sourceRow, const QModelIndex& parent);
 
+
 class LocationInformationModel : public QAbstractTableModel {
 Q_OBJECT
 public:
-	LocationInformationModel(QObject *obj = 0);
+	// Common columns, roles and accessor function for all dive-site models.
+	// Thus, different views can connect to different models.
 	enum Columns { UUID, NAME, LATITUDE, LONGITUDE, COORDS, DESCRIPTION, NOTES, TAXONOMY_1, TAXONOMY_2, TAXONOMY_3, COLUMNS};
 	enum Roles { UUID_ROLE = Qt::UserRole + 1 };
+	static QVariant getDiveSiteData(const struct dive_site *ds, int column, int role);
+
+	LocationInformationModel(QObject *obj = 0);
 	static LocationInformationModel *instance();
 	int columnCount(const QModelIndex &parent) const;
 	int rowCount(const QModelIndex &parent = QModelIndex()) const;