From 64f3781588001cd5ebeb44e4067d78012d7f3151 Mon Sep 17 00:00:00 2001
From: Stefan Reck <sreck@km3net.de>
Date: Mon, 27 Jun 2022 09:10:20 +0000
Subject: [PATCH] add w2lists_def readout
---
src/km3io/tools.py | 19 +++++++++++++++++++
1 file changed, 19 insertions(+)
diff --git a/src/km3io/tools.py b/src/km3io/tools.py
index c2d8a82..30e32c7 100644
--- a/src/km3io/tools.py
+++ b/src/km3io/tools.py
@@ -5,6 +5,7 @@ import numpy as np
import awkward as ak
import uproot3
+import km3io.definitions
from km3io.definitions import reconstruction as krec
from km3io.definitions import trigger as ktrg
from km3io.definitions import fitparameters as kfit
@@ -538,6 +539,24 @@ def is_3dmuon(trigger_mask):
return is_bit_set(trigger_mask, ktrg.JTRIGGER3DMUON)
+def get_w2list_idx(f):
+ """
+ Get the correct w2list_idx for the given file, or None if there is none.
+
+ Parameters
+ ----------
+ f : km3io.OfflineReader
+ The file.
+
+ """
+ w2s_idx = {
+ "genhen": km3io.definitions.w2list_genhen_idx,
+ "gseagen": km3io.definitions.w2list_gseagen_idx,
+ }
+ sim_program = f.header.simul.program.lower()
+ return w2s_idx.get(sim_program)
+
+
def is_nanobeacon(trigger_mask):
"""Returns True if the trigger mask contains the nano-beacon flag.
--
GitLab