add: added projectsHEADmain

1 files changed, 137 insertions, 0 deletions

diff --git a/ero1/src/helper/main_parcours.py b/ero1/src/helper/main_parcours.py
new file mode 100644
index 0000000..02b92de
--- /dev/null
+++ b/ero1/src/helper/main_parcours.py
@@ -0,0 +1,137 @@
+import time
+from src.helper.debug_printer import debug_print
+from src.helper.export_cost import export_cost
+from src.helper.export_to_kml import export_to_kml
+from src.deneigeuses.hierholzer import process_hierholzer
+from src.deneigeuses.directed_cpp import oriented_cpt
+from src.helper.display_graph import display_graph
+from src.helper.output_debug import output_debug
+from src.generation.graph_generation import generate_graph
+from src.helper.color_suburbs import display_suburbs_with_colors
+from src.helper.prune_maps import prune_verdun
+import networkx as nx
+
+
+# IMPORT FOR YAML
+from src.helper.export_import_yaml import load_paths_from_yaml
+from src.helper.partitions import partition
+from src.generation.suburb_snowplow_generation import generate_quartiers_from_path
+
+# IMPORT FOR DRONE
+from src.drone.postier_chinoisV2 import final_path
+# from src.drone.postier_chinoisV2 import postier_chinois_process
+
+def main_parcours(debug_mode=False, place_name=None, reversed_legend=None):
+    """
+    Fonction principale pour le parcours du graphe.
+    Parameters:
+        debug_mode : Indique si le mode debug est activé ou non.
+        place_name : Le nom de la place à parcourir.
+        reversed_legend : Si la légende est inversée.
+    """
+
+    #output_debug(None, None, None, debug_mode)
+
+    parcours_deneigeuses = []
+    # ---------------------- GENERATION DU GRAPHE  ---------------------------
+
+    debug_print("Génération du graph de Montréal en cours...", debug_mode)
+    graph = generate_graph("Montréal, Québec, Canada", debug_mode)
+    debug_print("Graph généré avec succès.", debug_mode)
+
+    # ---------------------- GENERATION DU GRAPHE  ---------------------------
+
+    # ---------------------- PARCOURS - DRONE  ---------------------------
+    
+    start_time_drone = time.time()
+    
+    #debug_print("Parcours du drone dans chaque arrondissement", debug_mode)
+    # - Load from the saved file directly
+    parcours_drone = load_paths_from_yaml(graph, filename="paths/paths-PostierChinoisV1.yml") # Nom à rajouter du coup
+    finalPath = final_path(graph, parcours_drone)
+    
+    # - Postier Chinois V{?}
+    # parcours_drone, finalPath = postier_chinois_process(graph, debug_mode)
+
+    end_time_drone = time.time()
+
+    # ---------------------- PARCOURS - DRONE  ---------------------------
+
+    # display_suburbs_with_colors(graph, parcours_drone, connections)
+
+    # ---------------------- RECUPERATION DU PARCOURS DU DRONE  ---------------------------
+    temp = []
+    # if place_name in parcours_drone:
+    #     debug_print("Parcours du drone dans l'arrondissement de " + place_name, debug_mode)
+    #     for elt in ["Outremont", "Verdun", "Anjou", "Rivière-des-Prairies–Pointe-aux-Trembles", "Le Plateau-Mont-Royal"] :
+    #         if elt in parcours_drone:
+    #             print(elt, len(parcours_drone[elt]))
+    #             temp.extend(parcours_drone[elt])
+    #         else:
+    #             print("Arrondissement non trouvé:", elt)
+    #     print("Total", len(temp))
+    #     parcours_drone = temp
+    # else:
+    #     debug_print("Parcours du drone dans l'arrondissement de " + place_name + " non trouvé", debug_mode)
+
+    # ---------------------- RECUPERATION DU PARCOURS DU DRONE  ---------------------------
+
+    #debug_print("Etablissement de connections entre arrondissement :", debug_mode)
+    #connections = connect_circuits(graph, parcours_drone)
+
+    #output_debug(graph, parcours_drone, connections, debug_mode)
+
+    # ---------------------- PARCOURS - DENEIGEUSE  ---------------------------
+
+    # G_partition = generate_quartiers_from_path(parcours_drone, graph, suburb_list=["Outremont", "Verdun", "Anjou", "Rivière-des-Prairies–Pointe-aux-Trembles", "Le Plateau-Mont-Royal"], debug_mode=True)
+    # partitions = partition(G_partition, 5, debug_mode=True)
+
+    # display_graph(graph, "Test 1", reversed_legend, parcours_drone, None, False)
+
+    start_time_deneigeuse = time.time()
+    
+    # TEMP FIX
+    # parcours_deneigeuse = parcours_drone
+
+    ## Postier chinois qui fonctionne que sur un graph fortement connexe
+    
+    verdun = generate_graph("Verdun, Montreal, Canada", debug_mode=debug_mode)
+    # verdun = prune_verdun(verdun)
+    parcours_deneigeuse = oriented_cpt(verdun, graph, debug_mode)
+
+    # for elt in partitions:
+    #     drone_edges = [(u, v) for u, v in elt.edges()]
+    #     parcours_deneigeuses.append(process_hierholzer(graph, drone_edges, debug_mode))
+    # parcours_deneigeuse = process_hierholzer(graph, parcours_drone, debug_mode)
+    end_time_deneigeuse = time.time()
+
+    # ---------------------- PARCOURS - DENEIGEUSE  ---------------------------
+
+    #display_graph(graph, "Test 2", reversed_legend, parcours_deneigeuse, None, False)
+
+    # ---------------------- EXPORT COST  ---------------------------
+
+    # parcours_deneigeuses.append(parcours_deneigeuse)
+    for elt in ["Outremont", "Verdun", "Anjou", "Rivière-des-Prairies–Pointe-aux-Trembles", "Le Plateau-Mont-Royal"] :
+        if elt in parcours_drone:
+            print(elt, len(parcours_drone[elt]))
+            temp.extend(parcours_drone[elt])
+        else:
+            print("Arrondissement non trouvé:", elt)
+    print("Total", len(temp))
+    parcours_drone = temp
+
+    if "verdun" in place_name.lower():
+        export_cost(parcours_drone, parcours_deneigeuses, graph, debug_mode, end_time_deneigeuse - start_time_deneigeuse, end_time_drone - start_time_drone)
+    else:
+        print("/!\\ Exportation du coût du parcours désactivée : Seul Verdun est supporté")
+
+    # ---------------------- EXPORT COST  ---------------------------
+
+    # ---------------------- EXPORT TO KML  ---------------------------
+
+    export_to_kml(parcours_deneigeuses, finalPath, graph, debug_mode)
+
+    # ---------------------- EXPORT TO KML  ---------------------------
+
+    return True