Add cryostream gantry device (#1664)

* Add cryostream_gantry device
* Add selection PVs to cryostream gantry

---------

Co-authored-by: Jacob Williamson <[email protected]>
Co-authored-by: Dominic Oram <[email protected]>

Author: 3 people

src/dodal/beamlines/i03.py

@@ -21,7 +21,7 @@
 from dodal.devices.backlight import Backlight
 from dodal.devices.baton import Baton
 from dodal.devices.collimation_table import CollimationTable
-from dodal.devices.cryostream import CryoStream
+from dodal.devices.cryostream import CryoStream, CryoStreamGantry
 from dodal.devices.detector.detector_motion import DetectorMotion
 from dodal.devices.diamond_filter import DiamondFilter, I03Filters
 from dodal.devices.eiger import EigerDetector
@@ -396,6 +396,14 @@ def cryostream() -> CryoStream:
     return CryoStream(PREFIX.beamline_prefix)
 
 
+@device_factory()
+def cryostream_gantry() -> CryoStreamGantry:
+    """Get the i03 cryostream gantry device, instantiate it if it hasn't already been.
+    If this is called when already instantiated in i03, it will return the existing object.
+    """
+    return CryoStreamGantry(PREFIX.beamline_prefix)
+
+
 @device_factory()
 def diamond_filter() -> DiamondFilter[I03Filters]:
     """Get the i03 diamond filter device, instantiate it if it hasn't already been.

src/dodal/devices/cryostream.py

@@ -32,6 +32,11 @@ class TurboEnum(StrictEnum):
     AUTO = "Auto"
 
 
+class CryoStreamSelection(StrictEnum):
+    CRYOJET = "CryoJet"
+    HC1 = "HC1"
+
+
 class OxfordCryoStreamController(StandardReadable):
     def __init__(self, prefix: str, name: str = ""):
         with self.add_children_as_readables(StandardReadableFormat.CONFIG_SIGNAL):
@@ -107,3 +112,19 @@ def __init__(self, prefix: str, name=""):
             self.status = OxfordCryoStreamStatus(prefix=prefix)
 
         super().__init__(name)
+
+
+class CryoStreamGantry(StandardReadable):
+    def __init__(self, prefix: str, name: str = ""):
+        with self.add_children_as_readables(StandardReadableFormat.CONFIG_SIGNAL):
+            self.cryostream_selector = epics_signal_r(
+                CryoStreamSelection, f"{prefix}-EA-GANT-01:CTRL"
+            )
+            self.hc1_selected = epics_signal_r(
+                int, f"{prefix}-MO-STEP-02:GPIO_INP_BITS.B2"
+            )
+            self.cryostream_selected = epics_signal_r(
+                int, f"{prefix}-MO-STEP-02:GPIO_INP_BITS.B3"
+            )
+
+        super().__init__(name)