commit 85f861f4cd5489e8140a82b74eb178b4a98e4eee
author Ryan Yin <100028731@mvla.net> Thu Sep 16 17:55:11 2021 -0700
committer Ryan Yin <100028731@mvla.net> Sat Sep 18 19:26:51 2021 -0700
tree 67145fb7846c1f6d05f5ae52fb9a832fcedf3263
parent 5a3d2fd9a50495e30043c01ad740fa4307d5abde

Add clear button

Change-Id: I8e01e79d432b1c0ce77af0410eec79b30c98e6c3
Signed-off-by: Ryan Yin <100028731@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 d743e0d..b7e0246 100755
--- a/frc971/control_loops/python/path_edit.py
+++ b/frc971/control_loops/python/path_edit.py
@@ -235,7 +235,6 @@
             if self.points.getSplines():
                 self.draw_splines(cr)
                 for i, points in enumerate(self.points.getSplines()):
-
                     points = [
                         np.array([self.mToPx(x), self.mToPx(y)])
                         for (x, y) in points
@@ -332,6 +331,17 @@
         self.queue_draw()
         self.graph.schedule_recalculate(self.points)
 
+    def clear_graph(self):
+        self.points = Points()
+        #recalulate graph using new points
+        self.graph.axis.clear()
+        self.graph.queue_draw()
+        #allow placing again
+        self.mode = Mode.kPlacing
+        #redraw entire graph
+        self.queue_draw()
+        #TODO: Make a way to undo clear
+
     def do_key_press_event(self, event):
         keyval = Gdk.keyval_to_lower(event.keyval)
 
@@ -354,7 +364,6 @@
             event.x, event.y)
         if self.mode == Mode.kEditing:
             if self.index_of_edit > -1 and self.held_x != self.mousex:
-
                 self.points.setSplines(self.spline_edit, self.index_of_edit,
                                        self.pxToM(self.mousex),
                                        self.pxToM(self.mousey))