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()
91 #os.system('killSalomeWithPort.py {0}'.format(self.port))
92 #os.system('killSalomeWithPort.py ' + self.port)
93 import killSalomeWithPort
94 killSalomeWithPort.killMyPort(self.port)
100 def test_values(value, et_value, check_error=0):
104 et_length = len(et_value)
105 if (length != et_length):
106 err_msg = "ERROR!!! There is different number of created " + str(length) + " and etalon " + str(et_length) + " values!!!"
110 for i in range(et_length):
111 if abs(et_value[i]) > 1:
112 max_val = abs(0.001 * et_value[i])
113 if abs(et_value[i] - value[i]) > max_val:
114 err_msg = "ERROR!!! Got value " + str(value[i]) + " is not equal to etalon value " + str(ret_value[i]) + "!!!"
119 if abs(et_value[i] - value[i]) > max_val:
120 err_msg = "ERROR!!! Got value " + value[i] + " is not equal to etalon value " + et_value[i] + "!!!"
122 if check_error and error > 0:
123 err_msg = ("There is(are) some error(s) was(were) found... "
124 "For more info see ERRORs above...")
125 raise RuntimeError(err_msg)
129 def get_picture_dir(pic_dir, subdir):
134 # Add current date and subdirectory for the case to the directory path
135 cur_date = date.today().strftime("%d%m%Y")
136 res_dir += "/test_" + cur_date + "/" + subdir
137 # Create the directory if doesn't exist
138 res_dir = os.path.normpath(res_dir)
139 if not os.path.exists(res_dir):
142 # Clean the directory
143 for root, dirs, files in os.walk(res_dir):
145 os.remove(os.path.join(root, f))
150 def call_and_check(prs, property_name, value, do_raise=1, compare_toler=-1.0):
151 """Utility function for 3D viewer test for common check of different
152 types of presentation parameters set"""
154 prs.SetPropertyWithName(property_name, value)
156 error_string = (str(value) + "value of " + property_name + " is not available for this type of presentations")
159 is_good = (error_string is None)
162 raise RuntimeError(error_string)
166 # compare just set value and the one got from presentation
167 really_set_value = prs.GetPropertyValue(property_name)
169 if compare_toler > 0:
170 is_equal = (fabs(really_set_value - value) < compare_toler)
172 is_equal = (really_set_value == value)
174 msg = str(really_set_value) + " has been set instead"
176 raise RuntimeError(msg)
184 def setShaded(view, shading):
185 """Utility function to set shaded mode in view"""
187 view.LightDiffuseColor = [1, 1, 1]
189 view.LightDiffuseColor = [0, 0, 0]
193 salome_session = SalomeSession()
197 print "Creating new study...",
198 aStudy = salome.myStudyManager.NewStudy("Study1")
200 raise RuntimeError("FAILED")
204 salome.myStudy = aStudy