M-Labs
/
artiq
mirror of https://github.com/m-labs/artiq.git
2
0
Fork 2
Code Issues Projects Releases Wiki Activity

kasli_tester: add Urukul support

This commit is contained in:
Sebastien Bourdeauducq 2018-05-18 23:18:03 +08:00
parent 72aef5799e
commit bdfd993818
1 changed files with 38 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
artiq/examples/kasli_basic/repository/kasli_tester.py
View File

@ -30,6 +30,8 @@ class KasliTester(EnvExperiment):
self.leds = dict() self.leds = dict()
self.ttl_outs = dict() self.ttl_outs = dict()
self.ttl_ins = dict() self.ttl_ins = dict()
self.urukul_cplds = dict()
self.urukuls = dict()
ddb = self.get_device_db() ddb = self.get_device_db()
for name, desc in ddb.items(): for name, desc in ddb.items():
@ -43,6 +45,10 @@ class KasliTester(EnvExperiment):
self.ttl_outs[name] = dev self.ttl_outs[name] = dev
elif (module, cls) == ("artiq.coredevice.ttl", "TTLInOut"): elif (module, cls) == ("artiq.coredevice.ttl", "TTLInOut"):
self.ttl_ins[name] = self.get_device(name) self.ttl_ins[name] = self.get_device(name)
elif (module, cls) == ("artiq.coredevice.urukul", "CPLD"):
self.urukul_cplds[name] = self.get_device(name)
elif (module, cls) == ("artiq.coredevice.ad9910", "AD9910"):
self.urukuls[name] = self.get_device(name)
# Remove Urukul control signals from TTL outs (tested separately) # Remove Urukul control signals from TTL outs (tested separately)
ddb = self.get_device_db() ddb = self.get_device_db()
@ -61,6 +67,7 @@ class KasliTester(EnvExperiment):
self.leds = sorted(self.leds.items(), key=lambda x: x[1].channel) self.leds = sorted(self.leds.items(), key=lambda x: x[1].channel)
self.ttl_outs = sorted(self.ttl_outs.items(), key=lambda x: x[1].channel) self.ttl_outs = sorted(self.ttl_outs.items(), key=lambda x: x[1].channel)
self.ttl_ins = sorted(self.ttl_ins.items(), key=lambda x: x[1].channel) self.ttl_ins = sorted(self.ttl_ins.items(), key=lambda x: x[1].channel)
self.urukuls = sorted(self.urukuls.items(), key=lambda x: x[1].sw.channel)
@kernel @kernel
def test_led(self, led): def test_led(self, led):
@ -102,8 +109,7 @@ class KasliTester(EnvExperiment):
print("pulses corresponds to its number in the group.") print("pulses corresponds to its number in the group.")
print("Press ENTER when done.") print("Press ENTER when done.")
ttl_outs = list(self.ttl_outs) for ttl_chunk in chunker(self.ttl_outs, 4):
for ttl_chunk in chunker(ttl_outs, 4):
print("Testing TTL outputs: {}.".format(", ".join(name for name, dev in ttl_chunk))) print("Testing TTL outputs: {}.".format(", ".join(name for name, dev in ttl_chunk)))
self.test_ttl_out_chunk([dev for name, dev in ttl_chunk]) self.test_ttl_out_chunk([dev for name, dev in ttl_chunk])
@ -133,9 +139,39 @@ class KasliTester(EnvExperiment):
else: else:
print("FAILED") print("FAILED")
@kernel
def init_urukul(self, cpld):
self.core.break_realtime()
cpld.init()
@kernel
def setup_urukul(self, channel, frequency):
self.core.break_realtime()
channel.set(frequency*MHz)
channel.sw.on()
channel.set_att(6.)
# We assume that RTIO channels for switches are grouped by card.
def test_urukuls(self):
print("*** Testing Urukul DDSes.")
print("Initializing CPLDs...")
for name, cpld in sorted(self.urukul_cplds.items(), key=lambda x: x[0]):
print(name + "...")
self.init_urukul(cpld)
print("...done")
print("Frequencies:")
for card_n, channels in enumerate(chunker(self.urukuls, 4)):
for channel_n, (channel_name, channel_dev) in enumerate(channels):
frequency = 10*(card_n + 1) + channel_n
print("{}\t{}MHz".format(channel_name, frequency))
self.setup_urukul(channel_dev, frequency)
print("Press ENTER when done.")
input()
def run(self): def run(self):
print("****** Kasli system tester ******") print("****** Kasli system tester ******")
print("") print("")
self.test_leds() self.test_leds()
self.test_ttl_outs() self.test_ttl_outs()
self.test_ttl_ins() self.test_ttl_ins()
self.test_urukuls()