iflow-mcp_xrds76354_sumo-mcp 0.1.0__py3-none-any.whl

workflows/sim_gen.py

@@ -0,0 +1,70 @@
+import os
+from mcp_tools.network import netgenerate
+from mcp_tools.route import random_trips, duarouter
+from mcp_tools.simulation import run_simple_simulation
+from mcp_tools.analysis import analyze_fcd
+
+def sim_gen_workflow(output_dir: str, grid_number: int = 3, steps: int = 100) -> str:
+    """
+    Executes the Simulation Generation & Evaluation workflow:
+    1. Generate Grid Network
+    2. Generate Random Trips
+    3. Compute Routes
+    4. Create Config
+    5. Run Simulation
+    6. Analyze Results
+    """
+    if not os.path.exists(output_dir):
+        os.makedirs(output_dir)
+        
+    net_file = os.path.join(output_dir, "grid.net.xml")
+    trips_file = os.path.join(output_dir, "trips.xml")
+    route_file = os.path.join(output_dir, "routes.xml")
+    sumocfg_file = os.path.join(output_dir, "sim.sumocfg")
+    fcd_file = os.path.join(output_dir, "fcd.xml")
+    
+    # 1. Generate Net
+    res = netgenerate(net_file, grid=True, grid_number=grid_number)
+    if "failed" in res.lower(): return f"Step 1 Failed: {res}"
+    
+    # 2. Generate Trips
+    res = random_trips(net_file, trips_file, end_time=steps)
+    if "failed" in res.lower(): return f"Step 2 Failed: {res}"
+    
+    # 3. Generate Routes
+    res = duarouter(net_file, trips_file, route_file)
+    if "failed" in res.lower(): return f"Step 3 Failed: {res}"
+    
+    # 4. Create Config
+    # We use absolute paths for safety or relative if SUMO expects it. 
+    # Usually relative to config file is best.
+    try:
+        with open(sumocfg_file, "w") as f:
+            f.write(f"""<configuration>
+    <input>
+        <net-file value="{os.path.basename(net_file)}"/>
+        <route-files value="{os.path.basename(route_file)}"/>
+    </input>
+    <time>
+        <begin value="0"/>
+        <end value="{steps}"/>
+    </time>
+    <output>
+        <fcd-output value="{os.path.basename(fcd_file)}"/>
+    </output>
+</configuration>""")
+    except Exception as e:
+        return f"Step 4 Failed: Could not write config file. {e}"
+    
+    # 5. Run Sim
+    res = run_simple_simulation(sumocfg_file, steps)
+    if "error" in res.lower(): return f"Step 5 Failed: {res}"
+    
+    # 6. Analyze
+    # FCD file should exist after sim
+    if not os.path.exists(fcd_file):
+        return "Step 6 Failed: FCD file not generated."
+        
+    res_analysis = analyze_fcd(fcd_file)
+    
+    return f"Workflow Completed Successfully.\n\nSimulation Output:\n{res}\n\nAnalysis Result:\n{res_analysis}"