Cura/cura_sf/fabmetheus_utilities/geometry/manipulation_meta/_carve.py

   1 """
   2 Boolean geometry carve.
   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.geometry_tools import path
  11 from fabmetheus_utilities.geometry.geometry_utilities.evaluate_elements import setting
  12 from fabmetheus_utilities.geometry.geometry_utilities import boolean_geometry
  13 from fabmetheus_utilities.geometry.geometry_utilities import evaluate
  14 from fabmetheus_utilities.geometry.geometry_utilities import matrix
  15 from fabmetheus_utilities.geometry.solids import triangle_mesh
  16 from fabmetheus_utilities import euclidean
  17 from fabmetheus_utilities import xml_simple_reader
  18
  19
  20 __author__ = 'Enrique Perez (perez_enrique@yahoo.com)'
  21 __credits__ = 'Art of Illusion <http://www.artofillusion.org/>'
  22 __date__ = '$Date: 2008/02/05 $'
  23 __license__ = 'GNU Affero General Public License http://www.gnu.org/licenses/agpl.html'
  24
  25
  26 def getLinkedElementNode(idSuffix, parentNode, target):
  27         'Get elementNode with identifiers and parentNode.'
  28         linkedElementNode = xml_simple_reader.ElementNode()
  29         euclidean.overwriteDictionary(target.attributes, ['id', 'name', 'quantity'], linkedElementNode.attributes)
  30         linkedElementNode.addSuffixToID(idSuffix)
  31         tagKeys = target.getTagKeys()
  32         tagKeys.append('carve')
  33         tagKeys.sort()
  34         tags = ', '.join(tagKeys)
  35         linkedElementNode.attributes['tags'] = tags
  36         linkedElementNode.setParentAddToChildNodes(parentNode)
  37         linkedElementNode.addToIdentifierDictionaries()
  38         return linkedElementNode
  39
  40 def getNewDerivation(elementNode):
  41         'Get new derivation.'
  42         return CarveDerivation(elementNode)
  43
  44 def processElementNode(elementNode):
  45         'Process the xml element.'
  46         processElementNodeByDerivation(None, elementNode)
  47
  48 def processElementNodeByDerivation(derivation, elementNode):
  49         'Process the xml element by derivation.'
  50         if derivation == None:
  51                 derivation = CarveDerivation(elementNode)
  52         targetElementNode = derivation.targetElementNode
  53         if targetElementNode == None:
  54                 print('Warning, carve could not get target for:')
  55                 print(elementNode)
  56                 return
  57         xmlObject = targetElementNode.xmlObject
  58         if xmlObject == None:
  59                 print('Warning, processElementNodeByDerivation in carve could not get xmlObject for:')
  60                 print(targetElementNode)
  61                 print(derivation.elementNode)
  62                 return
  63         matrix.getBranchMatrixSetElementNode(targetElementNode)
  64         transformedVertexes = xmlObject.getTransformedVertexes()
  65         if len(transformedVertexes) < 1:
  66                 print('Warning, transformedVertexes is zero in processElementNodeByDerivation in carve for:')
  67                 print(xmlObject)
  68                 print(targetElementNode)
  69                 print(derivation.elementNode)
  70                 return
  71         elementNode.localName = 'group'
  72         elementNode.getXMLProcessor().processElementNode(elementNode)
  73         minimumZ = boolean_geometry.getMinimumZ(xmlObject)
  74         maximumZ = euclidean.getTopPath(transformedVertexes)
  75         zoneArrangement = triangle_mesh.ZoneArrangement(derivation.layerHeight, transformedVertexes)
  76         oldVisibleString = targetElementNode.attributes['visible']
  77         targetElementNode.attributes['visible'] = True
  78         z = minimumZ + 0.5 * derivation.layerHeight
  79         loopLayers = boolean_geometry.getLoopLayers([xmlObject], derivation.importRadius, derivation.layerHeight, maximumZ, False, z, zoneArrangement)
  80         targetElementNode.attributes['visible'] = oldVisibleString
  81         for loopLayerIndex, loopLayer in enumerate(loopLayers):
  82                 if len(loopLayer.loops) > 0:
  83                         pathElement = getLinkedElementNode('_carve_%s' % loopLayerIndex, elementNode, targetElementNode)
  84                         vector3Loops = euclidean.getVector3Paths(loopLayer.loops, loopLayer.z)
  85                         path.convertElementNode(pathElement, vector3Loops)
  86
  87
  88 class CarveDerivation:
  89         "Class to hold carve variables."
  90         def __init__(self, elementNode):
  91                 'Set defaults.'
  92                 self.elementNode = elementNode
  93                 self.importRadius = setting.getImportRadius(elementNode)
  94                 self.layerHeight = setting.getLayerHeight(elementNode)
  95                 self.targetElementNode = evaluate.getElementNodeByKey(elementNode, 'target')