Source code for quantify_scheduler.device_under_test.edge

# Repository: https://gitlab.com/quantify-os/quantify-scheduler
# Licensed according to the LICENCE file on the main branch
"""The module contains definitions for edges."""
from __future__ import annotations

from typing import TYPE_CHECKING, Any

from qcodes.instrument.base import Instrument

from quantify_scheduler.device_under_test.device_element import DeviceElement
from quantify_scheduler.helpers.importers import export_python_object_to_path_string

if TYPE_CHECKING:
    from quantify_scheduler.backends.graph_compilation import DeviceCompilationConfig


[docs] class Edge(Instrument): """ Create an Edge. This class encapsulates the connection information between DeviceElements in the QuantumDevice. It provides an interface for the QuantumDevice to generate the edge information for use in the device compilation step. See :class:`quantify_scheduler.device_under_test.composite_square_edge` for an example edge implementation. """ def __init__( self, parent_element_name: str, child_element_name: str, **kwargs, # noqa: ANN003 ) -> None: self._parent_element_name = parent_element_name self._child_element_name = child_element_name super().__init__(name=f"{parent_element_name}_{child_element_name}", **kwargs) def __getstate__(self) -> dict[str, Any]: """ Serialize :class:`~Edge` into a dictionary. Serialization is performed by converting submodules of the object into the dictionary containing the parent and child element names of this edge and a dict for each submodule containing its parameter names and corresponding values. """ snapshot = self.snapshot() edge_data = { "parent_element_name": self._parent_element_name, "child_element_name": self._child_element_name, } for submodule_name, submodule_data in snapshot["submodules"].items(): edge_data[submodule_name] = { name: data["value"] for name, data in submodule_data["parameters"].items() } state = { "deserialization_type": export_python_object_to_path_string(self.__class__), "mode": "__init__", "data": edge_data, } return state @property
[docs] def parent_device_element(self) -> Instrument: """The parent DeviceElement connected by the edge.""" return self.find_instrument( name=self._parent_element_name, instrument_class=DeviceElement )
@property
[docs] def child_device_element(self) -> Instrument: """The child DeviceElement connected by the edge.""" return self.find_instrument( name=self._child_element_name, instrument_class=DeviceElement )
[docs] def generate_edge_config(self) -> DeviceCompilationConfig: """ Generate the device configuration for an edge. This method is intended to be used when this object is part of a device object containing multiple elements. """ raise NotImplementedError