add a script to concatenate output from grid jobs

1 files changed, 5 insertions, 1 deletions

diff --git a/utils/submit-grid-jobs b/utils/submit-grid-jobs
index 18c6e37..8aef916 100755
--- a/utils/submit-grid-jobs
+++ b/utils/submit-grid-jobs
@@ -110,10 +110,14 @@ if __name__ == '__main__':
     parser.add_argument("--dqxx-dir", type=str, help="dqxx directory", required=True)
     parser.add_argument("--min-nhit", type=int, help="minimum nhit to fit an event", default=100)
     parser.add_argument("--max-particles", type=int, help="maximum number of particles to fit for", default=3)
+    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()
 
     for filename in args.filenames:
-        output = check_output([join(args.dir,"zdab-cat"),filename])
+        if args.skip_second_event:
+            output = check_output([join(args.dir,"zdab-cat"),"--skip-second-event",filename])
+        else:
+            output = check_output([join(args.dir,"zdab-cat"),filename])
         data = yaml.load(output)
 
         for i, event in enumerate(data['data']):