Salome HOME
Merge changes from 'master' branch.
[modules/smesh.git] / src / Tools / blocFissure / materielCasTests / decoupeCylindre.py
1 # -*- coding: utf-8 -*-
2
3 import sys
4 import salome
5
6 salome.salome_init()
7
8 import salome_notebook
9 notebook = salome_notebook.notebook
10
11 import os
12 from blocFissure import gmu
13
14 ###
15 ### GEOM component
16 ###
17
18 import GEOM
19 from salome.geom import geomBuilder
20 import math
21 import SALOMEDS
22
23
24 geompy = geomBuilder.New()
25
26 O = geompy.MakeVertex(0, 0, 0)
27 OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
28 OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
29 OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
30 O_1 = geompy.MakeVertex(0, 0, 0)
31 OX_1 = geompy.MakeVectorDXDYDZ(1, 0, 0)
32 OY_1 = geompy.MakeVectorDXDYDZ(0, 1, 0)
33 OZ_1 = geompy.MakeVectorDXDYDZ(0, 0, 1)
34 Vertex_1 = geompy.MakeVertex(0, 0, 500)
35 Vertex_2 = geompy.MakeVertex(100, 0, 500)
36 Vertex_3 = geompy.MakeVertex(110, 0, 500)
37 Vertex_4 = geompy.MakeVertex(117.071068, -2.928932, 500)
38 Vertex_5 = geompy.MakeVertex(120, -10, 500)
39 Vertex_6 = geompy.MakeVertex(120, -30, 500)
40 Vertex_7 = geompy.MakeVertex(122.928932, -37.071068, 500)
41 Vertex_8 = geompy.MakeVertex(130, -40, 500)
42 Vertex_9 = geompy.MakeVertex(135, -40, 500)
43 Vertex_10 = geompy.MakeVertex(160, -40, 500)
44 Plane_1 = geompy.MakePlaneLCS(None, 2000, 2)
45 Mirror_1_1 = geompy.MakeMirrorByPlane(Vertex_2, Plane_1)
46 Mirror_1_2 = geompy.MakeMirrorByPlane(Vertex_3, Plane_1)
47 Mirror_1_3 = geompy.MakeMirrorByPlane(Vertex_4, Plane_1)
48 Mirror_1_4 = geompy.MakeMirrorByPlane(Vertex_5, Plane_1)
49 Mirror_1_5 = geompy.MakeMirrorByPlane(Vertex_6, Plane_1)
50 Mirror_1_6 = geompy.MakeMirrorByPlane(Vertex_7, Plane_1)
51 Mirror_1_7 = geompy.MakeMirrorByPlane(Vertex_8, Plane_1)
52 Mirror_1_8 = geompy.MakeMirrorByPlane(Vertex_9, Plane_1)
53 Mirror_1_9 = geompy.MakeMirrorByPlane(Vertex_10, Plane_1)
54 Curve_2 = geompy.MakeInterpol([Mirror_1_9, Mirror_1_8, Mirror_1_7, Mirror_1_6, Mirror_1_5, Mirror_1_4, Mirror_1_3, Mirror_1_2, Mirror_1_1, Vertex_1, Vertex_2, Vertex_3, Vertex_4, Vertex_5, Vertex_6, Vertex_7, Vertex_8, Vertex_9, Vertex_10], False, False)
55 Circle_1 = geompy.MakeCircle(Vertex_1, None, 145)
56 Vertex_11 = geompy.MakeVertex(0, -165, 500)
57 Curve_2_vertex_2 = geompy.GetSubShape(Curve_2, [2])
58 Curve_2_vertex_3 = geompy.GetSubShape(Curve_2, [3])
59 Arc_1 = geompy.MakeArc(Curve_2_vertex_2, Vertex_11, Curve_2_vertex_3)
60 FissInCylindre = geompy.MakeFaceWires([Curve_2, Arc_1], 1)
61 Divided_Cylinder_1 = geompy.MakeDividedCylinder(145, 800, GEOM.SQUARE)
62 CylindreSain = geompy.MakeRotation(Divided_Cylinder_1, OZ, 45*math.pi/180.0)
63 [Compound_1, vertical, radial, Compound_4] = geompy.Propagate(CylindreSain)
64 geompy.ExportBREP(FissInCylindre, os.path.join(gmu.pathBloc, "materielCasTests/FissInCylindre.brep"))
65 Vertex_12 = geompy.MakeVertex(0, -145, 500)
66 Circle_2 = geompy.MakeCircle(Vertex_12, None, 145)
67 Face_1 = geompy.MakeFaceWires([Circle_2], 1)
68 Vertex_13 = geompy.MakeVertex(0, 0, 500)
69 Disk_1 = geompy.MakeDiskPntVecR(Vertex_13, OZ_1, 170)
70 FissInCylindre2 = geompy.MakeCommon(Face_1, Disk_1)
71 geompy.ExportBREP(FissInCylindre2, os.path.join(gmu.pathBloc, "materielCasTests/FissInCylindre2.brep"))
72 geompy.addToStudy( O, 'O' )
73 geompy.addToStudy( OX, 'OX' )
74 geompy.addToStudy( OY, 'OY' )
75 geompy.addToStudy( OZ, 'OZ' )
76 geompy.addToStudy( O_1, 'O' )
77 geompy.addToStudy( OX_1, 'OX' )
78 geompy.addToStudy( OY_1, 'OY' )
79 geompy.addToStudy( OZ_1, 'OZ' )
80 geompy.addToStudy( Vertex_1, 'Vertex_1' )
81 geompy.addToStudy( Vertex_2, 'Vertex_2' )
82 geompy.addToStudy( Vertex_3, 'Vertex_3' )
83 geompy.addToStudy( Vertex_4, 'Vertex_4' )
84 geompy.addToStudy( Vertex_5, 'Vertex_5' )
85 geompy.addToStudy( Vertex_6, 'Vertex_6' )
86 geompy.addToStudy( Vertex_7, 'Vertex_7' )
87 geompy.addToStudy( Vertex_8, 'Vertex_8' )
88 geompy.addToStudy( Vertex_9, 'Vertex_9' )
89 geompy.addToStudy( Vertex_10, 'Vertex_10' )
90 geompy.addToStudy( Plane_1, 'Plane_1' )
91 geompy.addToStudy( Mirror_1_1, 'Mirror_1_1' )
92 geompy.addToStudy( Mirror_1_2, 'Mirror_1_2' )
93 geompy.addToStudy( Mirror_1_3, 'Mirror_1_3' )
94 geompy.addToStudy( Mirror_1_4, 'Mirror_1_4' )
95 geompy.addToStudy( Mirror_1_5, 'Mirror_1_5' )
96 geompy.addToStudy( Mirror_1_6, 'Mirror_1_6' )
97 geompy.addToStudy( Mirror_1_7, 'Mirror_1_7' )
98 geompy.addToStudy( Mirror_1_8, 'Mirror_1_8' )
99 geompy.addToStudy( Mirror_1_9, 'Mirror_1_9' )
100 geompy.addToStudy( Curve_2, 'Curve_2' )
101 geompy.addToStudy( Circle_1, 'Circle_1' )
102 geompy.addToStudy( Vertex_11, 'Vertex_11' )
103 geompy.addToStudyInFather( Curve_2, Curve_2_vertex_2, 'Curve_2:vertex_2' )
104 geompy.addToStudyInFather( Curve_2, Curve_2_vertex_3, 'Curve_2:vertex_3' )
105 geompy.addToStudy( Arc_1, 'Arc_1' )
106 geompy.addToStudy( FissInCylindre, 'FissInCylindre' )
107 geompy.addToStudy( Divided_Cylinder_1, 'Divided Cylinder_1' )
108 geompy.addToStudy( CylindreSain, 'CylindreSain' )
109 geompy.addToStudyInFather( CylindreSain, Compound_1, 'Compound_1' )
110 geompy.addToStudyInFather( CylindreSain, vertical, 'vertical' )
111 geompy.addToStudyInFather( CylindreSain, radial, 'radial' )
112 geompy.addToStudyInFather( CylindreSain, Compound_4, 'Compound_4' )
113 geompy.addToStudy( Vertex_12, 'Vertex_12' )
114 geompy.addToStudy( Circle_2, 'Circle_2' )
115 geompy.addToStudy( Face_1, 'Face_1' )
116 geompy.addToStudy( Vertex_13, 'Vertex_13' )
117 geompy.addToStudy( Disk_1, 'Disk_1' )
118 geompy.addToStudy( FissInCylindre2, 'FissInCylindre2' )
119
120 ###
121 ### SMESH component
122 ###
123
124 import  SMESH, SALOMEDS
125 from salome.smesh import smeshBuilder
126
127 smesh = smeshBuilder.New()
128 from salome.StdMeshers import StdMeshersBuilder
129 smeshObj_1 = smesh.CreateHypothesis('NumberOfSegments')
130 smeshObj_1.SetNumberOfSegments( 5 )
131 smeshObj_1.SetDistrType( 0 )
132 CylindreSain_1 = smesh.Mesh(CylindreSain)
133 Regular_1D = CylindreSain_1.Segment()
134 Nb_Segments_1 = Regular_1D.NumberOfSegments(15,[],[  ])
135 Nb_Segments_1.SetDistrType( 0 )
136 Quadrangle_2D = CylindreSain_1.Quadrangle(algo=smeshBuilder.QUADRANGLE)
137 Hexa_3D = CylindreSain_1.Hexahedron(algo=smeshBuilder.Hexa)
138 Regular_1D_1 = CylindreSain_1.Segment(geom=vertical)
139 Nb_Segments_2 = Regular_1D_1.NumberOfSegments(30,[],[  ])
140 Nb_Segments_2.SetDistrType( 0 )
141 Regular_1D_2 = CylindreSain_1.Segment(geom=radial)
142 Nb_Segments_3 = Regular_1D_2.NumberOfSegments(6,[],[  ])
143 Nb_Segments_3.SetDistrType( 0 )
144 isDone = CylindreSain_1.Compute()
145 smesh.SetName(CylindreSain_1, 'CylindreSain')
146 CylindreSain_1.ExportMED(os.path.join(gmu.pathBloc, "materielCasTests//CylindreSain.med"))
147 SubMesh_1 = Regular_1D_1.GetSubMesh()
148 SubMesh_2 = Regular_1D_2.GetSubMesh()
149
150 ## some objects were removed
151 aStudyBuilder = salome.myStudy.NewBuilder()
152 SO = salome.myStudy.FindObjectIOR(salome.myStudy.ConvertObjectToIOR(smeshObj_1))
153 if SO is not None: aStudyBuilder.RemoveObjectWithChildren(SO)
154 ## set object names
155 smesh.SetName(CylindreSain_1.GetMesh(), 'CylindreSain')
156 smesh.SetName(Regular_1D.GetAlgorithm(), 'Regular_1D')
157 smesh.SetName(Nb_Segments_1, 'Nb. Segments_1')
158 smesh.SetName(Quadrangle_2D.GetAlgorithm(), 'Quadrangle_2D')
159 smesh.SetName(Hexa_3D.GetAlgorithm(), 'Hexa_3D')
160 smesh.SetName(Nb_Segments_2, 'Nb. Segments_2')
161 smesh.SetName(Nb_Segments_3, 'Nb. Segments_3')
162 smesh.SetName(SubMesh_1, 'SubMesh_1')
163 smesh.SetName(SubMesh_2, 'SubMesh_2')
164
165 if salome.sg.hasDesktop():
166   salome.sg.updateObjBrowser()