diff --git a/km3io/daq.py b/km3io/daq.py index 807e59eb8ed06ec74a0bd7781b9d6aa31a19bbbc..9e58cecb95f7fc00916d09947b58b36c1fd744d2 100644 --- a/km3io/daq.py +++ b/km3io/daq.py @@ -150,6 +150,8 @@ class DAQTimeslices: TIMESLICE_FRAME_BASKET_CACHE_SIZE)) self._timeslices[stream.decode("ascii")] = (headers, superframes, hits_buffer) + setattr(self, stream.decode("ascii"), + DAQTimesliceStream(headers, superframes, hits_buffer)) def stream(self, stream, idx): ts = self._timeslices[stream] @@ -163,6 +165,21 @@ class DAQTimeslices: return str(self) +class DAQTimesliceStream: + def __init__(self, headers, superframes, hits_buffer): + # self.headers = headers.lazyarray( + # uproot.asjagged(uproot.astable( + # uproot.asdtype( + # np.dtype([('a', 'i4'), ('b', 'i4'), ('c', 'i4'), + # ('d', 'i4'), ('e', 'i4')]))), + # skipbytes=6), + # basketcache=uproot.cache.ThreadSafeArrayCache( + # TIMESLICE_FRAME_BASKET_CACHE_SIZE)) + self.headers = headers + self.superframes = superframes + self._hits_buffer = hits_buffer + + class DAQTimeslice: """A wrapper for a DAQ timeslice""" def __init__(self, header, superframe, hits_buffer, idx, stream):