commit 291f559ecc46f9f2b08d8efe9e5242b169aebad1
author Ravago Jones <ravagojones@gmail.com> Thu Jul 07 20:40:37 2022 -0700
committer Ravago Jones <ravagojones@gmail.com> Tue Jul 26 21:59:32 2022 -0700
tree 63ca03cabb800dcf1490f26dce1b4e14e21c84c5
parent fa8da564f2b12fa5bc6cfdb4d07bfa244a9ec8c7

Update Spline UI to graph multiple multisplines

Previously it would only graph the first one

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

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 82b95fc..91f31d4 100755
--- a/frc971/control_loops/python/path_edit.py
+++ b/frc971/control_loops/python/path_edit.py
@@ -247,18 +247,19 @@
             self.multisplines, mouse)
 
         # if the mouse is close enough, draw the robot
-        if result and result.fun < 2:
+        if result is not None and result.fun < 2:
             distance_spline = DistanceSpline(multispline.getLibsplines())
             x = result.x[0]
 
             # draw the robot to show its width
             self.draw_robot_at_point(cr, distance_spline, x)
 
-            self.graph.place_cursor(x)
-        elif self.graph.cursor:
+            multispline_index = self.multisplines.index(multispline)
+            self.graph.place_cursor(multispline_index, distance=result.x[0])
+        elif self.graph.cursor is not None:
+            multispline_index, x = self.graph.find_cursor()
             distance_spline = DistanceSpline(
-                self.multisplines[0].getLibsplines())
-            x = self.graph.find_cursor()
+                self.multisplines[multispline_index].getLibsplines())
 
             self.draw_robot_at_point(cr, distance_spline, x)