1 files changed, 130 insertions, 0 deletions
diff --git a/ui/qt/models/packet_list_model.h b/ui/qt/models/packet_list_model.h
new file mode 100644
index 0000000..807d884
--- /dev/null
+++ b/ui/qt/models/packet_list_model.h
@@ -0,0 +1,130 @@
+/** @file
+ *
+ * Wireshark - Network traffic analyzer
+ * By Gerald Combs <gerald@wireshark.org>
+ * Copyright 1998 Gerald Combs
+ *
+ * SPDX-License-Identifier: GPL-2.0-or-later
+ */
+
+#ifndef PACKET_LIST_MODEL_H
+#define PACKET_LIST_MODEL_H
+
+#include <config.h>
+
+#include <stdio.h>
+
+#include <glib.h>
+
+#include <epan/packet.h>
+
+#include <QAbstractItemModel>
+#include <QFont>
+#include <QVector>
+
+#include <ui/qt/progress_frame.h>
+
+#include "packet_list_record.h"
+
+#include "cfile.h"
+
+class QElapsedTimer;
+
+class PacketListModel : public QAbstractItemModel
+{
+    Q_OBJECT
+public:
+
+    enum {
+        HEADER_CAN_RESOLVE = Qt::UserRole,
+    };
+
+    explicit PacketListModel(QObject *parent = 0, capture_file *cf = NULL);
+    ~PacketListModel();
+    void setCaptureFile(capture_file *cf);
+    QModelIndex index(int row, int column,
+                      const QModelIndex & = QModelIndex()) const;
+    QModelIndex parent(const QModelIndex &) const;
+    int packetNumberToRow(int packet_num) const;
+    guint recreateVisibleRows();
+    void clear();
+
+    int rowCount(const QModelIndex &parent = QModelIndex()) const;
+    int columnCount(const QModelIndex & = QModelIndex()) const;
+    QVariant data(const QModelIndex &d_index, int role) const;
+    QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const;
+
+    gint appendPacket(frame_data *fdata);
+    frame_data *getRowFdata(QModelIndex idx);
+    frame_data *getRowFdata(int row);
+    void ensureRowColorized(int row);
+    int visibleIndexOf(frame_data *fdata) const;
+    /**
+     * @brief Invalidate any cached column strings.
+     */
+    void invalidateAllColumnStrings();
+    /**
+     * @brief Rebuild columns from settings.
+     */
+    void resetColumns();
+    void resetColorized();
+    void toggleFrameMark(const QModelIndexList &indeces);
+    void setDisplayedFrameMark(gboolean set);
+    void toggleFrameIgnore(const QModelIndexList &indeces);
+    void setDisplayedFrameIgnore(gboolean set);
+    void toggleFrameRefTime(const QModelIndex &rt_index);
+    void unsetAllFrameRefTime();
+    void addFrameComment(const QModelIndexList &indices, const QByteArray &comment);
+    void setFrameComment(const QModelIndex &index, const QByteArray &comment, guint c_number);
+    void deleteFrameComments(const QModelIndexList &indices);
+    void deleteAllFrameComments();
+
+    void setMaximumRowHeight(int height);
+
+signals:
+    void goToPacket(int);
+    void maxLineCountChanged(const QModelIndex &ih_index) const;
+    void itemHeightChanged(const QModelIndex &ih_index);
+
+    void bgColorizationProgress(int first, int last);
+
+public slots:
+    void sort(int column, Qt::SortOrder order = Qt::AscendingOrder);
+    void stopSorting();
+    void flushVisibleRows();
+    void dissectIdle(bool reset = false);
+
+private:
+    capture_file *cap_file_;
+    QList<QString> col_names_;
+    QVector<PacketListRecord *> physical_rows_;
+    QVector<PacketListRecord *> visible_rows_;
+    QVector<PacketListRecord *> new_visible_rows_;
+    QVector<int> number_to_row_;
+
+    int max_row_height_; // px
+    int max_line_count_;
+
+    static int sort_column_;
+    static int sort_column_is_numeric_;
+    static int text_sort_column_;
+    static Qt::SortOrder sort_order_;
+    static capture_file *sort_cap_file_;
+    static bool recordLessThan(PacketListRecord *r1, PacketListRecord *r2);
+    static double parseNumericColumn(const QString &val, bool *ok);
+
+    static gboolean stop_flag_;
+    static ProgressFrame *progress_frame_;
+    static double exp_comps_;
+    static double comps_;
+
+    QElapsedTimer *idle_dissection_timer_;
+    int idle_dissection_row_;
+
+    bool isNumericColumn(int column);
+
+private slots:
+    void emitItemHeightChanged(const QModelIndex &ih_index);
+};
+
+#endif // PACKET_LIST_MODEL_H