src/FeaturesPlugin/Test/TestExtrusionFuse.py

   1 """
   2       TestExtrusionFuse.py
   3       Unit test of FeaturesPlugin_ExtrusionFuse class
   4
   5       class FeaturesPlugin_ExtrusionFuse : public FeaturesPlugin_ExtrusionBoolean
   6                 static const std::string MY_EXTRUSION_ID("ExtrusionFuse");
   7         static const std::string MY_SKETCH_OBJECT_ID("sketch");
   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         static const std::string MY_BOOLEAN_OBJECTS_ID("boolean_objects");
  16
  17         data()->addAttribute(SKETCH_OBJECT_ID(), ModelAPI_AttributeReference::typeId());
  18         data()->addAttribute(CREATION_METHOD(), ModelAPI_AttributeString::typeId());
  19         data()->addAttribute(TO_SIZE_ID(), ModelAPI_AttributeDouble::typeId());
  20         data()->addAttribute(FROM_SIZE_ID(), ModelAPI_AttributeDouble::typeId());
  21         data()->addAttribute(TO_OBJECT_ID(), ModelAPI_AttributeSelection::typeId());
  22         data()->addAttribute(TO_OFFSET_ID(), ModelAPI_AttributeDouble::typeId());
  23         data()->addAttribute(FROM_OBJECT_ID(), ModelAPI_AttributeSelection::typeId());
  24         data()->addAttribute(FROM_OFFSET_ID(), ModelAPI_AttributeSelection::typeId());
  25         data()->addAttribute(BOOLEAN_OBJECTS_ID(), ModelAPI_AttributeSelectionList::typeId());
  26 """
  27 #=========================================================================
  28 # Initialization of the test
  29 #=========================================================================
  30 from ModelAPI import *
  31 from GeomDataAPI import *
  32 from GeomAlgoAPI import *
  33 from GeomAPI import *
  34 import math
  35
  36 __updated__ = "2014-12-16"
  37
  38 aSession = ModelAPI_Session.get()
  39 aDocument = aSession.moduleDocument()
  40 # Create a part for extrusion
  41 aSession.startOperation()
  42 aPartFeature = aDocument.addFeature("Part")
  43 aSession.finishOperation()
  44 assert (len(aPartFeature.results()) == 1)
  45 # Another way is:
  46 # aPart = aSession.activeDocument()
  47 aPartResult = modelAPI_ResultPart(aPartFeature.firstResult())
  48 aPart = aPartResult.partDoc()
  49 #=========================================================================
  50 # Create a sketch circle to extrude
  51 #=========================================================================
  52 aSession.startOperation()
  53 aCircleSketchFeature = featureToCompositeFeature(aPart.addFeature("Sketch"))
  54 origin = geomDataAPI_Point(aCircleSketchFeature.attribute("Origin"))
  55 origin.setValue(0, 0, 0)
  56 dirx = geomDataAPI_Dir(aCircleSketchFeature.attribute("DirX"))
  57 dirx.setValue(1, 0, 0)
  58 norm = geomDataAPI_Dir(aCircleSketchFeature.attribute("Norm"))
  59 norm.setValue(0, 0, 1)
  60 # Create circle
  61 aSketchCircle = aCircleSketchFeature.addFeature("SketchCircle")
  62 anCircleCentr = geomDataAPI_Point2D(aSketchCircle.attribute("CircleCenter"))
  63 aCircleRadius = aSketchCircle.real("CircleRadius")
  64 anCircleCentr.setValue(0, 0)
  65 aCircleRadius.setValue(20)
  66 aSession.finishOperation()
  67 #=========================================================================
  68 # Make extrusion on circle
  69 #=========================================================================
  70 # Build shape from sketcher results
  71 aCircleSketchResult = aCircleSketchFeature.firstResult()
  72 aCircleSketchEdges = modelAPI_ResultConstruction(aCircleSketchResult).shape()
  73 origin = geomDataAPI_Point(aCircleSketchFeature.attribute("Origin")).pnt()
  74 dirX = geomDataAPI_Dir(aCircleSketchFeature.attribute("DirX")).dir()
  75 norm = geomDataAPI_Dir(aCircleSketchFeature.attribute("Norm")).dir()
  76 aCircleSketchFaces = ShapeList()
  77 GeomAlgoAPI_SketchBuilder.createFaces(
  78     origin, dirX, norm, aCircleSketchEdges, aCircleSketchFaces)
  79 assert (len(aCircleSketchFaces) > 0)
  80 assert (aCircleSketchFaces[0] is not None)
  81 # Create extrusion
  82 aSession.startOperation()
  83 anExtrusionFt = aPart.addFeature("Extrusion")
  84 assert (anExtrusionFt.getKind() == "Extrusion")
  85 # selection type FACE=4
  86 anExtrusionFt.selectionList("base").append(
  87     aCircleSketchResult, aCircleSketchFaces[0])
  88 anExtrusionFt.string("CreationMethod").setValue("BySizes")
  89 anExtrusionFt.real("to_size").setValue(50)
  90 anExtrusionFt.real("from_size").setValue(0)
  91 anExtrusionFt.real("to_offset").setValue(0) #TODO: remove
  92 anExtrusionFt.real("from_offset").setValue(0) #TODO: remove
  93 anExtrusionFt.execute()
  94 aSession.finishOperation()
  95 assert (anExtrusionFt.real("to_size").value() == 50.0)
  96
  97 # Check extrusion results
  98 assert (len(anExtrusionFt.results()) > 0)
  99 anExtrusionResult = modelAPI_ResultBody(anExtrusionFt.firstResult())
 100 assert (anExtrusionResult is not None)
 101
 102 #=========================================================================
 103 # Make extrusion cut
 104 #=========================================================================
 105 aSession.startOperation()
 106 anExtrusionFuseFt = featureToCompositeFeature(aPart.addFeature("ExtrusionFuse"))
 107 assert (anExtrusionFuseFt.getKind() == "ExtrusionFuse")
 108 # selection type FACE=4
 109 aSession.startOperation()
 110 aCircleSketchFeature = featureToCompositeFeature(anExtrusionFuseFt.addFeature("Sketch"))
 111 origin = geomDataAPI_Point(aCircleSketchFeature.attribute("Origin"))
 112 origin.setValue(0, 0, 50)
 113 dirx = geomDataAPI_Dir(aCircleSketchFeature.attribute("DirX"))
 114 dirx.setValue(1, 0, 0)
 115 norm = geomDataAPI_Dir(aCircleSketchFeature.attribute("Norm"))
 116 norm.setValue(0, 0, 1)
 117 aCircleSketchFeature.selection("External").selectSubShape("face", "Extrusion_1/TopFace_1")
 118 aSketchCircle = aCircleSketchFeature.addFeature("SketchCircle")
 119 anCircleCentr = geomDataAPI_Point2D(aSketchCircle.attribute("CircleCenter"))
 120 aCircleRadius = aSketchCircle.real("CircleRadius")
 121 anCircleCentr.setValue(0, 0)
 122 aCircleRadius.setValue(10)
 123 aSession.finishOperation()
 124 aSession.startOperation()
 125 anExtrusionFuseFt.string("CreationMethod").setValue("BySizes")
 126 anExtrusionFuseFt.real("to_size").setValue(10)
 127 anExtrusionFuseFt.real("from_size").setValue(10)
 128 anExtrusionFuseFt.real("to_offset").setValue(0) #TODO: remove
 129 anExtrusionFuseFt.real("from_offset").setValue(0) #TODO: remove
 130 anExtrusionFuseFt.selectionList("boolean_objects").append(anExtrusionResult, anExtrusionResult.shape())
 131 anExtrusionFuseFt.execute()
 132 aSession.finishOperation()
 133 aSession.finishOperation()
 134
 135 #=========================================================================
 136 # Test results
 137 #=========================================================================
 138 aFactory = ModelAPI_Session.get().validators()
 139 assert (aFactory.validate(anExtrusionFuseFt))
 140 assert (len(anExtrusionFuseFt.results()) > 0)
 141 aCurrentResult = modelAPI_ResultBody(anExtrusionFuseFt.firstResult())
 142 assert (aCurrentResult is not None)