added doc strings; removed target param

This commit is contained in:
wakonig_k 2022-09-05 18:23:05 +02:00
parent 4884107f2d
commit 80b0b4701d
3 changed files with 69 additions and 16 deletions

View File

@ -95,9 +95,9 @@ class DeviceMsgMixin:
"""Trigger a reading and wait for completion. """Trigger a reading and wait for completion.
Args: Args:
group (str): waitingsub group wait_group (str): wait group
wait_group (str): _description_ device (list, optional): List of device names. Can be specified instead of group. Defaults to None.
device (list, optional): _description_. Defaults to None. group (str, optional): Group name of devices. Can be specified instead of device. Defaults to None.
pointID (int, optional): _description_. Defaults to None. pointID (int, optional): _description_. Defaults to None.
""" """
@ -106,6 +106,15 @@ class DeviceMsgMixin:
yield from self.wait(device=device, wait_type="read", group=group, wait_group=wait_group) yield from self.wait(device=device, wait_type="read", group=group, wait_group=wait_group)
def open_scan(self, *, scan_motors: list, num_pos: int, scan_name: str, scan_type: str): def open_scan(self, *, scan_motors: list, num_pos: int, scan_name: str, scan_type: str):
"""Open a new scan.
Args:
scan_motors (list): List of scan motors.
num_pos (int): Number of positions within the scope of this scan.
scan_name (str): Scan name.
scan_type (str): Scan type (e.g. 'step' or 'fly')
"""
yield self._device_msg( yield self._device_msg(
device=None, device=None,
action="open_scan", action="open_scan",
@ -118,6 +127,12 @@ class DeviceMsgMixin:
) )
def kickoff(self, *, device: str, parameter: dict = None): def kickoff(self, *, device: str, parameter: dict = None):
"""Kickoff a fly scan device.
Args:
device (str): Device name of flyer.
parameter (dict, optional): Additional parameters that should be forwarded to the device. Defaults to {}.
"""
parameter = parameter if not None else {} parameter = parameter if not None else {}
yield self._device_msg( yield self._device_msg(
device=device, device=device,
@ -127,15 +142,19 @@ class DeviceMsgMixin:
) )
def close_scan(self): def close_scan(self):
"""Close the scan."""
yield self._device_msg(device=None, action="close_scan", parameter={}) yield self._device_msg(device=None, action="close_scan", parameter={})
def stage(self): def stage(self):
"""Stage all devices"""
yield self._device_msg(device=None, action="stage", parameter={}) yield self._device_msg(device=None, action="stage", parameter={})
def unstage(self): def unstage(self):
"""Unstage all devices"""
yield self._device_msg(device=None, action="unstage", parameter={}) yield self._device_msg(device=None, action="unstage", parameter={})
def baseline_reading(self): def baseline_reading(self):
"""Run the baseline readings."""
yield self._device_msg( yield self._device_msg(
device=None, device=None,
action="baseline_reading", action="baseline_reading",
@ -147,12 +166,21 @@ class DeviceMsgMixin:
self, self,
*, *,
wait_type: str, wait_type: str,
device=None, device: Union[List[str], str] = None,
group: str = None, group: str = None,
wait_group: str = None, wait_group: str = None,
wait_time=None, wait_time: float = None,
): ):
"""Wait for an event.
Args:
wait_type (str): wait type
device (Union[List[str], str], optional): List of device names. Defaults to None.
group (str, optional): Device group that can be used instead of the device argument. Defaults to None.
wait_group (str, optional): Wait group. Defaults to None.
wait_time (float, optional): Wait time (for wait_type="trigger"). Defaults to None.
"""
self._check_device_and_groups(device, group) self._check_device_and_groups(device, group)
parameter = {"type": wait_type, "time": wait_time, "group": group, "wait_group": wait_group} parameter = {"type": wait_type, "time": wait_time, "group": group, "wait_group": wait_group}
self._exclude_nones(parameter) self._exclude_nones(parameter)
@ -168,11 +196,19 @@ class DeviceMsgMixin:
wait_group: str, wait_group: str,
device: list = None, device: list = None,
pointID: int = None, pointID: int = None,
target=None,
group: str = None, group: str = None,
): ):
"""_summary_
Args:
wait_group (str): Wait group.
device (list, optional): Device name. Can be used instead of group. Defaults to None.
pointID (int, optional): pointID to assign this reading to point within the scan. Defaults to None.
group (str, optional): Device group. Can be used instead of device. Defaults to None.
"""
self._check_device_and_groups(device, group) self._check_device_and_groups(device, group)
parameter = {"target": target, "group": group, "wait_group": wait_group} parameter = {"group": group, "wait_group": wait_group}
metadata = {"pointID": pointID} metadata = {"pointID": pointID}
self._exclude_nones(parameter) self._exclude_nones(parameter)
self._exclude_nones(metadata) self._exclude_nones(metadata)
@ -184,6 +220,13 @@ class DeviceMsgMixin:
) )
def trigger(self, *, group: str, pointID: int): def trigger(self, *, group: str, pointID: int):
"""Trigger a device group
Args:
group (str): Device group that should receive the trigger.
pointID (int): pointID that should be attached to this trigger event.
"""
yield self._device_msg( yield self._device_msg(
device=None, device=None,
action="trigger", action="trigger",
@ -192,6 +235,14 @@ class DeviceMsgMixin:
) )
def set(self, *, device: str, value: float, wait_group: str): def set(self, *, device: str, value: float, wait_group: str):
"""Set the device to a specific value.
Args:
device (str): Device name
value (float): Target value.
wait_group (str): wait group for this event.
"""
yield self._device_msg( yield self._device_msg(
device=device, device=device,
action="set", action="set",
@ -202,15 +253,25 @@ class DeviceMsgMixin:
) )
def open_scan_def(self): def open_scan_def(self):
"""open a new scan definition"""
yield self._device_msg(device=None, action="open_scan_def", parameter={}) yield self._device_msg(device=None, action="open_scan_def", parameter={})
def close_scan_def(self): def close_scan_def(self):
"""close a scan definition"""
yield self._device_msg(device=None, action="close_scan_def", parameter={}) yield self._device_msg(device=None, action="close_scan_def", parameter={})
def close_scan_group(self): def close_scan_group(self):
"""close a scan group"""
yield self._device_msg(device=None, action="close_scan_group", parameter={}) yield self._device_msg(device=None, action="close_scan_group", parameter={})
def rpc(self, *, device: str, parameter: dict): def rpc(self, *, device: str, parameter: dict):
"""Perfrom an RPC (remote procedure call) on a device.
Args:
device (str): Device name.
parameter (dict): parameters used for this rpc instructions.
"""
yield self._device_msg( yield self._device_msg(
device=device, device=device,
action="rpc", action="rpc",

View File

@ -337,10 +337,7 @@ class ScanBase(RequestBase):
yield from self.stubs.trigger(group="trigger", pointID=self.pointID) yield from self.stubs.trigger(group="trigger", pointID=self.pointID)
yield from self.stubs.wait(wait_type="trigger", group="trigger", wait_time=self.exp_time) yield from self.stubs.wait(wait_type="trigger", group="trigger", wait_time=self.exp_time)
yield from self.stubs.read( yield from self.stubs.read(
group="primary", group="primary", wait_group="readout_primary", pointID=self.pointID
wait_group="readout_primary",
pointID=self.pointID,
target="primary",
) )
yield from self.stubs.wait( yield from self.stubs.wait(
wait_type="read", group="scan_motor", wait_group="readout_primary" wait_type="read", group="scan_motor", wait_group="readout_primary"

View File

@ -245,7 +245,6 @@ def test_scan_move(mv_msg, reference_msg_list):
device=None, device=None,
action="read", action="read",
parameter={ parameter={
"target": "primary",
"group": "primary", "group": "primary",
"wait_group": "readout_primary", "wait_group": "readout_primary",
}, },
@ -306,7 +305,6 @@ def test_scan_move(mv_msg, reference_msg_list):
device=None, device=None,
action="read", action="read",
parameter={ parameter={
"target": "primary",
"group": "primary", "group": "primary",
"wait_group": "readout_primary", "wait_group": "readout_primary",
}, },
@ -371,7 +369,6 @@ def test_scan_move(mv_msg, reference_msg_list):
device=None, device=None,
action="read", action="read",
parameter={ parameter={
"target": "primary",
"group": "primary", "group": "primary",
"wait_group": "readout_primary", "wait_group": "readout_primary",
}, },
@ -757,7 +754,6 @@ def test_device_rpc():
device=None, device=None,
action="read", action="read",
parameter={ parameter={
"target": "primary",
"group": "primary", "group": "primary",
"wait_group": "readout_primary", "wait_group": "readout_primary",
}, },
@ -819,7 +815,6 @@ def test_device_rpc():
device=None, device=None,
action="read", action="read",
parameter={ parameter={
"target": "primary",
"group": "primary", "group": "primary",
"wait_group": "readout_primary", "wait_group": "readout_primary",
}, },