Salome HOME
ea6674715aabd9a0892cdfd7b74ae61fb24e42ec
[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 os.environ["PARAVIS_TEST_PICS"] = sys.argv[1]
31 picturedir = get_picture_dir("bugs/B1")
32
33 # 1. Import MED file
34 med_file_path = datadir + "resultat.01.med"
35
36 print 'Importing "resultat.01.med"....',
37 my_paravis.ImportFile(med_file_path)
38 med_reader = pvsimple.GetActiveSource()
39
40 if med_reader is None:
41     raise RuntimeError, "resultat.01.med was not imported!!!"
42 else:
43     print "OK"
44
45 # 2. Creation of presentation of each group
46 groups_on_cells = get_group_names(med_reader, "MAILLAGE_01_001", EntityType.CELL, wo_nogroups=True)
47 groups_on_nodes = get_group_names(med_reader, "MAILLAGE_01_001", EntityType.NODE, wo_nogroups=True)
48
49 groups = groups_on_cells + groups_on_nodes
50
51 errors = 0
52 i = 0
53 for group in groups:
54     i += 1
55     shor_name = group.split('/')[-1]
56     print "group: ", shor_name
57     prs = MeshOnGroup(med_reader, group)
58     
59     if prs is None : 
60         print "FAILED!!! Created presentation is None!!!"
61         errors += 1
62     else :
63         print "Presentation was created!"
64         pic_path = os.path.join(picturedir, shor_name.strip().split("_")[0]+str(i)+"."+pictureext)
65         process_prs_for_test(prs, pvsimple.GetRenderView(), pic_path)                    
66         
67
68 # check errors
69 if errors == 1:
70     raise RuntimeError, "There is an error was occured... For more info see ERROR message above.."
71 elif errors > 1:
72     raise RuntimeError, "There are some errors were occured... For more info see ERRORs messages above.."
73