116 lines
4.1 KiB
Python
116 lines
4.1 KiB
Python
"""
|
|
Equation for vertexes.
|
|
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
#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.
|
|
import __init__
|
|
|
|
from fabmetheus_utilities.geometry.geometry_utilities import evaluate
|
|
from fabmetheus_utilities.geometry.geometry_utilities import matrix
|
|
from fabmetheus_utilities import euclidean
|
|
import math
|
|
|
|
|
|
__author__ = 'Enrique Perez (perez_enrique@yahoo.com)'
|
|
__credits__ = 'Art of Illusion <http://www.artofillusion.org/>'
|
|
__date__ = '$Date: 2008/02/05 $'
|
|
__license__ = 'GNU Affero General Public License http://www.gnu.org/licenses/agpl.html'
|
|
|
|
|
|
globalExecutionOrder = -100
|
|
|
|
|
|
def equate(point, returnValue):
|
|
"Get equation for rectangular."
|
|
point.setToVector3(evaluate.getVector3ByDictionaryListValue(returnValue, point))
|
|
|
|
def equatePoints(elementNode, points, prefix, revolutions):
|
|
"Equate the points."
|
|
derivation = EquationDerivation(elementNode, prefix)
|
|
for equationResult in derivation.equationResults:
|
|
for point in points:
|
|
returnValue = equationResult.getReturnValue(point, revolutions)
|
|
if returnValue == None:
|
|
print('Warning, returnValue in alterVertexesByEquation in equation is None for:')
|
|
print(point)
|
|
print(elementNode)
|
|
else:
|
|
equationResult.equationFunction(point, returnValue)
|
|
|
|
def equateX(point, returnValue):
|
|
"Get equation for rectangular x."
|
|
point.x = returnValue
|
|
|
|
def equateY(point, returnValue):
|
|
"Get equation for rectangular y."
|
|
point.y = returnValue
|
|
|
|
def equateZ(point, returnValue):
|
|
"Get equation for rectangular z."
|
|
point.z = returnValue
|
|
|
|
def getManipulatedGeometryOutput(elementNode, geometryOutput, prefix):
|
|
"Get equated geometryOutput."
|
|
equatePoints(elementNode, matrix.getVertexes(geometryOutput), prefix, None)
|
|
return geometryOutput
|
|
|
|
def getManipulatedPaths(close, elementNode, loop, prefix, sideLength):
|
|
"Get equated paths."
|
|
equatePoints(elementNode, loop, prefix, 0.0)
|
|
return [loop]
|
|
|
|
def getNewDerivation(elementNode, prefix, sideLength):
|
|
'Get new derivation.'
|
|
return EquationDerivation(elementNode, prefix)
|
|
|
|
|
|
class EquationDerivation:
|
|
"Class to hold equation variables."
|
|
def __init__(self, elementNode, prefix):
|
|
'Set defaults.'
|
|
self.equationResults = []
|
|
self.addEquationResult(elementNode, equate, prefix)
|
|
self.addEquationResult(elementNode, equateX, prefix)
|
|
self.addEquationResult(elementNode, equateY, prefix)
|
|
self.addEquationResult(elementNode, equateZ, prefix)
|
|
|
|
def addEquationResult(self, elementNode, equationFunction, prefix):
|
|
'Add equation result to equationResults.'
|
|
prefixedEquationName = prefix + equationFunction.__name__[ len('equate') : ].replace('Dot', '.').lower()
|
|
if prefixedEquationName in elementNode.attributes:
|
|
self.equationResults.append(EquationResult(elementNode, equationFunction, prefixedEquationName))
|
|
|
|
|
|
class EquationResult:
|
|
"Class to get equation results."
|
|
def __init__(self, elementNode, equationFunction, key):
|
|
"Initialize."
|
|
self.distance = 0.0
|
|
elementNode.xmlObject = evaluate.getEvaluatorSplitWords(elementNode.attributes[key])
|
|
self.equationFunction = equationFunction
|
|
self.function = evaluate.Function(elementNode)
|
|
self.points = []
|
|
|
|
def getReturnValue(self, point, revolutions):
|
|
"Get return value."
|
|
if self.function == None:
|
|
return point
|
|
self.function.localDictionary['azimuth'] = math.degrees(math.atan2(point.y, point.x))
|
|
if len(self.points) > 0:
|
|
self.distance += abs(point - self.points[-1])
|
|
self.function.localDictionary['distance'] = self.distance
|
|
self.function.localDictionary['radius'] = abs(point.dropAxis())
|
|
if revolutions != None:
|
|
if len( self.points ) > 0:
|
|
revolutions += 0.5 / math.pi * euclidean.getAngleAroundZAxisDifference(point, self.points[-1])
|
|
self.function.localDictionary['revolutions'] = revolutions
|
|
self.function.localDictionary['vertex'] = point
|
|
self.function.localDictionary['vertexes'] = self.points
|
|
self.function.localDictionary['vertexindex'] = len(self.points)
|
|
self.function.localDictionary['x'] = point.x
|
|
self.function.localDictionary['y'] = point.y
|
|
self.function.localDictionary['z'] = point.z
|
|
self.points.append(point)
|
|
return self.function.getReturnValueWithoutDeletion()
|