diff --git a/trajectory_drawer.py b/trajectory_drawer.py index e648211..4759771 100644 --- a/trajectory_drawer.py +++ b/trajectory_drawer.py @@ -37,6 +37,11 @@ class TrajectoryDrawer: self.complete_trajectory() elif event.key == 'escape': self.clear_trajectory() + elif event.key == 'e': + if len(self.points): + self.points.append(self.points[0]) + self.update_plot() + def update_plot(self): if self.points: