Merge branch 'release/0.7.0'

[hubacji1/iamcar.git] / plot.py

diff --git a/plot.py b/plot.py
index 4680eddc6f24ce1e624d9130ff60fc74f8b73589..c9f1eb13aad7f300ec73e2ca8e6d13a78883c1db 100644 (file)
--- a/plot.py
+++ b/plot.py
@@ -129,13 +129,13 @@ def plot_segments(segments=[]):
 
 if __name__ == "__main__":
     SCEN_FILE = argv[1]
-    TRAJ_FILE = argv[2]
 
     s = load_scenario(SCEN_FILE)
     try:
+        TRAJ_FILE = argv[2]
         t = load_trajectory(TRAJ_FILE) # fixed to trajectories
     except:
-        pass
+        t = {"elap": 0}
 
     plt.rcParams["font.size"] = 24
     plt.rcParams['hatch.linewidth'] = 1.0
@@ -221,10 +221,18 @@ if __name__ == "__main__":
         except:
             print("No slot")
 
-    plt.plot(*car_frame(s["init"]), color="red", lw=2)
-    plt.plot(*car_frame(t["goal"]), color="red", lw=2)
-    plt.plot(*plot_nodes([s["init"]]), color="red", marker="+", ms=12)
-    plt.plot(*plot_nodes([t["goal"]]), color="red", marker="+", ms=12)
+    try: # init
+        plt.plot(*car_frame(t["init"]), color="red", lw=2)
+        plt.plot(*plot_nodes([t["init"]]), color="red", marker="+", ms=12)
+    except:
+        plt.plot(*car_frame(s["init"]), color="red", lw=2)
+        plt.plot(*plot_nodes([s["init"]]), color="red", marker="+", ms=12)
+    try: # goal
+        plt.plot(*plot_nodes([t["goal"]]), color="red", marker="+", ms=12)
+        plt.plot(*car_frame(t["goal"]), color="red", lw=2)
+    except:
+        plt.plot(*plot_nodes([s["goal"]]), color="red", marker="+", ms=12)
+        plt.plot(*car_frame(s["goal"]), color="red", lw=2)
     try: # middle
         plt.plot(*car_frame(t["midd"]), color="red", lw=2)
         plt.plot(*plot_nodes([t["midd"]]), color="red", marker="+", ms=12)