add rotation interface for basic nodes

KicadModTree/PolygonPoints.py

@@ -143,6 +143,11 @@ def cut(self, other):
 
         self.nodes.insert(idx_self+1, other[idx_other])
 
+    def rotate(self, angle, origin=(0, 0), use_degrees=True):
+        for p in self.nodes:
+            p.rotate(angle=angle, origin=origin, use_degrees=use_degrees)
+        return self
+
     def __iter__(self):
         for n in self.nodes:
             yield n

KicadModTree/Vector.py

@@ -191,21 +191,19 @@ def __iter__(self):
     def __copy__(self):
         return Vector2D(self.x, self.y)
 
-    def rotate(self, angle, origin=(0, 0), apply_on_copy=False, use_degrees=False):
-        point = self if not apply_on_copy else copy(self)
-
+    def rotate(self, angle, origin=(0, 0), use_degrees=True):
         op = Vector2D(origin)
 
         if use_degrees:
             angle = radians(angle)
 
-        temp = op.x + cos(angle) * (point.x - op.x) - sin(angle) * (point.y - op.y)
-        point.y = op.y + sin(angle) * (point.x - op.x) + cos(angle) * (point.y - op.y)
-        point.x = temp
+        temp = op.x + cos(angle) * (self.x - op.x) - sin(angle) * (self.y - op.y)
+        self.y = op.y + sin(angle) * (self.x - op.x) + cos(angle) * (self.y - op.y)
+        self.x = temp
 
-        return point
+        return self
 
-    def to_polar(self, origin=(0, 0), use_degrees=False):
+    def to_polar(self, origin=(0, 0), use_degrees=True):
         op = Vector2D(origin)
 
         diff = self - op
@@ -218,7 +216,7 @@ def to_polar(self, origin=(0, 0), use_degrees=False):
         return (radius, angle)
 
     @staticmethod
-    def from_polar(radius, angle, origin=(0, 0), use_degrees=False):
+    def from_polar(radius, angle, origin=(0, 0), use_degrees=True):
         if use_degrees:
             angle = radians(angle)
 

KicadModTree/nodes/base/Arc.py

@@ -118,6 +118,11 @@ def _initFromCenterAndEnd(self, **kwargs):
         else:
             raise KeyError('Arcs defined with center and endpoint must define the start point.')
 
+    def rotate(self, angle, origin=(0, 0), use_degrees=True):
+        self.center_pos.rotate(angle=angle, origin=origin, use_degrees=use_degrees)
+        self.start_pos.rotate(angle=angle, origin=origin, use_degrees=use_degrees)
+        return self
+
     def calculateBoundingBox(self):
         # TODO: finish implementation
         min_x = min(self.start_pos.x, self._calulateEndPos().x)

KicadModTree/nodes/base/Circle.py

@@ -44,11 +44,19 @@ def __init__(self, **kwargs):
         self.center_pos = Vector2D(kwargs['center'])
         self.radius = kwargs['radius']
 
-        self.end_pos = Vector2D([self.center_pos.x+self.radius, self.center_pos.y])
+        self._calcEndPos()
 
         self.layer = kwargs.get('layer', 'F.SilkS')
         self.width = kwargs.get('width')
 
+    def _calcEndPos(self):
+        self.end_pos = Vector2D([self.center_pos.x+self.radius, self.center_pos.y])
+
+    def rotate(self, angle, origin=(0, 0), use_degrees=True):
+        self.center_pos.rotate(angle=angle, origin=origin, use_degrees=use_degrees)
+        self._calcEndPos()
+        return self
+
     def calculateBoundingBox(self):
         min_x = self.center_pos.x-self.radius
         min_y = self.center_pos.y-self.radius

KicadModTree/nodes/base/Line.py

@@ -47,6 +47,11 @@ def __init__(self, **kwargs):
         self.layer = kwargs.get('layer', 'F.SilkS')
         self.width = kwargs.get('width')
 
+    def rotate(self, angle, origin=(0, 0), use_degrees=True):
+        self.start_pos.rotate(angle=angle, origin=origin, use_degrees=use_degrees)
+        self.end_pos.rotate(angle=angle, origin=origin, use_degrees=use_degrees)
+        return self
+
     def calculateBoundingBox(self):
         render_start_pos = self.getRealPosition(self.start_pos)
         render_end_pos = self.getRealPosition(self.end_pos)

KicadModTree/nodes/base/Pad.py

@@ -241,6 +241,14 @@ def _initShapeInZone(self, **kwargs):
             raise ValueError('{shape} is an illegal specifier for the shape in zone option'
                              .format(shape=self.shape_in_zone))
 
+    def rotate(self, angle, origin=(0, 0), use_degrees=True):
+        self.at.rotate(angle=angle, origin=origin, use_degrees=use_degrees)
+        a = angle if use_degrees else math.degrees(angle)
+
+        # subtraction because kicad text field rotation is the wrong way round
+        self.rotation -= a
+        return self
+
     # calculate the outline of a pad
     def calculateBoundingBox(self):
         return Node.calculateBoundingBox(self)

KicadModTree/nodes/base/Polygon.py

@@ -49,6 +49,10 @@ def __init__(self, **kwargs):
         self.layer = kwargs.get('layer', 'F.SilkS')
         self.width = kwargs.get('width')
 
+    def rotate(self, angle, origin=(0, 0), use_degrees=True):
+        self.nodes.rotate(angle=angle, origin=origin, use_degrees=use_degrees)
+        return self
+
     def calculateBoundingBox(self):
         return nodes.calculateBoundingBox()
 

KicadModTree/nodes/base/Text.py

@@ -48,9 +48,15 @@ class Text(Node):
     >>> Text(type='value', text="footprint name", at=[0, 3], layer='F.Fab')
     """
 
+    TYPE_REFERENCE = 'reference'
+    TYPE_VALUE = 'value'
+    TYPE_USER = 'user'
+    _TYPES = [TYPE_REFERENCE, TYPE_VALUE, TYPE_USER]
+
     def __init__(self, **kwargs):
         Node.__init__(self)
-        self.type = kwargs['type']
+        self._initType(**kwargs)
+
         self.text = kwargs['text']
         self.at = Vector2D(kwargs['at'])
         self.rotation = kwargs.get('rotation', 0)
@@ -61,6 +67,19 @@ def __init__(self, **kwargs):
 
         self.hide = kwargs.get('hide', False)
 
+    def _initType(self, **kwargs):
+        self.type = kwargs['type']
+        if self.type not in Text._TYPES:
+            raise ValueError('Illegal type selected for text field.')
+
+    def rotate(self, angle, origin=(0, 0), use_degrees=True):
+        self.at.rotate(angle=angle, origin=origin, use_degrees=use_degrees)
+        a = angle if use_degrees else math.degrees(angle)
+
+        # subtraction because kicad text field rotation is the wrong way round
+        self.rotation -= a
+        return self
+
     def calculateBoundingBox(self):
         width = len(self.text)*self.size['x']
         height = self.size['y']

KicadModTree/nodes/specialized/RingPad.py

@@ -143,7 +143,7 @@ def _generatePads(self):
         self._generatePastePads()
 
     def _generatePastePads(self):
-        a = 2*pi/self.num_paste_zones
+        a = 360/self.num_paste_zones
         pos = Vector2D(self.radius, 0).rotate(a/2)
         paste_width = self.width + 2*self.solder_paste_margin
 
@@ -195,7 +195,7 @@ def _generateCopperPads(self):
                 radius=self.radius
                 ))
 
-        a = 2*pi/self.num_anchor
+        a = 360/self.num_anchor
         pos = Vector2D(self.radius, 0)
         for i in range(1, self.num_anchor):
             pos.rotate(a)

KicadModTree/tests/moduletests/__init__.py

@@ -17,3 +17,4 @@
 from .test_kicad5_padshapes import Kicad5PadsTests
 from .test_exposed_pad import ExposedPadTests
 from .test_arc import ArcTests
+from .test_rotation import RotationTests