Update plot

Decide if there are some edges to plot and plot just one figure if not.

diff --git a/plot.py b/plot.py
index ff42fb894c0b077f782119a19a67f9a6f54ac434..43f6ee8093c618140c3e2535007820246aeb0f5e 100644 (file)
--- a/plot.py
+++ b/plot.py
@@ -3,7 +3,7 @@
 from json import loads
 from math import copysign, cos, pi, sin
 from matplotlib import pyplot as plt
 from json import loads
 from math import copysign, cos, pi, sin
 from matplotlib import pyplot as plt

-from sys import argv
+from sys import argv, exit

 
 sign = lambda x: copysign(1, x)
 
 
 sign = lambda x: copysign(1, x)
 


@@ -142,7 +142,10 @@ if __name__ == "__main__":
     ################
     ## 1st subplot
 
     ################
     ## 1st subplot
 

-    ax = fig.add_subplot(121)
+    if "edge" in t.keys():
+        ax = fig.add_subplot(121)
+    else:
+        ax = fig.add_subplot(111)

     ax.set_aspect("equal")
     ax.set_title("Final path")
     ax.set_xlabel("x [m]")
     ax.set_aspect("equal")
     ax.set_title("Final path")
     ax.set_xlabel("x [m]")


@@ -239,6 +242,11 @@ if __name__ == "__main__":
 
     handles, labels = ax.get_legend_handles_labels()
 
 
     handles, labels = ax.get_legend_handles_labels()
 

+    if not "edge" in t.keys():
+        plt.show()
+        plt.close(fig)
+        exit(0)
+

     ################
     ## 2nd subplot
 
     ################
     ## 2nd subplot