bec/ophyd_devices
1
0
Fork 0
Code Issues 9 Pull Requests 10 Actions Packages Projects Releases 126 Wiki Activity

refactor: cleanup and exclude ComplexConstantModel

Browse Source
This commit is contained in:
appel_c 2024-02-26 14:31:33 +01:00 committed by wakonig_k
parent 961041e072
commit 6eca704adc
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ophyd_devices/sim/sim_data.py
View File

@ -401,6 +401,7 @@ class SimulatedDataMonitor(SimulatedDataBase):
except TypeError: except TypeError:
is_model = False is_model = False
if is_model and name not in [ if is_model and name not in [
"ComplexConstantModel",
"Gaussian2dModel", "Gaussian2dModel",
"ExpressionModel", "ExpressionModel",
"Model", "Model",
@ -455,7 +456,8 @@ 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(self.bit_depth) noise = np.ceil(np.random.uniform(0, 1) * noise_multiplier).astype(int)
v += noise * (np.random.randint(0, 2) * 2 - 1)
return v return v
return v return v
@ -549,7 +551,7 @@ class SimulatedDataCamera(SimulatedDataBase):
) )
else: else:
value = self._compute_empty_image() value = self._compute_empty_image()
value = value.astype(self.bit_depth) value = self.bit_depth(value)
self.update_sim_state(signal_name, value) self.update_sim_state(signal_name, value)
def _compute_empty_image(self) -> np.ndarray: def _compute_empty_image(self) -> np.ndarray:
@ -570,7 +572,7 @@ class SimulatedDataCamera(SimulatedDataBase):
"""Compute a return value for SimulationType2D constant.""" """Compute a return value for SimulationType2D constant."""
try: try:
shape = self.parent.image_shape.get() shape = self.parent.image_shape.get()
v = self.sim_params.get("amplitude") * np.ones(shape, dtype=np.uint16) v = self.sim_params.get("amplitude") * np.ones(shape, dtype=np.float32)
v = self._add_noise(v, self.sim_params["noise"], self.sim_params["noise_multiplier"]) v = self._add_noise(v, self.sim_params["noise"], self.sim_params["noise_multiplier"])
return self._add_hot_pixel( return self._add_hot_pixel(
v, v,