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

use ArrayLike when accepting pseudo-ndarrays

Browse source
This commit is contained in:
Jan Petykiewicz 2021-06-22 18:59:13 -07:00
commit 4796676a4e
7 changed files with 31 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

9
masque/shapes/shape.py
View file

@ -2,6 +2,7 @@ from typing import List, Tuple, Callable, TypeVar, Optional, TYPE_CHECKING
from abc import ABCMeta, abstractmethod
import numpy # type: ignore
from numpy.typing import ArrayLike
from ..traits import (PositionableImpl, LayerableImpl, DoseableImpl,
Rotatable, Mirrorable, Copyable, Scalable,
@ -93,8 +94,8 @@ class Shape(PositionableImpl, LayerableImpl, DoseableImpl, Rotatable, Mirrorable
---- Non-abstract methods
'''
def manhattanize_fast(self,
grid_x: numpy.ndarray,
grid_y: numpy.ndarray,
grid_x: ArrayLike,
grid_y: ArrayLike,
) -> List['Polygon']:
"""
Returns a list of polygons with grid-aligned ("Manhattan") edges approximating the shape.
@ -200,8 +201,8 @@ class Shape(PositionableImpl, LayerableImpl, DoseableImpl, Rotatable, Mirrorable
return manhattan_polygons
def manhattanize(self,
grid_x: numpy.ndarray,
grid_y: numpy.ndarray
grid_x: ArrayLike,
grid_y: ArrayLike,
) -> List['Polygon']:
"""
Returns a list of polygons with grid-aligned ("Manhattan") edges approximating the shape.