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

test_pulse_rate: port to NAC3

This commit is contained in:
Sebastien Bourdeauducq 2022-06-07 00:15:51 +08:00
parent 85693e45e7
commit 8e206e92f5
1 changed files with 23 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
artiq/test/coredevice/test_rtio.py
View File

@ -9,6 +9,8 @@ from math import sqrt
from artiq.experiment import * from artiq.experiment import *
from artiq.test.hardware_testbench import ExperimentCase from artiq.test.hardware_testbench import ExperimentCase
from artiq.coredevice import exceptions from artiq.coredevice import exceptions
from artiq.coredevice.core import Core
from artiq.coredevice.ttl import TTLOut
from artiq.coredevice.comm_mgmt import CommMgmt from artiq.coredevice.comm_mgmt import CommMgmt
from artiq.coredevice.comm_analyzer import (StoppedMessage, OutputMessage, InputMessage, from artiq.coredevice.comm_analyzer import (StoppedMessage, OutputMessage, InputMessage,
decode_dump, get_analyzer_dump) decode_dump, get_analyzer_dump)
@ -118,27 +120,36 @@ class ClockGeneratorLoopback(EnvExperiment):
self.set_dataset("count", self.loop_clock_in.count(now_mu())) self.set_dataset("count", self.loop_clock_in.count(now_mu()))
@nac3
class PulseRate(EnvExperiment): class PulseRate(EnvExperiment):
core: KernelInvariant[Core]
ttl_out: KernelInvariant[TTLOut]
def build(self): def build(self):
self.setattr_device("core") self.setattr_device("core")
self.setattr_device("ttl_out") self.setattr_device("ttl_out")
@rpc
def set_pulse_rate(self, pulse_rate: float):
self.set_dataset("pulse_rate", pulse_rate)
@kernel @kernel
def run(self): def run(self):
self.core.reset() self.core.reset()
dt = self.core.seconds_to_mu(300*ns) dt = self.core.seconds_to_mu(300.*ns)
while True: i = 10000
for i in range(10000): while i > 0:
try: try:
self.ttl_out.pulse_mu(dt) self.ttl_out.pulse_mu(dt)
delay_mu(dt) delay_mu(dt)
except RTIOUnderflow: except RTIOUnderflow:
dt += 1 dt += int64(1)
self.core.break_realtime() i = 10000
break self.core.break_realtime()
else: else:
self.set_dataset("pulse_rate", self.core.mu_to_seconds(dt)) i -= 1
return self.set_pulse_rate(self.core.mu_to_seconds(dt))
class PulseRateAD9914DDS(EnvExperiment): class PulseRateAD9914DDS(EnvExperiment):