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

avoid unnecessarily nested classes

Browse Source 
Still provide a class variable for convenience
This commit is contained in:
Jan Petykiewicz 2020-05-11 19:03:55 -07:00
parent d33afc2bfd
commit 9bfb3bef9f
1 changed files with 24 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
masque/shapes/path.py
View File

@ -10,6 +10,13 @@ from ..utils import is_scalar, rotation_matrix_2d, vector2, layer_t
from ..utils import remove_colinear_vertices, remove_duplicate_vertices from ..utils import remove_colinear_vertices, remove_duplicate_vertices
class PathCap(Enum):
Flush = 0 # Path ends at final vertices
Circle = 1 # Path extends past final vertices with a semicircle of radius width/2
Square = 2 # Path extends past final vertices with a width-by-width/2 rectangle
SquareCustom = 4 # Path extends past final vertices with a rectangle of length
# defined by path.cap_extensions
class Path(Shape): class Path(Shape):
""" """
@ -21,15 +28,10 @@ class Path(Shape):
__slots__ = ('_vertices', '_width', '_cap', '_cap_extensions') __slots__ = ('_vertices', '_width', '_cap', '_cap_extensions')
_vertices: numpy.ndarray _vertices: numpy.ndarray
_width: float _width: float
_cap: 'Path.Cap'
_cap_extensions: numpy.ndarray or None _cap_extensions: numpy.ndarray or None
_cap: PathCap
class Cap(Enum): Cap = PathCap
Flush = 0 # Path ends at final vertices
Circle = 1 # Path extends past final vertices with a semicircle of radius width/2
Square = 2 # Path extends past final vertices with a width-by-width/2 rectangle
SquareCustom = 4 # Path extends past final vertices with a rectangle of length
# defined by path.cap_extensions
# width property # width property
@property @property
@ -49,17 +51,17 @@ class Path(Shape):
# cap property # cap property
@property @property
def cap(self) -> 'Path.Cap': def cap(self) -> PathCap:
""" """
Path end-cap Path end-cap
""" """
return self._cap return self._cap
@cap.setter @cap.setter
def cap(self, val: 'Path.Cap'): def cap(self, val: PathCap):
# TODO: Document that setting cap can change cap_extensions # TODO: Document that setting cap can change cap_extensions
self._cap = Path.Cap(val) self._cap = PathCap(val)
if self.cap != Path.Cap.SquareCustom: if self.cap != PathCap.SquareCustom:
self.cap_extensions = None self.cap_extensions = None
elif self.cap_extensions is None: elif self.cap_extensions is None:
# just got set to SquareCustom # just got set to SquareCustom
@ -77,8 +79,8 @@ class Path(Shape):
return self._cap_extensions return self._cap_extensions
@cap_extensions.setter @cap_extensions.setter
def cap_extensions(self, vals: numpy.ndarray or None): def cap_extensions(self, vals: Optional[numpy.ndarray]):
custom_caps = (Path.Cap.SquareCustom,) custom_caps = (PathCap.SquareCustom,)
if self.cap in custom_caps: if self.cap in custom_caps:
if vals is None: if vals is None:
raise Exception('Tried to set cap extensions to None on path with custom cap type') raise Exception('Tried to set cap extensions to None on path with custom cap type')
@ -138,7 +140,7 @@ class Path(Shape):
def __init__(self, def __init__(self,
vertices: numpy.ndarray, vertices: numpy.ndarray,
width: float = 0.0, width: float = 0.0,
cap: 'Path.Cap' = Cap.Flush, cap: PathCap = PathCap.Flush,
cap_extensions: numpy.ndarray = None, cap_extensions: numpy.ndarray = None,
offset: vector2 = (0.0, 0.0), offset: vector2 = (0.0, 0.0),
rotation: float = 0, rotation: float = 0,
@ -176,7 +178,7 @@ class Path(Shape):
@staticmethod @staticmethod
def travel(travel_pairs: Tuple[Tuple[float, float]], def travel(travel_pairs: Tuple[Tuple[float, float]],
width: float = 0.0, width: float = 0.0,
cap: 'Path.Cap' = Cap.Flush, cap: PathCap = PathCap.Flush,
cap_extensions = None, cap_extensions = None,
offset: vector2 = (0.0, 0.0), offset: vector2 = (0.0, 0.0),
rotation: float = 0, rotation: float = 0,
@ -285,7 +287,7 @@ class Path(Shape):
polys = [Polygon(offset=self.offset, vertices=verts, dose=self.dose, layer=self.layer)] polys = [Polygon(offset=self.offset, vertices=verts, dose=self.dose, layer=self.layer)]
if self.cap == Path.Cap.Circle: if self.cap == PathCap.Circle:
#for vert in v: # not sure if every vertex, or just ends? #for vert in v: # not sure if every vertex, or just ends?
for vert in [v[0], v[-1]]: for vert in [v[0], v[-1]]:
circ = Circle(offset=vert, radius=self.width / 2, dose=self.dose, layer=self.layer) circ = Circle(offset=vert, radius=self.width / 2, dose=self.dose, layer=self.layer)
@ -294,12 +296,12 @@ class Path(Shape):
return polys return polys
def get_bounds(self) -> numpy.ndarray: def get_bounds(self) -> numpy.ndarray:
if self.cap == Path.Cap.Circle: if self.cap == PathCap.Circle:
bounds = self.offset + numpy.vstack((numpy.min(self.vertices, axis=0) - self.width / 2, bounds = self.offset + numpy.vstack((numpy.min(self.vertices, axis=0) - self.width / 2,
numpy.max(self.vertices, axis=0) + self.width / 2)) numpy.max(self.vertices, axis=0) + self.width / 2))
elif self.cap in (Path.Cap.Flush, elif self.cap in (PathCap.Flush,
Path.Cap.Square, PathCap.Square,
Path.Cap.SquareCustom): PathCap.SquareCustom):
bounds = numpy.array([[+inf, +inf], [-inf, -inf]]) bounds = numpy.array([[+inf, +inf], [-inf, -inf]])
polys = self.to_polygons() polys = self.to_polygons()
for poly in polys: for poly in polys:
@ -383,9 +385,9 @@ class Path(Shape):
return self return self
def _calculate_cap_extensions(self) -> numpy.ndarray: def _calculate_cap_extensions(self) -> numpy.ndarray:
if self.cap == Path.Cap.Square: if self.cap == PathCap.Square:
extensions = numpy.full(2, self.width / 2) extensions = numpy.full(2, self.width / 2)
elif self.cap == Path.Cap.SquareCustom: elif self.cap == PathCap.SquareCustom:
extensions = self.cap_extensions extensions = self.cap_extensions
else: else:
# Flush or Circle # Flush or Circle