Add support for Raith MBMS paths (#252)

Also updates pki file.

Closes #255

Signed-off-by: Lucas Heitzmann Gabrielli <[email protected]>

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## Unreleased
+### Added
+- Support for Raith MBMS path data (thanks Matthew Mckee)
+
 ## 0.9.52 - 2024-04-18
 ### Fixed
 - Infinite loop in `Cell::remap_tags` (#246, thanks dtzitkas!)

CMakeLists.txt

@@ -24,7 +24,7 @@ if(CMAKE_PROJECT_NAME STREQUAL PROJECT_NAME)
         set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} /Zi")
         set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /Ox /GL -DNDEBUG")
     else()
-        set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall -Wextra -Wshadow -Wvla -Wformat -Wno-missing-field-initializers -Wno-missing-braces")
+        set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall -Wextra -Wshadow -Wvla -Wformat -Wno-missing-field-initializers -Wno-missing-braces -Wno-cast-function-type -Wno-unused-parameter")
         set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} -g")
         set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} -O3 -DNDEBUG")
     endif()

gdstk/_gdstk.pyi

@@ -1,7 +1,7 @@
 import pathlib
 import datetime
 import sys
-from typing import Optional, Iterable
+from typing import Optional, Iterable, Any
 from collections.abc import Callable, Sequence
 
 if sys.version_info >= (3, 8):
@@ -15,7 +15,7 @@ else:
     from typing_extensions import Self
 
 import numpy
-from numpy.typing import ArrayLike
+from numpy.typing import ArrayLike # type: ignore
 
 class Cell:
     labels: list[Label]
@@ -24,11 +24,11 @@ class Cell:
     polygons: list[Polygon]
     properties: list[list[str | bytes | float]]
     references: list[Reference]
-    def __init__(self, name: str): ...
+    def __init__(self, name: str) -> None: ...
     def add(self, *elements: Polygon | FlexPath | RobustPath | Label | Reference) -> Self: ...
     def area(self, by_spec: bool = False) -> float | dict[tuple[int, int], float]: ...
     def bounding_box(self) -> Optional[tuple[tuple[float, float], tuple[float, float]]]: ...
-    def convex_hull(self) -> numpy.ndarray: ...
+    def convex_hull(self) -> numpy.ndarray[Any, numpy.dtype[numpy.float64]]: ...
     def copy(
         self,
         name: str,
@@ -90,7 +90,7 @@ class Cell:
 
 class Curve:
     tolerance: float
-    def __init__(self, xy: tuple[float, float] | complex, tolerance: float = 0.01): ...
+    def __init__(self, xy: tuple[float, float] | complex, tolerance: float = 0.01) -> None: ...
     def arc(
         self,
         radius: float | tuple[float, float],
@@ -125,7 +125,7 @@ class Curve:
         curve_function: Callable[[float], tuple[float, float] | complex],
         relative: bool = True,
     ) -> Self: ...
-    def points(self) -> numpy.ndarray: ...
+    def points(self) -> numpy.ndarray[Any, numpy.dtype[numpy.float64]]: ...
     def quadratic(
         self, xy: Sequence[tuple[float, float] | complex], relative: bool = False
     ) -> Self: ...
@@ -140,6 +140,27 @@ class Curve:
     def turn(self, radius: float, angle: float) -> Self: ...
     def vertical(self, y: float | Sequence[float], relative: bool = False) -> Self: ...
 
+class RaithData:
+    base_cell_name: str
+    dwelltime_selection: int
+    pitch_parallel_to_path: float
+    pitch_perpendicular_to_path: float
+    pitch_scale: float
+    periods: int
+    grating_type: int
+    dots_per_cycle: int
+    def __init__(
+        self,
+        base_cell_name: str,
+        dwelltime_selection: int,
+        pitch_parallel_to_path: float,
+        pitch_perpendicular_to_path: float,
+        pitch_scale: float,
+        periods: int,
+        grating_type: int,
+        dots_per_cycle: int,
+    ) -> None: ...
+
 class FlexPath:
     bend_function: tuple[
         Optional[Callable[[float, float, float, float], list[tuple[float, float]]]], ...
@@ -165,6 +186,7 @@ class FlexPath:
     simple_path: bool
     size: int
     tolerance: float
+    raith_data: RaithData
     def __init__(
         self,
         points: tuple[float, float] | complex | Sequence[tuple[float, float] | complex],
@@ -200,8 +222,8 @@ class FlexPath:
         scale_width: bool = True,
         layer: int | Sequence[int] = 0,
         datatype: int | Sequence[int] = 0,
-    ): ...
-    def apply_repetition(self) -> list: ...
+    ) -> None: ...
+    def apply_repetition(self) -> list[Self]: ...
     def arc(
         self,
         radius: float | tuple[float, float],
@@ -261,15 +283,15 @@ class FlexPath:
     def mirror(
         self, p1: tuple[float, float] | complex, p2: tuple[float, float] | complex = (0, 0)
     ) -> Self: ...
-    def offsets(self) -> numpy.ndarray: ...
+    def offsets(self) -> numpy.ndarray[Any, numpy.dtype[numpy.float64]]: ...
     def parametric(
         self,
         path_function: Callable[[float], tuple[float, float] | complex],
         width: Optional[float] | Sequence[float] = None,
         offset: Optional[float] | Sequence[float] = None,
         relative: bool = True,
     ) -> Self: ...
-    def path_spines(self) -> list[numpy.ndarray]: ...
+    def path_spines(self) -> list[numpy.ndarray[Any, numpy.dtype[numpy.float64]]]: ...
     def quadratic(
         self,
         xy: Sequence[tuple[float, float] | complex],
@@ -317,7 +339,7 @@ class FlexPath:
     def set_property(
         self, name: str, value: str | bytes | float | Sequence[str | bytes | float]
     ) -> Self: ...
-    def spine(self) -> numpy.ndarray: ...
+    def spine(self) -> numpy.ndarray[Any, numpy.dtype[numpy.float64]]: ...
     def to_polygons(self) -> list[Polygon]: ...
     def translate(
         self, dx: float | tuple[float, float] | complex, dy: Optional[float] = None
@@ -336,7 +358,7 @@ class FlexPath:
         offset: Optional[float] | Sequence[float] = None,
         relative: bool = False,
     ) -> Self: ...
-    def widths(self) -> numpy.ndarray: ...
+    def widths(self) -> numpy.ndarray[Any, numpy.dtype[numpy.float64]]: ...
 
 class GdsWriter:
     def __init__(
@@ -347,8 +369,8 @@ class GdsWriter:
         precision: float = 1e-9,
         max_points: int = 199,
         timestamp: Optional[datetime.datetime] = None,
-    ): ...
-    def close(self): ...
+    ) -> None: ...
+    def close(self) -> None: ...
     def write(self, *cells: Cell | RawCell) -> Self: ...
 
 class Label:
@@ -372,7 +394,7 @@ class Label:
         x_reflection: bool = False,
         layer: int = 0,
         texttype: int = 0,
-    ): ...
+    ) -> None: ...
     def apply_repetition(self) -> list[Self]: ...
     def copy(self) -> Self: ...
     def delete_gds_property(self, attr: int) -> Self: ...
@@ -390,7 +412,9 @@ class Library:
     precision: float
     properties: list[list[str | bytes | float]]
     unit: float
-    def __init__(self, name: str = "library", unit: float = 1e-6, precision: float = 1e-9): ...
+    def __init__(
+        self, name: str = "library", unit: float = 1e-6, precision: float = 1e-9
+    ) -> None: ...
     def add(self, *cells: Cell | RawCell) -> Self: ...
     def delete_property(self, name: str) -> Self: ...
     def get_property(self, name: str) -> Optional[list[list[str | bytes | float]]]: ...
@@ -409,7 +433,7 @@ class Library:
         outfile: str | pathlib.Path,
         max_points: int = 199,
         timestamp: Optional[datetime.datetime] = None,
-    ): ...
+    ) -> None: ...
     def write_oas(
         self,
         outfile: str | pathlib.Path,
@@ -419,7 +443,7 @@ class Library:
         circletolerance: float = 0,
         standard_properties: bool = False,
         validation: Optional[Literal["crc32", "checksum32"]] = None,
-    ): ...
+    ) -> None: ...
 
 class Polygon:
     datatype: int
@@ -430,7 +454,7 @@ class Polygon:
     size: int
     def __init__(
         self, points: Sequence[tuple[float, float] | complex], layer: int = 0, datatype: int = 0
-    ): ...
+    )-> None: ...
     def apply_repetition(self) -> list[Self]: ...
     def area(self) -> float: ...
     def perimeter(self) -> float: ...
@@ -462,7 +486,7 @@ class Polygon:
         x_reflection: bool = False,
         rotation: float = 0,
         translation: Optional[tuple[float, float] | complex] = None,
-        matrix: Optional[ArrayLike] = None,
+        matrix: Optional[ArrayLike] = None, # type: ignore
     ) -> Self: ...
     def translate(
         self, dx: float | tuple[float, float] | complex, dy: Optional[float] = None
@@ -471,7 +495,7 @@ class Polygon:
 class RawCell:
     name: str
     size: int
-    def __init__(self, name: str): ...
+    def __init__(self, name: str)-> None: ...
     def dependencies(self, recursive: bool) -> list[RawCell]: ...
 
 class Reference:
@@ -493,10 +517,10 @@ class Reference:
         columns: int = 1,
         rows: int = 1,
         spacing: Optional[Sequence[float]] = ...,
-    ): ...
+    )-> None: ...
     def apply_repetition(self) -> list[Self]: ...
     def bounding_box(self) -> tuple[tuple[float, float], tuple[float, float]]: ...
-    def convex_hull(self) -> numpy.ndarray: ...
+    def convex_hull(self) -> numpy.ndarray[Any, numpy.dtype[numpy.float64]]: ...
     def copy(self) -> Self: ...
     def delete_gds_property(self, attr: int) -> Self: ...
     def delete_property(self, name: str) -> Self: ...
@@ -531,14 +555,14 @@ class Reference:
 
 class Repetition:
     columns: Optional[int]
-    offsets: Optional[numpy.ndarray]
+    offsets: Optional[numpy.ndarray[Any, numpy.dtype[numpy.float64]]]
     rows: Optional[int]
     size: int
     spacing: Optional[tuple[float, float]]
     v1: Optional[tuple[float, float]]
     v2: Optional[tuple[float, float]]
-    x_offsets: Optional[numpy.ndarray]
-    y_offsets: Optional[numpy.ndarray]
+    x_offsets: Optional[numpy.ndarray[Any, numpy.dtype[numpy.float64]]]
+    y_offsets: Optional[numpy.ndarray[Any, numpy.dtype[numpy.float64]]]
     def __init__(
         self,
         columns: Optional[int] = None,
@@ -549,8 +573,8 @@ class Repetition:
         offsets: Optional[Sequence[tuple[float, float] | complex]] = None,
         x_offsets: Optional[Sequence[float]] = None,
         y_offsets: Optional[Sequence[float]] = None,
-    ): ...
-    def get_offsets(self) -> numpy.ndarray: ...
+    )-> None: ...
+    def get_offsets(self) -> numpy.ndarray[Any, numpy.dtype[numpy.float64]]: ...
 
 class RobustPath:
     datatypes: tuple[int, ...]
@@ -588,7 +612,7 @@ class RobustPath:
         scale_width: bool = True,
         layer: int | list[int] = 0,
         datatype: int | list[int] = 0,
-    ): ...
+    )->None: ...
     def apply_repetition(self) -> list[Self]: ...
     def arc(
         self,
@@ -666,7 +690,7 @@ class RobustPath:
     def delete_property(self, name: str) -> Self: ...
     def get_gds_property(self, attr: int) -> Optional[str]: ...
     def get_property(self, name: str) -> Optional[list[list[str | bytes | float]]]: ...
-    def gradient(self, u: float, from_below: bool = True) -> numpy.ndarray: ...
+    def gradient(self, u: float, from_below: bool = True) -> numpy.ndarray[Any, numpy.dtype[numpy.float64]]: ...
     def horizontal(
         self,
         x: float,
@@ -710,7 +734,7 @@ class RobustPath:
     def mirror(
         self, p1: tuple[float, float] | complex, p2: tuple[float, float] | complex = (0, 0)
     ) -> Self: ...
-    def offsets(self, u: float, from_below: bool = True) -> numpy.ndarray: ...
+    def offsets(self, u: float, from_below: bool = True) -> numpy.ndarray[Any, numpy.dtype[numpy.float64]]: ...
     def parametric(
         self,
         path_function: Callable[[float], tuple[float, float] | complex],
@@ -729,8 +753,8 @@ class RobustPath:
         ] = None,
         relative: bool = True,
     ) -> Self: ...
-    def path_spines(self) -> list[numpy.ndarray]: ...
-    def position(self, u: float, from_below: bool = True) -> numpy.ndarray: ...
+    def path_spines(self) -> list[numpy.ndarray[Any, numpy.dtype[numpy.float64]]]: ...
+    def position(self, u: float, from_below: bool = True) -> numpy.ndarray[Any, numpy.dtype[numpy.float64]]: ...
     def quadratic(
         self,
         xy: Sequence[tuple[float, float] | complex],
@@ -796,7 +820,7 @@ class RobustPath:
     def set_property(
         self, name: str, value: str | bytes | float | Sequence[str | bytes | float]
     ) -> Self: ...
-    def spine(self) -> numpy.ndarray: ...
+    def spine(self) -> numpy.ndarray[Any, numpy.dtype[numpy.float64]]: ...
     def to_polygons(self) -> list[Polygon]: ...
     def translate(
         self, dx: float | tuple[float, float] | complex, dy: Optional[float] = None
@@ -825,7 +849,7 @@ class RobustPath:
         ] = None,
         relative: bool = False,
     ) -> Self: ...
-    def widths(self, u: float, from_below: bool = True) -> numpy.ndarray: ...
+    def widths(self, u: float, from_below: bool = True) -> numpy.ndarray[Any, numpy.dtype[numpy.float64]]: ...
 
 def all_inside(
     points: Sequence[tuple[float, float] | complex],
@@ -860,7 +884,7 @@ def boolean(
     datatype: int = 0,
 ) -> list[Polygon]: ...
 def contour(
-    data: ArrayLike,
+    data: ArrayLike, # type: ignore
     level: int = 0,
     length_scale: float = 1,
     precision: float = 0.01,
@@ -884,7 +908,7 @@ def ellipse(
     layer: int = 0,
     datatype: int = 0,
 ) -> Polygon: ...
-def gds_info(infile: str | pathlib.Path) -> dict: ...
+def gds_info(infile: str | pathlib.Path) -> dict[str, Any]: ...
 
 # def gds_timestamp(filename: str | pathlib.Path, timestamp:Optional[datetime.datetime]=None) -> datetime.datetime: ...
 def gds_units(infile: str | pathlib.Path) -> tuple[float, float]: ...
@@ -929,7 +953,7 @@ def read_gds(
     filter: Optional[Iterable[tuple[int, int]]] = None,
 ) -> Library: ...
 def read_oas(infile: str | pathlib.Path, unit: float = 0, tolerance: float = 0) -> Library: ...
-def read_rawcells(infile: str | pathlib.Path) -> dict: ...
+def read_rawcells(infile: str | pathlib.Path) -> dict[str, RawCell]: ...
 def rectangle(
     corner1: tuple[float, float] | complex,
     corner2: tuple[float, float] | complex,

include/gdstk/flexpath.hpp

@@ -12,14 +12,14 @@ LICENSE file or <http://www.boost.org/LICENSE_1_0.txt>
 #define _USE_MATH_DEFINES
 
 #include <stdint.h>
-#include <string.h>
 
 #include "array.hpp"
 #include "curve.hpp"
 #include "oasis.hpp"
 #include "pathcommon.hpp"
 #include "polygon.hpp"
 #include "property.hpp"
+#include "raithdata.hpp"
 #include "repetition.hpp"
 #include "utils.hpp"
 
@@ -72,6 +72,9 @@ struct FlexPath {
 
     Repetition repetition;
     Property* properties;
+
+    RaithData raith_data;
+
     // Used by the python interface to store the associated PyObject* (if any).
     // No functions in gdstk namespace should touch this value!
     void* owner;

include/gdstk/gdsii.hpp

@@ -78,7 +78,9 @@ enum struct GdsiiRecord : uint8_t {
     ENDMASKS = 0X38,
     LIBDIRSIZE = 0X39,
     SRFNAME = 0X3A,
-    LIBSECUR = 0X3B
+    LIBSECUR = 0X3B,
+    RAITHMBMSPATH = 0x5A,
+    RAITHPXXDATA = 0x62,
 };
 
 enum struct GdsiiDataType : uint8_t {

include/gdstk/gdstk.hpp

@@ -40,6 +40,7 @@ LICENSE file or <http://www.boost.org/LICENSE_1_0.txt>
 #include "oasis.hpp"
 #include "pathcommon.hpp"
 #include "polygon.hpp"
+#include "raithdata.hpp"
 #include "rawcell.hpp"
 #include "reference.hpp"
 #include "repetition.hpp"