##// END OF EJS Templates
Update scritps
Update scritps

File last commit:

r1429:96f17ea26f9f
r1505:8d720d40ced2
Show More
test_wradlib_RHI.py
72 lines | 2.5 KiB | text/x-python | PythonLexer
import numpy as np
import matplotlib.pyplot as plt
import wradlib as wrl
import warnings
# libreia nueva
#export WRADLIB_DATA="/home/soporte/Downloads/2014-06-09--185000.rhi.mvol"
from mpl_toolkits.axisartist.grid_finder import FixedLocator, DictFormatter
warnings.filterwarnings('ignore')
# lectura de gaMIC hdf5 file
filename = wrl.util.get_wradlib_data_file("/home/soporte/Downloads/2014-06-09--185000.rhi.mvol")
#filename = wrl.util.get_wradlib_data_file("2014-06-09--185000.rhi.mvol")
data1, metadata = wrl.io.read_gamic_hdf5(filename)
print(data1)
data1 = data1['SCAN0']['ZH']['data']
print(data1)
print("SHAPE Data",np.array(data1).shape)
r = metadata['SCAN0']['r']
print("r",r)
print("longitud r",len(r))
th = metadata['SCAN0']['el']
print("th",th)
print("longitud th",len(th))
az = metadata['SCAN0']['az']
print("az",az)
site = (metadata['VOL']['Longitude'], metadata['VOL']['Latitude'],
metadata['VOL']['Height'])
print("Longitud,Latitud,Altura",site)
ma1 = np.array(data1)
for i in range(3):
print("dark",ma1[i])
'''
mask_ind = np.where(data1 <= np.nanmin(data1))
data1[mask_ind] = np.nan
ma1 = np.ma.array(data1, mask=np.isnan(data1))
'''
####################### test ####################s
th=(np.arange(450)/10.0)+5
#th= np.roll(th,-2)
#th=np.where(a<7,np.nan,a)
ma1=np.roll(ma1,-2,axis=0)
for i in range(3):
print("green",ma1[i])
print("a",th)
#th = [i for i in reversed(a)]
######################### test
#cgax, pm = wrl.vis.plot_rhi(ma1,r=r,th=th,rf=1e3)
fig = plt.figure(figsize=(10,8))
cgax, pm = wrl.vis.plot_rhi(ma1,r=r,th=th,rf=1e3,fig=fig, ax=111,proj='cg')
caax = cgax.parasites[0]
paax = cgax.parasites[1]
cgax.set_ylim(0, 14)
#caax = cgax.parasites[0]
#paax = cgax.parasites[1]
#cgax, pm = wrl.vis.plot_rhi(ma1, r=r, th=th, rf=1e3, fig=fig, ax=111, proj='cg')
txt = plt.title('Simple RHI',y=1.05)
#cbar = plt.gcf().colorbar(pm, pad=0.05, ax=paax)
cbar = plt.gcf().colorbar(pm, pad=0.05)
cbar.set_label('reflectivity [dBZ]')
caax.set_xlabel('x_range [km]')
caax.set_ylabel('y_range [km]')
plt.text(1.0, 1.05, 'azimuth', transform=caax.transAxes, va='bottom',ha='right')
gh = cgax.get_grid_helper()
# set theta to some nice values
locs = [0., 1., 2., 3., 4., 5., 6., 7., 8., 9., 10., 11., 12., 13., 14.,
15., 16., 17., 18., 20., 22., 25., 30., 35., 40., 50., 60., 70., 80., 90.]
gh.grid_finder.grid_locator1 = FixedLocator(locs)
gh.grid_finder.tick_formatter1 = DictFormatter(dict([(i, r"${0:.0f}^\circ$".format(i)) for i in locs]))
plt.show()