#ifndef TREEMODEL_H

 #define TREEMODEL_H

 #include <QAbstractItemModel>

 #include <QModelIndex>

 #include <QVariant>

 class TreeItem;

 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(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;

 private:

     void setupModelData(TreeItem *parent);

     TreeItem *rootItem;

 };

 #endif