Add window reducer to process tool

Author: maxnoe

src/ctapipe/tools/process.py

@@ -12,6 +12,7 @@
 from ..image import ImageCleaner, ImageModifier, ImageProcessor
 from ..image.extractor import ImageExtractor
 from ..image.muon import MuonProcessor
+from ..image.reducer import ReadoutWindowReducer
 from ..instrument import SoftwareTrigger
 from ..io import (
     DataLevel,
@@ -75,6 +76,11 @@ class ProcessorTool(Tool):
         default_value=False,
     ).tag(config=True)
 
+    reduce_readout_window = Bool(
+        default_value=False,
+        help="If True, use ReadoutWindowReducer on waveforms.",
+    ).tag(config=True)
+
     aliases = {
         ("i", "input"): "EventSource.input_url",
         ("o", "output"): "DataWriter.output_path",
@@ -152,6 +158,7 @@ class ProcessorTool(Tool):
             metadata.Instrument,
             metadata.Contact,
             SoftwareTrigger,
+            ReadoutWindowReducer,
         ]
         + classes_with_traits(EventSource)
         + classes_with_traits(ImageCleaner)
@@ -179,6 +186,15 @@ def setup(self):
             sys.exit(1)
 
         subarray = self.event_source.subarray
+
+        if self.reduce_readout_window:
+            self.readout_window_reducer = ReadoutWindowReducer(
+                subarray=subarray, parent=self
+            )
+            subarray = self.readout_window_reducer.subarray
+        else:
+            self.readout_window_reducer = None
+
         self.software_trigger = SoftwareTrigger(parent=self, subarray=subarray)
         self.calibrate = CameraCalibrator(parent=self, subarray=subarray)
         self.process_images = ImageProcessor(subarray=subarray, parent=self)
@@ -300,6 +316,9 @@ def start(self):
                 )
                 continue
 
+            if self.readout_window_reducer is not None:
+                self.readout_window_reducer(event)
+
             if self.should_calibrate:
                 self.calibrate(event)
 

src/ctapipe/tools/tests/test_process.py

@@ -595,3 +595,46 @@ def test_prod6_issues(tmp_path):
         images = loader.read_telescope_events([32], true_images=True)
         images.add_index("event_id")
         np.testing.assert_array_equal(images.loc[1664106]["true_image"], -1)
+
+
+def test_readout_window_reducer(tmp_path, provenance):
+    """Test ReadoutWindowReducer in process tool."""
+    output = tmp_path / "test_reduced_window.dl1.h5"
+
+    config = {
+        "ProcessorTool": {
+            "reduce_readout_window": True,
+            "ReadoutWindowReducer": {
+                "window_start": [
+                    ("type", "*", None),
+                    ("type", "LST*", 10),
+                ],
+                "window_end": [
+                    ("type", "*", None),
+                    ("type", "LST*", 20),
+                ],
+            },
+            "DataWriter": {
+                "write_r1_waveforms": True,
+                "write_dl1_images": True,
+                "write_dl1_parameters": False,
+            },
+        }
+    }
+    config_path = tmp_path / "config.json"
+    config_path.write_text(json.dumps(config))
+
+    provenance_log = tmp_path / "provenance.log"
+    input_path = get_dataset_path("gamma_prod5.simtel.zst")
+    run_tool(
+        ProcessorTool(),
+        argv=[
+            f"--config={config_path}",
+            f"--input={input_path}",
+            f"--output={output}",
+            f"--provenance-log={provenance_log}",
+            "--overwrite",
+        ],
+        cwd=tmp_path,
+        raises=True,
+    )