schain Commit - r1519:9365e2e64639 · Jicamarca Repository

update script genera and script with library wradlib

avaldezp -

r1519:9365e2e64639

parent child

Context file:

r1519:9365e2e64639

Collapse all files

schainpy/scripts/sophy_proc.py +11 -2

             PATH = '/DATA_RM/DATA'
             PATH = '/media/jespinoza/Elements'
             PATH = '/media/jespinoza/data/SOPHY'
+            PATH = '/home/soporte/Documents/EVENTO'
             PARAM = {
                 'S': {'zmin': -45, 'zmax': -25, 'colormap': 'jet', 'label': 'Power', 'wrname': 'power','cb_label': 'dBm', 'ch':0},
                 N = int(1/(speed_axis[0]*ipp))                                               # 1 GRADO DE RESOLUCION
                 path = os.path.join(PATH, experiment, 'rawdata')
                 path_ped = os.path.join(PATH, experiment, 'position')
-                path_plots = os.path.join(PATH, experiment, 'plots')
+                path_plots = os.path.join(PATH, experiment, 'plotsC0N'+str(args.range))
-                path_save = os.path.join(PATH, experiment, 'param')
+                path_save = os.path.join(PATH, experiment, 'paramC0N'+str(args.range))
                 RMIX = 1.62
                 H0 = -1.68
                 MASK = 0.3
                     op = voltage.addOperation(name='PulsePair_vRF', optype='other')
                     op.addParameter(name='n', value=int(N)/ncode, format='int')
+                    #op.addParameter(name='removeDC', value=1, format='int')
                     proc = project.addProcUnit(datatype='ParametersProc', inputId=voltage.getId())
                     op = voltage1.addOperation(name='PulsePair_vRF', optype='other')
                     op.addParameter(name='n', value=int(conf['usrp_tx']['repetitions_1'])/2, format='int')
+                    #op.addParameter(name='removeDC', value=1, format='int')
                     proc1 = project.addProcUnit(datatype='ParametersProc', inputId=voltage1.getId())
                     proc1.addParameter(name='runNextUnit', value=True)
                     op = voltage2.addOperation(name='PulsePair_vRF', optype='other')
                     op.addParameter(name='n', value=int(conf['usrp_tx']['repetitions_2'])/ncode, format='int')
+                    #op.addParameter(name='removeDC', value=1, format='int')
                     proc2 = project.addProcUnit(datatype='ParametersProc', inputId=voltage2.getId())
                     proc2.addParameter(name='runNextUnit', value=True)
                 args = parser.parse_args()
                 main(args)
+            # python sophy_proc.py  HYO_PM@2022-06-09T15-05-12  --parameters V --plot --save --show --range 36

schainpy/scripts/sophy_proc_rev006.py +75 -17

             #LINUX bash: export WRADLIB_DATA=/path/to/wradlib-data
             warnings.filterwarnings('ignore')
             PARAM = {
-                'S': {'var': 'dataPP_POWER','vmin': -45, 'vmax': -15, 'cmap': 'jet', 'label': 'Power','unit': 'dBm'},
+                'S': {'var': 'power','vmin': -45, 'vmax': -15, 'cmap': 'jet', 'label': 'Power','unit': 'dBm'},
-                'V': {'var': 'velRadial_V', 'vmin': -10, 'vmax': 10 , 'cmap': 'sophy_v', 'label': 'Velocity','unit': 'm/s'},
+                'V': {'var': 'velocity', 'vmin': -10, 'vmax': 10 , 'cmap': 'sophy_v', 'label': 'Velocity','unit': 'm/s'},
-                'Z': {'var': 'Zdb',         'vmin': -30, 'vmax': 80 , 'cmap': 'sophy_r','label': 'Reflectivity','unit': 'dBZ'},
+                'Z': {'var': 'reflectivity','vmin': -30, 'vmax': 80 , 'cmap': 'sophy_r','label': 'Reflectivity','unit': 'dBZ'},
-                'W': {'var': 'Sigmav_W',    'vmin': 0  , 'vmax': 12 , 'cmap': 'sophy_w','label': 'Spectral Width','unit': 'hz'}
+                'W': {'var': 'spectral_width',    'vmin': 0  , 'vmax': 12 , 'cmap': 'sophy_w','label': 'Spectral Width','unit': 'hz'}
                 }
             class Readsophy():
                 def __init__(self):
                     self.grado     = None
                     self.variable  = None
                     self.save      = None
+                    self.range     = None
                 def read_files(self,path_file,grado=None, variable=None):
                     filter= "_E"+str(grado)+".0_"+variable
                         validFilelist.sort()
                     return validFilelist
-                def setup(self, path_file,grado,variable,save):
+                def setup(self, path_file,grado,range,variable,save):
                     self.path_file = path_file
+                    self.range     = range
                     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)
+                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
                 def run(self):
                     count= 0
                     len_files = len(self.list_file)
                         vmin   = PARAM[self.variable]['vmin']
                         vmax   = PARAM[self.variable]['vmax']
                         label  = PARAM[self.variable]['label']
-                        var_    = 'Data/'+var+'/channel00'
+                        var_    = 'Data/'+var+'/H'
                         data_arr   = numpy.array(test_hdf5[var_]['data']) # data
-                        utc_time   = numpy.array(test_hdf5['Data/utctime']['data'])
+                        utc_time   = numpy.array(test_hdf5['Data/time']['data'])
-                        data_azi   = numpy.array(test_hdf5['Metadata/data_azi']['data']) # th
+                        data_azi   = numpy.array(test_hdf5['Metadata/azimuth']['data']) # th
-                        data_ele   = numpy.array(test_hdf5["Metadata/data_ele"]['data'])
+                        data_ele   = numpy.array(test_hdf5["Metadata/elevation"]['data'])
-                        heightList = numpy.array(test_hdf5["Metadata/heightList"]['data']) # r
+                        heightList = numpy.array(test_hdf5["Metadata/range"]['data']) # r
+                        if self.range==0:
+                            self.range == heightList[-1]
+                        else:
+                            print(self.range)
+                        new_heightList,minIndex,maxIndex = self.selectHeights(heightList,0.06,self.range)
                         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))
-                            cgax, pm = wradlib.vis.plot_ppi(data_arr,r=heightList,az=data_azi,rf=1e3,fig=fig, ax=111,proj='cg',cmap=cmap,vmin=vmin, vmax=vmax)
+                            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)
                             caax  = cgax.parasites[0]
                             title = 'Simple PPI'+"-"+ my_time
                             t = pl.title(title, fontsize=12,y=1.05)
                             cbar.set_label(label+'[' + unit + ']')
                             gh = cgax.get_grid_helper()
                         else:
-                            cgax, pm = wradlib.vis.plot_ppi(data_arr,r=heightList,az=data_azi,rf=1e3,fig=fig, ax=111,proj='cg',cmap=cmap,vmin=vmin, vmax=vmax)
+                            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)
                             caax  = cgax.parasites[0]
                             title = 'Simple PPI'+"-"+my_time
                             t = pl.title(title, fontsize=12,y=1.05)
                                 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)
-                                os.mkdir(filesavepath)
+                                try:
+                                    os.mkdir(filesavepath)
+                                except:
+                                    pass
                             else:
                                 filename     = "SOPHY"+"_"+time_save+"_"+"E."+self.grado+"_"+self.variable+".png"
                             pl.savefig(filesavepath+filename)
-                        pl.pause(2)
+                        pl.pause(1)
                         pl.clf()
                     if count==len_files:
                         pl.close()
                     pl.show()
-            PATH = "/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/paramC0_FD_PL_R15.0km/2022-06-09T18-00-00/"
+            PATH = "/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/paramC0N36.0/2022-06-09T18-00-00/"
+            PATH = "/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/paramC0N36.0/2022-06-09T19-00-00/"
+            #PATH = "/home/soporte/Documents/EVENTO/HYO_PM@2022-05-31T12-00-17/paramC0N36.0/2022-05-31T16-00-00/"
             def main(args):
                 grado      = args.grado
                 parameters = args.parameters
                 save       = args.save
+                range      = args.range
                 obj        = Readsophy()
                 for param in parameters:
-                    obj.setup(path_file = PATH,grado = grado, variable=param,save=int(save))
+                    obj.setup(path_file = PATH,grado = grado,range=range, variable=param,save=int(save))
                     obj.run()
             if __name__ == '__main__':
                                     help='Angle in Elev to plot')
                 parser.add_argument('--save', default=0,
                                     help='Save plot')
+                parser.add_argument('--range', default=0, type=float,
+                                    help='Max range to plot')
                 args = parser.parse_args()
                 main(args)
+            #python sophy_proc_rev006.py  --parameters Z --grado 8 --save 1 --range 28
             '''
             def read_and_overview(filename):
                 """Read HDF5 using read_generic_hdf5 and print upper level dictionary keys

General Comments 0

Write
Preview

You need to be logged in to leave comments. Login now

No TODOs yet

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages