From 1206069a8fb811d1c1090ced1b7017f20377a8ff Mon Sep 17 00:00:00 2001
From: wyzula-jan <jan.wyzula@psi.ch>
Date: Mon, 3 Mar 2025 21:51:53 +0100
Subject: [PATCH] fix(plot_base): update mouse mode state on mode change

---
 .../plots_next_gen/toolbar_bundles/mouse_interactions.py        | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/bec_widgets/widgets/plots_next_gen/toolbar_bundles/mouse_interactions.py b/bec_widgets/widgets/plots_next_gen/toolbar_bundles/mouse_interactions.py
index d9ff4f40..e09fa1d0 100644
--- a/bec_widgets/widgets/plots_next_gen/toolbar_bundles/mouse_interactions.py
+++ b/bec_widgets/widgets/plots_next_gen/toolbar_bundles/mouse_interactions.py
@@ -86,6 +86,7 @@ class MouseInteractionToolbarBundle(ToolbarBundle):
         self.actions["switch_mouse"].actions["drag_mode"].action.setChecked(not checked)
         if self.target_widget and checked:
             self.target_widget.plot_item.getViewBox().setMouseMode(pg.ViewBox.RectMode)
+            self.mouse_mode = "RectMode"
 
     @SafeSlot(bool)
     def enable_mouse_pan_mode(self, checked: bool):
@@ -98,6 +99,7 @@ class MouseInteractionToolbarBundle(ToolbarBundle):
         self.actions["switch_mouse"].actions["rectangle_mode"].action.setChecked(not checked)
         if self.target_widget and checked:
             self.target_widget.plot_item.getViewBox().setMouseMode(pg.ViewBox.PanMode)
+            self.mouse_mode = "PanMode"
 
     @SafeSlot()
     def autorange_plot(self):