#!/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_spectrum'

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


duration = 50 #s

clock = 200835000

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

# B = 3.0V
m2_m1_freq = 197145000

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')

#scan range
res = m2_m1_freq
#res = clock  # 
start = res - 70000
end = res + 30000
step = 500
freqlist = list(range(int(start),int(end),int(step)))


#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'
pulse_length = 10 #11.5 17 24 6
pg.edit(pattfile,'PULSE_LENGTH',pulse_length)
pg.upload()
for i in range(0,1):
	for freq in freqlist:
		dds_mw.frequency(0,freq,'hz')		
		filename= run+'_'+date+'_'+str(freq)+'Hz.timestamp'
		time.sleep(1)
		print 'measuring...',filename
		readevents.main(filename,duration)

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