72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
from shapely.geometry import Polygon, box
|
|
|
|
from inire import CongestionOptions, NetSpec, RoutingOptions, RoutingProblem, SearchOptions, route
|
|
from inire.geometry.components import BendCollisionModel, BendPhysicalGeometry
|
|
from inire.geometry.primitives import Port
|
|
from inire.utils.visualization import plot_routing_results
|
|
|
|
|
|
def _run_session(
|
|
bounds: tuple[float, float, float, float],
|
|
net_id: str,
|
|
start: Port,
|
|
target: Port,
|
|
*,
|
|
bend_collision_type: BendCollisionModel = "arc",
|
|
bend_proxy_geometry: BendCollisionModel | None = None,
|
|
bend_physical_geometry: BendPhysicalGeometry | None = None,
|
|
) -> dict[str, object]:
|
|
problem = RoutingProblem(
|
|
bounds=bounds,
|
|
nets=(NetSpec(net_id, start, target, width=2.0),),
|
|
)
|
|
options = RoutingOptions(
|
|
search=SearchOptions(
|
|
bend_radii=(10.0,),
|
|
bend_collision_type=bend_collision_type,
|
|
bend_proxy_geometry=bend_proxy_geometry,
|
|
bend_physical_geometry=bend_physical_geometry,
|
|
sbend_radii=(),
|
|
),
|
|
congestion=CongestionOptions(max_iterations=1, use_tiered_strategy=False),
|
|
)
|
|
return route(problem, options=options).results_by_net
|
|
|
|
|
|
def main() -> None:
|
|
print("Running Example 08: Custom Bend Geometry...")
|
|
|
|
bounds = (0, 0, 150, 150)
|
|
start = Port(20, 20, 0)
|
|
target = Port(100, 100, 90)
|
|
custom_physical = Polygon([(0, -11), (11, -11), (11, 0), (9, 0), (9, -9), (0, -9)])
|
|
custom_proxy = box(0, -11, 11, 0)
|
|
|
|
print("Routing standard arc in its own session...")
|
|
results_std = _run_session(bounds, "standard_arc", start, target)
|
|
print("Routing custom geometry with a separate custom proxy in its own session...")
|
|
results_custom = _run_session(
|
|
bounds,
|
|
"custom_geometry_and_proxy",
|
|
start,
|
|
target,
|
|
bend_physical_geometry=custom_physical,
|
|
bend_proxy_geometry=custom_proxy,
|
|
)
|
|
|
|
all_results = {**results_std, **results_custom}
|
|
fig, _ax = plot_routing_results(
|
|
all_results,
|
|
[],
|
|
bounds,
|
|
netlist={
|
|
"standard_arc": (start, target),
|
|
"custom_geometry_and_proxy": (start, target),
|
|
},
|
|
)
|
|
fig.savefig("examples/08_custom_bend_geometry.png")
|
|
print("Saved plot to examples/08_custom_bend_geometry.png")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|