Source code for quantify_scheduler.gettables

# Repository: https://gitlab.com/quantify-os/quantify-scheduler
# Licensed according to the LICENCE file on the main branch
r"""
Module containing :class:`quantify_core.measurement.types.Gettable`\s for use with
quantify-scheduler.

.. warning::

    The gettable module is expected to change significantly as the
    acquisition protocols (#36 and #80) get fully supported by the scheduler.
    Currently different Gettables are required for different acquisition modes.
    The intent is to have one generic ``ScheduleGettable``.
    Expect breaking changes.
"""
# pylint: disable-all
from __future__ import annotations

import logging
import sys
from typing import TYPE_CHECKING, Any, Callable, Hashable

import numpy as np
from qcodes.parameters import Parameter

from quantify_scheduler.helpers.diagnostics_report import _generate_diagnostics_report

if TYPE_CHECKING:
    from numpy.typing import NDArray
    from xarray import Dataset

    from quantify_scheduler import CompiledSchedule, Schedule
    from quantify_scheduler.device_under_test.quantum_device import QuantumDevice

[docs] logger = logging.getLogger(__name__)
[docs] class AcquisitionProtocolError(TypeError): pass
[docs] class AcquisitionProtocolNotSupportedError(NotImplementedError): pass
[docs] class ScheduleGettable: """ Generic gettable for a quantify schedule using vector (I,Q) acquisition. The gettable evaluates the parameters passed as ``schedule_kwargs``, then generates the :class:`quantify_scheduler.schedules.schedule.Schedule` using the ``schedule_function``, this is then compiled and finally executed by the :class:`~.InstrumentCoordinator`. ``ScheduleGettable`` can be set to return either static (demodulated) I and Q values or magnitude and phase. Parameters ---------- quantum_device The qcodes instrument representing the quantum device under test (DUT) containing quantum device properties and setup configuration information. schedule_function A function which returns a :class:`quantify_scheduler.schedules.schedule.Schedule`. The function is required to have the ``repetitions`` keyword argument. schedule_kwargs The schedule function keyword arguments, when a value in this dictionary is a :class:`~qcodes.instrument.parameter.Parameter`, this parameter will be evaluated every time :code:`.get()` is called before being passed to the :code:`schedule_function`. num_channels The number of channels to expect in the acquisition data. data_labels Allows to specify custom labels. Needs to be precisely 2*num_channels if specified. The order is [Voltage I 0, Voltage Q 0, Voltage I 1, Voltage Q 1, ...], in case real_imag==True, otherwise [Magnitude 0, Phase 0, Magnitude 1, Phase 1, ...]. real_imag If true, the gettable returns I, Q values. Otherwise, magnitude and phase (degrees) are returned. batched Used to indicate if the experiment is performed in batches or in an iterative fashion. max_batch_size: Determines the maximum number of points to acquire when acquiring in batched mode. Can be used to split up a program in parts if required due to hardware constraints. always_initialize: If True, then reinitialize the schedule on each invocation of ``get``. If False, then only initialize the first invocation of ``get``. """ def __init__( self, quantum_device: QuantumDevice, schedule_function: Callable[..., Schedule], schedule_kwargs: dict[str, Any], num_channels: int = 1, data_labels: list[str] | None = None, real_imag: bool = True, batched: bool = False, max_batch_size: int = 1024, always_initialize: bool = True, ) -> None: self._data_labels_specified = data_labels is not None self.always_initialize = always_initialize self.is_initialized = False self._compiled_schedule: CompiledSchedule | None = None self.real_imag = real_imag if self.real_imag: self.name = ["I", "Q"] * num_channels if data_labels: self.label = data_labels else: self.label = [ f"Voltage {iq}{ch}" for ch in range(num_channels) for iq in ["I", "Q"] ] logger.info(f"Auto-generating labels. {self.label}") self.unit = ["V", "V"] * num_channels else: self.name = ["magn", "phase"] * num_channels if data_labels: self.label = data_labels else: self.label = [ f"{val_label}{ch}" for ch in range(num_channels) for val_label in ["Magnitude", "Phase"] ] logger.info(f"Auto-generating labels. {self.label}") self.unit = ["V", "deg"] * num_channels self.batched = batched self.batch_size = max_batch_size # schedule arguments self.schedule_function = schedule_function self.schedule_kwargs = schedule_kwargs self._evaluated_sched_kwargs = {} # the quantum device object containing setup configuration information self.quantum_device = quantum_device # The backend used for compilation. Available as a private variable # to facilitate debugging. Will be assigned upon compilation in self.initialize self._backend = None # Indicates whether compilation is done using the debug mode. # When using with diagnostics report, its value is set to `True`. self._debug_mode: bool = False def __call__(self) -> tuple[float, ...] | tuple[np.ndarray, ...]: """Acquire and return data""" return self.get()
[docs] def _compile(self, sched: Schedule) -> None: """Compile schedule, separated to allow for profiling compilation duration.""" compilation_config = self.quantum_device.generate_compilation_config() compilation_config.debug_mode = self._debug_mode # made into a private variable for debugging and future caching functionality self._backend = compilation_config.backend(name=compilation_config.name) self._compiled_schedule = self._backend.compile( schedule=sched, config=compilation_config )
[docs] def initialize(self) -> None: """ This generates the schedule and uploads the compiled instructions to the hardware using the instrument coordinator. """ logger.debug("Initializing schedule gettable.") self._evaluated_sched_kwargs = _evaluate_parameter_dict(self.schedule_kwargs) # generate a schedule using the evaluated keyword arguments dict self._compile( sched=self.schedule_function( **self._evaluated_sched_kwargs, repetitions=self.quantum_device.cfg_sched_repetitions(), ) ) instr_coordinator = self.quantum_device.instr_instrument_coordinator.get_instr() instr_coordinator.prepare(self._compiled_schedule) self.is_initialized = True
@property
[docs] def compiled_schedule(self) -> CompiledSchedule | None: """Return the schedule used in this class""" return self._compiled_schedule
[docs] def get(self) -> tuple[np.ndarray, ...]: """ Start the experimental sequence and retrieve acquisition data. The data format returned is dependent on the type of acquisitions used in the schedule. These data formats can be found in the :ref:`user guide <sec-user-guide-acquisition-data-schedulegettable>`. Returns ------- : A tuple of acquisition data per acquisition channel as specified above. """ instr_coordinator = self.quantum_device.instr_instrument_coordinator.get_instr() # ensure the instruments are not running and we are starting from a clean state # We switch on allow_failure so that some instruments which require the prepare # before the stop can fail. instr_coordinator.stop(allow_failure=True) if not self.is_initialized or self.always_initialize: self.initialize() if self.compiled_schedule is None: raise RuntimeError( "No compiled schedule was found. Either the schedule was not " "compiled, or the compiled schedule was not assigned to the " "correct attribute." ) instr_coordinator.start() acquired_data = instr_coordinator.retrieve_acquisition() instr_coordinator.stop() if len(acquired_data) == 0: raise RuntimeError( f"InstrumentCoordinator.retrieve_acquisition() " f"('{instr_coordinator.name}') " f"did not return any data, but was expected to return data." ) result = self.process_acquired_data(acquired_data) return result
[docs] def _reshape_data(self, acq_protocol: str, vals: NDArray) -> list[NDArray]: if acq_protocol == "TriggerCount": return [vals.real.astype(np.uint64)] if acq_protocol == "ThresholdedAcquisition": return [vals.real.astype(np.uint32)] if acq_protocol in ( "Trace", "SSBIntegrationComplex", "WeightedIntegratedSeparated", "NumericalSeparatedWeightedIntegration", "NumericalWeightedIntegration", ): ret_val = [] if self.real_imag: ret_val.append(vals.real) ret_val.append(vals.imag) return ret_val else: ret_val.append(np.abs(vals)) ret_val.append(np.angle(vals, deg=True)) return ret_val raise NotImplementedError( f"Acquisition protocol {acq_protocol} is not supported." )
[docs] def process_acquired_data( # noqa: PLR0912 self, acquired_data: Dataset, ) -> tuple[NDArray[np.float_], ...]: """ Reshapes the data as returned from the instrument coordinator into the form accepted by the measurement control. Parameters ---------- acquired_data Data that is returned by instrument coordinator. Returns ------- : A tuple of data, casted to a historical conventions on data format. """ # retrieve the acquisition results return_data = [] # We sort acquisition channels so that the user # has control over the order of the return data. # https://gitlab.com/quantify-os/quantify-scheduler/-/issues/466 sorted_acq_channels: list[Hashable] = sorted(acquired_data.data_vars) for idx, acq_channel in enumerate(sorted_acq_channels): acq_channel_data = acquired_data[acq_channel] acq_protocol = acq_channel_data.attrs["acq_protocol"] num_dims = len(acq_channel_data.dims) if acq_protocol == "Trace" and ( num_dims != 2 or not np.iscomplexobj(acq_channel_data) ): raise AcquisitionProtocolError( f"Data returned by an instrument coordinator component for " f"{acq_protocol} acquisition protocol is expected to be an " f"array of complex numbers with with two dimensions: " f"acquisition index and trace index. This is not the case for " f"acquisition channel {acq_channel}, that has data " f"type {acq_channel_data.dtype} and {num_dims} dimensions: " f"{', '.join(str(dim) for dim in acq_channel_data.dims)}." ) if acq_protocol in ( "SSBIntegrationComplex", "WeightedIntegratedSeparated", "NumericalSeparatedWeightedIntegration", "NumericalWeightedIntegration", "ThresholdedAcquisition", ) and num_dims not in (1, 2): raise AcquisitionProtocolError( f"Data returned by an instrument coordinator component for " f"{acq_protocol} acquisition protocol is expected to be an " f"array of complex numbers with with one or two dimensions: " f"acquisition index and optionally repetition index. This is not the case for " f"acquisition channel {acq_channel}, that has data " f"type {acq_channel_data.dtype} and {num_dims} dimensions: " f"{', '.join(str(dim) for dim in acq_channel_data.dims)}." ) if acq_protocol == "Trace" and acq_channel_data.shape[0] != 1: raise AcquisitionProtocolNotSupportedError( "Trace acquisition protocol with several acquisitions on the " "same acquisition channel is not supported by " "a ScheduleGettable" ) if acq_protocol not in ( "TriggerCount", "Trace", "SSBIntegrationComplex", "WeightedIntegratedSeparated", "NumericalSeparatedWeightedIntegration", "NumericalWeightedIntegration", "ThresholdedAcquisition", ): raise AcquisitionProtocolNotSupportedError( f"ScheduleGettable does not support {acq_protocol}." ) if not self._data_labels_specified and idx != acq_channel: logger.warning( f"Default data_labels may not match the acquisition channel. " f"Element {idx} with label {self.label[idx]} corresponds to " f"acq_channel {acq_channel}, while they were expected to match. To " f"fix this behavior either specify custom data_labels, or ensure " f"your acquisition channels are sequential starting from 0." ) vals = acq_channel_data.to_numpy().reshape((-1,)) if not self.batched and len(vals) != 1: raise ValueError( f"For iterative mode, only one value is expected for each " f"acquisition channel. Got {len(vals)} values for acquisition " f"channel '{acq_channel}' instead." ) return_data.extend(self._reshape_data(acq_protocol, vals)) logger.debug(f"Returning {len(return_data)} values.") return tuple(return_data)
[docs] def initialize_and_get_with_report(self) -> str: """ Create a report that saves all information from this experiment in a zipfile. Run :meth:`~.ScheduleGettable.initialize` and :meth:`~.ScheduleGettable.get` and capture all information from the experiment in a zipfile in the quantify datadir. The basic information in the report includes the schedule, device config and hardware config. The method attempts to compile the schedule, and if it succeeds, it runs the experiment and adds the compiled schedule, a snapshot of the instruments, and logs from the actual hardware (only Qblox instruments supported currently) to the zipfile. A full error trace is also included if any of these steps fail. Returns ------- : A path to the generated report. Directory name includes a flag indicating at which state the experiment and report retrieval stopped. Flags (defined in :func: `~._generate_diagnostics_report`): - ``failed_initialization``: The experiment failed during \ :meth:`~.ScheduleGettable.initialize`. - ``failed_exp``: The experiment initialized failed during \ :meth:`~.ScheduleGettable.get`. - ``failed_connection_to_hw``: The experiment initialized but both \ :meth:`~.ScheduleGettable.get` and \ :meth:`~.InstrumentCoordinator.retrieve_hardware_logs` failed. Connection \ to hardware was likely interrupted during runtime. - ``failed_hw_log_retrieval``: The experiment succeeded but \ :meth:`~.InstrumentCoordinator.retrieve_hardware_logs` failed. - ``completed_exp``: The experiment succeeded. """ if not self.quantum_device.instr_instrument_coordinator: raise AttributeError( "`InstrumentCoordinator` not found in the `QuantumDevice`, please\ add using\ `QuantumDevice.instr_instrument_coordinator(instrument_coordinator)`." ) if not self.quantum_device.hardware_config(): raise AttributeError( "Hardware configuration not found in the `QuantumDevice`, please add\ using `QuantumDevice.hardware_config(hardware_cfg)`." ) exception = None initialized = False acquisition_data = None self._debug_mode = True # Make sure only a compiled schedule generated in the block below is included in # the report self._compiled_schedule = None try: self.initialize() initialized = True acquisition_data = self.get() except: # pylint: disable=bare-except # noqa: E722 exception = sys.exc_info() gettable_config = { "repetitions": self.quantum_device.cfg_sched_repetitions(), "evaluated_schedule_kwargs": self._evaluated_sched_kwargs, } try: schedule = self.schedule_function( **self._evaluated_sched_kwargs, repetitions=self.quantum_device.cfg_sched_repetitions(), ) except ( TypeError ): # Not tested, included for guiding users with common MeasurementControl mistake # noqa: E501 raise TypeError( f"One or more keyword arguments of the schedule function in " f"schedule_kwargs are of unexpected type, please check the arguments " f"and try again.\n{self._evaluated_sched_kwargs=}" ) instrument_coordinator = ( self.quantum_device.instr_instrument_coordinator.get_instr() ) return _generate_diagnostics_report( quantum_device=self.quantum_device, gettable_config=gettable_config, schedule=schedule, instrument_coordinator=instrument_coordinator, # type: ignore compiled_schedule=self.compiled_schedule, acquisition_data=acquisition_data, initialized=initialized, experiment_exception=exception, )
[docs] def _evaluate_parameter_dict(parameters: dict[str, Any]) -> dict[str, Any]: r""" Loop over the keys and values in a dict and replaces parameters with their current value. Parameters ---------- parameters A dictionary containing a mix of :class:`~qcodes.instrument.parameter.Parameter`\s and normal values. Returns ------- : The ``parameters`` dictionary, but with the parameters replaced by their current value. Raises ------ TypeError If a parameter returns None """ evaluated_parameters = dict() for key, val in parameters.items(): if isinstance(val, Parameter): # evaluate the parameter evaluated_parameters[key] = val.get() # verify that the parameter has a value, a missing value typically indicates # that it was not initialized. if evaluated_parameters[key] is None: raise TypeError( f"{key}: parameter {val} returns None. " "It is possible this parameter was not configured correctly." ) else: evaluated_parameters[key] = val return evaluated_parameters