OSDN Git Service

a99777c909f2cea73e86aa261360680464ae8b6a
[neighbornote/NeighborNote.git] / src / cx / fbn / nevernote / filters / NoteSortFilterProxyModel.java
1 /*\r
2  * This file is part of NixNote \r
3  * Copyright 2009 Randy Baumgarte\r
4  * \r
5  * This file may be licensed under the terms of of the\r
6  * GNU General Public License Version 2 (the ``GPL'').\r
7  *\r
8  * Software distributed under the License is distributed\r
9  * on an ``AS IS'' basis, WITHOUT WARRANTY OF ANY KIND, either\r
10  * express or implied. See the GPL for the specific language\r
11  * governing rights and limitations.\r
12  *\r
13  * You should have received a copy of the GPL along with this\r
14  * program. If not, go to http://www.gnu.org/licenses/gpl.html\r
15  * or write to the Free Software Foundation, Inc.,\r
16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.\r
17  *\r
18 */\r
19 \r
20 package cx.fbn.nevernote.filters;\r
21 \r
22 import java.util.HashMap;\r
23 \r
24 import com.trolltech.qt.core.QAbstractItemModel;\r
25 import com.trolltech.qt.core.QModelIndex;\r
26 import com.trolltech.qt.core.QObject;\r
27 import com.trolltech.qt.core.Qt;\r
28 import com.trolltech.qt.gui.QIcon;\r
29 import com.trolltech.qt.gui.QImage;\r
30 import com.trolltech.qt.gui.QSortFilterProxyModel;\r
31 \r
32 import cx.fbn.nevernote.Global;\r
33 import cx.fbn.nevernote.evernote.NoteMetadata;\r
34 \r
35 public class NoteSortFilterProxyModel extends QSortFilterProxyModel {\r
36         private final HashMap<String, NoteMetadata> guids;\r
37         private final HashMap<String, NoteMetadata> pinnedGuids;\r
38         public Signal2<Integer,Integer> sortChanged;\r
39         public boolean blocked;\r
40         \r
41         public NoteSortFilterProxyModel(QObject parent) {\r
42                 super(parent);\r
43                 guids = new HashMap<String, NoteMetadata>();\r
44                 pinnedGuids = new HashMap<String, NoteMetadata>();\r
45                 setDynamicSortFilter(true);\r
46                 sortChanged = new Signal2<Integer,Integer>();\r
47         }\r
48         public void clear() {\r
49                 guids.clear();\r
50         }\r
51         public void addGuid(String guid, NoteMetadata meta) {\r
52                 if (!guids.containsKey(guid))\r
53                         guids.put(guid, meta);\r
54                 if (meta.isPinned() == true && !pinnedGuids.containsKey(guid))\r
55                         pinnedGuids.put(guid, meta);\r
56         }\r
57         public void filter() {\r
58                 invalidateFilter();\r
59         }\r
60         @Override\r
61         protected boolean filterAcceptsRow(int sourceRow, QModelIndex sourceParent) {\r
62                 QAbstractItemModel model = sourceModel();\r
63                 QModelIndex guidIndex = sourceModel().index(sourceRow, Global.noteTableGuidPosition);\r
64                 String guid = (String)model.data(guidIndex);\r
65                 \r
66                 if (guids.containsKey(guid) || pinnedGuids.containsKey(guid))\r
67                         return true;\r
68                 else\r
69                         return false;\r
70         }\r
71         \r
72         \r
73         @Override\r
74         public void sort(int col, Qt.SortOrder order) {\r
75                 if (col != Global.noteTableThumbnailPosition) {\r
76                         if (!blocked)   {\r
77                                 sortChanged.emit(col, order.value());    // Signal that the sort order has been modified\r
78                         }\r
79                         super.sort(col,order);\r
80                 }\r
81         }\r
82         \r
83         @Override\r
84         protected boolean lessThan(QModelIndex left, QModelIndex right) {\r
85                 Object leftData = sourceModel().data(left);\r
86                 Object rightData = sourceModel().data(right);\r
87                 \r
88                 if (rightData == null)\r
89                         return true;\r
90                 if (leftData instanceof QIcon)\r
91                         return true;\r
92                 if (leftData instanceof QImage && rightData instanceof QImage)\r
93                         return true;\r
94                 if (leftData instanceof Long && rightData instanceof Long) {\r
95                           Long leftLong = (Long)leftData;\r
96                           Long rightLong = (Long)rightData;\r
97                           return leftLong.compareTo(rightLong) < 0;            \r
98                 }\r
99                 if (leftData instanceof String && rightData instanceof String) {\r
100                         String leftString = (String)leftData;\r
101                         String rightString = (String)rightData;\r
102                         return leftString.compareTo(rightString) < 0;\r
103                 }\r
104                 \r
105                 return super.lessThan(left, right);\r
106         }\r
107 }