Fix recent list filtering

[quandoparte] / application / stationlistproxymodel.cpp

diff --git a/application/stationlistproxymodel.cpp b/application/stationlistproxymodel.cpp
index ab1dde3..295942e 100644 (file)
--- a/application/stationlistproxymodel.cpp
+++ b/application/stationlistproxymodel.cpp
@@ -10,11 +10,13 @@ Q_DECLARE_METATYPE(QGeoCoordinate)
 
 StationListProxyModel::StationListProxyModel(QObject *parent) :
     QSortFilterProxyModel(parent),
-    m_here(44.5, 9.0)
+    m_here(44.5, 9.0),
+    m_filterRecentOnly(false)
 {
 }
 
-bool StationListProxyModel::lessThan(const QModelIndex &left, const QModelIndex &right) const
+bool StationListProxyModel::lessThan(const QModelIndex &left,
+                                     const QModelIndex &right) const
 {
     int role = sortRole();
 
@@ -33,3 +35,23 @@ void StationListProxyModel::setUserPosition(const QtMobility::QGeoCoordinate &po
 {
     m_here = pos;
 }
+
+void StationListProxyModel::setRecentStations(const QStringList &stations)
+{
+    m_stations = stations;
+}
+
+bool StationListProxyModel::filterAcceptsRow(int sourceRow,
+                                             const QModelIndex &sourceParent) const
+{
+    if (m_filterRecentOnly) {
+        QModelIndex i = sourceModel()->index(sourceRow, 0, sourceParent);
+        return m_stations.contains(sourceModel()->data(i).toString());
+    }
+    return true;
+}
+
+void StationListProxyModel::setRecentOnlyFilter(bool activation)
+{
+    m_filterRecentOnly = activation;
+}