1 files changed, 48 insertions, 0 deletions
diff --git a/ui/qt/models/proto_tree_model.h b/ui/qt/models/proto_tree_model.h
new file mode 100644
index 00000000..df7cbba8
--- /dev/null
+++ b/ui/qt/models/proto_tree_model.h
@@ -0,0 +1,48 @@
+/** @file
+ *
+ * Wireshark - Network traffic analyzer
+ * By Gerald Combs <gerald@wireshark.org>
+ * Copyright 1998 Gerald Combs
+ *
+ * SPDX-License-Identifier: GPL-2.0-or-later
+ */
+
+#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, gpointer find_hfid_ptr);
+    static bool foreachFindField(ProtoNode *node, gpointer find_finfo_ptr);
+};
+
+#endif // PROTO_TREE_MODEL_H