From 0a3aad7404826821f7674f85f11ceece3f62700a Mon Sep 17 00:00:00 2001
From: Tamas Gal <tgal@km3net.de>
Date: Tue, 8 Dec 2020 19:51:08 +0100
Subject: [PATCH] Add alternative masking
---
km3io/tools.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/km3io/tools.py b/km3io/tools.py
index 9d5e9af..cf903fe 100644
--- a/km3io/tools.py
+++ b/km3io/tools.py
@@ -323,6 +323,12 @@ def mask(arr, sequence=None, startend=None, minmax=None, atleast=None):
return builder.snapshot()
+def mask_alt(arr, start, end):
+ nonempty = ak.num(arr, axis=-1) > 0
+ mask = (arr.mask[nonempty][..., 0] == start) & (arr.mask[nonempty][..., -1])
+ return ak.fill_none(mask, False)
+
+
@nb.njit
def _mask3d(arr, builder, sequence=None, startend=None, minmax=None, atleast=None):
for subarray in arr:
--
GitLab