1 files changed, 13 insertions, 1 deletions
diff --git a/qt-ui/filtermodels.cpp b/qt-ui/filtermodels.cpp
index be26a253b..aba480dd9 100644
--- a/qt-ui/filtermodels.cpp
+++ b/qt-ui/filtermodels.cpp
@@ -292,7 +292,7 @@ void LocationFilterModel::repopulate()
 	anyChecked = false;
 }
 
-MultiFilterSortModel::MultiFilterSortModel(QObject *parent) : QSortFilterProxyModel(parent), justCleared(false)
+MultiFilterSortModel::MultiFilterSortModel(QObject *parent) : QSortFilterProxyModel(parent), justCleared(false), curr_dive_site(NULL)
 {
 }
 
@@ -388,3 +388,15 @@ void MultiFilterSortModel::clearFilter()
 	justCleared = false;
 	myInvalidate();
 }
+
+void MultiFilterSortModel::startFilterDiveSite(int32_t uuid)
+{
+	curr_dive_site = get_dive_site_by_uuid(uuid);
+	myInvalidate();
+}
+
+void MultiFilterSortModel::stopFilterDiveSite()
+{
+	curr_dive_site = NULL;
+	myInvalidate();
+}