commit 295f730288ed492e2be14409683d1bbb10cd9318
author Nathan Leong <100028864@mvla.net> Sat Jul 23 16:09:31 2022 -0700
committer Nathan Leong <100028864@mvla.net> Sat Oct 08 14:49:39 2022 -0700
tree dabd56fddf1f34aaf4f017abf84a5febdce021d1
parent 87cadb7a4ba1a770439c4029e9fa77ff0619047b

Add a new spline button to the Spline UI

Change-Id: I43f96aa5e90127964bcb81b7df5a5e90009a3683
Signed-off-by: Nathan Leong <100028864@mvla.net>

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 b66a668..7c742c8 100755
--- a/frc971/control_loops/python/path_edit.py
+++ b/frc971/control_loops/python/path_edit.py
@@ -375,16 +375,24 @@
         keyval = Gdk.keyval_to_lower(event.keyval)
         if keyval == Gdk.KEY_z and event.state & Gdk.ModifierType.CONTROL_MASK:
             self.undo()
-        # TODO: This should be a button
+
         if keyval == Gdk.KEY_p:
-            self.mode = Mode.kPlacing
-            # F0 = A1
-            # B1 = 2F0 - E0
-            # C1= d0 + 4F0 - 4E0
-            multispline = self.active_multispline
-            multispline.extrapolate()
-            self.queue_draw()
+            self.new_spline()
         elif keyval == Gdk.KEY_m:
+            self.new_multispline()
+
+    def new_spline(self):
+        self.mode = Mode.kPlacing
+        # F0 = A1
+        # B1 = 2F0 - E0
+        # C1= d0 + 4F0 - 4E0
+        multispline = self.active_multispline
+        if len(multispline.getSplines()) != 0:
+            multispline.extrapolate(multispline.getSplines()[-1])
+        self.queue_draw()
+
+    def new_multispline(self):
+        if len(self.active_multispline.getSplines()) != 0:
             self.mode = Mode.kPlacing
             self.active_multispline_index += 1
             self.multisplines.insert(self.active_multispline_index,
@@ -392,7 +400,7 @@
 
             prev_multispline = self.multisplines[self.active_multispline_index
                                                  - 1]
-            if prev_multispline:
+            if len(prev_multispline.getSplines()) != 0:
                 self.active_multispline.extrapolate(
                     prev_multispline.getSplines()[-1])
             self.queue_draw()