1 files changed, 35 insertions, 0 deletions
diff --git a/ero1/src/deneigeuses/hangar_to_deneigeuse.py b/ero1/src/deneigeuses/hangar_to_deneigeuse.py
new file mode 100644
index 0000000..1bd6ae0
--- /dev/null
+++ b/ero1/src/deneigeuses/hangar_to_deneigeuse.py
@@ -0,0 +1,35 @@
+import networkx as nx
+
+def path_hangar_to_deneigeuse(Graphe, start_point):
+    """
+    Fait le lien entre le hangar "theorique" et le départ de la dénéigeuse
+    """
+    hangar = 299783596
+    res = []
+
+    try:
+        _, path_nodes = nx.single_source_dijkstra(Graphe, hangar, start_point, weight='length')
+    except nx.NetworkXNoPath:  
+        return []  
+
+    for i in range(len(path_nodes) - 1):
+        edge = (path_nodes[i], path_nodes[i+1])
+        res.append(edge)
+    return res
+
+def path_deneigeuse_to_hangar(Graphe, start_point):
+    """
+    Fait le lien entre l'emplacement de la deneigeuse et du hangar
+    """
+    hangar = 299783596
+    res = []
+
+    try:
+        _, path_nodes = nx.single_source_dijkstra(Graphe, start_point, hangar, weight='length')
+    except nx.NetworkXNoPath:  
+        return []  
+
+    for i in range(len(path_nodes) - 1):
+        edge = (path_nodes[i], path_nodes[i+1])
+        res.append(edge)
+    return res