test_wradlib_RHI.py
72 lines
| 2.5 KiB
| text/x-python
|
PythonLexer
avaldezp
|
r1411 | import numpy as np | ||
| import matplotlib.pyplot as plt | ||||
| import wradlib as wrl | ||||
| import warnings | ||||
| # libreia nueva | ||||
|
avaldezp
|
r1412 | #export WRADLIB_DATA="/home/soporte/Downloads/2014-06-09--185000.rhi.mvol" | ||
|
avaldezp
|
r1411 | 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") | ||||
|
avaldezp
|
r1429 | #filename = wrl.util.get_wradlib_data_file("2014-06-09--185000.rhi.mvol") | ||
|
avaldezp
|
r1411 | 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) | ||||
|
avaldezp
|
r1429 | for i in range(3): | ||
| print("dark",ma1[i]) | ||||
|
avaldezp
|
r1411 | ''' | ||
| mask_ind = np.where(data1 <= np.nanmin(data1)) | ||||
| data1[mask_ind] = np.nan | ||||
| ma1 = np.ma.array(data1, mask=np.isnan(data1)) | ||||
| ''' | ||||
|
avaldezp
|
r1429 | ####################### 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 | ||||
|
avaldezp
|
r1411 | #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() | ||||