diff options
Diffstat (limited to 'utils/sddm/plot_energy.py')
| -rw-r--r-- | utils/sddm/plot_energy.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/utils/sddm/plot_energy.py b/utils/sddm/plot_energy.py index 0ba4076..d9edf11 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): +def get_events(filenames, merge_fits=False, nhit_thresh=None, apply_nhit_trigger=True): 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) @@ -593,7 +593,8 @@ def get_events(filenames, merge_fits=False, nhit_thresh=None): ev = ev[ev.nhit_cal >= 100] # Require at least 1 NHIT trigger to fire - 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] + if apply_nhit_trigger: + 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 fits = fits.groupby(['run','gtid']).filter(lambda x: len(x) >= 10).reset_index() |