1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
"""Script that creates a mock-run for the basler camera"""
from .base import DeviceBase
class BaslerCamera(DeviceBase):
"""
Basler Camera device
Based on example /gpfs/exfel/exp/SPB/201930/p900061/raw/r0055/RAW-R0055-DA01-S00000.h5
"""
def __init__(self, device_id, nsamples=None, sensor_size=None):
"""Create a dummy basler device that inherits from Device Base"""
self.sensor_size = sensor_size or (2058, 2456)
super(BaslerCamera, self).__init__(device_id, nsamples=nsamples)
self.output_channels = ('daqOutput/data',)
# Technically, only the part before the / is the output channel.
# But there is a structure associated with the part one level after that,
# and we don't know what else to call it.
self.instrument_keys = [
('image/bitsPerPixel', 'i4', ()),
('image/dimTypes', 'i4', (2,)),
('image/dims', 'u8', (2,)),
('image/encoding', 'i4', ()),
('image/pixels', 'u2', self.sensor_size),
('image/roiOffsets', 'u8', (2,)),
('image/binning', 'u8', (2,)),
('image/flipX', 'u1', ()),
('image/flipY', 'u1', ())
]
def write_instrument(self, f):
super().write_instrument(f)
# Add fixed metadata
for channel in self.output_channels:
image_grp = 'INSTRUMENT/{}:{}/image/'.format(self.device_id, channel)
f[image_grp]['bitsPerPixel'][:self.nsamples] = 16
f[image_grp]['dims'][:self.nsamples] = self.sensor_size
|
