Salome HOME
9053652effa7d348d91aef9486c549ab3a38c479
[modules/shaper.git] / src / FeaturesPlugin / Test / TestTranslation_MultiLevelCompound_v95_2.py
1 # Copyright (C) 2020-2021  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 DX = 15
65 DY = 20
66 DZ = 25
67 TOLERANCE = 1.e-7
68 REFERENCE = []
69 for ind in range(0, Compound_2.result().numberOfSubs()):
70     p = Compound_2.result().subResult(ind).resultSubShapePair()[0].shape().middlePoint()
71     REFERENCE.append(p)
72
73 def assertResult(theFeature):
74     model.testNbResults(theFeature, 1)
75     model.testNbSubResults(theFeature, [3])
76     model.testNbSubShapes(theFeature, GeomAPI_Shape.SOLID, [10])
77     model.testNbSubShapes(theFeature, GeomAPI_Shape.FACE, [47])
78     model.testNbSubShapes(theFeature, GeomAPI_Shape.EDGE, [162])
79     model.testNbSubShapes(theFeature, GeomAPI_Shape.VERTEX, [324])
80     model.testResultsVolumes(theFeature, [14319.99602674256])
81
82     for ind in range(0, theFeature.result().numberOfSubs()):
83         ref = REFERENCE[ind]
84         midPoint = theFeature.result().subResult(ind).resultSubShapePair()[0].shape().middlePoint()
85         assert(midPoint.distance(ref) < TOLERANCE), "Sub-result {}; actual ({}, {}, {}) != expected ({}, {}, {})".format(ind, midPoint.x(), midPoint.y(), midPoint.z(), ref.x(), ref.y(), ref.z())
86
87
88 model.begin()
89 Translation_1 = model.addTranslation(Part_1_doc, [model.selection("SOLID", "Compound_2_1_1_5")], vector = [DX, 0, 0], keepSubResults = True)
90 model.end()
91 # selection of a compsolid part is prohibited
92 assert(Translation_1.feature().error() != "")
93
94 model.begin()
95 Translation_1.setMainObjects([model.selection("COMPSOLID", "Compound_2_1_1")])
96 REFERENCE[0].setX(REFERENCE[0].x() + DX)
97 assertResult(Translation_1)
98
99 Translation_2 = model.addTranslation(Part_1_doc, [model.selection("SOLID", "Translation_1_1_2_1")], vector = [0, DY, 0], keepSubResults = True)
100 REFERENCE[1].setY(REFERENCE[1].y() + DY / 2)
101 assertResult(Translation_2)
102
103 Translation_3 = model.addTranslation(Part_1_doc, [model.selection("SOLID", "Translation_2_1_3")], vector = [0, 0, DZ], keepSubResults = True)
104 REFERENCE[2].setZ(REFERENCE[2].z() + DZ)
105 assertResult(Translation_3)
106
107 Translation_4 = model.addTranslation(Part_1_doc, [model.selection("SOLID", "Translation_3_1_1_1"), model.selection("SOLID", "Translation_3_1_2_2"), model.selection("SOLID", "Translation_3_1_3")], vector = [DX, DY, DZ], keepSubResults = True)
108 model.end()
109 # selection of a compsolid part is prohibited
110 assert(Translation_4.feature().error() != "")
111
112 model.begin()
113 Translation_4.setMainObjects([model.selection("COMPSOLID", "Translation_3_1_1"), model.selection("SOLID", "Translation_3_1_2_2"), model.selection("SOLID", "Translation_3_1_3")])
114 REFERENCE = [GeomAPI_Pnt(REFERENCE[0].x() + DX, REFERENCE[0].y() + DY, REFERENCE[0].z() + DZ),
115              GeomAPI_Pnt(REFERENCE[1].x() + DX / 2, REFERENCE[1].y() + DY / 2, REFERENCE[1].z() + DZ / 2),
116              GeomAPI_Pnt(REFERENCE[2].x() + DX, REFERENCE[2].y() + DY, REFERENCE[2].z() + DZ)]
117 assertResult(Translation_4)
118
119 model.end()
120
121 assert(model.checkPythonDump())