Source code for artiq.coredevice.mirny

"""RTIO driver for Mirny (4 channel GHz PLLs)
"""

from artiq.language.core import kernel, delay, portable
from artiq.language.units import us

from numpy import int32

from artiq.coredevice import spi2 as spi


SPI_CONFIG = (
    0 * spi.SPI_OFFLINE
    | 0 * spi.SPI_END
    | 0 * spi.SPI_INPUT
    | 1 * spi.SPI_CS_POLARITY
    | 0 * spi.SPI_CLK_POLARITY
    | 0 * spi.SPI_CLK_PHASE
    | 0 * spi.SPI_LSB_FIRST
    | 0 * spi.SPI_HALF_DUPLEX
)

# SPI clock write and read dividers
SPIT_WR = 4
SPIT_RD = 16

SPI_CS = 1

WE = 1 << 24

# supported CPLD code version
PROTO_REV_MATCH = 0x0


[docs] class Mirny: """ Mirny PLL-based RF generator. :param spi_device: SPI bus device :param refclk: Reference clock (SMA, MMCX or on-board 100 MHz oscillator) frequency in Hz :param clk_sel: Reference clock selection. Valid options are: "XO" - onboard crystal oscillator; "SMA" - front-panel SMA connector; "MMCX" - internal MMCX connector. Passing an integer writes it as ``clk_sel`` in the CPLD's register 1. The effect depends on the hardware revision. :param core_device: Core device name (default: "core") """ kernel_invariants = {"bus", "core", "refclk", "clk_sel_hw_rev"} def __init__(self, dmgr, spi_device, refclk=100e6, clk_sel="XO", core_device="core"): self.core = dmgr.get(core_device) self.bus = dmgr.get(spi_device) # reference clock frequency self.refclk = refclk if not (10 <= self.refclk / 1e6 <= 600): raise ValueError("Invalid refclk") # reference clock selection try: self.clk_sel_hw_rev = { # clk source: [reserved, reserved, v1.1, v1.0] "xo": [-1, -1, 0, 0], "mmcx": [-1, -1, 3, 2], "sma": [-1, -1, 2, 3], }[clk_sel.lower()] except AttributeError: # not a string, fallback to int if clk_sel & 0x3 != clk_sel: raise ValueError("Invalid clk_sel") from None self.clk_sel_hw_rev = [clk_sel] * 4 except KeyError: raise ValueError("Invalid clk_sel") from None self.clk_sel = -1 # board hardware revision self.hw_rev = 0 # v1.0: 3, v1.1: 2 # TODO: support clk_div on v1.0 boards
[docs] @kernel def read_reg(self, addr): """Read a register""" self.bus.set_config_mu( SPI_CONFIG | spi.SPI_INPUT | spi.SPI_END, 24, SPIT_RD, SPI_CS ) self.bus.write((addr << 25)) return self.bus.read() & int32(0xFFFF)
[docs] @kernel def write_reg(self, addr, data): """Write a register""" self.bus.set_config_mu(SPI_CONFIG | spi.SPI_END, 24, SPIT_WR, SPI_CS) self.bus.write((addr << 25) | WE | ((data & 0xFFFF) << 8))
[docs] @kernel def init(self, blind=False): """ Initialize and detect Mirny. Select the clock source based the board's hardware revision. Raise ValueError if the board's hardware revision is not supported. :param blind: Verify presence and protocol compatibility. Raise ValueError on failure. """ reg0 = self.read_reg(0) self.hw_rev = reg0 & 0x3 if not blind: if (reg0 >> 2) & 0x3 != PROTO_REV_MATCH: raise ValueError("Mirny PROTO_REV mismatch") delay(100 * us) # slack # select clock source self.clk_sel = self.clk_sel_hw_rev[self.hw_rev] if self.clk_sel < 0: raise ValueError("Hardware revision not supported") self.write_reg(1, (self.clk_sel << 4)) delay(1000 * us)
[docs] @portable(flags={"fast-math"}) def att_to_mu(self, att): """Convert an attenuation setting in dB to machine units. :param att: Attenuation setting in dB. :return: Digital attenuation setting. """ code = int32(255) - int32(round(att * 8)) if code < 0 or code > 255: raise ValueError("Invalid Mirny attenuation!") return code
[docs] @kernel def set_att_mu(self, channel, att): """Set digital step attenuator in machine units. :param att: Attenuation setting, 8 bit digital. """ self.bus.set_config_mu(SPI_CONFIG | spi.SPI_END, 16, SPIT_WR, SPI_CS) self.bus.write(((channel | 8) << 25) | (att << 16))
[docs] @kernel def set_att(self, channel, att): """Set digital step attenuator in SI units. This method will write the attenuator settings of the selected channel. .. seealso:: :meth:`set_att_mu` :param channel: Attenuator channel (0-3). :param att: Attenuation setting in dB. Higher value is more attenuation. Minimum attenuation is 0*dB, maximum attenuation is 31.5*dB. """ self.set_att_mu(channel, self.att_to_mu(att))
[docs] @kernel def write_ext(self, addr, length, data, ext_div=SPIT_WR): """Perform SPI write to a prefixed address""" self.bus.set_config_mu(SPI_CONFIG, 8, SPIT_WR, SPI_CS) self.bus.write(addr << 25) self.bus.set_config_mu(SPI_CONFIG | spi.SPI_END, length, ext_div, SPI_CS) if length < 32: data <<= 32 - length self.bus.write(data)