Add a python script for running multiple tests with varying parameters.

1 files changed, 52 insertions, 0 deletions

diff --git a/contrib/run_template.py b/contrib/run_template.py
new file mode 100755
index 0000000..50c0484
--- /dev/null
+++ b/contrib/run_template.py
@@ -0,0 +1,52 @@
+#!/usr/bin/env python
+
+import multiprocessing
+import os
+import re
+import subprocess
+import sys
+
+PREFIX = '#TMPLT'
+
+if len(sys.argv) < 3:
+    sys.stderr.write('Usage: %s binary <templatized parfile> [number of jobs]\n' % sys.argv[0])
+    sys.stderr.write('A templatized parfile starts with a line in the following format:\n')
+    sys.stderr.write(PREFIX + ' <varname> value1 value2 ...\n')
+    sys.stderr.write('If number of jobs is specified, this number of processes will be executed in parallel.\n')
+    sys.exit(0)
+
+if len(sys.argv) >= 4:
+    nb_jobs = int(sys.argv[3])
+    if nb_jobs <= 0:
+        nb_jobs = multiprocessing.cpu_count()
+else:
+    nb_jobs = 1
+
+with open(sys.argv[2], 'r') as f:
+    line = f.readline()
+    if not line.startswith(PREFIX):
+        sys.stderr.write('The file does not contain a template line.\n')
+        sys.exit(1)
+
+    vals = line[len(PREFIX):].split()
+    var  = vals.pop(0)
+    vals = map(float, vals)
+
+    par = f.read()
+
+    def run_task(val):
+        parfile_name = '%s.par' % val
+        with open(parfile_name, 'w') as parfile:
+            par_expanded = re.sub(r'\$\{(.*?)\}', lambda match: str(eval(match.group(1), {var : val})), par)
+            parfile.write(par_expanded)
+            parfile.flush()
+            os.fsync(parfile.fileno())
+
+        subprocess.call([sys.argv[1], '--', parfile_name])
+        os.unlink(parfile_name)
+
+    if nb_jobs > 1:
+        pool = multiprocessing.Pool(nb_jobs)
+        pool.map(run_task, vals)
+    else:
+        map(run_task, vals)