Started to reorganize the Buffer{Model|View|Filter}. Mostly cleanup at the moment.

[quassel.git] / src / client / treemodel.h

diff --git a/src/client/treemodel.h b/src/client/treemodel.h
index 1941602..faad792 100644 (file)
--- a/src/client/treemodel.h
+++ b/src/client/treemodel.h
@@ -23,6 +23,7 @@
 
 #include <QList>
 #include <QVariant>
+#include <QHash>
 #include <QAbstractItemModel>
 
 /*****************************************
@@ -30,6 +31,7 @@
  *****************************************/
 class TreeItem : public QObject {
   Q_OBJECT
+  Q_PROPERTY(uint id READ id)
   
 public:
   TreeItem(const QList<QVariant> &data, TreeItem *parent = 0);
@@ -39,15 +41,19 @@ public:
   void appendChild(TreeItem *child);
   void removeChild(int row);
                    
-  TreeItem *child(int row);
+  virtual uint id() const;
+  TreeItem *child(int row) const;
+  TreeItem *childById(const uint &) const;
   int childCount() const;
   int columnCount() const;
   virtual QVariant data(int column, int role) const;
+  virtual Qt::ItemFlags flags() const;
   int row() const;
   TreeItem *parent();
     
 protected:
-  QList<TreeItem*> childItems;
+  QList<TreeItem *> childItems;
+  QHash<uint, TreeItem *> childHash; // uint to be compatible to qHash functions
   TreeItem *parentItem;
   QList<QVariant> itemData;
 };
@@ -67,12 +73,16 @@ public:
   virtual Qt::ItemFlags flags(const QModelIndex &index) const;
   QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const;
   QModelIndex index(int row, int column, const QModelIndex &parent = QModelIndex()) const;
+  QModelIndex indexById(uint id, const QModelIndex &parent = QModelIndex()) const;
   QModelIndex parent(const QModelIndex &index) const;
   int rowCount(const QModelIndex &parent = QModelIndex()) const;
   int columnCount(const QModelIndex &parent = QModelIndex()) const;
 
+  virtual void clear();
+
 protected:
   bool removeRow(int row, const QModelIndex &parent = QModelIndex());
+  bool removeRows(int row, int count, const QModelIndex &parent = QModelIndex());
   
   TreeItem *rootItem;
 };