Salome HOME
Insertion d'un cas test via xao
[modules/smesh.git] / src / Tools / blocFissure / materielCasTests / tube.py
1 # -*- coding: utf-8 -*-
2 # Copyright (C) 2014-2020  EDF R&D
3 #
4 # This library is free software; you can redistribute it and/or
5 # modify it under the terms of the GNU Lesser General Public
6 # License as published by the Free Software Foundation; either
7 # version 2.1 of the License, or (at your option) any later version.
8 #
9 # This library is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12 # Lesser General Public License for more details.
13 #
14 # You should have received a copy of the GNU Lesser General Public
15 # License along with this library; if not, write to the Free Software
16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
17 #
18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 #
20
21 """Cas-test de blocFissure sur un tube"""
22 __revision__ = "V02.04"
23
24 import logging
25
26 import sys
27 import salome
28
29 salome.salome_init()
30
31 import os
32 from blocFissure import gmu
33
34 #=============== Options ====================
35 # 1. NOM_OBJET = nom de l'objet
36 NOM_OBJET = "Tube"
37 # 2. CAO
38 # 2.1. Le tube : rayon intérieur, rayon extérieur, hauteur
39 RAYON_INT =  9.84
40 RAYON_EXT = 11.11
41 HAUTEUR = 50.
42 # 2.2. La fissure :
43 F_L_1 = 80.
44 F_THETA_1 = 3.
45 F_C_1 = 1.04
46 F_L_2 = 20.
47 # 3. Maillage
48 # Nombre de segments du tube
49 NB_S_E = 4
50 NB_S_H = 60
51 NB_S_P = 30
52 #============================================
53
54 ###
55 ### SHAPER component
56 ###
57
58 from SketchAPI import *
59
60 from salome.shaper import model
61
62 model.begin()
63 partSet = model.moduleDocument()
64
65 ### Create Part
66 Part_1 = model.addPart(partSet)
67 Part_1_doc = Part_1.document()
68 model.addParameter(Part_1_doc, "R_I", "%f" % RAYON_INT)
69 model.addParameter(Part_1_doc, "R_E", "%f" % RAYON_EXT)
70 model.addParameter(Part_1_doc, "H", "%f" % HAUTEUR)
71 model.addParameter(Part_1_doc, "F_L_1", "{}".format(F_L_1))
72 model.addParameter(Part_1_doc, "F_THETA_1", "{}".format(F_THETA_1))
73 model.addParameter(Part_1_doc, "F_R_1", "{}*{}".format(F_C_1,RAYON_INT))
74 model.addParameter(Part_1_doc, "F_L_2", "{}".format(F_L_2))
75
76 ### Create Sketch
77 Sketch_1 = model.addSketch(Part_1_doc, model.defaultPlane("XOZ"))
78
79 ### Create SketchLine
80 SketchLine_1 = Sketch_1.addLine(11.11, -22.82842712474619, 9.84, -22.82842712474619)
81
82 ### Create SketchLine
83 SketchLine_2 = Sketch_1.addLine(9.84, -22.82842712474619, 9.84, 27.17157287525381)
84
85 ### Create SketchLine
86 SketchLine_3 = Sketch_1.addLine(9.84, 27.17157287525381, 11.11, 27.17157287525381)
87
88 ### Create SketchLine
89 SketchLine_4 = Sketch_1.addLine(11.11, 27.17157287525381, 11.11, -22.82842712474619)
90 Sketch_1.setCoincident(SketchLine_4.endPoint(), SketchLine_1.startPoint())
91 Sketch_1.setCoincident(SketchLine_1.endPoint(), SketchLine_2.startPoint())
92 Sketch_1.setCoincident(SketchLine_2.endPoint(), SketchLine_3.startPoint())
93 Sketch_1.setCoincident(SketchLine_3.endPoint(), SketchLine_4.startPoint())
94 Sketch_1.setHorizontal(SketchLine_1.result())
95 Sketch_1.setVertical(SketchLine_2.result())
96 Sketch_1.setHorizontal(SketchLine_3.result())
97 Sketch_1.setVertical(SketchLine_4.result())
98 Sketch_1.setLength(SketchLine_4.result(), "H")
99
100 ### Create SketchProjection
101 SketchProjection_1 = Sketch_1.addProjection(model.selection("VERTEX", "PartSet/Origin"), False)
102 SketchPoint_1 = SketchProjection_1.createdFeature()
103 Sketch_1.setHorizontalDistance(SketchAPI_Point(SketchPoint_1).coordinates(), SketchLine_1.endPoint(), "R_I")
104 Sketch_1.setHorizontalDistance(SketchAPI_Point(SketchPoint_1).coordinates(), SketchLine_1.startPoint(), "R_E")
105 model.do()
106
107 ### Create Revolution
108 Revolution_1 = model.addRevolution(Part_1_doc, [model.selection("FACE", "Sketch_1/Face-SketchLine_4r-SketchLine_3r-SketchLine_2r-SketchLine_1r")], model.selection("EDGE", "PartSet/OZ"), 180, 0)
109 Revolution_1.result().setName("Tube")
110 Revolution_1.result().setTransparency(0.9399999999999999)
111
112 ### Create Sketch
113 Sketch_2 = model.addSketch(Part_1_doc, model.standardPlane("YOZ"))
114
115 ### Create SketchArc
116 SketchArc_1 = Sketch_2.addArc(-80, 0, 10.10993798722564, 4.722461741385164, 10.10993798722563, -4.722461741385561, True)
117
118 ### Create SketchProjection
119 SketchProjection_2 = Sketch_2.addProjection(model.selection("EDGE", "PartSet/OY"), False)
120 SketchLine_5 = SketchProjection_2.createdFeature()
121 Sketch_2.setCoincident(SketchArc_1.center(), SketchLine_5.result())
122
123 ### Create SketchProjection
124 SketchProjection_3 = Sketch_2.addProjection(model.selection("VERTEX", "PartSet/Origin"), False)
125 SketchPoint_2 = SketchProjection_3.createdFeature()
126 Sketch_2.setHorizontalDistance(SketchArc_1.center(), SketchAPI_Point(SketchPoint_2).coordinates(), "F_L_1")
127
128 ### Create SketchPoint
129 SketchPoint_3 = Sketch_2.addPoint(10.2336, 0)
130 SketchPoint_3.setAuxiliary(True)
131 Sketch_2.setCoincident(SketchPoint_3.coordinates(), SketchArc_1.results()[1])
132 Sketch_2.setMiddlePoint(SketchPoint_3.coordinates(), SketchArc_1.results()[1])
133 Sketch_2.setCoincident(SketchPoint_3.coordinates(), SketchLine_5.result())
134
135 ### Create SketchProjection
136 SketchProjection_4 = Sketch_2.addProjection(model.selection("VERTEX", "PartSet/Origin"), False)
137 SketchPoint_4 = SketchProjection_4.createdFeature()
138 Sketch_2.setHorizontalDistance(SketchPoint_3.coordinates(), SketchAPI_Point(SketchPoint_4).coordinates(), "F_R_1")
139
140 ### Create SketchLine
141 SketchLine_6 = Sketch_2.addLine(-80, 0, 10.10993798722564, 4.722461741385164)
142 SketchLine_6.setAuxiliary(True)
143 Sketch_2.setCoincident(SketchArc_1.center(), SketchLine_6.startPoint())
144 Sketch_2.setCoincident(SketchArc_1.startPoint(), SketchLine_6.endPoint())
145
146 ### Create SketchConstraintAngle
147 Sketch_2.setAngle(SketchLine_5.result(), SketchLine_6.result(), "F_THETA_1", type = "Direct")
148
149 ### Create SketchProjection
150 SketchProjection_5 = Sketch_2.addProjection(model.selection("VERTEX", "PartSet/Origin"), False)
151 SketchPoint_5 = SketchProjection_5.createdFeature()
152
153 ### Create SketchEllipticArc
154 SketchEllipticArc_1 = Sketch_2.addEllipticArc(-20, 8.956370781951521e-27, -10.21629725685072, 9.783702743149284, 10.10993798723031, 4.722461741243296, 9.440921421952831, 9.565935245237778, False)
155 [SketchPoint_6, SketchPoint_7, SketchPoint_8, SketchPoint_9, SketchPoint_10, SketchPoint_11, SketchPoint_12, SketchLine_7, SketchLine_8] = SketchEllipticArc_1.construction(center = "aux", firstFocus = "aux", secondFocus = "aux", majorAxisStart = "aux", majorAxisEnd = "aux", minorAxisStart = "aux", minorAxisEnd = "aux", majorAxis = "aux", minorAxis = "aux")
156 Sketch_2.setCoincident(SketchEllipticArc_1.startPoint(), SketchLine_6.endPoint())
157 Sketch_2.setCoincident(SketchAPI_Point(SketchPoint_6).coordinates(), SketchLine_5.result())
158 Sketch_2.setTangent(SketchEllipticArc_1.result(), SketchArc_1.results()[1])
159
160 ### Create SketchProjection
161 SketchProjection_6 = Sketch_2.addProjection(model.selection("VERTEX", "PartSet/Origin"), False)
162 SketchPoint_13 = SketchProjection_6.createdFeature()
163 Sketch_2.setHorizontalDistance(SketchAPI_Point(SketchPoint_6).coordinates(), SketchAPI_Point(SketchPoint_13).coordinates(), "F_L_2")
164
165 ### Create SketchLine
166 SketchLine_9 = Sketch_2.addLine(-20, 8.956370781951521e-27, 9.440921421952831, 9.565935245237778)
167 SketchLine_9.setAuxiliary(True)
168 Sketch_2.setCoincident(SketchEllipticArc_1.center(), SketchLine_9.startPoint())
169 Sketch_2.setCoincident(SketchEllipticArc_1.endPoint(), SketchLine_9.endPoint())
170
171 ### Create SketchConstraintAngle
172 Sketch_2.setAngle(SketchLine_9.result(), SketchLine_5.result(), "6.*F_THETA_1", type = "Direct")
173
174 ### Create SketchConstraintAngle
175 Sketch_2.setAngle(SketchLine_5.result(), SketchLine_7.result(), 45, type = "Supplementary")
176
177 ### Create SketchConstraintMirror
178 SketchConstraintMirror_1 = Sketch_2.addMirror(SketchLine_5.result(), [SketchEllipticArc_1.result()])
179 [SketchEllipticArc_2] = SketchConstraintMirror_1.mirrored()
180
181 ### Create SketchLine
182 SketchLine_10 = Sketch_2.addLine(9.440921421952831, 9.565935245237778, 9.440921421952837, -9.565935245237783)
183 Sketch_2.setCoincident(SketchEllipticArc_1.endPoint(), SketchLine_10.startPoint())
184 Sketch_2.setCoincident(SketchAPI_EllipticArc(SketchEllipticArc_2).endPoint(), SketchLine_10.endPoint())
185 model.do()
186
187 ### Create Face
188 Face_1 = model.addFace(Part_1_doc, [model.selection("FACE", "Sketch_2/Face-SketchArc_1_2r-SketchEllipticArc_1f-SketchLine_10f-SketchEllipticArc_2f")])
189 Face_1.result().setName("Fissure")
190
191 ### Create Group
192 Group_1 = model.addGroup(Part_1_doc, "Edges", [model.selection("EDGE", "Fissure/Modified_Edge&Sketch_2/SketchEllipticArc_1")])
193
194 ### Create Group
195 Group_2 = model.addGroup(Part_1_doc, "Edges", [model.selection("EDGE", "Fissure/Modified_Edge&Sketch_2/SketchArc_1_2")])
196
197 ### Create Group
198 Group_3 = model.addGroup(Part_1_doc, "Edges", [model.selection("EDGE", "Fissure/Modified_Edge&Sketch_2/SketchEllipticArc_2")])
199
200 ### Create Group
201 Group_4 = model.addGroup(Part_1_doc, "Edges", [model.selection("EDGE", "Fissure/Modified_Edge&Sketch_2/SketchLine_10")])
202
203 ### Create Group
204 Group_5 = model.addGroup(Part_1_doc, "Edges", [model.selection("EDGE", "[Tube/Generated_Face&Sketch_1/SketchLine_1][Tube/From_Face]")])
205 Group_5.setName("Epaisseur")
206 Group_5.result().setName("Epaisseur")
207
208 ### Create Group
209 Group_6 = model.addGroup(Part_1_doc, "Edges", [model.selection("EDGE", "[Tube/Generated_Face&Sketch_1/SketchLine_4][Tube/From_Face]")])
210 Group_6.setName("Hauteur")
211 Group_6.result().setName("Hauteur")
212
213 ### Create Group
214 Group_7 = model.addGroup(Part_1_doc, "Edges", [model.selection("EDGE", "[Tube/Generated_Face&Sketch_1/SketchLine_4][Tube/Generated_Face&Sketch_1/SketchLine_1]")])
215 Group_7.setName("Peripherie")
216 Group_7.result().setName("Peripherie")
217
218 ### Create Export
219 ficxao = os.path.join(gmu.pathBloc, "materielCasTests", "{}Fiss.xao".format(NOM_OBJET))
220 text = ".. Exportation de la géométrie de la fissure dans le fichier '{}'".format(ficxao)
221 logging.info(text)
222 _ = model.exportToXAO(Part_1_doc, ficxao, model.selection("FACE", "Fissure"), 'XAO')
223
224 model.end()
225
226 ###
227 ### SHAPERSTUDY component
228 ###
229
230 model.publishToShaperStudy()
231 import SHAPERSTUDY
232 l_aux = SHAPERSTUDY.shape(model.featureStringId(Revolution_1))
233 objet = l_aux[0]
234 l_groups = l_aux[1:]
235
236 ###
237 ### SMESH component
238 ###
239
240 import  SMESH, SALOMEDS
241 from salome.smesh import smeshBuilder
242
243 smesh = smeshBuilder.New()
244 from salome.StdMeshers import StdMeshersBuilder
245 Maillage_tube = smesh.Mesh(objet)
246 smesh.SetName(Maillage_tube, NOM_OBJET)
247
248 for groupe in l_groups:
249   groupe_nom = groupe.GetName()
250   if ( groupe_nom[:1] == "E" ):
251     group_e = groupe
252   elif ( groupe_nom[:1] == "H" ):
253     group_h = groupe
254   elif ( groupe_nom[:1] == "P" ):
255     group_p = groupe
256   _ = Maillage_tube.GroupOnGeom(groupe,groupe_nom,SMESH.EDGE)
257
258 Regular_1D = Maillage_tube.Segment()
259 Nb_Segments_1 = Regular_1D.NumberOfSegments(NB_S_P)
260 Nb_Segments_1.SetDistrType( 0 )
261 Quadrangle_2D = Maillage_tube.Quadrangle(algo=smeshBuilder.QUADRANGLE)
262 Hexa_3D = Maillage_tube.Hexahedron(algo=smeshBuilder.Hexa)
263
264 Regular_1D_1 = Maillage_tube.Segment(geom=group_e)
265 Number_of_Segments_1 = Regular_1D_1.NumberOfSegments(NB_S_E)
266 Propagation_of_1D_Hyp = Regular_1D_1.Propagation()
267
268 Regular_1D_2 = Maillage_tube.Segment(geom=group_h)
269 Number_of_Segments_2 = Regular_1D_2.NumberOfSegments(NB_S_H)
270 Propagation_of_1D_Hyp_1 = Regular_1D_2.Propagation()
271
272 is_done = Maillage_tube.Compute()
273 text = "Maillage_tube.Compute"
274 if is_done:
275   logging.info(text+" OK")
276 else:
277   text = "Erreur au calcul du maillage.\n" + text
278   logging.info(text)
279   raise Exception(text)
280
281 ficmed = os.path.join(gmu.pathBloc, "materielCasTests","{}.med".format(NOM_OBJET))
282 text = ".. Archivage du maillage dans le fichier '{}'".format(ficmed)
283 logging.info(text)
284 Maillage_tube.ExportMED(ficmed)
285
286 ## set object names
287 smesh.SetName(Maillage_tube.GetMesh(), 'Maillage_tube')
288 smesh.SetName(Regular_1D.GetAlgorithm(), 'Regular_1D')
289 smesh.SetName(Nb_Segments_1, 'Nb. Segments_1')
290 smesh.SetName(Quadrangle_2D.GetAlgorithm(), 'Quadrangle_2D')
291 smesh.SetName(Hexa_3D.GetAlgorithm(), 'Hexa_3D')
292
293 if salome.sg.hasDesktop():
294   salome.sg.updateObjBrowser()