##// END OF EJS Templates
schain
RSS
scripts de prueba para data procesada con signal chain y script proc_JC.py con los ultimos cambios y parametros de escritura
scripts de prueba para data procesada con signal chain y script proc_JC.py con los ultimos cambios y parametros de escritura
avaldezp - - Load All Authors

File last commit:

r1529:08e25d8a8316
r1535:0450a3a7a48e
Show More
sophy_proc_rev006.py
229 lines | 8.8 KiB | text/x-python | PythonLexer
/ schainpy / scripts / sophy_proc_rev006.py
History | Source | Raw |Copy content |Copy permalink
avaldezp
new script with wradlib library for process process data
 r1516 #!/usr/bin/env python
avaldezp
test
 r1520 # Ing. AVP
# 22/06/2022
# ARCHIVO DE LECTURA y PLOT
avaldezp
new script with wradlib library for process process data
 r1516 import matplotlib.pyplot as pl
import matplotlib
import wradlib
import numpy
import warnings
import argparse
from wradlib.io import read_generic_hdf5
from wradlib.util import get_wradlib_data_file
from plotting_codes import sophy_cb_tables
import os,time
for name, cb_table in sophy_cb_tables:
ncmap = matplotlib.colors.ListedColormap(cb_table, name=name)
matplotlib.pyplot.register_cmap(cmap=ncmap)
#LINUX bash: export WRADLIB_DATA=/path/to/wradlib-data
avaldezp
update script wradlib
 r1525 #example
#export WRADLIB_DATA="/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/paramC0N36.0/2022-06-09T18-00-00/"
avaldezp
new script with wradlib library for process process data
 r1516 warnings.filterwarnings('ignore')
PARAM = {
avaldezp
update script genera and script with library wradlib
 r1519 'S': {'var': 'power','vmin': -45, 'vmax': -15, 'cmap': 'jet', 'label': 'Power','unit': 'dBm'},
'V': {'var': 'velocity', 'vmin': -10, 'vmax': 10 , 'cmap': 'sophy_v', 'label': 'Velocity','unit': 'm/s'},
'Z': {'var': 'reflectivity','vmin': -30, 'vmax': 80 , 'cmap': 'sophy_r','label': 'Reflectivity','unit': 'dBZ'},
avaldezp
Update script test
 r1529 'W': {'var': 'spectral_width', 'vmin': 0 , 'vmax': 12 , 'cmap': 'sophy_w','label': 'Spectral Width','unit': 'm/s'}
avaldezp
new script with wradlib library for process process data
 r1516 }
class Readsophy():
def __init__(self):
self.list_file = None
self.grado = None
self.variable = None
self.save = None
avaldezp
update script genera and script with library wradlib
 r1519 self.range = None
avaldezp
new script with wradlib library for process process data
 r1516
def read_files(self,path_file,grado=None, variable=None):
filter= "_E"+str(grado)+".0_"+variable
validFilelist = []
fileList= os.listdir(path_file)
for thisFile in fileList:
if (os.path.splitext(thisFile)[0][-7:] != filter):
#print("s_:",os.path.splitext(thisFile)[0][-7:])
continue
validFilelist.append(thisFile)
validFilelist.sort()
return validFilelist
avaldezp
update script genera and script with library wradlib
 r1519 def setup(self, path_file,grado,range,variable,save):
avaldezp
new script with wradlib library for process process data
 r1516 self.path_file = path_file
avaldezp
update script genera and script with library wradlib
 r1519 self.range = range
avaldezp
new script with wradlib library for process process data
 r1516 self.grado = grado
self.variable = variable
self.save = save
self.list_file = self.read_files(path_file=self.path_file,grado=self.grado, variable=self.variable)
avaldezp
update script genera and script with library wradlib
 r1519 def selectHeights(self,heightList,minHei,maxHei):
if minHei and maxHei:
if (minHei < heightList[0]):
minHei = heightList[0]
if (maxHei > heightList[-1]):
maxHei = heightList[-1]
minIndex = 0
maxIndex = 0
heights = heightList
inda = numpy.where(heights >= minHei)
indb = numpy.where(heights <= maxHei)
try:
minIndex = inda[0][0]
except:
minIndex = 0
try:
maxIndex = indb[0][-1]
except:
maxIndex = len(heights)
new_heightList= self.selectHeightsByIndex(heightList=heightList,minIndex=minIndex, maxIndex=maxIndex)
return new_heightList, minIndex,maxIndex
def selectHeightsByIndex(self,heightList,minIndex, maxIndex):
if (minIndex < 0) or (minIndex > maxIndex):
raise ValueError("Height index range (%d,%d) is not valid" % (minIndex, maxIndex))
if (maxIndex >= len(heightList)):
maxIndex = len(heightList)
new_h = heightList[minIndex:maxIndex]
return new_h
avaldezp
update script wradlib
 r1525 def readAttributes(self,obj,variable):
var = PARAM[variable]['var']
unit = PARAM[variable]['unit']
cmap = PARAM[variable]['cmap']
vmin = PARAM[variable]['vmin']
vmax = PARAM[variable]['vmax']
label = PARAM[variable]['label']
var_ = 'Data/'+var+'/H'
data_arr = numpy.array(obj[var_]['data']) # data
utc_time = numpy.array(obj['Data/time']['data'])
data_azi = numpy.array(obj['Metadata/azimuth']['data']) # th
data_ele = numpy.array(obj["Metadata/elevation"]['data'])
heightList = numpy.array(obj["Metadata/range"]['data']) # r
return data_arr, utc_time, data_azi,data_ele, heightList,unit,cmap,vmin,vmax,label
avaldezp
new script with wradlib library for process process data
 r1516 def run(self):
count= 0
len_files = len(self.list_file)
for thisFile in self.list_file:
count = count +1
fullpathfile = self.path_file + thisFile
filename = get_wradlib_data_file(fullpathfile)
test_hdf5 = read_generic_hdf5(filename)
avaldezp
update script wradlib
 r1525 # LECTURA
data_arr, utc_time, data_azi,data_ele, heightList,unit,cmap,vmin,vmax,label = self.readAttributes(obj= test_hdf5,variable=self.variable)
avaldezp
update script genera and script with library wradlib
 r1519
if self.range==0:
self.range == heightList[-1]
new_heightList,minIndex,maxIndex = self.selectHeights(heightList,0.06,self.range)
avaldezp
update script wradlib
 r1525 # TIEMPO
avaldezp
new script with wradlib library for process process data
 r1516 my_time = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(utc_time[0]))
time_save = time.strftime('%Y%m%d_%H%M%S',time.localtime(utc_time[0]))
# PLOT DATA WITH ANNOTATION
if count ==1:
fig = pl.figure(figsize=(10,8))
avaldezp
update script genera and script with library wradlib
 r1519 cgax, pm = wradlib.vis.plot_ppi(data_arr[:,minIndex:maxIndex],r=new_heightList,az=data_azi,rf=1,fig=fig, ax=111,proj='cg',cmap=cmap,vmin=vmin, vmax=vmax)
avaldezp
new script with wradlib library for process process data
 r1516 caax = cgax.parasites[0]
avaldezp
fix range
 r1521 title = 'Simple PPI'+"-"+ my_time+" E."+self.grado
avaldezp
new script with wradlib library for process process data
 r1516 t = pl.title(title, fontsize=12,y=1.05)
cbar = pl.gcf().colorbar(pm, pad=0.075)
pl.text(1.0, 1.05, 'azimuth', transform=caax.transAxes, va='bottom',ha='right')
cbar.set_label(label+'[' + unit + ']')
gh = cgax.get_grid_helper()
else:
avaldezp
update script genera and script with library wradlib
 r1519 cgax, pm = wradlib.vis.plot_ppi(data_arr[:,minIndex:maxIndex],r=new_heightList,az=data_azi,rf=1,fig=fig, ax=111,proj='cg',cmap=cmap,vmin=vmin, vmax=vmax)
avaldezp
new script with wradlib library for process process data
 r1516 caax = cgax.parasites[0]
avaldezp
fix range
 r1521 title = 'Simple PPI'+"-"+my_time+" E."+self.grado
avaldezp
new script with wradlib library for process process data
 r1516 t = pl.title(title, fontsize=12,y=1.05)
cbar = pl.gcf().colorbar(pm, pad=0.075)
pl.text(1.0, 1.05, 'azimuth', transform=caax.transAxes, va='bottom',ha='right')
cbar.set_label(label+'[' + unit + ']')
gh = cgax.get_grid_helper()
if self.save == 1:
if count ==1:
filename = "SOPHY"+"_"+time_save+"_"+"E."+self.grado+"_"+self.variable+".png"
dir =self.variable+"_"+"E."+self.grado+"CH0/"
filesavepath = os.path.join(self.path_file,dir)
avaldezp
update script genera and script with library wradlib
 r1519 try:
os.mkdir(filesavepath)
except:
pass
avaldezp
new script with wradlib library for process process data
 r1516 else:
filename = "SOPHY"+"_"+time_save+"_"+"E."+self.grado+"_"+self.variable+".png"
pl.savefig(filesavepath+filename)
avaldezp
update script genera and script with library wradlib
 r1519 pl.pause(1)
avaldezp
new script with wradlib library for process process data
 r1516 pl.clf()
if count==len_files:
pl.close()
pl.show()
avaldezp
update script genera and script with library wradlib
 r1519 PATH = "/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/paramC0N36.0/2022-06-09T18-00-00/"
avaldezp
fix range
 r1521 #PATH = "/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/paramC0N36.0/2022-06-09T19-00-00/"
avaldezp
update script genera and script with library wradlib
 r1519
#PATH = "/home/soporte/Documents/EVENTO/HYO_PM@2022-05-31T12-00-17/paramC0N36.0/2022-05-31T16-00-00/"
avaldezp
new script with wradlib library for process process data
 r1516
def main(args):
grado = args.grado
parameters = args.parameters
save = args.save
avaldezp
update script genera and script with library wradlib
 r1519 range = args.range
avaldezp
new script with wradlib library for process process data
 r1516 obj = Readsophy()
for param in parameters:
avaldezp
update script genera and script with library wradlib
 r1519 obj.setup(path_file = PATH,grado = grado,range=range, variable=param,save=int(save))
avaldezp
new script with wradlib library for process process data
 r1516 obj.run()
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Script to process SOPHy data.')
parser.add_argument('--parameters', nargs='*', default=['S'],
help='Variables to process: P, Z, V ,W')
parser.add_argument('--grado', default=2,
help='Angle in Elev to plot')
parser.add_argument('--save', default=0,
help='Save plot')
avaldezp
update script genera and script with library wradlib
 r1519 parser.add_argument('--range', default=0, type=float,
help='Max range to plot')
avaldezp
new script with wradlib library for process process data
 r1516 args = parser.parse_args()
main(args)
avaldezp
update script genera and script with library wradlib
 r1519 #python sophy_proc_rev006.py --parameters Z --grado 8 --save 1 --range 28
avaldezp
new script with wradlib library for process process data
 r1516 '''
def read_and_overview(filename):
"""Read HDF5 using read_generic_hdf5 and print upper level dictionary keys
"""
test = read_generic_hdf5(filename)
print("\nPrint keys for file %s" % os.path.basename(filename))
for key in test.keys():
print("\t%s" % key)
return test
file__ = "/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/paramC0_FD_PL_R15.0km/2022-06-09T18-00-00/SOPHY_20220609_180229_E2.0_Z.hdf5"
filename = get_wradlib_data_file(file__)
print("filename:\n",filename )
test= read_and_overview(filename)
# ANADIR INFORMACION
# informacion de los pulsos de TX
# informacion de los ruidos
# informacion de los SNR ¿?
# Aumentar la amplitud de la USRP
avaldezp
last
 r1522 LAST_UPDATE
---- Noise
---- Mapas
avaldezp
new script with wradlib library for process process data
 r1516 '''