Added gen_exps.py script.

1 files changed, 98 insertions, 0 deletions

diff --git a/gen_exps.py b/gen_exps.py
index e69de29..e4e8187 100644..100755
--- a/gen_exps.py
+++ b/gen_exps.py
@@ -0,0 +1,98 @@
+#!/usr/bin/env python
+from __future__ import print_function
+
+import os
+import re
+import shutil as sh
+
+from gen.generators import GedfGenerator,PedfGenerator,CedfGenerator
+from optparse import OptionParser
+
+# There has to be a better way to do this...
+GENERATORS = {'C-EDF':CedfGenerator,
+              'P-EDF':PedfGenerator,
+              'G-EDF':GedfGenerator}
+
+def parse_args():
+    parser = OptionParser("usage: %prog [options] [files...] "
+                          "[generators...] [param=val[,val]...]")
+
+    parser.add_option('-o', '--out-dir', dest='out_dir',
+                      help='directory for data output',
+                      default=("%s/exps"%os.getcwd()))
+    parser.add_option('-f', '--force', action='store_true', default=False,
+                      dest='force', help='overwrite existing data')
+    parser.add_option('-l', '--list-generators', dest='list_gens',
+                      help='list allowed generators', action='store_true',
+                      default=False)
+    parser.add_option('-d', '--describe-generators', metavar='generator[,..]',
+                      dest='described', default=None,
+                      help='describe parameters for generator(s)')
+
+    return parser.parse_args()
+
+def load_file(fname):
+    with open(fname, 'r') as f:
+        data = f.read().strip()
+    try:
+        values = eval(data)
+        if 'generator' not in values:
+            raise ValueError()
+        generator = values['generator']
+        del values['generator']
+        return generator, values
+    except:
+           raise IOError("Invalid generation file: %s" % fname)
+
+def main():
+    opts, args = parse_args()
+
+    # Print generator information on the command line
+    if opts.list_gens:
+        print(", ".join(GENERATORS.keys()))
+    if opts.described != None:
+        for generator in opts.described.split(','):
+            if generator not in GENERATORS:
+                print("No generator '%s'" % generator)
+            else:
+                GENERATORS[generator]().print_help()
+    if opts.list_gens or opts.described:
+        return 0
+
+    params = filter(lambda x : re.match("\w+=\w+", x), args)
+
+    # Ensure some generator is loaded
+    args = list(set(args) - set(params))
+    #TODO: get every loaded plugin, try and use that generator
+    args = args or ['C-EDF', 'G-EDF', 'P-EDF']
+
+    # Split into files to load, named generators
+    files = filter(os.path.exists, args)
+    gen_list = list(set(args) - set(files))
+
+    # Parse all specified parameters to be applied to every experiment
+    global_params = dict(map(lambda x : tuple(x.split("=")), params))
+    for k, v in global_params.iteritems():
+        global_params[k] = v.split(',')
+
+    exp_sets  = map(load_file, files)
+    exp_sets += map(lambda x: (x, {}), gen_list)
+
+    if opts.force and os.path.exists(opts.out_dir):
+        sh.rmtree(opts.out_dir)
+    if not os.path.exists(opts.out_dir):
+        os.mkdir(opts.out_dir)
+
+    for gen_name, gen_params in exp_sets:
+        if gen_name not in GENERATORS:
+            raise ValueError("Invalid generator name: %s" % gen_name)
+
+        print("Creating experiments using %s generator..." % gen_name)
+
+        params = dict(gen_params.items() + global_params.items())
+        generator = GENERATORS[gen_name](params)
+
+        generator.create_exps(opts.out_dir, opts.force)
+
+if __name__ == '__main__':
+    main()