Salome HOME
Merge branch 'master' into gni/evolution
[modules/smesh.git] / src / Tools / blocFissure / materielCasTests / eprouvetteDroite.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 import logging
22
23 import sys
24 import salome
25
26 salome.salome_init()
27
28 import salome_notebook
29
30 import os
31 from blocFissure import gmu
32
33 ###
34 ### GEOM component
35 ###
36
37 import GEOM
38 from salome.geom import geomBuilder
39 import math
40 import SALOMEDS
41
42
43 geompy = geomBuilder.New()
44
45 O = geompy.MakeVertex(0, 0, 0)
46 OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
47 OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
48 OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
49 eprouvetteDroite = geompy.MakeBoxDXDYDZ(100, 200, 500)
50 [Compound_z, Compound_y, Compound_x] = geompy.Propagate(eprouvetteDroite)
51
52 Vertex_1 = geompy.MakeVertex(-10, -10, 200)
53 geomObj_1 = geompy.MakeMarker(0, 0, 0, 1, 0, 0, 0, 1, 0)
54 Plane_1 = geompy.MakePlane(Vertex_1, OZ, 2000)
55 geomObj_2 = geompy.MakeMarker(0, 0, 0, 1, 0, 0, 0, 1, 0)
56 Sketch_1 = geompy.MakeSketcherOnPlane("Sketcher:F 0.000000 0.000000:TT 120.000000 0.000000:T 0.000000 100.000000:T -120.000000 0.000000:WW", Plane_1 )
57 sectionDroite = geompy.MakeFaceWires([Sketch_1], 1)
58 Box_1_vertex_7 = geompy.GetSubShape(eprouvetteDroite, [7])
59 Box_1_vertex_16 = geompy.GetSubShape(eprouvetteDroite, [16])
60 Line_1 = geompy.MakeLineTwoPnt(Box_1_vertex_7, Box_1_vertex_16)
61 geomObj_3 = geompy.MakeMarker(0, 0, 0, 1, 0, 0, 0, 1, 0)
62 Plane_2 = geompy.MakePlane(Vertex_1, Line_1, 2000)
63 geomObj_4 = geompy.MakeMarker(0, 0, 0, 1, 0, 0, 0, 1, 0)
64 Sketch_2 = geompy.MakeSketcherOnPlane("Sketcher:F 0.000000 0.000000:T 120.000000 0.000000:T 0.000000 100.000000:T -120.000000 20.000000:WW", Plane_2 )
65 SectionInclinee = geompy.MakeFaceWires([Sketch_2], 1)
66 geompy.ExportBREP(sectionDroite, os.path.join(gmu.pathBloc, "materielCasTests", "EprouvetteDroiteFissPb1.brep"))
67 geompy.ExportBREP(SectionInclinee, os.path.join(gmu.pathBloc, "materielCasTests", "EprouvetteDroiteFiss2.brep"))
68 Vertex_2 = geompy.MakeVertex(110, -10, 200)
69 Vertex_3 = geompy.MakeVertex(110, 80, 200)
70 Vertex_4 = geompy.MakeVertex(-10, 80, 200)
71 Line_2 = geompy.MakeLineTwoPnt(Vertex_1, Vertex_2)
72 Line_3 = geompy.MakeLineTwoPnt(Vertex_2, Vertex_3)
73 Line_4 = geompy.MakeLineTwoPnt(Vertex_3, Vertex_4)
74 Line_5 = geompy.MakeLineTwoPnt(Vertex_4, Vertex_1)
75 Face_1 = geompy.MakeFaceWires([Line_2, Line_3, Line_4, Line_5], 1)
76 Vertex_5 = geompy.MakeVertex(110, -10, 180)
77 Vertex_6 = geompy.MakeVertex(110, 70, 180)
78 Face_1_vertex_4 = geompy.GetSubShape(Face_1, [4])
79 Line_6 = geompy.MakeLineTwoPnt(Face_1_vertex_4, Vertex_5)
80 Line_7 = geompy.MakeLineTwoPnt(Vertex_5, Vertex_6)
81 Face_1_vertex_9 = geompy.GetSubShape(Face_1, [9])
82 Line_8 = geompy.MakeLineTwoPnt(Vertex_6, Face_1_vertex_9)
83 Face_2 = geompy.MakeFaceWires([Line_5, Line_6, Line_7, Line_8], 1)
84 geompy.ExportBREP(Face_1, os.path.join(gmu.pathBloc, "materielCasTests", "EprouvetteDroiteFiss_1.brep"))
85 geompy.ExportBREP(Face_2, os.path.join(gmu.pathBloc, "materielCasTests", "EprouvetteDroiteFiss_2.brep"))
86 geompy.addToStudy( O, 'O' )
87 geompy.addToStudy( OX, 'OX' )
88 geompy.addToStudy( OY, 'OY' )
89 geompy.addToStudy( OZ, 'OZ' )
90 geompy.addToStudy( eprouvetteDroite, 'eprouvetteDroite' )
91 geompy.addToStudyInFather( eprouvetteDroite, Compound_z, 'Compound_z' )
92 geompy.addToStudyInFather( eprouvetteDroite, Compound_y, 'Compound_y' )
93 geompy.addToStudyInFather( eprouvetteDroite, Compound_x, 'Compound_x' )
94 geompy.addToStudy( Vertex_1, 'Vertex_1' )
95 geompy.addToStudy( Plane_1, 'Plane_1' )
96 geompy.addToStudy( Sketch_1, 'Sketch_1' )
97 geompy.addToStudy( sectionDroite, 'sectionDroite' )
98 geompy.addToStudyInFather( eprouvetteDroite, Box_1_vertex_7, 'Box_1:vertex_7' )
99 geompy.addToStudyInFather( eprouvetteDroite, Box_1_vertex_16, 'Box_1:vertex_16' )
100 geompy.addToStudy( Line_1, 'Line_1' )
101 geompy.addToStudy( Plane_2, 'Plane_2' )
102 geompy.addToStudy( Sketch_2, 'Sketch_2' )
103 geompy.addToStudy( SectionInclinee, 'SectionInclinee' )
104 geompy.addToStudy( Vertex_2, 'Vertex_2' )
105 geompy.addToStudy( Vertex_3, 'Vertex_3' )
106 geompy.addToStudy( Vertex_4, 'Vertex_4' )
107 geompy.addToStudy( Line_2, 'Line_2' )
108 geompy.addToStudy( Line_3, 'Line_3' )
109 geompy.addToStudy( Line_4, 'Line_4' )
110 geompy.addToStudy( Line_5, 'Line_5' )
111 geompy.addToStudy( Face_1, 'Face_1' )
112 geompy.addToStudy( Vertex_5, 'Vertex_5' )
113 geompy.addToStudy( Vertex_6, 'Vertex_6' )
114 geompy.addToStudyInFather( Face_1, Face_1_vertex_4, 'Face_1:vertex_4' )
115 geompy.addToStudy( Line_6, 'Line_6' )
116 geompy.addToStudy( Line_7, 'Line_7' )
117 geompy.addToStudyInFather( Face_1, Face_1_vertex_9, 'Face_1:vertex_9' )
118 geompy.addToStudy( Line_8, 'Line_8' )
119 geompy.addToStudy( Face_2, 'Face_2' )
120
121 ###
122 ### SMESH component
123 ###
124
125 import  SMESH, SALOMEDS
126 from salome.smesh import smeshBuilder
127
128 smesh = smeshBuilder.New()
129 from salome.StdMeshers import StdMeshersBuilder
130 eprouvetteDroite_1 = smesh.Mesh(eprouvetteDroite)
131 smesh.SetName(eprouvetteDroite_1, 'eprouvetteDroite')
132 Regular_1D = eprouvetteDroite_1.Segment()
133 Nb_Segments_1 = Regular_1D.NumberOfSegments(50,[],[  ])
134 Nb_Segments_1.SetDistrType( 0 )
135 Quadrangle_2D = eprouvetteDroite_1.Quadrangle(algo=smeshBuilder.QUADRANGLE)
136 Hexa_3D = eprouvetteDroite_1.Hexahedron(algo=smeshBuilder.Hexa)
137 Regular_1D_1 = eprouvetteDroite_1.Segment(geom=Compound_y)
138 Nb_Segments_2 = Regular_1D_1.NumberOfSegments(20,[],[  ])
139 Nb_Segments_2.SetDistrType( 0 )
140 Regular_1D_2 = eprouvetteDroite_1.Segment(geom=Compound_x)
141 Nb_Segments_3 = Regular_1D_2.NumberOfSegments(10,[],[  ])
142 Nb_Segments_3.SetDistrType( 0 )
143
144 is_done = eprouvetteDroite_1.Compute()
145 text = "eprouvetteDroite_1.Compute"
146 if is_done:
147   logging.info(text+" OK")
148 else:
149   text = "Erreur au calcul du maillage.\n" + text
150   logging.info(text)
151   raise Exception(text)
152
153 eprouvetteDroite_1.ExportMED(os.path.join(gmu.pathBloc, "materielCasTests", "eprouvetteDroite.med"))
154 SubMesh_1 = Regular_1D_1.GetSubMesh()
155 SubMesh_2 = Regular_1D_2.GetSubMesh()
156
157 ## set object names
158 smesh.SetName(eprouvetteDroite_1.GetMesh(), 'eprouvetteDroite')
159 smesh.SetName(Regular_1D.GetAlgorithm(), 'Regular_1D')
160 smesh.SetName(Nb_Segments_1, 'Nb. Segments_1')
161 smesh.SetName(Quadrangle_2D.GetAlgorithm(), 'Quadrangle_2D')
162 smesh.SetName(Hexa_3D.GetAlgorithm(), 'Hexa_3D')
163 smesh.SetName(Nb_Segments_2, 'Nb. Segments_2')
164 smesh.SetName(Nb_Segments_3, 'Nb. Segments_3')
165 smesh.SetName(SubMesh_1, 'SubMesh_1')
166 smesh.SetName(SubMesh_2, 'SubMesh_2')
167
168 if salome.sg.hasDesktop():
169   salome.sg.updateObjBrowser()