mirror of
https://github.com/m-labs/artiq.git
synced 2024-12-26 03:38:25 +08:00
shuttler: add AFE drivers
This commit is contained in:
parent
67b6588d95
commit
eb08c55abe
@ -3,6 +3,14 @@ import numpy
|
||||
from artiq.language.core import *
|
||||
from artiq.language.types import *
|
||||
from artiq.coredevice.rtio import rtio_output, rtio_input_data
|
||||
from artiq.coredevice import spi2 as spi
|
||||
from artiq.language.units import us
|
||||
|
||||
|
||||
@portable
|
||||
def shuttler_volt_to_mu(volt):
|
||||
# TODO: Check arg, raise exception if exceeds shuttler limit
|
||||
return int(round((1 << 14) * (volt / 20.0))) & 0x3fff
|
||||
|
||||
|
||||
class Config:
|
||||
@ -116,3 +124,140 @@ class Trigger:
|
||||
@kernel
|
||||
def trigger(self, trig_out):
|
||||
rtio_output(self.target_o, trig_out)
|
||||
|
||||
|
||||
RELAY_SPI_CONFIG = (0*spi.SPI_OFFLINE | 1*spi.SPI_END |
|
||||
0*spi.SPI_INPUT | 0*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_CS_RELAY = 1 << 0
|
||||
SPI_CS_LED = 1 << 1
|
||||
SPI_DIV = 4
|
||||
|
||||
ADC_SPI_CONFIG = (0*spi.SPI_OFFLINE | 0*spi.SPI_END |
|
||||
0*spi.SPI_INPUT | 0*spi.SPI_CS_POLARITY |
|
||||
1*spi.SPI_CLK_POLARITY | 1*spi.SPI_CLK_PHASE |
|
||||
0*spi.SPI_LSB_FIRST | 0*spi.SPI_HALF_DUPLEX)
|
||||
|
||||
ADC_CS = 1
|
||||
ADC_SPI_DIV = 32
|
||||
|
||||
|
||||
class Relay:
|
||||
kernel_invariant = {"core", "bus"}
|
||||
|
||||
def __init__(self, dmgr, spi_device, core_device="core"):
|
||||
self.core = dmgr.get(core_device)
|
||||
self.bus = dmgr.get(spi_device)
|
||||
|
||||
@kernel
|
||||
def init(self):
|
||||
self.bus.set_config_mu(
|
||||
RELAY_SPI_CONFIG, 16, SPI_DIV, SPI_CS_RELAY | SPI_CS_LED)
|
||||
|
||||
@kernel
|
||||
def set_led(self, leds: TInt32):
|
||||
self.bus.write(leds << 16)
|
||||
|
||||
|
||||
class ADC:
|
||||
kernel_invariant = {"core", "bus"}
|
||||
|
||||
def __init__(self, dmgr, spi_device, core_device="core"):
|
||||
self.core = dmgr.get(core_device)
|
||||
self.bus = dmgr.get(spi_device)
|
||||
|
||||
@kernel
|
||||
def read_id(self) -> TInt32:
|
||||
self.bus.set_config_mu(
|
||||
ADC_SPI_CONFIG | spi.SPI_END | spi.SPI_INPUT,
|
||||
24, ADC_SPI_DIV, ADC_CS)
|
||||
self.bus.write(0x47 << 24)
|
||||
return (self.bus.read() & 0xFFFF)
|
||||
|
||||
@kernel
|
||||
def read_ch(self, channel: TInt32) -> TFloat:
|
||||
# Always configure Profile 0
|
||||
self.bus.set_config_mu(
|
||||
ADC_SPI_CONFIG | spi.SPI_END, 24, ADC_SPI_DIV, ADC_CS)
|
||||
self.bus.write(0x10 << 24 | (0x8000 | ((channel * 2 + 1) << 4)) << 8)
|
||||
|
||||
# Configure Setup 0
|
||||
# Input buffer must be enabled to use REF pins correctly
|
||||
self.bus.set_config_mu(
|
||||
ADC_SPI_CONFIG | spi.SPI_END, 24, ADC_SPI_DIV, ADC_CS)
|
||||
self.bus.write((0x20 << 24) | (0x1300 << 8))
|
||||
|
||||
# Trigger single conversion
|
||||
self.bus.set_config_mu(
|
||||
ADC_SPI_CONFIG, 24, ADC_SPI_DIV, ADC_CS)
|
||||
self.bus.write((0x01 << 24) | (0x8010 << 8))
|
||||
|
||||
self.bus.set_config_mu(
|
||||
ADC_SPI_CONFIG | spi.SPI_INPUT, 16, ADC_SPI_DIV, ADC_CS)
|
||||
self.bus.write(0x40 << 24)
|
||||
while self.bus.read() & 0x80:
|
||||
delay(10*us)
|
||||
self.bus.write(0x40 << 24)
|
||||
|
||||
delay(10*us)
|
||||
self.bus.set_config_mu(
|
||||
ADC_SPI_CONFIG | spi.SPI_END | spi.SPI_INPUT,
|
||||
32, ADC_SPI_DIV, ADC_CS)
|
||||
self.bus.write(0x44 << 24)
|
||||
|
||||
adc_code = self.bus.read() & 0xFFFFFF
|
||||
return ((adc_code / (1 << 23)) - 1) * 2.5 / 0.1
|
||||
|
||||
@kernel
|
||||
def calibrate(self, volts, trigger, config, samples=[-5.0, 0.0, 5.0]):
|
||||
assert len(volts) == 16
|
||||
assert len(samples) > 1
|
||||
|
||||
measurements = [0.0] * len(samples)
|
||||
|
||||
for ch in range(16):
|
||||
# Find the average slope rate and offset
|
||||
for i in range(len(samples)):
|
||||
self.core.break_realtime()
|
||||
volts[ch].set_waveform(
|
||||
shuttler_volt_to_mu(samples[i]), 0, 0, 0)
|
||||
trigger.trigger(1 << ch)
|
||||
delay(1*us)
|
||||
measurements[i] = self.read_ch(ch ^ 1)
|
||||
|
||||
# Find the average output slope
|
||||
print(measurements)
|
||||
slope_sum = 0.0
|
||||
for i in range(len(samples) - 1):
|
||||
slope_sum += (measurements[i+1] - measurements[i])/(samples[i+1] - samples[i])
|
||||
slope_avg = slope_sum / (len(samples) - 1)
|
||||
print(slope_avg)
|
||||
|
||||
print("Suitable gain in Shuttler:")
|
||||
gain_code = int32(1 / slope_avg * (2 ** 16)) & 0xffff
|
||||
print(gain_code)
|
||||
|
||||
# Scale the measurements by 1/slope, find average offset
|
||||
offset_sum = 0.0
|
||||
for i in range(len(samples)):
|
||||
offset_sum += (measurements[i] / slope_avg) - samples[i]
|
||||
offset_avg = offset_sum / len(samples)
|
||||
print(offset_avg)
|
||||
|
||||
print("Suitable offset in Shuttler:")
|
||||
offset_code = shuttler_volt_to_mu(-offset_avg)
|
||||
print(offset_code)
|
||||
|
||||
self.core.break_realtime()
|
||||
config.set_gain(ch, gain_code)
|
||||
|
||||
delay_mu(int64(self.core.ref_multiplier))
|
||||
assert config.get_gain(ch) == gain_code
|
||||
|
||||
self.core.break_realtime()
|
||||
config.set_offset(ch, offset_code)
|
||||
|
||||
delay_mu(int64(self.core.ref_multiplier))
|
||||
assert config.get_offset(ch) == offset_code
|
||||
|
Loading…
Reference in New Issue
Block a user