schain Commit - r1531:a8e64958fb6a · Jicamarca Repository

Merge branch 'v3.0-WR' of http://intranet.igp.gob.pe:8082/schain into v3.0-WR

Juan C. Espinoza -

r1531:a8e64958fb6a

parent child

Context file:

r1531:a8e64958fb6a

Collapse all files

schainpy/scripts/giff_wr.py +23 -5

             # Importra librerías
             import os
             #import imageio
             import imageio.v2 as imageio
             # Ubicación de la base de datos
+            '''
             #path = '/home/soporte/Downloads/plots_C0/S/6/6_evento0/'
             path=  '/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/plotsC0_PL_R60.0km/V_PPI_EL_8.0CH0/'
             path=  '/home/soporte/Documents/EVENTO/HYO_PM@2022-05-31T12-00-17/plotsC0_PL_R60.0km_removeDC/V_PPI_EL_8.0CH0/'
             path=  '/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/plotsC0_FD_PL_R15.0km_removeDC/Z_PPI_EL_2.0CH0/'
             path=  '/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/plotsC0_FD_PL_R15.0km/Z_PPI_EL_8.0CH0/'
             path= '/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/paramC0N36.0/2022-06-09T18-00-00/W_E.2CH0/'
+            '''
+            #--------------------------------------------------------------------------------------------------------
+            #pulse pair
+            #path = '/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/plotsC0_PL_R28.0km/Z_PPI_EL_2.0CH0/'
+            #path = '/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/plotsC0_PL_R28.0km/W_PPI_WE_L8.0CH0/'
+            #---------------------------------------------------------------------------------------------------------
+            #pulse pair REMOVEDC
+            #path = '/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/plotsC0_PL_R28.0km_removeDC/V_PPI_EL_2.0CH0/'
+            #path = '/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/plotsC0_PL_R28.0km_removeDC/W_PPI_WE_L8.0CH0/'
+            #----------------------------------------------------------------------------------------------------------
+            #Spectro
+            #path = '/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/plotsC0_FD_PL_R28.0km/V_PPI_EL_2.0CH0/'
+            path = '/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/plotsC0_FD_PL_R28.0km/W_PPI_WE_L8.0CH0/'
+            #---------------------------------------------------------------------------------------------------------
+            #Spectro REMOVEDC
+            path = '/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/plotsC0_FD_PL_R28.0km_removeDC/V_PPI_EL_8.0CH0/'
+            #path = '/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/plotsC0_FD_PL_R28.0km_removeDC/W_PPI_WE_L2.0CH0/'
             archivos = sorted(os.listdir(path))
             img_array = []
             #Leer todos los archivos formato imagen desde path
             for x in range(0, len(archivos)):
                 nomArchivo = archivos[x]
                 dirArchivo = path + str(nomArchivo)
                 #Asignar a variable leer_imagen, el nombre de cada imagen
                 leer_imagen = imageio.imread(dirArchivo)
                 # añadir imágenes al arreglo img_array
                 img_array.append(leer_imagen)
             #Guardar Gif
-            imageio.mimwrite('evento_PL_PP_2_28KM_W', img_array, 'GIF', duration=0.5)
+            #imageio.mimwrite('/home/soporte/Documents/giff_28km/evento_PL_PP_8_28KM_W', img_array, 'GIF', duration=0.5)
+            #imageio.mimwrite('/home/soporte/Documents/giff_28km/evento_PL_PP_8_28KM_W_reDC', img_array, 'GIF', duration=0.5)
+            imageio.mimwrite('/home/soporte/Documents/giff_28kmFD/evento_PL_FD_8_28KM_V_reDC', img_array, 'GIF', duration=0.5)

schainpy/scripts/sophy_proc.py +2 -2

             # SOPHY PROC script
             import os, sys, json, argparse
             import datetime
             import time
             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},
                 'SNR': {'zmin': -40, 'zmax': -20, 'colormap': 'jet', 'label': 'SNR', 'wrname': 'snr','cb_label': 'dB', 'ch':0},
                 'V': {'zmin': -12, 'zmax': 12, 'colormap': 'sophy_v', 'label': 'Velocity', 'wrname': 'velocity', 'cb_label': 'm/s', 'ch':0},
                 'R': {'zmin': 0,   'zmax': 1,  'colormap': 'jet',    'label': 'RhoHV', 'wrname':'rhoHV', 'cb_label': '*',  'ch':0},
                 'P': {'zmin': -180,'zmax': 180,'colormap': 'RdBu_r', 'label': 'PhiDP', 'wrname':'phiDP' , 'cb_label': 'º',  'ch':0},
                 'D': {'zmin': -30, 'zmax': 80, 'colormap': 'sophy_r','label': 'ZDR','wrname':'differential_reflectivity' , 'cb_label': 'dBz','ch':0},
                 'Z':  {'zmin': -30, 'zmax': 80, 'colormap': 'sophy_r','label': 'Reflectivity ',  'wrname':'reflectivity', 'cb_label': 'dBz','ch':0},
-                'W':  {'zmin': 0, 'zmax': 15, 'colormap': 'sophy_w','label': 'Spectral Width', 'wrname':'spectral_width', 'cb_label': 'hz', 'ch':0}
+                'W':  {'zmin': 0, 'zmax': 15, 'colormap': 'sophy_w','label': 'Spectral Width', 'wrname':'spectral_width', 'cb_label': 'm/s', 'ch':0}
                 }
             def max_index(r, sample_rate, ipp):
                 return int(sample_rate*ipp*1e6 * r / 60) + int(sample_rate*ipp*1e6 * 1.2 / 60)
             def main(args):
                 experiment = args.experiment
                 fp = open(os.path.join(PATH, experiment, 'experiment.conf'))
                 conf = json.loads(fp.read())
                 ipp_km = conf['usrp_tx']['ipp']
                 ipp = ipp_km * 2 /300000
                 sample_rate  = conf['usrp_rx']['sample_rate']
                 axis = ['0' if x=='elevation' else '1' for x in conf['pedestal']['axis']]      # AZIMUTH 1 ELEVACION 0
                 speed_axis = conf['pedestal']['speed']
                 steps = conf['pedestal']['table']
                 time_offset = args.time_offset
                 parameters = args.parameters
                 start_date = experiment.split('@')[1].split('T')[0].replace('-', '/')
                 end_date = start_date
                 if args.start_time:
                     start_time = args.start_time
                 else:
                     start_time = experiment.split('@')[1].split('T')[1].replace('-', ':')
                 end_time = '23:59:59'
                 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, 'plotsC0N'+str(args.range))
                 path_save = os.path.join(PATH, experiment, 'paramC0N'+str(args.range))
                 RMIX = 1.62
                 H0 = -1.68
                 MASK = 0.3
                 from schainpy.controller import Project
                 project = Project()
                 project.setup(id='1', name='Sophy', description='sophy proc')
                 reader = project.addReadUnit(datatype='DigitalRFReader',
                     path=path,
                     startDate=start_date,
                     endDate=end_date,
                     startTime=start_time,
                     endTime=end_time,
                     delay=30,
                     online=args.online,
                     walk=1,
                     ippKm = ipp_km,
                     getByBlock = 1,
                     nProfileBlocks = N,
                 )
                 if not conf['usrp_tx']['enable_2']: # One Pulse
                     voltage = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId())
                     if conf['usrp_tx']['code_type_1'] != 'None':
                         codes = [ c.strip() for c in conf['usrp_tx']['code_1'].split(',')]
                         code = []
                         for c in codes:
                             code.append([int(x) for x in c])
                         op = voltage.addOperation(name='Decoder', optype='other')
                         op.addParameter(name='code', value=code)
                         op.addParameter(name='nCode', value=len(code), format='int')
                         op.addParameter(name='nBaud', value=len(code[0]), format='int')
                         op = voltage.addOperation(name='CohInt', optype='other') #Minimo integrar 2 perfiles por ser codigo complementario
                         op.addParameter(name='n', value=len(code), format='int')
                         ncode = len(code)
                     else:
                         ncode = 1
                         code = ['0']
                     op = voltage.addOperation(name='setH0')
                     op.addParameter(name='h0', value=H0)
                     if args.range > 0:
                         op = voltage.addOperation(name='selectHeights')
                         op.addParameter(name='minIndex', value='0', format='int')
                         op.addParameter(name='maxIndex', value=max_index(args.range, sample_rate, ipp), format='int')
                     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())
                     opObj10 = proc.addOperation(name="WeatherRadar")
                     opObj10.addParameter(name='tauW',value=(1e-6/sample_rate)*len(code[0]))
                     opObj10.addParameter(name='Pt',value=((1e-6/sample_rate)*len(code[0])/ipp)*200)
                     op = proc.addOperation(name='PedestalInformation')
                     op.addParameter(name='path', value=path_ped, format='str')
                     op.addParameter(name='interval', value='0.04')
                     op.addParameter(name='time_offset', value=time_offset)
                     op.addParameter(name='mode', value='PPI')
                     for param in parameters:
                         op = proc.addOperation(name='Block360')
                         op.addParameter(name='runNextOp', value=True)
                         op= proc.addOperation(name='WeatherParamsPlot')
                         if args.save: op.addParameter(name='save', value=path_plots, format='str')
                         op.addParameter(name='save_period', value=-1)
                         op.addParameter(name='show', value=args.show)
                         op.addParameter(name='channels', value='1,')
                         op.addParameter(name='zmin', value=PARAM[param]['zmin'])
                         op.addParameter(name='zmax', value=PARAM[param]['zmax'])
                         op.addParameter(name='attr_data', value=param, format='str')
                         op.addParameter(name='labels', value=[PARAM[param]['label']])
                         op.addParameter(name='save_code', value=param)
                         op.addParameter(name='cb_label', value=PARAM[param]['cb_label'])
                         op.addParameter(name='colormap', value=PARAM[param]['colormap'])
                         op.addParameter(name='bgcolor', value='black')
                         if MASK: op.addParameter(name='mask', value=MASK, format='float')
                         if args.server:
                             op.addParameter(name='server', value='0.0.0.0:4444')
                             op.addParameter(name='exp_code', value='400')
                         desc = {
                                 'Data': {
                                     param: PARAM[param]['wrname'],
                                     'utctime': 'time'
                                 },
                                  'Metadata': {
                                     'heightList': 'range',
                                     'data_azi': 'azimuth',
                                     'data_ele': 'elevation',
                                     'mode_op': 'scan_type',
                                     'h0': 'range_correction',
                                 }
                             }
                         if args.save:
                             opObj10 = proc.addOperation(name='HDFWriter')
                             writer.addParameter(name='path', value=path_save, format='str')
                             writer.addParameter(name='Reset', value=True)
                             writer.addParameter(name='setType', value='weather')
                             writer.addParameter(name='description', value=json.dumps(desc))
                             writer.addParameter(name='blocksPerFile', value='1',format='int')
                             writer.addParameter(name='metadataList', value='heightList,data_azi,data_ele,mode_op,latitude,longitude,altitude,heading,radar_name,institution,contact,h0,range_unit')
                             writer.addParameter(name='dataList', value='{},utctime'.format(param))
                             writer.addParameter(name='mask', value=MASK, format='float')
                             # meta
                             writer.addParameter(name='latitude', value='-12.040436')
                             writer.addParameter(name='longitude', value='-75.295893')
                             writer.addParameter(name='altitude', value='3379.2147')
                             writer.addParameter(name='heading', value='0')
                             writer.addParameter(name='radar_name', value='SOPHy')
                             writer.addParameter(name='institution', value='IGP')
                             writer.addParameter(name='contact', value='dscipion@igp.gob.pe')
                             writer.addParameter(name='created_by', value='Signal Chain (https://pypi.org/project/schainpy/)')
                             writer.addParameter(name='range_unit', value='km')
                 else: #Two pulses
                     voltage1 = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId())
                     op = voltage1.addOperation(name='ProfileSelector')
                     op.addParameter(name='profileRangeList', value='0,{}'.format(conf['usrp_tx']['repetitions_1']-1))
                     if conf['usrp_tx']['code_type_1'] != 'None':
                         codes = [ c.strip() for c in conf['usrp_tx']['code_1'].split(',')]
                         code = []
                         for c in codes:
                             code.append([int(x) for x in c])
                         op = voltage1.addOperation(name='Decoder', optype='other')
                         op.addParameter(name='code', value=code)
                         op.addParameter(name='nCode', value=len(code), format='int')
                         op.addParameter(name='nBaud', value=len(code[0]), format='int')
                     else:
                         code = ['0']
                     op = voltage1.addOperation(name='CohInt', optype='other') #Minimo integrar 2 perfiles por ser codigo complementario
                     op.addParameter(name='n', value=2, format='int')
                     if args.range > 0:
                         op = voltage1.addOperation(name='selectHeights')
                         op.addParameter(name='minIndex', value='0', format='int')
                         op.addParameter(name='maxIndex', value=max_index(RMIX, sample_rate, ipp), format='int')
                     op = voltage1.addOperation(name='setH0')
                     op.addParameter(name='h0', value=H0, format='float')
                     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)
                     opObj10 = proc1.addOperation(name="WeatherRadar")
                     opObj10.addParameter(name='tauW',value=(1e-6/sample_rate)*len(code[0]))
                     opObj10.addParameter(name='Pt',value=((1e-6/sample_rate)*len(code[0])/ipp)*200)
                     op = proc1.addOperation(name='PedestalInformation')
                     op.addParameter(name='path', value=path_ped, format='str')
                     op.addParameter(name='interval', value='0.04')
                     op.addParameter(name='time_offset', value=time_offset)
                     op.addParameter(name='mode', value='PPI')
                     op = proc1.addOperation(name='Block360')
                     op.addParameter(name='attr_data', value='data_param')
                     op.addParameter(name='runNextOp', value=True)
                     voltage2 = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId())
                     op = voltage2.addOperation(name='ProfileSelector')
                     op.addParameter(name='profileRangeList', value='{},{}'.format(conf['usrp_tx']['repetitions_1'], conf['usrp_tx']['repetitions_1']+conf['usrp_tx']['repetitions_2']-1))
                     if conf['usrp_tx']['code_type_2']:
                         codes = [ c.strip() for c in conf['usrp_tx']['code_2'].split(',')]
                         code = []
                         for c in codes:
                             code.append([int(x) for x in c])
                         op = voltage2.addOperation(name='Decoder', optype='other')
                         op.addParameter(name='code', value=code)
                         op.addParameter(name='nCode', value=len(code), format='int')
                         op.addParameter(name='nBaud', value=len(code[0]), format='int')
                         op = voltage2.addOperation(name='CohInt', optype='other') #Minimo integrar 2 perfiles por ser codigo complementario
                         op.addParameter(name='n', value=len(code), format='int')
                         ncode = len(code)
                     else:
                         ncode = 1
                     if args.range > 0:
                         op = voltage2.addOperation(name='selectHeights')
                         op.addParameter(name='minIndex', value=max_index(RMIX, sample_rate, ipp), format='int')
                         op.addParameter(name='maxIndex', value=max_index(args.range, sample_rate, ipp), format='int')
                     op = voltage2.addOperation(name='setH0')
                     op.addParameter(name='h0', value=H0, format='float')
                     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)
                     opObj10 = proc2.addOperation(name="WeatherRadar")
                     opObj10.addParameter(name='tauW',value=(1e-6/sample_rate)*len(code[0]))
                     opObj10.addParameter(name='Pt',value=((1e-6/sample_rate)*len(code[0])/ipp)*200)
                     op = proc2.addOperation(name='PedestalInformation')
                     op.addParameter(name='path', value=path_ped, format='str')
                     op.addParameter(name='interval', value='0.04')
                     op.addParameter(name='time_offset', value=time_offset)
                     op.addParameter(name='mode', value='PPI')
                     op = proc2.addOperation(name='Block360')
                     op.addParameter(name='attr_data', value='data_param')
                     op.addParameter(name='runNextOp', value=True)
                     merge = project.addProcUnit(datatype='MergeProc', inputId=[proc1.getId(), proc2.getId()])
                     merge.addParameter(name='attr_data', value='data_param')
                     merge.addParameter(name='mode', value='7') #RM
                     for param in parameters:
                         if args.plot:
                             op= merge.addOperation(name='WeatherParamsPlot')
                             if args.save:
                                 op.addParameter(name='save', value=path_plots, format='str')
                             op.addParameter(name='save_period', value=-1)
                             op.addParameter(name='show', value=args.show)
                             op.addParameter(name='channels', value='0,')
                             op.addParameter(name='zmin', value=PARAM[param]['zmin'], format='int')
                             op.addParameter(name='zmax', value=PARAM[param]['zmax'], format='int')
                             op.addParameter(name='attr_data', value=param, format='str')
                             op.addParameter(name='labels', value=[PARAM[param]['label']])
                             op.addParameter(name='save_code', value=param)
                             op.addParameter(name='cb_label', value=PARAM[param]['cb_label'])
                             op.addParameter(name='colormap', value=PARAM[param]['colormap'])
                             op.addParameter(name='bgcolor', value='black')
                             if MASK: op.addParameter(name='mask', value=MASK, format='float')
                             if args.server:
                                 op.addParameter(name='server', value='0.0.0.0:4444')
                                 op.addParameter(name='exp_code', value='400')
                         desc = {
                                 'Data': {
                                     'data_param': {PARAM[param]['wrname']: ['H', 'V']},
                                     'utctime': 'time'
                                 },
                                  'Metadata': {
                                     'heightList': 'range',
                                     'data_azi': 'azimuth',
                                     'data_ele': 'elevation',
                                     'mode_op': 'scan_type',
                                     'h0': 'range_correction',
                                 }
                             }
                         if args.save:
                             writer = merge.addOperation(name='HDFWriter')
                             writer.addParameter(name='path', value=path_save, format='str')
                             writer.addParameter(name='Reset', value=True)
                             writer.addParameter(name='setType', value='weather')
                             writer.addParameter(name='description', value=json.dumps(desc))
                             writer.addParameter(name='blocksPerFile', value='1',format='int')
                             writer.addParameter(name='metadataList', value='heightList,data_azi,data_ele,mode_op,latitude,longitude,altitude,heading,radar_name,institution,contact,h0,range_unit')
                             writer.addParameter(name='dataList', value='data_param,utctime')
                             writer.addParameter(name='weather_var', value=param)
                             writer.addParameter(name='mask', value=MASK, format='float')
                             # meta
                             writer.addParameter(name='latitude', value='-12.040436')
                             writer.addParameter(name='longitude', value='-75.295893')
                             writer.addParameter(name='altitude', value='3379.2147')
                             writer.addParameter(name='heading', value='0')
                             writer.addParameter(name='radar_name', value='SOPHy')
                             writer.addParameter(name='institution', value='IGP')
                             writer.addParameter(name='contact', value='dscipion@igp.gob.pe')
                             writer.addParameter(name='created_by', value='Signal Chain (https://pypi.org/project/schainpy/)')
                             writer.addParameter(name='range_unit', value='km')
                 project.start()
             if __name__ == '__main__':
                 parser = argparse.ArgumentParser(description='Script to process SOPHy data.')
                 parser.add_argument('experiment',
                                     help='Experiment name')
                 parser.add_argument('--parameters', nargs='*', default=['S'],
                                     help='Variables to process: P, Z, V')
                 parser.add_argument('--time_offset', default=0,
                                     help='Fix time offset')
                 parser.add_argument('--range', default=0, type=float,
                                     help='Max range to plot')
                 parser.add_argument('--save', action='store_true',
                                     help='Create output files')
                 parser.add_argument('--plot', action='store_true',
                                     help='Create plot files')
                 parser.add_argument('--show', action='store_true',
                                     help='Show matplotlib plot.')
                 parser.add_argument('--online', action='store_true',
                                     help='Set online mode.')
                 parser.add_argument('--server', action='store_true',
                                     help='Send to realtime')
                 parser.add_argument('--start_time', default='',
                                     help='Set start time.')
                 args = parser.parse_args()
                 main(args)
-            # python sophy_proc.py  HYO_PM@2022-06-09T15-05-12  --parameters V --plot --save --show --range 36
+            # python sophy_proc.py  HYO_PM@2022-06-09T15-05-12  --parameters V --plot --save --show --range 36S

schainpy/scripts/sophy_proc_rev004.py +46 -19

             # SOPHY PROC script
             import os, sys, json, argparse
             import datetime
             import time
             PATH = '/DATA_RM/DATA'
             # PATH = '/Users/jespinoza/workspace/data/'
             #PATH = '/home/roberto/DATA/data_WR_RHI/RHI'
             PATH = '/home/soporte/Downloads/data_WR_RHI'
             PATH = '/home/soporte/Documents/EVENTO/'
             PARAM = {
                 'S': {'zmin': -45, 'zmax': -15, 'colormap': 'jet', 'label': 'Power', 'wrname': 'power','cb_label': 'dBm', 'ch':0},
+                'SNR': {'zmin': -40, 'zmax': -20, 'colormap': 'jet', 'label': 'SNR', 'wrname': 'snr','cb_label': 'dB', 'ch':0},
                #'V': {'name': 'dataPP_DOP', 'zmin': -10, 'zmax': 10, 'colormap': 'sophy_v', 'label': 'Velocity', 'wrname': 'velocity', 'cb_label': 'm/s', 'ch':0},
                 'V': {'zmin': -10, 'zmax': 10, 'colormap': 'sophy_v', 'label': 'Velocity', 'wrname': 'velocity', 'cb_label': 'm/s', 'ch':0},
                 'R': {'zmin': 0,   'zmax': 1,  'colormap': 'jet',    'label': 'RhoHV', 'wrname':'rhoHV', 'cb_label': '*',  'ch':0},
                 'P': {'zmin': -180,'zmax': 180,'colormap': 'RdBu_r', 'label': 'PhiDP', 'wrname':'phiDP' , 'cb_label': 'º',  'ch':0},
-                'D': {'zmin': -20, 'zmax': 80, 'colormap': 'gist_ncar','label': 'ZDR','wrname':'differential_reflectivity' , 'cb_label': 'dBz','ch':0},
+                'D': {'zmin': -30, 'zmax': 80, 'colormap': 'gist_ncar','label': 'ZDR','wrname':'differential_reflectivity' , 'cb_label': 'dBz','ch':0},
                 'Z': {'zmin': -30, 'zmax': 80, 'colormap': 'sophy_r','label': 'Reflectivity',  'wrname':'reflectivity', 'cb_label': 'dBz','ch':1},
-                'W': {'zmin': 0, 'zmax': 12, 'colormap': 'sophy_w','label': 'Spectral Width', 'wrname':'spectral_width', 'cb_label': 'hz', 'ch':1}
+                'W': {'zmin': 0, 'zmax': 12, 'colormap': 'sophy_w','label': 'Spectral Width', 'wrname':'spectral_width', 'cb_label': 'm/s', 'ch':1}
                 }
             def max_index(r, sample_rate, ipp):
                 return int(sample_rate*ipp*1e6 * r / 60) + int(sample_rate*ipp*1e6 * 1.2 / 60)
             def main(args):
                 experiment = args.experiment
                 fp = open(os.path.join(PATH, experiment, 'experiment.conf'))
                 conf = json.loads(fp.read())
                 ipp_km = conf['usrp_tx']['ipp']
                 ipp = ipp_km * 2 /300000
                 sample_rate  = conf['usrp_rx']['sample_rate']
                 axis = ['0' if x=='elevation' else '1' for x in conf['pedestal']['axis']]      # AZIMUTH 1 ELEVACION 0
                 speed_axis = conf['pedestal']['speed']
                 steps = conf['pedestal']['table']
                 time_offset = args.time_offset
                 parameters = args.parameters
                 start_date = experiment.split('@')[1].split('T')[0].replace('-', '/')
                 end_date = start_date
                 if args.start_time:
                     start_time = args.start_time
                 else:
                     start_time = experiment.split('@')[1].split('T')[1].replace('-', ':')
                 #start_time = '16:15:00'
                 end_time = '23:59:59'
                 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, 'plotsC0_PL_R'+str(args.range)+'km_removeDC')
                 path_save = os.path.join(PATH, experiment, 'paramC0_PL_R'+str(args.range)+'km_removeDC')
                 #path_plots = os.path.join(PATH, experiment, 'plotsC0_PL_R'+str(args.range)+'km')
                 #path_save = os.path.join(PATH, experiment, 'paramC0_PL_R'+str(args.range)+'km')
                 RMIX = 1.62
+                H0=-1.68
+                MASK =0.3
                 from schainpy.controller import Project
                 project = Project()
                 project.setup(id='1', name='Sophy', description='sophy proc')
                 reader = project.addReadUnit(datatype='DigitalRFReader',
                     path=path,
                     startDate=start_date,
                     endDate=end_date,
                     startTime=start_time,
                     endTime=end_time,
                     delay=30,
                     online=args.online,
                     walk=1,
                     ippKm = ipp_km,
                     getByBlock = 1,
                     nProfileBlocks = N,
                 )
                 voltage = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId())
                 op = voltage.addOperation(name='ProfileSelector')
                 op.addParameter(name='profileRangeList', value='{},{}'.format(conf['usrp_tx']['repetitions_1'], conf['usrp_tx']['repetitions_1']+conf['usrp_tx']['repetitions_2']-1))
                 if conf['usrp_tx']['code_type_2']:
                     codes = [ c.strip() for c in conf['usrp_tx']['code_2'].split(',')]
                     code = []
                     for c in codes:
                         code.append([int(x) for x in c])
                     op = voltage.addOperation(name='Decoder', optype='other')
                     op.addParameter(name='code', value=code)
                     op.addParameter(name='nCode', value=len(code), format='int')
                     op.addParameter(name='nBaud', value=len(code[0]), format='int')
                     op = voltage.addOperation(name='CohInt', optype='other') #Minimo integrar 2 perfiles por ser codigo complementario
                     op.addParameter(name='n', value=len(code), format='int')
                     ncode = len(code)
+                    print("codes",codes)
+                    print("nCode",len(code))
                 else:
                     ncode = 1
-                op = voltage.addOperation(name='setH0')
-                op.addParameter(name='h0', value='-1.68')
                 if args.range > 0:
                     op = voltage.addOperation(name='selectHeights')
                     op.addParameter(name='minIndex', value='0', format='int')
                     op.addParameter(name='maxIndex', value=max_index(args.range, sample_rate, ipp), format='int')
+                op = voltage.addOperation(name='setH0')
+                op.addParameter(name='h0', value='-1.68' , format='float')
                 op = voltage.addOperation(name='PulsePair_vRF', optype='other')
                 op.addParameter(name='n', value=int(conf['usrp_tx']['repetitions_2'])/2, format='int')
-                #op.addParameter(name='removeDC', value=1, format='int')
+                op.addParameter(name='removeDC', value=1, format='int')
                 proc = project.addProcUnit(datatype='ParametersProc', inputId=voltage.getId())
+                proc.addParameter(name='runNextUnit', value=True)
                 opObj10 = proc.addOperation(name="WeatherRadar")
-                opObj10.addParameter(name='variableList',value='Reflectividad,VelocidadRadial,AnchoEspectral')
+                #opObj10.addParameter(name='variableList',value='Reflectividad,VelocidadRadial,AnchoEspectral')
                 opObj10.addParameter(name='tauW',value=(1e-6/sample_rate)*len(code[0]))
                 opObj10.addParameter(name='Pt',value=((1e-6/sample_rate)*len(code[0])/ipp)*200)
                 # {"latitude": -12.0404828587, "longitude": -75.2147483647, "altitude": 3379.2147483647}
                 op = proc.addOperation(name='PedestalInformation')
                 op.addParameter(name='path', value=path_ped, format='str')
                 op.addParameter(name='interval', value='0.04')
                 op.addParameter(name='time_offset', value=time_offset)
                 #op.addParameter(name='az_offset', value=-26.2)
                 op.addParameter(name='mode', value='PPI')
                 for param in parameters:
                     op = proc.addOperation(name='Block360')
                     op.addParameter(name='attr_data', value='data_param')
                     op.addParameter(name='runNextOp', value=True)
                     op= proc.addOperation(name='WeatherParamsPlot')
                     if args.save: op.addParameter(name='save', value=path_plots, format='str')
                     op.addParameter(name='save_period', value=-1)
                     op.addParameter(name='show', value=args.show)
                     op.addParameter(name='channels', value='0,')
                     op.addParameter(name='zmin', value=PARAM[param]['zmin'])
                     op.addParameter(name='zmax', value=PARAM[param]['zmax'])
                     op.addParameter(name='attr_data', value=param, format='str')
                     op.addParameter(name='labels', value=[PARAM[param]['label']])
                     op.addParameter(name='save_code', value=param)
                     op.addParameter(name='cb_label', value=PARAM[param]['cb_label'])
                     op.addParameter(name='colormap', value=PARAM[param]['colormap'])
-                    op.addParameter(name='bgcolor',value='black')
+                    op.addParameter(name='bgcolor', value='black')
-                    op.addParameter(name='snr_threshold',value=0.4)
+                    if MASK: op.addParameter(name='mask', value=MASK, format='float')
+                    if args.server:
+                        op.addParameter(name='server', value='0.0.0.0:4444')
+                        op.addParameter(name='exp_code', value='400')
                     desc = {
                             'Data': {
-                                'data_param': PARAM[param]['wrname'],
+                                'data_param': {PARAM[param]['wrname']: ['H', 'V']},
                                 'utctime': 'time'
                             },
                              'Metadata': {
                                 'heightList': 'range',
                                 'data_azi': 'azimuth',
                                 'data_ele': 'elevation',
+                                'mode_op': 'scan_type',
+                                'h0': 'range_correction',
                             }
                         }
                     if args.save:
-                        opObj10 = proc.addOperation(name='HDFWriter')
+                        writer = proc.addOperation(name='HDFWriter')
-                        opObj10.addParameter(name='path', value=path_save, format='str')
+                        writer.addParameter(name='path', value=path_save, format='str')
-                        opObj10.addParameter(name='Reset', value=True)
+                        writer.addParameter(name='Reset', value=True)
-                        opObj10.addParameter(name='setType', value='weather')
+                        writer.addParameter(name='setType', value='weather')
-                        opObj10.addParameter(name='    description', value=json.dumps(desc))
+                        writer.addParameter(name='description', value=json.dumps(desc))
-                        opObj10.addParameter(name='blocksPerFile', value='1',format='int')
+                        writer.addParameter(name='blocksPerFile', value='1',format='int')
-                        opObj10.addParameter(name='metadataList', value='heightList,data_azi,data_ele')
+                        writer.addParameter(name='metadataList', value='heightList,data_azi,data_ele,mode_op,latitude,longitude,altitude,heading,radar_name,institution,contact,h0,range_unit')
-                        opObj10.addParameter(name='dataList', value='{},utctime'.format(PARAM[param]['name']))
+                        writer.addParameter(name='dataList', value='data_param,utctime')
+                        writer.addParameter(name='weather_var', value=param)
+                        writer.addParameter(name='mask', value=MASK, format='float')
+                        # meta
+                        writer.addParameter(name='latitude', value='-12.040436')
+                        writer.addParameter(name='longitude', value='-75.295893')
+                        writer.addParameter(name='altitude', value='3379.2147')
+                        writer.addParameter(name='heading', value='0')
+                        writer.addParameter(name='radar_name', value='SOPHy')
+                        writer.addParameter(name='institution', value='IGP')
+                        writer.addParameter(name='contact', value='dscipion@igp.gob.pe')
+                        writer.addParameter(name='created_by', value='Signal Chain (https://pypi.org/project/schainpy/)')
+                        writer.addParameter(name='range_unit', value='km')
                 project.start()
             if __name__ == '__main__':
                 parser = argparse.ArgumentParser(description='Script to process SOPHy data.')
                 parser.add_argument('experiment',
                                     help='Experiment name')
                 parser.add_argument('--parameters', nargs='*', default=['S'],
                                     help='Variables to process: P, Z, V')
                 parser.add_argument('--time_offset', default=0,
                                     help='Fix time offset')
                 parser.add_argument('--range', default=0, type=float,
                                     help='Max range to plot')
                 parser.add_argument('--save', action='store_true',
                                     help='Create output files')
                 parser.add_argument('--show', action='store_true',
                                     help='Show matplotlib plot.')
                 parser.add_argument('--online', action='store_true',
                                     help='Set online mode.')
+                parser.add_argument('--server', action='store_true',
+                                    help='Send to realtime')
                 parser.add_argument('--start_time', default='',
                                     help='Set start time.')
                 args = parser.parse_args()
                 main(args)

schainpy/scripts/sophy_proc_rev005.py +36 -18

             # SOPHY PROC script
             import os, sys, json, argparse
             import datetime
             import time
             PATH = '/DATA_RM/DATA'
             # PATH = '/Users/jespinoza/workspace/data/'
             #PATH = '/home/roberto/DATA/data_WR_RHI/RHI'
             PATH = '/home/soporte/Downloads/data_WR_RHI'
             PATH = '/home/soporte/Documents/EVENTO/'
             PARAM = {
                 'S': {'name': 'dataPP_POWER', 'zmin': -45, 'zmax': -15, 'colormap': 'jet', 'label': 'Power', 'wrname': 'power','cb_label': 'dBm', 'ch':0},
                 'S2': {'name': 'data_pow', 'zmin': -45, 'zmax': -15, 'colormap': 'jet', 'label': 'Power', 'wrname': 'power','cb_label': 'dBm', 'ch':0},
                 #'V': {'name': 'dataPP_DOP', 'zmin': -10, 'zmax': 10, 'colormap': 'sophy_v', 'label': 'Velocity', 'wrname': 'velocity', 'cb_label': 'm/s', 'ch':0},
                 'V': {'name': 'velRadial_V', 'zmin': -10, 'zmax': 10, 'colormap': 'sophy_v', 'label': 'Velocity', 'wrname': 'velocity', 'cb_label': 'm/s', 'ch':0},
                 'R': {'name': 'RhoHV_R', 'zmin': 0,   'zmax': 1,  'colormap': 'jet',    'label': 'RhoHV', 'wrname':'rhoHV', 'cb_label': '*',  'ch':0},
                 'P': {'name': 'PhiD_P', 'zmin': -180,'zmax': 180,'colormap': 'RdBu_r', 'label': 'PhiDP', 'wrname':'phiDP' , 'cb_label': 'º',  'ch':0},
                 'D': {'name': 'Zdb_D', 'zmin': -20, 'zmax': 80, 'colormap': 'gist_ncar','label': 'ZDR','wrname':'differential_reflectivity' , 'cb_label': 'dBz','ch':0},
                 'Z':  {'name': 'Zdb', 'zmin': -30, 'zmax': 80, 'colormap': 'sophy_r','label': 'Reflectivity',  'wrname':'reflectivity', 'cb_label': 'dBz','ch':1},
-                'W':  {'name': 'Sigmav_W', 'zmin': 0, 'zmax': 12, 'colormap': 'sophy_w','label': 'Spectral Width', 'wrname':'spectral_width', 'cb_label': 'hz', 'ch':1}
+                'W':  {'name': 'Sigmav_W', 'zmin': 0, 'zmax': 12, 'colormap': 'sophy_w','label': 'Spectral Width', 'wrname':'spectral_width', 'cb_label': 'm/s', 'ch':1}
                 }
             def max_index(r, sample_rate, ipp):
                 return int(sample_rate*ipp*1e6 * r / 60) + int(sample_rate*ipp*1e6 * 1.2 / 60)
             def main(args):
                 experiment = args.experiment
                 fp = open(os.path.join(PATH, experiment, 'experiment.conf'))
                 conf = json.loads(fp.read())
                 ipp_km = conf['usrp_tx']['ipp']
                 ipp = ipp_km * 2 /300000
                 sample_rate  = conf['usrp_rx']['sample_rate']
                 axis = ['0' if x=='elevation' else '1' for x in conf['pedestal']['axis']]      # AZIMUTH 1 ELEVACION 0
                 speed_axis = conf['pedestal']['speed']
                 steps = conf['pedestal']['table']
                 time_offset = args.time_offset
                 parameters = args.parameters
                 start_date = experiment.split('@')[1].split('T')[0].replace('-', '/')
                 end_date = start_date
                 if args.start_time:
                     start_time = args.start_time
                 else:
                     start_time = experiment.split('@')[1].split('T')[1].replace('-', ':')
                 #start_time = '16:15:00'
                 end_time = '23:59:59'
                 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, 'plotsC0_FD_PL_R'+str(args.range)+'km_removeDC')
+                path_plots = os.path.join(PATH, experiment, 'plotsC0_FD_PL_R'+str(args.range)+'km_removeDC')
-                #path_save = os.path.join(PATH, experiment, 'paramC0_FD_PL_R'+str(args.range)+'km_removeDC')
+                path_save = os.path.join(PATH, experiment, 'paramC0_FD_PL_R'+str(args.range)+'km_removeDC')
-                path_plots = os.path.join(PATH, experiment, 'plotsC0_FD_PL_R'+str(args.range)+'km')
+                #path_plots = os.path.join(PATH, experiment, 'plotsC0_FD_PL_R'+str(args.range)+'km')
-                path_save = os.path.join(PATH, experiment, 'paramC0_FD_PL_R'+str(args.range)+'km')
+                #path_save = os.path.join(PATH, experiment, 'paramC0_FD_PL_R'+str(args.range)+'km')
                 RMIX = 1.62
+                MASK =0.3
                 from schainpy.controller import Project
                 project = Project()
                 project.setup(id='1', name='Sophy', description='sophy proc')
                 reader = project.addReadUnit(datatype='DigitalRFReader',
                     path=path,
                     startDate=start_date,
                     endDate=end_date,
                     startTime=start_time,
                     endTime=end_time,
                     delay=30,
                     online=args.online,
                     walk=1,
                     ippKm = ipp_km,
                     getByBlock = 1,
                     nProfileBlocks = N,
                 )
                 voltage = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId())
                 op =  voltage.addOperation(name='setAttribute')
                 op.addParameter(name='frequency', value='9345.e6', format='float')
                 op = voltage.addOperation(name='ProfileSelector')
                 op.addParameter(name='profileRangeList', value='{},{}'.format(conf['usrp_tx']['repetitions_1'], conf['usrp_tx']['repetitions_1']+conf['usrp_tx']['repetitions_2']-1))
                 if conf['usrp_tx']['code_type_2']:
                     codes = [ c.strip() for c in conf['usrp_tx']['code_2'].split(',')]
                     code = []
                     for c in codes:
                         code.append([int(x) for x in c])
                     op = voltage.addOperation(name='Decoder', optype='other')
                     op.addParameter(name='code', value=code)
                     op.addParameter(name='nCode', value=len(code), format='int')
                     op.addParameter(name='nBaud', value=len(code[0]), format='int')
                     op = voltage.addOperation(name='CohInt', optype='other') #Minimo integrar 2 perfiles por ser codigo complementario
                     op.addParameter(name='n', value=len(code), format='int')
                     ncode = len(code)
                 else:
                     ncode = 1
                 op = voltage.addOperation(name='setH0')
                 op.addParameter(name='h0', value='-1.68')
                 if args.range > 0:
                     op = voltage.addOperation(name='selectHeights')
                     op.addParameter(name='minIndex', value='0', format='int')
                     op.addParameter(name='maxIndex', value=max_index(args.range, sample_rate, ipp), format='int')
                 #---------------------------------------NEW PROCESSING -----------------------------------------------------
                 procB = project.addProcUnit(datatype='SpectraProc', inputId=voltage.getId())
                 procB.addParameter(name='nFFTPoints', value=int(conf['usrp_tx']['repetitions_2'])/2, format='int')
                 procB.addParameter(name='nProfiles', value=int(conf['usrp_tx']['repetitions_2'])/2, format='int')
-                #opObj11 = procB.addOperation(name='removeDC')
+                opObj11 = procB.addOperation(name='removeDC')
-                #opObj11.addParameter(name='mode', value=2)
+                opObj11.addParameter(name='mode', value=2)
                 proc= project.addProcUnit(datatype='ParametersProc',inputId=procB.getId())
                 opObj10 =  proc.addOperation(name='SpectralMoments')
                 opObj10.addParameter(name='wradar',value=True)
                 #---------------------------------------NEW PROCESSING -----------------------------------------------------
                 opObj10 = proc.addOperation(name="WeatherRadar")
-                opObj10.addParameter(name='variableList',value='Reflectividad,VelocidadRadial,AnchoEspectral')
+                #opObj10.addParameter(name='variableList',value='Reflectividad,VelocidadRadial,AnchoEspectral')
                 opObj10.addParameter(name='tauW',value=(1e-6/sample_rate)*len(code[0]))
                 opObj10.addParameter(name='Pt',value=((1e-6/sample_rate)*len(code[0])/ipp)*200)
                 # {"latitude": -12.0404828587, "longitude": -75.2147483647, "altitude": 3379.2147483647}
                 op = proc.addOperation(name='PedestalInformation')
                 op.addParameter(name='path', value=path_ped, format='str')
                 op.addParameter(name='interval', value='0.04')
                 op.addParameter(name='time_offset', value=time_offset)
                 #op.addParameter(name='az_offset', value=-26.2)
                 op.addParameter(name='mode', value='PPI')
                 for param in parameters:
                     op = proc.addOperation(name='Block360')
                     op.addParameter(name='attr_data', value='data_param')
                     op.addParameter(name='runNextOp', value=True)
                     op= proc.addOperation(name='WeatherParamsPlot')
                     if args.save: op.addParameter(name='save', value=path_plots, format='str')
                     op.addParameter(name='save_period', value=-1)
                     op.addParameter(name='show', value=args.show)
                     op.addParameter(name='channels', value='0,')
                     op.addParameter(name='zmin', value=PARAM[param]['zmin'])
                     op.addParameter(name='zmax', value=PARAM[param]['zmax'])
                     op.addParameter(name='attr_data', value=param, format='str')
                     op.addParameter(name='labels', value=[PARAM[param]['label']])
                     op.addParameter(name='save_code', value=param)
                     op.addParameter(name='cb_label', value=PARAM[param]['cb_label'])
                     op.addParameter(name='colormap', value=PARAM[param]['colormap'])
                     op.addParameter(name='bgcolor',value='black')
                     op.addParameter(name='snr_threshold',value=0.4)
+                    if MASK: op.addParameter(name='mask', value=MASK, format='float')
                     desc = {
                             'Data': {
-                                'data_param': PARAM[param]['wrname'],
+                                'data_param': {PARAM[param]['wrname']: ['H', 'V']},
                                 'utctime': 'time'
                             },
                              'Metadata': {
                                 'heightList': 'range',
                                 'data_azi': 'azimuth',
                                 'data_ele': 'elevation',
+                                'mode_op': 'scan_type',
+                                'h0': 'range_correction',
                             }
                         }
                     if args.save:
-                        opObj10 = proc.addOperation(name='HDFWriter')
+                        writer = proc.addOperation(name='HDFWriter')
-                        opObj10.addParameter(name='path', value=path_save, format='str')
+                        writer.addParameter(name='path', value=path_save, format='str')
-                        opObj10.addParameter(name='Reset', value=True)
+                        writer.addParameter(name='Reset', value=True)
-                        opObj10.addParameter(name='setType', value='weather')
+                        writer.addParameter(name='setType', value='weather')
-                        opObj10.addParameter(name='    description', value=json.dumps(desc))
+                        writer.addParameter(name='description', value=json.dumps(desc))
-                        opObj10.addParameter(name='blocksPerFile', value='1',format='int')
+                        writer.addParameter(name='blocksPerFile', value='1',format='int')
-                        opObj10.addParameter(name='metadataList', value='heightList,data_azi,data_ele')
+                        writer.addParameter(name='metadataList', value='heightList,data_azi,data_ele,mode_op,latitude,longitude,altitude,heading,radar_name,institution,contact,h0,range_unit')
-                        opObj10.addParameter(name='dataList', value='{},utctime'.format(PARAM[param]['name']))
+                        writer.addParameter(name='dataList', value='data_param,utctime')
+                        writer.addParameter(name='weather_var', value=param)
+                        writer.addParameter(name='mask', value=MASK, format='float')
+                        # meta
+                        writer.addParameter(name='latitude', value='-12.040436')
+                        writer.addParameter(name='longitude', value='-75.295893')
+                        writer.addParameter(name='altitude', value='3379.2147')
+                        writer.addParameter(name='heading', value='0')
+                        writer.addParameter(name='radar_name', value='SOPHy')
+                        writer.addParameter(name='institution', value='IGP')
+                        writer.addParameter(name='contact', value='dscipion@igp.gob.pe')
+                        writer.addParameter(name='created_by', value='Signal Chain (https://pypi.org/project/schainpy/)')
+                        writer.addParameter(name='range_unit', value='km')
                 project.start()
             if __name__ == '__main__':
                 parser = argparse.ArgumentParser(description='Script to process SOPHy data.')
                 parser.add_argument('experiment',
                                     help='Experiment name')
                 parser.add_argument('--parameters', nargs='*', default=['S'],
                                     help='Variables to process: P, Z, V')
                 parser.add_argument('--time_offset', default=0,
                                     help='Fix time offset')
                 parser.add_argument('--range', default=0, type=float,
                                     help='Max range to plot')
                 parser.add_argument('--save', action='store_true',
                                     help='Create output files')
                 parser.add_argument('--show', action='store_true',
                                     help='Show matplotlib plot.')
                 parser.add_argument('--online', action='store_true',
                                     help='Set online mode.')
                 parser.add_argument('--start_time', default='',
                                     help='Set start time.')
                 args = parser.parse_args()
-                main(args)
+                main(args) # Operator#ñ8

schainpy/scripts/sophy_proc_rev006.py +1 -1

             #!/usr/bin/env python
             # Ing. AVP
             # 22/06/2022
             # ARCHIVO DE LECTURA y PLOT
             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
             #example
             #export WRADLIB_DATA="/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/paramC0N36.0/2022-06-09T18-00-00/"
             warnings.filterwarnings('ignore')
             PARAM = {
                 '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'},
-                'W': {'var': 'spectral_width',    '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': 'm/s'}
                 }
             class Readsophy():
                 def __init__(self):
                     self.list_file = None
                     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 = []
                     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
                 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 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
                 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)
                         # LECTURA
                         data_arr, utc_time, data_azi,data_ele, heightList,unit,cmap,vmin,vmax,label = self.readAttributes(obj= test_hdf5,variable=self.variable)
                         if self.range==0:
                             self.range == heightList[-1]
                         new_heightList,minIndex,maxIndex = self.selectHeights(heightList,0.06,self.range)
                         # TIEMPO
                         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[:,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+"  E."+self.grado
                             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:
                             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+"  E."+self.grado
                             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)
                                 try:
                                     os.mkdir(filesavepath)
                                 except:
                                     pass
                             else:
                                 filename     = "SOPHY"+"_"+time_save+"_"+"E."+self.grado+"_"+self.variable+".png"
                             pl.savefig(filesavepath+filename)
                         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/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,range=range, variable=param,save=int(save))
                     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')
                 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
                 """
                 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
             LAST_UPDATE
             ---- Noise
             ---- Mapas
             '''

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