1 | /* |
2 | Copyright (c) 2009 Stephen Kelly <steveire@gmail.com> |
3 | |
4 | This library is free software; you can redistribute it and/or modify it |
5 | under the terms of the GNU Library General Public License as published by |
6 | the Free Software Foundation; either version 2 of the License, or (at your |
7 | option) any later version. |
8 | |
9 | This library is distributed in the hope that it will be useful, but WITHOUT |
10 | ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
11 | FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public |
12 | License for more details. |
13 | |
14 | You should have received a copy of the GNU Library General Public License |
15 | along with this library; see the file COPYING.LIB. If not, write to the |
16 | Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA |
17 | 02110-1301, USA. |
18 | */ |
19 | |
20 | #include "selectionproxymodel.h" |
21 | |
22 | #include "entitytreemodel.h" |
23 | |
24 | using namespace Akonadi; |
25 | |
26 | namespace Akonadi |
27 | { |
28 | |
29 | class SelectionProxyModelPrivate |
30 | { |
31 | public: |
32 | SelectionProxyModelPrivate(SelectionProxyModel *selectionProxyModel) |
33 | : q_ptr(selectionProxyModel) |
34 | { |
35 | Q_Q(SelectionProxyModel); |
36 | foreach (const QModelIndex &rootIndex, q->sourceRootIndexes()) { |
37 | rootIndexAdded(rootIndex); |
38 | } |
39 | } |
40 | ~SelectionProxyModelPrivate() |
41 | { |
42 | Q_Q(SelectionProxyModel); |
43 | foreach (const QModelIndex &idx, q->sourceRootIndexes()) { |
44 | rootIndexAboutToBeRemoved(idx); |
45 | } |
46 | } |
47 | |
48 | /** |
49 | Increases the refcount of the Collection in @p newRootIndex |
50 | */ |
51 | void rootIndexAdded(const QModelIndex &newRootIndex) |
52 | { |
53 | Q_Q(SelectionProxyModel); |
54 | // newRootIndex is already in the sourceModel. |
55 | q->sourceModel()->setData(newRootIndex, QVariant(), EntityTreeModel::CollectionRefRole); |
56 | q->sourceModel()->fetchMore(newRootIndex); |
57 | } |
58 | |
59 | /** |
60 | Decreases the refcount of the Collection in @p removedRootIndex |
61 | */ |
62 | void rootIndexAboutToBeRemoved(const QModelIndex &removedRootIndex) |
63 | { |
64 | Q_Q(SelectionProxyModel); |
65 | q->sourceModel()->setData(removedRootIndex, QVariant(), EntityTreeModel::CollectionDerefRole); |
66 | } |
67 | |
68 | Q_DECLARE_PUBLIC(SelectionProxyModel) |
69 | SelectionProxyModel *q_ptr; |
70 | }; |
71 | |
72 | } |
73 | |
74 | SelectionProxyModel::SelectionProxyModel(QItemSelectionModel *selectionModel, QObject *parent) |
75 | : KSelectionProxyModel(selectionModel, parent) |
76 | , d_ptr(new SelectionProxyModelPrivate(this)) |
77 | { |
78 | connect(this, SIGNAL(rootIndexAdded(QModelIndex)), SLOT(rootIndexAdded(QModelIndex))); |
79 | connect(this, SIGNAL(rootIndexAboutToBeRemoved(QModelIndex)), SLOT(rootIndexAboutToBeRemoved(QModelIndex))); |
80 | } |
81 | |
82 | SelectionProxyModel::~SelectionProxyModel() |
83 | { |
84 | delete d_ptr; |
85 | } |
86 | |
87 | #include "moc_selectionproxymodel.cpp" |
88 | |