From 3a73fb1d60502a20e00f4558f9f7a89e223c8685 Mon Sep 17 00:00:00 2001
From: Jan Petykiewicz <jan@mpxd.net>
Date: Fri, 25 Apr 2025 00:50:57 -0700
Subject: [PATCH] [PolyCollection] many fixes

---
 masque/shapes/__init__.py        |  1 +
 masque/shapes/poly_collection.py | 12 ++++++++----
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/masque/shapes/__init__.py b/masque/shapes/__init__.py
index 8ad46ef..fd66c59 100644
--- a/masque/shapes/__init__.py
+++ b/masque/shapes/__init__.py
@@ -10,6 +10,7 @@ from .shape import (
     )
 
 from .polygon import Polygon as Polygon
+from .poly_collection import PolyCollection as PolyCollection
 from .circle import Circle as Circle
 from .ellipse import Ellipse as Ellipse
 from .arc import Arc as Arc
diff --git a/masque/shapes/poly_collection.py b/masque/shapes/poly_collection.py
index 5d2ba49..bd2c23c 100644
--- a/masque/shapes/poly_collection.py
+++ b/masque/shapes/poly_collection.py
@@ -55,7 +55,11 @@ class PolyCollection(Shape):
         """
         Iterator which provides slices which index vertex_lists
         """
-        for ii, ff in chain(self._vertex_offsets, (self._vertex_lists.shape[0],)):
+        for ii, ff in zip(
+                self._vertex_offsets,
+                chain(self._vertex_offsets, (self._vertex_lists.shape[0],)),
+                strict=True,
+                ):
             yield slice(ii, ff)
 
     @property
@@ -144,7 +148,7 @@ class PolyCollection(Shape):
         return [Polygon(
             vertices = vv,
             offset = self.offset,
-            repetition = self.repetition.copy(),
+            repetition = copy.deepcopy(self.repetition),
             annotations = copy.deepcopy(self.annotations),
             ) for vv in self.polygon_vertices]
 
@@ -155,7 +159,7 @@ class PolyCollection(Shape):
     def rotate(self, theta: float) -> Self:
         if theta != 0:
             rot = rotation_matrix_2d(theta)
-            self._vertex_lists = numpy.einsum('ij,kj->ki', rot, self._vertex_lists_)
+            self._vertex_lists = numpy.einsum('ij,kj->ki', rot, self._vertex_lists)
         return self
 
     def mirror(self, axis: int = 0) -> Self:
@@ -163,7 +167,7 @@ class PolyCollection(Shape):
         return self
 
     def scale_by(self, c: float) -> Self:
-        self.vertex_lists *= c
+        self._vertex_lists *= c
         return self
 
     def normalized_form(self, norm_value: float) -> normalized_shape_tuple: