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
24using namespace Akonadi;
25
26namespace Akonadi
27{
28
29class SelectionProxyModelPrivate
30{
31public:
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
74SelectionProxyModel::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
82SelectionProxyModel::~SelectionProxyModel()
83{
84 delete d_ptr;
85}
86
87#include "moc_selectionproxymodel.cpp"
88