from pathlib import Path
from openalea.mtg import traversal
from openalea.plantgl.all import Scene, Viewer
import json
import hydroshoot
from hydroshoot import architecture, display, model

print(hydroshoot.__file__)

def main():
    # 📂 Percorso del progetto (dove sono params.json, digit.input, meteo.input, ecc.)
    path_project = Path(r".\hydroshoot\example\vsp_ws_grapevine")

    # =============================================================================
    # 🌱 Costruzione della vite a partire da digit.input
    # =============================================================================
    g = architecture.vine_mtg(path_project / 'digit.input')

    for v in traversal.iter_mtg2(g, g.root):
        architecture.vine_phyto_modular(g, v)
        architecture.vine_axeII(
            g, v,
            pruning_type='avg_field_model',
            N_max=6,
            insert_angle=90,
            N_max_order=6
        )
        architecture.vine_petiole(
            g, v,
            pet_ins=90., pet_ins_cv=0.,
            phyllo_angle=180.
        )
        architecture.vine_leaf(
            g, v,
            leaf_inc=-45., leaf_inc_cv=100.,
            lim_max=12.5, lim_min=5.,
            order_lim_max=6,
            max_order=55,
            rand_rot_angle=90.,
            cordon_vector=None
        )
        architecture.vine_mtg_properties(g, v)
        architecture.vine_mtg_geometry(g, v)
        architecture.vine_transform(g, v)

    # 👀 Visualizzazione (opzionale, richiede PlantGL viewer)
    scene = display.visu(
        g,
        def_elmnt_color_dict=True,
        scene=Scene(),
        view_result=False
    )

    Viewer.display(scene)
    Viewer.saveSnapshot("output_initial.png")
    print("✅ Immagine salvata come output.png")

    # =============================================================================
    # 🚀 Esegui la simulazione
    # =============================================================================
    model.run(
        g=g,
        wd=path_project,
        path_weather=path_project / 'meteo.input',
        scene=scene
    )

    # Stato finale
    scene_final = display.visu(
        g,
        def_elmnt_color_dict=True,
        scene=Scene(),
        view_result=False
    )
    Viewer.display(scene_final)
    Viewer.saveSnapshot("output_final.png")
    print("✅ Immagine finale salvata come output_final.png")

    # =============================================================================
    # 📊 Estrazione parametri fisici dalla pianta
    # =============================================================================
    positions = g.property("position")

    if not positions:
        print("⚠️ Nessuna proprietà 'position' trovata, provo con 'geometry'...")
        positions = g.property("geometry")

    if positions:
        height = max(pos[2] for pos in positions.values())
        print(f"Altezza pianta: {height:.2f} cm")
    else:
        print("❌ Non ci sono coordinate nella pianta, impossibile calcolare l’altezza")
        positions = g.property("position")
    
    classes = g.property("label")

    # Numero foglie
    n_leaves = sum(1 for vid, lbl in classes.items() if "Leaf" in lbl)
    print(f"Numero foglie: {n_leaves}")

    # Numero internodi / rami
    n_internodes = sum(1 for vid, lbl in classes.items() if "Internode" in lbl)
    print(f"Numero internodi (rami): {n_internodes}")

    # Numero frutti presenti
    n_fruits = sum(1 for vid, lbl in classes.items() if "Fruit" in lbl)
    print(f"Numero frutti: {n_fruits}")
    print("✅ Simulazione completata! Output in:")
    print(f" - {path_project}/time_series.output")
    print(f" - {path_project}/*.pckl (stati della pianta)")
    plant_summary = {
        "height_cm": height,
        "n_leaves": n_leaves,
        "n_internodes": n_internodes,
        "n_fruits": n_fruits,
        "initial_image": "output_initial.png",
        "final_image": "output_final.png",
        "time_series_output": str(path_project / "time_series.output"),
        "pickle_states": str(path_project / "*.pckl")
    }
    with open("plant_summary.json", "w", encoding="utf-8") as f:
        json.dump(plant_summary, f, indent=2, ensure_ascii=False)

if __name__ == '__main__':
    main()