Add support for SNAX triggers

gwdetchar · Feb 12, 2025 · bb78a4b · bb78a4b
1 parent e972226
commit bb78a4b
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/hveto/triggers.py b/hveto/triggers.py
@@ -48,6 +48,7 @@
     "pycbc_live": "hdf5.pycbc_live",
     "kleinewelle": "ligolw",
     "dmt_omega": "ligolw",
+    "snax": "hdf5.snax",
 }
 
 DEFAULT_TRIGFIND_OPTIONS = {
@@ -66,6 +67,9 @@
     },
     ("omicron", "root.omicron"): {
         "ext": "root",
+    },
+    ("snax", "hdf5"): {
+        "ext": "h5",
     }
 }
 
@@ -89,6 +93,9 @@
         "loudest": True,
         "columns": ["end_time", "template_duration", "new_snr"],
     },
+    ("snax", "hdf5.snax"): {
+        "columns": ["time", "frequency", "snr"],
+    },
 }
 
 
@@ -248,6 +255,8 @@ def _format_params(channel, etg, fmt, trigfind_kwargs, read_kwargs):
     # custom params for ETGs
     if etg == "pycbc_live":
         read_kwargs.setdefault("ifo", channel.split(":", 1)[0])
+    if etg == "snax":
+        read_kwargs.setdefault("channels", channel)
 
     return trigfind_kwargs, read_kwargs