X-Git-Url: https://git.quassel-irc.org/?p=quassel.git;a=blobdiff_plain;f=src%2Fclient%2Ftreemodel.h;h=6346d4f50850b9eecc785cdff467d0e3ca5a2304;hp=2b7928c53a3c3e1624c56c69ab08a03f26272060;hb=ef0e65458bf4e1cbc01399fe0e025a5b4fd327dc;hpb=44b22c4419f478a20f6324f9f3a700a2dec56302

diff --git a/src/client/treemodel.h b/src/client/treemodel.h
index 2b7928c5..6346d4f5 100644
--- 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,17 +41,23 @@ 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;
+  virtual void setFlags(Qt::ItemFlags);
   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;
+  Qt::ItemFlags _flags;
 };
 
 
@@ -67,16 +75,17 @@ 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;
 
-  void clear();
+  virtual void clear();
 
 protected:
   bool removeRow(int row, const QModelIndex &parent = QModelIndex());
   bool removeRows(int row, int count, const QModelIndex &parent = QModelIndex());
-
+  
   TreeItem *rootItem;
 };