]> SALOME platform Git repositories - modules/paravis.git/blob - test/VisuPrs/3D_viewer/A4.py
Salome HOME
Copyright update: 2016
[modules/paravis.git] / test / VisuPrs / 3D_viewer / A4.py
1 # Copyright (C) 2010-2016  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, or (at your option) any later version.
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/3D_viewer/A4 case
21 # Create 3D Viewer and test set view properties for Iso Surface presentation
22 # Author: POLYANSKIKH VERA
23 from paravistest import *
24 from presentations import *
25 from pvsimple import *
26 import sys
27 import time
28
29 # Directory for saving snapshots
30 picturedir = get_picture_dir("3D_viewer/A4")
31
32 # Add path separator to the end of picture path if necessery
33 if not picturedir.endswith(os.sep):
34     picturedir += os.sep
35
36 # Get view
37 my_view = GetRenderView()
38 reset_view(my_view)
39 Render(my_view)
40
41 file_name = datadir + "hydro_sea_alv.med"
42 print " --------------------------------- "
43 print "file ", file_name
44 print " --------------------------------- "
45
46 OpenDataFile(file_name)
47 proxy = GetActiveSource()
48 if proxy is None:
49     raise RuntimeError("Error: can't import file.")
50 else:
51     print "OK"
52
53 represents = [RepresentationType.POINTS, RepresentationType.WIREFRAME,\
54 RepresentationType.SURFACE, RepresentationType.VOLUME]
55 shrinks = [0, 1]
56 shadings = ["Flat", "Gouraud"]
57 opacities = [1.0, 0.5, 0.0]
58 linewidths = [1.0, 3.0, 10.0]
59 compare_prec = 0.00001
60
61 field_name = 'Head'
62
63 print "\nCreating iso surface.......",
64 iso_surf = IsoSurfacesOnField(proxy, EntityType.CELL, field_name, 1)
65 if iso_surf is None:
66     raise RuntimeError("Error!!! Presentation wasn't created...")
67
68 display_only(iso_surf, my_view)
69 reset_view(my_view)
70 Render(my_view)
71
72 print "\nChange Presentation Parameters..."
73
74 for reprCode in represents:
75     repr = RepresentationType.get_name(reprCode)
76     if reprCode == RepresentationType.VOLUME:
77         is_good = call_and_check(iso_surf, "Representation", repr, 0)
78         if is_good:
79             msg = "VOLUME representation must be not available for Iso Surface"
80             raise RuntimeError(msg)
81     else:
82         call_and_check(iso_surf, "Representation", repr, 1)
83
84     for sha in shadings:
85         setShaded(my_view, sha)
86         call_and_check(iso_surf, "Interpolation", sha, 1)
87         Render(my_view)
88
89         for opa in opacities:
90             call_and_check(iso_surf, "Opacity", opa, 1, compare_prec)
91
92             for lwi in linewidths:
93                 call_and_check(iso_surf, "LineWidth", lwi, 1, compare_prec)
94                 time.sleep(1)
95
96                 # save picture in file
97                 # Construct image file name
98                 pic_name = picturedir + "params_" + repr.replace(' ', '_') + "_any_" +  \
99                     str(sha) + "_" + str(opa) + "_" + str(lwi) + "." + pictureext
100
101                 # Show and record the presentation
102                 process_prs_for_test(iso_surf, my_view, pic_name)