Merge pull request #2386 from TjarkMiener/r0r1_merger

R0 and R1 waveform merging

ctapipe/io/hdf5merger.py

@@ -37,6 +37,8 @@ class NodeType(enum.Enum):
     "/simulation/event/telescope/impact": NodeType.TEL_GROUP,
     "/simulation/event/telescope/images": NodeType.TEL_GROUP,
     "/simulation/event/telescope/parameters": NodeType.TEL_GROUP,
+    "/r0/event/telescope": NodeType.TEL_GROUP,
+    "/r1/event/telescope": NodeType.TEL_GROUP,
     "/dl1/event/subarray/trigger": NodeType.TABLE,
     "/dl1/event/telescope/trigger": NodeType.TABLE,
     "/dl1/event/telescope/images": NodeType.TEL_GROUP,
@@ -115,6 +117,16 @@ class HDF5Merger(Component):
         help="Whether to include parameters calculated on true images in merged output",
     ).tag(config=True)
 
+    r0_waveforms = traits.Bool(
+        True,
+        help="Whether to include r0 waveforms in merged output",
+    ).tag(config=True)
+
+    r1_waveforms = traits.Bool(
+        True,
+        help="Whether to include r1 waveforms in merged output",
+    ).tag(config=True)
+
     dl1_images = traits.Bool(
         True,
         help="Whether to include dl1 images in merged output",
@@ -296,6 +308,16 @@ def _append(self, other):
         ):
             self._append_table_group(other, other.root[key])
 
+        # R0
+        key = "/r0/event/telescope/"
+        if self.telescope_events and self.r0_waveforms and key in other.root:
+            self._append_table_group(other, other.root[key])
+
+        # R1
+        key = "/r1/event/telescope/"
+        if self.telescope_events and self.r1_waveforms and key in other.root:
+            self._append_table_group(other, other.root[key])
+
         # DL1
         key = "/dl1/event/subarray/trigger"
         if key in other.root:

ctapipe/tools/merge.py

@@ -95,6 +95,18 @@ class MergeTool(Tool):
             "Include telescope-wise data",
             "Exclude telescope-wise data",
         ),
+        **flag(
+            "r0-waveforms",
+            "HDF5Merger.r0_waveforms",
+            "Include r0 waveforms",
+            "Exclude r0 waveforms",
+        ),
+        **flag(
+            "r1-waveforms",
+            "HDF5Merger.r1_waveforms",
+            "Include r1 waveforms",
+            "Exclude r1 waveforms",
+        ),
         **flag(
             "dl1-images",
             "HDF5Merger.dl1_images",

docs/changes/2386.feature.rst

@@ -0,0 +1 @@
+Add support for including r1 and r0 waveforms in the ``ctapipe-merge`` tool.