def get_two_best_sondes(date_str, **kwargs):
sonde_file=kwargs.get('sonde_file', '/bm/gdata/scollis/twpice/darwin.txt')
#outdir=kwargs.get('outdir', '/flurry/home/scollis/bom_mds/dealias/')
sonde_file=kwargs.get('sonde_file', '/bm/gdata/scollis/twpice/darwin.txt')
outdir=kwargs.get('outdir', '/flurry/home/scollis/bom_mds/dealias/')
tim_date=num2date(datestr2num(date_str))
sonde_list=read_sounding.read_sounding_within_a_day(sonde_file, tim_date)
launch_dates=[sonde['date_list'][0] for sonde in sonde_list]
#print launch_dates
launch_date_offset=[date2num(sonde['date_list'][0])- date2num(tim_date) for sonde in sonde_list]
sonde_made_it=False
candidate=0
while not(sonde_made_it):
best_sonde=sonde_list[argsort(abs(array(launch_date_offset)))[candidate]]
candidate=candidate+1
sonde_made_it=best_sonde['alt(m)'][-1] > 18000.
if not sonde_made_it: print "Sonde Burst at ", best_sonde['alt(m)'][-1], "m rejecting"
print "Sonde Burst at ", best_sonde['alt(m)'][-1], "m Accepting"
sonde_made_it=False
while not(sonde_made_it):
sec_best_sonde=sonde_list[argsort(abs(array(launch_date_offset)))[candidate]]
candidate=candidate+1
sonde_made_it=sec_best_sonde['alt(m)'][-1] > 18000.
if not sonde_made_it: print "Sonde Burst at ", sec_best_sonde['alt(m)'][-1], "m rejecting"
print "Sonde Burst at ", sec_best_sonde['alt(m)'][-1], "m Accepting"
print 'Time of radar: ', tim_date, ' Time of best sonde_launch: ', best_sonde['date_list'][0], ' Time of sonde_termination: ', best_sonde['date_list'][-1]
print 'Time of radar: ', tim_date, ' Time of second sonde_launch: ', sec_best_sonde['date_list'][0], ' Time of sonde_termination: ', best_sonde['date_list'][-1]
for i in range(len(sonde_list)):
best_sonde=sonde_list[argsort(abs(array(launch_date_offset)))[i]]
print 'Time of radar: ', tim_date, ' Time of best sonde_launch: ', best_sonde['date_list'][0], ' Offset', abs(date2num(best_sonde['date_list'][0])-date2num(tim_date))*24.0
return best_sonde, sec_best_sonde
def recon(date_str, latstr, lonstr):
use_guess='sonde'
sonde_file='/bm/gdata/scollis/twpice/darwin.txt'
#tim='1350'
tim_date=num2date(datestr2num(date_str))
ber, gp=netcdf_utis.load_cube('/bm/gdata/scollis/cube_data/'+std_datestr(tim_date)+'_deal.nc')
sonde_list=read_sounding.read_sounding_within_a_day(sonde_file, tim_date)
#launch_dates=[sonde['date_list'][0] for sonde in sonde_list]
#launch_date_offset=[date2num(sonde['date_list'][0])- date2num(tim_date) for sonde in sonde_list]
#best_sonde=sonde_list[argsort(abs(array(launch_date_offset)))[0]]
#print 'Time of radar: ', tim_date, ' Time of sonde_launch: ', best_sonde['date_list'][0], ' Time of sonde_termination: ', best_sonde['date_list'][-1]
req=[ 'alt(m)', 'wspd(m/s)', 'tdry(degs)', 'wdir(degs)']
first_sonde, second_sonde=read_sounding.get_two_best_conc_sondes(date_str, req_vars=req)
interp_sonde=read_sounding.interp_sonde_time(first_sonde, second_sonde, tim_date, gp['levs'])
u_sonde=ones(gp['CZ'].shape, dtype=float)
v_sonde=ones(gp['CZ'].shape, dtype=float)
w_sonde=zeros(gp['CZ'].shape, dtype=float)
for k in range(len(gp['levs'])):
u_sonde[:,:,k]=-1.0*u_sonde[:,:,k]*interp_sonde['wspd(m/s)'][k]*sin(pi*interp_sonde['wdir(degs)'][k]/180.0)
v_sonde[:,:,k]=-1.0*v_sonde[:,:,k]*interp_sonde['wspd(m/s)'][k]*cos(pi*interp_sonde['wdir(degs)'][k]/180.0)
Re=6371.0*1000.0
rad_at_radar=Re*sin(pi/2.0 -abs(gp['zero_loc'][0]*pi/180.0))#ax_radius(float(lat_cpol), units='degrees')
lons=gp['zero_loc'][1]+360.0*gp['xar']/(rad_at_radar*2.0*pi)
lats=gp['zero_loc'][0] + 360.0*gp['yar']/(Re*2.0*pi)
ber_loc=[-12.457, 130.925]
gp_loc= [-12.2492, 131.0444]
if use_guess=='none':
igu=ones(ber['CZ'].shape, dtype=float)*0.0
igv=ones(ber['CZ'].shape, dtype=float)*0.0
igw=ones(ber['CZ'].shape, dtype=float)*0.0
elif use_guess=='sonde':
igu=u_sonde
igv=v_sonde
igw=w_sonde
else:
ber_ig, gp_ig=netcdf_utis.load_cube(use_guess)
print gp_ig.keys()
igu=gp_ig['u_array']
igv=gp_ig['v_array']
igw=ones(ber['CZ'].shape, dtype=float)*0.0
#mywts=ones(ber['CZ'].shape, dtype=float)
angs=array(propigation.make_lobe_grid(ber_loc, gp_loc, lats,lons))
#wts_ang=zeros(gp['CZ'][:,:,0].shape, dtype=float)
#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_ang[i,j]=1.0
#for lvl_num in range(len(gp['levs'])):
# #create a weighting grid
# mask_reflect=10.0#dBZ
# mask=(gp['CZ'][:,:,lvl_num]/mask_reflect).round().clip(min=0., max=1.0)
# mask_vel_ber=(ber['VR'][:,:,lvl_num]+100.).clip(min=0., max=1.)
# mywts[:,:,lvl_num]=mask*mask_vel_ber*wts_ang
mywts=met.make_mask(ber, gp, angs, 1.0, 80.0)
print "Mean gp masked Velocity ", (gp['VE']*mywts).mean()
print "min gp masked Velocity ", (gp['VE']*mywts).min()
print "max gp masked Velocity ", (gp['VE']*mywts).max()
print "Mean Berrimah masked Velocity ", (ber['VE']*mywts).mean()
print "min Berrimah masked Velocity ", (ber['VE']*mywts).min()
print "max Berrimah masked Velocity ", (ber['VE']*mywts).max()
print "Mean gp masked CZ ", (gp['CZ']*mywts).mean()
print "min gp masked CZ ", (gp['CZ']*mywts).min()
print "max gp masked CZ ", (gp['CZ']*mywts).max()
print "Mean Berrimah masked CZ ", (ber['CZ']*mywts).mean()
print "min Berrimah masked CZ ", (ber['CZ']*mywts).min()
print "max Berrimah masked CZ ", (ber['CZ']*mywts).max()
print "Number of masked points", (mywts.shape[0]*mywts.shape[1]*mywts.shape[2])-mywts.sum()
print "Number of unmasked points ", mywts.sum()
print "**********************FALLSPEED INFO****************************"
#def terminal_velocity(refl, temps, levs, display=False):
tdry=interp_sonde['tdry(degs)']
pressure=interp_sonde['press(hPa)']
dummy=met.terminal_velocity(gp['CZ']*mywts, tdry, gp['levs'], display=True)
print "**********************FALLSPEED INFO****************************"
#print
f=0.0
X=[igu,igv,igw]
G,F,X=grad_conj_solver_3d.gracon_3d_packaged(X ,ber, gp, mywts, interp_sonde)
u_array,v_array,w_array=X
lvl=2500.0
lvl_num=argsort(abs(gp['levs']-lvl))[0]
print "Level_num=", lvl_num
if latstr=='max' or lonstr=='max':
maskedcz=gp['CZ'][:,:,lvl_num]*mywts[:,:,lvl_num]
i,j=mathematics.where_closest_2d(maskedcz.max(), maskedcz)
print i,j
lat=lats[j[0]]
lon=lons[i[0]]
print "Max CZ at ", lat, lon
else:
lon=float(lonstr)
lat=float(latstr)
f=figure()
#(lat_sl, lon_sl, lvl, data_cube, lats, lons, levs, u, v, w, angs, mask, par='CZ', w_mag=2.0, **kwargs)
alat, alon, alvl=pres.plot_slices(lat, lon, lvl, gp, lats, lons, gp['levs'], u_array, v_array, w_array, angs, mywts, par='CZ', w_mag=2.0,box=[130.5, 131.5, -12.7, -12.0], bquiver=[0.05, 0.75], ksp=0.05)
t1='Gunn Point reflectivity (dBZ) and reconstructed winds (m/s, *2.0 for w)\n sliced at %(alat)2.2fS and %(alon)3.2fE and %(alvl)d Metres on 22/01/06 at ' %{'alat':abs(alat), 'alon':alon, 'alvl':alvl}
t2=" %(HH)02d%(MM)02dZ" %{'HH':tim_date.hour, 'MM':tim_date.minute}
f.text( .1, .92, t1+t2)
inte_part=1000*(float(int(lat))-lat)
print {'alat':abs(alat), 'alon':alon, 'alvl':alvl}
ff=os.getenv('HOME')+'/bom_mds/output/recons_'+std_datestr(tim_date)[0:-5]+'/slicer3_%(alat)2.02f_%(alon)3.02f_%(alvl)05d_' %{'alat':abs(alat), 'alon':alon, 'alvl':alvl}
print ff
savefig(ff+t2+'.png', dpi=200)
gp.update({'u_array':u_array, 'v_array':v_array, 'w_array':w_array})
netcdf_utis.save_data_cube(ber, gp, '/bm/gdata/scollis/cube_data/'+std_datestr(tim_date)+'_winds.nc', gp_loc)
close(f)
def get_two_best_conc_sondes(date_str, **kwargs):
req_vars=kwargs.get('req_vars', [ 'alt(m)', 'press(hPa)', 'wspd(m/s)', 'tdry(degs)', 'wdir(degs)'])
min_ht=kwargs.get('min_ht', 17000.)
sonde_file=kwargs.get('sonde_file', '/bm/gdata/scollis/twpice/darwin.txt')
#outdir=kwargs.get('outdir', '/flurry/home/scollis/bom_mds/dealias/')
sonde_file=kwargs.get('sonde_file', '/bm/gdata/scollis/twpice/darwin.txt')
outdir=kwargs.get('outdir', '/flurry/home/scollis/bom_mds/dealias/')
tim_date=num2date(datestr2num(date_str))
sonde_list=read_sounding.read_sounding_within_a_day(sonde_file, tim_date)
launch_dates=[sonde['date_list'][0] for sonde in sonde_list]
#print launch_dates
launch_date_offset=[date2num(sonde['date_list'][0])- date2num(tim_date) for sonde in sonde_list]
sonde_made_it=False
candidate=0
while not(sonde_made_it):
sonde_num=argsort(abs(array(launch_date_offset)))[candidate]
best_sonde=sonde_list[sonde_num]
candidate=candidate+1
sonde_made_it=is_sonde_good(best_sonde, min_ht, req_vars)
#sonde_made_it=best_sonde['alt(m)'][-1] > min_ht
#if not sonde_made_it: print "Sonde Burst at ", best_sonde['alt(m)'][-1], "m rejecting"
#print "Sonde Burst at ", best_sonde['alt(m)'][-1], "m Accepting"
if launch_date_offset[sonde_num] < 0.:
print "Best Sonde flight is before the radar scan"
first_sonde=best_sonde
candidate=sonde_num +1
sonde_made_it=False
while not(sonde_made_it):
second_sonde=sonde_list[candidate]
candidate=candidate+1
sonde_made_it=is_sonde_good(second_sonde, min_ht, req_vars)
#sonde_made_it=second_sonde['alt(m)'][-1] > min_ht
#if not sonde_made_it: print "Sonde Burst at ", second_sonde['alt(m)'][-1], "m rejecting"
if (candidate==len(sonde_list)) and not(sonde_made_it):
print "I'm running off the end of the list, Bailing"
second_sonde=first_sonde
sonde_made_it=True
elif launch_date_offset[sonde_num] > 0.:
print "Best Sonde flight is after the radar scan"
second_sonde=best_sonde
candidate=sonde_num -1
sonde_made_it=False
while not(sonde_made_it):
first_sonde=sonde_list[candidate]
candidate=candidate-1
sonde_made_it=is_sonde_good(first_sonde, min_ht, req_vars)
#sonde_made_it=first_sonde['alt(m)'][-1] > min_ht
#if not sonde_made_it: print "Sonde Burst at ", first_sonde['alt(m)'][-1], "m rejecting"
if candidate==-1 and not(sonde_made_it):
print "I'm running off the start of the list, Bailing"
first_sonde=second_sonde
sonde_made_it=True
elif launch_date_offset[sonde_num]==0.:
print "Sonde launched at exactly the same time of the scan... Huey be praised"
first_sonde=second_sonde=best_sonde
print 'Time of radar: ', tim_date
print ' Time of first sonde_launch: ', first_sonde['date_list'][0], ' Time of first sonde_termination: ', first_sonde['date_list'][-1]
print "Sonde Burst at ", first_sonde['alt(m)'][-1], "m Accepting"
print ' Time of second sonde_launch: ', second_sonde['date_list'][0], ' Time of second sonde_termination: ', second_sonde['date_list'][-1]
print "Sonde Burst at ", second_sonde['alt(m)'][-1], "m Accepting"
for i in range(len(sonde_list)):
best_sonde=sonde_list[argsort(abs(array(launch_date_offset)))[i]]
print 'Time of radar: ', tim_date, ' Time of best sonde_launch: ', best_sonde['date_list'][0], ' Offset', abs(date2num(best_sonde['date_list'][0])-date2num(tim_date))*24.0
return first_sonde, second_sonde