##// END OF EJS Templates
schain
RSS
Script test de pruebas actuales con el update de heading en el block 360, el parametro adicional es heading que se lee del experiment.conf
Script test de pruebas actuales con el update de heading en el block 360, el parametro adicional es heading que se lee del experiment.conf
avaldez - - Load All Authors

File last commit:

r1591:d9024fd73e87
r1759:c8ee9da660f1
Show More
test_zdr.py
486 lines | 21.4 KiB | text/x-python | PythonLexer
/ schainpy / scripts / test_zdr.py
History | Source | Raw |Copy content |Copy permalink
avaldezp
update new plots y processing
 r1581 import numpy,os,time
import matplotlib
import argparse
import matplotlib.pyplot as plt
from wradlib.io import read_generic_hdf5
from wradlib.util import get_wradlib_data_file
from plotting_codes import sophy_cb_tables
avaldezp
UPDATE ZDR
 r1582 from scipy import stats
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 import wradlib
from wradlib.io import read_generic_hdf5
from wradlib.util import get_wradlib_data_file
import warnings
warnings.filterwarnings('ignore')
avaldezp
update new plots y processing
 r1581
for name, cb_table in sophy_cb_tables:
ncmap = matplotlib.colors.ListedColormap(cb_table, name=name)
matplotlib.pyplot.register_cmap(cmap=ncmap)
'''
NOTA:
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 #python test_zdr.py --parameters Z --grado 90 --mode 'PPI' --range 10 --type plot_WR --channel H
avaldezp
update new plots y processing
 r1581 - python3.10
- Conda environment:
WR_CONDA_JUN14 * /home/soporte/anaconda3/envs/WR_CONDA_JUN14
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 export WRADLIB_DATA="/media/soporte/TOSHIBAEXT/sophy/HYO_CC4_CC64_COMB@2022-12-26T00-00-32/param-EVENTO/"
avaldezp
update new plots y processing
 r1581 - Update de plotting_codes
'''
PARAM = {
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 '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': -20, 'vmax': 80 , 'cmap': 'sophy_z','label': 'Reflectivity' ,'unit': 'dBZ'},
'W': {'var': 'spectral_width' ,'vmin': 0 , 'vmax': 12 , 'cmap': 'sophy_w','label': 'Spectral Width','unit': 'm/s'},
'R': {'var': 'rhoHV' ,'vmin': 0.2, 'vmax': 1 , 'cmap': 'sophy_r','label': 'RhoHV' , 'unit': ' '},
'D': {'var': 'differential_reflectivity','vmin': -9 , 'vmax': 12 , 'cmap': 'sophy_d','label': 'ZDR' , 'unit': 'dB'}
avaldezp
update new plots y processing
 r1581 }
class Readsophy():
def __init__(self):
self.list_file = None
self.grado = None
self.variable = None
self.save = None
self.range = None
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 def setup(self, path_file,mode,channel,type,grado,range,r_min,variable,save):
avaldezp
update new plots y processing
 r1581 self.path_file = path_file
self.mode = mode
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 self.channel = channel
avaldezp
update new plots y processing
 r1581 self.range = range
self.grado = grado
avaldezp
UPDATE ZDR
 r1582 self.r_min = r_min
avaldezp
update new plots y processing
 r1581 self.variable = variable
self.save = save
avaldezp
UPDATE ZDR
 r1584 self.type_ = type
avaldezp
update new plots y processing
 r1581 self.list_file = self.read_files(path_file=self.path_file,mode=self.mode,grado=self.grado, variable=self.variable)
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 #print("self.list_file",self.list_file)
avaldezp
update new plots y processing
 r1581
def read_files(self,path_file,mode=None,grado=None, variable=None):
if mode =='PPI':
filter= "_E"+str(grado)+".0_"+variable
else:
filter= "_A"+str(grado)+".0_"+variable
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 #print("Filter :",filter)
avaldezp
update new plots y processing
 r1581 validFilelist = []
fileList= os.listdir(path_file)
for thisFile in fileList:
#print(thisFile)
if not os.path.splitext(thisFile)[0][-7:] in filter:
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 #print("s_:",os.path.splitext(thisFile)[0][-7:])
avaldezp
update new plots y processing
 r1581 continue
validFilelist.append(thisFile)
validFilelist.sort()
return validFilelist
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 def readAttributes(self,obj,variable,channel,type_):
avaldezp
update new plots y processing
 r1581 var = PARAM[variable]['var']
unit = PARAM[variable]['unit']
cmap = PARAM[variable]['cmap']
vmin = PARAM[variable]['vmin']
vmax = PARAM[variable]['vmax']
label = PARAM[variable]['label']
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591
avaldezp
update new plots y processing
 r1581 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
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 if type_ =='Diagonal':
var_H = 'Data/'+var+'/'+str('H')
var_V = 'Data/'+var+'/'+str('V')
data_arr_H = numpy.array(obj[var_H]['data']) # data
data_arr_V = numpy.array(obj[var_V]['data']) # data
data_arr = numpy.array([data_arr_H ,data_arr_V])
else:
var_ = 'Data/'+var+'/'+str(channel)
data_arr = numpy.array(obj[var_]['data']) # data
avaldezp
update new plots y processing
 r1581 return data_arr, utc_time, data_azi,data_ele, heightList,unit,cmap,vmin,vmax,label
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591
avaldezp
update new plots y processing
 r1581
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 ZDR
 r1584 def plot_RTI_PPI_RHI(self,count,x,y,z,cmap,my_time,vmin,vmax,label,unit,mode,grado):
avaldezp
UPDATE ZDR
 r1582 if count==1:
fig = plt.figure(figsize=(8,6))
plt.pcolormesh(x,y,z,cmap =cmap, vmin = vmin, vmax = vmax)
avaldezp
UPDATE TEST ZDR
 r1586 title = 'Sophy Plot '+label+"-"+ my_time+" "+mode+" "+grado+" N° "+str(count)
avaldezp
UPDATE ZDR
 r1582 t = plt.title(title, fontsize=12,y=1.05)
cbar = plt.colorbar()
cbar.set_label(label+'[' + unit + ']')
else:
plt.pcolormesh(x,y,z, cmap =cmap, vmin = vmin, vmax = vmax)
avaldezp
UPDATE TEST ZDR
 r1586 title = 'Sophy Plot '+label+"-"+ my_time+" "+mode+" "+grado+" N° "+str(count)
avaldezp
UPDATE ZDR
 r1582 t = plt.title(title, fontsize=12,y=1.05)
cbar = plt.colorbar()
cbar.set_label(label+'[' + unit + ']')
avaldezp
UPDATE ZDR
 r1584 def plot_PROFILE(self,count,z,y,my_time,label,mode,grado):
if count==1:
fig = plt.figure(figsize=(8,6))
avaldezp
UPDATE TEST ZDR
 r1586 plt.plot(numpy.nanmean(z,1),y)
title = 'Sophy Plot '+label+"-"+ my_time+" "+mode+" "+grado+" N° "+str(count)
avaldezp
UPDATE ZDR
 r1584 t = plt.title(title, fontsize=12,y=1.05)
plt.ylim(0,self.range+1)
plt.xlabel(label)
plt.ylabel('Height(Km)')
if self.variable=="R":
plt.xlim(-1,3)
if self.variable=='D':
plt.xlim(-10,10)
avaldezp
UPDATE TEST ZDR
 r1586 if self.variable=='Z':
plt.xlim(-20,80)
avaldezp
UPDATE ZDR
 r1584 else:
avaldezp
UPDATE TEST ZDR
 r1586 plt.plot(numpy.nanmean(z,1),y)
title = 'Sophy Plot '+label+"-"+ my_time+" "+mode+" "+grado+" N° "+str(count)
avaldezp
UPDATE ZDR
 r1584 t = plt.title(title, fontsize=12,y=1.05)
plt.ylim(0,self.range+1)
plt.xlabel(label)
plt.ylabel('Height(Km)')
if self.variable=="R":
plt.xlim(-1,3)
if self.variable=='D':
plt.xlim(-10,10)
avaldezp
UPDATE TEST ZDR
 r1586 if self.variable=='Z':
plt.xlim(-20,80)
avaldezp
UPDATE ZDR
 r1584
def save_PIC(self,count,time_save):
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 print("save",count)
if self.channel=='H':
ch_='0'
else:
ch_='1'
avaldezp
UPDATE ZDR
 r1584 if count ==1:
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 #filename = "SOPHY"+"_"+time_save+"_"+self.mode+"_"+self.grado+"_"+self.variable+str(self.range)+".png"
filename = "SOPHY"+"_"+time_save+"_"+"E."+self.grado+"_"+self.variable+" "+self.channel+" "+str(self.range)+".png"
dir =self.variable+"_"+self.mode+"_"+self.grado+"_"+"CH"+ch_+"/"
avaldezp
UPDATE ZDR
 r1584 filesavepath = os.path.join(self.path_file,dir)
try:
os.mkdir(filesavepath)
except:
pass
else:
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 dir =self.variable+"_"+self.mode+"_"+self.grado+"_"+"CH"+ch_+"/"
avaldezp
UPDATE ZDR
 r1584 filesavepath = os.path.join(self.path_file,dir)
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 filename = "SOPHY"+"_"+time_save+"_"+"E."+self.grado+"_"+self.variable+" "+self.channel+" "+str(self.range)+".png"
avaldezp
UPDATE ZDR
 r1584 plt.savefig(filesavepath+filename)
avaldezp
UPDATE TEST ZDR
 r1586 def seleccion_roHV_min(self,count,z,y,arr_):
##print("y",y)
avaldezp
UPDATE ZDR
 r1584 if self.variable=='R':
len_Z= z.shape[1]
min_CC=numpy.zeros(len_Z)
min_index_CC = numpy.zeros(len_Z)
for i in range(len_Z):
tmp=numpy.nanmin(z[:,i])
tmp_index = numpy.nanargmin((z[:,i]))
avaldezp
UPDATE TEST ZDR
 r1586 if tmp <0.5:
avaldezp
UPDATE ZDR
 r1584 tmp_index= numpy.nan
value = numpy.nan
else:
avaldezp
UPDATE TEST ZDR
 r1586 value= y[tmp_index]
avaldezp
UPDATE ZDR
 r1584 min_CC[i] =value
moda_,count_m_ = stats.mode(min_CC)
avaldezp
UPDATE TEST ZDR
 r1586 #print("MODA",moda_)
avaldezp
UPDATE ZDR
 r1584 for i in range(len_Z):
if min_CC[i]>moda_[0]+0.15 or min_CC[i]<moda_[0]-0.15:
min_CC[i]=numpy.nan
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 #print("MIN_CC",min_CC)
#print("y[0]",y[0])
avaldezp
UPDATE TEST ZDR
 r1586 min_index_CC=((min_CC-y[0])/0.06)
avaldezp
UPDATE ZDR
 r1584 if count == 0:
arr_ = min_index_CC
else:
arr_ = numpy.append(arr_,min_index_CC)
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 #print("arr_",min_index_CC)
avaldezp
UPDATE ZDR
 r1584 return arr_
else:
avaldezp
UPDATE TEST ZDR
 r1586 print("Operation JUST for roHV - EXIT ")
exit()
def pp_BB(self,count,x,y,z,filename,c_max,prom_List):
#print("z shape",z.shape)
len_Z = z.shape[1]
len_X = x.shape[0]
try:
min_CC = numpy.load(filename)
except:
print("There is no file")
exit()
if count ==1:
c_min = 0
c_max = c_max+ len_X
plt.plot(x,y[0]+min_CC[c_min:c_max]*0.06,'yo')
else:
c_min = c_max
c_max = c_min+len_X
try:
plt.plot(x,y[0]+min_CC[c_min:c_max]*0.06,'yo')
except:
print("Check number of file")
return 0
bb = numpy.zeros(len_Z)
min_READ_CC = min_CC[c_min:c_max]
#print(min_READ_CC[0:50])
for i in range(len_Z):
if min_READ_CC[i]==numpy.nan:
bb[i]=numpy.nan
try:
bb[i]=z[int(min_READ_CC[i])][i]
except:
bb[i]=numpy.nan
print("bb _ prom_ZDR",numpy.nanmean(bb))
prom_List.append(numpy.nanmean(bb))
return c_max
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 def plot_WR(self,count,fig,z,r,az,rf,my_time,label,unit,cmap,vmin,vmax):
if count ==1:
plt.rcParams['axes.facecolor']='black'
cgax, pm = wradlib.vis.plot_ppi(data=z,r=r,az=az,rf=1,fig=fig,ax=111,proj='cg',cmap=cmap,vmin=vmin,vmax=vmax)
caax = cgax.parasites[0]
title = 'Sophy Plot'+"-"+label+"-"+ my_time+" "+self.mode+"_"+self.channel+"_"+self.grado+"_"+" N° "+str(count)
t = plt.title(title, fontsize=10,y=1.05)
cbar = plt.gcf().colorbar(pm,ax=cgax,shrink=0.7)
plt.text(1.0, 1.05, 'azimuth', transform=caax.transAxes, va='bottom',ha='right')
cbar.set_label(label+'[' + unit + ']')
caax.set_xlabel('x_range [km]')
caax.set_ylabel('y_range [km]')
gh = cgax.get_grid_helper()
else:
cgax, pm = wradlib.vis.plot_ppi(data=z,r=r,az=az,rf=1,fig=fig,ax=111,proj='cg',cmap=cmap,vmin=vmin,vmax=vmax)
caax = cgax.parasites[0]
title = 'Sophy Plot'+"-"+label+"-"+ my_time+" "+self.mode+"_"+self.channel+"_"+self.grado+"_"+" N° "+str(count)
t = plt.title(title, fontsize=10,y=1.05)
cbar = plt.gcf().colorbar(pm,ax=cgax,shrink=0.7)
plt.text(1.0, 1.05, 'azimuth', transform=caax.transAxes, va='bottom',ha='right')
cbar.set_label(label+'[' + unit + ']')
caax.set_xlabel('x_range [km]')
caax.set_ylabel('y_range [km]')
gh = cgax.get_grid_helper()
def plot_D(self,count,h,x,y,my_time,label,mode,grado):
if count==1:
fig =plt.figure(figsize=(16,8))
ax1=plt.subplot(1,2,1)
ax2=plt.subplot(1,2,2)
ax1.plot(h,x,"bo",color="red", linestyle="dotted", lw=3,label='Zh')
ax1.plot(h,y,"bo",color="blue", linestyle="dotted", lw=3,label='Zv')
title = 'Sophy Plot '+ "Zh Zv"+"-"+ my_time+" "+mode+" "+grado+" N° "+str(count)
fig.suptitle(title, fontsize=18)
#plt.ylim(0,self.range+1)
ax1.legend(loc=1,prop={'size': 16})
ax1.set_title("Z- Reflectividad vs H(Km)", fontsize=15)
ax1.set_ylabel('Z Reflectivity')
ax1.set_xlabel('H(Km)')
ax1.set_ylim(-20,70)
m= numpy.arange(100)-20
ax2.plot(x,y,"bo",color="red", linestyle="dotted", lw=3,label='Zh')
ax2.plot(m,m,"bo",color="green", linestyle="dotted", lw=0.5,label='Zh')
ax2.set_title('ZV vs ZH', fontsize=16)
ax2.set_xlabel('Zh dBZ')
ax2.set_xlim(-20,70)
ax2.set_ylim(-20,70)
ax2.set_ylabel('Zv dBZ')
ax2.set_xlabel('Zh dBZ')
ax2.legend(loc=1,prop={'size': 16})
else:
ax1=plt.subplot(1,2,1)
ax2=plt.subplot(1,2,2)
ax1.plot(h,x,"bo",color="red", linestyle="dotted", lw=3,label='Zh')
ax1.plot(h,y,"bo",color="blue", linestyle="dotted", lw=3,label='Zv')
title = 'Sophy Plot '+ "Zh Zv"+"-"+ my_time+" "+mode+" "+grado+" N° "+str(count)
plt.suptitle(title, fontsize=18)
ax1.legend(loc=1,prop={'size': 16})
ax1.set_title("Z- Reflectividad vs H(Km)", fontsize=15)
ax1.set_ylabel('Z Reflectivity')
ax1.set_xlabel('H(Km)')
ax1.set_ylim(-20,70)
m= numpy.arange(100)-20
ax2.plot(x,y,"bo",color="red", linestyle="dotted", lw=3,label='Zv vs Zh')
ax2.plot(m,m,"bo",color="green", linestyle="dotted", lw=0.5,label='Zh=Zv')
ax2.set_title('ZV vs ZH', fontsize=16)
ax2.set_xlim(-20,70)
ax2.set_ylim(-20,70)
ax2.set_ylabel('Zv dBZ')
ax2.set_xlabel('Zh dBZ')
ax2.legend(loc=1,prop={'size': 16})
#plt.gcf()
avaldezp
UPDATE ZDR
 r1584
avaldezp
update new plots y processing
 r1581 def run(self):
count = 0
len_files = len(self.list_file)
avaldezp
UPDATE ZDR
 r1584 SAVE_PARAM= []
avaldezp
update new plots y processing
 r1581 for thisFile in self.list_file:
count= count +1
print("Count :", count)
fullpathfile = self.path_file + thisFile
filename = get_wradlib_data_file(fullpathfile)
test_hdf5 = read_generic_hdf5(filename)
# LECTURA
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 data_arr, utc_time, data_azi,data_ele, heightList,unit,cmap,vmin,vmax,label = self.readAttributes(obj= test_hdf5,variable=self.variable,channel=self.channel,type_=self.type_)
avaldezp
UPDATE ZDR
 r1584 # SELECCION DE ALTURAS
avaldezp
update new plots y processing
 r1581 if self.range==0:
self.range == heightList[-1]
avaldezp
UPDATE ZDR
 r1582 if self.r_min==0:
self.r_min = 0.01
new_heightList,minIndex,maxIndex = self.selectHeights(heightList,self.r_min,self.range)
avaldezp
update new plots y processing
 r1581 # TIEMPO
utc_time[0] = utc_time[0]+60*60*5
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]))
avaldezp
UPDATE ZDR
 r1582 # VARIABLES
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 if self.type_=='Diagonal':
x= numpy.nanmean(data_arr[0][:,minIndex:maxIndex].transpose(),1)
y= numpy.nanmean(data_arr[1][:,minIndex:maxIndex].transpose(),1)
h= new_heightList
else:
len_X= data_arr.shape[0]
data_arr_= data_arr[:,minIndex:maxIndex].transpose()
x=numpy.linspace(1,len_X,len_X)
y= new_heightList
z=data_arr_
profile = numpy.mean(z,1)
avaldezp
update new plots y processing
 r1581 if count ==1:
avaldezp
UPDATE ZDR
 r1584 if self.type_ =="PROFILE":
self.plot_PROFILE(count=count ,z=z,y=y,my_time=my_time,label=label,mode=self.mode,grado=self.grado)
if self.type_ =="RTI":
self.plot_RTI_PPI_RHI(count=count,x=x,y=y,z=z,cmap=cmap,my_time=my_time,vmin=vmin,vmax=vmax,label =label,unit=unit, mode=self.mode,grado=self.grado)
if self.type_ =='roHV_MIN':
avaldezp
UPDATE TEST ZDR
 r1586 arr_= self.seleccion_roHV_min(count=count,z=z,y=y,arr_= 0)
if self.type_ =='pp_BB':
self.plot_RTI_PPI_RHI(count=count,x=x,y=y,z=z,cmap=cmap,my_time=my_time,vmin=vmin,vmax=vmax,label =label,unit=unit, mode=self.mode,grado=self.grado)
c_max = self.pp_BB(count=count,x=x,y=y,z=z,filename='index.npy',c_max=0,prom_List=SAVE_PARAM)
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 if self.type_ == "plot_WR":
fig =plt.figure(figsize=(10,8))
self.plot_WR(count=count,fig=fig,z=data_arr[:,minIndex:maxIndex],r=y,az=data_azi,rf=1,my_time=my_time,label=label,unit=unit,cmap=cmap,vmin=vmin,vmax=vmax)
if self.type_ =="Diagonal":
self.plot_D(count=count,h=h,x=x,y=y,my_time=my_time,label="Zv vs Zh",mode=self.mode,grado=self.grado)
avaldezp
update new plots y processing
 r1581 else:
avaldezp
UPDATE ZDR
 r1584 if self.type_=="RTI":
self.plot_RTI_PPI_RHI(count=count,x=x,y=y,z=z,cmap=cmap,my_time=my_time,vmin=vmin,vmax=vmax,label =label,unit=unit, mode=self.mode,grado=self.grado)
if self.type_ =="PROFILE":
self. plot_PROFILE(count=count,z=z,y=y,my_time=my_time,label=label,mode=self.mode,grado=self.grado)
if self.type_ =='roHV_MIN':
avaldezp
UPDATE TEST ZDR
 r1586 arr_= self.seleccion_roHV_min(count=count,z=z,y=y,arr_= arr_)
if self.type_ =='pp_BB':
self.plot_RTI_PPI_RHI(count=count,x=x,y=y,z=z,cmap=cmap,my_time=my_time,vmin=vmin,vmax=vmax,label =label,unit=unit, mode=self.mode,grado=self.grado)
c_max = self.pp_BB(count=count,x=x,y=y,z=z,filename='index.npy',c_max=c_max,prom_List=SAVE_PARAM)
if c_max ==0:
count=len_files
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 if self.type_ == "plot_WR":
self.plot_WR(count=count,fig=fig,z=data_arr[:,minIndex:maxIndex],r=y,az=data_azi,rf=1,my_time=my_time,label=label,unit=unit,cmap=cmap,vmin=vmin,vmax=vmax)
if self.type_ =="Diagonal":
self.plot_D(count=count,h=h,x=x,y=y,my_time=my_time,label="Zv vs Zh",mode=self.mode,grado=self.grado)
avaldezp
update new plots y processing
 r1581 if self.save == 1:
avaldezp
UPDATE ZDR
 r1584 self.save_PIC(count=count,time_save=time_save)
plt.pause(1)
avaldezp
update new plots y processing
 r1581 plt.clf()
if count == len_files:
avaldezp
UPDATE ZDR
 r1584 if self.type_ =='roHV_MIN':
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 numpy.save("/media/soporte/DATA/soporte/WRJAN2023/schain/schainpy/scripts/index.npy",arr_)
avaldezp
UPDATE TEST ZDR
 r1586 if self.type_ =='pp_BB':
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 numpy.save("/media/soporte/DATA/soporte/WRJAN2023/schain/schainpy/scripts/index_"+self.variable+"_prom.npy",SAVE_PARAM)
print("---------------------END---------------------")
avaldezp
update new plots y processing
 r1581 plt.close()
avaldezp
UPDATE TEST ZDR
 r1586 exit()
avaldezp
update new plots y processing
 r1581 plt.show()
def main(args):
grado = args.grado
parameters = args.parameters
avaldezp
UPDATE ZDR
 r1582 r_min = args.r_min
avaldezp
update new plots y processing
 r1581 save = args.save
range = args.range
mode = args.mode
avaldezp
UPDATE ZDR
 r1584 type = args.type
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 channel = args.channel
avaldezp
update new plots y processing
 r1581 obj = Readsophy()
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 print("------------------START-------------------")
avaldezp
update new plots y processing
 r1581 print("MODE :", mode)
if not mode =='PPI' and not mode =='RHI':
print("Error - Choose Mode RHI or PPI")
return None
for param in parameters:
print("Parameters : ", param)
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 DIR_ = "NORMAL"
avaldezp
update new plots y processing
 r1581 if mode =='PPI':
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 #PATH = "/media/soporte/TOSHIBAEXT/sophy/HYO_CC4_CC64_COMB@2022-12-26T00-00-32/param-EVENTO/"+str(param)+"_PPI_EL_"+str(grado)+".0/"
#PATH = "/media/soporte/TOSHIBAEXT/sophy/HYO_CC4_CC64_COMB@2022-12-27T00-00-32/param-FIXIT/"+str(param)+"_PPI_EL_"+str(grado)+".0/"
PATH = "/media/soporte/DATA/sophy/HYO_CC4_CC64_COMB@2022-12-27T00-00-32/param-"+DIR_+"/"+str(param)+"_PPI_EL_"+str(grado)+".0/"
avaldezp
update new plots y processing
 r1581 else:
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 #PATH = "/media/soporte/TOSHIBAEXT/sophy/HYO_CC4_CC64_COMB@2022-12-26T00-00-32/param-EVENTO/"+str(param)+"_RHI_AZ_"+str(grado)+".0/"
PATH = "/media/soporte/DATA/sophy/HYO_CC4_CC64_COMB@2022-12-27T00-00-32/param-"+DIR_+"/"+str(param)+"_PPI_EL_"+str(grado)+".0/"
avaldezp
update new plots y processing
 r1581 print("Path : ",PATH)
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 obj.setup(path_file =PATH,mode=mode,channel=channel,type=type,grado = grado,range=range,r_min=r_min, variable=param,save=int(save))
avaldezp
update new plots y processing
 r1581 print("SETUP OK")
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,D')
parser.add_argument('--grado', default=2,
help='Angle in Elev to plot')
parser.add_argument('--mode',default='PPI',
help='MODE PPI or RHI')
parser.add_argument('--save', default=0,
help='Save plot')
parser.add_argument('--range', default=0, type=float,
help='Max range to plot')
avaldezp
UPDATE ZDR
 r1582 parser.add_argument('--r_min', default=0, type=float,
help='Min range to plot')
avaldezp
UPDATE ZDR
 r1584 parser.add_argument('--type', default='RTI',
help='TYPE Profile or RTI')
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591 parser.add_argument('--channel', default='H',
help='Choose H or V')
avaldezp
update new plots y processing
 r1581 args = parser.parse_args()
main(args)
avaldez
Update para ploteo de Zh vs Zv , METODO plot_D, atributo Diagonal
 r1591
#python test_zdr.py --parameters Z --grado 90 --mode 'PPI' --range 6 --r_min 0.5 --type plot_WR --channel H --save 1