bmp = export_beam.beam_real(aa[0], ntop, shape0, 'x')
freq, fbmamp = export_beam.beam_fourier(bmp, d, 400)
print 'Time to initialize:', sys_time.clock(), 'seconds'
d = select_pair.pair_coarse(aa, src,times_coarse,dist,2.) #coarsely determine crossings
print 'Time after coarse selection:', sys_time.clock(), 'seconds'
# pairs_sorted = select_pair.pair_sort(d,freq,fbmamp) # sort crossings
# clos_app = select_pair.get_closest(pairs_sorted) # determine closest approach points
nproc = 4
nkey = len(d)
chunk,remainder = nkey/nproc, nkey%nproc
clos_app = select_pair.alter_clos(d,freq,fbmamp) # determine closest approach points
print 'Found closest approach points after:', sys_time.clock(), 'seconds'
pairs_final = select_pair.pair_fin(clos_app,dt,aa,src,freq,fbmamp,corr_tol) # output final sorted pairs
print 'Total time:', sys_time.clock(), 'seconds'
# write result to file and screen
f1 = open('./Pairing.out', 'a')
for j in n.arange(len(pairs_final)):
# print pairs_final[j]
f1.write(str(pairs_final[j])+'\n')
f1.close()
# call plotting routines
figname = './corr'+str(int(corr_tol))+'.png'
plot_pair.plot_closapp(clos_app,corr_tol,figname)
pair_xampl = select_pair.test_sample(pairs_final,dt,aa,src,freq,fbmamp,3000.)
plot_pair.plot_pair_xampl(pair_xampl)
d = select_pair.pair_coarse(aa, src, times_coarse, dist,
2.) #coarsely determine crossings
print 'Time after coarse selection:', sys_time.clock(), 'seconds'
# pairs_sorted = select_pair.pair_sort(d,freq,fbmamp) # sort crossings
# clos_app = select_pair.get_closest(pairs_sorted) # determine closest approach points
nproc = 4
nkey = len(d)
chunk, remainder = nkey / nproc, nkey % nproc
clos_app = select_pair.alter_clos(d, freq,
fbmamp) # determine closest approach points
print 'Found closest approach points after:', sys_time.clock(), 'seconds'
pairs_final = select_pair.pair_fin(clos_app, dt, aa, src, freq, fbmamp,
corr_tol) # output final sorted pairs
print 'Total time:', sys_time.clock(), 'seconds'
# write result to file and screen
f1 = open('./Pairing.out', 'a')
for j in n.arange(len(pairs_final)):
# print pairs_final[j]
f1.write(str(pairs_final[j]) + '\n')
f1.close()
# call plotting routines
figname = './corr' + str(int(corr_tol)) + '.png'
plot_pair.plot_closapp(clos_app, corr_tol, figname)
pair_xampl = select_pair.test_sample(pairs_final, dt, aa, src, freq, fbmamp,
3000.)
plot_pair.plot_pair_xampl(pair_xampl)
f1 = open(Oname, 'a')
for j in n.arange(len(pairs_final)):
#print pairs_final[j]
f1.write(str(pairs_final[j])+'\n')
f1.close()
f1 = open(Cname, 'a')
f1.write("bl1_bl2,T2-T1,Opp \n")
for j in n.arange(len(pairs_final)):
#print pairs_final[j]
T1, T2 = float(pairs_final[j][2][1]), float(pairs_final[j][3][1])
#fn1, fn2 = get_files.get_file(T1,dt_file, fdict1), get_files.get_file(T2,dt_file, fdict2)
blstr = str(pairs_final[j][2][0][0])+'_'+str(pairs_final[j][2][0][1])+'_'+str(pairs_final[j][3][0][0])+'_'+str(pairs_final[j][3][0][1])
lststr = str(-pairs_final[j][2][1]+pairs_final[j][3][1]) #time lag T2-T1
OPP = pairs_final[j][1]
#f1.write(blstr+','+lststr+' '+str(fn1)+' '+str(fn2)+' '+str(OPP)+'\n')
f1.write(blstr+','+lststr+','+str(OPP)+'\n')
f1.close()
#call plotting routines
figname = './corr'+str(int(corr_tol))+str(n.around(list_freq[ni],decimals=3))+'.png'
print "Saving scatterplot", figname
plot_pair.plot_closapp(clos_app,corr_tol,figname)
#plot sample approach points, puv in pair_fin must be True
pair_xampl = select_pair.test_sample(pairs_final)
plot_pair.plot_pair_xampl(pair_xampl)
