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
