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

   1 """
   2 Boolean geometry disjoin.
   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 difference
  16 from fabmetheus_utilities.geometry.solids import triangle_mesh
  17 from fabmetheus_utilities import euclidean
  18 from fabmetheus_utilities import xml_simple_reader
  19 from fabmetheus_utilities.vector3 import Vector3
  20
  21
  22 __author__ = 'Enrique Perez (perez_enrique@yahoo.com)'
  23 __credits__ = 'Art of Illusion <http://www.artofillusion.org/>'
  24 __date__ = '$Date: 2008/02/05 $'
  25 __license__ = 'GNU Affero General Public License http://www.gnu.org/licenses/agpl.html'
  26
  27
  28 def getLinkedElementNode(idSuffix, parentNode, target):
  29         'Get elementNode with identifiers and parentNode.'
  30         linkedElementNode = xml_simple_reader.ElementNode()
  31         euclidean.overwriteDictionary(target.attributes, ['id', 'name', 'quantity'], linkedElementNode.attributes)
  32         linkedElementNode.addSuffixToID(idSuffix)
  33         tagKeys = target.getTagKeys()
  34         tagKeys.append('disjoin')
  35         tagKeys.sort()
  36         tags = ', '.join(tagKeys)
  37         linkedElementNode.attributes['tags'] = tags
  38         linkedElementNode.setParentAddToChildNodes(parentNode)
  39         linkedElementNode.addToIdentifierDictionaries()
  40         return linkedElementNode
  41
  42 def getNewDerivation(elementNode):
  43         'Get new derivation.'
  44         return DisjoinDerivation(elementNode)
  45
  46 def processElementNode(elementNode):
  47         'Process the xml element.'
  48         processElementNodeByDerivation(None, elementNode)
  49
  50 def processElementNodeByDerivation(derivation, elementNode):
  51         'Process the xml element by derivation.'
  52         if derivation == None:
  53                 derivation = DisjoinDerivation(elementNode)
  54         targetElementNode = derivation.targetElementNode
  55         if targetElementNode == None:
  56                 print('Warning, disjoin could not get target for:')
  57                 print(elementNode)
  58                 return
  59         xmlObject = targetElementNode.xmlObject
  60         if xmlObject == None:
  61                 print('Warning, processElementNodeByDerivation in disjoin could not get xmlObject for:')
  62                 print(targetElementNode)
  63                 print(derivation.elementNode)
  64                 return
  65         matrix.getBranchMatrixSetElementNode(targetElementNode)
  66         transformedVertexes = xmlObject.getTransformedVertexes()
  67         if len(transformedVertexes) < 1:
  68                 print('Warning, transformedVertexes is zero in processElementNodeByDerivation in disjoin for:')
  69                 print(xmlObject)
  70                 print(targetElementNode)
  71                 print(derivation.elementNode)
  72                 return
  73         elementNode.localName = 'group'
  74         elementNode.getXMLProcessor().processElementNode(elementNode)
  75         targetChainMatrix = matrix.Matrix(xmlObject.getMatrixChainTetragrid())
  76         minimumZ = boolean_geometry.getMinimumZ(xmlObject)
  77         z = minimumZ + 0.5 * derivation.sheetThickness
  78         zoneArrangement = triangle_mesh.ZoneArrangement(derivation.layerHeight, transformedVertexes)
  79         oldVisibleString = targetElementNode.attributes['visible']
  80         targetElementNode.attributes['visible'] = True
  81         loops = boolean_geometry.getEmptyZLoops([xmlObject], derivation.importRadius, False, z, zoneArrangement)
  82         targetElementNode.attributes['visible'] = oldVisibleString
  83         vector3Loops = euclidean.getVector3Paths(loops, z)
  84         pathElement = getLinkedElementNode('_sheet', elementNode, targetElementNode)
  85         path.convertElementNode(pathElement, vector3Loops)
  86         targetOutput = xmlObject.getGeometryOutput()
  87         differenceElement = getLinkedElementNode('_solid', elementNode, targetElementNode)
  88         targetElementCopy = targetElementNode.getCopy('_positive', differenceElement)
  89         targetElementCopy.attributes['visible'] = True
  90         targetElementCopy.attributes.update(targetChainMatrix.getAttributes('matrix.'))
  91         complexMaximum = euclidean.getMaximumByVector3Path(transformedVertexes).dropAxis()
  92         complexMinimum = euclidean.getMinimumByVector3Path(transformedVertexes).dropAxis()
  93         centerComplex = 0.5 * (complexMaximum + complexMinimum)
  94         centerVector3 = Vector3(centerComplex.real, centerComplex.imag, minimumZ)
  95         slightlyMoreThanHalfExtent = 0.501 * (complexMaximum - complexMinimum)
  96         inradius = Vector3(slightlyMoreThanHalfExtent.real, slightlyMoreThanHalfExtent.imag, derivation.sheetThickness)
  97         cubeElement = xml_simple_reader.ElementNode()
  98         cubeElement.attributes['inradius'] = str(inradius)
  99         if not centerVector3.getIsDefault():
 100                 cubeElement.attributes['translate.'] = str(centerVector3)
 101         cubeElement.localName = 'cube'
 102         cubeElement.setParentAddToChildNodes(differenceElement)
 103         difference.processElementNode(differenceElement)
 104
 105
 106 class DisjoinDerivation:
 107         "Class to hold disjoin variables."
 108         def __init__(self, elementNode):
 109                 'Set defaults.'
 110                 self.elementNode = elementNode
 111                 self.importRadius = setting.getImportRadius(elementNode)
 112                 self.layerHeight = setting.getLayerHeight(elementNode)
 113                 self.sheetThickness = setting.getSheetThickness(elementNode)
 114                 self.targetElementNode = evaluate.getElementNodeByKey(elementNode, 'target')