Cura/cura_sf/fabmetheus_utilities/geometry/creation/polygon.py

   1 """
   2 Polygon path.
   3
   4 """
   5
   6 from __future__ import absolute_import
   7 #Init has to be imported first because it has code to workaround the python bug where relative imports don't work if the module is imported as a main module.
   8 import __init__
   9
  10 from fabmetheus_utilities.geometry.creation import lineation
  11 from fabmetheus_utilities.geometry.geometry_tools import path
  12 from fabmetheus_utilities.geometry.geometry_utilities import evaluate
  13 from fabmetheus_utilities.vector3 import Vector3
  14 from fabmetheus_utilities import euclidean
  15 import math
  16
  17
  18 __author__ = 'Enrique Perez (perez_enrique@yahoo.com)'
  19 __credits__ = 'Art of Illusion <http://www.artofillusion.org/>'
  20 __date__ = '$Date: 2008/02/05 $'
  21 __license__ = 'GNU Affero General Public License http://www.gnu.org/licenses/agpl.html'
  22
  23
  24 def getGeometryOutput(derivation, elementNode):
  25         "Get vector3 vertexes from attribute dictionary."
  26         if derivation == None:
  27                 derivation = PolygonDerivation(elementNode)
  28         loop = []
  29         spiral = lineation.Spiral(derivation.spiral, 0.5 * derivation.sideAngle / math.pi)
  30         for side in xrange(derivation.start, derivation.start + derivation.extent + 1):
  31                 angle = float(side) * derivation.sideAngle
  32                 unitPolar = euclidean.getWiddershinsUnitPolar(angle)
  33                 vertex = spiral.getSpiralPoint(unitPolar, Vector3(unitPolar.real * derivation.radius.real, unitPolar.imag * derivation.radius.imag))
  34                 loop.append(vertex)
  35         loop = euclidean.getLoopWithoutCloseEnds(0.000001 * max(derivation.radius.real, derivation.radius.imag), loop)
  36         lineation.setClosedAttribute(elementNode, derivation.revolutions)
  37         return lineation.getGeometryOutputByLoop(elementNode, lineation.SideLoop(loop, derivation.sideAngle))
  38
  39 def getGeometryOutputByArguments(arguments, elementNode):
  40         "Get vector3 vertexes from attribute dictionary by arguments."
  41         evaluate.setAttributesByArguments(['sides', 'radius'], arguments, elementNode)
  42         return getGeometryOutput(None, elementNode)
  43
  44 def getNewDerivation(elementNode):
  45         'Get new derivation.'
  46         return PolygonDerivation(elementNode)
  47
  48 def processElementNode(elementNode):
  49         "Process the xml element."
  50         path.convertElementNode(elementNode, getGeometryOutput(None, elementNode))
  51
  52
  53 class PolygonDerivation:
  54         "Class to hold polygon variables."
  55         def __init__(self, elementNode):
  56                 'Set defaults.'
  57                 self.sides = evaluate.getEvaluatedFloat(4.0, elementNode, 'sides')
  58                 self.sideAngle = 2.0 * math.pi / self.sides
  59                 cosSide = math.cos(0.5 * self.sideAngle)
  60                 self.radius = lineation.getComplexByMultiplierPrefixes(elementNode, cosSide, ['apothem', 'inradius'], complex(1.0, 1.0))
  61                 self.radius = lineation.getComplexByPrefixes(elementNode, ['demisize', 'radius'], self.radius)
  62                 self.radius = lineation.getComplexByMultiplierPrefixes(elementNode, 2.0, ['diameter', 'size'], self.radius)
  63                 self.sidesCeiling = int(math.ceil(abs(self.sides)))
  64                 self.start = evaluate.getEvaluatedInt(0, elementNode, 'start')
  65                 end = evaluate.getEvaluatedInt(self.sidesCeiling, elementNode, 'end')
  66                 self.revolutions = evaluate.getEvaluatedInt(1, elementNode, 'revolutions')
  67                 self.extent = evaluate.getEvaluatedInt(end - self.start, elementNode, 'extent')
  68                 self.extent += self.sidesCeiling * (self.revolutions - 1)
  69                 self.spiral = evaluate.getVector3ByPrefix(None, elementNode, 'spiral')