commit 017812088ddaaf3bbf0fb8831f2d0be52bda019d
author Ravago Jones <ravagojones@gmail.com> Sun Aug 01 15:25:11 2021 -0700
committer Ravago Jones <ravagojones@gmail.com> Sat Aug 14 16:33:37 2021 -0700
tree 34e5f442fdebd6148546999980406b3f9b33059a
parent 797c49cac9f8426e1c18ee5ee15537bf807f9239

Add zoom to the Spline UI

Signed-off-by: Ravago Jones <ravagojones@gmail.com>
Change-Id: Id20a21c6c6caf041bef5ce518c1ddcbe75216d66

frc971/control_loops/python/path_edit.py

diff --git a/frc971/control_loops/python/path_edit.py b/frc971/control_loops/python/path_edit.py
index 6f54119..959b382 100755
--- a/frc971/control_loops/python/path_edit.py
+++ b/frc971/control_loops/python/path_edit.py
@@ -413,3 +413,42 @@
 
                 self.index_of_edit = -1
                 self.spline_edit = -1
+
+    def mouse_scroll(self, event):
+        self.mousex, self.mousey = self.input_transform.transform_point(
+            event.x, event.y)
+
+        step_size = 20  # px
+
+        if event.direction == Gdk.ScrollDirection.UP:
+            # zoom out
+            scale_by = step_size
+        elif event.direction == Gdk.ScrollDirection.DOWN:
+            # zoom in
+            scale_by = -step_size
+        else:
+            return
+
+        apparent_width, apparent_height = self.transform.transform_distance(
+            self.mToPx(FIELD.width), self.mToPx(FIELD.length))
+        scale = (apparent_width + scale_by) / apparent_width
+
+        # scale from point in field coordinates
+        point = self.mousex, self.mousey
+
+        # move the origin to point
+        self.transform.translate(point[0], point[1])
+
+        # scale from new origin
+        self.transform.scale(scale, scale)
+
+        # move back
+        self.transform.translate(-point[0], -point[1])
+
+        # snap to the edge when near 1x scaling
+        if 0.99 < self.transform.xx < 1.01 and -50 < self.transform.x0 < 50:
+            self.transform.x0 = 0
+            self.transform.y0 = 0
+            print("snap")
+
+        self.queue_draw()

frc971/control_loops/python/spline_graph.py

diff --git a/frc971/control_loops/python/spline_graph.py b/frc971/control_loops/python/spline_graph.py
index f56b0d3..a8e5c78 100755
--- a/frc971/control_loops/python/spline_graph.py
+++ b/frc971/control_loops/python/spline_graph.py
@@ -80,6 +80,7 @@
         self.method_connect("button-release-event", self.field.button_release)
         self.method_connect("configure-event", self.configure)
         self.method_connect("motion_notify_event", self.field.mouse_move)
+        self.method_connect("scroll_event", self.field.mouse_scroll)
 
         self.file_name_box = Gtk.Entry()
         self.file_name_box.set_size_request(200, 40)