mx-bluesky 1.5.0__py3-none-any.whl → 1.5.1__py3-none-any.whl

mx_bluesky/hyperion/experiment_plans/load_centre_collect_full_plan.py

@@ -76,17 +76,29 @@ def load_centre_collect_full(
             flyscan_event_handler,
         )
 
-        locations_to_collect_um: list[np.ndarray] = []
+        locations_to_collect_um: list[np.ndarray]
+        samples_to_collect: list[int]
 
         if flyscan_event_handler.xray_centre_results:
             selection_func = flyscan_result.resolve_selection_fn(
                 parameters.selection_params
             )
             hits = selection_func(flyscan_event_handler.xray_centre_results)
-            locations_to_collect_um = [hit.centre_of_mass_mm * 1000 for hit in hits]
+            hits_to_collect = []
+            for hit in hits:
+                if hit.sample_id is None:
+                    LOGGER.warning(
+                        f"Diffracting centre {hit} not collected because no sample id was assigned."
+                    )
+                else:
+                    hits_to_collect.append(hit)
 
+            locations_to_collect_um = [
+                hit.centre_of_mass_mm * 1000 for hit in hits_to_collect
+            ]
+            samples_to_collect = [hit.sample_id for hit in hits_to_collect]
             LOGGER.info(
-                f"Selected hits {hits} using {selection_func}, args={parameters.selection_params}"
+                f"Selected hits {hits_to_collect} using {selection_func}, args={parameters.selection_params}"
             )
         else:
             # If the xray centring hasn't found a result but has not thrown an error it
@@ -98,6 +110,7 @@ def load_centre_collect_full(
             locations_to_collect_um = [
                 np.array([initial_x_mm, initial_y_mm, initial_z_mm]) * 1000
             ]
+            samples_to_collect = [parameters.sample_id]
 
         multi_rotation = parameters.multi_rotation_scan
         rotation_template = multi_rotation.rotation_scans.copy()
@@ -108,9 +121,11 @@ def load_centre_collect_full(
 
         generator = rotation_scan_generator(is_alternating)
         next(generator)
-        for location in locations_to_collect_um:
+        for location, sample_id in zip(
+            locations_to_collect_um, samples_to_collect, strict=True
+        ):
             for rot in rotation_template:
-                combination = generator.send((rot, location))
+                combination = generator.send((rot, location, sample_id))
                 multi_rotation.rotation_scans.append(combination)
         multi_rotation = RotationScan.model_validate(multi_rotation)
 
@@ -125,8 +140,10 @@ def load_centre_collect_full(
 
 def rotation_scan_generator(
     is_alternating: bool,
-) -> Generator[RotationScanPerSweep, tuple[RotationScanPerSweep, np.ndarray], None]:
-    scan_template, location = yield  # type: ignore
+) -> Generator[
+    RotationScanPerSweep, tuple[RotationScanPerSweep, np.ndarray, int], None
+]:
+    scan_template, location, sample_id = yield  # type: ignore
     next_rotation_direction = scan_template.rotation_direction
     while True:
         scan = scan_template.model_copy()
@@ -135,6 +152,7 @@ def rotation_scan_generator(
             scan.y_start_um,
             scan.z_start_um,
         ) = location
+        scan.sample_id = sample_id
         if is_alternating:
             if next_rotation_direction != scan.rotation_direction:
                 # If originally specified direction of the current scan is different
@@ -146,4 +164,4 @@ def rotation_scan_generator(
                 scan.rotation_direction = next_rotation_direction
             next_rotation_direction = next_rotation_direction.opposite
 
-        scan_template, location = yield scan
+        scan_template, location, sample_id = yield scan

mx_bluesky/hyperion/experiment_plans/robot_load_and_change_energy.py

@@ -10,22 +10,25 @@ import bluesky.preprocessors as bpp
 import pydantic
 from blueapi.core import BlueskyContext
 from bluesky.utils import Msg
-from dodal.devices.aperturescatterguard import ApertureScatterguard, ApertureValue
+from dodal.devices.aperturescatterguard import ApertureScatterguard
 from dodal.devices.attenuator.attenuator import BinaryFilterAttenuator
 from dodal.devices.backlight import Backlight, BacklightPosition
 from dodal.devices.focusing_mirror import FocusingMirrorWithStripes, MirrorVoltages
 from dodal.devices.i03.dcm import DCM
 from dodal.devices.i03.undulator_dcm import UndulatorDCM
-from dodal.devices.motors import XYZPositioner
+from dodal.devices.motors import XYZStage
 from dodal.devices.oav.oav_detector import OAV
 from dodal.devices.robot import BartRobot, SampleLocation
-from dodal.devices.smargon import CombinedMove, Smargon, StubPosition
+from dodal.devices.smargon import Smargon
 from dodal.devices.thawer import Thawer
 from dodal.devices.webcam import Webcam
 from dodal.devices.xbpm_feedback import XBPMFeedback
-from dodal.plan_stubs.motor_utils import MoveTooLarge, home_and_reset_wrapper
 
-from mx_bluesky.common.utils.log import LOGGER
+from mx_bluesky.common.device_setup_plans.robot_load_unload import (
+    do_plan_while_lower_gonio_at_home,
+    prepare_for_robot_load,
+    wait_for_smargon_not_disabled,
+)
 from mx_bluesky.hyperion.experiment_plans.set_energy_plan import (
     SetEnergyComposite,
     set_energy_plan,
@@ -47,7 +50,7 @@ class RobotLoadAndEnergyChangeComposite:
     # RobotLoad fields
     robot: BartRobot
     webcam: Webcam
-    lower_gonio: XYZPositioner
+    lower_gonio: XYZStage
     thawer: Thawer
     oav: OAV
     smargon: Smargon
@@ -61,25 +64,6 @@ def create_devices(context: BlueskyContext) -> RobotLoadAndEnergyChangeComposite
     return device_composite_from_context(context, RobotLoadAndEnergyChangeComposite)
 
 
-def wait_for_smargon_not_disabled(smargon: Smargon, timeout=60):
-    """Waits for the smargon disabled flag to go low. The robot hardware is responsible
-    for setting this to low when it is safe to move. It does this through a physical
-    connection between the robot and the smargon.
-    """
-    LOGGER.info("Waiting for smargon enabled")
-    SLEEP_PER_CHECK = 0.1
-    times_to_check = int(timeout / SLEEP_PER_CHECK)
-    for _ in range(times_to_check):
-        smargon_disabled = yield from bps.rd(smargon.disabled)
-        if not smargon_disabled:
-            LOGGER.info("Smargon now enabled")
-            return
-        yield from bps.sleep(SLEEP_PER_CHECK)
-    raise TimeoutError(
-        "Timed out waiting for smargon to become enabled after robot load"
-    )
-
-
 def take_robot_snapshots(oav: OAV, webcam: Webcam, directory: Path):
     time_now = datetime.now()
     snapshot_format = f"{time_now.strftime('%H%M%S')}_{{device}}_after_load"
@@ -93,28 +77,15 @@ def take_robot_snapshots(oav: OAV, webcam: Webcam, directory: Path):
         yield from bps.wait("snapshots")
 
 
-def prepare_for_robot_load(
-    aperture_scatterguard: ApertureScatterguard, smargon: Smargon
-):
-    yield from bps.abs_set(
-        aperture_scatterguard.selected_aperture,
-        ApertureValue.OUT_OF_BEAM,
-        group="prepare_robot_load",
-    )
-
-    yield from bps.mv(smargon.stub_offsets, StubPosition.RESET_TO_ROBOT_LOAD)
-
-    yield from bps.mv(smargon, CombinedMove(x=0, y=0, z=0, chi=0, phi=0, omega=0))
-
-    yield from bps.wait("prepare_robot_load")
-
-
 def do_robot_load(
     composite: RobotLoadAndEnergyChangeComposite,
     sample_location: SampleLocation,
+    sample_id: int,
     demand_energy_ev: float | None,
     thawing_time: float,
 ):
+    yield from bps.abs_set(composite.robot.next_sample_id, sample_id, wait=True)
+
     yield from bps.abs_set(
         composite.robot,
         sample_location,
@@ -133,16 +104,6 @@ def do_robot_load(
     yield from wait_for_smargon_not_disabled(composite.smargon)
 
 
-def raise_exception_if_moved_out_of_cryojet(exception):
-    yield from bps.null()
-    if isinstance(exception, MoveTooLarge):
-        raise Exception(
-            f"Moving {exception.axis} back to {exception.position} after \
-                        robot load would move it out of the cryojet. The max safe \
-                        distance is {exception.maximum_move}"
-        )
-
-
 def pin_already_loaded(
     robot: BartRobot, sample_location: SampleLocation
 ) -> Generator[Msg, None, bool]:
@@ -158,6 +119,7 @@ def robot_load_and_snapshots(
     composite: RobotLoadAndEnergyChangeComposite,
     location: SampleLocation,
     snapshot_directory: Path,
+    sample_id: int,
     thawing_time: float,
     demand_energy_ev: float | None,
 ):
@@ -166,37 +128,25 @@ def robot_load_and_snapshots(
     robot_load_plan = do_robot_load(
         composite,
         location,
+        sample_id,
         demand_energy_ev,
         thawing_time,
     )
 
-    # The lower gonio must be in the correct position for the robot load and we
-    # want to put it back afterwards. Note we don't wait the robot is interlocked
-    # to the lower gonio and the  move is quicker than the robot takes to get to the
-    # load position.
-    yield from bpp.contingency_wrapper(
-        home_and_reset_wrapper(
-            robot_load_plan,
-            composite.lower_gonio,
-            BartRobot.LOAD_TOLERANCE_MM,
-            CONST.HARDWARE.CRYOJET_MARGIN_MM,
-            "lower_gonio",
-            wait_for_all=False,
-        ),
-        except_plan=raise_exception_if_moved_out_of_cryojet,
+    gonio_finished = yield from do_plan_while_lower_gonio_at_home(
+        robot_load_plan, composite.lower_gonio
     )
-
     yield from bps.wait(group="snapshot")
 
     yield from take_robot_snapshots(composite.oav, composite.webcam, snapshot_directory)
 
-    yield from bps.create(name=CONST.DESCRIPTORS.ROBOT_LOAD)
-    yield from bps.read(composite.robot.barcode)
+    yield from bps.create(name=CONST.DESCRIPTORS.ROBOT_UPDATE)
+    yield from bps.read(composite.robot)
     yield from bps.read(composite.oav.snapshot)
     yield from bps.read(composite.webcam)
     yield from bps.save()
 
-    yield from bps.wait("reset-lower_gonio")
+    yield from bps.wait(gonio_finished)
 
 
 def robot_load_and_change_energy_plan(
@@ -218,17 +168,13 @@ def robot_load_and_change_energy_plan(
                 composite,
                 sample_location,
                 params.snapshot_directory,
+                params.sample_id,
                 params.thawing_time,
                 params.demand_energy_ev,
             ),
             md={
                 "subplan_name": CONST.PLAN.ROBOT_LOAD,
-                "metadata": {
-                    "visit": params.visit,
-                    "sample_id": params.sample_id,
-                    "sample_puck": sample_location.puck,
-                    "sample_pin": sample_location.pin,
-                },
+                "metadata": {"visit": params.visit, "sample_id": params.sample_id},
                 "activate_callbacks": [
                     "RobotLoadISPyBCallback",
                 ],

mx_bluesky/hyperion/experiment_plans/robot_load_then_centre_plan.py

@@ -18,7 +18,7 @@ from dodal.devices.focusing_mirror import FocusingMirrorWithStripes, MirrorVolta
 from dodal.devices.i03 import Beamstop
 from dodal.devices.i03.dcm import DCM
 from dodal.devices.i03.undulator_dcm import UndulatorDCM
-from dodal.devices.motors import XYZPositioner
+from dodal.devices.motors import XYZStage
 from dodal.devices.oav.oav_detector import OAV
 from dodal.devices.oav.pin_image_recognition import PinTipDetection
 from dodal.devices.robot import BartRobot, SampleLocation
@@ -96,7 +96,7 @@ class RobotLoadThenCentreComposite:
     # RobotLoad fields
     robot: BartRobot
     webcam: Webcam
-    lower_gonio: XYZPositioner
+    lower_gonio: XYZStage
     beamstop: Beamstop
 
 

mx_bluesky/hyperion/experiment_plans/rotation_scan_plan.py

@@ -342,10 +342,15 @@ def _move_and_rotation(
 
     if params.take_snapshots:
         yield from bps.wait(CONST.WAIT.MOVE_GONIO_TO_START)
-        yield from setup_beamline_for_OAV(
-            composite.smargon, composite.backlight, composite.aperture_scatterguard
-        )
-        yield from bps.wait(group=CONST.WAIT.READY_FOR_OAV)
+
+        if not params.use_grid_snapshots:
+            yield from setup_beamline_for_OAV(
+                composite.smargon,
+                composite.backlight,
+                composite.aperture_scatterguard,
+                wait=True,
+            )
+
         if params.selected_aperture:
             yield from bps.prepare(
                 composite.aperture_scatterguard,

mx_bluesky/hyperion/external_interaction/agamemnon.py

@@ -148,6 +148,7 @@ def populate_parameters_from_agamemnon(agamemnon_params) -> Sequence[LoadCentreC
                     "name": "TopNByMaxCount",
                     "n": pin_type.expected_number_of_crystals,
                 },
+                "features": {"use_gpu_results": True},
                 "robot_load_then_centre": {
                     "storage_directory": str(visit_directory) + "/xraycentring",
                     "file_name": file_name,
@@ -156,7 +157,6 @@ def populate_parameters_from_agamemnon(agamemnon_params) -> Sequence[LoadCentreC
                     "omega_start_deg": 0.0,
                     "chi_start_deg": collection["chi"],
                     "transmission_frac": 1.0,
-                    "features": {"use_gpu_results": True},
                     **with_energy_params,
                 },
                 "multi_rotation_scan": {
@@ -228,10 +228,10 @@ def update_params_from_agamemnon(parameters: T) -> T:
             parameters.robot_load_then_centre.grid_width_um = pin_type.full_width
             parameters.select_centres.n = pin_type.expected_number_of_crystals
             if pin_type != SinglePin():
-                # Snapshots between each collection take a lot of time.
-                # Before we do https://github.com/DiamondLightSource/mx-bluesky/issues/226
-                # this will give no snapshots but that's preferable
+                # Rotation snapshots will be generated from the gridscan snapshots,
+                # no need to specify snapshot omega.
                 parameters.multi_rotation_scan.snapshot_omegas_deg = []
+                parameters.multi_rotation_scan.use_grid_snapshots = True
     except (ValueError, ValidationError) as e:
         LOGGER.warning(f"Failed to update parameters: {e}")
     except Exception as e:

mx_bluesky/hyperion/external_interaction/callbacks/__main__.py

@@ -29,7 +29,7 @@ from mx_bluesky.common.utils.log import (
     _get_logging_dirs,
     tag_filter,
 )
-from mx_bluesky.hyperion.external_interaction.callbacks.robot_load.ispyb_callback import (
+from mx_bluesky.hyperion.external_interaction.callbacks.robot_actions.ispyb_callback import (
     RobotLoadISPyBCallback,
 )
 from mx_bluesky.hyperion.external_interaction.callbacks.rotation.ispyb_callback import (

mx_bluesky/hyperion/external_interaction/callbacks/{robot_load → robot_actions}/ispyb_callback.py

@@ -12,6 +12,7 @@ from mx_bluesky.common.external_interaction.ispyb.exp_eye_store import (
     BLSampleStatus,
     ExpeyeInteraction,
     RobotActionID,
+    create_update_data_from_event_doc,
 )
 from mx_bluesky.common.utils.log import ISPYB_ZOCALO_CALLBACK_LOGGER
 from mx_bluesky.hyperion.parameters.constants import CONST
@@ -20,11 +21,21 @@ if TYPE_CHECKING:
     from event_model.documents import Event, EventDescriptor, RunStart, RunStop
 
 
+robot_update_mapping = {
+    "robot-barcode": "sampleBarcode",
+    "robot-current_pin": "containerLocation",
+    "robot-current_puck": "dewarLocation",
+    # I03 uses webcam/oav snapshots in place of before/after snapshots
+    "webcam-last_saved_path": "xtalSnapshotBefore",
+    "oav-snapshot-last_saved_path": "xtalSnapshotAfter",
+}
+
+
 class RobotLoadISPyBCallback(PlanReactiveCallback):
     def __init__(self) -> None:
         ISPYB_ZOCALO_CALLBACK_LOGGER.debug("Initialising ISPyB Robot Load Callback")
         super().__init__(log=ISPYB_ZOCALO_CALLBACK_LOGGER)
-        self._metadata: dict | None = None
+        self._sample_id: int | None = None
 
         self.run_uid: str | None = None
         self.descriptors: dict[str, EventDescriptor] = {}
@@ -35,22 +46,24 @@ class RobotLoadISPyBCallback(PlanReactiveCallback):
         ISPYB_ZOCALO_CALLBACK_LOGGER.debug(
             "ISPyB robot load callback received start document."
         )
-        if doc.get("subplan_name") == CONST.PLAN.ROBOT_LOAD:
+        subplan = doc.get("subplan_name")
+        if subplan == CONST.PLAN.ROBOT_LOAD or subplan == CONST.PLAN.ROBOT_UNLOAD:
             ISPYB_ZOCALO_CALLBACK_LOGGER.debug(
                 f"ISPyB robot load callback received: {doc}"
             )
             self.run_uid = doc.get("uid")
-            self._metadata = doc.get("metadata")
-            assert isinstance(self._metadata, dict)
+            metadata = doc.get("metadata")
+            assert isinstance(metadata, dict)
+            self._sample_id = metadata["sample_id"]
+            assert isinstance(self._sample_id, int)
             proposal, session = get_proposal_and_session_from_visit_string(
-                self._metadata["visit"]
+                metadata["visit"]
             )
-            self.action_id = self.expeye.start_load(
+            self.action_id = self.expeye.start_robot_action(
+                "LOAD" if subplan == CONST.PLAN.ROBOT_LOAD else "UNLOAD",
                 proposal,
                 session,
-                self._metadata["sample_id"],
-                self._metadata["sample_puck"],
-                self._metadata["sample_pin"],
+                self._sample_id,
             )
         return super().activity_gated_start(doc)
 
@@ -62,18 +75,14 @@ class RobotLoadISPyBCallback(PlanReactiveCallback):
         event_descriptor = self.descriptors.get(doc["descriptor"])
         if (
             event_descriptor
-            and event_descriptor.get("name") == CONST.DESCRIPTORS.ROBOT_LOAD
+            and event_descriptor.get("name") == CONST.DESCRIPTORS.ROBOT_UPDATE
         ):
             assert self.action_id is not None, (
                 "ISPyB Robot load callback event called unexpectedly"
             )
-            barcode = doc["data"]["robot-barcode"]
-            oav_snapshot = doc["data"]["oav-snapshot-last_saved_path"]
-            webcam_snapshot = doc["data"]["webcam-last_saved_path"]
             # I03 uses webcam/oav snapshots in place of before/after snapshots
-            self.expeye.update_barcode_and_snapshots(
-                self.action_id, barcode, webcam_snapshot, oav_snapshot
-            )
+            update_data = create_update_data_from_event_doc(robot_update_mapping, doc)
+            self.expeye.update_robot_action(self.action_id, update_data)
 
         return super().activity_gated_event(doc)
 
@@ -87,12 +96,12 @@ class RobotLoadISPyBCallback(PlanReactiveCallback):
             )
             exit_status = doc.get("exit_status")
             assert exit_status, "Exit status not available in stop document!"
-            assert self._metadata, "Metadata not received before stop document."
+            assert self._sample_id is not None, "Stop called before start"
             reason = doc.get("reason") or "OK"
 
-            self.expeye.end_load(self.action_id, exit_status, reason)
+            self.expeye.end_robot_action(self.action_id, exit_status, reason)
             self.expeye.update_sample_status(
-                self._metadata["sample_id"],
+                self._sample_id,
                 BLSampleStatus.LOADED
                 if exit_status == "success"
                 else BLSampleStatus.ERROR_BEAMLINE,

mx_bluesky/hyperion/external_interaction/callbacks/rotation/ispyb_callback.py

@@ -54,6 +54,7 @@ class RotationISPyBCallback(BaseISPyBCallback):
         super().__init__(emit=emit)
         self.last_sample_id: int | None = None
         self.ispyb_ids: IspybIds = IspybIds()
+        self.ispyb = StoreInIspyb(self.ispyb_config)
 
     def activity_gated_start(self, doc: RunStart):
         if doc.get("subplan_name") == CONST.PLAN.ROTATION_OUTER:
@@ -82,7 +83,6 @@ class RotationISPyBCallback(BaseISPyBCallback):
                     f"Collection is {self.params.ispyb_experiment_type} - storing sampleID to bundle images"
                 )
                 self.last_sample_id = self.params.sample_id
-            self.ispyb = StoreInIspyb(self.ispyb_config)
             ISPYB_ZOCALO_CALLBACK_LOGGER.info("Beginning ispyb deposition")
             data_collection_group_info = populate_data_collection_group(self.params)
             data_collection_info = populate_data_collection_info_for_rotation(

mx_bluesky/hyperion/external_interaction/callbacks/snapshot_callback.py

@@ -1,4 +1,5 @@
 import dataclasses
+import os
 import re
 from collections.abc import Iterator
 from datetime import datetime
@@ -172,6 +173,8 @@ class BeamDrawingCallback(PlanReactiveCallback):
                 f"Generating snapshot at {current_sample_pos_mm} from base snapshot {snapshot_info}"
             )
             output_snapshot_directory = data["oav-snapshot-directory"]
+            if not os.path.exists(output_snapshot_directory):
+                os.mkdir(output_snapshot_directory)
             base_file_stem = Path(snapshot_info.snapshot_path).stem
             output_snapshot_filename = _snapshot_filename(base_file_stem)
             output_snapshot_path = (

mx_bluesky/hyperion/external_interaction/config_server.py

@@ -1,7 +1,6 @@
 from functools import cache
 
 from daq_config_server.client import ConfigServer
-from pydantic import model_validator
 
 from mx_bluesky.common.external_interaction.config_server import FeatureFlags
 from mx_bluesky.common.utils.log import LOGGER
@@ -14,8 +13,6 @@ class HyperionFeatureFlags(FeatureFlags):
 
     Attributes:
         use_panda_for_gridscan:         If True then the PandA is used for gridscans, otherwise the zebra is used
-        compare_cpu_and_gpu_zocalo:     If True then GPU result processing is enabled
-            alongside CPU and the results are compared. The CPU result is still take.n
         use_gpu_results:                If True then GPU result processing is enabled
             and the GPU result is taken.
         set_stub_offsets:               If True then set the stub offsets after moving to the crystal (ignored for
@@ -31,15 +28,7 @@ class HyperionFeatureFlags(FeatureFlags):
     def get_config_server() -> ConfigServer:
         return ConfigServer(CONST.CONFIG_SERVER_URL, LOGGER)
 
-    @model_validator(mode="after")
-    def use_gpu_and_compare_cannot_both_be_true(self):
-        assert not (self.use_gpu_results and self.compare_cpu_and_gpu_zocalo), (
-            "Cannot both use GPU results and compare them to CPU"
-        )
-        return self
-
     use_panda_for_gridscan: bool = CONST.I03.USE_PANDA_FOR_GRIDSCAN
-    compare_cpu_and_gpu_zocalo: bool = CONST.I03.COMPARE_CPU_AND_GPU_ZOCALO
     use_gpu_results: bool = CONST.I03.USE_GPU_RESULTS
     set_stub_offsets: bool = CONST.I03.SET_STUB_OFFSETS
     omega_flip: bool = CONST.I03.OMEGA_FLIP

mx_bluesky/hyperion/parameters/constants.py

@@ -29,12 +29,8 @@ class I03Constants:
     OMEGA_FLIP = True
     ALTERNATE_ROTATION_DIRECTION = True
 
-    # Turns on GPU processing for zocalo and logs a comparison between GPU and CPU-
-    # processed results.
-    COMPARE_CPU_AND_GPU_ZOCALO = False
-
     # Turns on GPU processing for zocalo and uses the results that come back
-    USE_GPU_RESULTS = False
+    USE_GPU_RESULTS = True
 
 
 @dataclass(frozen=True)

mx_bluesky/hyperion/parameters/gridscan.py

@@ -20,9 +20,7 @@ class GridCommonWithHyperionDetectorParams(GridCommon, WithHyperionUDCFeatures):
     @property
     def detector_params(self):
         params = super().detector_params
-        params.enable_dev_shm = (
-            self.features.compare_cpu_and_gpu_zocalo or self.features.use_gpu_results
-        )
+        params.enable_dev_shm = self.features.use_gpu_results
         return params
 
 
@@ -35,9 +33,7 @@ class HyperionSpecifiedThreeDGridScan(WithHyperionUDCFeatures, SpecifiedThreeDGr
     @property
     def detector_params(self):
         params = super().detector_params
-        params.enable_dev_shm = (
-            self.features.compare_cpu_and_gpu_zocalo or self.features.use_gpu_results
-        )
+        params.enable_dev_shm = self.features.use_gpu_results
         return params
 
     # Relative to common grid scan, stub offsets are defined by config server

mx_bluesky/hyperion/parameters/load_centre_collect.py

@@ -50,6 +50,12 @@ class LoadCentreCollect(
             f"Unexpected fields found in LoadCentreCollect {disallowed_keys}"
         )
 
+        assert "features" not in values["robot_load_then_centre"], (
+            "Features flags must be specified at top-level in LoadCentreCollect"
+        )
+        assert "features" not in values["multi_rotation_scan"], (
+            "Features flags must be specified at top-level in LoadCentreCollect"
+        )
         keys_from_outer_load_centre_collect = (
             MxBlueskyParameters.model_fields.keys()
             | WithSample.model_fields.keys()
@@ -70,6 +76,9 @@ class LoadCentreCollect(
             f"Unexpected keys in multi_rotation_scan: {', '.join(duplicated_multi_rotation_scan_keys)}"
         )
 
+        for rotation in values["multi_rotation_scan"]["rotation_scans"]:
+            rotation["sample_id"] = values["sample_id"]
+
         new_robot_load_then_centre_params = construct_from_values(
             values, values["robot_load_then_centre"], RobotLoadThenCentre
         )
@@ -80,6 +89,12 @@ class LoadCentreCollect(
         values["robot_load_then_centre"] = new_robot_load_then_centre_params
         return values
 
+    @model_validator(mode="after")
+    def _ensure_features_are_internally_consistent(self) -> Self:
+        self.robot_load_then_centre.features = self.features
+        self.multi_rotation_scan.features = self.features
+        return self
+
     @model_validator(mode="after")
     def _check_rotation_start_xyz_is_not_specified(self) -> Self:
         for scan in self.multi_rotation_scan.single_rotation_scans:

mx_bluesky/hyperion/parameters/rotation.py

@@ -18,12 +18,14 @@ from scanspec.core import Path as ScanPath
 from scanspec.specs import Line
 
 from mx_bluesky.common.parameters.components import (
+    DiffractionExperiment,
     DiffractionExperimentWithSample,
     IspybExperimentType,
     OptionalGonioAngleStarts,
     OptionalXyzStarts,
     RotationAxis,
     SplitScan,
+    WithSample,
     WithScan,
 )
 from mx_bluesky.hyperion.parameters.components import WithHyperionUDCFeatures
@@ -33,7 +35,7 @@ from mx_bluesky.hyperion.parameters.constants import (
 )
 
 
-class RotationScanPerSweep(OptionalGonioAngleStarts, OptionalXyzStarts):
+class RotationScanPerSweep(OptionalGonioAngleStarts, OptionalXyzStarts, WithSample):
     """
     Describes a rotation scan about the specified axis.
 
@@ -54,7 +56,7 @@ class RotationScanPerSweep(OptionalGonioAngleStarts, OptionalXyzStarts):
     nexus_vds_start_img: int = Field(default=0, ge=0)
 
 
-class RotationExperiment(DiffractionExperimentWithSample, WithHyperionUDCFeatures):
+class RotationExperiment(DiffractionExperiment, WithHyperionUDCFeatures):
     shutter_opening_time_s: float = Field(default=CONST.I03.SHUTTER_TIME_S)
     rotation_increment_deg: float = Field(default=0.1, gt=0)
     ispyb_experiment_type: IspybExperimentType = Field(
@@ -105,7 +107,9 @@ class RotationExperiment(DiffractionExperimentWithSample, WithHyperionUDCFeature
             return aperture_position
 
 
-class SingleRotationScan(WithScan, RotationScanPerSweep, RotationExperiment):
+class SingleRotationScan(
+    WithScan, RotationScanPerSweep, RotationExperiment, DiffractionExperimentWithSample
+):
     @property
     def detector_params(self):
         return self._detector_params(self.omega_start_deg)

{mx_bluesky-1.5.0.dist-info → mx_bluesky-1.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mx-bluesky
-Version: 1.5.0
+Version: 1.5.1
 Summary: Bluesky tools for MX Beamlines at DLS
 Author-email: Dominic Oram <dominic.oram@diamond.ac.uk>
 License:                                  Apache License
@@ -221,7 +221,7 @@ Requires-Dist: ispyb
 Requires-Dist: jupyterlab
 Requires-Dist: matplotlib
 Requires-Dist: nexgen>=0.11.0
-Requires-Dist: numpy
+Requires-Dist: numpy==2.2.6
 Requires-Dist: opencv-python
 Requires-Dist: opentelemetry-distro
 Requires-Dist: opentelemetry-exporter-otlp
@@ -233,13 +233,14 @@ Requires-Dist: requests
 Requires-Dist: scanspec
 Requires-Dist: scipy
 Requires-Dist: semver
+Requires-Dist: deepdiff
 Requires-Dist: matplotlib
 Requires-Dist: blueapi>=0.11.1
 Requires-Dist: daq-config-server==0.1.1
 Requires-Dist: ophyd>=1.10.5
 Requires-Dist: ophyd-async>=0.10.0a2
 Requires-Dist: bluesky>=1.13.1
-Requires-Dist: dls-dodal==1.49.0
+Requires-Dist: dls-dodal==1.51.0
 Provides-Extra: dev
 Requires-Dist: black; extra == "dev"
 Requires-Dist: build; extra == "dev"