proto_tree_model.h

Go to the documentation of this file.

 
#ifndef PROTO_TREE_MODEL_H
#define PROTO_TREE_MODEL_H
 
#include <ui/qt/utils/field_information.h>
#include <ui/qt/utils/proto_node.h>
 
#include <QAbstractItemModel>
#include <QModelIndex>

class ProtoTreeModel : public QAbstractItemModel
{
    Q_OBJECT
 
public:
    enum {
        ColoringRuleEntryRole = Qt::UserRole,
    };

    explicit ProtoTreeModel(QObject *parent = 0);

    ~ProtoTreeModel();

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

    QModelIndex index(int row, int, const QModelIndex &parent = QModelIndex()) const override;

    virtual QModelIndex parent(const QModelIndex &index) const override;

    virtual int rowCount(const QModelIndex &parent = QModelIndex()) const override;

    virtual int columnCount(const QModelIndex &) const  override{ return 1; }

    virtual QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;

    void setRootNode(proto_node *root_node);

    ProtoNode *protoNodeFromIndex(const QModelIndex &index) const;

    QModelIndex indexFromProtoNode(ProtoNode *index_node) const;

    QModelIndex findFirstHfid(int hf_id);

    QModelIndex findFieldInformation(FieldInformation *finfo);
 
private:
    ProtoNode *root_node_; 

    static bool foreachFindHfid(ProtoNode *node, void *find_hfid_ptr);

    static bool foreachFindField(ProtoNode *node, void *find_finfo_ptr);
};
 
#endif // PROTO_TREE_MODEL_H