From 561ec6c2902364a21bc65d72c304926ab8efc7e8 Mon Sep 17 00:00:00 2001
From: Tamas Gal <himself@tamasgal.com>
Date: Thu, 4 Nov 2021 21:21:35 +0100
Subject: [PATCH] Make is_bit_set numba compatible
---
km3io/tools.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/km3io/tools.py b/km3io/tools.py
index db7f530..c684219 100644
--- a/km3io/tools.py
+++ b/km3io/tools.py
@@ -493,15 +493,16 @@ def usr(objects, field):
return objects.usr[:, idx]
+@nb.vectorize(("boolean(int64, int64)", "boolean(uint64, int64)"), nopython=True)
def is_bit_set(value, bit_position):
"""Returns true if a bit at the given position is 1.
- value: int or array(int)
+ value: int or array([u]int64)
The value to check, can be a single value or an array of values.
bit_position: int
0 for the first position, 1 for the second etc.
"""
- return (np.array(value) & (1 << bit_position)).astype(bool)
+ return bool(value & (1 << bit_position))
def is_3dshower(trigger_mask):
--
GitLab