commit d8be38815eb75245dc473d58e1d7cc8afea54ea8
author Ryan Yin <100028731@mvla.net> Wed Oct 13 20:59:41 2021 -0700
committer Ryan Yin <100028731@mvla.net> Wed Nov 10 14:03:12 2021 -0700
tree 94b80331abd87045eb06f99c65b2fd948addf088
parent cef51f465f274f4927c1170a9c15a61335c098e7

Added undo button to spline UI

Signed-off-by: Ryan Yin <100028731@mvla.net>
Change-Id: I39a2f1ea9695206fba78cbe1c0c4dbbe525c469f

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 e07bc80..2e67480 100755
--- a/frc971/control_loops/python/spline_graph.py
+++ b/frc971/control_loops/python/spline_graph.py
@@ -34,6 +34,8 @@
             self.field.points.getConstraint("LATERAL_ACCELERATION"))
         self.vol_input.set_value(self.field.points.getConstraint("VOLTAGE"))
 
+    def undo_func(self, *args):
+        self.field.undo()
     def long_changed(self, button):
         value = self.long_input.get_value()
         self.field.points.setConstraint("LONGITUDINAL_ACCELERATION", value)
@@ -135,6 +137,10 @@
         self.clear = Gtk.Button.new_with_label("Clear")
         self.clear.set_size_request(100, 40)
         self.clear.connect("clicked", self.clear_clicked)
+
+        self.undo = Gtk.Button.new_with_label("Undo (Ctrl + Z)")
+        self.undo.set_size_request(100, 40)
+        self.undo.connect("clicked", self.undo_func)
         #Dropdown feature
         self.label = Gtk.Label()
         self.label.set_text("Change Field:")
@@ -171,11 +177,12 @@
         container.attach(limitControls, 5, 1, 1, 1)
 
         jsonControls = Gtk.FlowBox()
-        jsonControls.set_min_children_per_line(3)
+        jsonControls.set_min_children_per_line(5)
         jsonControls.add(self.file_name_box)
         jsonControls.add(self.output_json)
         jsonControls.add(self.input_json)
         jsonControls.add(self.clear)
+        jsonControls.add(self.undo)
         container.attach(jsonControls, 1, 0, 1, 1)
 
         container.attach(self.label, 4, 0, 1, 1)