src/FeaturesPlugin/Test/TestPartition.py

   1 #=========================================================================
   2 # Initialization of the test
   3 #=========================================================================
   4 from ModelAPI import *
   5 from GeomDataAPI import *
   6 from GeomAlgoAPI import *
   7 from GeomAPI import *
   8 import math
   9
  10 __updated__ = "2014-12-16"
  11
  12 aSession = ModelAPI_Session.get()
  13 aDocument = aSession.moduleDocument()
  14 # Create a part for extrusion
  15 aSession.startOperation()
  16 aPartFeature = aDocument.addFeature("Part")
  17 aSession.finishOperation()
  18 assert (len(aPartFeature.results()) == 1)
  19 # Another way is:
  20 # aPart = aSession.activeDocument()
  21 aPartResult = modelAPI_ResultPart(aPartFeature.firstResult())
  22 aPart = aPartResult.partDoc()
  23
  24 #=========================================================================
  25 # Create a sketch circle to extrude
  26 #=========================================================================
  27 aSession.startOperation()
  28 aCircleSketchFeature = featureToCompositeFeature(aPart.addFeature("Sketch"))
  29 origin = geomDataAPI_Point(aCircleSketchFeature.attribute("Origin"))
  30 origin.setValue(0, 0, 0)
  31 dirx = geomDataAPI_Dir(aCircleSketchFeature.attribute("DirX"))
  32 dirx.setValue(1, 0, 0)
  33 norm = geomDataAPI_Dir(aCircleSketchFeature.attribute("Norm"))
  34 norm.setValue(0, 0, 1)
  35 # Create circle
  36 aSketchCircle = aCircleSketchFeature.addFeature("SketchCircle")
  37 anCircleCentr = geomDataAPI_Point2D(aSketchCircle.attribute("CircleCenter"))
  38 aCircleRadius = aSketchCircle.real("CircleRadius")
  39 anCircleCentr.setValue(0, 0)
  40 aCircleRadius.setValue(50)
  41 aSession.finishOperation()
  42
  43 #=========================================================================
  44 # Make extrusion on circle
  45 #=========================================================================
  46 # Build shape from sketcher results
  47 aCircleSketchResult = aCircleSketchFeature.firstResult()
  48 aCircleSketchEdges = modelAPI_ResultConstruction(aCircleSketchResult).shape()
  49 origin = geomDataAPI_Point(aCircleSketchFeature.attribute("Origin")).pnt()
  50 dirX = geomDataAPI_Dir(aCircleSketchFeature.attribute("DirX")).dir()
  51 norm = geomDataAPI_Dir(aCircleSketchFeature.attribute("Norm")).dir()
  52 aCircleSketchFaces = ShapeList()
  53 GeomAlgoAPI_SketchBuilder.createFaces(
  54     origin, dirX, norm, aCircleSketchEdges, aCircleSketchFaces)
  55 assert (len(aCircleSketchFaces) > 0)
  56 assert (aCircleSketchFaces[0] is not None)
  57 # Create extrusion
  58 aSession.startOperation()
  59 anExtrusionFt = aPart.addFeature("Extrusion")
  60 assert (anExtrusionFt.getKind() == "Extrusion")
  61 # selection type FACE=4
  62 anExtrusionFt.selectionList("base").append(
  63     aCircleSketchResult, aCircleSketchFaces[0])
  64 anExtrusionFt.string("CreationMethod").setValue("BySizes")
  65 anExtrusionFt.real("to_size").setValue(50)
  66 anExtrusionFt.real("from_size").setValue(0)
  67 anExtrusionFt.real("to_offset").setValue(0) #TODO: remove
  68 anExtrusionFt.real("from_offset").setValue(0) #TODO: remove
  69 anExtrusionFt.execute()
  70 aSession.finishOperation()
  71 assert (anExtrusionFt.real("to_size").value() == 50)
  72
  73 # Check extrusion results
  74 assert (len(anExtrusionFt.results()) > 0)
  75 anExtrusionResult = modelAPI_ResultBody(anExtrusionFt.firstResult())
  76 assert (anExtrusionResult is not None)
  77
  78 #=========================================================================
  79 # Create a partition
  80 #=========================================================================
  81 aXOZPlane = modelAPI_Result(aDocument.objectByName("Construction", "XOZ"))
  82 aSession.startOperation()
  83 aPartitionFt = aPart.addFeature("Partition")
  84 assert (aPartitionFt.getKind() == "Partition")
  85 aPartitionFt.selectionList("main_objects").append(anExtrusionResult, anExtrusionResult.shape())
  86 aPartitionFt.selectionList("tool_objects").append(aXOZPlane, None)
  87 aPartitionFt.execute()
  88 aSession.finishOperation()
  89
  90 assert (len(aPartitionFt.results()) > 0)
  91 aPartitionResult = modelAPI_ResultBody(aPartitionFt.firstResult())
  92 assert (aPartitionResult is not None)