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