src/FeaturesPlugin/Test/TestRotation_MultiLevelCompound_v0_4.py

   1 # Copyright (C) 2020-2023  CEA/DEN, EDF R&D
   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 from salome.shaper import model
  21 from GeomAPI import *
  22
  23 model.begin()
  24 partSet = model.moduleDocument()
  25 Part_1 = model.addPart(partSet)
  26 Part_1_doc = Part_1.document()
  27 Sketch_1 = model.addSketch(Part_1_doc, model.defaultPlane("XOY"))
  28 SketchPoint_1 = Sketch_1.addPoint(44.29784155360136, 17.09942588468031)
  29 SketchArc_1 = Sketch_1.addArc(44.29784155360136, 17.09942588468031, 47.1668727423061, 12.27945348765633, 45.38299232715926, 22.60269052200972, False)
  30 SketchConstraintCoincidence_1 = Sketch_1.setCoincident(SketchPoint_1.coordinates(), SketchArc_1.center())
  31 SketchLine_1 = Sketch_1.addLine(42.5764228403785, 22.14892077680068, 39.70739165167375, 16.41085839939117)
  32 SketchLine_2 = Sketch_1.addLine(39.70739165167375, 16.41085839939117, 43.03546783057126, 12.04993099255995)
  33 SketchConstraintCoincidence_2 = Sketch_1.setCoincident(SketchLine_1.endPoint(), SketchLine_2.startPoint())
  34 SketchLine_3 = Sketch_1.addLine(28.57555063949931, 12.96802097294547, 15.72229091410204, 12.96802097294547)
  35 SketchLine_4 = Sketch_1.addLine(15.72229091410204, 12.96802097294547, 15.72229091410204, 21.46035329151154)
  36 SketchLine_5 = Sketch_1.addLine(15.72229091410204, 21.46035329151154, 28.57555063949931, 21.46035329151154)
  37 SketchLine_6 = Sketch_1.addLine(28.57555063949931, 21.46035329151154, 28.57555063949931, 12.96802097294547)
  38 SketchConstraintCoincidence_3 = Sketch_1.setCoincident(SketchLine_6.endPoint(), SketchLine_3.startPoint())
  39 SketchConstraintCoincidence_4 = Sketch_1.setCoincident(SketchLine_3.endPoint(), SketchLine_4.startPoint())
  40 SketchConstraintCoincidence_5 = Sketch_1.setCoincident(SketchLine_4.endPoint(), SketchLine_5.startPoint())
  41 SketchConstraintCoincidence_6 = Sketch_1.setCoincident(SketchLine_5.endPoint(), SketchLine_6.startPoint())
  42 SketchConstraintHorizontal_1 = Sketch_1.setHorizontal(SketchLine_3.result())
  43 SketchConstraintVertical_1 = Sketch_1.setVertical(SketchLine_4.result())
  44 SketchConstraintHorizontal_2 = Sketch_1.setHorizontal(SketchLine_5.result())
  45 SketchConstraintVertical_2 = Sketch_1.setVertical(SketchLine_6.result())
  46 SketchLine_7 = Sketch_1.addLine(-10.67279602198167, 14.78814178154371, -28.34602814440294, 14.78814178154371)
  47 SketchLine_8 = Sketch_1.addLine(-28.34602814440294, 14.78814178154371, -28.34602814440294, 25.13271321305362)
  48 SketchLine_9 = Sketch_1.addLine(-28.34602814440294, 25.13271321305362, -10.67279602198167, 25.13271321305362)
  49 SketchLine_10 = Sketch_1.addLine(-10.67279602198167, 25.13271321305362, -10.67279602198167, 14.78814178154371)
  50 SketchConstraintCoincidence_7 = Sketch_1.setCoincident(SketchLine_10.endPoint(), SketchLine_7.startPoint())
  51 SketchConstraintCoincidence_8 = Sketch_1.setCoincident(SketchLine_7.endPoint(), SketchLine_8.startPoint())
  52 SketchConstraintCoincidence_9 = Sketch_1.setCoincident(SketchLine_8.endPoint(), SketchLine_9.startPoint())
  53 SketchConstraintCoincidence_10 = Sketch_1.setCoincident(SketchLine_9.endPoint(), SketchLine_10.startPoint())
  54 SketchConstraintHorizontal_3 = Sketch_1.setHorizontal(SketchLine_7.result())
  55 SketchConstraintVertical_3 = Sketch_1.setVertical(SketchLine_8.result())
  56 SketchConstraintHorizontal_4 = Sketch_1.setHorizontal(SketchLine_9.result())
  57 SketchConstraintVertical_4 = Sketch_1.setVertical(SketchLine_10.result())
  58 SketchLine_11 = Sketch_1.addLine(-17.67323212242127, 25.13271321305362, -21.80463703415611, 14.78814178154371)
  59 SketchConstraintCoincidence_11 = Sketch_1.setCoincident(SketchLine_11.startPoint(), SketchLine_9.result())
  60 SketchConstraintCoincidence_12 = Sketch_1.setCoincident(SketchLine_11.endPoint(), SketchLine_7.result())
  61 model.do()
  62 Vertex_1 = model.addVertex(Part_1_doc, [model.selection("VERTEX", "Sketch_1/SketchArc_1")], False)
  63 Edge_1 = model.addEdge(Part_1_doc, [model.selection("EDGE", "Sketch_1/SketchArc_1_2")], False)
  64 Wire_1 = model.addWire(Part_1_doc, [model.selection("EDGE", "Sketch_1/SketchLine_1"), model.selection("EDGE", "Sketch_1/SketchLine_2")], False)
  65 Face_1 = model.addFace(Part_1_doc, [model.selection("FACE", "Sketch_1/Face-SketchLine_6r-SketchLine_5r-SketchLine_4r-SketchLine_3r")])
  66 Shell_1 = model.addShell(Part_1_doc, [model.selection("FACE", "Sketch_1/Face-SketchLine_10r-SketchLine_9r-SketchLine_11f-SketchLine_7r"), model.selection("FACE", "Sketch_1/Face-SketchLine_11r-SketchLine_9r-SketchLine_8r-SketchLine_7r")])
  67 Box_1 = model.addBox(Part_1_doc, 10, 10, 10)
  68 Translation_1 = model.addTranslation(Part_1_doc, [model.selection("SOLID", "Box_1_1")], model.selection("EDGE", "PartSet/OX"), 50)
  69 Box_2 = model.addBox(Part_1_doc, 10, 10, 10)
  70 Cylinder_1 = model.addCylinder(Part_1_doc, model.selection("VERTEX", "PartSet/Origin"), model.selection("EDGE", "PartSet/OZ"), 5, 10)
  71 Partition_1 = model.addPartition(Part_1_doc, [model.selection("SOLID", "Box_2_1"), model.selection("SOLID", "Cylinder_1_1")])
  72 Box_3 = model.addBox(Part_1_doc, 10, 10, 10)
  73 Translation_2 = model.addTranslation(Part_1_doc, [model.selection("SOLID", "Box_3_1")], model.selection("EDGE", "PartSet/OY"), 50)
  74 AngularCopy_1 = model.addMultiRotation(Part_1_doc, [model.selection("SOLID", "Translation_2_1")], model.selection("EDGE", "PartSet/OZ"), 30, 3)
  75 Compound_1_objects = [model.selection("VERTEX", "Vertex_1_1"), model.selection("EDGE", "Edge_1_1"), model.selection("WIRE", "Wire_1_1"), model.selection("FACE", "Face_1_1"), model.selection("SHELL", "Shell_1_1"), model.selection("SOLID", "Translation_1_1"), model.selection("COMPSOLID", "Partition_1_1"), model.selection("COMPOUND", "AngularCopy_1_1")]
  76 Compound_1 = model.addCompound(Part_1_doc, Compound_1_objects)
  77 model.end()
  78
  79 TOLERANCE = 1.e-7
  80 ANGLE = 30
  81 AXIS = GeomAPI_Ax1(GeomAPI_Pnt(0, 0, 0), GeomAPI_Dir(0, 0, 1))
  82
  83
  84 model.begin()
  85 Rotation_1 = model.addRotation(Part_1_doc, [model.selection("VERTEX", "Compound_1_1_1")], model.selection("EDGE", "PartSet/OZ"), ANGLE)
  86 model.testNbResults(Rotation_1, 1)
  87 model.testNbSubResults(Rotation_1, [0])
  88 model.testNbSubShapes(Rotation_1, GeomAPI_Shape.SOLID, [0])
  89 model.testNbSubShapes(Rotation_1, GeomAPI_Shape.FACE, [0])
  90 model.testNbSubShapes(Rotation_1, GeomAPI_Shape.EDGE, [0])
  91 model.testNbSubShapes(Rotation_1, GeomAPI_Shape.VERTEX, [1])
  92 model.testResultsVolumes(Rotation_1, [0])
  93 model.testResultsAreas(Rotation_1, [0])
  94 refPoint = Compound_1.result().subResult(0).resultSubShapePair()[0].shape().middlePoint()
  95 refPoint.rotate(AXIS, ANGLE)
  96 midPoint = Rotation_1.defaultResult().shape().middlePoint()
  97 assert(midPoint.distance(refPoint) < TOLERANCE)
  98
  99 Recover_1 = model.addRecover(Part_1_doc, Rotation_1, [Compound_1.result()], True)
 100 Rotation_2 = model.addRotation(Part_1_doc, [model.selection("EDGE", "Recover_1_1_2")], model.selection("EDGE", "PartSet/OZ"), ANGLE)
 101 model.testNbResults(Rotation_2, 1)
 102 model.testNbSubResults(Rotation_2, [0])
 103 model.testNbSubShapes(Rotation_2, GeomAPI_Shape.SOLID, [0])
 104 model.testNbSubShapes(Rotation_2, GeomAPI_Shape.FACE, [0])
 105 model.testNbSubShapes(Rotation_2, GeomAPI_Shape.EDGE, [1])
 106 model.testNbSubShapes(Rotation_2, GeomAPI_Shape.VERTEX, [2])
 107 model.testResultsVolumes(Rotation_2, [0])
 108 model.testResultsAreas(Rotation_2, [0])
 109 refPoint = Compound_1.result().subResult(1).resultSubShapePair()[0].shape().middlePoint()
 110 refPoint.rotate(AXIS, ANGLE)
 111 midPoint = Rotation_2.defaultResult().shape().middlePoint()
 112 assert(midPoint.distance(refPoint) < TOLERANCE)
 113
 114 Recover_2 = model.addRecover(Part_1_doc, Rotation_2, [Recover_1.result()])
 115 Rotation_3 = model.addRotation(Part_1_doc, [model.selection("WIRE", "Recover_2_1_3")], model.selection("EDGE", "PartSet/OZ"), ANGLE)
 116 model.testNbResults(Rotation_3, 1)
 117 model.testNbSubResults(Rotation_3, [0])
 118 model.testNbSubShapes(Rotation_3, GeomAPI_Shape.SOLID, [0])
 119 model.testNbSubShapes(Rotation_3, GeomAPI_Shape.FACE, [0])
 120 model.testNbSubShapes(Rotation_3, GeomAPI_Shape.EDGE, [2])
 121 model.testNbSubShapes(Rotation_3, GeomAPI_Shape.VERTEX, [4])
 122 model.testResultsVolumes(Rotation_3, [0])
 123 model.testResultsAreas(Rotation_3, [0])
 124 refPoint = Compound_1.result().subResult(2).resultSubShapePair()[0].shape().middlePoint()
 125 refPoint.rotate(AXIS, ANGLE)
 126 midPoint = Rotation_3.defaultResult().shape().middlePoint()
 127 assert(midPoint.distance(refPoint) < TOLERANCE)
 128
 129 Recover_3 = model.addRecover(Part_1_doc, Rotation_3, [Recover_2.result()])
 130 Rotation_4 = model.addRotation(Part_1_doc, [model.selection("FACE", "Recover_3_1_4")], model.selection("EDGE", "PartSet/OZ"), ANGLE)
 131 model.testNbResults(Rotation_4, 1)
 132 model.testNbSubResults(Rotation_4, [0])
 133 model.testNbSubShapes(Rotation_4, GeomAPI_Shape.SOLID, [0])
 134 model.testNbSubShapes(Rotation_4, GeomAPI_Shape.FACE, [1])
 135 model.testNbSubShapes(Rotation_4, GeomAPI_Shape.EDGE, [4])
 136 model.testNbSubShapes(Rotation_4, GeomAPI_Shape.VERTEX, [8])
 137 model.testResultsVolumes(Rotation_4, [0])
 138 model.testResultsAreas(Rotation_4, [109.154152964914914])
 139 refPoint = Compound_1.result().subResult(3).resultSubShapePair()[0].shape().middlePoint()
 140 refPoint.rotate(AXIS, ANGLE)
 141 midPoint = Rotation_4.defaultResult().shape().middlePoint()
 142 assert(midPoint.distance(refPoint) < TOLERANCE)
 143
 144 Recover_4 = model.addRecover(Part_1_doc, Rotation_4, [Recover_3.result()])
 145 Rotation_5 = model.addRotation(Part_1_doc, [model.selection("SHELL", "Recover_4_1_5")], model.selection("EDGE", "PartSet/OZ"), ANGLE)
 146 model.testNbResults(Rotation_5, 1)
 147 model.testNbSubResults(Rotation_5, [0])
 148 model.testNbSubShapes(Rotation_5, GeomAPI_Shape.SOLID, [0])
 149 model.testNbSubShapes(Rotation_5, GeomAPI_Shape.FACE, [2])
 150 model.testNbSubShapes(Rotation_5, GeomAPI_Shape.EDGE, [8])
 151 model.testNbSubShapes(Rotation_5, GeomAPI_Shape.VERTEX, [16])
 152 model.testResultsVolumes(Rotation_5, [0])
 153 model.testResultsAreas(Rotation_5, [182.822012116])
 154 refPoint = Compound_1.result().subResult(4).resultSubShapePair()[0].shape().middlePoint()
 155 refPoint.rotate(AXIS, ANGLE)
 156 midPoint = Rotation_5.defaultResult().shape().middlePoint()
 157 assert(midPoint.distance(refPoint) < TOLERANCE)
 158
 159 Recover_5 = model.addRecover(Part_1_doc, Rotation_5, [Recover_4.result()])
 160 Rotation_6 = model.addRotation(Part_1_doc, [model.selection("SOLID", "Recover_5_1_6")], model.selection("EDGE", "PartSet/OZ"), ANGLE)
 161 model.testNbResults(Rotation_6, 1)
 162 model.testNbSubResults(Rotation_6, [0])
 163 model.testNbSubShapes(Rotation_6, GeomAPI_Shape.SOLID, [1])
 164 model.testNbSubShapes(Rotation_6, GeomAPI_Shape.FACE, [6])
 165 model.testNbSubShapes(Rotation_6, GeomAPI_Shape.EDGE, [24])
 166 model.testNbSubShapes(Rotation_6, GeomAPI_Shape.VERTEX, [48])
 167 model.testResultsVolumes(Rotation_6, [1000])
 168 refPoint = Compound_1.result().subResult(5).resultSubShapePair()[0].shape().middlePoint()
 169 refPoint.rotate(AXIS, ANGLE)
 170 midPoint = Rotation_6.defaultResult().shape().middlePoint()
 171 assert(midPoint.distance(refPoint) < TOLERANCE)
 172
 173 Recover_6 = model.addRecover(Part_1_doc, Rotation_6, [Recover_5.result()])
 174 Rotation_7 = model.addRotation(Part_1_doc, [model.selection("COMPSOLID", "Recover_6_1_7")], model.selection("EDGE", "PartSet/OZ"), ANGLE)
 175 model.testNbResults(Rotation_7, 1)
 176 model.testNbSubResults(Rotation_7, [3])
 177 model.testNbSubShapes(Rotation_7, GeomAPI_Shape.SOLID, [3])
 178 model.testNbSubShapes(Rotation_7, GeomAPI_Shape.FACE, [17])
 179 model.testNbSubShapes(Rotation_7, GeomAPI_Shape.EDGE, [66])
 180 model.testNbSubShapes(Rotation_7, GeomAPI_Shape.VERTEX, [132])
 181 model.testResultsVolumes(Rotation_7, [1589.0486226])
 182 refPoint = GeomAPI_Pnt(1.830127, 4.330127, 5.0)
 183 midPoint = Rotation_7.defaultResult().shape().middlePoint()
 184 assert(midPoint.distance(refPoint) < TOLERANCE)
 185
 186 Recover_7 = model.addRecover(Part_1_doc, Rotation_7, [Recover_6.result()])
 187 Rotation_8 = model.addRotation(Part_1_doc, [model.selection("COMPOUND", "Recover_7_1_8")], model.selection("EDGE", "PartSet/OZ"), ANGLE)
 188 model.testNbResults(Rotation_8, 1)
 189 model.testNbSubResults(Rotation_8, [3])
 190 model.testNbSubShapes(Rotation_8, GeomAPI_Shape.SOLID, [3])
 191 model.testNbSubShapes(Rotation_8, GeomAPI_Shape.FACE, [18])
 192 model.testNbSubShapes(Rotation_8, GeomAPI_Shape.EDGE, [72])
 193 model.testNbSubShapes(Rotation_8, GeomAPI_Shape.VERTEX, [144])
 194 model.testResultsVolumes(Rotation_8, [3000])
 195 refPoint = GeomAPI_Pnt(-38.169872981, 28.4807621, 5.0)
 196 midPoint = Rotation_8.defaultResult().shape().middlePoint()
 197 assert(midPoint.distance(refPoint) < TOLERANCE)
 198
 199 model.end()
 200
 201 assert(model.checkPythonDump())