]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESH_I/SMESH_test.py
Salome HOME
3023a28d01c5a529d91eb607f19ccce1f813c4cd
[modules/smesh.git] / src / SMESH_I / SMESH_test.py
1 #  SMESH SMESH_I : idl implementation based on 'SMESH' unit's calsses
2 #
3 #  Copyright (C) 2003  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 #  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS 
5
6 #  This library is free software; you can redistribute it and/or 
7 #  modify it under the terms of the GNU Lesser General Public 
8 #  License as published by the Free Software Foundation; either 
9 #  version 2.1 of the License. 
10
11 #  This library is distributed in the hope that it will be useful, 
12 #  but WITHOUT ANY WARRANTY; without even the implied warranty of 
13 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU 
14 #  Lesser General Public License for more details. 
15
16 #  You should have received a copy of the GNU Lesser General Public 
17 #  License along with this library; if not, write to the Free Software 
18 #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA 
19
20 #  See http://www.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org 
21 #
22 #
23 #
24 #  File   : SMESH_test.py
25 #  Module : SMESH
26
27 import SMESH
28 import smeshpy
29 import salome
30 from salome import sg
31 import math
32 #import SMESH_BasicHypothesis_idl
33
34 import geompy
35
36 geom = salome.lcc.FindOrLoadComponent("FactoryServer", "GEOM")
37 myBuilder = salome.myStudy.NewBuilder()
38 from geompy import gg
39
40 ShapeTypeCompSolid = 1
41 ShapeTypeSolid = 2
42 ShapeTypeShell = 3
43 ShapeTypeFace = 4
44 ShapeTypeWire = 5
45 ShapeTypeEdge = 6
46 ShapeTypeVertex = 7
47
48 # ---- define a box
49
50 box = geompy.MakeBox(0., 0., 0., 100., 200., 300.)
51 idb = geompy.addToStudy(box,"box")
52
53 # ---- add first face of box in study
54
55 subShapeList=geompy.SubShapeAll(box,ShapeTypeFace)
56 face=subShapeList[0]
57 name = geompy.SubShapeName( face._get_Name(), box._get_Name() )
58 print name
59 idf=geompy.addToStudyInFather(box,face,name)
60
61 # ---- add shell from box  in study
62
63 subShellList=geompy.SubShapeAll(box,ShapeTypeShell)
64 shell = subShellList[0]
65 name = geompy.SubShapeName( shell._get_Name(), box._get_Name() )
66 print name
67 ids=geompy.addToStudyInFather(box,shell,name)
68
69 # ---- add first edge of face in study
70
71 edgeList = geompy.SubShapeAll(face,ShapeTypeEdge)
72 edge=edgeList[0];
73 name = geompy.SubShapeName( edge._get_Name(), face._get_Name() )
74 print name
75 ide=geompy.addToStudyInFather(face,edge,name)
76
77 # ---- launch SMESH, init a Mesh with the box
78 gen=smeshpy.smeshpy()
79 mesh=gen.Init(idb)
80
81 # ---- create Hypothesis
82
83 print "-------------------------- create Hypothesis"
84 print "-------------------------- LocalLength"
85 hyp1=gen.CreateHypothesis("LocalLength")
86 print hyp1.GetName()
87 print hyp1.GetId()
88 hypo1 = hyp1._narrow(SMESH.SMESH_LocalLength)
89 print hypo1.GetLength()
90 hypo1.SetLength(100)
91 print hypo1.GetLength()
92
93 print "-------------------------- bidon"
94 hyp3=gen.CreateHypothesis("bidon")
95
96 print "-------------------------- NumberOfSegments"
97 hyp3=gen.CreateHypothesis("NumberOfSegments")
98 hypo3=hyp3._narrow(SMESH.SMESH_NumberOfSegments)
99 hypo3.SetNumberOfSegments(7)
100 print hypo3.GetName()
101 print hypo3.GetNumberOfSegments()
102 print hypo3.GetId()
103
104 print "-------------------------- MaxElementArea"
105 hyp4=gen.CreateHypothesis("MaxElementArea")
106 hypo4=hyp4._narrow(SMESH.SMESH_MaxElementArea)
107 hypo4.SetMaxElementArea(5000)
108 print hypo4.GetName()
109 print hypo4.GetMaxElementArea()
110 print hypo4.GetId()
111
112 print "-------------------------- Regular_1D"
113 alg1=gen.CreateHypothesis("Regular_1D")
114 print alg1.GetName()
115 print alg1.GetId()
116 algo1=alg1._narrow(SMESH.SMESH_Algo)
117 listHyp=algo1.GetCompatibleHypothesis()
118 for hyp in listHyp:
119     print hyp
120     
121 algo_1=alg1._narrow(SMESH.SMESH_Regular_1D)
122 print algo_1.GetId()
123
124 print "-------------------------- MEFISTO_2D"
125 alg2=gen.CreateHypothesis("MEFISTO_2D")
126 print alg2.GetName()
127 print alg2.GetId()
128 algo2=alg2._narrow(SMESH.SMESH_Algo)
129 listHyp=algo2.GetCompatibleHypothesis()
130 for hyp in listHyp:
131     print hyp
132 algo_2=alg2._narrow(SMESH.SMESH_MEFISTO_2D)
133 print algo_2.GetId()
134
135 # ---- add hypothesis to edge
136
137 print "-------------------------- add hypothesis to edge"
138 edge=salome.IDToObject(ide)
139 submesh=mesh.GetElementsOnShape(edge)
140 ret=mesh.AddHypothesis(edge,algo_1)
141 print ret
142 ret=mesh.AddHypothesis(edge,hypo1)
143 print ret
144
145 # ---- compute edge
146
147 ##print "-------------------------- compute edge"
148 ##ret=gen.Compute(mesh,ide)
149 ##print ret
150 ##log=mesh.GetLog(1);
151 ##for a in log:
152 ##    print a
153
154 # ---- add hypothesis to box
155
156 print "-------------------------- add hypothesis to box"
157 box=salome.IDToObject(idb)
158 submesh=mesh.GetElementsOnShape(box)
159 ret=mesh.AddHypothesis(box,algo_1)
160 print ret
161 ret=mesh.AddHypothesis(box,hypo1)
162 print ret
163 ret=mesh.AddHypothesis(box,algo_2)
164 print ret
165 ret=mesh.AddHypothesis(box,hypo4)
166 print ret
167
168 # ---- compute face
169
170 print "-------------------------- compute face"
171 ret=gen.Compute(mesh,idf)
172 print ret
173 log=mesh.GetLog(1);
174 for a in log:
175     print "-------"
176     ii = 0
177     ir = 0
178     comType = a.commandType
179     if comType == 0:
180         for i in range(a.number):
181             ind = a.indexes[ii]
182             ii = ii+1
183             r1 = a.coords[ir]
184             ir = ir+1
185             r2 = a.coords[ir]
186             ir = ir+1
187             r3 = a.coords[ir]
188             ir = ir+1
189             print "AddNode %i - %g %g %g" % (ind, r1, r2, r3)
190     elif comType == 1:
191         for i in range(a.number):
192             ind = a.indexes[ii]
193             ii = ii+1
194             i1 = a.indexes[ii]
195             ii = ii+1
196             i2 = a.indexes[ii]
197             ii = ii+1
198             print "AddEdge %i - %i %i" % (ind, i1, i2)
199     elif comType == 2:
200         for i in range(a.number):
201             ind = a.indexes[ii]
202             ii = ii+1
203             i1 = a.indexes[ii]
204             ii = ii+1
205             i2 = a.indexes[ii]
206             ii = ii+1
207             i3 = a.indexes[ii]
208             ii = ii+1
209             print "AddTriangle %i - %i %i %i" % (ind, i1, i2, i3)
210
211 # ---- compute box
212
213 ##print "-------------------------- compute box"
214 ##ret=gen.Compute(mesh,idb)
215 ##print ret
216 ##log=mesh.GetLog(1);
217 ##print log
218
219 ##shell=salome.IDToObject(ids)
220 ##submesh=mesh.GetElementsOnShape(shell)
221 ##ret=mesh.AddHypothesis(shell,algo_1)
222 ##print ret
223 ##ret=mesh.AddHypothesis(shell,hypo1)
224 ##print ret
225 ##ret=gen.Compute(mesh,ids)
226 ##print ret
227