bec/ophyd_devices
0
0
Fork 0
mirror of https://github.com/bec-project/ophyd_devices.git synced 2026-02-06 23:28:41 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: improvements from review

Browse Source
This commit is contained in:
perl_d
2025-11-26 09:40:36 +01:00
committed by David Perl
parent 288096b8ff
commit ad1b042f2e
1 changed files with 9 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
ophyd_devices/devices/optics_shutter.py
View File

@@ -4,6 +4,7 @@ from ophyd import Component as Cpt
from ophyd import EpicsSignal, EpicsSignalRO, Kind
from ophyd.status import SubscriptionStatus
from ophyd_devices import CompareStatus
from ophyd_devices.interfaces.base_classes.psi_device_base import PSIDeviceBase
@@ -17,13 +18,6 @@ class ShutterEnabled(IntEnum):
DISABLED = 0
def _is_state(state: int | str):
def _cb(*, old_value, value, **kwargs):
return value == state
return _cb
class Shutter(PSIDeviceBase):
"""A generic optics shutter device, for IOCs with the format '[BEAMLINE]-EH1-PSYS:SH-[A/B]-'
@@ -47,11 +41,13 @@ class Shutter(PSIDeviceBase):
"""
is_open = Cpt(EpicsSignalRO, "OPEN", kind=Kind.hinted)
USER_ACCESS = ["open", "close", "status", "enabled"]
is_open = Cpt(EpicsSignalRO, "OPEN", kind=Kind.config, auto_monitor=True)
is_closed = Cpt(EpicsSignalRO, "CLOSE", kind=Kind.omitted)
is_enabled = Cpt(EpicsSignalRO, "ENABLE")
is_ok = Cpt(EpicsSignalRO, "OK")
alarm = Cpt(EpicsSignalRO, "ALARM")
is_enabled = Cpt(EpicsSignalRO, "ENABLE", kind=Kind.config, auto_monitor=True)
is_ok = Cpt(EpicsSignalRO, "OK", kind=Kind.omitted, auto_monitor=True)
alarm = Cpt(EpicsSignalRO, "ALARM", kind=Kind.omitted, auto_monitor=True)
set_open = Cpt(EpicsSignal, "OPEN-SET", kind=Kind.omitted)
set_closed = Cpt(EpicsSignal, "CLOSE-SET", kind=Kind.omitted)
@@ -66,7 +62,7 @@ class Shutter(PSIDeviceBase):
"""
self._check_enabled()
self.set_open.put(1)
return SubscriptionStatus(self.is_open, _is_state(ShutterOpenState.OPEN))
return CompareStatus(self.is_open, ShutterOpenState.OPEN)
def close(self):
"""Close the shutter.
@@ -75,7 +71,7 @@ class Shutter(PSIDeviceBase):
"""
self._check_enabled()
self.set_closed.put(1)
return SubscriptionStatus(self.is_open, _is_state(ShutterOpenState.CLOSED))
return CompareStatus(self.is_open, ShutterOpenState.CLOSED)
def status(self) -> ShutterOpenState:
return ShutterOpenState(self.is_open.get())