Source code for quantify_scheduler.backends.qblox.operation_handling.virtual

# Repository: https://gitlab.com/quantify-os/quantify-scheduler
# Licensed according to the LICENCE file on the main branch
"""Classes for handling operations that are neither pulses nor acquisitions."""

from typing import Any, Dict

import numpy as np
from quantify_scheduler.backends.qblox import constants, helpers, q1asm_instructions
from quantify_scheduler.backends.qblox.operation_handling.base import IOperationStrategy
from quantify_scheduler.backends.qblox.qasm_program import QASMProgram
from quantify_scheduler.backends.types import qblox as types
from quantify_scheduler.backends.qblox.conditional import FeedbackTriggerCondition


[docs] class IdleStrategy(IOperationStrategy): """ Defines the behavior for an operation that does not produce any output. Parameters ---------- operation_info : quantify_scheduler.backends.types.qblox.OpInfo The operation info that corresponds to this operation. """ def __init__(self, operation_info: types.OpInfo): self._op_info = operation_info @property
[docs] def operation_info(self) -> types.OpInfo: """Property for retrieving the operation info.""" return self._op_info
[docs] def generate_data(self, wf_dict: Dict[str, Any]): """Returns None as no waveforms are generated in this strategy.""" return None
[docs] def insert_qasm(self, qasm_program: QASMProgram): """ Add the assembly instructions for the Q1 sequence processor that corresponds to this operation. Not an abstractmethod, since it is allowed to use the IdleStrategy directly (e.g. for IdlePulses), but can be overridden in subclass to add some assembly instructions despite not outputting any data. Parameters ---------- qasm_program The QASMProgram to add the assembly instructions to. """
[docs] class NcoPhaseShiftStrategy(IdleStrategy): """ Strategy for operation that does not produce any output, but rather applies a phase shift to the NCO. Implemented as ``set_ph_delta`` and an ``upd_param`` of 8 ns, leading to a total duration of 8 ns before the next command can be issued. """
[docs] def insert_qasm(self, qasm_program: QASMProgram): """ Inserts the instructions needed to shift the NCO phase by a specific amount. Parameters ---------- qasm_program The QASMProgram to add the assembly instructions to. """ phase = self.operation_info.data.get("phase_shift") if phase != 0: phase_arg = helpers.get_nco_phase_arguments(phase) qasm_program.emit( q1asm_instructions.INCR_NCO_PHASE_OFFSET, phase_arg, comment=f"increment nco phase by {phase:.2f} deg", ) qasm_program.emit( q1asm_instructions.UPDATE_PARAMETERS, constants.NCO_SET_PH_DELTA_WAIT, comment="apply nco phase shift", ) qasm_program.elapsed_time += constants.NCO_SET_PH_DELTA_WAIT
[docs] class NcoResetClockPhaseStrategy(IdleStrategy): """ Strategy for operation that does not produce any output, but rather resets the phase of the NCO. """
[docs] def insert_qasm(self, qasm_program: QASMProgram): """ Inserts the instructions needed to reset the NCO phase. Parameters ---------- qasm_program The QASMProgram to add the assembly instructions to. """ reset_clock_phase = self.operation_info.data.get("reset_clock_phase") if reset_clock_phase is None: raise KeyError( "NcoResetClockPhaseStrategy called, " "but reset_clock_phase not present in operation_info.data" ) qasm_program.emit(q1asm_instructions.RESET_PHASE)
[docs] class NcoSetClockFrequencyStrategy(IdleStrategy): """ Strategy for operation that does not produce any output, but rather sets the frequency of the NCO. Implemented as ``set_freq`` and an ``upd_param`` of 8 ns, leading to a total duration of 8 ns before the next command can be issued. """
[docs] def insert_qasm(self, qasm_program: QASMProgram): """ Inserts the instructions needed to set the NCO frequency. Parameters ---------- qasm_program The QASMProgram to add the assembly instructions to. """ clock_freq_new = self.operation_info.data.get("clock_freq_new") clock_freq_old = self.operation_info.data.get("clock_freq_old") interm_freq_old = self.operation_info.data.get("interm_freq_old") if clock_freq_old is None or np.isnan(clock_freq_old): raise RuntimeError( f"Clock '{self.operation_info.data.get('clock')}' has an undefined " f"initial frequency ({clock_freq_old=}); " f"ensure this resource has been added to the schedule or to the device " f"config." ) if interm_freq_old is None: raise RuntimeError( f"Clock '{self.operation_info.data.get('clock')}' has an undefined " f"associated intermodulation frequency ({interm_freq_old=}); make " f"sure an 'interm_freq' is supplied or that 'mix_lo' is set to true in " f"the hardware config." ) iterm_freq_new = interm_freq_old + clock_freq_new - clock_freq_old frequency_args = helpers.get_nco_set_frequency_arguments(iterm_freq_new) qasm_program.emit( q1asm_instructions.SET_FREQUENCY, frequency_args, comment=f"set nco frequency to {iterm_freq_new:e} Hz", ) qasm_program.emit( q1asm_instructions.UPDATE_PARAMETERS, constants.NCO_SET_FREQ_WAIT, comment="apply nco frequency change", ) qasm_program.elapsed_time += constants.NCO_SET_FREQ_WAIT
[docs] class AwgOffsetStrategy(IdleStrategy): """ Strategy for compiling a DC voltage offset instruction. The generated Q1ASM contains only the ``set_awg_offs`` instruction and no ``upd_param`` instruction. """
[docs] def insert_qasm(self, qasm_program: QASMProgram): """ Add the Q1ASM instruction for a DC voltage offset. Parameters ---------- qasm_program : QASMProgram The QASMProgram to add the assembly instructions to. """ path_I_amp = qasm_program.expand_awg_from_normalised_range( val=self.operation_info.data["offset_path_I"], immediate_size=constants.IMMEDIATE_SZ_OFFSET, param="offset_awg_path_I", operation=self.operation_info, ) path_Q_amp = qasm_program.expand_awg_from_normalised_range( val=self.operation_info.data["offset_path_Q"], immediate_size=constants.IMMEDIATE_SZ_OFFSET, param="offset_awg_path_Q", operation=self.operation_info, ) qasm_program.emit( q1asm_instructions.SET_AWG_OFFSET, path_I_amp, path_Q_amp, comment=f"setting offset for {self.operation_info.name}", )
[docs] class ResetFeedbackTriggersStrategy(IdleStrategy): """Strategy for resetting the count of feedback trigger addresses."""
[docs] def insert_qasm(self, qasm_program: QASMProgram): """ Add the assembly instructions for the Q1 sequence processor that corresponds to this pulse. Parameters ---------- qasm_program The QASMProgram to add the assembly instructions to. """ duration = round(1e9 * self.operation_info.data.get("duration")) qasm_program.emit( q1asm_instructions.FEEDBACK_TRIGGERS_RST, duration, comment="reset trigger count", ) qasm_program.elapsed_time += duration
[docs] class UpdateParameterStrategy(IdleStrategy): """Strategy for compiling an "update parameters" real-time instruction."""
[docs] def insert_qasm(self, qasm_program: QASMProgram): """ Add the ``upd_param`` assembly instruction for the Q1 sequence processor. Parameters ---------- qasm_program : QASMProgram The QASMProgram to add the assembly instructions to. """ qasm_program.emit( q1asm_instructions.UPDATE_PARAMETERS, constants.GRID_TIME, ) qasm_program.elapsed_time += constants.GRID_TIME
[docs] class LoopStrategy(IdleStrategy): """ Strategy for compiling a "Loop" control flow instruction. Empty as it is used for isinstance. """
[docs] class ConditionalStrategy(IdleStrategy): """Strategy for compiling a "Conditional" control flow instruction.""" def __init__( self, operation_info: types.OpInfo, trigger_condition: FeedbackTriggerCondition ): super().__init__(operation_info=operation_info) self.trigger_condition = trigger_condition
[docs] class ControlFlowReturnStrategy(IdleStrategy): """ Strategy for compiling "ControlFlowReturn" control flow instruction. Empty as it is used for isinstance. """