add peaks ext and NaI

XAMS-nikhef · Oct 24, 2024 · 073ed08 · 073ed08
1 parent 8366293
commit 073ed08
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/amstrax/auto_processing_new/offline_processing.py b/amstrax/auto_processing_new/offline_processing.py
@@ -29,7 +29,7 @@ def parse_args():
     parser.add_argument(
         "--targets",
         nargs="+",
-        default=["peak_basics", "event_info"],
+        default=["peak_basics", "event_info", "event_area_per_channel"],
         help="List of targets to process (e.g., raw_records, peaks, events).",
     )
     parser.add_argument("--output_folder", type=str, default=None, help="Path to output folder.")

diff --git a/amstrax/auto_processing_new/process.py b/amstrax/auto_processing_new/process.py
@@ -112,6 +112,14 @@ def infer_special_modes(self, st):
             self.targets = ["raw_records", "records_led", "led_calibration"]
             log.info(f"Overriding targets to {self.targets}")
 
+
+        if "_nai" in self.run_doc.get('mode'):
+            # Add the NAI plugin to the context
+            # All the _ext plugins should be already in the context
+            log.info("Detected NaI run.")
+            log.info("Adding peak_basics_ext to list of targets to process.")
+            self.targets.append("peak_basics_ext")          
+
         return st
 
     def setup_production(self):