m-labs · Feb 23, 2016 · Feb 23, 2016
Showing with 45 additions and 0 deletions.

+2 −0 artiq/wavesynth/coefficients.py

+43 −0 examples/master/repository/coredevice_examples/pdq2_simple.py
diff --git a/artiq/wavesynth/coefficients.py b/artiq/wavesynth/coefficients.py
@@ -73,6 +73,8 @@ def build_segment(durations, coefficients, target="bias",
                 if cdj or abs(yijk) or not compress:
                     cdj.append(float(yijk))
             cdj.reverse()
+            if not cdj:
+                cdj.append(float(yij[0]))
             cd.append({target: {variable: cdj}})
         yield {"duration": int(dxi), "channel_data": cd}
 

diff --git a/examples/master/repository/coredevice_examples/pdq2_simple.py b/examples/master/repository/coredevice_examples/pdq2_simple.py
@@ -0,0 +1,43 @@
+import numpy as np
+
+from artiq.experiment import *
+from artiq.wavesynth.coefficients import build_segment
+
+
+class PDQ2Simple(EnvExperiment):
+    """Set PDQ2 voltages."""
+    def build(self):
+        self.setattr_device("core")
+        self.setattr_device("pmt")
+        self.setattr_device("electrodes")
+
+        # 1 device, 3 board each, 3 dacs each
+        self.u = np.arange(4*3)[None, :, None]*.1
+
+    def setup(self, offset):
+        self.electrodes.disarm()
+        self.load = self.electrodes.create_frame()
+        segment = self.load.create_segment()
+        for line in build_segment([100], self.u + offset):
+            segment.add_line(**line)
+        self.detect = self.electrodes.create_frame()
+        segment = self.detect.create_segment()
+        for line in build_segment([100], -self.u + offset):
+            segment.add_line(**line)
+        self.electrodes.arm()
+
+    @kernel
+    def one(self):
+        self.load.advance()
+        delay(1*ms)
+        self.detect.advance()
+        delay(1*ms)
+        self.pmt.gate_rising(100*us)
+        return self.pmt.count()
+
+    def run(self):
+        offsets = np.linspace(0, 10, 3)
+        for o in offsets:
+            self.setup(o)
+            print(self.electrodes.get_program())
+            self.one()