Filter mask

src/graphnet/data/dataconverter.py

@@ -107,6 +107,7 @@ def __init__(
         workers: int = 1,
         index_column: str = "event_no",
         icetray_verbose: int = 0,
+        I3filters: List[str] = [],
     ):
         """Construct DataConverter.
 
@@ -166,6 +167,7 @@ def __init__(
         self._sequential_batch_pattern = sequential_batch_pattern
         self._input_file_batch_pattern = input_file_batch_pattern
         self._workers = workers
+        self._I3filters = I3filters
 
         # Create I3Extractors
         self._extractors = I3ExtractorCollection(*extractors)
@@ -435,6 +437,8 @@ def _extract_data(self, fileset: FileSet) -> List[OrderedDict]:
                     continue
             if self._skip_frame(frame):
                 continue
+            if self._filter_mask(frame, self._I3filters):
+                continue
 
             # Try to extract data from I3Frame
             results = self._extractors(frame)
@@ -566,3 +570,27 @@ def _skip_frame(self, frame: "icetray.I3Frame") -> bool:
         if frame["I3EventHeader"].sub_event_stream == "NullSplit":
             return True
         return False
+
+    def _filter_mask(
+        self, frame: "icetray.I3Frame", I3filters: List[str]
+    ) -> bool:
+        """Check if specified condition(s) are met.
+
+        Args:
+            frame: I3Frame to check.
+            I3filters: List of I3Filters to check for pass.
+        """
+        if "FilterMask" not in frame:
+            self.warning_once(
+                "FilterMask not found in frame. Skipping filter checks."
+            )
+            return False
+        for filter in I3filters:
+            if filter not in frame["FilterMask"]:
+                self.warning_once(
+                    f"Filter {filter} not found in frame. Skipping."
+                )
+                continue
+            if frame["FilterMask"][filter].condition_passed is False:
+                return True
+        return False