Salome HOME
Merge branch 'V7_dev'
[modules/paravis.git] / test / VisuPrs / ScalarMap / G2.py
index 7ccf93115629f628fdbf1e20a2905936be217e8f..d8eda87473bc2e7061c4a742cde61361bf987ad2 100644 (file)
@@ -1,4 +1,4 @@
-# Copyright (C) 2010-2014  CEA/DEN, EDF R&D
+# Copyright (C) 2010-2016  CEA/DEN, EDF R&D
 #
 # This library is free software; you can redistribute it and/or
 # modify it under the terms of the GNU Lesser General Public
 
 import sys
 import os
-from paravistest import datadir, pictureext, get_picture_dir
-import paravis
-from pvsimple import GetActiveSource, GetRenderView, Render
+from paravistest import datadir, pictureext, get_picture_dir, get_png_picture_resolution
+from pvsimple import GetActiveSource, GetRenderView, Render, OpenDataFile
 from presentations import ScalarMapOnField, hide_all, EntityType, PrsTypeEnum,reset_view,process_prs_for_test
 
-
-# Create presentations
-myParavis = paravis.myParavis
-
 picturedir = get_picture_dir("ScalarMap/G2")
 
 theFileName = datadir +  "Bug829_resu_mode.med"
 print " --------------------------------- "
 print "file ", theFileName
 print " --------------------------------- "
-    
+
 """Build presentations of the given types for all fields of the given file."""
 #print "Import %s..." % theFileName.split('/')[-1],
-result = myParavis.ImportFile(theFileName)
+result = OpenDataFile(theFileName)
 proxy = GetActiveSource()
 if proxy is None:
-       raise RuntimeError, "Error: can't import file."
+        raise RuntimeError, "Error: can't import file."
 else: print "OK"
 # Get view
 aView = GetRenderView()
 
 
+import time
+
 aFieldEntity = EntityType.NODE
 aFieldName = "MODES___DEPL____________________"
 #create list to store picture files sizes
-sizes=[]
+sizesw=[]
+sizesh=[]
 #create Scalar Map presentations for 10 timestamps
 for i in range(1,11):
     hide_all(aView, True)
@@ -62,27 +60,38 @@ for i in range(1,11):
     #display only current scalar map
     aPrs.Visibility=1
     reset_view(aView)
-    Render(aView)    
-    
+    Render(aView)
+
     # Add path separator to the end of picture path if necessery
     if not picturedir.endswith(os.sep):
             picturedir += os.sep
     prs_type = PrsTypeEnum.SCALARMAP
-            
+
     # Get name of presentation type
-    prs_name = PrsTypeEnum.get_name(prs_type)    
+    prs_name = PrsTypeEnum.get_name(prs_type)
     f_prs_type = prs_name.replace(' ', '').upper()
     # Construct image file name
     pic_name = picturedir + aFieldName + "_" + str(i) + "_" + f_prs_type + "." + pictureext
-    
+
     # Show and record the presentation
     process_prs_for_test(aPrs, aView, pic_name)
-    sizes.append(os.path.getsize(pic_name))
+    (w,h) = get_png_picture_resolution(pic_name)
+    sizesw.append(w)
+    sizesh.append(h)
 
-# check sizes of pictures      
-if abs(max(sizes)-min(sizes)) > 0.01*max(sizes):
-    print "<b>WARNING!!! Pictures have different sizes!!!</b>"; 
+# check sizes of pictures: width
+if abs(max(sizesw)-min(sizesw)) > 0:
+    print "<b>ERROR!!! Pictures have different width !!!</b>";
     for i in range(1,11):
         picture_name = "time_stamp_"+str(i)+"."+pictureext
-        print "Picture: "+picture_name+"; size: "+str(sizes[i-1]) 
-    raise RuntimeError 
+        print "Picture: "+picture_name+"; width : "+str(sizesw[i-1])
+    raise RuntimeError
+
+# check sizes of pictures: height
+if abs(max(sizesh)-min(sizesh)) > 0:
+    print "<b>WARNING!!! Pictures have different height !!!</b>";
+    for i in range(1,11):
+        picture_name = "time_stamp_"+str(i)+"."+pictureext
+        print "Picture: "+picture_name+"; height : "+str(sizesh[i-1])
+    raise RuntimeError
+