# Repository: https://gitlab.com/quantify-os/quantify-scheduler
# Licensed according to the LICENCE file on the main branch
"""NV-center-specific operations for use with the quantify_scheduler."""
from __future__ import annotations
from typing import TYPE_CHECKING, Hashable, Iterable, Literal
from .operation import Operation, _generate_acq_indices_for_gate
if TYPE_CHECKING:
from quantify_scheduler.enums import BinMode
[docs]
class ChargeReset(Operation):
r"""
Prepare a NV to its negative charge state NV$^-$.
Create a new instance of ChargeReset operation that is used to initialize the
charge state of an NV center.
Parameters
----------
qubit
The qubit to charge-reset. NB one or more qubits can be specified, e.g.,
:code:`ChargeReset("qe0")`, :code:`ChargeReset("qe0", "qe1", "qe2")`, etc..
"""
def __init__(self, *qubits: str) -> None:
device_elements = qubits
super().__init__(name=f"ChargeReset {', '.join(device_elements)}")
self.data.update(
{
"name": f"ChargeReset {', '.join(device_elements)}",
"gate_info": {
"unitary": None,
"plot_func": "quantify_scheduler.schedules._visualization."
+ "circuit_diagram.reset",
"tex": r"$NV^-$",
"device_elements": list(device_elements),
"operation_type": "charge_reset",
},
}
)
self.update()
def __str__(self) -> str:
device_elements = map(lambda x: f"'{x}'", self.data["gate_info"]["device_elements"])
return f'{self.__class__.__name__}({",".join(device_elements)})'
[docs]
class CRCount(Operation):
r"""
Operate ionization and spin pump lasers for charge and resonance counting.
Gate level description for an optical CR count measurement.
The measurement is compiled according to the type of acquisition specified
in the device configuration.
Parameters
----------
qubits
The qubits you want to measure
acq_channel
Only for special use cases.
By default (if None): the acquisition channel specified in the device element is used.
If set, this acquisition channel is used for this measurement.
acq_index
Index of the register where the measurement is stored.
If None specified, it will default to a list of zeros of len(qubits)
acq_protocol
Acquisition protocol (currently ``"TriggerCount"`` and ``"Trace"``)
are supported. If ``None`` is specified, the default protocol is chosen
based on the device and backend configuration.
bin_mode
The binning mode that is to be used. If not None, it will overwrite
the binning mode used for Measurements in the quantum-circuit to
quantum-device compilation step.
"""
def __init__(
self,
*qubits: str,
acq_channel: Hashable | None = None,
acq_index: tuple[int, ...] | int = None,
# These are the currently supported acquisition protocols.
acq_protocol: Literal[
"Trace",
"TriggerCount",
None,
] = None,
bin_mode: BinMode = None,
) -> None:
device_elements = qubits
acq_index: int | Iterable[int] = _generate_acq_indices_for_gate(
device_elements=device_elements, acq_index=acq_index
)
plot_func = "quantify_scheduler.schedules._visualization.circuit_diagram.acq_meter_text"
super().__init__(f"CRCount {', '.join(device_elements)}")
self.data.update(
{
"name": f"CRCount {', '.join(device_elements)}",
"gate_info": {
"unitary": None,
"plot_func": plot_func,
"tex": r"CR",
"device_elements": list(device_elements),
"acq_channel_override": acq_channel,
"acq_index": acq_index,
"acq_protocol": acq_protocol,
"bin_mode": bin_mode,
"operation_type": "cr_count",
},
}
)
self._update()
def __str__(self) -> str:
gate_info = self.data["gate_info"]
device_elements = map(lambda x: f"'{x}'", gate_info["device_elements"])
acq_channel = gate_info["acq_channel_override"]
acq_index = gate_info["acq_index"]
acq_protocol = gate_info["acq_protocol"]
bin_mode = gate_info["bin_mode"]
return (
f'{self.__class__.__name__}({",".join(device_elements)}, '
f"acq_channel={acq_channel}, "
f"acq_index={acq_index}, "
f'acq_protocol="{acq_protocol}", '
f"bin_mode={str(bin_mode)})"
)
