##// 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
avaldezp - - Load All Authors

File last commit:

r1464:3250059d6d95
r1759:c8ee9da660f1
Show More
sophy_proc_test001.py
195 lines | 8.8 KiB | text/x-python | PythonLexer
/ schainpy / scripts / sophy_proc_test001.py
History | Source | Raw |Copy content |Copy permalink
avaldezp
fix en escritura y prueba de script con ploteo general
 r1463 # 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 = {
avaldezp
fix save data and test multiple plot
 r1464 '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}
avaldezp
fix en escritura y prueba de script con ploteo general
 r1463 }
avaldezp
fix save data and test multiple plot
 r1464
#---------------------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'
}
}
'''
avaldezp
fix en escritura y prueba de script con ploteo general
 r1463 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:
avaldezp
fix save data and test multiple plot
 r1464 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'
}
}
avaldezp
fix en escritura y prueba de script con ploteo general
 r1463 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')
avaldezp
fix save data and test multiple plot
 r1464 #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')
avaldezp
fix en escritura y prueba de script con ploteo general
 r1463 opObj10.addParameter(name='dataList',value=','.join([PARAM[param]['name'],'utctime']))
avaldezp
fix save data and test multiple plot
 r1464 opObj10.addParameter(name='description',value=json.dumps(desc_wr))
avaldezp
fix en escritura y prueba de script con ploteo general
 r1463
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)