Skip drs4 corrections already applied by EVB

Author: maxnoe

src/ctapipe_io_lst/calibration.py

@@ -18,6 +18,7 @@
 
 from .compat import CTAPIPE_GE_0_21
 from .containers import LSTArrayEventContainer
+from .evb_preprocessing import EVBPreprocessingFlag
 
 
 from .constants import (
@@ -225,6 +226,9 @@ def __init__(self, subarray, config=None, parent=None, **kwargs):
     def apply_drs4_corrections(self, event: LSTArrayEventContainer):
 
         for tel_id in event.trigger.tels_with_trigger:
+            tdp_action = event.lst.tel[tel_id].evt.tdp_action
+            preprocessing = EVBPreprocessingFlag(tdp_action or 0)
+
             r1 = event.r1.tel[tel_id]
             # If r1 was not yet filled, copy of r0 converted
             if r1.waveform is None:
@@ -244,21 +248,20 @@ def apply_drs4_corrections(self, event: LSTArrayEventContainer):
             r1.waveform = r1.waveform.astype(np.float32, copy=False)
 
             # apply drs4 corrections
-            if self.apply_drs4_pedestal_correction:
+            if self.apply_drs4_pedestal_correction and EVBPreprocessingFlag.BASELINE_SUBTRACTION not in preprocessing:
                 self.subtract_pedestal(event, tel_id)
 
-            if self.apply_timelapse_correction:
+            if self.apply_timelapse_correction and EVBPreprocessingFlag.DELTA_T_CORRECTION not in preprocessing:
                 self.time_lapse_corr(event, tel_id)
             else:
                 self.update_last_readout_times(event, tel_id)
 
-            if self.apply_spike_correction:
+            if self.apply_spike_correction and EVBPreprocessingFlag.SPIKE_REMOVAL not in preprocessing:
                 if self.spike_correction_method == 'subtraction':
                     self.subtract_spikes(event, tel_id)
                 else:
                     self.interpolate_spikes(event, tel_id)
 
-
             # remove samples at beginning / end of waveform
             start = self.r1_sample_start.tel[tel_id]
             end = self.r1_sample_end.tel[tel_id]