1 files changed, 8 insertions, 1 deletions
diff --git a/utils/dc b/utils/dc
index 6c7d86a..40ae13e 100755
--- a/utils/dc
+++ b/utils/dc
@@ -273,13 +273,20 @@ if __name__ == '__main__':
     parser.add_argument("--steps", type=int, default=100000, help="number of steps in the MCMC chain")
     parser.add_argument("--save", action="store_true", default=False, help="save plots")
     parser.add_argument("--mc", nargs='+', required=True, help="atmospheric MC files")
+    parser.add_argument("--nhit-thresh", type=int, default=None, help="nhit threshold to apply to events before processing (should only be used for testing to speed things up)")
     args = parser.parse_args()
 
     setup_matplotlib(args.save)
 
     import matplotlib.pyplot as plt
 
-    ev = get_events(args.filenames,merge_fits=True)
+    # Loop over runs to prevent using too much memory
+    evs = []
+    rhdr = pd.concat([read_hdf(filename, "rhdr").assign(filename=filename) for filename in args.filenames],ignore_index=True)
+    for run, df in rhdr.groupby('run'):
+        evs.append(get_events(df.filename.values, merge_fits=True, nhit_thresh=args.nhit_thresh))
+    ev = pd.concat(evs)
+
     ev = ev[ev.prompt]
     ev = ev[ev.nhit_cal > 100]