Discretization Example

from functools import partial
from types import SimpleNamespace

import gdsfactory as gf
from gdsfactory.gpdk import PDK

import gdsfactoryplus as gfp
dr = gfp.routing.doroutes

PDK.activate()

import matplotlib.pyplot as plt
import numpy as np
import shapely.geometry as sg
from tqdm.notebook import tqdm

c = gf.Component()
ref = c << dr.pcells.fanout_frame2(
    transition="ew",
    add_frame=False,
    num_inputs=2,
    width=100,
)
c.add_ports(ref)

straight = "straight"
bend = {"component": "bend_euler", "settings": {"radius": 5}}
spacing = 1.0

_starts, _legs1 = dr.add_fan_in(
    component=c,
    inputs=[p for p in c.ports if str(p.name).startswith("in")],
    straight=straight,
    bend=bend,
    spacing=spacing,
)
_stops, _legs2 = dr.add_fan_in(
    component=c,
    inputs=[p for p in c.ports if str(p.name).startswith("out")],
    straight=straight,
    bend=bend,
    spacing=spacing,
)

if len(_starts) != len(_stops):
    msg = "number of start ports is different than number of end ports"
    raise ValueError(msg)

dr.routing.add_route_astar(
    component=c,
    start=(*np.mean(_starts, 0), "e"),
    stop=(*np.mean(_stops, 0), "e"),
    straight=partial(dr.pcells.straights, straight, len(_starts), spacing),
    bend=partial(dr.pcells.bends, bend, straight, len(_starts), spacing),
    layers=["WG"],
    grid_unit=500,
)
c.show()
dr.util.show_cell(c)

API key for organization 'GDSFactory' found.

bbox = dr.util.as_kcell(c).bbox()
bbox = SimpleNamespace(
    north=bbox.top, east=bbox.right, south=bbox.bottom, west=bbox.left
)
gu = 3000
start = (int(np.mean(_starts[:, 0])), int(np.mean(_starts[:, 1])), "e")
stop = (int(np.mean(_stops[:, 0])), int(np.mean(_stops[:, 1])), "e")
m = (bbox.east - bbox.west) // gu
n = (bbox.north - bbox.south) // gu
xc = np.arange(0, m) * gu + bbox.west
yc = np.arange(0, n) * gu + bbox.south
start_gu = (np.argmin(np.abs(xc - start[0])), np.argmin(np.abs(yc - start[1])))
stop_gu = (np.argmin(np.abs(xc - stop[0])), np.argmin(np.abs(yc - stop[1])))
_start = (start_gu[0] * gu + bbox.west, start_gu[1] * gu + bbox.south)
_stop = (stop_gu[0] * gu + bbox.west, stop_gu[1] * gu + bbox.south)
xb = np.arange(0, m + 1) * gu + bbox.west - gu / 2
yb = np.arange(0, n + 1) * gu + bbox.south - gu / 2
Y, X = np.meshgrid(yb, xb)
grid = np.zeros((m, n), dtype=int)
layer = dr.types.validate_layer(gf.kcl, "WG")
polys = dr.util.extract_polys(c, [layer])
mp = sg.MultiPolygon([sg.Polygon(p) for p in polys])
for i, x in enumerate(tqdm(xc)):
    for j, y in enumerate(yc):
        cell = sg.Polygon(
            [
                (x - gu / 2, y - gu / 2),
                (x - gu / 2, y + gu / 2),
                (x + gu / 2, y + gu / 2),
                (x + gu / 2, y - gu / 2),
            ]
        )
        grid[i, j] = cell.intersects(mp)
grid[start_gu] = 2
grid[stop_gu] = 2
plt.pcolormesh(X, Y, grid, cmap="Greys")
for poly in polys:
    plt.plot(*poly.T)
plt.plot(start[0], start[1], "o", color="green")
plt.plot(stop[0], stop[1], "o", color="green")
plt.plot(_start[0], _start[1], "o", color="red")
plt.plot(_stop[0], _stop[1], "o", color="red")
plt.xticks(xb, ["" for _ in xb])
plt.yticks(yb, ["" for _ in yb])
plt.grid(visible=True)
plt.axis("scaled")
plt.show()

  0%|          | 0/33 [00:00<?, ?it/s]