Skip to content

Commit d739f0f

Browse files
Use enum on aravis driver (#412)
* Use enum on aravis driver * Update src/ophyd_async/epics/areadetector/drivers/aravis_driver.py Co-authored-by: Tom C (DLS) <[email protected]> --------- Co-authored-by: Tom C (DLS) <[email protected]>
1 parent 9165c5a commit d739f0f

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/ophyd_async/epics/areadetector/drivers/aravis_driver.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from enum import Enum
2-
from typing import Literal
32

3+
from ophyd_async.core import SubsetEnum
44
from ophyd_async.epics.areadetector.drivers import ADBase
55
from ophyd_async.epics.signal.signal import epics_signal_rw_rbv
66

@@ -19,7 +19,7 @@ class AravisTriggerMode(str, Enum):
1919
To prevent requiring one Enum class per possible configuration, we set as this Enum
2020
but read from the underlying signal as a str.
2121
"""
22-
AravisTriggerSource = Literal["Freerun", "Line1", "Line2", "Line3", "Line4"]
22+
AravisTriggerSource = SubsetEnum["Freerun", "Line1"]
2323

2424

2525
class AravisDriver(ADBase):
@@ -34,5 +34,7 @@ def __init__(self, prefix: str, name: str = "") -> None:
3434
self.trigger_mode = epics_signal_rw_rbv(
3535
AravisTriggerMode, prefix + "TriggerMode"
3636
)
37-
self.trigger_source = epics_signal_rw_rbv(str, prefix + "TriggerSource")
37+
self.trigger_source = epics_signal_rw_rbv(
38+
AravisTriggerSource, prefix + "TriggerSource"
39+
)
3840
super().__init__(prefix, name=name)

0 commit comments

Comments
 (0)