src/FeaturesPlugin/Test/TestMultiTranslation_MultiLevelCompound_v0_2.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
  22 from GeomAPI import *
  23 from GeomAlgoAPI import *
  24 from SketchAPI import *
  25
  26 model.begin()
  27 partSet = model.moduleDocument()
  28 Part_1 = model.addPart(partSet)
  29 Part_1_doc = Part_1.document()
  30 Sketch_1 = model.addSketch(Part_1_doc, model.standardPlane("XOY"))
  31 SketchEllipse_1 = Sketch_1.addEllipse(11.18033988749894, -50, 22.36067977499789, -50, 10)
  32 [SketchPoint_1, SketchPoint_2, SketchPoint_3, SketchPoint_4, SketchPoint_5, SketchPoint_6, SketchPoint_7, SketchLine_1, SketchLine_2] = SketchEllipse_1.construction(center = "aux", firstFocus = "aux", secondFocus = "aux", majorAxisStart = "aux", majorAxisEnd = "aux", minorAxisStart = "aux", minorAxisEnd = "aux", majorAxis = "aux", minorAxis = "aux")
  33 SketchConstraintHorizontal_1 = Sketch_1.setHorizontal(SketchLine_1.result())
  34 SketchConstraintLength_1 = Sketch_1.setLength(SketchLine_1.result(), 30)
  35 SketchConstraintLength_2 = Sketch_1.setLength(SketchLine_2.result(), 20)
  36 SketchProjection_1 = Sketch_1.addProjection(model.selection("EDGE", "PartSet/OY"), False)
  37 SketchLine_3 = SketchProjection_1.createdFeature()
  38 SketchConstraintCoincidence_1 = Sketch_1.setCoincident(SketchAPI_Point(SketchPoint_3).coordinates(), SketchLine_3.result())
  39 SketchConstraintDistance_1 = Sketch_1.setDistance(SketchAPI_Line(SketchLine_3).startPoint(), SketchAPI_Point(SketchPoint_3).coordinates(), 50, True)
  40 SketchCircle_1 = Sketch_1.addCircle(41.18033988749897, -50, 5)
  41 SketchConstraintCoincidence_2 = Sketch_1.setCoincident(SketchCircle_1.center(), SketchLine_1.result())
  42 SketchConstraintDistance_2 = Sketch_1.setDistance(SketchEllipse_1.majorAxisPositive(), SketchCircle_1.center(), 15, True)
  43 SketchConstraintRadius_1 = Sketch_1.setRadius(SketchCircle_1.results()[1], 5)
  44 SketchMultiRotation_1 = Sketch_1.addRotation([SketchEllipse_1.result(), SketchCircle_1.results()[1]], SketchAPI_Point(SketchPoint_3).coordinates(), 360, 3, True)
  45 [SketchEllipse_2, SketchEllipse_3, SketchCircle_2, SketchCircle_3] = SketchMultiRotation_1.rotated()
  46 model.do()
  47 Sketch_1.changeFacesOrder([[SketchEllipse_1.result(), SketchEllipse_1.result(), SketchEllipse_2.result(), SketchEllipse_3.result()],
  48                            [SketchCircle_1.results()[1]],
  49                            [SketchEllipse_2.result(), SketchEllipse_2.result(), SketchEllipse_3.result(), SketchEllipse_1.result()],
  50                            [SketchEllipse_1.result(), SketchEllipse_3.result(), SketchEllipse_2.result()],
  51                            [SketchEllipse_3.result(), SketchEllipse_2.result(), SketchEllipse_1.result()],
  52                            [SketchEllipse_1.result(), SketchEllipse_2.result(), SketchEllipse_3.result()],
  53                            [SketchEllipse_1.result(), SketchEllipse_3.result(), SketchEllipse_2.result()],
  54                            [SketchEllipse_2.result(), SketchEllipse_3.result(), SketchEllipse_3.result(), SketchEllipse_1.result()],
  55                            [SketchCircle_2.results()[1]],
  56                            [SketchCircle_3.results()[1]]
  57                           ])
  58 model.do()
  59 Extrusion_1 = model.addExtrusion(Part_1_doc, [model.selection("COMPOUND", "Sketch_1")], model.selection(), 10, 0)
  60 Compound_1 = model.addCompound(Part_1_doc, [model.selection("SOLID", "Extrusion_1_2"), model.selection("SOLID", "Extrusion_1_3")])
  61 Compound_2_objects = [model.selection("COMPSOLID", "Extrusion_1_1"), model.selection("COMPOUND", "Compound_1_1"), model.selection("SOLID", "Extrusion_1_4")]
  62 Compound_2 = model.addCompound(Part_1_doc, Compound_2_objects)
  63 model.end()
  64
  65 TOLERANCE = 1.e-7
  66 DISTANCE_1 = 20
  67 COPIES_1 = 3
  68 DISTANCE_2 = 100
  69 COPIES_2 = 2
  70 COPIES = COPIES_1 * COPIES_2
  71
  72 model.begin()
  73 MultiTranslation_1 = model.addMultiTranslation(Part_1_doc, [model.selection("SOLID", "Compound_2_1_1_5")], model.selection("EDGE", "PartSet/OZ"), DISTANCE_1, COPIES_1, model.selection("EDGE", "PartSet/OY"), DISTANCE_2, COPIES_2)
  74 model.testNbResults(MultiTranslation_1, 1)
  75 model.testNbSubResults(MultiTranslation_1, [COPIES])
  76 model.testNbSubShapes(MultiTranslation_1, GeomAPI_Shape.SOLID, [1 * COPIES])
  77 model.testNbSubShapes(MultiTranslation_1, GeomAPI_Shape.FACE, [5 * COPIES])
  78 model.testNbSubShapes(MultiTranslation_1, GeomAPI_Shape.EDGE, [18 * COPIES])
  79 model.testNbSubShapes(MultiTranslation_1, GeomAPI_Shape.VERTEX, [36 * COPIES])
  80 model.testResultsVolumes(MultiTranslation_1, [542.746463956 * COPIES])
  81
  82 comp = GeomAlgoAPI_CompoundBuilder.compound([Extrusion_1.results()[0].subResult(4).resultSubShapePair()[0].shape()])
  83 refPoint = comp.middlePoint()
  84 refPoint.setY(refPoint.y() + DISTANCE_2 / 2 * (COPIES_2 - 1))
  85 refPoint.setZ(refPoint.z() + DISTANCE_1 / 2 * (COPIES_1 - 1))
  86 midPoint = MultiTranslation_1.defaultResult().shape().middlePoint()
  87 assert(midPoint.distance(refPoint) < TOLERANCE)
  88
  89
  90 Recover_1 = model.addRecover(Part_1_doc, MultiTranslation_1, [Compound_2.result()], True)
  91 MultiTranslation_2 = model.addMultiTranslation(Part_1_doc, [model.selection("SOLID", "Recover_1_1_2_1")], model.selection("EDGE", "PartSet/OZ"), DISTANCE_1, COPIES_1, model.selection("EDGE", "PartSet/OY"), DISTANCE_2, COPIES_2)
  92 model.testNbResults(MultiTranslation_2, 1)
  93 model.testNbSubResults(MultiTranslation_2, [COPIES])
  94 model.testNbSubShapes(MultiTranslation_2, GeomAPI_Shape.SOLID, [1 * COPIES])
  95 model.testNbSubShapes(MultiTranslation_2, GeomAPI_Shape.FACE, [3 * COPIES])
  96 model.testNbSubShapes(MultiTranslation_2, GeomAPI_Shape.EDGE, [6 * COPIES])
  97 model.testNbSubShapes(MultiTranslation_2, GeomAPI_Shape.VERTEX, [12 * COPIES])
  98 model.testResultsVolumes(MultiTranslation_2, [785.39816339745 * COPIES])
  99
 100 refPoint = Recover_1.result().subResult(1).subResult(0).resultSubShapePair()[0].shape().middlePoint()
 101 refPoint.setY(refPoint.y() + DISTANCE_2 / 2 * (COPIES_2 - 1))
 102 refPoint.setZ(refPoint.z() + DISTANCE_1 / 2 * (COPIES_1 - 1))
 103 midPoint = MultiTranslation_2.defaultResult().shape().middlePoint()
 104 assert(midPoint.distance(refPoint) < TOLERANCE)
 105
 106
 107 Recover_2 = model.addRecover(Part_1_doc, MultiTranslation_2, [Recover_1.result()], True)
 108 MultiTranslation_3 = model.addMultiTranslation(Part_1_doc, [model.selection("SOLID", "Recover_2_1_3")], model.selection("EDGE", "PartSet/OZ"), DISTANCE_1, COPIES_1, model.selection("EDGE", "PartSet/OY"), DISTANCE_2, COPIES_2)
 109 model.testNbResults(MultiTranslation_3, 1)
 110 model.testNbSubResults(MultiTranslation_3, [COPIES])
 111 model.testNbSubShapes(MultiTranslation_3, GeomAPI_Shape.SOLID, [1 * COPIES])
 112 model.testNbSubShapes(MultiTranslation_3, GeomAPI_Shape.FACE, [3 * COPIES])
 113 model.testNbSubShapes(MultiTranslation_3, GeomAPI_Shape.EDGE, [6 * COPIES])
 114 model.testNbSubShapes(MultiTranslation_3, GeomAPI_Shape.VERTEX, [12 * COPIES])
 115 model.testResultsVolumes(MultiTranslation_3, [785.39816339745 * COPIES])
 116
 117 refPoint = Recover_2.result().subResult(2).resultSubShapePair()[0].shape().middlePoint()
 118 refPoint.setY(refPoint.y() + DISTANCE_2 / 2 * (COPIES_2 - 1))
 119 refPoint.setZ(refPoint.z() + DISTANCE_1 / 2 * (COPIES_1 - 1))
 120 midPoint = MultiTranslation_3.defaultResult().shape().middlePoint()
 121 assert(midPoint.distance(refPoint) < TOLERANCE)
 122
 123
 124 Recover_3 = model.addRecover(Part_1_doc, MultiTranslation_3, [Recover_2.result()], True)
 125 MultiTranslation_4 = model.addMultiTranslation(Part_1_doc, [model.selection("SOLID", "Recover_3_1_1_1"), model.selection("SOLID", "Recover_3_1_2_2"), model.selection("SOLID", "Recover_3_1_3")], model.selection("EDGE", "PartSet/OZ"), DISTANCE_1, COPIES_1, model.selection("EDGE", "PartSet/OY"), DISTANCE_2, COPIES_2)
 126 model.testNbResults(MultiTranslation_4, 3)
 127 model.testNbSubResults(MultiTranslation_4, [COPIES, COPIES, COPIES])
 128 model.testNbSubShapes(MultiTranslation_4, GeomAPI_Shape.SOLID, [1 * COPIES, 1 * COPIES, 1 * COPIES])
 129 model.testNbSubShapes(MultiTranslation_4, GeomAPI_Shape.FACE, [6 * COPIES, 3 * COPIES, 3 * COPIES])
 130 model.testNbSubShapes(MultiTranslation_4, GeomAPI_Shape.EDGE, [24 * COPIES, 6 * COPIES, 6 * COPIES])
 131 model.testNbSubShapes(MultiTranslation_4, GeomAPI_Shape.VERTEX, [48 * COPIES, 12 * COPIES, 12 * COPIES])
 132 model.testResultsVolumes(MultiTranslation_4, [3444.394198615 * COPIES, 785.39816339745 * COPIES, 785.39816339745 * COPIES])
 133
 134 comp = GeomAlgoAPI_CompoundBuilder.compound([Recover_3.result().subResult(0).subResult(0).resultSubShapePair()[0].shape()])
 135 REFERENCE = [comp.middlePoint(),
 136              Recover_3.result().subResult(1).subResult(1).resultSubShapePair()[0].shape().middlePoint(),
 137              Recover_3.result().subResult(2).resultSubShapePair()[0].shape().middlePoint()]
 138 for res, ref in zip(MultiTranslation_4.results(), REFERENCE):
 139     ref.setY(ref.y() + DISTANCE_2 / 2 * (COPIES_2 - 1))
 140     ref.setZ(ref.z() + DISTANCE_1 / 2 * (COPIES_1 - 1))
 141     midPoint = res.resultSubShapePair()[0].shape().middlePoint()
 142     assert(midPoint.distance(ref) < TOLERANCE)
 143
 144 model.end()
 145
 146 assert(model.checkPythonDump())