diff options
| -rwxr-xr-x | utils/submit-grid-jobs | 20 |
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) |