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

   1 """
   2 Boolean geometry disjoin.
   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 difference
  14 from fabmetheus_utilities.geometry.solids import triangle_mesh
  15 from fabmetheus_utilities import euclidean
  16 from fabmetheus_utilities import xml_simple_reader
  17 from fabmetheus_utilities.vector3 import Vector3
  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('disjoin')
  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 DisjoinDerivation(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 = DisjoinDerivation(elementNode)
  52         targetElementNode = derivation.targetElementNode
  53         if targetElementNode == None:
  54                 print('Warning, disjoin could not get target for:')
  55                 print(elementNode)
  56                 return
  57         xmlObject = targetElementNode.xmlObject
  58         if xmlObject == None:
  59                 print('Warning, processElementNodeByDerivation in disjoin 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 disjoin for:')
  67                 print(xmlObject)
  68                 print(targetElementNode)
  69                 print(derivation.elementNode)
  70                 return
  71         elementNode.localName = 'group'
  72         elementNode.getXMLProcessor().processElementNode(elementNode)
  73         targetChainMatrix = matrix.Matrix(xmlObject.getMatrixChainTetragrid())
  74         minimumZ = boolean_geometry.getMinimumZ(xmlObject)
  75         z = minimumZ + 0.5 * derivation.sheetThickness
  76         zoneArrangement = triangle_mesh.ZoneArrangement(derivation.layerHeight, transformedVertexes)
  77         oldVisibleString = targetElementNode.attributes['visible']
  78         targetElementNode.attributes['visible'] = True
  79         loops = boolean_geometry.getEmptyZLoops([xmlObject], derivation.importRadius, False, z, zoneArrangement)
  80         targetElementNode.attributes['visible'] = oldVisibleString
  81         vector3Loops = euclidean.getVector3Paths(loops, z)
  82         pathElement = getLinkedElementNode('_sheet', elementNode, targetElementNode)
  83         path.convertElementNode(pathElement, vector3Loops)
  84         targetOutput = xmlObject.getGeometryOutput()
  85         differenceElement = getLinkedElementNode('_solid', elementNode, targetElementNode)
  86         targetElementCopy = targetElementNode.getCopy('_positive', differenceElement)
  87         targetElementCopy.attributes['visible'] = True
  88         targetElementCopy.attributes.update(targetChainMatrix.getAttributes('matrix.'))
  89         complexMaximum = euclidean.getMaximumByVector3Path(transformedVertexes).dropAxis()
  90         complexMinimum = euclidean.getMinimumByVector3Path(transformedVertexes).dropAxis()
  91         centerComplex = 0.5 * (complexMaximum + complexMinimum)
  92         centerVector3 = Vector3(centerComplex.real, centerComplex.imag, minimumZ)
  93         slightlyMoreThanHalfExtent = 0.501 * (complexMaximum - complexMinimum)
  94         inradius = Vector3(slightlyMoreThanHalfExtent.real, slightlyMoreThanHalfExtent.imag, derivation.sheetThickness)
  95         cubeElement = xml_simple_reader.ElementNode()
  96         cubeElement.attributes['inradius'] = str(inradius)
  97         if not centerVector3.getIsDefault():
  98                 cubeElement.attributes['translate.'] = str(centerVector3)
  99         cubeElement.localName = 'cube'
 100         cubeElement.setParentAddToChildNodes(differenceElement)
 101         difference.processElementNode(differenceElement)
 102
 103
 104 class DisjoinDerivation(object):
 105         "Class to hold disjoin variables."
 106         def __init__(self, elementNode):
 107                 'Set defaults.'
 108                 self.elementNode = elementNode
 109                 self.importRadius = setting.getImportRadius(elementNode)
 110                 self.layerHeight = setting.getLayerHeight(elementNode)
 111                 self.sheetThickness = setting.getSheetThickness(elementNode)
 112                 self.targetElementNode = evaluate.getElementNodeByKey(elementNode, 'target')