DeviceLib/trees.py at master · tpolakovic/DeviceLib · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
#!/usr/bin/env python3

from phidl import Device, CrossSection
import phidl.geometry as pg
import phidl.path as pp
import phidl.routing as pr

def _branch_start(width=2, size=10, layer=2):
    D = Device()
    P1 = pp.Path()
    P1.append( pp.euler(angle=90,radius=size/2, use_eff=True) )
    P1.append( pp.euler(angle=-90,radius=size/2, use_eff=True) )
    P2 = pp.Path()
    P2.append( pp.euler(angle=-90,radius=size/2, use_eff=True) )
    P2.append( pp.euler(angle=90,radius=size/2, use_eff=True) )
    XS = CrossSection()
    XS.add(width=width, ports=('in', 'out'))
    bend1 = XS.extrude(P1)
    bend2 = XS.extrude(P2)
    D.add_ref([bend1,bend2])
    D.add_port(name='in', port=bend1.ports['in'])
    D.add_port(name='out1', port=bend1.ports['out'])
    D.add_port(name='out2', port=bend2.ports['out'])
    out = pg.boolean(D, None, operation = 'or', layer=layer)
    out.add_port(name='in', port=D.ports['in'])
    out.add_port(name='out1', port=D.ports['out1'])
    out.add_port(name='out2', port=D.ports['out2'])
    return out

def _choke(width, choke_width, choke_length, layer):
    CD = Device()
    C = pg.optimal_step(start_width=width, end_width=choke_width, symmetric=True, layer=layer)
    R = pg.compass((choke_length, choke_width), layer=layer)
    choke1 = CD.add_ref(C)
    choke2 = CD.add_ref(C)
    choke_bridge = CD.add_ref(R)
    choke_bridge.connect('W', choke1.ports[2])
    choke2.connect(2, choke_bridge.ports['E'])
    CD.add_port('in', port=choke1.ports[1])
    CD.add_port('out', port=choke2.ports[1])
    CN = pg.connector(midpoint=choke_bridge.center, width=choke_length, orientation=90)
    chan_ports = CD.add_ref(CN)
    CD.add_port(name='L', port=chan_ports.ports[1])
    CD.add_port(name='R', port=chan_ports.ports[2])
    CD.flatten()
    return CD

def _tree_branch(width=5, pitch=50, choke_width=1, choke_length=2, choke_dist=10, layer=2,
               layer_channel=12, negative=True, trench=2):
    if not negative:
        trench = 0
    size = pitch/2
    D = Device()
    B = _branch_start(width=width, size=size, layer=layer)
    branch = D.add_ref(B)

    C = _choke(width, choke_width, choke_length, layer)
    c_len = C.xsize

    choke1 = D.add_ref(C)
    choke1.connect('in', branch.ports['out1'])
    R1 = pg.straight(size=(width,c_len), layer=layer)
    rec1 = D.add_ref(R1)
    rec1.connect(1, branch.ports['out2'])

    R2 = pg.straight(size=(width, choke_dist), layer=layer)
    cbridge1 = D.add_ref(R2)
    cbridge2 = D.add_ref(R2)
    cbridge1.connect(1, choke1.ports['out'])
    cbridge2.connect(1, rec1.ports[2])

    choke2 = D.add_ref(C)
    choke2.connect('in', cbridge2.ports[2])
    rec2 = D.add_ref(R1)
    rec2.connect(1, cbridge1.ports[2])

    D.add_port(name='in', port=branch.ports['in'])
    D.add_port(name='out1', port=rec2.ports[2])
    D.add_port(name='out2', port=choke2.ports['out'])
    l1, r1 = choke1.ports['L'], choke1.ports['R']
    l2, r2 = choke2.ports['L'], choke2.ports['R']

    if negative:
        D = pg.outline(D, distance=trench, open_ports=trench+1, layer=layer)

    RS = pg.straight(size=(choke_length, width/2+trench), layer=layer_channel)
    ch1 = D.add_ref(RS)
    ch1.connect(1, l1)
    ch2 = D.add_ref(RS)
    ch2.connect(1, r2)
    RL = pg.straight(size=(choke_length, 2*size + width/2 + trench), layer=layer_channel)
    ch3 = D.add_ref(RL)
    ch3.connect(1, r1)
    ch4 = D.add_ref(RL)
    ch4.connect(1, l2)

    D.add_port(name='L1', port=ch1.ports[2])
    D.add_port(name='L2', port=ch4.ports[2])
    D.add_port(name='R1', port=ch3.ports[2])
    D.add_port(name='R2', port=ch2.ports[2])

    D.flatten()
    return D


def tree(n_levels = 4, width=5, pitch=50, choke_width=1, choke_length=2, choke_dist=10, layer=2,
               layer_channel=12, negative=True, trench=2):
    D = Device()
    branches = []
    for i in range(n_levels):
        _pitch = 2**(n_levels-i)*pitch
        branches.append(
            _tree_branch(width, _pitch, choke_width, choke_length, choke_dist, layer, layer_channel, negative, trench)
        )

    levels = []
    for i in range(n_levels):
        devices_in_level = 2**i
        devices = []
        for _ in range(devices_in_level):
            devices.append(D.add_ref(branches[i]))
        levels.append(devices)

    for i in range(1, n_levels):
        devices_in_level = 2**i
        for j in range(0,devices_in_level, 2):
            levels[i][j].connect('in', levels[i-1][j//2].ports['out1'])
            levels[i][j+1].connect('in', levels[i-1][j//2].ports['out2'])

    for (i, level) in enumerate(levels):
        devices_in_level = 2**i
        for j in range(devices_in_level-1):
            PR1 = pr.route_quad(level[j+1].ports['L1'], level[j].ports['R1'], layer=layer_channel)
            PR2 = pr.route_quad(level[j+1].ports['L2'], level[j].ports['R2'], layer=layer_channel)
            D.add(PR1)
            D.add(PR2)
    D.add_port(name='in', port=levels[0][0].ports['in'])
    for (i, device) in enumerate(levels[-1]):
        D.add_port(name='out_{}'.format(2*i), port=device.ports['out1'])
        D.add_port(name='out_{}'.format(2*i+1), port=device.ports['out2'])

    for (i, level) in enumerate(levels):
        D.add_port(name='L1_{}'.format(i), port=level[0].ports['L1'])
        D.add_port(name='R1_{}'.format(i), port=level[0].ports['L2'])
        D.add_port(name='L2_{}'.format(i), port=level[-1].ports['R1'])
        D.add_port(name='R2_{}'.format(i), port=level[-1].ports['R2'])
    return D