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 05: Orientation Stress Test...")

    # 1. Setup Environment
    bounds = (0, 0, 200, 200)
    engine = CollisionEngine(clearance=2.0)
    danger_map = DangerMap(bounds=bounds)
    danger_map.precompute([])

    evaluator = CostEvaluator(engine, danger_map, bend_penalty=50.0)
    context = AStarContext(evaluator, snap_size=5.0, bend_radii=[20.0])
    metrics = AStarMetrics()
    pf = PathFinder(context, metrics)

    # 2. Define Netlist
    # Challenging orientation combinations
    netlist = {
        "u_turn": (Port(50, 50, 0), Port(50, 70, 180)),
        "loop": (Port(100, 100, 90), Port(100, 80, 270)),
        "zig_zag": (Port(20, 150, 0), Port(180, 150, 0)),
    }
    net_widths = {nid: 2.0 for nid in netlist}

    # 3. Route
    print("Routing complex orientation nets...")
    results = pf.route_all(netlist, net_widths)

    # 4. Check Results
    for nid, res in results.items():
        status = "Success" if res.is_valid else "Failed"
        print(f"  {nid}: {status}")

    # 5. Visualize
    fig, ax = plot_routing_results(results, [], bounds, netlist=netlist)
    fig.savefig("examples/05_orientation_stress.png")
    print("Saved plot to examples/05_orientation_stress.png")


if __name__ == "__main__":
    main()