pxi6733: minor mediator fixes

sbourdeauducq · sbourdeauducq · commit 2921bd65019f · 2015-09-04T18:53:50.000-06:00
diff --git a/artiq/devices/pxi6733/mediator.py b/artiq/devices/pxi6733/mediator.py
@@ -58,6 +58,9 @@ def add_line(self, duration, channel_data):
             raise ArmError
         self.lines.append((duration, channel_data))
 
+    def get_sample_count(self):
+        return sum(duration for duration, _ in self.lines)
+
     @kernel
     def advance(self):
         if self.frame.invalidated:
@@ -107,13 +110,13 @@ def _invalidate(self):
         self.invalidated = True
 
     def _get_samples(self):
-        program = [
+        program = [[
             {
                 "dac_divider": 1,
                 "duration": duration,
                 "channel_data": channel_data,
-            } for duration, channel_data in segment.lines 
-                for segment in self.segments]
+            } for segment in self.segments
+                for duration, channel_data in segment.lines]]
         synth = Synthesizer(self.daqmx.channel_count, program)
         synth.select(0)
         # not setting any trigger flag in the program causes the whole
@@ -145,7 +148,7 @@ def __init__(self, dmgr, daqmx_device, clock_device, channel_count,
         self.daqmx = dmgr.get(daqmx_device)
         self.clock = dmgr.get(clock_device)
         self.channel_count = channel_count
-        if self.sample_rate_in_mu:
+        if sample_rate_in_mu:
             self.sample_rate = sample_rate
         else:
             self.sample_rate = self.clock.frequency_to_ftw(sample_rate)
