Source code for qcodes.parameters.multi_channel_instrument_parameter

from __future__ import annotations

import logging
import sys
from typing import TYPE_CHECKING, Any, Generic, TypeVar

from .multi_parameter import MultiParameter

    from import Sequence

    from import InstrumentModule

    from .parameter_base import ParamRawDataType

InstrumentModuleType = TypeVar("InstrumentModuleType", bound="InstrumentModule")
_LOG = logging.getLogger(__name__)

[docs] class MultiChannelInstrumentParameter(MultiParameter, Generic[InstrumentModuleType]): """ Parameter to get or set multiple channels simultaneously. Will normally be created by a :class:`ChannelList` and not directly by anything else. Args: channels: A list of channels which we can operate on simultaneously. param_name: Name of the multichannel parameter """ def __init__( self, channels: Sequence[InstrumentModuleType], param_name: str, *args: Any, **kwargs: Any, ) -> None: super().__init__(*args, **kwargs) self._channels = channels self._param_name = param_name
[docs] def get_raw(self) -> tuple[ParamRawDataType, ...]: """ Return a tuple containing the data from each of the channels in the list. """ return tuple(chan.parameters[self._param_name].get() for chan in self._channels)
[docs] def set_raw(self, value: ParamRawDataType | Sequence[ParamRawDataType]) -> None: """ Set all parameters to this/these value(s). Args: value: The value(s) to set to. The type is given by the underlying parameter. """ try: for chan in self._channels: getattr(chan, self._param_name).set(value) except Exception as err: try: # Catch wrong length of value before any setting is done value_list = list(value) if len(value_list) != len(self._channels): raise ValueError for chan, val in zip(self._channels, value_list): getattr(chan, self._param_name).set(val) except (TypeError, ValueError): note = ( "Value should either be valid for a single parameter of the channel list " "or a sequence of valid values of the same length as the list." ) if sys.version_info >= (3, 11): err.add_note(note) else: _LOG.error(note) raise err from None
@property def full_names(self) -> tuple[str, ...]: """ Overwrite full_names because the instrument name is already included in the name. This happens because the instrument name is included in the channel name merged into the parameter name above. """ return self.names