# Repository: https://gitlab.com/quantify-os/quantify-scheduler
# Licensed according to the LICENCE file on the main branch
# pylint: disable=invalid-name
"""NV-center-specific operations for use with the quantify_scheduler."""
from typing import Literal, Tuple, Union, Optional, Hashable
from .operation import Operation
from ..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):
super().__init__(name=f"ChargeReset {', '.join(qubits)}")
self.data.update(
{
"name": f"ChargeReset {', '.join(qubits)}",
"gate_info": {
"unitary": None,
"plot_func": "quantify_scheduler.schedules._visualization."
+ "circuit_diagram.reset",
"tex": r"$NV^-$",
"qubits": list(qubits),
"operation_type": "charge_reset",
},
}
)
self.update()
def __str__(self) -> str:
qubits = map(lambda x: f"'{x}'", self.data["gate_info"]["qubits"])
return f'{self.__class__.__name__}({",".join(qubits)})'
[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: Optional[Hashable] = None,
acq_index: Union[Tuple[int, ...], int] = None,
# These are the currently supported acquisition protocols.
acq_protocol: Literal[
"Trace",
"TriggerCount",
None,
] = None,
bin_mode: BinMode = None,
):
# this if else statement a workaround to support multiplexed measurements (#262)
# this snippet has some automatic behaviour that is error prone.
# see #262
if len(qubits) == 1:
if acq_index is None:
acq_index = 0
else:
if isinstance(acq_index, int):
acq_index = [
acq_index,
] * len(qubits)
elif acq_index is None:
# defaults to writing the result of all qubits to acq_index 0.
# note that this will result in averaging data together if multiple
# measurements are present in the same schedule (#262)
acq_index = list(0 for i in range(len(qubits)))
plot_func = (
"quantify_scheduler.schedules._visualization.circuit_diagram.acq_meter_text"
)
super().__init__(f"CRCount {', '.join(qubits)}")
self.data.update(
{
"name": f"CRCount {', '.join(qubits)}",
"gate_info": {
"unitary": None,
"plot_func": plot_func,
"tex": r"CR",
"qubits": list(qubits),
"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"]
qubits = map(lambda x: f"'{x}'", gate_info["qubits"])
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(qubits)}, '
f"acq_channel={acq_channel}, "
f"acq_index={acq_index}, "
f'acq_protocol="{acq_protocol}", '
f"bin_mode={str(bin_mode)})"
)
