Source code for atomiq.components.dummies

"""
Dummies
=======

This module gathers differnt dummy components that mimic hardware but do nothing. They are mainly used for debugging
atomiq
"""
from __future__ import annotations

from atomiq.components.primitives import Component, Parametrizable, Measurable
from atomiq.components.electronics.adc import ADC
from atomiq.components.electronics.voltagesource import DAC
from atomiq.components.electronics.rfsource import RFSource

from artiq.experiment import kernel
from artiq.language.core import now_mu
from artiq.language.types import TFloat, TArray, TInt32




class DummyADC(ADC):

    kernel_invariants = {"return_value"}

    def __init__(self, return_value: TFloat, *args, **kwargs):
        ADC.__init__(self, *args, **kwargs)
        self.return_value = return_value

    @kernel
    def _measure(self, target_arr: TArray(TFloat)) -> TArray(TFloat):
        for i in range(self.num_chan):
            target_arr[i] = self.return_value
        return target_arr





class DummyRFSource(RFSource):
    """A dummy RFSource used for debugging
    """

    @kernel
    def _set_frequency(self, frequency):
        self.frequency = frequency
        self.experiment.log.info(self.identifier + ": Set frequency to {0:.0f} MHz", [frequency/1e6])

    @kernel
    def _set_amplitude(self, amplitude):
        self.amplitude = amplitude
        self.experiment.log.info(self.identifier + ": Set amplitude to {0:.3f}", [amplitude])

    @kernel
    def _set_phase(self, phase):
        self.phase = phase
        self.experiment.log.info(self.identifier + ": Set phase to {0}/pi", [phase/3.14])





class DummyDAC(DAC):
    def __init__(self, *args, **kwargs):
        DAC.__init__(self, *args, **kwargs)

        self.values = [0.0]*self.num_chan



    @kernel
    def set_channel_voltage(self, channel: TInt32, voltage: TFloat):
        self.values[channel] = voltage




    @kernel
    def update(self):
        self.experiment.log.info("time: {0}", [self.experiment.core.mu_to_seconds(now_mu())])
        self.experiment.log.info(self.identifier + ": Setting output voltages {0}", [self.values])






class DummyActor(Component, Parametrizable):
    def __init__(self, *args, **kwargs):
        Component.__init__(self, *args, **kwargs)
        Parametrizable.__init__(self, ["amplitude"])

        self.amplitude = 0.0



    @kernel
    def set_amplitude(self, amp: TFloat):
        self.amplitude = amp
        self.experiment.log.info("DummyActor("+self.identifier+"): setting amplitude {0}", [amp])






class DummyMonitor(Component, Measurable):
    def __init__(self, dummy_actor: DummyActor, *args, **kwargs):
        Component.__init__(self, *args, **kwargs)
        Measurable.__init__(self, ["voltage"])

        self.dummy_actor = dummy_actor



    @kernel
    def get_voltage(self) -> TFloat:
        return 0.2*self.dummy_actor.amplitude