Decide what scenario to generate

diff --git a/scripts/generate_simple_json_scenario.py b/scripts/generate_simple_json_scenario.py
index da1a4833b61c539aa5395ebc1aeefc5194945763..88724e7e1ff182a8b40065f5f63b16c091019166 100644 (file)
--- a/scripts/generate_simple_json_scenario.py
+++ b/scripts/generate_simple_json_scenario.py
@@ -11,6 +11,7 @@ The scenario contains at least:
 from json import dumps
 from math import cos, pi, sin
 from random import random, uniform
+import sys
 
 W = 1.625
 L = 3.760
@@ -198,12 +199,20 @@ def gen_obst():
     return obstacles
 
 
+SCENARIOS = {
+    "pa01": {
+        "init": gen_init(),
+        "slot": gen_slot_at(15, 0, 0, True),
+        "obst": [gen_obst_at(p) for p in [[], [], []]]}}
+
 if __name__ == "__main__":
-    init = gen_init()
-    slot = gen_slot()
-    obst = gen_obst()
-    print(dumps({
-        "init": init,
-        "slot": slot,
-        "obst": obst,
-    }))
+    sc = ""
+    if len(sys.argv) == 2:
+        sc = sys.argv[1]
+    if sc in SCENARIOS:
+        print(dumps(SCENARIOS()[sc]))
+    else:
+        print(dumps({
+            "init": gen_init(),
+            "slot": gen_slot(),
+            "obst": gen_obst()}))