# 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' PARAM = { 'P': {'name': 'dataPP_POWER','zmin': 35, 'zmax': 60, 'colormap': 'jet', 'label': 'Power', 'wrname':'Pow', 'cb_label': 'dB', 'ch':1}, 'V': {'name': 'dataPP_DOP', 'zmin': -20, 'zmax': 20, 'colormap': 'seismic','label': 'Velocity', 'wrname':'Dop', 'cb_label': 'm/s','ch':1}, 'RH': {'name': 'RhoHV_R', 'zmin': 0, 'zmax': 1, 'colormap': 'jet', 'label': 'Coef.Correlacion', 'wrname':'R', 'cb_label': '*', 'ch':0}, 'FD': {'name': 'PhiD_P', 'zmin': -180,'zmax': 180,'colormap': 'RdBu_r', 'label': 'Fase Diferencial', 'wrname':'P' , 'cb_label': 'ยบ', 'ch':0}, 'ZD': {'name': 'Zdb_D', 'zmin': -20, 'zmax': 80, 'colormap': 'viridis','label': 'Reflect.Diferencial','wrname':'D' , 'cb_label': 'dBz','ch':0}, 'Z': {'name': 'Zdb', 'zmin': -20, 'zmax': 60, 'colormap': 'viridis','label': 'Reflectividad', 'wrname':'Z', 'cb_label': 'dBz','ch':1}, 'W': {'name': 'Sigmav_W', 'zmin': -20, 'zmax': 60, 'colormap': 'viridis','label': 'AnchoEspectral', 'wrname':'S', 'cb_label': 'hz', 'ch':1} } #---------------------SIGNAL CHAIN ------------------------------------ # Definido por el usuario puede ser modificado solo se necesita definir. Ejemplo ''' desc_wr= { 'Data': { 'dataPP_POW': 'Power', 'utctime': 'Time', 'azimuth': 'az', 'elevation':'el' }, 'Metadata': { 'heightList': 'range', 'channelList': 'Channels' } } ''' 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 samp_rate = conf['usrp_rx']['sample_rate'] axis = ['0' if x=='elevation' else '1' for x in conf['pedestal']['speed']] # AZIMUTH 1 ELEVACION 0 speed_axis = conf['pedestal']['speed'] steeps = conf['pedestal']['table'] time_offset = args.time_offset parameters = args.parameters #start_date = experiment.split('@')[1].split('T')[0].replace('-', '/') start_date = '2022/04/22' end_date = start_date #start_time = experiment.split('@')[1].split('T')[1] start_time ='17:42:55' end_time = '23:59:59' max_index = int(samp_rate*ipp*1e6 * args.range / 60) + int(samp_rate*ipp*1e6 * 1.2 / 60) 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_save = os.path.join(PATH, experiment, 'param') dBmin = 35 dBmax = 60 Vmin = -20 Vmax = 20 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=0, online=0, walk=1, ippKm = ipp_km, getByBlock = 1, nProfileBlocks = N, ) voltage = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId()) op = voltage.addOperation(name='setH0') op.addParameter(name='h0', value='-1.2') if args.range > 0: op = voltage.addOperation(name='selectHeights') op.addParameter(name='minIndex', value='0', format='int') op.addParameter(name='maxIndex', value=max_index, format='int') op = voltage.addOperation(name='PulsePair_vRF', optype='other') op.addParameter(name='n', value=int(N), format='int') proc = project.addProcUnit(datatype='ParametersProc', inputId=voltage.getId()) #-----------------------new--------- variables polarimetricas--------------- opObj10 = proc.addOperation(name="WeatherRadar") opObj10.addParameter(name='variableList',value='Reflectividad,ReflectividadDiferencial,CoeficienteCorrelacion,FaseDiferencial,VelocidadRadial,AnchoEspectral') op = proc.addOperation(name='PedestalInformation') op.addParameter(name='path', value=path_ped, format='str') op.addParameter(name='interval', value='0.04', format='float') op.addParameter(name='time_offset', value=time_offset) #op.addParameter(name='axis', value=','.join(axis)) #Preguntar en pedestal si todos los elementos #de aqui son iguales, si lo fueran, que considere #el primero dato como el modo (PPI o RHI) y que ya #no pregunte por el modo porque este no cambia for param in parameters: op = proc.addOperation(name='Block360_vRF4') #op.addParameter(name='axis', value=','.join(axis)) op.addParameter(name='runNextOp', value=True) op.addParameter(name='attr_data', value=PARAM[param]['name']) path_fig = '/AUTO{}km'.format(args.range) op = proc.addOperation(name='WeatherParamsPlot') op.addParameter(name='save', value=path_plots+path_fig, 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[param]['name'], 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']) if args.save: desc_wr= { 'Data': { PARAM[param]['name']: PARAM[param]['wrname'],#PARAM[param]['name']: {PARAM[param]['wrname']:['P0','P1']}, 'utctime': 'Time' }, 'Metadata': { 'heightList': 'range', 'channelList': 'Channels', 'data_azi': 'azimuth', 'data_ele': 'elevation' } } opObj10 = proc.addOperation(name='HDFWriter') opObj10.addParameter(name='path',value=path_save, format='str') opObj10.addParameter(name='Reset',value=True) opObj10.addParameter(name='setType',value='weather') opObj10.addParameter(name='blocksPerFile',value='1',format='int') #opObj10.addParameter(name='channel',value=PARAM[param]['ch'],format='int') opObj10.addParameter(name='metadataList',value='heightList,channelList,Typename,Datatype,Scantype,Latitude,Longitud,Heading,Waveform,PRF,CreatedBy,ContactInformation,data_azi,data_ele') opObj10.addParameter(name='Typename', value=PARAM[param]['label']) opObj10.addParameter(name='Datatype', value='RadialSet') opObj10.addParameter(name='Scantype', value='PPI') opObj10.addParameter(name='Latitude', value='-11.96') opObj10.addParameter(name='Longitud', value='-76.54') opObj10.addParameter(name='Heading', value='293') opObj10.addParameter(name='Height', value='293') opObj10.addParameter(name='Waveform', value='OFM') opObj10.addParameter(name='PRF', value='2500') opObj10.addParameter(name='CreatedBy', value='WeatherRadarJROTeam') opObj10.addParameter(name='ContactInformation', value='dscipion@igp.gob.pe') opObj10.addParameter(name='dataList',value=','.join([PARAM[param]['name'],'utctime'])) opObj10.addParameter(name='description',value=json.dumps(desc_wr)) 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=['P'], 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=int, 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.') args = parser.parse_args() print (args) main(args)