src/FeaturesPlugin/Test/TestExtrusion.py

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