1 files changed, 76 insertions, 0 deletions
diff --git a/ero1/src/helper/prune_maps.py b/ero1/src/helper/prune_maps.py
new file mode 100644
index 0000000..6559a77
--- /dev/null
+++ b/ero1/src/helper/prune_maps.py
@@ -0,0 +1,76 @@
+import networkx as nx
+import osmnx as ox
+from src.helper.debug_printer import debug_print
+from src.helper.display_graph import display_graph
+
+def prune_verdun(G, debug_mode=False):
+    """
+    Nettoie le graph en retirant les arrêtes problématiques
+    afin que le graph soit orienté et fortement connexe.
+    Les Retraits sont pour la plupart minimes et concernent
+    des routes à sens unique qui vont vers d'autre quartiers.
+    (Impossibles à parcourir dans le cadre d'un quartier à moins de sortir)
+    Retourne le graph nettoyé.
+    Parameters:
+    G: Graph du quartier de Verdun
+    debug_mode: [OPTIONEL | DEFAUT = False]
+    """
+    debug_print("Nettoyage du graph Verdun", debug_mode)
+
+    G.remove_edge(32764413, 248511841)
+    G.remove_edge(615028247, 615028248)
+    G.remove_edge(615028248, 4503982628)
+    G.remove_edge(615035051, 615028249)
+    G.remove_edge(4503982628, 615035051)
+    G.remove_edge(5342978418, 615028328)
+
+    G.remove_node(8640521401)
+    G.remove_node(32764413)
+    G.remove_node(248511841)
+    G.remove_node(615028247)
+    G.remove_node(615028248)
+    G.remove_node(4503982628)
+    G.remove_node(615035051)
+    G.remove_node(615028328)
+
+    debug_print("Verdun nettoyé! Supprimé 8 arcs et 8 noeuds", debug_mode)
+
+    return G
+
+def prune_outremont(G, debug_mode=False):
+    """
+    Nettoie le graph en retirant les arrêtes problématiques
+    afin que le graph soit orienté et fortement connexe.
+    Les Retraits sont pour la plupart minimes et concernent
+    des routes à sens unique qui vont vers d'autre quartiers.
+    (Impossibles à parcourir dans le cadre d'un quartier à moins de sortir)
+    Retourne le graph nettoyé.
+    Parameters:
+    G: Graph du quartier d'Outremont
+    debug_mode: [OPTIONEL | DEFAUT = False]
+    """
+    debug_print("Nettoyage du graph Outremont...", debug_mode)
+
+    # Avenue Willowdale
+    G.remove_node(209387127)
+    G.remove_node(3165394666)
+    G.remove_node(209387136)
+    G.remove_node(209387140)
+    G.remove_node(209387103)
+    G.remove_node(209387147)
+
+    # Avenue Gare de Triage
+    G.remove_node(5412399376)
+
+    # Avenue Durocher/Atlantic
+    G.remove_node(437865622)
+    G.remove_node(12844070435)
+    G.remove_node(5412399379)
+
+    # Sortie autoroute Avenue Davaar et Rockland
+    G.remove_node(213955306)
+    G.remove_node(213955379)
+
+    return G
+
+# TESTINGGGGGG