Salome HOME
0022071: EDF 2497 PARAVIS : Porting of VISU tests for PARAVIS
[modules/paravis.git] / test / VisuPrs / bugs / B1.py
1 # Copyright (C) 2010-2013  CEA/DEN, EDF R&D
2 #
3 # This library is free software; you can redistribute it and/or
4 # modify it under the terms of the GNU Lesser General Public
5 # License as published by the Free Software Foundation; either
6 # version 2.1 of the License.
7 #
8 # This library is distributed in the hope that it will be useful,
9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11 # Lesser General Public License for more details.
12 #
13 # You should have received a copy of the GNU Lesser General Public
14 # License along with this library; if not, write to the Free Software
15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
16 #
17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 #
19
20 # This case corresponds to: /visu/bugs1/B1 case
21
22
23 import sys
24 from paravistest import datadir, pictureext, get_picture_dir
25 from presentations import *
26 import paravis
27 import pvsimple
28
29 my_paravis = paravis.myParavis
30 picturedir = get_picture_dir(sys.argv[1], "bugs/B1")
31
32 # 1. Import MED file
33 med_file_path = datadir + "resultat.01.med"
34
35 print 'Importing "resultat.01.med"....',
36 my_paravis.ImportFile(med_file_path)
37 med_reader = pvsimple.GetActiveSource()
38
39 if med_reader is None:
40     raise RuntimeError, "resultat.01.med was not imported!!!"
41 else:
42     print "OK"
43
44 # 2. Creation of presentation of each group
45 groups_on_cells = get_group_names(med_reader, "MAILLAGE_01_001", EntityType.CELL, wo_nogroups=True)
46 groups_on_nodes = get_group_names(med_reader, "MAILLAGE_01_001", EntityType.NODE, wo_nogroups=True)
47
48 groups = groups_on_cells + groups_on_nodes
49
50 errors = 0
51 i = 0
52 for group in groups:
53     i += 1
54     shor_name = group.split('/')[-1]
55     print "group: ", shor_name
56     prs = MeshOnGroup(med_reader, group)
57     
58     if prs is None : 
59         print "FAILED!!! Created presentation is None!!!"
60         errors += 1
61     else :
62         print "Presentation was created!"
63         pic_path = os.path.join(picturedir, shor_name.strip().split("_")[0]+str(i)+"."+pictureext)
64         process_prs_for_test(prs, pvsimple.GetRenderView(), pic_path)                    
65         
66
67 # check errors
68 if errors == 1:
69     raise RuntimeError, "There is an error was occured... For more info see ERROR message above.."
70 elif errors > 1:
71     raise RuntimeError, "There are some errors were occured... For more info see ERRORs messages above.."
72