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)
def main(): args = get_argparser().parse_args() init_logger(args) dev = Client(args.server, args.port, "pdq2") dev.init() if args.reset: dev.write(b"\x00\x00") # flush any escape dev.cmd("RESET", True) time.sleep(.1) dev.cmd("START", False) dev.cmd("ARM", False) dev.cmd("DCM", args.dcm) freq = 100e6 if args.dcm else 50e6 dev.set_freq(freq) 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: for ch, channel in enumerate(dev.channels): entry = [] for fr in range(dev.channels[0].num_frames): vi = .1*fr + ch + voltages entry.append(channel.segment(times, vi, order=args.order, end=False, aux=args.aux)) pi = 2*np.pi*(-.5 + .01*fr + .1*ch + 0*voltages) fi = 10e6*times/times[-1] channel.segment(2*times, voltages, pi, fi, trigger=False, silence=True, aux=args.aux) dev.write_channel(channel, entry) elif args.bit: v = [-1, 0, -1] # for i in range(15): # v.extend([(1 << i) - 1, 1 << i]) v = np.array(v)*dev.channels[0].max_out/dev.channels[0].max_val t = np.arange(len(v)) for channel in dev.channels: s = channel.segment(t, v, order=0, shift=15, stop=False, trigger=False) dev.write_channel(channel, [s for i in range(channel.num_frames)]) else: c = dev.channels[args.channel] map = [None] * c.num_frames map[args.frame] = c.segment(times, voltages, order=args.order, aux=args.aux) dev.write_channel(c, map) dev.cmd("START", True) dev.cmd("ARM", not args.disarm) dev.cmd("TRIGGER", args.free) if args.plot: from matplotlib import pyplot as plt fig, ax = plt.subplots() ax.plot(times, voltages, "xk", label="points") if args.order > 0: spline = interpolate.splrep(times, voltages, k=args.order) ttimes = np.arange(0, times[-1], 1/freq) vvoltages = interpolate.splev(ttimes, spline) ax.plot(ttimes, vvoltages, ",b", label="interpolation") fig.savefig(args.plot)
def main(): args = get_argparser().parse_args() init_logger(args) dev = Client(args.server, args.port, "pdq2") dev.init() if args.reset: dev.write(b"\x00\x00") # flush any escape dev.cmd("RESET", True) time.sleep(.1) dev.cmd("START", False) dev.cmd("ARM", False) dev.cmd("DCM", args.dcm) freq = 100e6 if args.dcm else 50e6 dev.set_freq(freq) 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: for ch, channel in enumerate(dev.channels): entry = [] for fr in range(dev.channels[0].num_frames): vi = .1 * fr + ch + voltages entry.append( channel.segment(times, vi, order=args.order, end=False, aux=args.aux)) pi = 2 * np.pi * (-.5 + .01 * fr + .1 * ch + 0 * voltages) fi = 10e6 * times / times[-1] channel.segment(2 * times, voltages, pi, fi, trigger=False, silence=True, aux=args.aux) dev.write_channel(channel, entry) elif args.bit: v = [-1, 0, -1] # for i in range(15): # v.extend([(1 << i) - 1, 1 << i]) v = np.array(v) * dev.channels[0].max_out / dev.channels[0].max_val t = np.arange(len(v)) for channel in dev.channels: s = channel.segment(t, v, order=0, shift=15, stop=False, trigger=False) dev.write_channel(channel, [s for i in range(channel.num_frames)]) else: c = dev.channels[args.channel] map = [None] * c.num_frames map[args.frame] = c.segment(times, voltages, order=args.order, aux=args.aux) dev.write_channel(c, map) dev.cmd("START", True) dev.cmd("ARM", not args.disarm) dev.cmd("TRIGGER", args.free) if args.plot: from matplotlib import pyplot as plt fig, ax = plt.subplots() ax.plot(times, voltages, "xk", label="points") if args.order > 0: spline = interpolate.splrep(times, voltages, k=args.order) ttimes = np.arange(0, times[-1], 1 / freq) vvoltages = interpolate.splev(ttimes, spline) ax.plot(ttimes, vvoltages, ",b", label="interpolation") fig.savefig(args.plot)