Salome HOME
Merge branch 'master' into gni/evolution
[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.06"
23
24 import logging
25 import os
26
27 from blocFissure import gmu
28
29 import salome
30 from SketchAPI import *
31 from salome.shaper import model
32 import SHAPERSTUDY
33 import SMESH
34 from salome.smesh import smeshBuilder
35
36 #=============== Options ====================
37 # 1. NOM_OBJET = nom de l'objet
38 NOM_OBJET = "Tube"
39 # 2. CAO
40 # 2.1. Le tube : rayon intérieur, rayon extérieur, hauteur
41 RAYON_INT =  9.84
42 RAYON_EXT = 11.11
43 HAUTEUR = 50.
44 # 2.2. La fissure :
45 F_L_1 = 80.
46 F_THETA_1 = 3.
47 F_C_1 = 1.04
48 F_L_2 = 20.
49 F_C_3 = 0.99
50 # 3. Maillage
51 # Nombre de segments du tube
52 NB_S_E = 4
53 NB_S_H = 60
54 NB_S_P = 30
55 #============================================
56
57 salome.salome_init()
58
59 ###
60 ### SHAPER component
61 ###
62
63 model.begin()
64 partSet = model.moduleDocument()
65
66 ### Create Part
67 Part_1 = model.addPart(partSet)
68 Part_1_doc = Part_1.document()
69 model.addParameter(Part_1_doc, "R_I", "%f" % RAYON_INT)
70 model.addParameter(Part_1_doc, "R_E", "%f" % RAYON_EXT)
71 model.addParameter(Part_1_doc, "H", "%f" % HAUTEUR)
72 model.addParameter(Part_1_doc, "F_L_1", "{}".format(F_L_1))
73 model.addParameter(Part_1_doc, "F_THETA_1", "{}".format(F_THETA_1))
74 model.addParameter(Part_1_doc, "F_R_1", "{}*{}".format(F_C_1,RAYON_INT))
75 model.addParameter(Part_1_doc, "F_L_2", "{}".format(F_L_2))
76 model.addParameter(Part_1_doc, "F_D_3", "{}*{}".format(F_C_3,RAYON_INT))
77
78 ### Create Sketch
79 Sketch_1 = model.addSketch(Part_1_doc, model.defaultPlane("XOZ"))
80
81 ### Create SketchLine
82 SketchLine_1 = Sketch_1.addLine(11.11, -22.82842712474619, 9.84, -22.82842712474619)
83
84 ### Create SketchLine
85 SketchLine_2 = Sketch_1.addLine(9.84, -22.82842712474619, 9.84, 27.17157287525381)
86
87 ### Create SketchLine
88 SketchLine_3 = Sketch_1.addLine(9.84, 27.17157287525381, 11.11, 27.17157287525381)
89
90 ### Create SketchLine
91 SketchLine_4 = Sketch_1.addLine(11.11, 27.17157287525381, 11.11, -22.82842712474619)
92 Sketch_1.setCoincident(SketchLine_4.endPoint(), SketchLine_1.startPoint())
93 Sketch_1.setCoincident(SketchLine_1.endPoint(), SketchLine_2.startPoint())
94 Sketch_1.setCoincident(SketchLine_2.endPoint(), SketchLine_3.startPoint())
95 Sketch_1.setCoincident(SketchLine_3.endPoint(), SketchLine_4.startPoint())
96 Sketch_1.setHorizontal(SketchLine_1.result())
97 Sketch_1.setVertical(SketchLine_2.result())
98 Sketch_1.setHorizontal(SketchLine_3.result())
99 Sketch_1.setVertical(SketchLine_4.result())
100 Sketch_1.setLength(SketchLine_4.result(), "H")
101
102 ### Create SketchProjection
103 SketchProjection_1 = Sketch_1.addProjection(model.selection("VERTEX", "PartSet/Origin"), False)
104 SketchPoint_1 = SketchProjection_1.createdFeature()
105 Sketch_1.setHorizontalDistance(SketchAPI_Point(SketchPoint_1).coordinates(), SketchLine_1.endPoint(), "R_I")
106 Sketch_1.setHorizontalDistance(SketchAPI_Point(SketchPoint_1).coordinates(), SketchLine_1.startPoint(), "R_E")
107 model.do()
108
109 ### Create Revolution
110 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)
111 Revolution_1.result().setName("Tube")
112 Revolution_1.result().setTransparency(0.37)
113
114 ### Create Sketch
115 Sketch_2 = model.addSketch(Part_1_doc, model.standardPlane("YOZ"))
116
117 ### Create SketchArc
118 SketchArc_1 = Sketch_2.addArc(-80, 0, 10.10993798722564, 4.722461741385164, 10.10993798722563, -4.722461741385561, True)
119
120 ### Create SketchProjection
121 SketchProjection_2 = Sketch_2.addProjection(model.selection("EDGE", "PartSet/OY"), False)
122 SketchLine_5 = SketchProjection_2.createdFeature()
123 Sketch_2.setCoincident(SketchArc_1.center(), SketchLine_5.result())
124
125 ### Create SketchProjection
126 SketchProjection_3 = Sketch_2.addProjection(model.selection("VERTEX", "PartSet/Origin"), False)
127 SketchPoint_2 = SketchProjection_3.createdFeature()
128 Sketch_2.setHorizontalDistance(SketchArc_1.center(), SketchAPI_Point(SketchPoint_2).coordinates(), "F_L_1")
129
130 ### Create SketchPoint
131 SketchPoint_3 = Sketch_2.addPoint(10.2336, 0)
132 SketchPoint_3.setAuxiliary(True)
133 Sketch_2.setCoincident(SketchPoint_3.coordinates(), SketchArc_1.results()[1])
134 Sketch_2.setMiddlePoint(SketchPoint_3.coordinates(), SketchArc_1.results()[1])
135 Sketch_2.setCoincident(SketchPoint_3.coordinates(), SketchLine_5.result())
136
137 ### Create SketchProjection
138 SketchProjection_4 = Sketch_2.addProjection(model.selection("VERTEX", "PartSet/Origin"), False)
139 SketchPoint_4 = SketchProjection_4.createdFeature()
140 Sketch_2.setHorizontalDistance(SketchPoint_3.coordinates(), SketchAPI_Point(SketchPoint_4).coordinates(), "F_R_1")
141
142 ### Create SketchLine
143 SketchLine_6 = Sketch_2.addLine(-80, 0, 10.10993798722564, 4.722461741385164)
144 SketchLine_6.setAuxiliary(True)
145 Sketch_2.setCoincident(SketchArc_1.center(), SketchLine_6.startPoint())
146 Sketch_2.setCoincident(SketchArc_1.startPoint(), SketchLine_6.endPoint())
147
148 ### Create SketchConstraintAngle
149 Sketch_2.setAngle(SketchLine_5.result(), SketchLine_6.result(), "F_THETA_1", type = "Direct")
150
151 ### Create SketchEllipticArc
152 SketchEllipticArc_1 = Sketch_2.addEllipticArc(-20, 8.956370781951521e-27, -10.21629725685072, 9.783702743149284, 10.10993798723031, 4.722461741243296, 9.7416, 8.052073253504034, False)
153 [SketchPoint_5, SketchPoint_6, SketchPoint_7, SketchPoint_8, SketchPoint_9, SketchPoint_10, SketchPoint_11, SketchLine_7, SketchLine_8] = \
154   SketchEllipticArc_1.construction(center = "aux", firstFocus = "aux", secondFocus = "aux", majorAxisStart = "aux", majorAxisEnd = "aux", minorAxisStart = "aux", minorAxisEnd = "aux", majorAxis = "aux", minorAxis = "aux")
155 Sketch_2.setCoincident(SketchEllipticArc_1.startPoint(), SketchLine_6.endPoint())
156 Sketch_2.setCoincident(SketchAPI_Point(SketchPoint_5).coordinates(), SketchLine_5.result())
157 Sketch_2.setTangent(SketchEllipticArc_1.result(), SketchArc_1.results()[1])
158
159 ### Create SketchProjection
160 SketchProjection_5 = Sketch_2.addProjection(model.selection("VERTEX", "PartSet/Origin"), False)
161 SketchPoint_12 = SketchProjection_5.createdFeature()
162 Sketch_2.setHorizontalDistance(SketchAPI_Point(SketchPoint_5).coordinates(), SketchAPI_Point(SketchPoint_12).coordinates(), "F_L_2")
163
164 ### Create SketchConstraintAngle
165 Sketch_2.setAngle(SketchLine_5.result(), SketchLine_7.result(), 45, type = "Supplementary")
166
167 ### Create SketchProjection
168 SketchProjection_6 = Sketch_2.addProjection(model.selection("VERTEX", "PartSet/Origin"), False)
169 SketchPoint_13 = SketchProjection_6.createdFeature()
170 Sketch_2.setHorizontalDistance(SketchEllipticArc_1.endPoint(), SketchAPI_Point(SketchPoint_13).coordinates(), "F_D_3")
171
172 ### Create SketchConstraintMirror
173 SketchConstraintMirror_1 = Sketch_2.addMirror(SketchLine_5.result(), [SketchEllipticArc_1.result()])
174 [SketchEllipticArc_2] = SketchConstraintMirror_1.mirrored()
175
176 ### Create SketchLine
177 SketchLine_9 = Sketch_2.addLine(9.7416, 8.052073253504034, 9.7416, -8.052073253504034)
178 Sketch_2.setCoincident(SketchEllipticArc_1.endPoint(), SketchLine_9.startPoint())
179 Sketch_2.setCoincident(SketchAPI_EllipticArc(SketchEllipticArc_2).endPoint(), SketchLine_9.endPoint())
180 model.do()
181
182 ### Create Face
183 Face_1 = model.addFace(Part_1_doc, [model.selection("FACE", "Sketch_2/Face-SketchArc_1_2r-SketchEllipticArc_1f-SketchLine_9f-SketchEllipticArc_2f")])
184 Face_1.result().setName("Fissure")
185
186 ### Create Group
187 Group_1 = model.addGroup(Part_1_doc, "Edges", [model.selection("EDGE", "Fissure/Modified_Edge&Sketch_2/SketchEllipticArc_1")])
188
189 ### Create Group
190 Group_2 = model.addGroup(Part_1_doc, "Edges", [model.selection("EDGE", "Fissure/Modified_Edge&Sketch_2/SketchArc_1_2")])
191
192 ### Create Group
193 Group_3 = model.addGroup(Part_1_doc, "Edges", [model.selection("EDGE", "Fissure/Modified_Edge&Sketch_2/SketchEllipticArc_2")])
194
195 ### Create Group
196 Group_4 = model.addGroup(Part_1_doc, "Edges", [model.selection("EDGE", "Fissure/Modified_Edge&Sketch_2/SketchLine_9")])
197
198 ### Create Group
199 Group_5 = model.addGroup(Part_1_doc, "Edges", [model.selection("EDGE", "[Tube/Generated_Face&Sketch_1/SketchLine_1][Tube/From_Face]")])
200 Group_5.setName("Epaisseur")
201 Group_5.result().setName("Epaisseur")
202
203 ### Create Group
204 Group_6 = model.addGroup(Part_1_doc, "Edges", [model.selection("EDGE", "[Tube/Generated_Face&Sketch_1/SketchLine_4][Tube/From_Face]")])
205 Group_6.setName("Hauteur")
206 Group_6.result().setName("Hauteur")
207
208 ### Create Group
209 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]")])
210 Group_7.setName("Peripherie")
211 Group_7.result().setName("Peripherie")
212
213 ### Create Export
214 ficcao = os.path.join(gmu.pathBloc, "materielCasTests", "{}Fiss.xao".format(NOM_OBJET))
215 text = ".. Exportation de la géométrie de la fissure dans le fichier '{}'".format(ficcao)
216 logging.info(text)
217 _ = model.exportToXAO(Part_1_doc, ficcao, model.selection("FACE", "Fissure"), 'XAO')
218
219 model.end()
220
221 ###
222 ### SHAPERSTUDY component
223 ###
224
225 model.publishToShaperStudy()
226
227 l_aux = SHAPERSTUDY.shape(model.featureStringId(Revolution_1))
228 objet = l_aux[0]
229 l_groups = l_aux[1:]
230
231 ###
232 ### SMESH component
233 ###
234
235 smesh = smeshBuilder.New()
236 Maillage_tube = smesh.Mesh(objet)
237 smesh.SetName(Maillage_tube, NOM_OBJET)
238
239 for groupe in l_groups:
240   groupe_nom = groupe.GetName()
241   if ( groupe_nom[:1] == "E" ):
242     group_e = groupe
243   elif ( groupe_nom[:1] == "H" ):
244     group_h = groupe
245   elif ( groupe_nom[:1] == "P" ):
246     group_p = groupe
247   _ = Maillage_tube.GroupOnGeom(groupe,groupe_nom,SMESH.EDGE)
248
249 Regular_1D = Maillage_tube.Segment()
250 Nb_Segments_1 = Regular_1D.NumberOfSegments(NB_S_P)
251 Nb_Segments_1.SetDistrType( 0 )
252 Quadrangle_2D = Maillage_tube.Quadrangle(algo=smeshBuilder.QUADRANGLE)
253 Hexa_3D = Maillage_tube.Hexahedron(algo=smeshBuilder.Hexa)
254
255 Regular_1D_1 = Maillage_tube.Segment(geom=group_e)
256 Number_of_Segments_1 = Regular_1D_1.NumberOfSegments(NB_S_E)
257 Propagation_of_1D_Hyp = Regular_1D_1.Propagation()
258
259 Regular_1D_2 = Maillage_tube.Segment(geom=group_h)
260 Number_of_Segments_2 = Regular_1D_2.NumberOfSegments(NB_S_H)
261 Propagation_of_1D_Hyp_1 = Regular_1D_2.Propagation()
262
263 is_done = Maillage_tube.Compute()
264 text = "Maillage_tube.Compute"
265 if is_done:
266   logging.info(text+" OK")
267 else:
268   text = "Erreur au calcul du maillage.\n" + text
269   logging.info(text)
270   raise Exception(text)
271
272 ficmed = os.path.join(gmu.pathBloc, "materielCasTests","{}.med".format(NOM_OBJET))
273 text = ".. Archivage du maillage dans le fichier '{}'".format(ficmed)
274 logging.info(text)
275 Maillage_tube.ExportMED(ficmed)
276
277 ## set object names
278 smesh.SetName(Regular_1D.GetAlgorithm(), 'Regular_1D')
279 smesh.SetName(Nb_Segments_1, 'Nb. Segments_1')
280 smesh.SetName(Quadrangle_2D.GetAlgorithm(), 'Quadrangle_2D')
281 smesh.SetName(Hexa_3D.GetAlgorithm(), 'Hexa_3D')
282
283 if salome.sg.hasDesktop():
284   salome.sg.updateObjBrowser()