1 files changed, 128 insertions, 0 deletions
diff --git a/ui/qt/models/timeline_delegate.cpp b/ui/qt/models/timeline_delegate.cpp
new file mode 100644
index 0000000..07a7409
--- /dev/null
+++ b/ui/qt/models/timeline_delegate.cpp
@@ -0,0 +1,128 @@
+/* timeline_delegate.cpp
+ *
+ * Wireshark - Network traffic analyzer
+ * By Gerald Combs <gerald@wireshark.org>
+ * Copyright 1998 Gerald Combs
+ *
+ * SPDX-License-Identifier: GPL-2.0-or-later
+ */
+
+#include <ui/qt/models/timeline_delegate.h>
+#include <ui/qt/models/atap_data_model.h>
+
+#include <ui/qt/utils/color_utils.h>
+
+#include <QApplication>
+#include <QPainter>
+#include <QTreeView>
+#include <QAbstractProxyModel>
+
+// XXX We might want to move this to conversation_dialog.cpp.
+
+// PercentBarDelegate uses a stronger blend value, but its bars are also
+// more of a prominent feature. Make the blend weaker here so that we don't
+// obscure our text.
+static const double bar_blend_ = 0.08;
+
+TimelineDelegate::TimelineDelegate(QWidget *parent) :
+    QStyledItemDelegate(parent)
+{
+    _dataRole = Qt::UserRole;
+}
+
+void TimelineDelegate::setDataRole(int dataRole)
+{
+    _dataRole = dataRole;
+}
+
+void TimelineDelegate::paint(QPainter *painter, const QStyleOptionViewItem &option,
+                               const QModelIndex &index) const
+{
+    QStyleOptionViewItem option_vi = option;
+    QStyledItemDelegate::initStyleOption(&option_vi, index);
+
+    bool drawBar = false;
+    struct timeline_span span_px = index.data(_dataRole).value<struct timeline_span>();
+    if (_dataRole == ATapDataModel::TIMELINE_DATA) {
+        double span_s = span_px.maxRelTime - span_px.minRelTime;
+        QTreeView * tree = qobject_cast<QTreeView *>(parent());
+        if (tree) {
+            QAbstractProxyModel * proxy = qobject_cast<QAbstractProxyModel *>(tree->model());
+            if (proxy && proxy->sourceModel()) {
+                QModelIndex indexStart = proxy->mapFromSource(proxy->sourceModel()->index(0, span_px.colStart));
+                int colStart = -1;
+                int start_px = 0;
+                if (indexStart.isValid()) {
+                    colStart = indexStart.column();
+                    start_px = tree->columnWidth(colStart);
+                }
+                int colDuration = -1;
+                int column_px = start_px;
+                QModelIndex indexDuration = proxy->mapFromSource(proxy->sourceModel()->index(0, span_px.colDuration));
+                if (indexDuration.isValid()) {
+                    colDuration = indexDuration.column();
+                    column_px += tree->columnWidth(colDuration);
+                }
+
+                span_px.start = ((span_px.startTime - span_px.minRelTime) * column_px) / span_s;
+                span_px.width = ((span_px.stopTime - span_px.startTime) * column_px) / span_s;
+
+                if (index.column() == colStart) {
+                    drawBar = true;
+                } else if (index.column() == colDuration) {
+                    drawBar = true;
+                    span_px.start -= start_px;
+                }
+            }
+        }
+    }
+
+    if (!drawBar) {
+        QStyledItemDelegate::paint(painter, option, index);
+        return;
+    }
+
+    // Paint our rect with no text using the current style, then draw our
+    // bar and text over it.
+    option_vi.text = QString();
+    QStyle *style = option_vi.widget ? option_vi.widget->style() : QApplication::style();
+    style->drawControl(QStyle::CE_ItemViewItem, &option_vi, painter, option_vi.widget);
+
+    if (QApplication::style()->objectName().contains("vista")) {
+        // QWindowsVistaStyle::drawControl does this internally. Unfortunately there
+        // doesn't appear to be a more general way to do this.
+        option_vi.palette.setColor(QPalette::All, QPalette::HighlightedText,
+                               option_vi.palette.color(QPalette::Active, QPalette::Text));
+    }
+
+    QPalette::ColorGroup cg = option_vi.state & QStyle::State_Enabled
+                              ? QPalette::Normal : QPalette::Disabled;
+    QColor text_color = option_vi.palette.color(cg, QPalette::Text);
+    QColor bar_color = ColorUtils::alphaBlend(option_vi.palette.windowText(),
+                                              option_vi.palette.window(), bar_blend_);
+
+    if (cg == QPalette::Normal && !(option_vi.state & QStyle::State_Active))
+        cg = QPalette::Inactive;
+    if (option_vi.state & QStyle::State_Selected) {
+        text_color = option_vi.palette.color(cg, QPalette::HighlightedText);
+        bar_color = ColorUtils::alphaBlend(option_vi.palette.color(cg, QPalette::Window),
+                                           option_vi.palette.color(cg, QPalette::Highlight),
+                                           bar_blend_);
+    }
+
+    painter->save();
+    int border_radius = 3; // We use 3 px elsewhere, e.g. filter combos.
+    QRect timeline_rect = option.rect;
+    timeline_rect.adjust(span_px.start, 1, 0, -1);
+    timeline_rect.setWidth(span_px.width);
+    painter->setClipRect(option.rect);
+    painter->setPen(Qt::NoPen);
+    painter->setBrush(bar_color);
+    painter->drawRoundedRect(timeline_rect, border_radius, border_radius);
+    painter->restore();
+
+    painter->save();
+    painter->setPen(text_color);
+    painter->drawText(option.rect, Qt::AlignCenter, index.data(Qt::DisplayRole).toString());
+    painter->restore();
+}