58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
|
from shapely.geometry import Polygon
|
||
|
|
|
||
|
|
from inire.geometry.collision import CollisionEngine
|
||
|
|
from inire.geometry.primitives import Port
|
||
|
|
from inire.router.astar import AStarRouter
|
||
|
|
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
|
||
|
|
# Define the routing area bounds (minx, miny, maxx, maxy)
|
||
|
|
bounds = (0, 0, 100, 100)
|
||
|
|
|
||
|
|
engine = CollisionEngine(clearance=2.0)
|
||
|
|
danger_map = DangerMap(bounds=bounds)
|
||
|
|
|
||
|
|
# Add a simple rectangular obstacle
|
||
|
|
obstacle = Polygon([(30, 20), (70, 20), (70, 40), (30, 40)])
|
||
|
|
engine.add_static_obstacle(obstacle)
|
||
|
|
|
||
|
|
# Precompute the danger map (distance field) for heuristics
|
||
|
|
danger_map.precompute([obstacle])
|
||
|
|
|
||
|
|
evaluator = CostEvaluator(engine, danger_map)
|
||
|
|
router = AStarRouter(evaluator)
|
||
|
|
pf = PathFinder(router, evaluator)
|
||
|
|
|
||
|
|
# 2. Define Netlist
|
||
|
|
# Route from (10, 10) to (90, 50)
|
||
|
|
# The obstacle at y=20-40 blocks the direct path.
|
||
|
|
netlist = {
|
||
|
|
"simple_net": (Port(10, 10, 0), Port(90, 50, 0)),
|
||
|
|
}
|
||
|
|
net_widths = {"simple_net": 2.0}
|
||
|
|
|
||
|
|
# 3. Route
|
||
|
|
results = pf.route_all(netlist, net_widths)
|
||
|
|
|
||
|
|
# 4. Check Results
|
||
|
|
if results["simple_net"].is_valid:
|
||
|
|
print("Success! Route found.")
|
||
|
|
print(f"Path collisions: {results['simple_net'].collisions}")
|
||
|
|
else:
|
||
|
|
print("Failed to route.")
|
||
|
|
|
||
|
|
# 5. Visualize
|
||
|
|
fig, ax = plot_routing_results(results, [obstacle], bounds)
|
||
|
|
fig.savefig("examples/simple_route.png")
|
||
|
|
print("Saved plot to examples/simple_route.png")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|