From ed75ea1836a5500e0e83d8244e5c38a952ea7cf6 Mon Sep 17 00:00:00 2001
From: Willem Melching <willem.melching@gmail.com>
Date: Thu, 9 Mar 2023 22:14:06 +0100
Subject: [PATCH] cabana: scrub by shift dragging on graph (#27545)

* cabana: scrub by shift dragging on graph

* not while live streaming
old-commit-hash: b79f4594c42b4af0fc371b5d48445ce86c1cb3ff
---
 tools/cabana/chartswidget.cc | 26 ++++++++++++++++++++++++++
 tools/cabana/chartswidget.h  |  2 ++
 2 files changed, 28 insertions(+)

diff --git a/tools/cabana/chartswidget.cc b/tools/cabana/chartswidget.cc
index 54966be578..8ef2ae7b4f 100644
--- a/tools/cabana/chartswidget.cc
+++ b/tools/cabana/chartswidget.cc
@@ -668,6 +668,16 @@ void ChartView::mousePressEvent(QMouseEvent *event) {
     if (dropAction == Qt::MoveAction) {
       return;
     }
+  } else if (event->button() == Qt::LeftButton && QApplication::keyboardModifiers().testFlag(Qt::ShiftModifier)) {
+    if (!can->liveStreaming()) {
+      // Save current playback state when scrubbing
+      resume_after_scrub = !can->isPaused();
+      if (resume_after_scrub) {
+        can->pause(true);
+      }
+
+      is_scrubbing = true;
+    }
   } else {
     QChartView::mousePressEvent(event);
   }
@@ -701,9 +711,25 @@ void ChartView::mouseReleaseEvent(QMouseEvent *event) {
   } else {
     QGraphicsView::mouseReleaseEvent(event);
   }
+
+  // Resume playback if we were scrubbing
+  is_scrubbing = false;
+  if (resume_after_scrub) {
+    can->pause(false);
+    resume_after_scrub = false;
+  }
 }
 
 void ChartView::mouseMoveEvent(QMouseEvent *ev) {
+  // Scrubbing
+  if (is_scrubbing && QApplication::keyboardModifiers().testFlag(Qt::ShiftModifier)) {
+    double t = chart()->mapToValue({(double)ev->x(), (double)ev->y()}).x();
+    // Prevent seeking past the end of the route
+    t = std::clamp(t, 0., can->totalSeconds());
+    can->seekTo(t);
+    return;
+  }
+
   auto rubber = findChild<QRubberBand *>();
   bool is_zooming = rubber && rubber->isVisible();
   const auto plot_area = chart()->plotArea();
diff --git a/tools/cabana/chartswidget.h b/tools/cabana/chartswidget.h
index c86c19a04e..22ba020b29 100644
--- a/tools/cabana/chartswidget.h
+++ b/tools/cabana/chartswidget.h
@@ -97,6 +97,8 @@ private:
   double cur_sec = 0;
   const QString mime_type = "application/x-cabanachartview";
   SeriesType series_type = SeriesType::Line;
+  bool is_scrubbing = false;
+  bool resume_after_scrub = false;
   friend class ChartsWidget;
  };