def main(): args = get_argparser().parse_args() dev = Client(args.server, args.port, "pdq2") dev.init() if args.reset: dev.flush_escape() dev.write_cmd("RESET_EN") time.sleep(.1) if args.dcm: dev.write_cmd("DCM_EN") dev.set_freq(100e6) elif args.dcm == 0: dev.write_cmd("DCM_DIS") dev.set_freq(50e6) dev.write_cmd("START_DIS") num_channels = dev.get_num_channels() num_frames = dev.get_num_frames() times = eval(args.times, globals(), {}) voltages = eval(args.voltages, globals(), dict(t=times)) if args.demo: # FIXME channels = [args.channel] if args.channel < num_channels \ else range(num_channels) frames = [args.frame] if args.frame < num_frames \ else range(num_frames) for channel in channels: f = [] for frame in frames: vi = .1 * frame + channel + voltages pi = 2 * np.pi * (.01 * frame + .1 * channel + 0 * voltages) fi = 10e6 * times / times[-1] f.append(b"".join([ dev.frame(times, vi, order=args.order, end=False), dev.frame(2 * times, voltages, pi, fi, trigger=False), # dev.frame(2*times, 0*vi+.1, 0*pi, 0*fi+1e6), # dev.frame(times, 0*vi, order=args.order, silence=True), ])) board, dac = divmod(channel, dev.num_dacs) dev.write_data(dev.add_mem_header(board, dac, dev.map_frames(f))) elif args.bit: map = [0] * num_frames t = np.arange(2 * 16) * 1. v = [-1, 0, -1] for i in range(15): vi = 1 << i v.extend([vi - 1, vi]) v = np.array(v) * dev.get_max_out() / (1 << 15) t, v = t[:3], v[:3] # print(t, v) for channel in range(num_channels): dev.multi_frame([(t, v)], channel=channel, order=0, map=map, shift=15, stop=False, trigger=False) else: tv = [(times, voltages)] map = [None] * num_frames map[args.frame] = 0 dev.multi_frame(tv, channel=args.channel, order=args.order, map=map) dev.write_cmd("START_EN") if not args.disarm: dev.write_cmd("ARM_EN") if args.free: dev.write_cmd("TRIGGER_EN") if args.plot: from matplotlib import pyplot as plt fig, ax0 = plt.subplots() ax0.plot(times, voltages, "xk", label="points") if args.order: spline = interpolate.splrep(times, voltages, k=args.order) ttimes = np.arange(0, times[-1], 1 / dev.get_freq()) vvoltages = interpolate.splev(ttimes, spline) ax0.plot(ttimes, vvoltages, ",b", label="interpolation") fig.savefig(args.plot)