diff options
Diffstat (limited to 'utils/sddm/plot_energy.py')
| -rw-r--r-- | utils/sddm/plot_energy.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/utils/sddm/plot_energy.py b/utils/sddm/plot_energy.py index bba0e75..be13d78 100644 --- a/utils/sddm/plot_energy.py +++ b/utils/sddm/plot_energy.py @@ -441,7 +441,7 @@ def michel_spectrum(T): y *= 2*MUON_MASS return y -def get_events(filenames, merge_fits=False, nhit_thresh=None, apply_nhit_trigger=True): +def get_events(filenames, merge_fits=False, nhit_thresh=None, mc=False): ev = pd.concat([read_hdf(filename, "ev").assign(filename=filename) for filename in filenames],ignore_index=True) fits = pd.concat([read_hdf(filename, "fits") for filename in filenames],ignore_index=True) rhdr = pd.concat([read_hdf(filename, "rhdr") for filename in filenames],ignore_index=True) @@ -576,7 +576,11 @@ def get_events(filenames, merge_fits=False, nhit_thresh=None, apply_nhit_trigger ev = ev.groupby('run',group_keys=False).apply(breakdown_follower_cut) # retrigger cut - ev = ev.groupby('run',group_keys=False).apply(retrigger_cut) + if not mc: + # We don't apply the retrigger cut to MC because it requires that the + # previous GTID be in the run, but quite a few of the MC events failed + # to simulate + ev = ev.groupby('run',group_keys=False).apply(retrigger_cut) # Label instrumentals ev['noise'] = ev.dc & (DC_JUNK | DC_CRATE_ISOTROPY | DC_QVNHIT | DC_ITC | DC_ESUM) != 0 @@ -604,7 +608,7 @@ def get_events(filenames, merge_fits=False, nhit_thresh=None, apply_nhit_trigger ev = ev[ev.nhit_cal >= 100] # Require at least 1 NHIT trigger to fire - if apply_nhit_trigger: + if not mc: ev = ev[(ev.trg_type & (TRIG_NHIT_100_LO | TRIG_NHIT_100_MED | TRIG_NHIT_100_HI | TRIG_NHIT_20 | TRIG_NHIT_20_LB)) != 0] # Require all 10 fits |