# KicadModTree is free software: you can redistribute it and/or
# modify it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# KicadModTree is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with kicad-footprint-generator. If not, see < http://www.gnu.org/licenses/ >.
#
# (C) 2016 by Thomas Pointhuber, <thomas.pointhuber@gmx.at>
import math
from KicadModTree.Vector import *
from KicadModTree.nodes.Node import Node
[docs]class Rotation(Node):
"""Apply rotation to the child tree
:param r: angle which the child should rotate
:type r: ``float``
:Example:
>>> from KicadModTree import *
>>> Rotation(90)
"""
def __init__(self, r):
Node.__init__(self)
self.rotation = r # in degree
[docs] def getRealPosition(self, coordinate, rotation=None):
if rotation is None:
rotation = 0
parsed_coordinate = Vector2D(coordinate)
phi = self.rotation*math.pi/180
rotation_coordinate = {'x': parsed_coordinate.x*math.cos(phi) + parsed_coordinate.y*math.sin(phi),
'y': -parsed_coordinate.x*math.sin(phi) + parsed_coordinate.y*math.cos(phi)}
if not self._parent:
if rotation is None:
return rotation_coordinate
else:
return rotation_coordinate, rotation + self.rotation
else:
if rotation is None:
rotation = 0
return self._parent.getRealPosition(rotation_coordinate, rotation + self.rotation)
def _getRenderTreeText(self):
render_text = Node._getRenderTreeText(self)
render_text += " [r: {r}]".format(r=self.rotation)
return render_text