1 files changed, 69 insertions, 0 deletions
diff --git a/qt-models/completionmodels.cpp b/qt-models/completionmodels.cpp
new file mode 100644
index 000000000..f2e70afd1
--- /dev/null
+++ b/qt-models/completionmodels.cpp
@@ -0,0 +1,69 @@
+#include "completionmodels.h"
+#include "dive.h"
+#include "mainwindow.h"
+
+#define CREATE_UPDATE_METHOD(Class, diveStructMember)          \
+	void Class::updateModel()                              \
+	{                                                      \
+		QStringList list;                              \
+		struct dive *dive;                             \
+		int i = 0;                                     \
+		for_each_dive (i, dive)                        \
+		{                                              \
+			QString buddy(dive->diveStructMember); \
+			if (!list.contains(buddy)) {           \
+				list.append(buddy);            \
+			}                                      \
+		}                                              \
+		std::sort(list.begin(), list.end());           \
+		setStringList(list);                           \
+	}
+
+#define CREATE_CSV_UPDATE_METHOD(Class, diveStructMember)                                        \
+	void Class::updateModel()                                                                \
+	{                                                                                        \
+		QSet<QString> set;                                                               \
+		struct dive *dive;                                                               \
+		int i = 0;                                                                       \
+		for_each_dive (i, dive)                                                          \
+		{                                                                                \
+			QString buddy(dive->diveStructMember);                                   \
+			foreach (const QString &value, buddy.split(",", QString::SkipEmptyParts)) \
+			{                                                                        \
+				set.insert(value.trimmed());                                     \
+			}                                                                        \
+		}                                                                                \
+		QStringList setList = set.toList();                                              \
+		std::sort(setList.begin(), setList.end());                                       \
+		setStringList(setList);                                                     \
+	}
+
+CREATE_CSV_UPDATE_METHOD(BuddyCompletionModel, buddy);
+CREATE_CSV_UPDATE_METHOD(DiveMasterCompletionModel, divemaster);
+CREATE_UPDATE_METHOD(SuitCompletionModel, suit);
+
+void LocationCompletionModel::updateModel()
+{
+	QStringList list;
+	struct dive_site *ds;
+	int i = 0;
+	for_each_dive_site(i, ds) {
+		if (!list.contains(ds->name))
+			list.append(ds->name);
+	}
+	std::sort(list.begin(), list.end());
+	setStringList(list);
+}
+
+void TagCompletionModel::updateModel()
+{
+	if (g_tag_list == NULL)
+		return;
+	QStringList list;
+	struct tag_entry *current_tag_entry = g_tag_list->next;
+	while (current_tag_entry != NULL) {
+		list.append(QString(current_tag_entry->tag->name));
+		current_tag_entry = current_tag_entry->next;
+	}
+	setStringList(list);
+}