mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2025-06-15 00:37:13 +02:00
28 lines
815 B
Python
28 lines
815 B
Python
import numpy as np
|
|
from . import _aare
|
|
|
|
|
|
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 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])
|
|
|
|
|
|
#on import generate the pixel maps to avoid doing it every time
|
|
moench05 = Moench05Transform()
|
|
matterhorn02 = Matterhorn02Transform() |