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)
corr_tol = 5000. # cutoff of minimum correlation 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)
for ni in range(len(list_freq)): bmp = bmp_list[ni] 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) #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 clos_app = select_pair.alter_clos( d, freq, fbmamp, nproc=2) #determine closest approach points print 'Found closest approach points after:', sys_time.clock(), 'seconds' pairs_final = select_pair.pair_fin(clos_app, dt_fine, aa, src, freq, fbmamp, False, False, False, 5000.) #output final sorted pairs print 'Total time:', sys_time.clock(), 'seconds' #write result to file and screen Oname = './P' + str(n.around(list_freq[ni], decimals=3)) + '.out' f1 = open(Oname, 'w') f1.close() 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() #call plotting routines figname = './corr' + str(int(corr_tol)) + str(
bmp_list = export_beam.beam_real(aa[0], ntop, shape0, 'x') for ni in range(len(list_freq)): bmp = bmp_list[ni] freq, fbmamp = export_beam.beam_fourier(bmp, sp, 400) bm_intpl = export_beam.beam_interpol(freq,fbmamp,'cubic') print 'Time to initialize:', sys_time.clock(), 'seconds' print 'fbmampshape, midval', fbmamp.shape, fbmamp[200][200] d = select_pair.pair_coarse(aa, src,times_coarse,dist,False, 0.1, True) #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 clos_app = select_pair.alter_clos(d,bm_intpl) #determine closest approach points print 'Found closest approach points after:', sys_time.clock(), 'seconds' pairs_final = select_pair.pair_fin(clos_app,dt_fine,aa,src,freq,fbmamp,multweight=True,noiseweight=True,ovlpweight=True,puv=True) print 'Total time:', sys_time.clock(), 'seconds' #write result to file and screen Oname = './P'+str(n.around(list_freq[ni],decimals=3))+'.out' Cname = './P'+str(n.around(list_freq[ni],decimals=3))+'.cue' f1 = open(Oname, 'w') f1.close() f1 = open(Cname, 'w') f1.close() print "Writting ourput files", Oname, Cname 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()
sz = 200 d = 1./sz img = a.img.Img(200,res=0.5) #400 by 400 image, i.e. 200 boxes, with 4 pixels per box,freq space kmax=100, dk=0.5 X,Y,Z = img.get_top(center=(200,200)) shape0 = X.shape X,Y,Z = X.flatten(),Y.flatten(),Z.flatten() ntop = n.array([X,Y,Z]) list_freq = [.15] dt = 0.001 dt_fine = 0.0004971027374267578 times_coarse = n.arange(2456249.30169,2456249.35791, dt) times_fine = n.arange(2456249.30169,2456249.40791, dt_fine) dist = 1. #size of cells to store in dictionary. corr_tol = 5000. #cutoff of minimum correlation aa = a.cal.get_aa('psa6240_v003',n.array(list_freq)) #aa = a.cal.get_aa('psa898_v003',n.array(list_freq)) src = a.fit.RadioFixedBody(0, aa.lat, janskies=0., mfreq=.18, name='test') nants = len(aa) bmp_list = export_beam.beam_real(aa[0], ntop, shape0, 'x') ni = 0 bmp = bmp_list[ni] freq, fbmamp = export_beam.beam_fourier(bmp, d, 400) f = export_beam.beam_interpol(freq,fbmamp,'cubic') d = select_pair.pair_coarse(aa, src,times_coarse,dist) #coarsely determine crossings clos_app = select_pair.alter_clos(d,f) #determine closest approach points pairs_final = select_pair.pair_fin(clos_app,dt_fine,aa,src,freq, fbmamp,False,False,False,5000.) #output final sorted pairs