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:
    explicit ProtoTreeModel(QObject * parent = 0);
    ~ProtoTreeModel();
 
    virtual Qt::ItemFlags flags(const QModelIndex &index) const;
    QModelIndex index(int row, int, const QModelIndex &parent = QModelIndex()) const;
    virtual QModelIndex parent(const QModelIndex &index) const;
    virtual int rowCount(const QModelIndex &parent = QModelIndex()) const;
    virtual int columnCount(const QModelIndex &) const { return 1; }
    virtual QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const;
 
    // root_node can be NULL.
    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