Cura/cura_sf/fabmetheus_utilities/geometry/manipulation_paths/bevel.py

   1 """
   2 Add material to support overhang or remove material at the overhang angle.
   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.creation import lineation
  11 from fabmetheus_utilities.geometry.geometry_utilities import evaluate
  12 from fabmetheus_utilities.vector3 import Vector3
  13 from fabmetheus_utilities import euclidean
  14
  15
  16 __author__ = 'Enrique Perez (perez_enrique@yahoo.com)'
  17 __credits__ = 'Art of Illusion <http://www.artofillusion.org/>'
  18 __date__ = '$Date: 2008/02/05 $'
  19 __license__ = 'GNU Affero General Public License http://www.gnu.org/licenses/agpl.html'
  20
  21
  22 globalExecutionOrder = 20
  23
  24
  25 def getBevelPath( begin, center, close, end, radius ):
  26         "Get bevel path."
  27         beginComplex = begin.dropAxis()
  28         centerComplex = center.dropAxis()
  29         endComplex = end.dropAxis()
  30         beginComplexSegmentLength = abs( centerComplex - beginComplex )
  31         endComplexSegmentLength = abs( centerComplex - endComplex )
  32         minimumRadius = lineation.getMinimumRadius( beginComplexSegmentLength, endComplexSegmentLength, radius )
  33         if minimumRadius <= close:
  34                 return [ center ]
  35         beginBevel = center + minimumRadius / beginComplexSegmentLength * ( begin - center )
  36         endBevel = center + minimumRadius / endComplexSegmentLength * ( end - center )
  37         if radius > 0.0:
  38                 return [ beginBevel, endBevel ]
  39         midpointComplex = 0.5 * ( beginBevel.dropAxis() + endBevel.dropAxis() )
  40         spikeComplex = centerComplex + centerComplex - midpointComplex
  41         return [ beginBevel, Vector3( spikeComplex.real, spikeComplex.imag, center.z ), endBevel ]
  42
  43 def getManipulatedPaths(close, elementNode, loop, prefix, sideLength):
  44         "Get bevel loop."
  45         if len(loop) < 3:
  46                 return [loop]
  47         derivation = BevelDerivation(elementNode, prefix, sideLength)
  48         if derivation.radius == 0.0:
  49                 return loop
  50         bevelLoop = []
  51         for pointIndex in xrange(len(loop)):
  52                 begin = loop[(pointIndex + len(loop) - 1) % len(loop)]
  53                 center = loop[pointIndex]
  54                 end = loop[(pointIndex + 1) % len(loop)]
  55                 bevelLoop += getBevelPath(begin, center, close, end, derivation.radius)
  56         return [euclidean.getLoopWithoutCloseSequentialPoints(close, bevelLoop)]
  57
  58 def getNewDerivation(elementNode, prefix, sideLength):
  59         'Get new derivation.'
  60         return BevelDerivation(elementNode, prefix, sideLength)
  61
  62 def processElementNode(elementNode):
  63         "Process the xml element."
  64         lineation.processElementNodeByFunction(elementNode, getManipulatedPaths)
  65
  66
  67 class BevelDerivation:
  68         "Class to hold bevel variables."
  69         def __init__(self, elementNode, prefix, sideLength):
  70                 'Set defaults.'
  71                 self.radius = lineation.getFloatByPrefixSide(0.0, elementNode, prefix + 'radius', sideLength)