bec/ophyd_devices
0
0
Fork 0
mirror of https://github.com/bec-project/ophyd_devices.git synced 2025-06-24 03:38:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

refactor: cleanup

Browse Source
This commit is contained in:
appel_c
2024-02-26 13:59:49 +01:00
committed by wakonig_k
parent dc9634b739
commit 961041e072
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
ophyd_devices/sim/sim_data.py
View File

@ -422,7 +422,8 @@ class SimulatedDataMonitor(SimulatedDataBase):
if compute_readback: if compute_readback:
method = self._compute method = self._compute
value = self.execute_simulation_method(method=method, signal_name=signal_name) value = self.execute_simulation_method(method=method, signal_name=signal_name)
self.update_sim_state(signal_name, self.bit_depth(value)) value = self.bit_depth(value)
self.update_sim_state(signal_name, value)
def _compute(self, *args, **kwargs) -> int: def _compute(self, *args, **kwargs) -> int:
""" """
@ -454,7 +455,7 @@ class SimulatedDataMonitor(SimulatedDataBase):
v = np.random.poisson(v) v = np.random.poisson(v)
return v return v
elif noise == NoiseType.UNIFORM: elif noise == NoiseType.UNIFORM:
v += np.round(np.random.uniform(-1, 1) * noise_multiplier).astype(int) v += np.round(np.random.uniform(-1, 1) * noise_multiplier).astype(self.bit_depth)
return v return v
return v return v
@ -684,12 +685,10 @@ class SimulatedDataCamera(SimulatedDataBase):
noise (NoiseType): Type of noise to add. noise (NoiseType): Type of noise to add.
""" """
if noise == NoiseType.POISSON: if noise == NoiseType.POISSON:
v = np.random.poisson(np.round(v), v.shape).astype("uint16") v = np.random.poisson(np.round(v), v.shape)
return v return v
if noise == NoiseType.UNIFORM: if noise == NoiseType.UNIFORM:
v += np.round(np.random.uniform(-noise_multiplier, noise_multiplier, v.shape)).astype( v += np.random.uniform(-noise_multiplier, noise_multiplier, v.shape)
"uint16"
)
return v return v
if noise == NoiseType.NONE: if noise == NoiseType.NONE:
return v return v