forked from M-Labs/artiq
kasli_tester: add Urukul support
This commit is contained in:
parent
72aef5799e
commit
bdfd993818
|
@ -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()
|
||||||
|
|
Loading…
Reference in New Issue