jan/masque
1
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

speed up GDS writing

Browse Source 
by preallocating xy array and directly rounding to target dtype
This commit is contained in:
Jan Petykiewicz 2022-06-08 20:55:25 -07:00
parent bfabd351c4
commit 48b28351ed
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
masque/file/gdsii.py
View File

@ -462,16 +462,18 @@ def _shapes_to_elements(
elements.append(path) elements.append(path)
elif isinstance(shape, Polygon): elif isinstance(shape, Polygon):
polygon = shape polygon = shape
xy_open = numpy.round(polygon.vertices + polygon.offset).astype(int) xy_closed = numpy.empty((polygon.vertices.shape[0] + 1, 2), dtype=numpy.int32)
xy_closed = numpy.vstack((xy_open, xy_open[0, :])) numpy.rint(polygon.vertices + polygon.offset, out=xy_closed[:-1], casting='unsafe')
xy_closed[-1] = xy_closed[0]
boundary = klamath.elements.Boundary(layer=(layer, data_type), boundary = klamath.elements.Boundary(layer=(layer, data_type),
xy=xy_closed, xy=xy_closed,
properties=properties) properties=properties)
elements.append(boundary) elements.append(boundary)
else: else:
for polygon in shape.to_polygons(): for polygon in shape.to_polygons():
xy_open = numpy.round(polygon.vertices + polygon.offset).astype(int) xy_closed = numpy.empty((polygon.vertices.shape[0] + 1, 2), dtype=numpy.int32)
xy_closed = numpy.vstack((xy_open, xy_open[0, :])) numpy.rint(polygon.vertices + polygon.offset, out=xy_closed[:-1], casting='unsafe')
xy_closed[-1] = xy_closed[0]
boundary = klamath.elements.Boundary(layer=(layer, data_type), boundary = klamath.elements.Boundary(layer=(layer, data_type),
xy=xy_closed, xy=xy_closed,
properties=properties) properties=properties)