update submit-grid-jobs to create a new directory

1 files changed, 19 insertions, 1 deletions

diff --git a/utils/submit-grid-jobs b/utils/submit-grid-jobs
index 8aef916..ac74867 100755
--- a/utils/submit-grid-jobs
+++ b/utils/submit-grid-jobs
@@ -17,7 +17,7 @@
 from __future__ import print_function, division
 import yaml
 import string
-from os.path import split, splitext, join
+from os.path import split, splitext, join, abspath
 import uuid
 from subprocess import check_call
 
@@ -103,6 +103,7 @@ def submit_job(filename, run, gtid, dir, dqxx_dir, min_nhit, max_particles):
 if __name__ == '__main__':
     import argparse
     from subprocess import check_output
+    import os
 
     parser = argparse.ArgumentParser("submit grid jobs")
     parser.add_argument("filenames", nargs='+', help="input files")
@@ -113,6 +114,13 @@ if __name__ == '__main__':
     parser.add_argument("--skip-second-event", action='store_true', help="only fit the first event after a MAST bank", default=False)
     args = parser.parse_args()
 
+    # get the current working directory
+    home_dir = os.getcwd()
+
+    # get absolute paths since we are going to create a new directory
+    args.dir = abspath(args.dir)
+    args.dqxx_dir = abspath(args.dqxx_dir)
+
     for filename in args.filenames:
         if args.skip_second_event:
             output = check_output([join(args.dir,"zdab-cat"),"--skip-second-event",filename])
@@ -120,6 +128,14 @@ if __name__ == '__main__':
             output = check_output([join(args.dir,"zdab-cat"),filename])
         data = yaml.load(output)
 
+        head, tail = split(filename)
+        root, ext = splitext(tail)
+
+        new_dir = "%s_%s" % (root,ID.hex)
+
+        os.mkdir(new_dir)
+        os.chdir(new_dir)
+
         for i, event in enumerate(data['data']):
             for ev in event['ev']:
                 run = ev['run']
@@ -128,3 +144,5 @@ if __name__ == '__main__':
 
                 if nhit >= args.min_nhit:
                     submit_job(filename, run, gtid, args.dir, args.dqxx_dir, args.min_nhit, args.max_particles)
+
+        os.chdir(home_dir)