test_wradlib_h5.py
150 lines
| 4.9 KiB
| text/x-python
|
PythonLexer
| r1325 | #!/usr/bin/env python | |||
| #LINUX | ||||
| #bash: export WRADLIB_DATA=/path/to/wradlib-data | ||||
| import os | ||||
| import matplotlib.pyplot as pl | ||||
| import wradlib as wrl | ||||
| import numpy as np | ||||
| pl.switch_backend("TKAgg") | ||||
| #filename = wrl.util.get_wradlib_data_file('dx/raa00-dx_10908-0806021655-fbg---bin.gz') | ||||
| validFilelist = [] | ||||
| path="/home/alex/Downloads/hdf5_wr/d2020204/" | ||||
| ext=".hdf5" | ||||
| ##path= '/home/alex/Downloads/2019-12-29/' | ||||
| fileList= os.listdir(path) | ||||
| for thisFile in fileList: | ||||
| ##if (os.path.splitext(thisFile)[0][-4:] != 'dBuZ'): | ||||
| ## continue | ||||
| validFilelist.append(thisFile) | ||||
| validFilelist.sort() | ||||
| #---print ("Files",validFilelist) | ||||
| #print (validFilelist) | ||||
| i=0 | ||||
| for thisFile in validFilelist: | ||||
| i=i+1 | ||||
| fpath = path+thisFile | ||||
| print("fpath",fpath) | ||||
| f = wrl.util.get_wradlib_data_file(fpath) | ||||
| #print("f",f) | ||||
| ####fcontent = wrl.io.read_rainbow(f) | ||||
| fcontent = wrl.io.read_generic_hdf5(f) | ||||
| #print(fcontent.keys()) | ||||
| #print (fcontent) | ||||
| #print(sfcontent['volume'] | ||||
| #Get azimuth data | ||||
| azi = fcontent['Pedestal/azimuth']['data'] | ||||
| ##print("azi0",azi['data'].shape) | ||||
| ##print("azi",azi['data']) | ||||
| ''' | ||||
| azi = fcontent['volume']['scan']['slice']['slicedata']['rayinfo']['data'] | ||||
| azidepth = float(fcontent['volume']['scan']['slice']['slicedata']['rayinfo']['@depth']) | ||||
| azirange = float(fcontent['volume']['scan']['slice']['slicedata']['rayinfo']['@rays']) | ||||
| azires = float(fcontent['volume']['scan']['slice']['anglestep']) | ||||
| azi = (azi * azirange / 2**azidepth) * azires | ||||
| ''' | ||||
| #---print("AZI",azi.shape) | ||||
| #---print("AZI",type(azi)) | ||||
| ##print("azi_range",azi) | ||||
| #print("azidepth",azidepth) | ||||
| #print("azirange",azirange) | ||||
| #print("azires",azires) | ||||
| #print("azi azi",azi) | ||||
| # Create range array | ||||
| ''' | ||||
| stoprange = float(fcontent['volume']['scan']['slice']['stoprange']) | ||||
| rangestep = float(fcontent['volume']['scan']['slice']['rangestep']) | ||||
| ''' | ||||
| stoprange = float(33*1.5) | ||||
| rangestep = float(0.15) | ||||
| r = np.arange(0, stoprange, rangestep) | ||||
| #---print("stoprange",stoprange) | ||||
| #---print("rangestep",rangestep) | ||||
| #---print("r",r.shape) | ||||
| #---print("r",type(r)) | ||||
| # GET reflectivity data#print (fcontent) | ||||
| #print(sfcontent['volume'] | ||||
| ''' | ||||
| data = fcontent['volume']['scan']['slice']['slicedata']['rawdata']['data'] | ||||
| datadepth = float(fcontent['volume']['scan']['slice']['slicedata']['rawdata']['@depth']) | ||||
| datamin = float(fcontent['volume']['scan']['slice']['slicedata']['rawdata']['@min']) | ||||
| datamax = float(fcontent['volume']['scan']['slice']['slicedata']['rawdata']['@max']) | ||||
| data = datamin + data * (datamax - datamin) / 2 ** datadepth | ||||
| ''' | ||||
| data = fcontent['Data/dataPP_POW/table00']['data'] | ||||
| data = 10*np.log10(data/(625**2)) | ||||
| #---print("DATA",data.shape) | ||||
| #---print("DATA",type(data)) | ||||
| ##data =fcontent['Data/dataPP_POW/table00']['data'] | ||||
| ##print("data 1",data.shape) | ||||
| ##data = 10*np.log10(fcontent['Data/dataPP_POW/table00']['data']/(625**2)) | ||||
| ##print("test data",data.shape) | ||||
| #print("datadepth",datadepth) | ||||
| #print("datamin",datamin) | ||||
| #print("datamax",datamax) | ||||
| #print("data",data) | ||||
| if i==1: | ||||
| azi2= azi | ||||
| data2= data | ||||
| #---print("azi2 shape I es 1",azi2.shape) | ||||
| else: | ||||
| if i==4: | ||||
| #---print (azi[60:].shape) | ||||
| azi2= np.concatenate((azi2,azi[40:])) | ||||
| data2= np.concatenate((data2,data[40:,:]),axis=0) | ||||
| else: | ||||
| azi2= np.concatenate((azi2,azi)) | ||||
| data2= np.concatenate((data2,data),axis=0) | ||||
| #---print("azi2 shape",azi2.shape) | ||||
| #print(fcontent['volume']['sensorinfo']) | ||||
| #print("ESPACIO") | ||||
| #print(data) | ||||
| #GET ANNOTATION data | ||||
| ''' | ||||
| unit = fcontent['volume']['scan']['slice']['slicedata']['rawdata']['@type'] | ||||
| time = fcontent['volume']['scan']['slice']['slicedata']['@time'] | ||||
| date = fcontent['volume']['scan']['slice']['slicedata']['@date'] | ||||
| lon = fcontent['volume']['sensorinfo']['lon'] | ||||
| lat = fcontent['volume']['sensorinfo']['lat'] | ||||
| sensortype = fcontent['volume']['sensorinfo']['@type'] | ||||
| sensorname = fcontent['volume']['sensorinfo']['@name'] | ||||
| ''' | ||||
| #---print("AZI2",azi2.shape) | ||||
| #---print("data2",data2.shape) | ||||
| # PLOT DATA WITH ANNOTATION | ||||
| fig = pl.figure(figsize=(8,8)) | ||||
| cgax, pm = wrl.vis.plot_ppi(data2, r=r, az=azi2, fig=fig, proj='cg') | ||||
| #title = '{0} {1} {2} {3}\n{4}E {5}N'.format(sensortype, sensorname, date,time, lon, lat) | ||||
| caax = cgax.parasites[0] | ||||
| #t = pl.title(title, fontsize=12) | ||||
| #t.set_y(1.1) | ||||
| cbar = pl.gcf().colorbar(pm, pad=0.075) | ||||
| caax.set_xlabel('x_range [km]') | ||||
| caax.set_ylabel('y_range [km]') | ||||
| pl.text(1.0, 1.05, 'azimuth', transform=caax.transAxes, va='bottom', | ||||
| ha='right') | ||||
| #cbar.set_label('reflectivity [' + unit + ']') | ||||
| pl.show() | ||||
| #print ("FIN") | ||||
| #import time | ||||
| #time.sleep(2) | ||||
| #pl.close() | ||||
| # stop here | ||||
| #print(fcontent['data']) | ||||
| #data, metadata = wrl.io.read_dx(filename) | ||||
| #ax, pm = wrl.vis.plot_ppi(data) # simple diagnostic plot | ||||
| #cbar = pl.colorbar(pm, shrink=0.75) | ||||
