1/***************************************************************************
2 * Copyright (C) 2005-2014 by the Quassel Project *
3 * devel@quassel-irc.org *
4 * *
5 * This program is free software; you can redistribute it and/or modify *
6 * it under the terms of the GNU General Public License as published by *
7 * the Free Software Foundation; either version 2 of the License, or *
8 * (at your option) version 3. *
9 * *
10 * This program is distributed in the hope that it will be useful, *
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of *
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
13 * GNU General Public License for more details. *
14 * *
15 * You should have received a copy of the GNU General Public License *
16 * along with this program; if not, write to the *
17 * Free Software Foundation, Inc., *
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. *
19 ***************************************************************************/
20
21#ifndef FLATPROXYMODEL_H
22#define FLATPROXYMODEL_H
23
24#include <QAbstractProxyModel>
25
26class FlatProxyModel : public QAbstractProxyModel
27{
28 Q_OBJECT
29
30public:
31 FlatProxyModel(QObject *parent = 0);
32
33 virtual QModelIndex mapFromSource(const QModelIndex &sourceIndex) const;
34 virtual QModelIndex mapToSource(const QModelIndex &proxyIndex) const;
35
36 virtual QItemSelection mapSelectionFromSource(const QItemSelection &sourceSelection) const;
37 virtual QItemSelection mapSelectionToSource(const QItemSelection &proxySelection) const;
38
39 virtual void setSourceModel(QAbstractItemModel *sourceModel);
40
41 virtual QModelIndex index(int row, int column, const QModelIndex &parent) const;
42 virtual QModelIndex parent(const QModelIndex &index) const;
43
44 virtual int rowCount(const QModelIndex &index) const;
45 virtual int columnCount(const QModelIndex &index) const;
46
47public slots:
48 void linkTest() const;
49 void completenessTest() const;
50
51private slots:
52 void on_columnsAboutToBeInserted(const QModelIndex &parent, int start, int end);
53 void on_columnsAboutToBeRemoved(const QModelIndex &parent, int start, int end);
54 void on_columnsInserted(const QModelIndex &parent, int start, int end);
55 void on_columnsRemoved(const QModelIndex &parent, int start, int end);
56
57 void on_dataChanged(const QModelIndex &topLeft, const QModelIndex &bottomRight);
58// void on_headerDataChanged(Qt::Orientation orientation, int first, int last);
59
60 void on_layoutAboutToBeChanged();
61 void on_layoutChanged();
62
63 inline void on_modelAboutToBeReset() { beginResetModel(); endResetModel(); }
64 // void on_modelReset();
65
66 void on_rowsAboutToBeInserted(const QModelIndex &parent, int start, int end);
67 void on_rowsAboutToBeRemoved(const QModelIndex &parent, int start, int end);
68 void on_rowsInserted(const QModelIndex &parent, int start, int end);
69 void on_rowsRemoved(const QModelIndex &parent, int start, int end);
70
71private:
72 QList<int> _childCount;
73
74 class SourceItem;
75 SourceItem *_rootSourceItem;
76
77 void insertSubTree(const QModelIndex &source_idx, bool emitInsert = true);
78 SourceItem *insertSubTreeHelper(SourceItem *parentItem, SourceItem *lastItem_, const QModelIndex &source_idx);
79
80 void removeSubTree(const QModelIndex &source_idx, bool emitRemove = true);
81
82 SourceItem *sourceToInternal(const QModelIndex &sourceIndex) const;
83
84 void checkChildCount(const QModelIndex &index, const SourceItem *item, int &pos) const;
85
86 class _RangeRect
87 {
88public:
89 int left, right, top, bottom;
90 SourceItem *topItem, *bottomItem;
91 bool operator<(const _RangeRect &other) const;
92 };
93};
94
95
96class FlatProxyModel::SourceItem
97{
98public:
99 SourceItem(int row = 0, SourceItem *parent = 0);
100 ~SourceItem();
101
102 inline SourceItem *parent() const { return _parent; }
103 inline SourceItem *child(int i) const { return _childs[i]; }
104 inline int childCount() const { return _childs.count(); }
105
106 inline int pos() const { return _pos; }
107 inline SourceItem *next() const { return _next; }
108
109 int sourceRow() const;
110 SourceItem *findChild(int proxyPos) const;
111
112private:
113 inline void removeChild(SourceItem *item) { _childs.removeAt(_childs.indexOf(item)); }
114 inline void setPos(int i) { _pos = i; }
115 inline void setNext(SourceItem *next) { _next = next; }
116
117 SourceItem *_parent;
118 QList<SourceItem *> _childs;
119 int _pos;
120 SourceItem *_next;
121
122 friend class FlatProxyModel;
123};
124
125
126#endif //FLATPROXYMODEL_H
127