Waveform cleaner

ctapipe/image/waveform_cleaning.py

@@ -13,7 +13,7 @@
                                              LocalPeakIntegrator)
 
 __all__ = ['WaveformCleanerFactory', 'CHECMWaveformCleanerAverage',
-           'CHECMWaveformCleanerLocal',
+           'CHECMWaveformCleanerLocal','BaselineWaveformCleaner',
            'NullWaveformCleaner']
 
 
@@ -67,6 +67,26 @@ def apply(self, waveforms):
         return waveforms
 
 
+class BaselineWaveformCleaner(WaveformCleaner):
+    """
+    Basic waveform cleaner that subtracts the waveform baseline as
+    estimated  in a window of "baseline_width" samples, shifted of
+    "window_shift" samples
+    """
+    baseline_width = Int(10, help='Define then number of samples for estimating the '
+                                  'baseline').tag(config=True)
+
+    window_shift = Int(0, help='Define the shift of the window on which calculate '
+                               'the baseline.').tag(config=True)
+
+    def apply(self, waveforms):
+        # Subtract initial baseline
+        baseline_sub = waveforms - np.mean(waveforms[:, :, self.window_shift:self.baseline_width],
+                                           axis=2)[:, :, None]
+
+        return baseline_sub
+
+
 class CHECMWaveformCleaner(WaveformCleaner):
     """
     Waveform cleaner used by CHEC-M.