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 plane
  80 #=========================================================================
  81 aSession.startOperation()
  82 aPlaneFeature = aPart.addFeature("Plane")
  83 aPlaneFeature.string("creation_method").setValue("by_general_equation")
  84 aPlaneFeature.string("by_other_plane_option").setValue("by_distance_from_other") # TODO: remove
  85 aPlaneFeature.real("A").setValue(0.)
  86 aPlaneFeature.real("B").setValue(1.)
  87 aPlaneFeature.real("C").setValue(0.)
  88 aPlaneFeature.real("D").setValue(1.)
  89 aSession.finishOperation()
  90 aPlaneResult = aPlaneFeature.firstResult()
  91
  92 #=========================================================================
  93 # Create a partition
  94 #=========================================================================
  95 aSession.startOperation()
  96 aPartitionFt = aPart.addFeature("Partition")
  97 assert (aPartitionFt.getKind() == "Partition")
  98 aPartitionFt.selectionList("base_objects").append(anExtrusionResult, anExtrusionResult.shape())
  99 aPartitionFt.selectionList("base_objects").append(aPlaneResult, None)
 100 aPartitionFt.execute()
 101 aSession.finishOperation()
 102
 103 #=========================================================================
 104 # Test results
 105 #=========================================================================
 106 aFactory = ModelAPI_Session.get().validators()
 107 assert (aFactory.validate(aPartitionFt))
 108 assert (len(aPartitionFt.results()) > 0)
 109 aPartitionResult = modelAPI_ResultBody(aPartitionFt.firstResult())
 110 assert (aPartitionResult is not None)
 111 aSession.undo()
 112
 113 #=========================================================================
 114 # Create a partition
 115 #=========================================================================
 116 aSession.startOperation()
 117 aPartitionFt = aPart.addFeature("Partition")
 118 assert (aPartitionFt.getKind() == "Partition")
 119 aPartitionFt.selectionList("base_objects").append(anExtrusionResult, anExtrusionResult.shape())
 120 aPartitionFt.selectionList("base_objects").append(aPlaneResult, None)
 121 aPartitionFt.execute()
 122 aSession.finishOperation()
 123
 124 #=========================================================================
 125 # Test results
 126 #=========================================================================
 127 aFactory = ModelAPI_Session.get().validators()
 128 assert (aFactory.validate(aPartitionFt))
 129 assert (len(aPartitionFt.results()) > 0)
 130 aPartitionResult = modelAPI_ResultBody(aPartitionFt.firstResult())
 131 assert (aPartitionResult is not None)
 132
 133 from salome.shaper import model
 134 assert(model.checkPythonDump())