Skip to content

Commit

Permalink
variable name changes
Browse files Browse the repository at this point in the history
  • Loading branch information
ZohebShaikh committed Sep 27, 2024
1 parent 2ab2da8 commit 26ccd6f
Show file tree
Hide file tree
Showing 8 changed files with 20 additions and 20 deletions.
22 changes: 10 additions & 12 deletions src/ophyd_async/core/_detector.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
from abc import ABC, abstractmethod
from collections.abc import AsyncGenerator, AsyncIterator, Callable, Iterator, Sequence
from enum import Enum
from functools import cached_property
from typing import (
Generic,
)
Expand Down Expand Up @@ -70,8 +71,8 @@ class TriggerInfo(BaseModel):
multiplier: int = 1

@computed_field
@property
def total_frames(self) -> int:
@cached_property
def total_number_of_triggers(self) -> int:
return (
sum(self.number_of_triggers)
if isinstance(self.number_of_triggers, list)
Expand Down Expand Up @@ -334,10 +335,10 @@ async def kickoff(self):
assert (
self._trigger_info and self._number_of_triggers_iter
), "Prepare must be called before kickoff!"
if self._frames_completed >= self._trigger_info.total_frames:
raise Exception(
if self._frames_completed >= self._trigger_info.total_number_of_triggers:
raise RuntimeError(
f"Kickoff called more than the configured number of "
f"{self._trigger_info.total_frames} iteration(s)!"
f"{self._trigger_info.total_number_of_triggers} iteration(s)!"
)
self._frames_to_complete = next(self._number_of_triggers_iter)
self._frames_completed += self._frames_to_complete
Expand All @@ -355,16 +356,11 @@ async def complete(self):
)
try:
async for index in indices_written:
target = (
self._frames_to_complete
if isinstance(self._trigger_info.number_of_triggers, list)
else self._trigger_info.number_of_triggers
)
yield WatcherUpdate(
name=self.name,
current=index,
initial=self._initial_frame,
target=target,
target=self._frames_to_complete,
unit="",
precision=0,
time_elapsed=time.monotonic() - self._fly_start
Expand All @@ -375,8 +371,10 @@ async def complete(self):
break
finally:
await indices_written.aclose()
if self._frames_completed == self._trigger_info.total_frames:
if self._frames_completed == self._trigger_info.total_number_of_triggers:
self._frames_completed = 0
self._frames_to_complete = 0
self._number_of_triggers_iter = None
await self.controller.wait_for_idle()

async def describe_collect(self) -> dict[str, DataKey]:
Expand Down
4 changes: 2 additions & 2 deletions src/ophyd_async/epics/adaravis/_aravis_controller.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ def get_deadtime(self, exposure: float | None) -> float:
return _HIGHEST_POSSIBLE_DEADTIME

async def prepare(self, trigger_info: TriggerInfo):
if trigger_info.total_frames == 0:
if trigger_info.total_number_of_triggers == 0:
image_mode = adcore.ImageMode.continuous
else:
image_mode = adcore.ImageMode.multiple
Expand All @@ -43,7 +43,7 @@ async def prepare(self, trigger_info: TriggerInfo):

await asyncio.gather(
self._drv.trigger_source.set(trigger_source),
self._drv.num_images.set(trigger_info.total_frames),
self._drv.num_images.set(trigger_info.total_number_of_triggers),
self._drv.image_mode.set(image_mode),
)

Expand Down
2 changes: 1 addition & 1 deletion src/ophyd_async/epics/adkinetix/_kinetix_controller.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ def get_deadtime(self, exposure: float | None) -> float:
async def prepare(self, trigger_info: TriggerInfo):
await asyncio.gather(
self._drv.trigger_mode.set(KINETIX_TRIGGER_MODE_MAP[trigger_info.trigger]),
self._drv.num_images.set(trigger_info.total_frames),
self._drv.num_images.set(trigger_info.total_number_of_triggers),
self._drv.image_mode.set(adcore.ImageMode.multiple),
)
if trigger_info.livetime is not None and trigger_info.trigger not in [
Expand Down
4 changes: 3 additions & 1 deletion src/ophyd_async/epics/adpilatus/_pilatus_controller.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,9 @@ async def prepare(self, trigger_info: TriggerInfo):
await asyncio.gather(
self._drv.trigger_mode.set(self._get_trigger_mode(trigger_info.trigger)),
self._drv.num_images.set(
999_999 if trigger_info.total_frames == 0 else trigger_info.total_frames
999_999
if trigger_info.total_number_of_triggers == 0
else trigger_info.total_number_of_triggers
),
self._drv.image_mode.set(adcore.ImageMode.multiple),
)
Expand Down
2 changes: 1 addition & 1 deletion src/ophyd_async/epics/adsimdetector/_sim_controller.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ async def prepare(self, trigger_info: TriggerInfo):
DEFAULT_TIMEOUT + await self.driver.acquire_time.get_value()
)
await asyncio.gather(
self.driver.num_images.set(trigger_info.total_frames),
self.driver.num_images.set(trigger_info.total_number_of_triggers),
self.driver.image_mode.set(adcore.ImageMode.multiple),
)

Expand Down
2 changes: 1 addition & 1 deletion src/ophyd_async/epics/advimba/_vimba_controller.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ async def prepare(self, trigger_info: TriggerInfo):
await asyncio.gather(
self._drv.trigger_mode.set(TRIGGER_MODE[trigger_info.trigger]),
self._drv.exposure_mode.set(EXPOSE_OUT_MODE[trigger_info.trigger]),
self._drv.num_images.set(trigger_info.total_frames),
self._drv.num_images.set(trigger_info.total_number_of_triggers),
self._drv.image_mode.set(adcore.ImageMode.multiple),
)
if trigger_info.livetime is not None and trigger_info.trigger not in [
Expand Down
2 changes: 1 addition & 1 deletion src/ophyd_async/epics/eiger/_eiger_controller.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ async def prepare(self, trigger_info: TriggerInfo):
self._drv.trigger_mode.set(
EIGER_TRIGGER_MODE_MAP[trigger_info.trigger].value
),
self._drv.num_images.set(trigger_info.total_frames),
self._drv.num_images.set(trigger_info.total_number_of_triggers),
]
if trigger_info.livetime is not None:
coros.extend(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ async def arm(self):
self._coroutine_for_image_writing(
self._trigger_info.livetime,
self.period,
self._trigger_info.total_frames,
self._trigger_info.total_number_of_triggers,
)
)

Expand Down

0 comments on commit 26ccd6f

Please sign in to comment.