add --run-list argument to chi2 and dm-search

This commit updates the chi2 and dm-search scripts to add the ability to
pass a run list on the comand line.

2 files changed, 10 insertions, 0 deletions

diff --git a/utils/chi2 b/utils/chi2
index 39db081..5d56f17 100755
--- a/utils/chi2
+++ b/utils/chi2
@@ -431,6 +431,7 @@ if __name__ == '__main__':
     parser.add_argument("--print-nll", action='store_true', default=False, help="print nll values")
     parser.add_argument("--walkers", type=int, default=100, help="number of walkers")
     parser.add_argument("--thin", type=int, default=10, help="number of steps to thin")
+    parser.add_argument("--run-list", default=None, help="run list")
     args = parser.parse_args()
 
     setup_matplotlib(args.save)
@@ -444,6 +445,10 @@ if __name__ == '__main__':
         evs.append(get_events(df.filename.values, merge_fits=True, nhit_thresh=args.nhit_thresh))
     ev = pd.concat(evs)
 
+    if args.run_list is not None:
+        runs = np.genfromtxt(args.run_list)
+        ev = ev[ev.run.isin(runs)]
+
     ev = correct_energy_bias(ev)
 
     # Note: We loop over the MC filenames here instead of just passing the
diff --git a/utils/dm-search b/utils/dm-search
index 17f107a..6c78d4e 100755
--- a/utils/dm-search
+++ b/utils/dm-search
@@ -481,6 +481,7 @@ if __name__ == '__main__':
     parser.add_argument("--walkers", type=int, default=100, help="number of walkers")
     parser.add_argument("--thin", type=int, default=10, help="number of steps to thin")
     parser.add_argument("--test", type=int, default=0, help="run tests to check discovery threshold")
+    parser.add_argument("--run-list", default=None, help="run list")
     args = parser.parse_args()
 
     setup_matplotlib(args.save)
@@ -494,6 +495,10 @@ if __name__ == '__main__':
         evs.append(get_events(df.filename.values, merge_fits=True, nhit_thresh=args.nhit_thresh))
     ev = pd.concat(evs)
 
+    if args.run_list is not None:
+        runs = np.genfromtxt(args.run_list)
+        ev = ev[ev.run.isin(runs)]
+
     livetime = 0.0
     livetime_pulse_gt = 0.0
     for ev in evs: