1259 test oav plans

src/mx_bluesky/beamlines/i04/__init__.py

@@ -1,6 +1,13 @@
 from mx_bluesky.beamlines.i04.experiment_plans.i04_grid_detect_then_xray_centre_plan import (
     i04_grid_detect_then_xray_centre,
 )
+from mx_bluesky.beamlines.i04.test_plans.oav_automation import (
+    feedback_check,
+    move_scintillator,
+    open_close_fast_shutter,
+    set_transmission_percentage,
+    take_OAV_image,
+)
 from mx_bluesky.beamlines.i04.thawing_plan import (
     thaw,
     thaw_and_murko_centre,
@@ -12,4 +19,11 @@
     "thaw_and_stream_to_redis",
     "i04_grid_detect_then_xray_centre",
     "thaw_and_murko_centre",
+    "feedback_check",
+    "move_scintillator",
+    "open_close_fast_shutter",
+    "set_transmission_percentage",
+    "take_OAV_image",
 ]
+
+# testing my git script

src/mx_bluesky/beamlines/i04/test_plans/__init__.py

@@ -0,0 +1,15 @@
+from mx_bluesky.beamlines.i04.test_plans.oav_automation import (
+    feedback_check,
+    move_scintillator,
+    open_close_fast_shutter,
+    set_transmission_percentage,
+    take_OAV_image,
+)
+
+__all__ = [
+    "feedback_check",
+    "move_scintillator",
+    "open_close_fast_shutter",
+    "set_transmission_percentage",
+    "take_OAV_image",
+]

src/mx_bluesky/beamlines/i04/test_plans/oav_automation.py

@@ -0,0 +1,66 @@
+import bluesky.plan_stubs as bps
+from bluesky.utils import MsgGenerator
+from dodal.common import inject
+from dodal.devices.attenuator.attenuator import BinaryFilterAttenuator
+from dodal.devices.oav.oav_detector import OAV
+from dodal.devices.scintillator import InOut, Scintillator
+from dodal.devices.xbpm_feedback import XBPMFeedback
+from dodal.devices.zebra.zebra_controlled_shutter import (
+    ZebraShutter,
+    ZebraShutterControl,
+    ZebraShutterState,
+)
+
+# from dodal.devices.oav.snapshots.snapshot_with_grid import SnapshotWithGrid
+
+"""
+My task:
+    Move the scintillator in and out
+    Change transmission percentage
+    Take OAV images
+    Open and close fast shutter
+"""
+
+
+# def oav_automation_test()
+def feedback_check(
+    feedback: XBPMFeedback = inject("xbpm_feedback"),
+):
+    yield from bps.trigger(feedback, wait=True)
+
+
+def set_transmission_percentage(
+    percentage: float,
+    attenuator: BinaryFilterAttenuator = inject("attenuator"),
+) -> MsgGenerator:
+    yield from bps.mv(attenuator, percentage / 100)
+
+
+def open_close_fast_shutter(
+    shutter_state: ZebraShutterState,
+    shutter: ZebraShutter = inject("sample_shutter"),
+) -> MsgGenerator:
+    base_control_mode = yield from bps.rd(shutter.control_mode)
+
+    yield from bps.abs_set(shutter.control_mode, ZebraShutterControl.MANUAL, wait=True)
+    yield from bps.abs_set(shutter, shutter_state, wait=True)
+
+    yield from bps.abs_set(shutter.control_mode, base_control_mode, wait=True)
+
+
+def take_OAV_image(
+    file_path: str,
+    file_name: str,
+    oav: OAV = inject("oav"),
+) -> MsgGenerator:
+    group = "path setting"
+    yield from bps.abs_set(oav.snapshot.filename, file_name, group=group)
+    yield from bps.abs_set(oav.snapshot.directory, file_path, group=group)
+    yield from bps.wait(group)
+    yield from bps.trigger(oav.snapshot, wait=True)
+
+
+def move_scintillator(
+    scintillator_state: InOut, scintillator: Scintillator = inject("scintillator")
+) -> MsgGenerator:
+    yield from bps.mv(scintillator.selected_pos, scintillator_state)