]> SALOME platform Git repositories - modules/kernel.git/blob - doc/salome/examples/example19
Salome HOME
Merge from V6_main_20120808 08Aug12
[modules/kernel.git] / doc / salome / examples / example19
1
2 #======================================================================
3 #1. Create Study and add some components to it
4 #======================================================================
5
6 #=================================
7 #       create AttributeReal      
8 #=================================
9 A = batchmode_geompy.myBuilder.FindOrCreateAttribute(batchmode_geompy.father, "AttributeReal")
10 if A == None :
11         raise  RuntimeError, "Can't create AttributeReal attribute"
12 A = A._narrow(SALOMEDS.AttributeReal)
13 A.SetValue(0.0001)
14 if A.Value() != 0.0001:
15         raise  RuntimeError, "Error : wrong value of  AttributeReal"
16
17 # ===  Geometry  ==================================================
18
19 import batchmode_geompy
20  
21 #===================================
22 #     define a box
23 #===================================
24
25 box = batchmode_geompy.MakeBox(0., 0., 0., 100., 200., 300.)
26 idbox = batchmode_geompy.addToStudy(box,"box")
27
28
29 geom = batchmode_geompy.geom
30 import SMESH
31 import batchmode_smesh
32 ShapeType = batchmode_smesh.ShapeType
33
34 # ---- add first face of box in study
35
36 subShapeList = batchmode_geompy.SubShapeAll(box,ShapeType["Face"])
37 face=subShapeList[0]
38 name = "box_face"
39 print name
40 idface= batchmode_geompy.addToStudyInFather(box,face,name)
41
42 # ---- add shell from box  in study
43
44 subShellList= batchmode_geompy.SubShapeAll(box,ShapeType["Shell"])
45 shell = subShellList[0]
46 name = "box_shell"
47 print name
48 idshell= batchmode_geompy.addToStudyInFather(box,shell,name)
49
50 # ---- add first edge of face in study
51
52 edgeList =  batchmode_geompy.SubShapeAll(face,ShapeType["Edge"])
53 edge=edgeList[0];
54 name = "face_edge"
55 print name
56 idedge= batchmode_geompy.addToStudyInFather(face,edge,name)
57
58 # ---- launch SMESH, init a Mesh with the box
59
60 smesh =  batchmode_smesh.smesh
61 #  --  Init  --
62 shape = batchmode_geompy.IDToObject(idbox)
63 mesh=smesh.Init(geom, batchmode_geompy.myStudyId, shape)
64
65 orb = batchmode_geompy.orb
66
67 idmesh = batchmode_smesh.AddNewMesh( orb.object_to_string(mesh) )
68 batchmode_smesh.SetName(idmesh, "Meshbox");
69 batchmode_smesh.SetShape(idbox, idmesh);
70
71 # ---- create Hypothesis
72
73 print "-------------------------- create Hypothesis ----------------------"
74 print "-------------------------- LocalLength"
75 hyp1 = smesh.CreateHypothesis("LocalLength", batchmode_geompy.myStudyId )
76 hypLen1 = hyp1._narrow(SMESH.SMESH_LocalLength)
77 hypLen1.SetLength(100)
78 print hypLen1.GetName()
79 print hypLen1.GetId()
80 print hypLen1.GetLength()
81
82 idlength = batchmode_smesh.AddNewHypothesis( orb.object_to_string(hypLen1) );
83 batchmode_smesh.SetName(idlength, "Local_Length_100");
84
85 print "-------------------------- NumberOfSegments"
86 hyp2 =  smesh.CreateHypothesis("NumberOfSegments", batchmode_geompy.myStudyId )
87 hypNbSeg1=hyp2._narrow(SMESH.SMESH_NumberOfSegments)
88 hypNbSeg1.SetNumberOfSegments(7)
89 print hypNbSeg1.GetName()
90 print hypNbSeg1.GetId()
91 print hypNbSeg1.GetNumberOfSegments()
92
93 idseg =  batchmode_smesh.AddNewHypothesis( orb.object_to_string(hypNbSeg1) );
94 batchmode_smesh.SetName(idseg, "NumberOfSegments_7");
95
96 print "-------------------------- MaxElementArea"
97 hyp3 = smesh.CreateHypothesis("MaxElementArea",  batchmode_geompy.myStudyId)
98 hypArea1=hyp3._narrow(SMESH.SMESH_MaxElementArea)
99 hypArea1.SetMaxElementArea(2500)
100 print hypArea1.GetName()
101 print hypArea1.GetId()
102 print hypArea1.GetMaxElementArea()
103
104 idarea1 = batchmode_smesh.AddNewHypothesis( orb.object_to_string(hypArea1) );
105 batchmode_smesh.SetName(idarea1, "MaxElementArea_2500");
106
107 print "-------------------------- MaxElementArea"
108 hyp3 = smesh.CreateHypothesis("MaxElementArea",  batchmode_geompy.myStudyId)
109 hypArea2 = hyp3._narrow(SMESH.SMESH_MaxElementArea)
110 hypArea2.SetMaxElementArea(500)
111 print hypArea2.GetName()
112 print hypArea2.GetId()
113 print hypArea2.GetMaxElementArea()
114
115 idarea2 = batchmode_smesh.AddNewHypothesis( orb.object_to_string(hypArea2) );
116 batchmode_smesh.SetName(idarea2, "MaxElementArea_500");
117
118 print "-------------------------- Regular_1D"
119 alg1 = smesh.CreateHypothesis("Regular_1D",  batchmode_geompy.myStudyId)
120 algo1 = alg1._narrow(SMESH.SMESH_Algo)
121 listHyp = algo1.GetCompatibleHypothesis()
122 for hyp in listHyp:
123     print hyp
124 algoReg=alg1._narrow(SMESH.SMESH_Regular_1D)
125 print algoReg.GetName()
126 print algoReg.GetId()
127
128 idreg = batchmode_smesh.AddNewAlgorithms( orb.object_to_string(algoReg) );
129 batchmode_smesh.SetName(idreg, "Regular_1D");
130
131 print "-------------------------- MEFISTO_2D"
132 alg2 = smesh.CreateHypothesis("MEFISTO_2D",  batchmode_geompy.myStudyId)
133 algo2 = alg2._narrow(SMESH.SMESH_Algo)
134 listHyp=algo2.GetCompatibleHypothesis()
135 for hyp in listHyp:
136     print hyp
137 algoMef=alg2._narrow(SMESH.SMESH_MEFISTO_2D)
138 print algoMef.GetName()
139 print algoMef.GetId()
140
141 idmef = batchmode_smesh.AddNewAlgorithms( orb.object_to_string(algoMef) );
142 batchmode_smesh.SetName(idmef, "MEFISTO_2D");
143
144
145 # ---- add hypothesis to edge
146
147 print "-------------------------- add hypothesis to edge"
148 edge = batchmode_geompy.IDToObject(idedge)
149 submesh = mesh.GetElementsOnShape(edge)
150 ret = mesh.AddHypothesis(edge,algoReg)
151 print ret
152 ret=mesh.AddHypothesis(edge,hypLen1)
153 print ret
154
155 idsm1 = batchmode_smesh.AddSubMeshOnShape( idmesh,
156                                            idedge,
157                                            orb.object_to_string(submesh),
158                                            ShapeType["Edge"] )
159 batchmode_smesh.SetName(idsm1, "SubMeshEdge")
160 batchmode_smesh.SetAlgorithms( idsm1, idreg );
161 batchmode_smesh.SetHypothesis( idsm1, idlength );
162
163 print "-------------------------- add hypothesis to face"
164 face=batchmode_geompy.IDToObject(idface)
165 submesh=mesh.GetElementsOnShape(face)
166 ret=mesh.AddHypothesis(face,hypArea2)
167 print ret
168
169 idsm2 = batchmode_smesh.AddSubMeshOnShape( idmesh,
170                                     idface,
171                                     orb.object_to_string(submesh),
172                                     ShapeType["Face"] )
173 batchmode_smesh.SetName(idsm2, "SubMeshFace")
174 batchmode_smesh.SetHypothesis( idsm2, idarea2 );
175
176 # ---- add hypothesis to box
177
178 print "-------------------------- add hypothesis to box"
179 box=batchmode_geompy.IDToObject(idbox)
180 submesh=mesh.GetElementsOnShape(box)
181 ret=mesh.AddHypothesis(box,algoReg)
182 print ret
183 ret=mesh.AddHypothesis(box,hypNbSeg1)
184 print ret
185 ret=mesh.AddHypothesis(box,algoMef)
186 print ret
187 ret=mesh.AddHypothesis(box,hypArea1)
188 print ret
189
190 batchmode_smesh.SetAlgorithms( idmesh, idreg );
191 batchmode_smesh.SetHypothesis( idmesh, idseg );
192 batchmode_smesh.SetAlgorithms( idmesh, idmef );
193 batchmode_smesh.SetHypothesis( idmesh, idarea1 );
194
195 # ---- compute box
196
197 print "-------------------------- compute box"
198 ret=smesh.Compute(mesh,box)
199 print ret
200 log=mesh.GetLog(0); # no erase trace
201 for linelog in log:
202     print linelog
203
204
205 #=======================================================
206 #    add  SuperVision component
207 #=======================================================
208
209 from batchmode_SuperV import *
210 import os
211 dir= os.getenv("SUPERV_ROOT_DIR")
212 if dir == None:
213         raise RuntimeError, "SUPERV_ROOT_DIR is not defined"
214 xmlfile = dir +"/examples/GraphEssai.xml"
215 print "--------------\n"+xmlfile+"\n--------------\n"
216
217 myGraph = Graph ( xmlfile )
218
219 # This DataFlow is "valid" : no loop, correct links between Nodes etc...
220 print myGraph.IsValid()
221
222 # Get Nodes
223 myGraph.PrintNodes()
224 Add,Sub,Mul,Div = myGraph.Nodes()
225
226 # Load Datas
227 Addx = Add.Input("x",3.)
228 Addy = Add.Input("y",4.5)
229 Subx = Sub.Input("x",1.5)
230
231 # Get Output Port
232 Addz = Add.Port('z')
233 Subz = Sub.Port('z')
234 Mulz = Mul.Port('z')
235 Divz = Div.Port('z')
236
237 # This DataFlow is "executable" : all pending Ports are defined with Datas
238 print myGraph.IsExecutable()
239
240 # Starts only execution of that DataFlow and gets control immediatly
241 print myGraph.Run()
242
243 # That DataFlow is running ==> 0 (false)
244 print myGraph.IsDone()
245
246 # Events of execution :
247 aStatus,aNode,anEvent,aState = myGraph.Event()
248 while aStatus :
249     print aNode.Thread(),aNode.SubGraph(),aNode.Name(),anEvent,aState
250     aStatus,aNode,anEvent,aState = myGraph.Event()
251 print myGraph.IsDone()
252
253 # Wait for Completion (but it is already done after event loop ...)
254 print "Done : ",myGraph.DoneW()
255
256 # Get result
257 print "Result : ",Divz.ToString()
258
259 # Intermediate results :
260 print "Intermediate Result Add\z : ",Addz.ToString()
261 print "Intermediate Result Sub\z : ",Subz.ToString()
262 print "Intermediate Result Mul\z : ",Mulz.ToString()
263
264 print " "
265 #print "Type : print myGraph.IsDone()"
266 #print "       If execution is finished ==> 1 (true)"
267 res=myGraph.IsDone()
268 if res != 1:
269         raise RuntimeError, "myGraph.Run() is not done"
270
271 print " "
272 print "Type : print Divz.ToString()"
273 print "       You will get the result"
274 Divz.ToString()
275
276 print " "
277 print "Type : myGraph.PrintPorts()"
278 print "       to see input and output values of the graph"
279 myGraph.PrintPorts()
280
281 print " "
282 print "Type : Add.PrintPorts()"
283 Add.PrintPorts()
284
285 print "Type : Sub.PrintPorts()"
286 Sub.PrintPorts()
287
288 print "Type : Mul.PrintPorts()"
289 Mul.PrintPorts()
290
291 print "Type : Div.PrintPorts()"
292 print "       to see input and output values of nodes"
293 Div.PrintPorts()
294
295 # Export will create newsupervisionexample.xml and the corresponding .py file
296 tmpdir=os.getenv("TmpDir")
297 if tmpdir is None:
298         tmpdir="/tmp"
299 file = tmpdir + "/newsupervisionexample"
300 print "--------------\n"+file+"\n--------------\n"
301 myGraph.Export(file)
302
303 ior = batchmode_geompy.orb.object_to_string(myGraph.G)
304 addStudy(ior)
305
306 GraphName = myGraph.Name()
307 print "Befor save ",
308 nodes = myGraph.Nodes()
309 length_bs = len(nodes)
310 print "ListOfNodes length = ", length_bs
311 names=[]
312 for node in nodes:
313         names.append(node.Name())
314 print names
315
316
317 #=================================
318 #      save / restore study      
319 #=================================
320 str= os.getenv("TmpDir")
321 if str == None:  
322         str = "/tmp"
323 file = str+"/test.hdf"
324
325
326 #==================================================
327 #1. SaveAs
328 #==================================================
329 print " -------  We will save to", file, "-----------"
330
331 batchmode_geompy.myStudyManager.SaveAs(file, batchmode_geompy.myStudy)
332 batchmode_geompy.myStudyManager.Close(batchmode_geompy.myStudy)
333
334 #rename the file and try to reread it again
335
336 os.mkdir(str + "/test_dir")
337
338 #2.==================================================
339 print " ------- We rename the file as " + str + "/test_dir/test_new.hdf"
340 os.rename(file, str+"/test_dir/test_new.hdf")
341
342 #==================================================
343 #3. Open
344 #==================================================
345 print " ------- We try to open " + str + "/test_dir/test_new.hdf"
346
347 try:
348         openedStudy=batchmode_geompy.myStudyManager.Open(str+"/test_dir/test_new.hdf")
349 except Exception:
350         raise  RuntimeError, "Can't open saved study!"
351
352 if openedStudy == None:
353         raise  RuntimeError, "Can't open saved study!"
354
355 father = openedStudy.FindComponent("GEOM")
356 if father is None:
357          raise  RuntimeError, "GEOM component is not found!  Wrong study is opened." 
358
359
360 #=================================
361 #     find AttributeReal
362 #=================================
363
364 res,A=father.FindAttribute("AttributeReal")
365 if res == 0 or A == None:
366         raise  RuntimeError, "Error:  not found AttributeReal"
367
368 A = A._narrow(SALOMEDS.AttributeReal)
369 if A.Value() != 0.0001:
370         raise  RuntimeError, "Error : wrong value of  AttributeReal"
371
372 #==================================================
373 #     find  box
374 #==================================================
375 box = openedStudy.FindObject("box")
376 if box is None :
377         raise  RuntimeError, "box was not found! Wrong study is opened."
378
379 edge=openedStudy.FindObject("face_edge")
380 if edge is None :
381         raise  RuntimeError, "face_edge was not found! Wrong study is opened."
382
383
384 father = openedStudy.FindComponent("MESH")
385 if father is None:
386          raise  RuntimeError, "MESH component is not found!  Wrong study is opened." 
387
388 hp=openedStudy.FindObject("Meshbox")
389 if hp is None:
390         raise  RuntimeError, "Meshbox object was not found! Wrong study is opened."
391
392 hp = openedStudy.FindObject("Algorithms Definition")
393 if hp is None:
394         raise  RuntimeError, "Algorithms Definition object was not found! Wrong study is opened."
395
396 hp = openedStudy.FindObject("MEFISTO_2D")
397 if hp is None:
398         raise  RuntimeError, "MEFISTO_2D object was not found! Wrong study is opened."
399
400 hp = openedStudy.FindObject("Hypothesis Definition")
401 if hp is None:
402         raise  RuntimeError, "Algoriths Definition was not found! Wrong study is opened."
403
404 #==================================================
405 # Find Supervisor
406 #==================================================
407 father = openedStudy.FindComponent("SUPERV")
408 if father is None: 
409          raise  RuntimeError, "SUPERV component is not found!  Wrong study is opened." 
410
411 SuperV = lcc.FindOrLoadComponent("SuperVisionContainer","Supervision")
412 Builder = openedStudy.NewBuilder()
413 Builder.LoadWith(father, SuperV)
414 import SALOMEDS
415
416 aChildIterator = openedStudy.NewChildIterator(father)
417
418 #while aChildIterator.More():
419 anSObject = aChildIterator.Value()
420 #print "iterate: ", anSObject.GetID()
421 res, anAttr=anSObject.FindAttribute("AttributeIOR")
422 if res :
423         anAttr=anAttr._narrow(SALOMEDS.AttributeIOR)
424         ior = anAttr.Value()
425         Graph=SuperV.getGraph(ior)
426         ListOfNodes=Graph.Nodes()
427         length_as= len(ListOfNodes)
428         print "ListOfNodes length = ", length_as
429         if length_as != length_bs:
430                 raise RuntimeErrror, "defferent length of nodes after study open"
431 #aChildIterator.Next()
432
433 Names = []
434 for node in ListOfNodes:
435         Names.append(node.Name())
436 names.sort()
437 Names.sort()
438 if names != Names :
439         raise RuntimeError, "List of dataflow nodes after save differs from one befor save operation"
440
441 #==================================================
442 #4. Save
443 #==================================================
444 batchmode_geompy.myStudyManager.Save(openedStudy)
445 batchmode_geompy.myStudyManager.Close(openedStudy)
446 #==================================================
447 #5. Open
448 #==================================================
449 try:
450         openedStudy = batchmode_geompy.myStudyManager.Open(str+"/test_dir/test_new.hdf")
451 except Exception:
452         raise  RuntimeError, "Can't open saved study!"
453
454
455
456 if openedStudy == None:
457         raise  RuntimeError, "Can't open saved study!"
458
459 father = openedStudy.FindComponent("GEOM")
460 if father is None:
461          raise  RuntimeError, "Geom component is not found!  Wrong study is opened." 
462
463
464 #find AttributeReal
465 res,A=father.FindAttribute("AttributeReal")
466 if res == 0 or A == None:
467         raise  RuntimeError, "Error:  not found AttributeReal"
468
469 A = A._narrow(SALOMEDS.AttributeReal)
470 if A.Value() != 0.0001:
471         raise  RuntimeError, "Error : wrong value of  AttributeReal"
472 #==================================================
473 #     find  box
474 #==================================================
475 box = openedStudy.FindObject("box")
476 if box is None :
477         raise  RuntimeError, "box was not found! Wrong study is opened."
478
479 edge=openedStudy.FindObject("face_edge")
480 if edge is None :
481         raise  RuntimeError, "face_edge was not found! Wrong study is opened."
482
483
484 father = openedStudy.FindComponent("MESH")
485 if father is None:
486          raise  RuntimeError, "MESH component is not found!  Wrong study is opened." 
487
488 hp=openedStudy.FindObject("Meshbox")
489 if hp is None:
490         raise  RuntimeError, "Meshbox object was not found! Wrong study is opened."
491
492 hp = openedStudy.FindObject("Algorithms Definition")
493 if hp is None:
494         raise  RuntimeError, "Algorithms Definition object was not found! Wrong study is opened."
495
496 hp = openedStudy.FindObject("MEFISTO_2D")
497 if hp is None:
498         raise  RuntimeError, "MEFISTO_2D object was not found! Wrong study is opened."
499
500 hp = openedStudy.FindObject("Hypothesis Definition")
501 if hp is None:
502         raise  RuntimeError, "Algoriths Definition was not found! Wrong study is opened."
503
504 #==================================================
505 # Find Supervisor
506 #==================================================
507 father = openedStudy.FindComponent("SUPERV")
508 if father is None:
509          raise  RuntimeError, "SUPERV component is not found!  Wrong study is opened." 
510
511 SuperV = lcc.FindOrLoadComponent("SuperVisionContainer","Supervision")
512 Builder = openedStudy.NewBuilder()
513 Builder.LoadWith(father, SuperV)
514 import SALOMEDS
515
516 aChildIterator = openedStudy.NewChildIterator(father)
517
518 #while aChildIterator.More():
519 anSObject = aChildIterator.Value()
520
521 res, anAttr=anSObject.FindAttribute("AttributeIOR")
522 if res :
523         anAttr=anAttr._narrow(SALOMEDS.AttributeIOR)
524         ior = anAttr.Value()
525         Graph=SuperV.getGraph(ior)
526         ListOfNodes=Graph.Nodes()
527         length_as= len(ListOfNodes)
528         print "ListOfNodes length = ", length_as
529         if length_as != length_bs:
530                 raise RuntimeErrror, "defferent length of nodes after study open"
531 #aChildIterator.Next()
532
533 Names = []
534 for node in ListOfNodes:
535         Names.append(node.Name())
536 names.sort()
537 Names.sort()
538 if names != Names :
539         raise RuntimeError, "List of dataflow nodes after save differs from one befor save operation"
540