mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2025-12-15 01:21:30 +01:00
80a2b02345
This PR adds support for decoding digital data from the my320 test chip. - Added BitOffset (strong type) - Expand 24 to 32 bit - Python bindings for decoding my302 - Improved docs
96 lines
2.7 KiB
Python
96 lines
2.7 KiB
Python
# SPDX-License-Identifier: MPL-2.0
|
|
import numpy as np
|
|
from . import _aare
|
|
|
|
|
|
class AdcSar04Transform64to16:
|
|
def __call__(self, data):
|
|
return _aare.adc_sar_04_decode64to16(data)
|
|
|
|
class AdcSar05Transform64to16:
|
|
def __call__(self, data):
|
|
return _aare.adc_sar_05_decode64to16(data)
|
|
|
|
class Moench05Transform:
|
|
#Could be moved to C++ without changing the interface
|
|
def __init__(self):
|
|
self.pixel_map = _aare.GenerateMoench05PixelMap()
|
|
|
|
def __call__(self, data):
|
|
return np.take(data.view(np.uint16), self.pixel_map)
|
|
|
|
|
|
class Moench05Transform1g:
|
|
#Could be moved to C++ without changing the interface
|
|
def __init__(self):
|
|
self.pixel_map = _aare.GenerateMoench05PixelMap1g()
|
|
|
|
def __call__(self, data):
|
|
return np.take(data.view(np.uint16), self.pixel_map)
|
|
|
|
|
|
class Moench05TransformOld:
|
|
#Could be moved to C++ without changing the interface
|
|
def __init__(self):
|
|
self.pixel_map = _aare.GenerateMoench05PixelMapOld()
|
|
|
|
def __call__(self, data):
|
|
return np.take(data.view(np.uint16), self.pixel_map)
|
|
|
|
|
|
class Matterhorn02Transform:
|
|
def __init__(self):
|
|
self.pixel_map = _aare.GenerateMH02FourCounterPixelMap()
|
|
|
|
def __call__(self, data):
|
|
counters = int(data.size / 48**2 / 2)
|
|
if counters == 1:
|
|
return np.take(data.view(np.uint16), self.pixel_map[0])
|
|
else:
|
|
return np.take(data.view(np.uint16), self.pixel_map[0:counters])
|
|
|
|
class Mythen302Transform:
|
|
"""
|
|
Transform Mythen 302 test chip data from a buffer of bytes (uint8_t)
|
|
to a uint32 numpy array of [64,3] representing channels and counters.
|
|
Assumes data taken with rx_dbitlist 17 6, rx_dbitreorder 1 and Digital
|
|
Samples = 2310 [(64x3x24)/2 + some extra]
|
|
|
|
.. note::
|
|
|
|
The offset is in number of bits 0-7
|
|
|
|
"""
|
|
_n_channels = 64
|
|
_n_counters = 3
|
|
|
|
def __init__(self, offset=4):
|
|
self.offset = offset
|
|
|
|
def __call__(self, data : np.ndarray):
|
|
"""
|
|
Transform buffer of data to a [64,3] np.ndarray of uint32.
|
|
|
|
Parameters
|
|
----------
|
|
data : np.ndarray
|
|
Expected dtype: uint8
|
|
|
|
Returns
|
|
----------
|
|
image : np.ndarray
|
|
uint32 array of size 64, 3
|
|
"""
|
|
res = _aare.decode_my302(data, self.offset)
|
|
res = res.reshape(
|
|
Mythen302Transform._n_channels, Mythen302Transform._n_counters
|
|
)
|
|
return res
|
|
|
|
#on import generate the pixel maps to avoid doing it every time
|
|
moench05 = Moench05Transform()
|
|
moench05_1g = Moench05Transform1g()
|
|
moench05_old = Moench05TransformOld()
|
|
matterhorn02 = Matterhorn02Transform()
|
|
adc_sar_04_64to16 = AdcSar04Transform64to16()
|
|
adc_sar_05_64to16 = AdcSar05Transform64to16() |