1 /***************************************************************************
2 * Copyright (C) 2005-08 by the Quassel Project *
3 * devel@quassel-irc.org *
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. *
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. *
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 * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
19 ***************************************************************************/
25 #include <QStringList>
28 #include <QAbstractItemModel>
30 /*****************************************
31 * general item used in the Tree Model
32 *****************************************/
33 class AbstractTreeItem : public QObject {
35 Q_PROPERTY(quint64 id READ id)
38 AbstractTreeItem(AbstractTreeItem *parent = 0);
39 virtual ~AbstractTreeItem();
41 void appendChild(int column, AbstractTreeItem *child);
42 void appendChild(AbstractTreeItem *child);
44 void removeChild(int column, int row);
45 void removeChild(int row);
46 void removeAllChilds();
48 virtual quint64 id() const;
50 AbstractTreeItem *child(int column, int row) const;
51 AbstractTreeItem *child(int row) const;
53 AbstractTreeItem *childById(int column, const quint64 &id) const;
54 AbstractTreeItem *childById(const quint64 &id) const;
56 int childCount(int column) const;
57 int childCount() const;
59 virtual int columnCount() const = 0;
61 virtual QVariant data(int column, int role) const = 0;
63 virtual Qt::ItemFlags flags() const;
64 virtual void setFlags(Qt::ItemFlags);
68 AbstractTreeItem *parent() const;
71 void dataChanged(int column = -1);
72 void newChild(AbstractTreeItem *);
73 void childRemoved(int row);
74 void childsRemoved(int firstRow, int lastRow);
76 void beginRemoveChilds(int firstRow, int lastRow);
77 void endRemoveChilds();
80 void childDestroyed();
83 QHash<int, QList<AbstractTreeItem *> > _childItems;
84 QHash<int, QHash<quint64, AbstractTreeItem *> > _childHash; // uint to be compatible to qHash functions FIXME test this
87 int defaultColumn() const;
91 /*****************************************
93 *****************************************/
94 class SimpleTreeItem : public AbstractTreeItem {
98 SimpleTreeItem(const QList<QVariant> &data, AbstractTreeItem *parent = 0);
99 virtual ~SimpleTreeItem();
100 virtual QVariant data(int column, int role) const;
101 virtual int columnCount() const;
104 QList<QVariant> _itemData;
107 /*****************************************
109 *****************************************/
110 class PropertyMapItem : public AbstractTreeItem {
114 PropertyMapItem(const QStringList &propertyOrder, AbstractTreeItem *parent = 0);
115 PropertyMapItem(AbstractTreeItem *parent = 0);
117 virtual ~PropertyMapItem();
119 virtual QVariant data(int column, int role) const;
120 virtual int columnCount() const;
122 void appendProperty(const QString &property);
125 QStringList _propertyOrder;
129 /*****************************************
131 *****************************************/
132 class TreeModel : public QAbstractItemModel {
136 TreeModel(const QList<QVariant> &, QObject *parent = 0);
137 virtual ~TreeModel();
139 QVariant data(const QModelIndex &index, int role) const;
140 virtual Qt::ItemFlags flags(const QModelIndex &index) const;
141 QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const;
143 QModelIndex index(int row, int column, const QModelIndex &parent = QModelIndex()) const;
144 QModelIndex indexById(quint64 id, const QModelIndex &parent = QModelIndex()) const;
145 QModelIndex indexByItem(AbstractTreeItem *item) const;
147 QModelIndex parent(const QModelIndex &index) const;
149 int rowCount(const QModelIndex &parent = QModelIndex()) const;
150 int columnCount(const QModelIndex &parent = QModelIndex()) const;
152 virtual void clear();
155 void itemDataChanged(int column = -1);
156 void newChild(AbstractTreeItem *child);
158 void beginRemoveChilds(int firstRow, int lastRow);
159 void endRemoveChilds();
161 void childRemoved(int row);
162 void childsRemoved(int firstRow, int lastRow);
165 void appendChild(AbstractTreeItem *parent, AbstractTreeItem *child);
167 bool removeRow(int row, const QModelIndex &parent = QModelIndex());
168 bool removeRows(int row, int count, const QModelIndex &parent = QModelIndex());
170 AbstractTreeItem *rootItem;