commit 4890c205ed0a70486c5dc1206e157505d8f0504d
author Maxwell Henderson <mxwhenderson@gmail.com> Tue Mar 21 12:09:04 2023 -0700
committer Maxwell Henderson <mxwhenderson@gmail.com> Tue Mar 21 18:11:33 2023 -0700
tree a0998b70ab386a27a32f48f6cc0de326631f40c1
parent effa797531d3916a1df5049b915815674ddd8e24

Add option to stop editing control points

Signed-off-by: Maxwell Henderson <mxwhenderson@gmail.com>
Change-Id: I5e26ff2ae14a607b8f511fb626752dd605590eb3

y2023/control_loops/python/graph_edit.py

diff --git a/y2023/control_loops/python/graph_edit.py b/y2023/control_loops/python/graph_edit.py
index e47128a..37e951f 100644
--- a/y2023/control_loops/python/graph_edit.py
+++ b/y2023/control_loops/python/graph_edit.py
@@ -289,6 +289,8 @@
         # Lets you only view selected path
         self.view_current = False
 
+        self.editing = True
+
     def _do_button_press_internal(self, event):
         o_x = event.x
         o_y = event.y
@@ -573,6 +575,9 @@
             print("Switched to segment:", self.segments[self.index].name)
             self.segments[self.index].Print(graph_paths.points)
 
+        elif keyval == Gdk.KEY_d:
+            self.editing = not self.editing
+
         elif keyval == Gdk.KEY_l:
             self.view_current = not self.view_current
 
@@ -608,6 +613,7 @@
         self.redraw()
 
     def do_button_press(self, event):
+
         last_pos = self.last_pos
         self.last_pos = (event.x, event.y)
         pt_theta = self.cur_pt_in_theta()
@@ -617,10 +623,11 @@
 
         self.now_segment_pt = np.array(shift_angles(pt_theta))
 
-        if self.edit_control1:
-            self.segments[self.index].control1 = self.now_segment_pt
-        else:
-            self.segments[self.index].control2 = self.now_segment_pt
+        if self.editing:
+            if self.edit_control1:
+                self.segments[self.index].control1 = self.now_segment_pt
+            else:
+                self.segments[self.index].control2 = self.now_segment_pt
 
         print('Clicked at theta: np.array([%s, %s])' %
               (self.now_segment_pt[0], self.now_segment_pt[1]))