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
19 __author__ = 'Enrique Perez (perez_enrique@yahoo.com)'
20 __credits__ = 'Art of Illusion <http://www.artofillusion.org/>'
21 __date__ = '$Date: 2008/02/05 $'
22 __license__ = 'GNU Affero General Public License http://www.gnu.org/licenses/agpl.html'
25 def addGridRow(diameter, gridPath, loopsComplex, maximumComplex, rowIndex, x, y, zigzag):
28 while x < maximumComplex.real:
30 if euclidean.getIsInFilledRegion(loopsComplex, point):
33 if zigzag and rowIndex % 2 == 1:
37 def getGeometryOutput(elementNode):
38 'Get vector3 vertexes from attribute dictionary.'
39 derivation = GridDerivation(elementNode)
40 diameter = derivation.radius + derivation.radius
41 typeStringTwoCharacters = derivation.typeString.lower()[: 2]
42 typeStringFirstCharacter = typeStringTwoCharacters[: 1]
43 topRight = complex(derivation.demiwidth, derivation.demiheight)
44 loopsComplex = [euclidean.getSquareLoopWiddershins(-topRight, topRight)]
45 if len(derivation.target) > 0:
46 loopsComplex = euclidean.getComplexPaths(derivation.target)
47 maximumComplex = euclidean.getMaximumByComplexPaths(loopsComplex)
48 minimumComplex = euclidean.getMinimumByComplexPaths(loopsComplex)
50 if typeStringTwoCharacters == 'he':
51 gridPath = getHexagonalGrid(diameter, loopsComplex, maximumComplex, minimumComplex, derivation.zigzag)
52 elif typeStringTwoCharacters == 'ra' or typeStringFirstCharacter == 'a':
53 gridPath = getRandomGrid(derivation, diameter, elementNode, loopsComplex, maximumComplex, minimumComplex)
54 elif typeStringTwoCharacters == 're' or typeStringFirstCharacter == 'e':
55 gridPath = getRectangularGrid(diameter, loopsComplex, maximumComplex, minimumComplex, derivation.zigzag)
57 print('Warning, the step type was not one of (hexagonal, random or rectangular) in getGeometryOutput in grid for:')
58 print(derivation.typeString)
61 loop = euclidean.getVector3Path(gridPath)
62 elementNode.attributes['closed'] = 'false'
63 return lineation.getGeometryOutputByLoop(elementNode, lineation.SideLoop(loop, 0.5 * math.pi))
65 def getGeometryOutputByArguments(arguments, elementNode):
66 'Get vector3 vertexes from attribute dictionary by arguments.'
67 if len(arguments) < 1:
68 return getGeometryOutput(elementNode)
69 inradius = 0.5 * euclidean.getFloatFromValue(arguments[0])
70 elementNode.attributes['inradius.x'] = str(inradius)
71 if len(arguments) > 1:
72 inradius = 0.5 * euclidean.getFloatFromValue(arguments[1])
73 elementNode.attributes['inradius.y'] = str(inradius)
74 return getGeometryOutput(elementNode)
76 def getHexagonalGrid(diameter, loopsComplex, maximumComplex, minimumComplex, zigzag):
78 diameter = complex(diameter.real, math.sqrt(0.75) * diameter.imag)
79 demiradius = 0.25 * diameter
80 xRadius = 0.5 * diameter.real
81 xStart = minimumComplex.real - demiradius.real
82 y = minimumComplex.imag - demiradius.imag
85 while y < maximumComplex.imag:
89 addGridRow(diameter, gridPath, loopsComplex, maximumComplex, rowIndex, x, y, zigzag)
94 def getIsPointInsideZoneAwayOthers(diameterReciprocal, loopsComplex, point, pixelDictionary):
95 'Determine if the point is inside the loops zone and and away from the other points.'
96 if not euclidean.getIsInFilledRegion(loopsComplex, point):
98 pointOverDiameter = complex(point.real * diameterReciprocal.real, point.imag * diameterReciprocal.imag)
99 squareValues = euclidean.getSquareValuesFromPoint(pixelDictionary, pointOverDiameter)
100 for squareValue in squareValues:
101 if abs(squareValue - pointOverDiameter) < 1.0:
103 euclidean.addElementToPixelListFromPoint(pointOverDiameter, pixelDictionary, pointOverDiameter)
106 def getNewDerivation(elementNode):
107 'Get new derivation.'
108 return GridDerivation(elementNode)
110 def getRandomGrid(derivation, diameter, elementNode, loopsComplex, maximumComplex, minimumComplex):
111 'Get rectangular grid.'
113 diameterReciprocal = complex(1.0 / diameter.real, 1.0 / diameter.imag)
114 diameterSquared = diameter.real * diameter.real + diameter.imag * diameter.imag
115 elements = int(math.ceil(derivation.density * euclidean.getAreaLoops(loopsComplex) / diameterSquared / math.sqrt(0.75)))
116 elements = evaluate.getEvaluatedInt(elements, elementNode, 'elements')
117 failedPlacementAttempts = 0
119 if derivation.seed != None:
120 random.seed(derivation.seed)
121 successfulPlacementAttempts = 0
122 while failedPlacementAttempts < 100:
123 point = euclidean.getRandomComplex(minimumComplex, maximumComplex)
124 if getIsPointInsideZoneAwayOthers(diameterReciprocal, loopsComplex, point, pixelDictionary):
125 gridPath.append(point)
126 euclidean.addElementToPixelListFromPoint(point, pixelDictionary, point)
127 successfulPlacementAttempts += 1
129 failedPlacementAttempts += 1
130 if successfulPlacementAttempts >= elements:
134 def getRectangularGrid(diameter, loopsComplex, maximumComplex, minimumComplex, zigzag):
135 'Get rectangular grid.'
136 demiradius = 0.25 * diameter
137 xStart = minimumComplex.real - demiradius.real
138 y = minimumComplex.imag - demiradius.imag
141 while y < maximumComplex.imag:
142 addGridRow(diameter, gridPath, loopsComplex, maximumComplex, rowIndex, xStart, y, zigzag)
147 def processElementNode(elementNode):
148 'Process the xml element.'
149 path.convertElementNode(elementNode, getGeometryOutput(elementNode))
152 class GridDerivation:
153 'Class to hold grid variables.'
154 def __init__(self, elementNode):
156 self.inradius = lineation.getInradius(complex(10.0, 10.0), elementNode)
157 self.demiwidth = lineation.getFloatByPrefixBeginEnd(elementNode, 'demiwidth', 'width', self.inradius.real)
158 self.demiheight = lineation.getFloatByPrefixBeginEnd(elementNode, 'demiheight', 'height', self.inradius.imag)
159 self.density = evaluate.getEvaluatedFloat(0.2, elementNode, 'density')
160 self.radius = lineation.getComplexByPrefixBeginEnd(elementNode, 'elementRadius', 'elementDiameter', complex(1.0, 1.0))
161 self.radius = lineation.getComplexByPrefixBeginEnd(elementNode, 'radius', 'diameter', self.radius)
162 self.seed = evaluate.getEvaluatedInt(None, elementNode, 'seed')
163 self.target = evaluate.getTransformedPathsByKey([], elementNode, 'target')
164 self.typeMenuRadioStrings = 'hexagonal random rectangular'.split()
165 self.typeString = evaluate.getEvaluatedString('rectangular', elementNode, 'type')
166 self.zigzag = evaluate.getEvaluatedBoolean(True, elementNode, 'zigzag')