Skip to content
Snippets Groups Projects
Commit f4f48a33 authored by Tamas Gal's avatar Tamas Gal :speech_balloon:
Browse files

Merge branch 'master' into 'add_w2lists_def_readout'

# Conflicts:
#   src/km3io/tools.py
parents 1e203885 b1985a63
No related branches found
No related tags found
1 merge request!72add w2lists_def readout
Pipeline #29106 passed with warnings
......@@ -556,3 +556,13 @@ def get_w2list_idx(f):
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.
Parameters
----------
trigger_mask : int or array(int)
A value or an array of the trigger_mask, either of an event, or a hit.
"""
return is_bit_set(trigger_mask, ktrg.JTRIGGERNB)
......@@ -31,6 +31,7 @@ from km3io.tools import (
is_3dshower,
is_mxshower,
is_3dmuon,
is_nanobeacon,
)
OFFLINE_FILE = OfflineReader(data_path("offline/km3net_offline.root"))
......@@ -642,3 +643,13 @@ class TestTriggerMaskChecks(unittest.TestCase):
[False, False, False, True, False, False, True, False, True, True],
list(is_3dmuon(GENHEN_OFFLINE_FILE.events.trigger_mask)),
)
def test_is_nanobeacon(self):
assert np.allclose(
[False, False, False, False, False, False, False, False, False, False],
list(is_nanobeacon(OFFLINE_FILE.events.trigger_mask)),
)
assert np.allclose(
[False, False, False, False, False, False, False, False, False, False],
list(is_nanobeacon(GENHEN_OFFLINE_FILE.events.trigger_mask)),
)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment