from inire.geometry.collision import CollisionEngine
from inire.geometry.primitives import Port
from inire.router.astar import AStarContext, AStarMetrics
from inire.router.cost import CostEvaluator
from inire.router.danger_map import DangerMap
from inire.router.pathfinder import PathFinder
from inire.utils.visualization import plot_routing_results


def main() -> None:
    print("Running Example 01: Simple Route...")

    # 1. Setup Environment
    # We define a 100um x 100um routing area
    bounds = (0, 0, 100, 100)
    
    # Clearance of 2.0um between waveguides
    engine = CollisionEngine(clearance=2.0)
    
    # Precompute DangerMap for heuristic speedup
    danger_map = DangerMap(bounds=bounds)
    danger_map.precompute([]) # No obstacles yet

    # 2. Configure Router
    evaluator = CostEvaluator(engine, danger_map)
    context = AStarContext(evaluator, snap_size=1.0, bend_radii=[10.0])
    metrics = AStarMetrics()
    pf = PathFinder(context, metrics)

    # 3. Define Netlist
    # Start at (10, 50) pointing East (0 deg)
    # Target at (90, 50) pointing East (0 deg)
    netlist = {
        "net1": (Port(10, 50, 0), Port(90, 50, 0)),
    }
    net_widths = {"net1": 2.0}

    # 4. Route
    results = pf.route_all(netlist, net_widths)

    # 5. Check Results
    res = results["net1"]
    if res.is_valid:
        print("Success! Route found.")
        print(f"Path collisions: {res.collisions}")
    else:
        print("Failed to find route.")

    # 6. Visualize
    # plot_routing_results takes a dict of RoutingResult objects
    fig, ax = plot_routing_results(results, [], bounds)
    fig.savefig("examples/01_simple_route.png")
    print("Saved plot to examples/01_simple_route.png")


if __name__ == "__main__":
    main()