m-labs · Feb 23, 2016 · Feb 23, 2016 · Feb 23, 2016
Showing with 54 additions and 14 deletions.

+47 −8 artiq/sim/devices.py

+1 −1 examples/master/repository/coredevice_examples/pdq2_simple.py

+0 −1 examples/master/repository/coredevice_examples/transport.py

+6 −4 examples/sim/al_spectroscopy.py
diff --git a/artiq/sim/devices.py b/artiq/sim/devices.py
@@ -1,7 +1,6 @@
 from random import Random
 
-from artiq.language.core import delay, kernel
-from artiq.language import units
+from artiq.language.core import delay, at_mu, kernel
 from artiq.sim import time
 
 
@@ -16,6 +15,7 @@ def run(self, k_function, k_args, k_kwargs):
         self._level -= 1
         if self._level == 0:
             print(time.manager.format_timeline())
+            time.manager.timeline.clear()
         return r
 
 
@@ -27,19 +27,58 @@ def __init__(self, dmgr, name):
         self.prng = Random()
 
     @kernel
-    def wait_edge(self):
-        duration = self.prng.randrange(0, 20)*units.ms
-        time.manager.event(("wait_edge", self.name, duration))
+    def gate_rising(self, duration):
+        time.manager.event(("gate_rising", self.name, duration))
         delay(duration)
 
     @kernel
-    def count_gate(self, duration):
-        result = self.prng.randrange(0, 100)
-        time.manager.event(("count_gate", self.name, duration, result))
+    def gate_falling(self, duration):
+        time.manager.event(("gate_falling", self.name, duration))
+        delay(duration)
+
+    @kernel
+    def gate_both(self, duration):
+        time.manager.event(("gate_both", self.name, duration))
         delay(duration)
+
+    @kernel
+    def count(self):
+        result = self.prng.randrange(0, 100)
+        time.manager.event(("count", self.name, result))
+        return result
+
+    @kernel
+    def timestamp_mu(self):
+        result = time.manager.get_time_mu()
+        result += self.prng.randrange(100, 1000)
+        time.manager.event(("timestamp_mu", self.name, result))
+        at_mu(result)
         return result
 
 
+class Output:
+    def __init__(self, dmgr, name):
+        self.core = dmgr.get("core")
+        self.name = name
+
+    @kernel
+    def set_o(self, value):
+        time.manager.event(("set", self.name, value))
+
+    @kernel
+    def pulse(self, duration):
+        time.manager.event(("pulse", self.name, duration))
+        delay(duration)
+
+    @kernel
+    def on(self):
+        self.set_o(True)
+
+    @kernel
+    def off(self):
+        self.set_o(False)
+
+
 class WaveOutput:
     def __init__(self, dmgr, name):
         self.core = dmgr.get("core")

diff --git a/examples/master/repository/coredevice_examples/pdq2_simple.py b/examples/master/repository/coredevice_examples/pdq2_simple.py
@@ -36,7 +36,7 @@ def one(self):
         return self.pmt.count()
 
     def run(self):
-        offsets = np.linspace(0, 10, 3)
+        offsets = np.arange(0, 3)
         for o in offsets:
             self.setup(o)
             self.one()
diff --git a/examples/master/repository/coredevice_examples/transport.py b/examples/master/repository/coredevice_examples/transport.py
@@ -82,7 +82,6 @@ def repeat(self):
 
     def scan(self, stops):
         for s in stops:
-            self.histogram = [0 for _ in range(self.bins)]
             # non-kernel, build frames
             # could also be rpc'ed from repeat()
             self.calc_waveforms(s)

diff --git a/examples/sim/al_spectroscopy.py b/examples/sim/al_spectroscopy.py
@@ -20,7 +20,8 @@ def build(self):
     def run(self):
         state_0_count = 0
         for count in range(100):
-            self.mains_sync.wait_edge()
+            self.mains_sync.gate_rising(1*s/60)
+            at_mu(self.mains_sync.timestamp_mu() + 100*us)
             delay(10*us)
             self.laser_cooling.pulse(100*MHz, 100*us)
             delay(5*us)
@@ -34,9 +35,10 @@ def run(self):
                 delay(5*us)
                 with parallel:
                     self.state_detection.pulse(100*MHz, 10*us)
-                    photon_count = self.pmt.count_gate(10*us)
-                if (photon_count < self.photon_limit_low
-                        or photon_count > self.photon_limit_high):
+                    self.pmt.gate_rising(10*us)
+                photon_count = self.pmt.count()
+                if (photon_count < self.photon_limit_low or
+                        photon_count > self.photon_limit_high):
                     break
             if photon_count < self.photon_limit_low:
                 state_0_count += 1