From 8e206e92f538139a10fd668321b66c122a6ed7d0 Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Tue, 7 Jun 2022 00:15:51 +0800
Subject: [PATCH] test_pulse_rate: port to NAC3

---
 artiq/test/coredevice/test_rtio.py | 35 ++++++++++++++++++++----------
 1 file changed, 23 insertions(+), 12 deletions(-)

diff --git a/artiq/test/coredevice/test_rtio.py b/artiq/test/coredevice/test_rtio.py
index 20a8d0ed6..6c9969399 100644
--- a/artiq/test/coredevice/test_rtio.py
+++ b/artiq/test/coredevice/test_rtio.py
@@ -9,6 +9,8 @@ from math import sqrt
 from artiq.experiment import *
 from artiq.test.hardware_testbench import ExperimentCase
 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_analyzer import (StoppedMessage, OutputMessage, InputMessage,
                                             decode_dump, get_analyzer_dump)
@@ -118,27 +120,36 @@ class ClockGeneratorLoopback(EnvExperiment):
         self.set_dataset("count", self.loop_clock_in.count(now_mu()))
 
 
+@nac3
 class PulseRate(EnvExperiment):
+    core: KernelInvariant[Core]
+    ttl_out: KernelInvariant[TTLOut]
+
     def build(self):
         self.setattr_device("core")
         self.setattr_device("ttl_out")
 
+    @rpc
+    def set_pulse_rate(self, pulse_rate: float):
+        self.set_dataset("pulse_rate", pulse_rate)
+
     @kernel
     def run(self):
         self.core.reset()
-        dt = self.core.seconds_to_mu(300*ns)
-        while True:
-            for i in range(10000):
-                try:
-                    self.ttl_out.pulse_mu(dt)
-                    delay_mu(dt)
-                except RTIOUnderflow:
-                    dt += 1
-                    self.core.break_realtime()
-                    break
+        dt = self.core.seconds_to_mu(300.*ns)
+        i = 10000
+        while i > 0:
+            try:
+                self.ttl_out.pulse_mu(dt)
+                delay_mu(dt)
+            except RTIOUnderflow:
+                dt += int64(1)
+                i = 10000
+                self.core.break_realtime()
             else:
-                self.set_dataset("pulse_rate", self.core.mu_to_seconds(dt))
-                return
+                i -= 1
+        self.set_pulse_rate(self.core.mu_to_seconds(dt))
+
 
 
 class PulseRateAD9914DDS(EnvExperiment):