forked from jan/fatamorgana
206 lines
7.5 KiB
Python
206 lines
7.5 KiB
Python
# type: ignore
|
|
|
|
from typing import List, Tuple, Iterable
|
|
from itertools import chain
|
|
from io import BytesIO, BufferedIOBase
|
|
import struct
|
|
|
|
import pytest # type: ignore
|
|
import numpy
|
|
from numpy.testing import assert_equal
|
|
|
|
from .utils import HEADER, FOOTER
|
|
from ..basic import write_uint, write_sint, read_uint, read_sint, write_bstring, write_byte, PathExtensionScheme
|
|
from ..basic import InvalidRecordError, InvalidDataError
|
|
from ..main import OasisLayout
|
|
|
|
|
|
def base_tests(layout: OasisLayout) -> None:
|
|
assert layout.version.string == '1.0'
|
|
assert layout.unit == 1000
|
|
assert layout.validation.checksum_type == 0
|
|
|
|
assert not layout.properties
|
|
assert not layout.propnames
|
|
assert not layout.xnames
|
|
assert not layout.textstrings
|
|
assert not layout.cellnames
|
|
assert not layout.layers
|
|
|
|
assert len(layout.cells) == 1
|
|
assert layout.cells[0].name.string == 'ABC'
|
|
assert not layout.cells[0].properties
|
|
|
|
|
|
def write_file_1(buf: BufferedIOBase) -> BufferedIOBase:
|
|
'''
|
|
'''
|
|
buf.write(HEADER)
|
|
|
|
write_uint(buf, 14) # CELL record (explicit)
|
|
write_bstring(buf, b'ABC') # Cell name
|
|
|
|
# PATH 0
|
|
write_uint(buf, 22) # PATH record
|
|
write_byte(buf, 0b1111_1011) # EWPX_YRDL
|
|
write_uint(buf, 1) # layer
|
|
write_uint(buf, 2) # datatype
|
|
write_uint(buf, 10) # half-width
|
|
write_byte(buf, 0b0000_1111) # extension-scheme 0000_SSEE
|
|
write_sint(buf, 5) # (extension-scheme) start
|
|
write_sint(buf, -5) # (extension-scheme) end
|
|
write_uint(buf, 0) # pointlist: 1-delta, horiz-fisrt
|
|
write_uint(buf, 3) # (pointlist) dimension
|
|
write_sint(buf, 150) # (pointlist)
|
|
write_sint(buf, 50) # (pointlist)
|
|
write_sint(buf, -50) # (pointlist)
|
|
write_sint(buf, 0) # geometry-x (absolute)
|
|
write_sint(buf, 100) # geometry-y (absolute)
|
|
|
|
write_uint(buf, 16) # XYRELATIVE record
|
|
|
|
# PATH 1
|
|
write_uint(buf, 22) # PATH record
|
|
write_byte(buf, 0b1110_1011) # EWPX_YRDL
|
|
write_uint(buf, 1) # layer
|
|
write_uint(buf, 2) # datatype
|
|
write_uint(buf, 10) # half-width
|
|
write_byte(buf, 0b0000_0000) # extension-scheme 0000_SSEE
|
|
write_uint(buf, 0) # pointlist: 1-delta, horiz-fisrt
|
|
write_uint(buf, 3) # (pointlist) dimension
|
|
write_sint(buf, 150) # (pointlist)
|
|
write_sint(buf, 50) # (pointlist)
|
|
write_sint(buf, -50) # (pointlist)
|
|
write_sint(buf, 200) # geometry-y (relative)
|
|
|
|
# PATH 2
|
|
write_uint(buf, 22) # PATH record
|
|
write_byte(buf, 0b1110_1001) # EWPX_YRDL
|
|
write_uint(buf, 1) # layer
|
|
write_uint(buf, 10) # half-width
|
|
write_byte(buf, 0b0000_0100) # extension-scheme 0000_SSEE
|
|
write_uint(buf, 0) # pointlist: 1-delta, horiz-fisrt
|
|
write_uint(buf, 3) # (pointlist) dimension
|
|
write_sint(buf, 150) # (pointlist)
|
|
write_sint(buf, 50) # (pointlist)
|
|
write_sint(buf, -50) # (pointlist)
|
|
write_sint(buf, 200) # geometry-y (relative)
|
|
|
|
# PATH 3
|
|
write_uint(buf, 22) # PATH record
|
|
write_byte(buf, 0b1110_1010) # EWPX_YRDL
|
|
write_uint(buf, 2) # datatype
|
|
write_uint(buf, 12) # half-width
|
|
write_byte(buf, 0b0000_0101) # extension-scheme 0000_SSEE
|
|
write_uint(buf, 0) # pointlist: 1-delta, horiz-fisrt
|
|
write_uint(buf, 3) # (pointlist) dimension
|
|
write_sint(buf, 150) # (pointlist)
|
|
write_sint(buf, 50) # (pointlist)
|
|
write_sint(buf, -50) # (pointlist)
|
|
write_sint(buf, 200) # geometry-y (relative)
|
|
|
|
# PATH 4
|
|
write_uint(buf, 22) # PATH record
|
|
write_byte(buf, 0b1010_1011) # EWPX_YRDL
|
|
write_uint(buf, 1) # layer
|
|
write_uint(buf, 2) # datatype
|
|
write_byte(buf, 0b0000_1010) # extension-scheme 0000_SSEE
|
|
write_uint(buf, 0) # pointlist: 1-delta, horiz-fisrt
|
|
write_uint(buf, 3) # (pointlist) dimension
|
|
write_sint(buf, 150) # (pointlist)
|
|
write_sint(buf, 50) # (pointlist)
|
|
write_sint(buf, -50) # (pointlist)
|
|
write_sint(buf, 200) # geometry-y (relative)
|
|
|
|
# PATH 5
|
|
write_uint(buf, 22) # PATH record
|
|
write_byte(buf, 0b0000_1011) # EWPX_YRDL
|
|
write_uint(buf, 2) # layer
|
|
write_uint(buf, 3) # datatype
|
|
write_sint(buf, 200) # geometry-y (relative)
|
|
|
|
# PATH 6
|
|
write_uint(buf, 22) # PATH record
|
|
write_byte(buf, 0b0000_1111) # EWPX_YRDL
|
|
write_uint(buf, 2) # layer
|
|
write_uint(buf, 3) # datatype
|
|
write_sint(buf, 200) # geometry-y (relative)
|
|
write_uint(buf, 1) # repetition (3x4 matrix)
|
|
write_uint(buf, 1) # (repetition) x-dimension
|
|
write_uint(buf, 2) # (repetition) y-dimension
|
|
write_uint(buf, 200) # (repetition) x-spacing
|
|
write_uint(buf, 300) # (repetition) y-spacing
|
|
|
|
write_uint(buf, 16) # XYRELATIVE record
|
|
|
|
# PATH 7
|
|
write_uint(buf, 22) # PATH record
|
|
write_byte(buf, 0b0001_0101) # EWPX_YRDL
|
|
write_uint(buf, 1) # layer
|
|
write_sint(buf, 1000) # geometry-x (relative)
|
|
write_uint(buf, 0) # repetition (reuse)
|
|
|
|
buf.write(FOOTER)
|
|
return buf
|
|
|
|
|
|
def test_file_1() -> None:
|
|
buf = write_file_1(BytesIO())
|
|
|
|
buf.seek(0)
|
|
layout = OasisLayout.read(buf)
|
|
|
|
base_tests(layout)
|
|
|
|
geometry = layout.cells[0].geometry
|
|
assert len(geometry) == 8
|
|
|
|
for ii, gg in enumerate(geometry):
|
|
msg = f'Failed on path {ii}'
|
|
if ii < 7:
|
|
assert gg.y == 100 + ii * 200, msg
|
|
assert gg.x == 0, msg
|
|
else:
|
|
assert gg.x == 1000, msg
|
|
assert gg.y == 1300, msg
|
|
|
|
if ii < 5:
|
|
assert gg.layer == 1, msg
|
|
assert gg.datatype == 2, msg
|
|
elif ii < 7:
|
|
assert gg.layer == 2, msg
|
|
assert gg.datatype == 3, msg
|
|
else:
|
|
assert gg.layer == 1, msg
|
|
assert gg.datatype == 3, msg
|
|
|
|
if ii < 6:
|
|
assert gg.repetition is None, msg
|
|
elif ii in (7, 8):
|
|
assert gg.repetition.a_count == 3, msg
|
|
assert gg.repetition.b_count == 4, msg
|
|
assert gg.repetition.a_vector == [200, 0], msg
|
|
assert gg.repetition.b_vector == [0, 300], msg
|
|
assert not gg.properties, msg
|
|
|
|
if ii < 3:
|
|
assert gg.half_width == 10, msg
|
|
else:
|
|
assert gg.half_width == 12, msg
|
|
|
|
assert len(gg.point_list) == 3, msg
|
|
assert_equal(gg.point_list, [[150, 0], [0, 50], [-50, 0]], err_msg=msg)
|
|
|
|
if ii >= 4:
|
|
assert gg.extension_start == (PathExtensionScheme.HalfWidth, None)
|
|
assert gg.extension_end == (PathExtensionScheme.HalfWidth, None)
|
|
|
|
assert geometry[0].extension_start == (PathExtensionScheme.Arbitrary, 5)
|
|
assert geometry[1].extension_start == (PathExtensionScheme.Arbitrary, 5)
|
|
assert geometry[2].extension_start == (PathExtensionScheme.Flush, None)
|
|
assert geometry[3].extension_start == (PathExtensionScheme.Flush, None)
|
|
assert geometry[0].extension_end == (PathExtensionScheme.Arbitrary, -5)
|
|
assert geometry[1].extension_end == (PathExtensionScheme.Arbitrary, -5)
|
|
assert geometry[2].extension_end == (PathExtensionScheme.Arbitrary, -5)
|
|
assert geometry[3].extension_end == (PathExtensionScheme.Flush, None)
|