#!/usr/bin/python

import patterngenerator as pg
import readevents
import datetime as dt
import usbdds
import serial
import time
import caprel_optimize
import subprocess as sp

date = dt.date.today().isoformat()
run = 'mw_rabi'

dds_mw = usbdds.dds('dds_QO0035') #channel0-sw dipole
dds_mw2 = usbdds.dds('dds_QO0034') #channel0-sw dipole


duration = 80 #s

clock = 200835000

# B = 1.5V
m2_m1_freq = 200237000
m1_m0_freq = 200631500
m0_p1_freq = 201038500
p1_p2_freq = 201444700

# B = 9.0V
m2_m1_freq = 197085500

dds_mw.frequency(0,m2_m1_freq,'hz')
dds_mw.frequency(1,m1_m0_freq,'hz')
dds_mw2.frequency(0,m0_p1_freq,'hz')	
dds_mw2.frequency(1,p1_p2_freq ,'hz')


#pattfile = '/home/qitlab/superlens/patt_gen/lin_trap/mw_timescan_red.patt'  # prepares in dark state, F=1
pattfile = '/home/qitlab/superlens/patt_gen/lin_trap/mw_timescan_OP_red.patt' # prepares in bright state, F=2,m=-2
#pattfile = '/home/qitlab/superlens/patt_gen/lin_trap/mw_timescan_OP_red_2nd_trans.patt'
#pattfile = '/home/qitlab/superlens/patt_gen/lin_trap/mw_timescan_OP_red_3rd_trans.patt'
#pattfile = '/home/qitlab/superlens/patt_gen/lin_trap/mw_timescan_OP_red_4th_trans.patt'

for i in range(0,1):
	for pulse_length in list(range(1,301,4)):#+list(range(300,400,4)):
	#for pulse_length in [20]+range(1000,5001,1000):
		filename= run+'_'+date+'_'+str(pulse_length).zfill(4)+'us.timestamp'
		pg.edit(pattfile,'PULSE_LENGTH',pulse_length)
		pg.upload()
		time.sleep(2)
		print 'measuring...',filename
		readevents.main(filename,duration)

pg.edit(pattfile,'PULSE_LENGTH',1)
pg.upload()