From 88eb25614ffdbd2b077cf4bef86693c5ed8b316c Mon Sep 17 00:00:00 2001
From: Grufoony <gregorio.berselli@studio.unibo.it>
Date: Thu, 5 Mar 2026 17:37:30 +0100
Subject: [PATCH] Prova

---
 examples/modena.py | 75 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100644 examples/modena.py

diff --git a/examples/modena.py b/examples/modena.py
new file mode 100644
index 00000000..7df1e4f7
--- /dev/null
+++ b/examples/modena.py
@@ -0,0 +1,75 @@
+from datetime import datetime
+import logging
+
+from dsf.cartography import get_cartography
+from dsf.mobility import RoadNetwork, Dynamics
+
+from tqdm import trange
+import numpy as np
+
+SEED = 42
+np.random.seed(SEED)
+
+logging.basicConfig(level=logging.INFO)
+
+if __name__ == "__main__":
+    logging.info("Getting data from OpenStreetMap...")
+    # Get the cartography of Modena, Italy
+    G, df_edges, df_nodes = get_cartography("Parma, Emilia-Romagna, Italy")
+    # set nlanes to 1 if 0
+    df_edges["nlanes"] = df_edges["nlanes"].replace(0, 1).fillna(1).astype(int)
+
+    df_edges.to_csv("modena_edges.csv", sep=";", index=False)
+    df_nodes.to_csv("modena_nodes.csv", sep=";", index=False)
+
+    nodes = G.nodes(data=False)
+    # Extract 10% random node ids as origins and destinations for the traffic simulation
+    origin_ids = np.random.choice(
+        list(nodes), size=int(0.1 * len(nodes)), replace=False
+    )
+    destination_ids = np.random.choice(
+        list(nodes), size=int(0.1 * len(nodes)), replace=False
+    )
+
+    del df_edges, df_nodes, G
+
+    logging.info("Creating road network and dynamics model...")
+
+    # Create a road network from the cartography
+    road_network = RoadNetwork()
+    road_network.importEdges("modena_edges.csv", ";")
+    road_network.importNodeProperties("modena_nodes.csv", ";")
+    # Adjust network parameters
+    road_network.adjustNodeCapacities()
+    road_network.autoMapStreetLanes()
+    road_network.autoAssignRoadPriorities()
+    road_network.autoInitTrafficLights()
+
+    # Generaate a random vector of integer values for vehicle input
+    # We want values to have a 10s entry for a whole day
+    vehicle_input = np.random.randint(0, 10, size=8640)
+
+    # Create a dynamics model for the road network
+    dynamics = Dynamics(road_network, seed=SEED, alpha=0.8)
+    # Get epoch time of today at midnight
+    epoch_time = int(
+        datetime.combine(datetime.today(), datetime.min.time()).timestamp()
+    )
+
+    dynamics.setInitTime(epoch_time)
+    dynamics.connectDataBase("modena.db")
+    dynamics.saveData(300, True, True, True)
+
+    dynamics.setOriginNodes(origin_ids)
+    dynamics.setDestinationNodes(destination_ids)
+
+    # Simulate traffic for 24 hours with a time step of 10 seconds
+    for time_step in trange(8640):
+        # Update paths every 5 minutes (300 seconds)
+        if time_step % 300 == 0:
+            dynamics.updatePaths()
+        # Add agents every 10 seconds
+        if time_step % 10 == 0:
+            dynamics.addAgentsRandomly(vehicle_input[time_step])
+
+    dynamics.summary()