def main(): args = get_argparser().parse_args() init_logger(args) dev = Client(args.server, args.port, "pdq2") freq = dev.get_freq() times = np.around(eval(args.times, globals(), {})*freq) voltages = eval(args.voltages, globals(), dict(t=times/freq)) dt = np.diff(times.astype(np.int)) if args.order: tck = interpolate.splrep(times, voltages, k=args.order, s=0) u = interpolate.spalde(times, tck) else: u = voltages[:, None] segment = [] for dti, ui in zip(dt, u): segment.append({ "duration": int(dti), "channel_data": [{ "bias": { "amplitude": [float(uij) for uij in ui] } }] }) program = [[] for i in range(args.frame)] program.append(segment) dev.park() dev.program(program, [args.channel]) dev.unpark() dev.cmd("TRIGGER", args.free)
def main(): args = get_argparser().parse_args() init_logger(args) dev = Client(args.server, args.port, "pdq2") freq = dev.get_freq() times = np.around(eval(args.times, globals(), {}) * freq) voltages = eval(args.voltages, globals(), dict(t=times / freq)) dt = np.diff(times.astype(np.int)) if args.order: tck = interpolate.splrep(times, voltages, k=args.order, s=0) u = interpolate.spalde(times, tck) else: u = voltages[:, None] segment = [] for dti, ui in zip(dt, u): segment.append({ "duration": int(dti), "channel_data": [{ "bias": { "amplitude": [float(uij) for uij in ui] } }] }) program = [[] for i in range(args.frame)] program.append(segment) dev.park() dev.program(program, [args.channel]) dev.unpark() dev.cmd("TRIGGER", args.free)
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) servers = { idx: { "host": ip, "notify": 3250, "control": 3251 } for idx, ip in enumerate(args.server) } while True: measurements = [] for _, server in servers.items(): try: client = RPCClient(server["host"], server["control"]) lasers = client.get_laser_db() for laser in lasers: meas = client.get_freq(laser, age=args.poll_time, priority=3, get_osa_trace=False, blocking=True, mute=False, offset_mode=False) status, freq, _ = meas if status != WLMMeasurementStatus.OKAY: logger.info("{}: measurement error") continue f_ref = lasers[laser]["f_ref"] delta = freq - lasers[laser]["f_ref"] measurements.append({ "measurement": laser, "fields": { "freq": freq, "f_ref": f_ref, "detuning": delta } }) logger.info("{}: freq {} THz, f_ref {} THz, " "detuning {} MHz".format( laser, freq, f_ref, delta)) except OSError: logger.warning("Error querying server {}".format(server)) finally: client.close_rpc() if measurements == []: time.sleep(args.poll_time) continue try: influx = influxdb.InfluxDBClient(host="10.255.6.4", database=args.database, username="******", password="******") influx.write_points(measurements) finally: influx.close() time.sleep(args.poll_time)