1 # Copyright (C) 2010-2012 CEA/DEN, EDF R&D
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.
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.
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
17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 This module provides auxiliary classes, functions and variables for testing.
24 #from __future__ import print_function
28 from datetime import date
35 samples_dir = os.getenv("DATA_DIR")
38 if samples_dir is not None:
39 samples_dir = os.path.normpath(samples_dir)
40 datadir = samples_dir + "/MedFiles/"
41 tablesdir = samples_dir + "/Tables/"
43 # Graphica files extension
44 pictureext = os.getenv("PIC_EXT")
45 if pictureext == None:
50 class RepresentationType:
52 Types of representation.
62 _type2name = {OUTLINE: 'Outline',
64 WIREFRAME: 'Wireframe',
66 SURFACEEDGES: 'Surface With Edges',
68 POINTSPRITE: 'Point Sprite'}
71 def get_name(cls, type):
72 """Return paraview representation type by the primitive type."""
73 return cls._type2name[type]
76 class SalomeSession(object):
80 #sys.argv += ["--killall"]
81 #sys.argv += ["--portkill=" + port]
82 sys.argv += ["--show-desktop=1"]
83 sys.argv += ["--splash=0"]
84 sys.argv += ["--modules=MED,VISU,PARAVIS"]
85 clt, d = runSalome.main()
90 #VTN: workaround for crash on CentOS.6.3.64
92 #os.system('killSalomeWithPort.py {0}'.format(self.port))
93 #os.system('killSalomeWithPort.py ' + self.port)
94 #import killSalomeWithPort
95 #killSalomeWithPort.killMyPort(self.port)
100 # Auxiliary functions
101 def test_values(value, et_value, check_error=0):
105 et_length = len(et_value)
106 if (length != et_length):
107 err_msg = "ERROR!!! There is different number of created " + str(length) + " and etalon " + str(et_length) + " values!!!"
111 for i in range(et_length):
112 if abs(et_value[i]) > 1:
113 max_val = abs(0.001 * et_value[i])
114 if abs(et_value[i] - value[i]) > max_val:
115 err_msg = "ERROR!!! Got value " + str(value[i]) + " is not equal to etalon value " + str(ret_value[i]) + "!!!"
120 if abs(et_value[i] - value[i]) > max_val:
121 err_msg = "ERROR!!! Got value " + value[i] + " is not equal to etalon value " + et_value[i] + "!!!"
123 if check_error and error > 0:
124 err_msg = ("There is(are) some error(s) was(were) found... "
125 "For more info see ERRORs above...")
126 raise RuntimeError(err_msg)
130 def get_picture_dir(pic_dir, subdir):
135 # Add current date and subdirectory for the case to the directory path
136 cur_date = date.today().strftime("%d%m%Y")
137 res_dir += "/test_" + cur_date + "/" + subdir
138 # Create the directory if doesn't exist
139 res_dir = os.path.normpath(res_dir)
140 if not os.path.exists(res_dir):
143 # Clean the directory
144 for root, dirs, files in os.walk(res_dir):
146 os.remove(os.path.join(root, f))
151 def call_and_check(prs, property_name, value, do_raise=1, compare_toler=-1.0):
152 """Utility function for 3D viewer test for common check of different
153 types of presentation parameters set"""
155 prs.SetPropertyWithName(property_name, value)
157 error_string = (str(value) + "value of " + property_name + " is not available for this type of presentations")
160 is_good = (error_string is None)
163 raise RuntimeError(error_string)
167 # compare just set value and the one got from presentation
168 really_set_value = prs.GetPropertyValue(property_name)
170 if compare_toler > 0:
171 is_equal = (fabs(really_set_value - value) < compare_toler)
173 is_equal = (really_set_value == value)
175 msg = str(really_set_value) + " has been set instead"
177 raise RuntimeError(msg)
185 def setShaded(view, shading):
186 """Utility function to set shaded mode in view"""
188 view.LightDiffuseColor = [1, 1, 1]
190 view.LightDiffuseColor = [0, 0, 0]
194 salome_session = SalomeSession()
198 print "Creating new study...",
199 aStudy = salome.myStudyManager.NewStudy("Study1")
201 raise RuntimeError("FAILED")
205 salome.myStudy = aStudy