src/FeaturesPlugin/Test/TestMultiTranslation_MultiLevelCompound_v95_2.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
  22 from GeomAPI import *
  23 from SketchAPI import *
  24
  25 model.begin()
  26 partSet = model.moduleDocument()
  27 Part_1 = model.addPart(partSet)
  28 Part_1_doc = Part_1.document()
  29 Sketch_1 = model.addSketch(Part_1_doc, model.standardPlane("XOY"))
  30 SketchEllipse_1 = Sketch_1.addEllipse(11.18033988749894, -50, 22.36067977499789, -50, 10)
  31 [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")
  32 SketchConstraintHorizontal_1 = Sketch_1.setHorizontal(SketchLine_1.result())
  33 SketchConstraintLength_1 = Sketch_1.setLength(SketchLine_1.result(), 30)
  34 SketchConstraintLength_2 = Sketch_1.setLength(SketchLine_2.result(), 20)
  35 SketchProjection_1 = Sketch_1.addProjection(model.selection("EDGE", "PartSet/OY"), False)
  36 SketchLine_3 = SketchProjection_1.createdFeature()
  37 SketchConstraintCoincidence_1 = Sketch_1.setCoincident(SketchAPI_Point(SketchPoint_3).coordinates(), SketchLine_3.result())
  38 SketchConstraintDistance_1 = Sketch_1.setDistance(SketchAPI_Line(SketchLine_3).startPoint(), SketchAPI_Point(SketchPoint_3).coordinates(), 50, True)
  39 SketchCircle_1 = Sketch_1.addCircle(41.18033988749897, -50, 5)
  40 SketchConstraintCoincidence_2 = Sketch_1.setCoincident(SketchCircle_1.center(), SketchLine_1.result())
  41 SketchConstraintDistance_2 = Sketch_1.setDistance(SketchEllipse_1.majorAxisPositive(), SketchCircle_1.center(), 15, True)
  42 SketchConstraintRadius_1 = Sketch_1.setRadius(SketchCircle_1.results()[1], 5)
  43 SketchMultiRotation_1 = Sketch_1.addRotation([SketchEllipse_1.result(), SketchCircle_1.results()[1]], SketchAPI_Point(SketchPoint_3).coordinates(), 360, 3, True)
  44 [SketchEllipse_2, SketchEllipse_3, SketchCircle_2, SketchCircle_3] = SketchMultiRotation_1.rotated()
  45 model.do()
  46 Sketch_1.changeFacesOrder([[SketchEllipse_1.result(), SketchEllipse_1.result(), SketchEllipse_2.result(), SketchEllipse_3.result()],
  47                            [SketchCircle_1.results()[1]],
  48                            [SketchEllipse_2.result(), SketchEllipse_2.result(), SketchEllipse_3.result(), SketchEllipse_1.result()],
  49                            [SketchEllipse_1.result(), SketchEllipse_3.result(), SketchEllipse_2.result()],
  50                            [SketchEllipse_3.result(), SketchEllipse_2.result(), SketchEllipse_1.result()],
  51                            [SketchEllipse_1.result(), SketchEllipse_2.result(), SketchEllipse_3.result()],
  52                            [SketchEllipse_1.result(), SketchEllipse_3.result(), SketchEllipse_2.result()],
  53                            [SketchEllipse_2.result(), SketchEllipse_3.result(), SketchEllipse_3.result(), SketchEllipse_1.result()],
  54                            [SketchCircle_2.results()[1]],
  55                            [SketchCircle_3.results()[1]]
  56                           ])
  57 model.do()
  58 Extrusion_1 = model.addExtrusion(Part_1_doc, [model.selection("COMPOUND", "Sketch_1")], model.selection(), 10, 0)
  59 Compound_1 = model.addCompound(Part_1_doc, [model.selection("SOLID", "Extrusion_1_2"), model.selection("SOLID", "Extrusion_1_3")])
  60 Compound_2_objects = [model.selection("COMPSOLID", "Extrusion_1_1"), model.selection("COMPOUND", "Compound_1_1"), model.selection("SOLID", "Extrusion_1_4")]
  61 Compound_2 = model.addCompound(Part_1_doc, Compound_2_objects)
  62 model.end()
  63
  64 TOLERANCE = 1.e-7
  65 DISTANCE_1 = 100
  66 COPIES_1 = 2
  67 DISTANCE_2 = 50
  68 COPIES_2 = 3
  69 COPIES = COPIES_1 * COPIES_2
  70
  71
  72 def subsRefPoints(theResult):
  73     res = []
  74     for ind in range(0, theResult.numberOfSubs()):
  75         res.append( theResult.subResult(ind).resultSubShapePair()[0].shape().middlePoint())
  76     return res
  77
  78 REFERENCE = subsRefPoints(Compound_2.result())
  79
  80 def assertResult(theFeature, theNbSolids, theNbFaces, theNbEdges, theNbVertices, theVolume):
  81     model.testNbResults(theFeature, 1)
  82     model.testNbSubResults(theFeature, [3])
  83     model.testNbSubShapes(theFeature, GeomAPI_Shape.SOLID, [theNbSolids])
  84     model.testNbSubShapes(theFeature, GeomAPI_Shape.FACE, [theNbFaces])
  85     model.testNbSubShapes(theFeature, GeomAPI_Shape.EDGE, [theNbEdges])
  86     model.testNbSubShapes(theFeature, GeomAPI_Shape.VERTEX, [theNbVertices])
  87     model.testResultsVolumes(theFeature, [theVolume])
  88
  89     for ind in range(0, theFeature.result().numberOfSubs()):
  90         ref = REFERENCE[ind]
  91         midPoint = theFeature.result().subResult(ind).resultSubShapePair()[0].shape().middlePoint()
  92         assert(midPoint.distance(ref) < TOLERANCE), "Sub-result {}; actual ({}, {}, {}) != expected ({}, {}, {})".format(ind, midPoint.x(), midPoint.y(), midPoint.z(), ref.x(), ref.y(), ref.z())
  93
  94
  95 model.begin()
  96 MultiTranslation_1 = model.addMultiTranslation(Part_1_doc, [model.selection("SOLID", "Compound_2_1_1_5")], model.selection("EDGE", "PartSet/OY"), DISTANCE_1, COPIES_1, model.selection("EDGE", "PartSet/OZ"), DISTANCE_2, COPIES_2, keepSubResults = True)
  97 model.end()
  98 # selection of a compsolid part is prohibited
  99 assert(MultiTranslation_1.feature().error() != "")
 100
 101 model.begin()
 102 MultiTranslation_1.setMainObjects([model.selection("COMPSOLID", "Compound_2_1_1")])
 103 REFERENCE[0].setY(REFERENCE[0].y() + DISTANCE_1 / 2 * (COPIES_1 - 1))
 104 REFERENCE[0].setZ(REFERENCE[0].z() + DISTANCE_2 / 2 * (COPIES_2 - 1))
 105 assertResult(MultiTranslation_1, 45, 237, 882, 1764, 74139.0037)
 106
 107 MultiTranslation_2 = model.addMultiTranslation(Part_1_doc, [model.selection("SOLID", "LinearCopy_1_1_2_1")], model.selection("EDGE", "PartSet/OY"), DISTANCE_1, COPIES_1, model.selection("EDGE", "PartSet/OZ"), DISTANCE_2, COPIES_2, keepSubResults = True)
 108 ref1Ycopy = REFERENCE[1].y()
 109 REFERENCE[1].setY(0)
 110 REFERENCE[1].setZ(REFERENCE[1].z() + DISTANCE_2 / 2 * (COPIES_2 - 1))
 111 assertResult(MultiTranslation_2, 50, 252, 912, 1824, 78065.9945)
 112
 113 MultiTranslation_3 = model.addMultiTranslation(Part_1_doc, [model.selection("SOLID", "LinearCopy_2_1_3")], model.selection("EDGE", "PartSet/OY"), DISTANCE_1, COPIES_1, model.selection("EDGE", "PartSet/OZ"), DISTANCE_2, COPIES_2, keepSubResults = True)
 114 REFERENCE[2].setY(REFERENCE[2].y() + DISTANCE_1 / 2 * (COPIES_1 - 1))
 115 REFERENCE[2].setZ(REFERENCE[2].z() + DISTANCE_2 / 2 * (COPIES_2 - 1))
 116 assertResult(MultiTranslation_3, 55, 267, 942, 1884, 81992.9853)
 117
 118 MultiTranslation_4 = model.addMultiTranslation(Part_1_doc, [model.selection("COMPSOLID", "LinearCopy_3_1_1_5"), model.selection("SOLID", "LinearCopy_3_1_2_2"), model.selection("COMPOUND", "LinearCopy_3_1_3")], model.selection("EDGE", "PartSet/OX"), -DISTANCE_2, COPIES_2, model.selection("EDGE", "PartSet/OY"), -DISTANCE_1, COPIES_1, keepSubResults = True)
 119 REFERENCE[0].setX(REFERENCE[0].x() - DISTANCE_2 / 2 * (COPIES_2 - 1))
 120 REFERENCE[0].setY(REFERENCE[0].y() - DISTANCE_1 / 2 * (COPIES_1 - 1))
 121 REFERENCE[1].setX(REFERENCE[1].x() - DISTANCE_2 / 2 * (COPIES_2 - 1))
 122 REFERENCE[1].setY(ref1Ycopy)
 123 REFERENCE[2].setX(REFERENCE[2].x() - DISTANCE_2 / 2 * (COPIES_2 - 1))
 124 REFERENCE[2].setY(REFERENCE[2].y() - DISTANCE_1 / 2 * (COPIES_1 - 1))
 125 assertResult(MultiTranslation_4, 125, 562, 1872, 3744, 169300.9287)
 126
 127 model.end()
 128
 129 assert(model.checkPythonDump())