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.
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
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'
24 def getGeometryOutput(derivation, elementNode):
25 "Get vector3 vertexes from attribute dictionary."
26 if derivation == None:
27 derivation = PolygonDerivation(elementNode)
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))
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))
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)
44 def getNewDerivation(elementNode):
46 return PolygonDerivation(elementNode)
48 def processElementNode(elementNode):
49 "Process the xml element."
50 path.convertElementNode(elementNode, getGeometryOutput(None, elementNode))
53 class PolygonDerivation:
54 "Class to hold polygon variables."
55 def __init__(self, elementNode):
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')