nsqdriver 0.12.11__cp312-cp312-win_amd64.whl → 0.12.13__cp312-cp312-win_amd64.whl

nsqdriver/NS_DDS_v3.py

@@ -1,7 +1,9 @@
 import copy
 from enum import Enum
+from math import ceil
 from collections import namedtuple
-from waveforms import Waveform, wave_eval
+from waveforms import Waveform, wave_eval, WaveVStack
+from waveforms.waveform import _zero
 from waveforms.math.signal import getFTMatrix, shift
 import nsqdriver.nswave as nw
 
@@ -37,28 +39,6 @@ except ImportError as e:
 DEBUG_PRINT = False
 
 
-@nw.kernel
-def program_cap(param: nw.Var):
-    #
-    nw.wait_for_trigger()
-    i: nw.Var
-    # param: [[100e-9, 1e-6], [200e-9, 1e-6]]
-    for i in param:
-        nw.wait(i[0])
-        nw.capture(i[1], i[2], i[3])
-
-
-ProbeSegment = namedtuple('ProbeSegment', ['start', 'stop', 'freq'])
-
-
-CaptureCmd = namedtuple('CaptureCmd', ['start', 'ad_duration', 'delay', 'da_duration', 'freqs'])
-
-
-class DemodulateMode(str, Enum):
-    MORE_QUBIT = 'more_qubit'
-    COMPLEX_SEQ = 'complex_seq'
-
-
 def get_coef(coef_info, sampleRate):
     start, stop = coef_info['start'], coef_info['stop']
     numberOfPoints = int(
@@ -103,56 +83,69 @@ def get_coef(coef_info, sampleRate):
     return np.asarray(wList), fList, numberOfPoints, phases, round((stop - t0) * sampleRate), t
 
 
-def granularity4ns(delay):
-    # 4ns取整
-    out_delay = round(
-        delay // 4e-9 * 4e-9, 10
-    )
-    return out_delay
-
-
-def generate_para(device, ch, coef_info):
-    global RES_MAP
-    global PROGRAME_PARA
-    RES_MAP = []
-    PROGRAME_PARA = []
-
-    res_coef = get_coef(coef_info, 4e9)
-    da_ad_delay = 150e-9  # ad相对于da的延迟
-    ad_start = []
-    freq_map = []
-
-    for num, (orig, i) in enumerate(zip(coef_info["wList"], res_coef[0])):
-        RES_MAP.append([None, None])  # 第一个字段标记第几个频点，第二个字段标记第几次采集
-        if orig['Delta'] not in freq_map:
-            # Delta 字段是解模频点
-            # 新增加了频点
-            freq_map.append(orig['Delta'])
-        RES_MAP[-1][0] = freq_map.index(orig['Delta'])
-
-        # 5.43这个频率点包含了第
-        ad_line = res_coef[-1][np.nonzero(i)]
-        ad_width = granularity4ns(ad_line[-1] - ad_line[0] + 0.25e-9)
-        da_width = ad_width  # 设置播放时长=采样时长
-        if orig["t0"] not in ad_start:
-            # 这个采集的起始点以前没配置过
-            PROGRAME_PARA.append([orig["t0"], ad_width, da_ad_delay, da_width])
-            ad_start.append(orig["t0"])
-        # else:
-        #     # 这个采集的起点以前配置过
-        #     cap_num += 1
-        # cap_map[num] = [cap_num, ]
-        RES_MAP[-1][1] = len(ad_start) - 1  # len(ad_start) - 1 表明了当前是第几次采集
-    # print(programe_para, freq_map)
-    device.set("Program", program_cap(PROGRAME_PARA), ch)
-    device.set("FreqList", freq_map, ch)
-
-
-def get_coef_res(iq_res):
-    res = []
-    for (freq_num, cap_num) in RES_MAP:
-        res.append(iq_res[freq_num][cap_num::len(PROGRAME_PARA)])
-    return res
+def get_demod_envelope(coef_info, demod_map, sampleRate):
+    start, stop = coef_info['start'], coef_info['stop']
+    t0 = coef_info['t0']
+    numberOfPoints = int(
+        (stop - start) * sampleRate)
+    if numberOfPoints % 64 != 0:
+        numberOfPoints = numberOfPoints + 64 - numberOfPoints % 64
+    t = np.arange(numberOfPoints) / sampleRate + start
+    demod_map_list = demod_map
+    weight_sum = np.empty(numberOfPoints)
+    for weight in demod_map_list:
+        if isinstance(weight, np.ndarray):
+            pass
+        else:
+            if isinstance(weight, str):
+                fun = wave_eval(weight)
+            elif isinstance(weight, Waveform):
+                fun = weight
+            else:
+                raise TypeError(f'Unsupported type {weight}')
+            weight = fun(t)
+            weight_sum += weight
+    return weight_sum
+
+
+
+
+@nw.kernel
+def program_cap(param: nw.Var):
+    #
+    nw.wait_for_trigger()
+    i: nw.Var
+    # param: [[100e-9, 1e-6], [200e-9, 1e-6]]
+    # nw.wait(150e-9)
+    for i in param:
+        nw.wait(i[0])
+        nw.capture(i[1], 0, 0)
+
+
+@nw.kernel
+def program_da(p: nw.Var):
+    i: nw.Var
+    e: nw.Var
+    nw.init_frame(0, 0)
+    nw.wait_for_trigger()
+    # nw.reset_frame()
+    e = nw.ins_envelope(p[0][1])
+    for i in p:
+        nw.wait(i[0])
+        # e = nw.ins_envelope(i[1])
+        nw.play_wave(e, 1, 0, 0)
+
+
+ProbeSegment = namedtuple('ProbeSegment', ['start', 'stop', 'freq', 'demod', 'idx'])
+
+CaptureCmd = namedtuple('CaptureCmd', [
+    'start', 'ad_duration', 'delay', 'da_duration', 'freqs', 'delays', 'demod_wave_list', 'idx_list'
+])
+
+
+class DemodulateMode(str, Enum):
+    MORE_QUBIT = 'more_qubit'
+    COMPLEX_SEQ = 'complex_seq'
 
 
 class Driver(BaseDriver):
@@ -222,15 +215,22 @@ class Driver(BaseDriver):
         self.handle = None
         self.model = 'NS_MCI'  # 默认为设备名字
         self.srate = 8e9
+        self.ad_srate = 4e9
         self.addr = addr
         self.timeout = timeout
         self.chs = set()  # 记录配置过的ch通道
         self.IQ_cache = {}
         self.coef_cache = {}
         self.res_maps = {}
+        self.demod_maps = {}
         self.probe_da_wave = {}
-        self.capture_cmds: "dict[int, CaptureCmd]" = {}
+        # self.probe_delay = 32e-9
+        self.probe_delay = 0
+        self.capture_cmds: "dict[int, list[CaptureCmd]]" = {}
+        self.capture_cali_param: "dict[int, np.ndarray]" = {}
+        self.capture_points: "dict[int, np.ndarray]" = {}
         self.demodulate_mode = DemodulateMode.MORE_QUBIT
+        self.demode_calculus: "dict[int, np.ndarray]" = {}
 
     def open(self, **kw):
         """
@@ -242,40 +242,42 @@ class Driver(BaseDriver):
         DArate = 8e9
         ADrate = 4e9
         sysparam = {
-            "MixMode": 1,
+            "MixMode": 2,
             "RefClock": "out",
             "DArate": DArate,
             "ADrate": ADrate,
             "CaptureMode": 0,
             "INMixMode": 2,  # 4～6 GHz 取 1， 6 ～ 8 GHz 取 2
         }
+        sysparam.update(kw.get('system_parameter', {}))
 
         device = MCIDriver(self.addr, self.timeout)
         device.open(system_parameter=sysparam)
         self.handle = device
 
     def granularity4ns(self, delay):
-        # 4ns取整
-        out_delay = round(
-            delay // 4e-9 * 4e-9, 10
-        )
-        return out_delay
+        # points_4ns = 16  # self.ad_srate*4e-6
+        return delay // 4 * 4
 
     @staticmethod
-    def get_sequence_in_time(coef_info: dict) -> list[CaptureCmd]:
+    def _delay2_phase(delay, freq):
+        return 2 * np.pi * freq * (delay * 1e-9)
+
+    def in_sequence_in_time(self, coef_info: dict) -> list[CaptureCmd]:
         w_list = coef_info.get('wList', [])
         time_segments: "list[ProbeSegment]" = []
 
-        for wave in w_list:
-            t0 = wave['t0']
+        for idx, wave in enumerate(w_list):
+            t0 = int(round(wave['t0'] * 1e9))
             weight_expr = wave['weight']
 
             # 假设 weight 表达式格式为 "square(X) >> Y"，我们提取实际时间宽度
             # duration = float(weight_expr.split('>>')[1].strip())
             _start, _stop, _ = wave_eval(weight_expr).bounds
+            _start, _stop = int(round(_start * 1e9)), int(round(_stop * 1e9))
 
             # 将区间加入列表
-            seg = ProbeSegment(t0+_start, t0+_stop, wave['Delta'])
+            seg = ProbeSegment(t0 + _start, t0 + _stop, wave['Delta'], weight_expr, idx)
             time_segments.append(seg)
 
         # 按起始时间排序
@@ -284,61 +286,154 @@ class Driver(BaseDriver):
         # 结果存储
         non_overlapping_segments: list[CaptureCmd] = []
         current_start, current_end = time_segments[0].start, time_segments[0].stop
-        current_cmd = CaptureCmd(0, 0, 0, 0, [time_segments[0].freq])
+        current_cmd = CaptureCmd(0, 0, 0, 0, [time_segments[0].freq], [0.],[time_segments[0].demod],
+                                 [time_segments[0].idx])
         pointer = 0
-
         for seg in time_segments[1:]:
             if seg.start > current_end:
                 # 如果不重叠，保存当前段并移动到下一段
-                current_cmd = current_cmd._replace(start=current_start-pointer)
+                if pointer == 0:
+                    current_cmd = current_cmd._replace(start=current_start)
+                else:
+                    current_cmd = current_cmd._replace(start=current_start - self.probe_delay)
                 current_cmd = current_cmd._replace(ad_duration=current_end - current_start)
-                current_cmd = current_cmd._replace(delay=230e-9)
+                current_cmd = current_cmd._replace(delay=self.probe_delay)
                 current_cmd = current_cmd._replace(da_duration=current_end - current_start)
+                current_cmd = current_cmd._replace(demod_wave_list=[time_segments[0].demod])
                 non_overlapping_segments.append(current_cmd)
 
-                current_cmd = CaptureCmd(0, 0, 0, 0, [seg.freq])
-                pointer += current_end
+                current_cmd = CaptureCmd(0, 0, 0, 0, [seg.freq], [0.], [seg.demod], [seg.idx])
+                pointer = current_end
                 current_start, current_end = seg.start, seg.stop
             else:
                 # 如果有重叠，扩展当前段
                 current_end = max(current_end, seg.stop)
+                current_cmd.idx_list.append(seg.idx)
                 current_cmd.freqs.append(seg.freq)
+                current_cmd.demod_wave_list.append(seg.demod)
+                # 由delay换算解缠绕相位
+                current_cmd.delays.append(seg.start - current_start)
         else:
             # 添加最后一个段
-            current_cmd = current_cmd._replace(start=current_start - pointer)
+            current_cmd = current_cmd._replace(start=current_start - self.probe_delay)
             current_cmd = current_cmd._replace(ad_duration=current_end - current_start)
-            current_cmd = current_cmd._replace(delay=230e-9)
+            current_cmd = current_cmd._replace(delay=self.probe_delay)
             current_cmd = current_cmd._replace(da_duration=current_end - current_start)
             non_overlapping_segments.append(current_cmd)
         return non_overlapping_segments
 
-    def generate_para(self, coef_info, ch):
-        res_map = []
+    def generate_in_program(self, coef_info, ch):
         freq_map = []
+        demod_wave_map = []
         seq_param = []
 
-        self.capture_cmds[ch] = seq = self.get_sequence_in_time(coef_info)
+        self.capture_cmds[ch] = seq = self.in_sequence_in_time(coef_info)
 
         for segment in seq:
+            demod_wave_map.extend(segment.demod_wave_list)
+            demod_wave_map = list(set(demod_wave_map))
             freq_map.extend(segment.freqs)
-        freq_map = list(set(freq_map))
+            freq_map = list(set(freq_map))
 
+        _t_end = 0
+        res_map = [[]]*len(coef_info['wList'])
+        phase_map = [0]*len(coef_info['wList'])
+        points_map = [0]*len(coef_info['wList'])
         for cap_num, segment in enumerate(seq):
+            _align_start = self.granularity4ns(segment.start)  # 向前取整
+            _start_diff = segment.start - _align_start
+            _align_end = ceil((segment.start+segment.ad_duration)/4)*4  #向上取整
             seq_param.append([
-                segment.start, segment.ad_duration, segment.delay, segment.da_duration
+                (_align_start-_t_end)*1e-9,
+                (_align_end-_align_start)*1e-9,
+                segment.delay*1e-9,
+                (_align_end-_align_start)*1e-9,
             ])
-            for freq in segment.freqs:
-                res_map.append([freq_map.index(freq), cap_num])
+            _t_end = _align_end
+            for idx, delay, freq, demod_wave in zip(segment.idx_list, segment.delays, segment.freqs,
+                                        segment.demod_wave_list):
+                res_map[idx] = [freq_map.index(freq), cap_num]
+                phase_map[idx] = self._delay2_phase(delay + _start_diff, freq)  # 向前取整的缩进加上起始时间的差值来计算相位
+                points_map[idx] = segment.ad_duration * 1e-9 * self.ad_srate
 
         self.res_maps[ch] = res_map
-        self.handle.set("Program", program_cap(seq_param), ch)
-        self.handle.set("FreqList", freq_map, ch)
-        self.handle.set("TimeWidth", 2e-6, ch)
+        self.capture_cali_param[ch] = np.exp(-1j * np.array(phase_map)).reshape((-1, 1))
+        self.capture_points[ch] = np.array(points_map).reshape((-1, 1))
+        return program_cap(seq_param), freq_map, demod_wave_map
+
+    def out_sequence_in_time(self, wave_list: list):
+        last_start = wave_list[0][0]
+        last_stop = wave_list[0][1]
+        temp_w = [wave_list[0][2]]
+        _res = []
+
+        for idx, (start, stop, seg) in enumerate(wave_list[1:]):
+            if start > last_stop:
+                _res.append([last_start, last_stop, np.hstack(temp_w)])
+                last_start = start
+                last_stop = stop
+                temp_w.clear()
+                temp_w.append(seg)
+            else:
+                last_stop = max(last_stop, stop)
+                temp_w.append(seg)
+        else:
+            _res.append([last_start, last_stop, np.hstack(temp_w)])
+        return _res
+
+    def gen_wave_frag(self, x, wave: "Waveform"):
+        range_list = np.searchsorted(x, wave.bounds)
+        # ret = np.zeros_like(x)
+        ret = []
+        start, stop = 0, 0
+        for i, stop in enumerate(range_list):
+            if start < stop and wave.seq[i] != _zero:
+                _w = copy.deepcopy(wave)
+                _w.start = start / self.srate
+                _w.stop = stop / self.srate
+                part = _w.sample(self.srate)
+                part = part if part is None else part[:(stop-start)]
+                ret.append((start, stop, part))
+            start = stop
+        else:
+            if not ret:
+                ret.append((0, 128, np.zeros((128,))))
+        return ret
+
+    def generate_out_program(self, _wave, ch):
+        align_points = 32  # 4ns*8e9
+        if isinstance(_wave, WaveVStack):
+            _wave = _wave.simplify()
+        if len(_wave.seq) == 1 and _wave.seq[0] == _zero:
+            wave_list = [(0, 128, np.zeros((128,)))]
+        else:
+            _wave.stop = _wave.bounds[-2]
+            wave_list = self.gen_wave_frag(
+                np.linspace(_wave.start, _wave.stop, int((_wave.stop - _wave.start) * self.srate)), _wave)
+        # print(f'generate_out_program: {_wave.start=}, {_wave.stop=}, {wave_list=}, {ch=}')
+        _t_end = 0
+        para = []
+        wave = self.out_sequence_in_time(wave_list)
+
+        for i in wave:
+            wait = (i[0] - _t_end)
+            align_wait = wait // align_points * align_points
+            zero_num = wait - align_wait
+            align_end = ceil(i[1] / align_points) * align_points
+            align_wave = [np.zeros(zero_num), i[2], np.zeros([align_end - i[1]])]
+            para.append([align_wait / self.srate, np.hstack(align_wave)])
+            _t_end = align_end
+        return program_da(para)
 
     def get_coef_res(self, iq_res, ch):
         res = []
         for (freq_num, cap_num) in self.res_maps[ch]:
             res.append(iq_res[freq_num][cap_num::len(self.capture_cmds[ch])])
+        # 采样点归一化
+        res = np.array(res) / self.demode_calculus[ch]
+        # 校准相位
+        res *= self.capture_cali_param[ch]
+
         return res
 
     def close(self, **kw):
@@ -358,24 +453,29 @@ class Driver(BaseDriver):
 
     def write(self, name: str, value, **kw):
         channel = kw.get('ch', 1)
-        print(name, value, "write" * 4)
+        # print(f'NS_DDS_v3 write: {name=}, {channel=}')
         if name in {'Coefficient'}:
-            # data, f_list, numberOfPoints, phases, points, _ = get_coef(value, 4e9)
-
-            # print('DemoPoints'*10, points, channel)
-            # self.handle.set('DemoPoints', points, channel)      ###############   16ns-oscillation problem
-            # self.handle.set('DemodulationParam', data, channel)
             coef_info = value
             self.chs.add(channel)
-            self.generate_para(coef_info, channel)
+            kernel, freq_map, demod_wave_map = self.generate_in_program(coef_info, channel)
+            self.handle.set("ProgramIN", kernel, channel)
+            demode_wave = get_demod_envelope(coef_info, demod_wave_map, sampleRate=8e9)
+            self.demode_calculus[channel] = np.sum(demode_wave)
+            self.handle.set("DemodulationParam", demode_wave, channel)
+            self.handle.set('TimeWidth', 2e-6, channel)
+
+            self.handle.set("FreqList", freq_map, channel)
             self.coef_cache.update({channel: coef_info})
-        # elif name in {'Waveform'} and isinstance(value, waveforms.Waveform):
-        #     self.probe_da_wave[channel] = value
         elif name in {
             'CaptureMode', 'SystemSync', 'ResetTrig', 'TrigPeriod',
             'TrigFrom'
         }:
             pass
+        elif name in {
+            'GenWave', 'Waveform'
+        } and isinstance(value, Waveform):
+            kernel_da = self.generate_out_program(value, channel)
+            self.handle.set("ProgramOUT", kernel_da, channel)
         else:
             if name in {"Shot"}:
                 self.shots = value
@@ -383,18 +483,13 @@ class Driver(BaseDriver):
 
     def read(self, name: str, **kw):
         channel = kw.get('ch', 1)
-        print(name, kw, "READ " * 6)
         if name in {"IQ"}:
-            # if channel not in self.chs:
-            #     result = self.IQ_cache[channel]
-            #     print("result.shape, cache", result)
-            # else:
-            iq_res = self.handle.get("IQ", channel, round(self.shots * len(self.capture_cmds[channel])))  # / self.points
-
-            # print(np.array(iq_res).shape, "iq shape")
-            result = np.array(self.get_coef_res(iq_res, channel)).T
-            # print(result.shape, "result.shape," * 3)
-            if len(self.chs) != 0:
+            iq_res = self.handle.get(
+                "IQ", channel, round(self.shots * len(self.capture_cmds[channel]))
+            )
+
+            result = self.get_coef_res(iq_res, channel).T
+            if len(self.chs) != 0 and channel in self.chs:
                 self.chs.remove(channel)
             # self.IQ_cache.update({channel: result})
             if len(self.chs) == 0:
@@ -402,3 +497,95 @@ class Driver(BaseDriver):
         else:
             result = self.handle.get(name, channel)
         return result
+
+
+if __name__ == '__main__':
+    co = {'start': 0.0, 'stop': 10e-06, 'wList': [
+        {'Delta': 6796613333.333333, 'phase': -0.0, 'weight': 'square(8e-07) >> 4e-07', 'window': (0, 1024), 'w': None,
+         't0': 1.5e-08, 'phi': 2.4311851282940524, 'threshold': 9.645718574523926},
+        {'Delta': 6965129999.666667, 'phase': -0.0, 'weight': 'square(8e-07) >> 4e-07', 'window': (0, 1024), 'w': None,
+         't0': 2.345e-06, 'phi': 2.1739656328752264, 'threshold': 20.36802101135254},
+        {'Delta': 6866353333.333333, 'phase': -0.0, 'weight': 'square(8e-07) >> 4e-07', 'window': (0, 1024), 'w': None,
+         't0': 2.345e-06, 'phi': 1.851749364542847, 'threshold': 21.65827751159668},
+        {'Delta': 6796613333.333333, 'phase': -0.0, 'weight': 'square(8e-07) >> 4e-07', 'window': (0, 1024), 'w': None,
+         't0': 5.175e-06, 'phi': 2.4311851282940524, 'threshold': 9.645718574523926},
+        {'Delta': 6965129999.666667, 'phase': -0.0, 'weight': 'square(8e-07) >> 4e-07', 'window': (0, 1024), 'w': None,
+         't0': 5.175e-06, 'phi': 2.1739656328752264, 'threshold': 20.36802101135254},
+        {'Delta': 6866353333.333333, 'phase': -0.0, 'weight': 'square(8e-07) >> 4e-07', 'window': (0, 1024), 'w': None,
+         't0': 5.175e-06, 'phi': 1.851749364542847, 'threshold': 21.65827751159668},
+        {'Delta': 6796613333.333333, 'phase': -0.0, 'weight': 'square(8e-07) >> 4e-07', 'window': (0, 1024), 'w': None,
+         't0': 7.805e-06, 'phi': 2.4311851282940524, 'threshold': 9.645718574523926},
+        {'Delta': 6796613333.333333, 'phase': -0.0, 'weight': 'square(8e-07) >> 4e-07', 'window': (0, 1024), 'w': None,
+         't0': 8.805e-06, 'phi': 2.4311851282940524, 'threshold': 9.645718574523926},
+        {'Delta': 6965129999.666667, 'phase': -0.0, 'weight': 'square(8e-07) >> 4e-07', 'window': (0, 1024), 'w': None,
+         't0': 9.005e-06, 'phi': 2.4311851282940524, 'threshold': 9.645718574523926}
+    ]}
+    import matplotlib.pyplot as plt
+    from nsqdriver import QSYNCDriver
+    import time
+    import waveforms as wf
+    from nsqdriver.NS_MCI import SHARED_DEVICE_MEM
+
+    SHARED_DEVICE_MEM.clear_ip()
+    _d = Driver('192.168.0.43', 300)
+    qsync = QSYNCDriver("192.168.0.43")
+    _d = Driver('192.168.0.228', 30)
+    _q = QSYNCDriver('192.168.0.228')
+    time.sleep(5)
+    _wave = wf.zero()
+    for _w in co['wList']:
+        t0 = _w['t0']
+        _wave += (wf.wave_eval(_w['weight']) * wf.cos(2 * np.pi * _w['Delta'])) >> t0
+    # _wave = _wave << 50e-9
+    _wave.start = 0
+    _wave.stop = 60e-6
+
+    _wave(np.linspace(0, 20e-6, int(20e-6 * 8e9)), frag=True)
+
+    wave = _wave.sample(8e9) / 3
+    _wave
+
+    for ch in range(3, 4):
+        _d.write('GenWave', _wave, ch=ch)
+        _d.write('Coefficient', co, ch=ch)
+    # res = _d.generate_para(co, 1)
+    _d.write("Coefficient", co, ch=2)
+    ch = 2
+    # wave = wf.cos(2*wf.pi*0.15e9)
+    # wave.start = 0
+    # wave.stop = 1e-6
+
+    _wave = waveforms.zero()
+    for _w in co['wList']:
+        t0 = _w['t0']
+        _wave += (waveforms.wave_eval(_w['weight']) * waveforms.cos(2 * np.pi * _w['Delta'])) >> t0
+    # _wave = _wave << 50e-9
+    _wave.start = 0
+    _wave.stop = co["stop"]
+    plt.plot(_wave.sample(8e9))
+    plt.show()
+    _d.write("GenWave", _wave, ch=2)
+    _d.set('CaptureMode', 1)
+    shots = 1024
+    qsync.set('Shot', shots)
+    _d.write('Shot', shots)
+    _d.set('StartCapture')
+    qsync.set('GenerateTrig', 400e-6)
+    time.sleep(400e-6 * shots + 0.5)
+    data = _d.get('TraceIQ', ch)
+    data = np.array(data)
+    print(data.shape)
+    data = data.reshape((shots, -1))
+    # qsync.set("Shot", 0xFFFFFFFF)
+    # qsync.set("GenerateTrig", 500e-6)
+    plt.figure()
+    plt.plot(data.mean(axis=0)[:])
+    plt.show()
+    _d.set('CaptureMode', 0)
+    _d.write("Coefficient", co, ch=2)
+    _d.write("GenWave", _wave, ch=2)
+    _d.set('StartCapture')
+    qsync.set('GenerateTrig', 40e-6)
+    data = _d.read("IQ", ch=2)
+    print(20 * np.log10(np.abs(data.mean(axis=0))))
+    pass