from inire import NetSpec, ObjectiveWeights, Port, RoutingOptions, RoutingProblem, SearchOptions, route
from inire.utils.visualization import plot_routing_results


def main() -> None:
    print("Running Example 04: S-Bends and Multiple Radii...")

    bounds = (0, 0, 100, 100)
    netlist = {
        "sbend_only": (Port(10, 50, 0), Port(60, 55, 0)),
        "multi_radii": (Port(10, 10, 0), Port(90, 90, 0)),
    }
    problem = RoutingProblem(
        bounds=bounds,
        nets=tuple(NetSpec(net_id, start, target, width=2.0) for net_id, (start, target) in netlist.items()),
    )
    options = RoutingOptions(
        search=SearchOptions(
            node_limit=50000,
            bend_radii=(10.0, 30.0),
            sbend_offsets=(5.0,),
        ),
        objective=ObjectiveWeights(
            unit_length_cost=1.0,
            bend_penalty=10.0,
            sbend_penalty=20.0,
        ),
    )

    run = route(problem, options=options)
    for net_id, result in run.results_by_net.items():
        status = "Success" if result.is_valid else "Failed"
        print(f"{net_id}: {status}, collisions={result.collisions}")

    fig, _ax = plot_routing_results(run.results_by_net, [], bounds, netlist=netlist)
    fig.savefig("examples/04_sbends_and_radii.png")
    print("Saved plot to examples/04_sbends_and_radii.png")


if __name__ == "__main__":
    main()