from __future__ import annotations
import gdsfactory as gf
from gdsfactory.component import Component
from gdsfactory.components.bend_circular import bend_circular180
from gdsfactory.components.bend_euler import bend_euler180
from gdsfactory.components.component_sequence import component_sequence
from gdsfactory.components.mmi2x2 import mmi2x2
from gdsfactory.components.straight import straight
from gdsfactory.typings import ComponentSpec, ComponentSpecOrList, CrossSectionSpec
@gf.cell_with_child
def bendu_double(
component: ComponentSpec,
cross_section: CrossSectionSpec = "xs_sc",
bend180: ComponentSpec | ComponentSpecOrList = bend_circular180,
port1: str = "o1",
port2: str = "o2",
) -> ComponentSpec:
"""Returns double bend.
Args:
component: for cutback.
cross_section: specification (CrossSection, string or dict).
bend180: ubend. If a list, each 180 bend is different.
port1: name of first optical port.
port2: name of second optical port.
"""
xs = gf.get_cross_section(cross_section)
xs_r2 = xs.copy(
radius=xs.radius - (component.ports[port1].y - component.ports[port2].y)
)
bendu = gf.Component()
if isinstance(bend180, list):
bend_r = bendu << bend180[0](cross_section=xs)
bend_r2 = bendu << bend180[1](cross_section=xs_r2)
else:
bend_r = bendu << bend180(cross_section=xs)
bend_r2 = bendu << bend180(
cross_section=xs_r2,
)
bend_r2 = bend_r2.move(
origin=(0, 0),
destination=(0, component.ports[port1].y - component.ports[port2].y),
)
bendu.add_port("o1", port=bend_r.ports["o1"])
bendu.add_port("o2", port=bend_r2.ports["o1"])
bendu.add_port("o3", port=bend_r2.ports["o2"])
bendu.add_port("o4", port=bend_r.ports["o2"])
bendu.copy_child_info(component)
return bendu
@gf.cell
def straight_double(
component: ComponentSpec,
cross_section: CrossSectionSpec = "xs_sc",
port1: str = "o1",
port2: str = "o2",
straight_length: float | None = None,
) -> ComponentSpec:
"""Returns double straight.
Args:
component: for cutback.
cross_section: specification (CrossSection, string or dict).
port1: name of first optical port.
port2: name of second optical port.
straight_length: length of straight.
"""
xs = gf.get_cross_section(cross_section)
straight_double = gf.Component()
straight_component = straight(
length=straight_length or xs.radius * 2, cross_section=xs
)
straight_component2 = straight(
length=straight_length or xs.radius * 2, cross_section=xs
)
straight_r = straight_double << straight_component
straight_r2 = straight_double << straight_component2.mirror((1, 0))
straight_r2 = straight_r2.move(
origin=(0, 0),
destination=(0, -component.ports[port1].y + component.ports[port2].y),
)
straight_double.add_port("o1", port=straight_r.ports["o1"])
straight_double.add_port("o2", port=straight_r2.ports["o1"])
straight_double.add_port("o3", port=straight_r2.ports["o2"])
straight_double.add_port("o4", port=straight_r.ports["o2"])
return straight_double
[docs]
@gf.cell
def cutback_2x2(
component: ComponentSpec = mmi2x2,
cols: int = 4,
rows: int = 5,
port1: str = "o1",
port2: str = "o2",
port3: str = "o3",
port4: str = "o4",
bend180: ComponentSpec | ComponentSpecOrList = bend_circular180,
mirror: bool = False,
straight_length: float | None = None,
cross_section: CrossSectionSpec = "xs_sc",
) -> Component:
"""Returns a daisy chain of splitters for measuring their loss.
Args:
component: for cutback.
cols: number of columns.
rows: number of rows.
port1: name of first optical port.
port2: name of second optical port.
bend180: ubend. If a list, a different bend is used for each of the 2 ports.
straight: waveguide spec to connect both sides.
mirror: Flips component. Useful when 'o2' is the port that you want to route to.
straight_length: length of the straight section between cutbacks.
cross_section: specification (CrossSection, string or dict).
"""
component = gf.get_component(component)
bendu = bendu_double(
component=component,
cross_section=cross_section,
bend180=bend180,
port1=port1,
port2=port2,
)
straight = straight_double(
component=component,
cross_section=cross_section,
straight_length=straight_length,
port1=port1,
port2=port2,
)
# Define a map between symbols and (component, input port, output port)
symbol_to_component = {
"A": (component, port1, port3),
"B": (component, port4, port2),
"D": (bendu, "o2", "o3"),
"C": (bendu, "o4", "o1"),
"-": (straight, "o1", "o3"),
"_": (straight, "o2", "o4"),
}
# Generate the sequence of staircases
s = ""
for i in range(rows):
s += "AB" * cols
if mirror:
s += "C" if i % 2 == 0 else "D"
else:
s += "D" if i % 2 == 0 else "C"
s = s[:-1]
s += "-_"
for i in range(rows):
s += "AB" * cols
s += "D" if (i + rows) % 2 == 0 else "C"
s = s[:-1]
n = cols * rows * 2
seq = component_sequence(sequence=s, symbol_to_component=symbol_to_component)
c = gf.Component()
_ = c << seq
c.add_port("o1", port=seq.named_references["A1"]["o1"])
c.add_port("o2", port=seq.named_references["A1"]["o2"])
c.add_port("o3", port=seq.named_references[f"B{n}"]["o2"])
c.add_port("o4", port=seq.named_references[f"B{n}"]["o1"])
c.copy_child_info(component)
c.info["components"] = 2 * n
return c
if __name__ == "__main__":
# c = bendu_double(
# component = gf.get_component(mmi2x2),
# cross_section = "xs_sc",
# bend180 = [bend_circular180, bend_euler180],
# port1 = "o1",
# port2 = "o2",
# )
c = cutback_2x2(
cols=3, rows=2, mirror=True, bend180=[bend_circular180, bend_euler180]
)
cols = range(1, 3)
rows = range(1, 3)
cs = [cutback_2x2(cols=col, rows=row) for col in cols for row in rows]
ncomponent_expected = [4 * col * row for col in cols for row in rows]
ncomponents = [c.info["components"] for c in cs]
print(ncomponents, ncomponent_expected)
c.show(show_ports=True)
