1 # Copyright (C) 2020 CEA/DEN, EDF R&D
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.
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.
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
17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 from salome.shaper import model
23 from SketchAPI import *
28 partSet = model.moduleDocument()
29 Part_1 = model.addPart(partSet)
30 Part_1_doc = Part_1.document()
31 Sketch_1 = model.addSketch(Part_1_doc, model.standardPlane("XOY"))
32 SketchEllipse_1 = Sketch_1.addEllipse(11.18033988749894, -50, 22.36067977499789, -50, 10)
33 [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")
34 SketchConstraintHorizontal_1 = Sketch_1.setHorizontal(SketchLine_1.result())
35 SketchConstraintLength_1 = Sketch_1.setLength(SketchLine_1.result(), 30)
36 SketchConstraintLength_2 = Sketch_1.setLength(SketchLine_2.result(), 20)
37 SketchProjection_1 = Sketch_1.addProjection(model.selection("EDGE", "PartSet/OY"), False)
38 SketchLine_3 = SketchProjection_1.createdFeature()
39 SketchConstraintCoincidence_1 = Sketch_1.setCoincident(SketchAPI_Point(SketchPoint_3).coordinates(), SketchLine_3.result())
40 SketchConstraintDistance_1 = Sketch_1.setDistance(SketchAPI_Line(SketchLine_3).startPoint(), SketchAPI_Point(SketchPoint_3).coordinates(), 50, True)
41 SketchCircle_1 = Sketch_1.addCircle(41.18033988749897, -50, 5)
42 SketchConstraintCoincidence_2 = Sketch_1.setCoincident(SketchCircle_1.center(), SketchLine_1.result())
43 SketchConstraintDistance_2 = Sketch_1.setDistance(SketchEllipse_1.majorAxisPositive(), SketchCircle_1.center(), 15, True)
44 SketchConstraintRadius_1 = Sketch_1.setRadius(SketchCircle_1.results()[1], 5)
45 SketchMultiRotation_1 = Sketch_1.addRotation([SketchEllipse_1.result(), SketchCircle_1.results()[1]], SketchAPI_Point(SketchPoint_3).coordinates(), 360, 3, True)
46 [SketchEllipse_2, SketchEllipse_3, SketchCircle_2, SketchCircle_3] = SketchMultiRotation_1.rotated()
48 Sketch_1.changeFacesOrder([[SketchEllipse_1.result(), SketchEllipse_1.result(), SketchEllipse_2.result(), SketchEllipse_3.result()],
49 [SketchCircle_1.results()[1]],
50 [SketchEllipse_2.result(), SketchEllipse_2.result(), SketchEllipse_3.result(), SketchEllipse_1.result()],
51 [SketchEllipse_1.result(), SketchEllipse_3.result(), SketchEllipse_2.result()],
52 [SketchEllipse_3.result(), SketchEllipse_2.result(), SketchEllipse_1.result()],
53 [SketchEllipse_1.result(), SketchEllipse_2.result(), SketchEllipse_3.result()],
54 [SketchEllipse_1.result(), SketchEllipse_3.result(), SketchEllipse_2.result()],
55 [SketchEllipse_2.result(), SketchEllipse_3.result(), SketchEllipse_3.result(), SketchEllipse_1.result()],
56 [SketchCircle_2.results()[1]],
57 [SketchCircle_3.results()[1]]
60 Extrusion_1 = model.addExtrusion(Part_1_doc, [model.selection("COMPOUND", "Sketch_1")], model.selection(), 10, 0)
61 Compound_1 = model.addCompound(Part_1_doc, [model.selection("SOLID", "Extrusion_1_2"), model.selection("SOLID", "Extrusion_1_3")])
62 Compound_2_objects = [model.selection("COMPSOLID", "Extrusion_1_1"), model.selection("COMPOUND", "Compound_1_1"), model.selection("SOLID", "Extrusion_1_4")]
63 Compound_2 = model.addCompound(Part_1_doc, Compound_2_objects)
66 def symmetryOX(thePoint):
67 return GeomAPI_Pnt(thePoint.x(), -thePoint.y(), -thePoint.z())
69 def middleOZ(thePoint):
70 return GeomAPI_Pnt(0, 0, thePoint.z())
72 def average(thePoints):
75 x += p.x(); y += p.y(); z+= p.z()
79 return GeomAPI_Pnt(x, y, z)
81 def subsRefPoints(theResult):
83 for ind in range(0, theResult.numberOfSubs()):
84 res.append( theResult.subResult(ind).resultSubShapePair()[0].shape().middlePoint())
89 ORIGIN = GeomAPI_Pnt(0, 0, 0)
90 REFERENCE = subsRefPoints(Compound_2.result())
93 def assertResult(theFeature, theNbSolids, theNbFaces, theNbEdges, theNbVertices, theVolume):
94 model.testNbResults(theFeature, 1)
95 model.testNbSubResults(theFeature, [3])
96 model.testNbSubShapes(theFeature, GeomAPI_Shape.SOLID, [theNbSolids])
97 model.testNbSubShapes(theFeature, GeomAPI_Shape.FACE, [theNbFaces])
98 model.testNbSubShapes(theFeature, GeomAPI_Shape.EDGE, [theNbEdges])
99 model.testNbSubShapes(theFeature, GeomAPI_Shape.VERTEX, [theNbVertices])
100 model.testResultsVolumes(theFeature, [theVolume])
102 for ind in range(0, theFeature.result().numberOfSubs()):
104 midPoint = theFeature.result().subResult(ind).resultSubShapePair()[0].shape().middlePoint()
105 assert(midPoint.distance(ref) < TOLERANCE), "Sub-result {}; actual ({}, {}, {}) != expected ({}, {}, {})".format(ind, midPoint.x(), midPoint.y(), midPoint.z(), ref.x(), ref.y(), ref.z())
109 Symmetry_1 = model.addSymmetry(Part_1_doc, [model.selection("SOLID", "Compound_2_1_1_5")], model.selection("EDGE", "PartSet/OX"), keepOriginal = False, keepSubResults = True)
111 # selection of a compsolid part is prohibited
112 assert(Symmetry_1.feature().error() != "")
115 Symmetry_1.setMainObjects([model.selection("COMPSOLID", "Compound_2_1_1")])
116 REFERENCE[0] = symmetryOX(REFERENCE[0])
117 assertResult(Symmetry_1, 10, 47, 162, 324, 14319.99602674256)
119 Symmetry_2 = model.addSymmetry(Part_1_doc, [model.selection("SOLID", "Symmetry_1_1_2_1")], model.selection("EDGE", "PartSet/OX"), keepOriginal = False, keepSubResults = True)
120 subs = subsRefPoints(Compound_2.result().subResult(1))
121 subs[0] = symmetryOX(subs[0])
122 REFERENCE[1] = average(subs)
123 assertResult(Symmetry_2, 10, 47, 162, 324, 14319.99602674256)
125 Symmetry_3 = model.addSymmetry(Part_1_doc, [model.selection("SOLID", "Symmetry_2_1_3")], model.selection("EDGE", "PartSet/OX"), keepOriginal = False, keepSubResults = True)
126 REFERENCE[2] = symmetryOX(REFERENCE[2])
127 assertResult(Symmetry_3, 10, 47, 162, 324, 14319.99602674256)
129 Symmetry_4 = model.addSymmetry(Part_1_doc, [model.selection("SOLID", "Symmetry_3_1_1_1"), model.selection("SOLID", "Symmetry_3_1_2_2"), model.selection("SOLID", "Symmetry_3_1_3")], model.selection("EDGE", "PartSet/OX"), keepOriginal = False, keepSubResults = True)
131 # selection of a compsolid part is prohibited
132 assert(Symmetry_4.feature().error() != "")
135 Symmetry_4.setMainObjects([model.selection("COMPSOLID", "Symmetry_3_1_1"), model.selection("SOLID", "Symmetry_3_1_2_2"), model.selection("SOLID", "Symmetry_3_1_3")])
136 REFERENCE[0] = symmetryOX(REFERENCE[0])
137 subs = subsRefPoints(Compound_2.result().subResult(1))
138 for ind in range(0, len(subs)):
139 subs[ind] = symmetryOX(subs[ind])
140 REFERENCE[1] = average(subs)
141 REFERENCE[2] = symmetryOX(REFERENCE[2])
142 assertResult(Symmetry_4, 10, 47, 162, 324, 14319.99602674256)
145 Symmetry_5 = model.addSymmetry(Part_1_doc, [model.selection("COMPSOLID", "Symmetry_4_1_1")], model.selection("EDGE", "PartSet/OZ"), keepOriginal = True, keepSubResults = True)
146 REFERENCE[0] = middleOZ(REFERENCE[0])
147 assertResult(Symmetry_5, 17, 85, 306, 612, 26283.79756329278)
149 Symmetry_6 = model.addSymmetry(Part_1_doc, [model.selection("SOLID", "Symmetry_5_1_2_1")], model.selection("EDGE", "PartSet/OZ"), keepOriginal = True, keepSubResults = True)
150 REFERENCE[1] = middleOZ(REFERENCE[1])
151 assertResult(Symmetry_6, 18, 88, 312, 624, 27069.19572669)
153 Symmetry_7 = model.addSymmetry(Part_1_doc, [model.selection("SOLID", "Symmetry_6_1_3")], model.selection("EDGE", "PartSet/OZ"), keepOriginal = True, keepSubResults = True)
154 REFERENCE[2] = middleOZ(REFERENCE[2])
155 assertResult(Symmetry_7, 19, 91, 318, 636, 27854.59389)
157 Symmetry_8 = model.addSymmetry(Part_1_doc, [model.selection("COMPSOLID", "Symmetry_7_1_1"), model.selection("SOLID", "Symmetry_7_1_2_2"), model.selection("SOLID", "Symmetry_7_1_3")], model.selection("EDGE", "PartSet/OY"), keepOriginal = True, keepSubResults = True)
158 REFERENCE[0] = ORIGIN
159 REFERENCE[1] = ORIGIN
160 REFERENCE[2] = ORIGIN
161 assertResult(Symmetry_8, 36, 176, 624, 1248, 54138.39145338)
165 assert(model.checkPythonDump())