feat(sim-test-devices): Add test devices

Author: cappel89

ophyd_devices/sim/sim_test_devices.py

@@ -7,11 +7,13 @@
 from bec_lib.endpoints import MessageEndpoints
 from bec_lib.logger import bec_logger
 from ophyd import Component as Cpt
-from ophyd import Device, DeviceStatus, Kind, OphydObject, PositionerBase, Staged
+from ophyd import Device, DeviceStatus, Kind, OphydObject, PositionerBase, Signal, Staged
 
+from ophyd_devices.interfaces.base_classes.psi_pseudo_motor_base import PSIPseudoMotorBase
 from ophyd_devices.sim.sim_camera import SimCamera
 from ophyd_devices.sim.sim_positioner import SimPositioner
 from ophyd_devices.sim.sim_signals import SetableSignal
+from ophyd_devices.utils.bec_processed_signal import BECProcessedSignal, ProcessedSignalModel
 from ophyd_devices.utils.bec_signals import (
     AsyncSignal,
     DynamicSignal,
@@ -424,6 +426,49 @@ def complete_cam():
         return status
 
 
+class VirtualSlitCenter(PSIPseudoMotorBase):
+
+    left_edge = Cpt(SimPositioner, name="left_edge", kind=Kind.normal)
+    right_edge = Cpt(SimPositioner, name="right_edge", kind=Kind.normal)
+
+    def __init__(self, name, device_manager, **kwargs):
+        super().__init__(name, device_manager, **kwargs)
+        positioners = {"left_edge": self.left_edge, "right_edge": self.right_edge}
+        self.set_positioner_objects(positioners)
+
+    def forward_calculation(self, left_edge: Signal, right_edge: Signal) -> float:
+        return float((left_edge.get() + right_edge.get()) / 2)
+
+    def inverse_calculation(
+        self, position, left_edge: Signal, right_edge: Signal
+    ) -> dict[str, float]:
+        left_pos = left_edge.root.readback.get()
+        right_pos = right_edge.root.readback.get()
+        width = right_pos - left_pos
+        new_right_pos = position + width / 2
+        new_left_pos = position - width / 2
+        return {"left_edge": new_left_pos, "right_edge": new_right_pos}
+
+    def motors_are_moving(self, left_edge: Signal, right_edge: Signal) -> int:
+        left_moving = left_edge.get()
+        right_moving = right_edge.get()
+        return int(left_moving or right_moving)
+
+
+class BECPseudoSignal(BECProcessedSignal):
+
+    def __init__(self, name, signal_1: str, signal_2: str, device_manager=None, **kwargs):
+        model = ProcessedSignalModel(
+            devices={"signal_1": signal_1, "signal_2": signal_2},
+            compute_method=self.compute,
+            return_type=float,
+        )
+        super().__init__(name, model=model, device_manager=device_manager, **kwargs)
+
+    def compute(self, signal_1: Signal, signal_2: Signal) -> float:
+        return float(signal_1.get() + signal_2.get())
+
+
 if __name__ == "__main__":
     cam = SimCameraWithPSIComponents(name="cam")
     cam.read()