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)