Merge pull request #259 from goktug97/close_on_key

Close on key

AerialNavigation/rocket_powered_landing/rocket_powered_landing.py

@@ -567,6 +567,9 @@ def plot_animation(X, U):  # pragma: no cover
 
     fig = plt.figure()
     ax = fig.gca(projection='3d')
+    # for stopping simulation with the esc key.
+    fig.canvas.mpl_connect('key_release_event',
+            lambda event: [exit(0) if event.key == 'escape' else None])
 
     for k in range(K):
         plt.cla()