src/FeaturesPlugin/Test/TestExtrusion.py

   1 """
   2       TestExtrusion.py
   3       Unit test of FeaturesPlugin_Extrusion class
   4
   5       class FeaturesPlugin_Extrusion : public ModelAPI_Feature
   6         static const std::string MY_EXTRUSION_ID("Extrusion");
   7         static const std::string LIST_ID("base");
   8         static const std::string METHOD_ATTR("CreationMethod");
   9         static const std::string MY_TO_SIZE_ID("to_size");
  10         static const std::string MY_FROM_SIZE_ID("from_size");
  11         static const std::string MY_TO_OBJECT_ID("to_object");
  12         static const std::string MY_TO_OFFSET_ID("to_offset");
  13         static const std::string MY_FROM_OBJECT_ID("from_object");
  14         static const std::string MY_FROM_OFFSET_ID("from_offset");
  15
  16         data()->addAttribute(LIST_ID(), ModelAPI_AttributeSelection::typeId());
  17         data()->addAttribute(CREATION_METHOD(), ModelAPI_AttributeString::typeId());
  18         data()->addAttribute(TO_SIZE_ID(), ModelAPI_AttributeDouble::typeId());
  19         data()->addAttribute(FROM_SIZE_ID(), ModelAPI_AttributeDouble::typeId());
  20         data()->addAttribute(TO_OBJECT_ID(), ModelAPI_AttributeSelection::typeId());
  21         data()->addAttribute(TO_OFFSET_ID(), ModelAPI_AttributeDouble::typeId());
  22         data()->addAttribute(FROM_OBJECT_ID(), ModelAPI_AttributeSelection::typeId());
  23         data()->addAttribute(FROM_OFFSET_ID(), ModelAPI_AttributeSelection::typeId());
  24 """
  25 #=========================================================================
  26 # Initialization of the test
  27 #=========================================================================
  28 from ModelAPI import *
  29 from GeomDataAPI import *
  30 from GeomAlgoAPI import *
  31 from GeomAPI import *
  32 import math
  33
  34 __updated__ = "2014-12-16"
  35
  36 aSession = ModelAPI_Session.get()
  37 aDocument = aSession.moduleDocument()
  38 # Create a part for extrusion
  39 aSession.startOperation()
  40 aPartFeature = aDocument.addFeature("Part")
  41 aSession.finishOperation()
  42 assert (len(aPartFeature.results()) == 1)
  43 # Another way is:
  44 # aPart = aSession.activeDocument()
  45 aPartResult = modelAPI_ResultPart(aPartFeature.firstResult())
  46 aPart = aPartResult.partDoc()
  47 #=========================================================================
  48 # Create a sketch circle to extrude
  49 #=========================================================================
  50 aSession.startOperation()
  51 aCircleSketchFeature = featureToCompositeFeature(aPart.addFeature("Sketch"))
  52 origin = geomDataAPI_Point(aCircleSketchFeature.attribute("Origin"))
  53 origin.setValue(0, 0, 0)
  54 dirx = geomDataAPI_Dir(aCircleSketchFeature.attribute("DirX"))
  55 dirx.setValue(1, 0, 0)
  56 norm = geomDataAPI_Dir(aCircleSketchFeature.attribute("Norm"))
  57 norm.setValue(0, 0, 1)
  58 # Create circle
  59 aSketchCircle = aCircleSketchFeature.addFeature("SketchCircle")
  60 anCircleCentr = geomDataAPI_Point2D(aSketchCircle.attribute("circle_center"))
  61 aCircleRadius = aSketchCircle.real("circle_radius")
  62 anCircleCentr.setValue(50., 50)
  63 aCircleRadius.setValue(20.)
  64 aSession.finishOperation()
  65 #=========================================================================
  66 # Make extrusion on circle
  67 #=========================================================================
  68 # Build shape from sketcher results
  69 aCircleSketchResult = aCircleSketchFeature.firstResult()
  70 aCircleSketchEdges = modelAPI_ResultConstruction(aCircleSketchResult).shape()
  71 origin = geomDataAPI_Point(aCircleSketchFeature.attribute("Origin")).pnt()
  72 dirX = geomDataAPI_Dir(aCircleSketchFeature.attribute("DirX")).dir()
  73 norm = geomDataAPI_Dir(aCircleSketchFeature.attribute("Norm")).dir()
  74 aCircleSketchFaces = ShapeList()
  75 GeomAlgoAPI_SketchBuilder.createFaces(
  76     origin, dirX, norm, aCircleSketchEdges, aCircleSketchFaces)
  77 assert (len(aCircleSketchFaces) > 0)
  78 assert (aCircleSketchFaces[0] is not None)
  79 # Create extrusion
  80 aSession.startOperation()
  81 anExtrusionFt = aPart.addFeature("Extrusion")
  82 assert (anExtrusionFt.getKind() == "Extrusion")
  83 # selection type FACE=4
  84 anExtrusionFt.selectionList("base").append(
  85     aCircleSketchResult, aCircleSketchFaces[0])
  86 anExtrusionFt.string("CreationMethod").setValue("BySizes")
  87 anExtrusionFt.real("to_size").setValue(50)
  88 anExtrusionFt.real("from_size").setValue(0)
  89 anExtrusionFt.real("to_offset").setValue(0) #TODO: remove
  90 anExtrusionFt.real("from_offset").setValue(0) #TODO: remove
  91 anExtrusionFt.execute()
  92 aSession.finishOperation()
  93 assert (anExtrusionFt.real("to_size").value() == 50.0)
  94
  95 # Check extrusion results
  96 assert (len(anExtrusionFt.results()) > 0)
  97 anExtrusionResult = modelAPI_ResultBody(anExtrusionFt.firstResult())
  98 assert (anExtrusionResult is not None)
  99
 100 #=========================================================================
 101 # Test extrusion between bounding planes
 102 #=========================================================================
 103 # Create from plane
 104 aSession.startOperation()
 105 aFromPlaneFeature = aPart.addFeature("Plane")
 106 aFromPlaneFeature.string("creation_method").setValue("by_general_equation")
 107 aFromPlaneFeature.string("by_other_plane_option").setValue("by_distance_from_other") # TODO: remove
 108 aFromPlaneFeature.real("A").setValue(0.)
 109 aFromPlaneFeature.real("B").setValue(0.)
 110 aFromPlaneFeature.real("C").setValue(1.)
 111 aFromPlaneFeature.real("D").setValue(30.)
 112 aSession.finishOperation()
 113
 114 # Create to plane
 115 aSession.startOperation()
 116 aToPlaneFeature = aPart.addFeature("Plane")
 117 aToPlaneFeature.string("creation_method").setValue("by_general_equation")
 118 aToPlaneFeature.string("by_other_plane_option").setValue("by_distance_from_other") # TODO: remove
 119 aToPlaneFeature.real("A").setValue(0.)
 120 aToPlaneFeature.real("B").setValue(0.)
 121 aToPlaneFeature.real("C").setValue(1.)
 122 aToPlaneFeature.real("D").setValue(-30.)
 123 aSession.finishOperation()
 124
 125 # Create extrusion
 126 aSession.startOperation()
 127 anExtrusionFt = aPart.addFeature("Extrusion")
 128 assert (anExtrusionFt.getKind() == "Extrusion")
 129 # selection type FACE=4
 130 anExtrusionFt.selectionList("base").append(
 131     aCircleSketchResult, aCircleSketchFaces[0])
 132 anExtrusionFt.string("CreationMethod").setValue("ByPlanesAndOffsets")
 133 anExtrusionFt.real("to_size").setValue(0) #TODO: remove
 134 anExtrusionFt.real("from_size").setValue(0) #TODO: remove
 135 aToResult = aToPlaneFeature.firstResult()
 136 aToShape = modelAPI_ResultConstruction(aToResult).shape()
 137 anExtrusionFt.selection("to_object").setValue(aToResult, aToShape)
 138 anExtrusionFt.real("to_offset").setValue(10)
 139 aFromResult = aFromPlaneFeature.firstResult()
 140 aFromShape = modelAPI_ResultConstruction(aFromResult).shape()
 141 anExtrusionFt.selection("from_object").setValue(aFromResult, aFromShape)
 142 anExtrusionFt.real("from_offset").setValue(10)
 143 anExtrusionFt.execute()
 144 aSession.finishOperation()
 145
 146 # Check extrusion results
 147 assert (len(anExtrusionFt.results()) > 0)
 148 anExtrusionResult = modelAPI_ResultBody(anExtrusionFt.firstResult())
 149 assert (anExtrusionResult is not None)
 150
 151 # Check extrusion volume
 152 aRefVolume = 100530.96491487337
 153 aResVolume = GeomAlgoAPI_ShapeTools_volume(anExtrusionResult.shape())
 154 assert (math.fabs(aResVolume - aRefVolume) < 10 ** -5)
 155
 156 #=========================================================================
 157 # Test extrusion between bounding faces from other sketch result
 158 #=========================================================================
 159 aSession.startOperation()
 160 aClampSketchFeature = featureToCompositeFeature(aPart.addFeature("Sketch"))
 161 origin = geomDataAPI_Point(aClampSketchFeature.attribute("Origin"))
 162 origin.setValue(0, 0, 0)
 163 dirx = geomDataAPI_Dir(aClampSketchFeature.attribute("DirX"))
 164 dirx.setValue(1, 0, 0)
 165 norm = geomDataAPI_Dir(aClampSketchFeature.attribute("Norm"))
 166 norm.setValue(0, 1, 0)
 167 # Create clamp
 168 aSketchLineA = aClampSketchFeature.addFeature("SketchLine")
 169 aSketchLineB = aClampSketchFeature.addFeature("SketchLine")
 170 aSketchLineC = aClampSketchFeature.addFeature("SketchLine")
 171 aSketchLineD = aClampSketchFeature.addFeature("SketchLine")
 172 aSketchLineE = aClampSketchFeature.addFeature("SketchLine")
 173 aLineAStartPoint = geomDataAPI_Point2D(aSketchLineA.attribute("StartPoint"))
 174 aLineAEndPoint = geomDataAPI_Point2D(aSketchLineA.attribute("EndPoint"))
 175 aLineBStartPoint = geomDataAPI_Point2D(aSketchLineB.attribute("StartPoint"))
 176 aLineBEndPoint = geomDataAPI_Point2D(aSketchLineB.attribute("EndPoint"))
 177 aLineCStartPoint = geomDataAPI_Point2D(aSketchLineC.attribute("StartPoint"))
 178 aLineCEndPoint = geomDataAPI_Point2D(aSketchLineC.attribute("EndPoint"))
 179 aLineDStartPoint = geomDataAPI_Point2D(aSketchLineD.attribute("StartPoint"))
 180 aLineDEndPoint = geomDataAPI_Point2D(aSketchLineD.attribute("EndPoint"))
 181 aLineEStartPoint = geomDataAPI_Point2D(aSketchLineE.attribute("StartPoint"))
 182 aLineEEndPoint = geomDataAPI_Point2D(aSketchLineE.attribute("EndPoint"))
 183 aLineAStartPoint.setValue(0., -50.)
 184 aLineAEndPoint.setValue(0., 50.)
 185 aLineBStartPoint.setValue(0., 50.)
 186 aLineBEndPoint.setValue(100., 50.)
 187 aLineCStartPoint.setValue(100., 50.)
 188 aLineCEndPoint.setValue(10., 0.)
 189 aLineDStartPoint.setValue(10., 0.)
 190 aLineDEndPoint.setValue(100., -50.)
 191 aLineEStartPoint.setValue(100., -50.)
 192 aLineEEndPoint.setValue(0., -50.)
 193 aSession.finishOperation()
 194
 195 # Extrude clamp
 196 aClampSketchResult = aClampSketchFeature.firstResult()
 197 aClampSketchEdges = modelAPI_ResultConstruction(aClampSketchResult).shape()
 198 origin = geomDataAPI_Point(aClampSketchFeature.attribute("Origin")).pnt()
 199 dirX = geomDataAPI_Dir(aClampSketchFeature.attribute("DirX")).dir()
 200 norm = geomDataAPI_Dir(aClampSketchFeature.attribute("Norm")).dir()
 201 aClampSketchFaces = ShapeList()
 202 GeomAlgoAPI_SketchBuilder.createFaces(
 203     origin, dirX, norm, aClampSketchEdges, aClampSketchFaces)
 204 aSession.startOperation()
 205 aClampExtrusionFt = aPart.addFeature("Extrusion")
 206 assert (aClampExtrusionFt.getKind() == "Extrusion")
 207 # selection type FACE=4
 208 aClampExtrusionFt.selectionList("base").append(
 209     aClampSketchResult, aClampSketchFaces[0])
 210 aClampExtrusionFt.string("CreationMethod").setValue("BySizes")
 211 aClampExtrusionFt.real("to_size").setValue(70)
 212 aClampExtrusionFt.real("from_size").setValue(0)
 213 aClampExtrusionFt.real("to_offset").setValue(0) #TODO: remove
 214 aClampExtrusionFt.real("from_offset").setValue(0) #TODO: remove
 215 aClampExtrusionFt.execute()
 216 aSession.finishOperation()
 217
 218 # Check extrusion results
 219 assert (len(aClampExtrusionFt.results()) > 0)
 220 anExtrusionResult = modelAPI_ResultBody(aClampExtrusionFt.firstResult())
 221 assert (anExtrusionResult is not None)
 222
 223 # Extrude circle
 224 aSession.startOperation()
 225 anExtrusionFt = aPart.addFeature("Extrusion")
 226 assert (anExtrusionFt.getKind() == "Extrusion")
 227 # selection type FACE=4
 228 anExtrusionFt.selectionList("base").append(
 229     aCircleSketchResult, aCircleSketchFaces[0])
 230 anExtrusionFt.string("CreationMethod").setValue("ByPlanesAndOffsets")
 231 anExtrusionFt.real("to_size").setValue(0) #TODO: remove
 232 anExtrusionFt.real("from_size").setValue(0) #TODO: remove
 233 anExtrusionFt.selection("to_object").selectSubShape("face", "Extrusion_3_1/Generated_Face_2")
 234 anExtrusionFt.real("to_offset").setValue(0)
 235 anExtrusionFt.selection("from_object").selectSubShape("face", "Extrusion_3_1/Generated_Face_1")
 236 anExtrusionFt.real("from_offset").setValue(0)
 237 anExtrusionFt.execute()
 238 aSession.finishOperation()
 239
 240 # Check extrusion results
 241 assert (len(anExtrusionFt.results()) > 0)
 242 anExtrusionResult = modelAPI_ResultBody(anExtrusionFt.firstResult())
 243 assert (anExtrusionResult is not None)
 244
 245 #=========================================================================
 246 # End of test
 247 #=========================================================================
 248
 249 from salome.shaper import model
 250 assert(model.checkPythonDump())