Merge pull request #309 from ISISComputingGroup/ophyd_async_0.15

Migrate to `ophyd-async v0.15` (breaking changes for user-defined devices)

Author: FreddieAkeroyd

doc/conf.py

@@ -89,7 +89,7 @@
 intersphinx_mapping = {
     "python": ("https://docs.python.org/3", None),
     "bluesky": ("https://blueskyproject.io/bluesky/main/", None),
-    "ophyd_async": ("https://blueskyproject.io/ophyd-async/v0.14.0/", None),
+    "ophyd_async": ("https://blueskyproject.io/ophyd-async/v0.15/", None),
     "event_model": ("https://blueskyproject.io/event-model/main/", None),
     "scipp": ("https://scipp.github.io/", None),
     "scippneutron": ("https://scipp.github.io/scippneutron/", None),

pyproject.toml

@@ -42,7 +42,7 @@ classifiers = [
 dependencies = [
   "bluesky",  # Bluesky framework
   "confluent-kafka",  # Kafka producer
-  "ophyd-async[ca] == 0.14.2",  # Device abstraction. When changing, also change in doc/conf.py
+  "ophyd-async[ca] == 0.15.0",  # Device abstraction. When changing, also change in doc/conf.py
   "lmfit",  # Fitting
   "matplotlib",  # Plotting
   "msgpack-numpy",  # Encoding kafka messages

src/ibex_bluesky_core/devices/block.py

@@ -249,13 +249,15 @@ def plan():
                 be set to empty string to read and write to exactly the same PV.
 
         """
+        self._write_config: BlockWriteConfig[T] = write_config or BlockWriteConfig()
+
         self.setpoint: SignalRW[T] = epics_signal_rw(
-            datatype, f"{prefix}CS:SB:{block_name}{sp_suffix}"
+            datatype,
+            f"{prefix}CS:SB:{block_name}{sp_suffix}",
+            wait=self._write_config.use_completion_callback,
         )
         """The setpoint for this block."""
 
-        self._write_config: BlockWriteConfig[T] = write_config or BlockWriteConfig()
-
         if self._write_config.use_global_moving_flag:
             # Misleading PV name... says it's a str but it's really a bi record.
             # Only link to this if we need to (i.e. if use_global_moving_flag was requested)
@@ -283,9 +285,7 @@ def my_plan():
 
         async def do_set(setpoint: T) -> None:
             logger.info("Setting Block %s to %s", self.name, setpoint)
-            await self.setpoint.set(
-                setpoint, wait=self._write_config.use_completion_callback, timeout=None
-            )
+            await self.setpoint.set(setpoint, timeout=None)
             logger.info("Got completion callback from setting block %s to %s", self.name, setpoint)
 
             if self._write_config.use_global_moving_flag:

src/ibex_bluesky_core/devices/dae/__init__.py

@@ -108,7 +108,7 @@ async def set(self, value: T) -> None:
             OSError: If the signal failed to be set to the specified value.
 
         """
-        await self.signal.set(value, wait=True, timeout=None)
+        await self.signal.set(value, timeout=None)
         actual_value = await self.signal.get_value()
         if value != actual_value:
             raise OSError(
@@ -252,7 +252,7 @@ async def _trigger_and_get_raw_specdata(self) -> npt.NDArray[np.int32]:
         time channels (including the "junk" time-channel 0).
         """
         await self.controls.update_run.trigger()
-        await self.raw_spec_data_proc.set(1, wait=True)
+        await self.raw_spec_data_proc.set(1)
         (raw_data, nord) = await asyncio.gather(
             self.raw_spec_data.get_value(),
             self.raw_spec_data_nord.get_value(),

src/ibex_bluesky_core/devices/dae/_controls.py

@@ -98,5 +98,5 @@ async def set(self, value: BeginRunExBits) -> None:
 
         """
         logger.info("starting run with options %s", value)
-        await self._raw_begin_run_ex.set(value, wait=True, timeout=None)
+        await self._raw_begin_run_ex.set(value, timeout=None)
         logger.info("start run complete")

src/ibex_bluesky_core/devices/dae/_period_settings.py

@@ -180,4 +180,4 @@ async def set(self, value: DaePeriodSettingsData) -> None:
         current_xml = await self._raw_period_settings.get_value()
         to_write = _convert_period_settings_to_xml(current_xml, value)
         logger.info("set period settings: %s", to_write)
-        await self._raw_period_settings.set(to_write, wait=True, timeout=None)
+        await self._raw_period_settings.set(to_write, timeout=None)

src/ibex_bluesky_core/devices/dae/_settings.py

@@ -277,4 +277,4 @@ async def set(self, value: DaeSettingsData) -> None:
         current_xml = await self._raw_dae_settings.get_value()
         to_write = _convert_dae_settings_to_xml(current_xml, value)
         logger.info("set dae settings: %s", to_write)
-        await self._raw_dae_settings.set(to_write, wait=True, timeout=None)
+        await self._raw_dae_settings.set(to_write, timeout=None)

src/ibex_bluesky_core/devices/dae/_tcb_settings.py

@@ -199,4 +199,4 @@ async def set(self, value: DaeTCBSettingsData) -> None:
         xml = _convert_tcb_settings_to_xml(current_xml_dehexed, value)
         the_value_to_write = compress_and_hex(xml).decode()
         logger.info("set tcb settings: %s", the_value_to_write)
-        await self._raw_tcb_settings.set(the_value_to_write, wait=True, timeout=None)
+        await self._raw_tcb_settings.set(the_value_to_write, timeout=None)

src/ibex_bluesky_core/devices/reflectometry.py

@@ -78,7 +78,7 @@ async def set(self, value: float) -> None:
         indicate it has finished.
         """
         logger.info("setting %s to %s", self.setpoint.source, value)
-        await self.setpoint.set(value, wait=True, timeout=None)
+        await self.setpoint.set(value, timeout=None)
         await asyncio.sleep(0.1)
         logger.info("waiting for %s", self.changing.source)
         async for chg in observe_value(self.changing, done_timeout=self.changing_timeout):
@@ -118,7 +118,7 @@ async def set(self, value: float) -> None:
         if in_manager_mode != NoYesChoice.YES:
             raise ValueError(f"Cannot redefine {self.define_pos_sp.source} as not in manager mode.")
         logger.info("setting %s to %s", self.define_pos_sp.source, value)
-        await self.define_pos_sp.set(value, wait=True, timeout=None)
+        await self.define_pos_sp.set(value, timeout=None)
         logger.info("waiting for 1s for redefine to finish")
         # The Reflectometry server has a CHANGED PV for a redefine, but it doesn't actually
         # give a monitor update, so just wait an arbitrary length of time for it to be done.

src/ibex_bluesky_core/devices/simpledae/_controllers.py

@@ -18,11 +18,11 @@
 async def _end_or_abort_run(dae: Dae, save: bool) -> None:
     if save:
         logger.info("ending run")
-        await dae.controls.end_run.trigger(wait=True, timeout=None)
+        await dae.controls.end_run.trigger(timeout=None)
         logger.info("run ended")
     else:
         logger.info("aborting run")
-        await dae.controls.abort_run.trigger(wait=True, timeout=None)
+        await dae.controls.abort_run.trigger(timeout=None)
         logger.info("run aborted")
 
 
@@ -65,7 +65,7 @@ async def start_counting(self, dae: Dae) -> None:
         """
         logger.info("start counting")
         self._current_period += 1
-        await dae.period_num.set(self._current_period, wait=True, timeout=None)
+        await dae.period_num.set(self._current_period, timeout=None)
 
         # Error if the period change didn't work (e.g. we have exceeded max periods)
         await wait_for_value(
@@ -80,7 +80,7 @@ async def start_counting(self, dae: Dae) -> None:
         await wait_for_value(dae.period.raw_frames, 0, timeout=10)
 
         logger.info("resuming run")
-        await dae.controls.resume_run.trigger(wait=True, timeout=None)
+        await dae.controls.resume_run.trigger(timeout=None)
         await wait_for_value(
             dae.run_state,
             lambda v: v in {RunstateEnum.RUNNING, RunstateEnum.WAITING, RunstateEnum.VETOING},
@@ -90,7 +90,7 @@ async def start_counting(self, dae: Dae) -> None:
     async def stop_counting(self, dae: Dae) -> None:
         """Stop counting a scan point, by pausing the run."""
         logger.info("stop counting")
-        await dae.controls.pause_run.trigger(wait=True, timeout=None)
+        await dae.controls.pause_run.trigger(timeout=None)
         await wait_for_value(dae.run_state, RunstateEnum.PAUSED, timeout=10)
 
     async def teardown(self, dae: Dae) -> None:
@@ -133,7 +133,7 @@ def __init__(self, save_run: bool) -> None:
     async def start_counting(self, dae: Dae) -> None:
         """Start counting a scan point, by starting a DAE run."""
         logger.info("start counting")
-        await dae.controls.begin_run.trigger(wait=True, timeout=None)
+        await dae.controls.begin_run.trigger(timeout=None)
         await wait_for_value(
             dae.run_state,
             lambda v: v in {RunstateEnum.RUNNING, RunstateEnum.WAITING, RunstateEnum.VETOING},