1 files changed, 134 insertions, 0 deletions
diff --git a/ui/qt/io_graph_action.cpp b/ui/qt/io_graph_action.cpp
new file mode 100644
index 00000000..777f87b1
--- /dev/null
+++ b/ui/qt/io_graph_action.cpp
@@ -0,0 +1,134 @@
+/* @file
+ *
+ * Wireshark - Network traffic analyzer
+ * By Gerald Combs <gerald@wireshark.org>
+ * Copyright 1998 Gerald Combs
+ *
+ * SPDX-License-Identifier: GPL-2.0-or-later
+ */
+
+#include "io_graph_action.h"
+
+#include <ui/qt/main_application.h>
+#include <ui/qt/main_window.h>
+#include <ui/qt/io_graph_dialog.h>
+#include <ui/qt/utils/field_information.h>
+
+#include <ui/io_graph_item.h>
+
+#include <wsutil/filesystem.h>
+
+#include <QMenu>
+
+IOGraphAction::IOGraphAction(QObject *parent, io_graph_item_unit_t unit, QString field) :
+    QAction(parent),
+    unit_(unit),
+    field_(field)
+{
+    setText(unitName(unit));
+    connect(this, &QAction::triggered, [&](){ emit openIOGraphDialog(unit_, field_); });
+}
+
+const QString IOGraphAction::unitName(io_graph_item_unit_t unit) {
+    switch (unit) {
+    case IOG_ITEM_UNIT_PACKETS:
+        if (is_packet_configuration_namespace()) {
+            return QObject::tr("PACKETS");
+        }
+        return QObject::tr("EVENTS");
+    case IOG_ITEM_UNIT_BYTES:
+        return QObject::tr("BYTES");
+    case IOG_ITEM_UNIT_BITS:
+        return QObject::tr("BITS");
+    case IOG_ITEM_UNIT_CALC_FRAMES:
+        return QObject::tr("COUNT FRAMES");
+    case IOG_ITEM_UNIT_CALC_FIELDS:
+        return QObject::tr("COUNT FIELDS");
+    case IOG_ITEM_UNIT_CALC_SUM:
+        return QObject::tr("SUM");
+    case IOG_ITEM_UNIT_CALC_MAX:
+        return QObject::tr("MAX");
+    case IOG_ITEM_UNIT_CALC_MIN:
+        return QObject::tr("MIN");
+    case IOG_ITEM_UNIT_CALC_AVERAGE:
+        return QObject::tr("AVERAGE");
+    case IOG_ITEM_UNIT_CALC_THROUGHPUT:
+        return QObject::tr("THROUGHPUT");
+    case IOG_ITEM_UNIT_CALC_LOAD:
+        return QObject::tr("LOAD");
+    default:
+        return QObject::tr("UNKNOWN");
+    }
+}
+
+QList<io_graph_item_unit_t> IOGraphAction::unitTypes(const FieldInformation::HeaderInfo& headerinfo)
+{
+    static const QList<io_graph_item_unit_t> simple_types_ = QList<io_graph_item_unit_t>()
+        << IOG_ITEM_UNIT_CALC_FRAMES
+        << IOG_ITEM_UNIT_CALC_FIELDS;
+
+    static const QList<io_graph_item_unit_t> number_types_ = QList<io_graph_item_unit_t>()
+        << IOG_ITEM_UNIT_CALC_SUM
+        << IOG_ITEM_UNIT_CALC_FRAMES
+        << IOG_ITEM_UNIT_CALC_FIELDS
+        << IOG_ITEM_UNIT_CALC_MAX
+        << IOG_ITEM_UNIT_CALC_MIN
+        << IOG_ITEM_UNIT_CALC_THROUGHPUT
+        << IOG_ITEM_UNIT_CALC_AVERAGE;
+
+    static const QList<io_graph_item_unit_t> time_types_ = QList<io_graph_item_unit_t>(number_types_)
+        << IOG_ITEM_UNIT_CALC_LOAD;
+
+    switch (headerinfo.type) {
+    case FT_UINT8:
+    case FT_UINT16:
+    case FT_UINT24:
+    case FT_UINT32:
+    case FT_UINT64:
+    case FT_INT8:
+    case FT_INT16:
+    case FT_INT24:
+    case FT_INT32:
+    case FT_INT64:
+    case FT_FLOAT:
+    case FT_DOUBLE:
+        return number_types_;
+    case FT_RELATIVE_TIME:
+        return time_types_;
+    default:
+        return simple_types_;
+    }
+}
+
+QMenu * IOGraphAction::createMenu(const FieldInformation::HeaderInfo& headerinfo, QWidget * parent)
+{
+    MainWindow *mw(nullptr);
+    if (mainApp)
+    {
+        QWidget * mainWin = mainApp->mainWindow();
+        if (qobject_cast<MainWindow *>(mainWin)) {
+            mw = qobject_cast<MainWindow *>(mainWin);
+        }
+    }
+
+    QString title("I/O Graph");
+    QMenu * submenu = new QMenu(title, parent);
+
+    int one_em = submenu->fontMetrics().height();
+    QString prep_text = QString("%1: %2").arg(title).arg(headerinfo.abbreviation);
+    prep_text = submenu->fontMetrics().elidedText(prep_text, Qt::ElideRight, one_em * 40);
+    QAction * comment = submenu->addAction(prep_text);
+    comment->setEnabled(false);
+    submenu->addSeparator();
+
+    IOGraphAction *graphAction;
+    for (const auto &unit : IOGraphAction::unitTypes(headerinfo)) {
+        graphAction = new IOGraphAction(submenu, unit, headerinfo.abbreviation);
+        if (mw) {
+                connect(graphAction, &IOGraphAction::openIOGraphDialog, mw, &MainWindow::showIOGraphDialog);
+        }
+        submenu->addAction(graphAction);
+    }
+
+    return submenu;
+}