272 lines
9.4 KiB
Python
272 lines
9.4 KiB
Python
import pytest
|
|
import numpy
|
|
from numpy import pi
|
|
from masque import Pather, RenderPather, Library, Pattern, Port
|
|
from masque.builder.tools import PathTool
|
|
from masque.error import BuildError
|
|
|
|
def test_pather_trace_basic() -> None:
|
|
lib = Library()
|
|
tool = PathTool(layer='M1', width=1000)
|
|
p = Pather(lib, tools=tool)
|
|
|
|
# Port rotation 0 points in +x (INTO device).
|
|
# To extend it, we move in -x direction.
|
|
p.pattern.ports['A'] = Port((0, 0), rotation=0)
|
|
|
|
# Trace single port
|
|
p.at('A').trace(None, 5000)
|
|
assert numpy.allclose(p.pattern.ports['A'].offset, (-5000, 0))
|
|
|
|
# Trace with bend
|
|
p.at('A').trace(True, 5000) # CCW bend
|
|
# Port was at (-5000, 0) rot 0.
|
|
# New wire starts at (-5000, 0) rot 0.
|
|
# Output port of wire before rotation: (5000, 500) rot -pi/2
|
|
# Rotate by pi (since dev port rot is 0 and tool port rot is 0):
|
|
# (-5000, -500) rot pi - pi/2 = pi/2
|
|
# Add to start: (-10000, -500) rot pi/2
|
|
assert numpy.allclose(p.pattern.ports['A'].offset, (-10000, -500))
|
|
assert p.pattern.ports['A'].rotation is not None
|
|
assert numpy.isclose(p.pattern.ports['A'].rotation, pi/2)
|
|
|
|
def test_pather_trace_to() -> None:
|
|
lib = Library()
|
|
tool = PathTool(layer='M1', width=1000)
|
|
p = Pather(lib, tools=tool)
|
|
|
|
p.pattern.ports['A'] = Port((0, 0), rotation=0)
|
|
|
|
# Trace to x=-10000
|
|
p.at('A').trace_to(None, x=-10000)
|
|
assert numpy.allclose(p.pattern.ports['A'].offset, (-10000, 0))
|
|
|
|
# Trace to position=-20000
|
|
p.at('A').trace_to(None, p=-20000)
|
|
assert numpy.allclose(p.pattern.ports['A'].offset, (-20000, 0))
|
|
|
|
def test_pather_bundle_trace() -> None:
|
|
lib = Library()
|
|
tool = PathTool(layer='M1', width=1000)
|
|
p = Pather(lib, tools=tool)
|
|
|
|
p.pattern.ports['A'] = Port((0, 0), rotation=0)
|
|
p.pattern.ports['B'] = Port((0, 2000), rotation=0)
|
|
|
|
# Straight bundle - all should align to same x
|
|
p.at(['A', 'B']).straight(xmin=-10000)
|
|
assert numpy.isclose(p.pattern.ports['A'].offset[0], -10000)
|
|
assert numpy.isclose(p.pattern.ports['B'].offset[0], -10000)
|
|
|
|
# Bundle with bend
|
|
p.at(['A', 'B']).ccw(xmin=-20000, spacing=2000)
|
|
# Traveling in -x direction. CCW turn turns towards -y.
|
|
# A is at y=0, B is at y=2000.
|
|
# Rotation center is at y = -R.
|
|
# A is closer to center than B. So A is inner, B is outer.
|
|
# xmin is coordinate of innermost bend (A).
|
|
assert numpy.isclose(p.pattern.ports['A'].offset[0], -20000)
|
|
# B's bend is further out (more negative x)
|
|
assert numpy.isclose(p.pattern.ports['B'].offset[0], -22000)
|
|
|
|
def test_pather_each_bound() -> None:
|
|
lib = Library()
|
|
tool = PathTool(layer='M1', width=1000)
|
|
p = Pather(lib, tools=tool)
|
|
|
|
p.pattern.ports['A'] = Port((0, 0), rotation=0)
|
|
p.pattern.ports['B'] = Port((-1000, 2000), rotation=0)
|
|
|
|
# Each should move by 5000 (towards -x)
|
|
p.at(['A', 'B']).trace(None, each=5000)
|
|
assert numpy.allclose(p.pattern.ports['A'].offset, (-5000, 0))
|
|
assert numpy.allclose(p.pattern.ports['B'].offset, (-6000, 2000))
|
|
|
|
def test_selection_management() -> None:
|
|
lib = Library()
|
|
p = Pather(lib)
|
|
p.pattern.ports['A'] = Port((0, 0), rotation=0)
|
|
p.pattern.ports['B'] = Port((0, 0), rotation=0)
|
|
|
|
pp = p.at('A')
|
|
assert pp.ports == ['A']
|
|
|
|
pp.select('B')
|
|
assert pp.ports == ['A', 'B']
|
|
|
|
pp.deselect('A')
|
|
assert pp.ports == ['B']
|
|
|
|
pp.select(['A'])
|
|
assert pp.ports == ['B', 'A']
|
|
|
|
pp.drop()
|
|
assert 'A' not in p.pattern.ports
|
|
assert 'B' not in p.pattern.ports
|
|
assert pp.ports == []
|
|
|
|
def test_mark_fork() -> None:
|
|
lib = Library()
|
|
p = Pather(lib)
|
|
p.pattern.ports['A'] = Port((100, 200), rotation=1)
|
|
|
|
pp = p.at('A')
|
|
pp.mark('B')
|
|
assert 'B' in p.pattern.ports
|
|
assert numpy.allclose(p.pattern.ports['B'].offset, (100, 200))
|
|
assert p.pattern.ports['B'].rotation == 1
|
|
assert pp.ports == ['A'] # mark keeps current selection
|
|
|
|
pp.fork('C')
|
|
assert 'C' in p.pattern.ports
|
|
assert pp.ports == ['C'] # fork switches to new name
|
|
|
|
def test_rename() -> None:
|
|
lib = Library()
|
|
p = Pather(lib)
|
|
p.pattern.ports['A'] = Port((0, 0), rotation=0)
|
|
|
|
p.at('A').rename('B')
|
|
assert 'A' not in p.pattern.ports
|
|
assert 'B' in p.pattern.ports
|
|
|
|
p.pattern.ports['C'] = Port((0, 0), rotation=0)
|
|
pp = p.at(['B', 'C'])
|
|
pp.rename({'B': 'D', 'C': 'E'})
|
|
assert 'B' not in p.pattern.ports
|
|
assert 'C' not in p.pattern.ports
|
|
assert 'D' in p.pattern.ports
|
|
assert 'E' in p.pattern.ports
|
|
assert set(pp.ports) == {'D', 'E'}
|
|
|
|
def test_renderpather_uturn_fallback() -> None:
|
|
lib = Library()
|
|
tool = PathTool(layer='M1', width=1000)
|
|
rp = RenderPather(lib, tools=tool)
|
|
rp.pattern.ports['A'] = Port((0, 0), rotation=0)
|
|
|
|
# PathTool doesn't implement planU, so it should fall back to two planL calls
|
|
rp.at('A').uturn(offset=10000, length=5000)
|
|
|
|
# Two steps should be added
|
|
assert len(rp.paths['A']) == 2
|
|
assert rp.paths['A'][0].opcode == 'L'
|
|
assert rp.paths['A'][1].opcode == 'L'
|
|
|
|
rp.render()
|
|
assert rp.pattern.ports['A'].rotation is not None
|
|
assert numpy.isclose(rp.pattern.ports['A'].rotation, pi)
|
|
|
|
def test_autotool_uturn() -> None:
|
|
from masque.builder.tools import AutoTool
|
|
lib = Library()
|
|
|
|
# Setup AutoTool with a simple straight and a bend
|
|
def make_straight(length: float) -> Pattern:
|
|
pat = Pattern()
|
|
pat.rect(layer='M1', xmin=0, xmax=length, yctr=0, ly=1000)
|
|
pat.ports['in'] = Port((0, 0), 0)
|
|
pat.ports['out'] = Port((length, 0), pi)
|
|
return pat
|
|
|
|
bend_pat = Pattern()
|
|
bend_pat.polygon(layer='M1', vertices=[(0, -500), (0, 500), (1000, -500)])
|
|
bend_pat.ports['in'] = Port((0, 0), 0)
|
|
bend_pat.ports['out'] = Port((500, -500), pi/2)
|
|
lib['bend'] = bend_pat
|
|
|
|
tool = AutoTool(
|
|
straights=[AutoTool.Straight(ptype='wire', fn=make_straight, in_port_name='in', out_port_name='out')],
|
|
bends=[AutoTool.Bend(abstract=lib.abstract('bend'), in_port_name='in', out_port_name='out', clockwise=True)],
|
|
sbends=[],
|
|
transitions={},
|
|
default_out_ptype='wire'
|
|
)
|
|
|
|
p = Pather(lib, tools=tool)
|
|
p.pattern.ports['A'] = Port((0, 0), 0)
|
|
|
|
# CW U-turn (jog < 0)
|
|
# R = 500. jog = -2000. length = 1000.
|
|
# p0 = planL(length=1000) -> out at (1000, -500) rot pi/2
|
|
# R2 = 500.
|
|
# l2_length = abs(-2000) - abs(-500) - 500 = 1000.
|
|
p.at('A').uturn(offset=-2000, length=1000)
|
|
|
|
# Final port should be at (-1000, 2000) rot pi
|
|
# Start: (0,0) rot 0. Wire direction is rot + pi = pi (West, -x).
|
|
# Tool planU returns (length, jog) = (1000, -2000) relative to (0,0) rot 0.
|
|
# Rotation of pi transforms (1000, -2000) to (-1000, 2000).
|
|
# Final rotation: 0 + pi = pi.
|
|
assert numpy.allclose(p.pattern.ports['A'].offset, (-1000, 2000))
|
|
assert p.pattern.ports['A'].rotation is not None
|
|
assert numpy.isclose(p.pattern.ports['A'].rotation, pi)
|
|
|
|
def test_pather_trace_into() -> None:
|
|
lib = Library()
|
|
tool = PathTool(layer='M1', width=1000)
|
|
p = Pather(lib, tools=tool)
|
|
|
|
# 1. Straight connector
|
|
p.pattern.ports['A'] = Port((0, 0), rotation=0)
|
|
p.pattern.ports['B'] = Port((-10000, 0), rotation=pi)
|
|
p.at('A').trace_into('B', plug_destination=False)
|
|
assert 'B' in p.pattern.ports
|
|
assert 'A' in p.pattern.ports
|
|
assert numpy.allclose(p.pattern.ports['A'].offset, (-10000, 0))
|
|
|
|
# 2. Single bend
|
|
p.pattern.ports['C'] = Port((0, 0), rotation=0)
|
|
p.pattern.ports['D'] = Port((-5000, 5000), rotation=pi/2)
|
|
p.at('C').trace_into('D', plug_destination=False)
|
|
assert 'D' in p.pattern.ports
|
|
assert 'C' in p.pattern.ports
|
|
assert numpy.allclose(p.pattern.ports['C'].offset, (-5000, 5000))
|
|
|
|
# 3. Jog (S-bend)
|
|
p.pattern.ports['E'] = Port((0, 0), rotation=0)
|
|
p.pattern.ports['F'] = Port((-10000, 2000), rotation=pi)
|
|
p.at('E').trace_into('F', plug_destination=False)
|
|
assert 'F' in p.pattern.ports
|
|
assert 'E' in p.pattern.ports
|
|
assert numpy.allclose(p.pattern.ports['E'].offset, (-10000, 2000))
|
|
|
|
# 4. U-bend (0 deg angle)
|
|
p.pattern.ports['G'] = Port((0, 0), rotation=0)
|
|
p.pattern.ports['H'] = Port((-10000, 2000), rotation=0)
|
|
p.at('G').trace_into('H', plug_destination=False)
|
|
assert 'H' in p.pattern.ports
|
|
assert 'G' in p.pattern.ports
|
|
# A U-bend with length=-travel=10000 and jog=-2000 from (0,0) rot 0
|
|
# ends up at (-10000, 2000) rot pi.
|
|
assert numpy.allclose(p.pattern.ports['G'].offset, (-10000, 2000))
|
|
assert p.pattern.ports['G'].rotation is not None
|
|
assert numpy.isclose(p.pattern.ports['G'].rotation, pi)
|
|
|
|
|
|
def test_pather_jog_failed_fallback_is_atomic() -> None:
|
|
lib = Library()
|
|
tool = PathTool(layer='M1', width=2, ptype='wire')
|
|
p = Pather(lib, tools=tool)
|
|
p.pattern.ports['A'] = Port((0, 0), rotation=0, ptype='wire')
|
|
|
|
with pytest.raises(BuildError, match='shorter than required bend'):
|
|
p.jog('A', 1.5, length=5)
|
|
|
|
assert numpy.allclose(p.pattern.ports['A'].offset, (0, 0))
|
|
assert p.pattern.ports['A'].rotation == 0
|
|
assert len(p.paths['A']) == 0
|
|
|
|
|
|
def test_pather_uturn_failed_fallback_is_atomic() -> None:
|
|
lib = Library()
|
|
tool = PathTool(layer='M1', width=2, ptype='wire')
|
|
p = Pather(lib, tools=tool)
|
|
p.pattern.ports['A'] = Port((0, 0), rotation=0, ptype='wire')
|
|
|
|
with pytest.raises(BuildError, match='shorter than required bend'):
|
|
p.uturn('A', 1.5, length=0)
|
|
|
|
assert numpy.allclose(p.pattern.ports['A'].offset, (0, 0))
|
|
assert p.pattern.ports['A'].rotation == 0
|
|
assert len(p.paths['A']) == 0
|