srenblad
/
artiq
forked from M-Labs/artiq
1
0
Fork 0
Code Pull Requests Activity

shuttler: implement gain & offset register access

This commit is contained in:
occheung 2023-09-13 16:45:48 -07:00 committed by Sébastien Bourdeauducq
parent 1bb7e9ceef
commit 67b6588d95
1 changed files with 32 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
artiq/coredevice/shuttler.py
View File

@ -2,20 +2,47 @@ import numpy
from artiq.language.core import * from artiq.language.core import *
from artiq.language.types import * from artiq.language.types import *
from artiq.coredevice.rtio import rtio_output from artiq.coredevice.rtio import rtio_output, rtio_input_data
class Config: class Config:
kernel_invariants = {"core", "channel", "target_o"} kernel_invariants = {
"core", "channel", "target_base", "target_read",
"target_gain", "target_offset", "target_clr"
}
def __init__(self, dmgr, channel, core_device="core"): def __init__(self, dmgr, channel, core_device="core"):
self.core = dmgr.get(core_device) self.core = dmgr.get(core_device)
self.channel = channel self.channel = channel
self.target_o = channel << 8 self.target_base = channel << 8
self.target_read = 1 << 6
self.target_gain = 0 * (1 << 4)
self.target_offset = 1 * (1 << 4)
self.target_clr = 1 * (1 << 5)
@kernel @kernel
def set_config(self, config): def set_clr(self, clr):
rtio_output(self.target_o, config) rtio_output(self.target_base | self.target_clr, clr)
@kernel
def set_gain(self, channel, gain):
rtio_output(self.target_base | self.target_gain | channel, gain)
@kernel
def get_gain(self, channel):
rtio_output(self.target_base | self.target_gain |
self.target_read | channel, 0)
return rtio_input_data(self.channel)
@kernel
def set_offset(self, channel, offset):
rtio_output(self.target_base | self.target_offset | channel, offset)
@kernel
def get_offset(self, channel):
rtio_output(self.target_base | self.target_offset |
self.target_read | channel, 0)
return rtio_input_data(self.channel)
class Volt: class Volt: