src/PythonAPI/model/features/revolution.py

   1 """Revolution Interface
   2 Author: Sergey Pokhodenko
   3 Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   4 """
   5
   6 from model.roots import Interface
   7
   8
   9 def addRevolution(part, *args):
  10     """Add an Revolution feature to the Part and return Revolution.
  11
  12     Pass all args to Revolution __init__ function.
  13     """
  14     assert(len(args) > 0 and args[0] is not None)
  15     feature = part.addFeature("Revolution")
  16     return Revolution(feature, *args)
  17
  18
  19 class Revolution(Interface):
  20     """Interface on an Revolution feature."""
  21
  22     def __init__(self, feature, *args):
  23         """Initialize an Revolution feature with given parameters.
  24
  25         Expected arguments for all modes:
  26         feature -- an Revolution feature
  27
  28         Expected arguments for initializing the feature:
  29         base -- name, sketch or list of names and sketches.
  30         If base is None then don't change the feature.
  31         axis_object -- name, edge for axis
  32
  33         For ByAngles mode (expect 2 arguments):
  34         to_angle -- upper angle
  35         from_angle -- lower angle
  36
  37         For ByPlanesAndOffsets mode (expect 4 arguments):
  38         to_object -- upper object (plane)
  39         to_offset -- offset from upper object
  40         from_object -- lower object (plane)
  41         from_offset -- offset from lower object
  42         """
  43         Interface.__init__(self, feature)
  44         assert(self._feature.getKind() == "Revolution")
  45
  46         self._base = self._feature.data().selectionList("base")
  47         self._axis_object = self._feature.data().selection("axis_object")
  48         self._CreationMethod = self._feature.string("CreationMethod")
  49         self._to_angle = self._feature.data().real("to_angle")
  50         self._from_angle = self._feature.data().real("from_angle")
  51         self._to_object = self._feature.data().selection("to_object")
  52         self._to_offset = self._feature.data().real("to_offset")
  53         self._from_object = self._feature.data().selection("from_object")
  54         self._from_offset = self._feature.data().real("from_offset")
  55
  56         assert(self._base)
  57         assert(self._axis_object)
  58         assert(self._CreationMethod)
  59         assert(self._to_angle)
  60         assert(self._from_angle)
  61         assert(self._to_object)
  62         assert(self._to_offset)
  63         assert(self._from_object)
  64         assert(self._from_offset)
  65
  66         if not args:
  67             return
  68
  69         assert(len(args) in (4, 6))
  70
  71         base, axis_object = args[:2]
  72         args = args[2:]
  73
  74         self.setBase(base)
  75         self.setAxisObject(axis_object)
  76
  77         if len(args) == 4:
  78             self.setPlanesAndOffsets(*args)
  79         elif len(args) == 2:
  80             self.setAngles(*args)
  81
  82         self._execute()
  83         pass
  84
  85     def __clear(self):
  86         self._CreationMethod.setValue("ByAngles")
  87         self._fill_attribute(self._to_angle, 0)
  88         self._fill_attribute(self._from_angle, 0)
  89         self._fill_attribute(self._to_object, None)
  90         self._fill_attribute(self._to_offset, 0)
  91         self._fill_attribute(self._from_object, None)
  92         self._fill_attribute(self._from_offset, 0)
  93         pass
  94
  95     def setBase(self, base):
  96         """Modify base attribute of the feature.
  97
  98         See __init__.
  99         """
 100         self._fill_attribute(self._base, base)
 101         pass
 102
 103     def setAxisObject(self, axis_object):
 104         """Modify axis_object attribute of the feature.
 105
 106         See __init__.
 107         """
 108         self._fill_attribute(self._axis_object, axis_object)
 109         pass
 110
 111     def setAngles(self, to_angle, from_angle):
 112         """Modify the to_angle, from_angle attributes of the feature.
 113
 114         See __init__.
 115         """
 116         self.__clear()
 117         self._CreationMethod.setValue("ByAngles")
 118         self._fill_attribute(self._to_angle, to_angle)
 119         self._fill_attribute(self._from_angle, from_angle)
 120         pass
 121
 122     def setPlanesAndOffsets(self, to_object, to_offset,
 123                             from_object, from_offset):
 124         """Modify planes and offsets attributes of the feature.
 125
 126         See __init__.
 127         """
 128         self.__clear()
 129         self._CreationMethod.setValue("ByPlanesAndOffsets")
 130         self._fill_attribute(self._to_object, to_object)
 131         self._fill_attribute(self._to_offset, to_offset)
 132         self._fill_attribute(self._from_object, from_object)
 133         self._fill_attribute(self._from_offset, from_offset)
 134         pass