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

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