Source code for quantify_scheduler.enums

# Repository: https://gitlab.com/quantify-os/quantify-scheduler
# Licensed according to the LICENCE file on the main branch
"""Enums for quantify-scheduler."""

from typing import Any

try:
    from enum import StrEnum, unique  # type: ignore
except ImportError:
    from enum import Enum, unique



    class StrEnum(Enum):
        """Enum that can be directly serialized to string."""

        def __str__(self) -> str:
            return str(self.value)

        def __eq__(self, other: Any) -> bool:  # noqa: ANN401
            if type(self) is type(other):
                return str(self) == str(other)
            elif other in [str(val) for val in type(self)]:
                return str(self) == other
            else:
                return NotImplemented

        def __lt__(self, other: Any) -> bool:  # noqa: ANN401
            if type(self) is type(other):
                return str(self) < str(other)
            elif other in [str(val) for val in type(self)]:
                return str(self) < other
            else:
                return NotImplemented

        def __gt__(self, other: Any) -> bool:  # noqa: ANN401
            if type(self) is type(other):
                return str(self) > str(other)
            elif other in [str(val) for val in type(self)]:
                return str(self) > other
            else:
                return NotImplemented

        def __le__(self, other: Any) -> bool:  # noqa: ANN401
            if type(self) is type(other):
                return str(self) <= str(other)
            elif other in [str(val) for val in type(self)]:
                return str(self) <= other
            else:
                return NotImplemented

        def __ge__(self, other: Any) -> bool:  # noqa: ANN401
            if type(self) is type(other):
                return str(self) >= str(other)
            elif other in [str(val) for val in type(self)]:
                return str(self) >= other
            else:
                return NotImplemented

        def __key__(self) -> str:
            return str(self)

        def __hash__(self) -> int:
            return hash(self.__key__())



@unique


class BinMode(StrEnum):  # type: ignore
    """
    Describes how to handle `Acquisitions` that write to the same `AcquisitionIndex`.

    A BinMode is a property of an `AcquisitionChannel` that describes how to
    handle multiple
    :class:`~quantify_scheduler.operations.acquisition_library.Acquisition` s
    that write data to the same `AcquisitionIndex` on a channel.

    The most common use-case for this is when iterating over multiple
    repetitions of a :class:`~quantify_scheduler.schedules.schedule.Schedule`
    When the BinMode is set to `APPEND` new entries will be added as a list
    along the `repetitions` dimension.

    When the BinMode is set to `AVERAGE` the outcomes are averaged together
    into one value.

    Note that not all `AcquisitionProtocols` and backends support all possible
    BinModes. For more information, please see the :ref:`sec-acquisition-protocols`
    reference guide and some of the Qblox-specific :ref:`acquisition details
    <sec-qblox-acquisition-details>`.
    """



    APPEND = "append"



    AVERAGE = "average"



    FIRST = "first"



    DISTRIBUTION = "distribution"



    SUM = "sum"


    # N.B. in principle it is possible to specify other behaviours for
    # BinMode such as `SUM` or `OVERWRITE` but these are not
    # currently supported by any backend.




class TimeSource(StrEnum):  # type: ignore
    """
    Selects the timetag data source for timetag (trace) acquisitions.

    See :class:`~quantify_scheduler.operations.acquisition_library.Timetag` and
    :class:`~quantify_scheduler.operations.acquisition_library.TimetagTrace`.
    """



    FIRST = "first"



    SECOND = "second"



    LAST = "last"






class TimeRef(StrEnum):  # type: ignore
    """
    Selects the event that counts as a time reference (i.e. t=0) for timetags.

    See :class:`~quantify_scheduler.operations.acquisition_library.Timetag` and
    :class:`~quantify_scheduler.operations.acquisition_library.TimetagTrace`.
    """



    START = "start"



    END = "end"



    FIRST = "first"



    TIMESTAMP = "timestamp"



    PORT = "port"






class TriggerCondition(StrEnum):  # type: ignore
    """Comparison condition for the thresholded trigger count acquisition."""



    LESS_THAN = "less_than"



    GREATER_THAN_EQUAL_TO = "greater_than_equal_to"






class DualThresholdedTriggerCountLabels(StrEnum):  # type: ignore
    """
    All suffixes for the feedback trigger labels that can be used by
    DualThresholdedTriggerCount.
    """



    LOW = "low"



    MID = "mid"



    HIGH = "high"



    INVALID = "invalid"