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

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