treemodel.h

Go to the documentation of this file.

#ifndef TREEMODEL_H
#define TREEMODEL_H

#include <QAbstractItemModel>

#include "database.h"

class TreeItem;
class TreePath;
class QSqlTableModel;

class ProjectTreeModel : public QAbstractItemModel, public SqlHelper {
    Q_OBJECT
public:
    enum TreeModelRoles {
        TreeItemPointer = Qt::UserRole + 1100,
        TreeId,
        TreeItemPath
    };

    ProjectTreeModel(QObject *parent = 0);
    ~ProjectTreeModel();

    QVariant data(const QModelIndex &index, int role) const;
    QVariant headerData(int section, Qt::Orientation orientation,
                        int role = Qt::DisplayRole) const;
    TreeItem *treeItemPointer(const QModelIndex &index) const;
    TreeItem *treeItemPointer(const int tree_id) const;

    QModelIndex index(int row, int column,
                      const QModelIndex &parent = QModelIndex()) const;
    QModelIndex index(const TreePath &path);

    QModelIndex parent(const QModelIndex &index) const;
    int rowCount(const QModelIndex &parent = QModelIndex()) const;
    int columnCount(const QModelIndex &parent = QModelIndex()) const;

    Qt::ItemFlags flags(const QModelIndex &index) const;
    bool setData(const QModelIndex &index, const QVariant &value,
                 int role = Qt::EditRole);

    QMimeData *mimeData(const QModelIndexList &indexes) const;
    QStringList mimeTypes() const;
    Qt::DropActions supportedDropActions() const;

    bool insertRows(int position, int rows, const QModelIndex &parent = QModelIndex());
    bool removeRows(int position, int rows, const QModelIndex &parent = QModelIndex());

    void reLoad();

    virtual QModelIndex match(const QModelIndex &start, const TreePath &path) const ;

private slots:
    virtual void onTableChange(const QString &tableName);

private:
    void setupModelData(TreeItem *parent);

    QString formName(int row, int column, QSqlTableModel &table, const QString &suffix = QString());
    bool columnIsPrimary(int column, QSqlTableModel &table);
    bool columnIsReserved(int column, QSqlTableModel &table);

private:
    TreeItem *rootItem;
};

#endif // TREEMODEL_H