def test_gracon():
#setup
noise_level=0.0#m/s
nx=40
ny=40
fw=0.1
m_bump=10.00
t0=systime()
lats=linspace(-13.5, -12.0, 40)
lons=linspace(130.5, 131.5, 40)
ber_loc=[-12.4, 130.85] #location of Berrimah radar
gp_loc=[-12.2492, 131.0444]#location of CPOL at Gunn Point
h=2.5*1000.0
print 'calculating berimah UV', systime()-t0
i_ber, j_ber, k_ber=propigation.unit_vector_grid(lats, lons, h, ber_loc)
print 'calculating gp UV', systime()-t0
i_gp, j_gp, k_gp=propigation.unit_vector_grid(lats, lons, h, gp_loc)
#make winds
u,v=simul_winds.unif_wind(lats, lons, 10.0, 75.0)
up,vp=array(simul_winds.vortex(lats, lons, [-12.5, 131.1], fw))*m_bump
#make V_r measurements
vr_ber=i_ber*(up+u)+j_ber*(vp+v) + (random.random([nx,ny])-0.5)*(noise_level*2.0)
vr_gp=i_gp*(up+u)+j_gp*(vp+v)+ (random.random([nx,ny])-0.5)*(noise_level*2.0)
#try to reconstruct the wind field
igu, igv= simul_winds.unif_wind(lats, lons, 0.0, 90.0)
gv_u=zeros(u.shape)
gv_v=zeros(v.shape)
f=0.0
print igu.mean()
angs=array(propigation.make_lobe_grid(ber_loc, gp_loc, lats,lons))
wts=zeros(angs.shape, dtype=float)+1.0
#for i in range(angs.shape[0]):
# for j in range(angs.shape[1]):
# if (angs[i,j] < 150.0) and (angs[i,j] > 30.0): wts[i,j]=1.0
print 'Into fortran'
gv_u,gv_v,f,u_array,v_array = gracon_vel2d.gracon_vel2d(gv_u,gv_v,f,igu,igv,i_ber,j_ber,i_gp,j_gp,vr_ber,vr_gp,wts, nx=nx,ny=ny)
print u_array.mean()
print f
bnds=[0.,20.]
f=figure()
mapobj=pres.generate_darwin_plot()
pres.quiver_contour_winds(mapobj, lats, lons, (up+u),(vp+v), bounds=bnds)
savefig(os.getenv('HOME')+'/bom_mds/output/orig_winds_clean.png')
close(f)
f=figure()
mapobj=pres.generate_darwin_plot()
pres.quiver_contour_winds(mapobj, lats, lons, (wts*u_array +0.001),(wts*v_array +0.001), bounds=bnds)
savefig(os.getenv('HOME')+'/bom_mds/output/recon_winds_clean.png')
close(f)
f=figure()
mapobj=pres.generate_darwin_plot()
pres.quiver_contour_winds(mapobj, lats, lons, (wts*u_array - (up+u)),(wts*v_array -(vp+v)))
savefig(os.getenv('HOME')+'/bom_mds/output/errors_clean.png')
close(f)
def test_radar_view():
lats=linspace(-13.0, -11.5, 40)
lons=linspace(130., 132., 40)
ber_loc=[-12.4, 130.85] #location of Berrimah radar
gp_loc=[-12.2492, 131.0444]#location of CPOL at Gunn Point
h=2.5*1000.0
t0=systime()
i_a, j_a, k_a=propigation.unit_vector_grid(lats, lons, h, gp_loc)
print "unit_vector_compute took ", systime()-t0, " seconds"
fw=0.1#degrees
m_bump=5.0
#b1=simul_winds.speed_bump(lats, lons, [-12.0, 131.0], fw)*m_bump
#b2=-1.0*simul_winds.speed_bump(lats, lons, [-12.0, 131.1], fw)*m_bump
u,v=simul_winds.unif_wind(lats, lons, 5.0, 75.0)
up,vp=array(simul_winds.vortex(lats, lons, [-12.5, 131.1], fw))*m_bump
#up=u+(b1-b2)
#vp=v-(b1-b2)
w=v*0.0
v_r=i_a*(up+u)+j_a*(vp+v)+k_a*w
f=figure()
mapobj=pres.generate_darwin_plot()
pres.contour_vr(mapobj,lats, lons, v_r)
savefig(os.getenv('HOME')+'/bom_mds/output/test_radar_view_gp.png')
close(f)
f=figure()
mapobj=pres.generate_darwin_plot()
pres.quiver_contour_winds(mapobj, lats, lons, (up+u),(vp+v))
savefig(os.getenv('HOME')+'/bom_mds/output/test_pert2.png')
close(f)
def test_angle_grid2():
lats=linspace(-13.0, -11.5, 100)
lons=linspace(130., 132., 100)
ber_loc=[-12.4, 130.85] #location of Berrimah radar
gp_loc=[-12.2492, 131.0444]#location of CPOL at Gunn Point
h=10.0*1000.0
t0=systime()
i_a, j_a, k_a=propigation.unit_vector_grid(lats, lons, h, ber_loc)
print "compute took ", systime()-t0, " seconds"
f=figure()
mapobj=pres.generate_darwin_plot()
pres.contour_comp(mapobj,lats, lons, k_a)
savefig(os.getenv('HOME')+'/bom_mds/output/test_angs.png')
close(f)