update cat-grid-jobs to take a directory as an argument instead of files

1 files changed, 3 insertions, 2 deletions

diff --git a/utils/cat-grid-jobs b/utils/cat-grid-jobs
index 83a8004..1f83bea 100755
--- a/utils/cat-grid-jobs
+++ b/utils/cat-grid-jobs
@@ -130,10 +130,11 @@ if __name__ == '__main__':
     import os
     import sys
     import h5py
+    import glob
 
     parser = argparse.ArgumentParser("concatenate fit results from grid jobs into a single file")
     parser.add_argument("zdab", help="zdab input file")
-    parser.add_argument("filenames", nargs='+', help="input files")
+    parser.add_argument("directory", help="directory with grid results")
     parser.add_argument("-o", "--output", type=str, help="output filename", required=True)
     args = parser.parse_args()
 
@@ -158,7 +159,7 @@ if __name__ == '__main__':
 
     with h5py.File(args.output,"a") as fout:
         total_events = fout['ev'].shape[0]
-        for filename in args.filenames:
+        for filename in glob.glob(join(args.directory,'*.hdf5')):
             with h5py.File(filename) as f:
                 # Check to see if the git sha1 match
                 if fout.attrs['git_sha1'] != f.attrs['git_sha1']: