#ifndef TREEMODEL_H

 #define TREEMODEL_H

 #include <QAbstractItemModel>

 #include <QModelIndex>

 #include <QVariant>

 class BranchItem;

 class TreeItem;

 class LinkableMapObj;

 class TreeModel : public QAbstractItemModel

 {

     Q_OBJECT

 public:

     TreeModel(QObject *parent = 0);

     ~TreeModel();

     QVariant data(const QModelIndex &index, int role) const;

     Qt::ItemFlags flags(const QModelIndex &index) const;

     QVariant headerData(int section, Qt::Orientation orientation,

                         int role = Qt::DisplayRole) const;

 	QModelIndex index (TreeItem* ti);

     QModelIndex index(int row, int column,

                       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;

 	BranchItem* nextBranch (BranchItem* &current, BranchItem* &previous, bool deepLevelsFirst=true, BranchItem* start=NULL);

 	bool removeRows ( int row, int count, 

 				const QModelIndex & parent = QModelIndex() ); 

 	TreeItem* getItem (const QModelIndex &index) const;

 	BranchItem* getRootItem();

 protected:

     BranchItem *rootItem;

 };

 #endif