Skip to content

Commit 033eb80

Browse files
committed
Allow multiple children cells with equal names
1 parent 758e6d1 commit 033eb80

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

gdshelpers/geometry/chip.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -214,11 +214,13 @@ def get_desc(self):
214214
"""
215215
def walk_cells(cell, out_dict):
216216
if cell.name not in out_dict:
217-
out_dict[cell.name] = {
218-
'cells': {child['cell'].name: dict(offset=tuple(child['origin']),
219-
angle=child['angle'] or 0) for child in cell.cells},
220-
**cell.desc
221-
}
217+
cellrefs = []
218+
for child in cell.cells:
219+
child_dict = child.copy()
220+
child_dict['cell'] = child_dict['cell'].name
221+
cellrefs.append(child_dict)
222+
223+
out_dict[cell.name] = {'cells': cellrefs, **cell.desc}
222224

223225
for child in cell.cells:
224226
walk_cells(child['cell'], out_dict)

gdshelpers/tests/test_chip.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,3 +158,4 @@ def test_desc(self):
158158
self.assertTrue("desctest" in d["cells"])
159159
self.assertEqual("desctest", d["root"])
160160
self.assertEqual(2, len(d["cells"]["desctest"]["cells"]))
161+
self.assertEqual("child1", d["cells"]["desctest"]["cells"][0]["cell"])

0 commit comments

Comments
 (0)