src/FeaturesPlugin/Test/TestSymmetry_MultiLevelCompound_v95_5.py

   1 # Copyright (C) 2020-2022  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")], keepSubResults = True)
  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 # collect reference data
  80 REFERENCE = []
  81 TOLERANCE = 1.e-7
  82 for ind in range(0, Compound_1.result().numberOfSubs()):
  83     p = Compound_1.result().subResult(ind).resultSubShapePair()[1].middlePoint()
  84     REFERENCE.append(p)
  85
  86 def assertResult(theFeature, theNbMoved):
  87     model.testNbResults(theFeature, 1)
  88     model.testNbSubResults(theFeature, [8])
  89     model.testNbSubShapes(theFeature, GeomAPI_Shape.SOLID, [7])
  90     model.testNbSubShapes(theFeature, GeomAPI_Shape.FACE, [44])
  91     model.testNbSubShapes(theFeature, GeomAPI_Shape.EDGE, [177])
  92     model.testNbSubShapes(theFeature, GeomAPI_Shape.VERTEX, [355])
  93     model.testResultsVolumes(theFeature, [5589.0486226])
  94
  95     for ind in range(0, theFeature.result().numberOfSubs()):
  96         ref = GeomAPI_Pnt(REFERENCE[ind].x(), REFERENCE[ind].y(), REFERENCE[ind].z())
  97         if ind < theNbMoved:
  98             ref.setY(-ref.y())
  99             ref.setZ(-ref.z())
 100         midPoint = theFeature.result().subResult(ind).resultSubShapePair()[0].shape().middlePoint()
 101         assert(midPoint.distance(ref) < TOLERANCE), "Sub-result {}; actual ({}, {}, {}) != expected ({}, {}, {})".format(ind, midPoint.x(), midPoint.y(), midPoint.z(), ref.x(), ref.y(), ref.z())
 102
 103
 104 model.begin()
 105 Symmetry_1 = model.addSymmetry(Part_1_doc, [model.selection("VERTEX", "Compound_1_1_1")], model.selection("EDGE", "PartSet/OX"), keepOriginal = False, keepSubResults = True)
 106 assertResult(Symmetry_1, 1)
 107
 108 Symmetry_2 = model.addSymmetry(Part_1_doc, [model.selection("EDGE", "Symmetry_1_1_2")], model.selection("EDGE", "PartSet/OX"), keepOriginal = False, keepSubResults = True)
 109 assertResult(Symmetry_2, 2)
 110
 111 Symmetry_3 = model.addSymmetry(Part_1_doc, [model.selection("WIRE", "Symmetry_2_1_3")], model.selection("EDGE", "PartSet/OX"), keepOriginal = False, keepSubResults = True)
 112 assertResult(Symmetry_3, 3)
 113
 114 Symmetry_4 = model.addSymmetry(Part_1_doc, [model.selection("FACE", "Symmetry_3_1_4")], model.selection("EDGE", "PartSet/OX"), keepOriginal = False, keepSubResults = True)
 115 assertResult(Symmetry_4, 4)
 116
 117 Symmetry_5 = model.addSymmetry(Part_1_doc, [model.selection("SHELL", "Symmetry_4_1_5")], model.selection("EDGE", "PartSet/OX"), keepOriginal = False, keepSubResults = True)
 118 assertResult(Symmetry_5, 5)
 119
 120 Symmetry_6 = model.addSymmetry(Part_1_doc, [model.selection("SOLID", "Symmetry_5_1_6")], model.selection("EDGE", "PartSet/OX"), keepOriginal = False, keepSubResults = True)
 121 assertResult(Symmetry_6, 6)
 122
 123 Symmetry_7 = model.addSymmetry(Part_1_doc, [model.selection("COMPSOLID", "Symmetry_6_1_7")], model.selection("EDGE", "PartSet/OX"), keepOriginal = False, keepSubResults = True)
 124 assertResult(Symmetry_7, 7)
 125
 126 Symmetry_8 = model.addSymmetry(Part_1_doc, [model.selection("COMPOUND", "Symmetry_7_1_8")], model.selection("EDGE", "PartSet/OX"), keepOriginal = False, keepSubResults = True)
 127 assertResult(Symmetry_8, 8)
 128
 129 model.end()
 130
 131 assert(model.checkPythonDump())