Allow specifying the name of the output `.pkl` file on the command line.

Change-Id: I837dc25d6c24bf7430765a62374213b57a205806

1 files changed, 3 insertions, 1 deletions

diff --git a/benchmarking/java-ubenchs/run.py b/benchmarking/java-ubenchs/run.py
index 53dbdde..4e61d6f 100755
--- a/benchmarking/java-ubenchs/run.py
+++ b/benchmarking/java-ubenchs/run.py
@@ -81,6 +81,8 @@ def BuildOptions():
                         target.''')
     parser.add_argument('-f', '--filter', action = 'store', default = '*',
                         help='Quoted (benchmark name) filter pattern.')
+    parser.add_argument('--output-pkl', action = 'store',
+                        help='Specify a name for the output `.pkl` file.')
     return parser.parse_args()
 
 
@@ -231,7 +233,7 @@ if __name__ == "__main__":
     utils.PrintStats(result, iterations = args.iterations)
     print('')
     # Write the results to a file so they can later be used with `compare.py`.
-    res_filename = 'res.' + time.strftime("%Y.%m.%d-%H:%M:%S") + '.pkl'
+    res_filename = args.output_pkl if args.output_pkl is not None else 'res.' + time.strftime("%Y.%m.%d-%H:%M:%S") + '.pkl'
     with open(res_filename, 'wb') as pickle_file:
         pickle.dump(result, pickle_file)
         print(('Wrote results to %s.' % res_filename))