add fast-path for 0-degree rotations

4 years ago · c0f0b2190a
parent e1d890b44e
commit c0f0b2190a
2 changed files with 4 additions and 2 deletions
--- a/masque/shapes/path.py
+++ b/masque/shapes/path.py
@ -317,7 +317,8 @@ class Path(Shape, metaclass=AutoSlots):
        return bounds

    def rotate(self, theta: float) -> 'Path':
-        self.vertices = numpy.dot(rotation_matrix_2d(theta), self.vertices.T).T
+        if theta != 0:
+            self.vertices = numpy.dot(rotation_matrix_2d(theta), self.vertices.T).T
        return self

    def mirror(self, axis: int) -> 'Path':
--- a/masque/shapes/polygon.py
+++ b/masque/shapes/polygon.py
@ -266,7 +266,8 @@ class Polygon(Shape, metaclass=AutoSlots):
                             self.offset + numpy.max(self.vertices, axis=0)))

    def rotate(self, theta: float) -> 'Polygon':
-        self.vertices = numpy.dot(rotation_matrix_2d(theta), self.vertices.T).T
+        if theta != 0:
+            self.vertices = numpy.dot(rotation_matrix_2d(theta), self.vertices.T).T
        return self

    def mirror(self, axis: int) -> 'Polygon':