util: added --visualize-dry

benchpress/util.py

@@ -6,11 +6,11 @@
 import numpy as np
 import atexit
 
-
 gfx_handle = None
 visualization_param = None
 visualization_trace = {'org': [], 'zip': []}
 visualization_trace_fname = None  # When None, no tracing
+visualization_dry = False
 
 # Check whether the numpy module is overruled by Bohrium
 bh_is_loaded_as_np = np.__name__ == "bohrium"
@@ -97,7 +97,7 @@ def surface3d():
 
 
 def confirm_exit(msg="Hit Enter to exit..."):
-    if visualization_param is None:
+    if visualization_trace_fname is None and not visualization_dry:
         if sys.version_info[0] == 2:
             raw_input(msg)
         else:
@@ -108,16 +108,19 @@ def plot_surface_wrapper(*args):
     global visualization_trace
     from bohrium import visualization
 
-    if visualization_trace_fname is None:  # We don't show visualization when tracing
-        if visualization_param is None:
-            visualization.plot_surface(*args)
-        else:
-            visualization.plot_surface(*args, param=visualization_param)
+    if visualization_dry:  # We force the visualization process on a dry run
+        visualization.compressed_copy(args[0], param=visualization_param).copy2numpy()
     else:
-        org = args[0].copy2numpy()
-        compressed = visualization.compressed_copy(args[0], param=visualization_param).copy2numpy()
-        visualization_trace['org'].append(org)
-        visualization_trace['zip'].append(compressed)
+        if visualization_trace_fname is None:  # We don't show visualization when tracing
+            if visualization_param is None:
+                visualization.plot_surface(*args)
+            else:
+                visualization.plot_surface(*args, param=visualization_param)
+        else:
+            org = args[0].copy2numpy()
+            compressed = visualization.compressed_copy(args[0], param=visualization_param).copy2numpy()
+            visualization_trace['org'].append(org)
+            visualization_trace['zip'].append(compressed)
 
 
 try:
@@ -167,7 +170,7 @@ class Benchmark:
     """
 
     def __init__(self):
-        global visualization_param, visualization_trace_fname
+        global visualization_param, visualization_trace_fname, visualization_dry
 
         self.__elapsed = 0.0  # The quantity measured
         self.__script = sys.argv[0]  # The script being run
@@ -228,6 +231,11 @@ def __init__(self):
                        type=str,
                        help="Dump frames to files instead of showing them"
                        )
+        p.add_argument('--visualize-dry',
+                       default=False,
+                       action='store_true',
+                       help="Do the data process but don't show any visualization"
+                       )
         p.add_argument('--verbose',
                        default=False,
                        action='store_true',
@@ -285,6 +293,7 @@ def __init__(self):
         self.verbose = args.verbose
         visualization_param = args.visualize_param
         visualization_trace_fname = args.visualize_trace
+        visualization_dry = args.visualize_dry
 
     def start(self):
         flush()