##// END OF EJS Templates
add heading to script
jespinoza -
r1719:500e3b37e54e
parent child
Show More
@@ -1,489 +1,491
1 # SOPHY PROC script
1 # SOPHY PROC script
2 import os, sys, json, argparse
2 import os, sys, json, argparse
3 import multiprocessing
3 import multiprocessing
4 import datetime
4 import datetime
5 import time
5 import time
6
6
7 PATH = '/data'
7 PATH = '/data'
8 #PATH = "/media/soporte/TOSHIBAEXT/sophy/"
8 #PATH = "/media/soporte/TOSHIBAEXT/sophy/"
9
9
10
10
11
11
12 # SNR ZMIN -40 A ZMAX -20
12 # SNR ZMIN -40 A ZMAX -20
13 PARAM = {
13 PARAM = {
14 'S': {'zmin': -45, 'zmax':-25, 'colormap': 'jet' , 'label': 'Power', 'wrname': 'power','cb_label': 'dBm', 'ch':0},
14 'S': {'zmin': -45, 'zmax':-25, 'colormap': 'jet' , 'label': 'Power', 'wrname': 'power','cb_label': 'dBm', 'ch':0},
15 'SNR':{'zmin': -10, 'zmax': 15, 'colormap': 'jet' , 'label': 'SNR', 'wrname': 'snr','cb_label': 'dB', 'ch':0},
15 'SNR':{'zmin': -10, 'zmax': 15, 'colormap': 'jet' , 'label': 'SNR', 'wrname': 'snr','cb_label': 'dB', 'ch':0},
16 'V': {'zmin': -12, 'zmax': 12, 'colormap': 'sophy_v', 'label': 'Velocity', 'wrname': 'velocity', 'cb_label': 'm/s', 'ch':0},
16 'V': {'zmin': -12, 'zmax': 12, 'colormap': 'sophy_v', 'label': 'Velocity', 'wrname': 'velocity', 'cb_label': 'm/s', 'ch':0},
17 'R': {'zmin': 0.5, 'zmax': 1 , 'colormap': 'sophy_r', 'label': 'RhoHV', 'wrname':'rhoHV', 'cb_label': '', 'ch':0},
17 'R': {'zmin': 0.5, 'zmax': 1 , 'colormap': 'sophy_r', 'label': 'RhoHV', 'wrname':'rhoHV', 'cb_label': '', 'ch':0},
18 'P': {'zmin': -180,'zmax': 180,'colormap': 'sophy_p', 'label': 'PhiDP', 'wrname':'phiDP' , 'cb_label': 'degrees', 'ch':0},
18 'P': {'zmin': -180,'zmax': 180,'colormap': 'sophy_p', 'label': 'PhiDP', 'wrname':'phiDP' , 'cb_label': 'degrees', 'ch':0},
19 'D': {'zmin': -9 , 'zmax': 12, 'colormap': 'sophy_d', 'label': 'ZDR','wrname':'differential_reflectivity' , 'cb_label': 'dB','ch':0},
19 'D': {'zmin': -9 , 'zmax': 12, 'colormap': 'sophy_d', 'label': 'ZDR','wrname':'differential_reflectivity' , 'cb_label': 'dB','ch':0},
20 'Z': {'zmin': -20, 'zmax': 80, 'colormap': 'sophy_z', 'label': 'Reflectivity ', 'wrname':'reflectivity', 'cb_label': 'dBz','ch':0},
20 'Z': {'zmin': -20, 'zmax': 80, 'colormap': 'sophy_z', 'label': 'Reflectivity ', 'wrname':'reflectivity', 'cb_label': 'dBz','ch':0},
21 'W': {'zmin': 0 , 'zmax': 12, 'colormap': 'sophy_w', 'label': 'Spectral Width', 'wrname':'spectral_width', 'cb_label': 'm/s', 'ch':0}
21 'W': {'zmin': 0 , 'zmax': 12, 'colormap': 'sophy_w', 'label': 'Spectral Width', 'wrname':'spectral_width', 'cb_label': 'm/s', 'ch':0}
22 }
22 }
23
23
24 META = ['heightList', 'data_azi', 'data_ele', 'mode_op', 'latitude', 'longitude', 'altitude', 'heading', 'radar_name',
24 META = ['heightList', 'data_azi', 'data_ele', 'mode_op', 'latitude', 'longitude', 'altitude', 'heading', 'radar_name',
25 'institution', 'contact', 'h0', 'range_unit', 'prf', 'prf_unit', 'variable', 'variable_unit', 'n_pulses',
25 'institution', 'contact', 'h0', 'range_unit', 'prf', 'prf_unit', 'variable', 'variable_unit', 'n_pulses',
26 'pulse1_range', 'pulse1_width', 'pulse2_width', 'pulse1_repetitions', 'pulse2_repetitions', 'pulse_width_unit',
26 'pulse1_range', 'pulse1_width', 'pulse2_width', 'pulse1_repetitions', 'pulse2_repetitions', 'pulse_width_unit',
27 'snr_threshold', 'data_noise']
27 'snr_threshold', 'data_noise']
28
28
29
29
30 def max_index(r, sample_rate, ipp, h0):
30 def max_index(r, sample_rate, ipp, h0):
31
31
32 return int(sample_rate*ipp*1e6 * r / 60) + int(sample_rate*ipp*1e6 * -h0 / 60)
32 return int(sample_rate*ipp*1e6 * r / 60) + int(sample_rate*ipp*1e6 * -h0 / 60)
33
33
34 def main(args):
34 def main(args):
35
35
36 experiment = args.experiment
36 experiment = args.experiment
37 fp = open(os.path.join(PATH, experiment, 'experiment.json'))
37 fp = open(os.path.join(PATH, experiment, 'experiment.json'))
38 conf = json.loads(fp.read())
38 conf = json.loads(fp.read())
39
39
40 ipp_km = conf['usrp_tx']['ipp']
40 ipp_km = conf['usrp_tx']['ipp']
41 ipp = ipp_km * 2 /300000
41 ipp = ipp_km * 2 /300000
42 sample_rate = conf['usrp_rx']['sample_rate']
42 sample_rate = conf['usrp_rx']['sample_rate']
43 speed_axis = conf['pedestal']['speed']
43 speed_axis = conf['pedestal']['speed']
44 if args.angles:
44 if args.angles:
45 angles = args.angles
45 angles = args.angles
46 else:
46 else:
47 angles = conf['pedestal']['table']
47 angles = conf['pedestal']['table']
48 time_offset = args.time_offset
48 time_offset = args.time_offset
49 parameters = args.parameters
49 parameters = args.parameters
50 start_date = conf['name'].split('@')[1].split('T')[0].replace('-', '/')
50 start_date = conf['name'].split('@')[1].split('T')[0].replace('-', '/')
51 end_date = start_date
51 end_date = start_date
52 if args.start_time:
52 if args.start_time:
53 start_time = args.start_time
53 start_time = args.start_time
54 else:
54 else:
55 start_time = conf['name'].split('@')[1].split('T')[1].replace('-', ':')
55 start_time = conf['name'].split('@')[1].split('T')[1].replace('-', ':')
56
56
57 if args.end_time:
57 if args.end_time:
58 end_time = args.end_time
58 end_time = args.end_time
59 else:
59 else:
60 end_time = '23:59:59'
60 end_time = '23:59:59'
61
61
62 N = int(1.0/(abs(speed_axis[0])*ipp)) # 1 GRADO DE RESOLUCION
62 N = int(1.0/(abs(speed_axis[0])*ipp)) # 1 GRADO DE RESOLUCION
63
63
64 path = os.path.join(PATH, experiment, 'rawdata')
64 path = os.path.join(PATH, experiment, 'rawdata')
65 path_ped = os.path.join(PATH, experiment, 'position')
65 path_ped = os.path.join(PATH, experiment, 'position')
66 if args.label:
66 if args.label:
67 label = '-{}'.format(args.label)
67 label = '-{}'.format(args.label)
68 else:
68 else:
69 label = ''
69 label = ''
70 path_plots = os.path.join(PATH, experiment, 'plots{}'.format(label))
70 path_plots = os.path.join(PATH, experiment, 'plots{}'.format(label))
71 path_save = os.path.join(PATH, experiment, 'param{}'.format(label))
71 path_save = os.path.join(PATH, experiment, 'param{}'.format(label))
72 RMIX = 4.8#5.8 #4.8#5.68#4.8#4.8#2.64#10#2.64
72 RMIX = 4.8#5.8 #4.8#5.68#4.8#4.8#2.64#10#2.64
73 H0 = -1.68 #-1.68# -1.2#-1.68#-1.2#0.5#-1.2
73 H0 = -1.68 #-1.68# -1.2#-1.68#-1.2#0.5#-1.2
74 MASK = 0.6
74 MASK = 0.6
75 #MASK = 0.4#0.35
75 #MASK = 0.4#0.35
76
76
77 from schainpy.controller import Project
77 from schainpy.controller import Project
78
78
79 project = Project()
79 project = Project()
80 project.setup(id='1', name='Sophy', description='sophy proc')
80 project.setup(id='1', name='Sophy', description='sophy proc')
81
81
82 reader = project.addReadUnit(datatype='DigitalRFReader',
82 reader = project.addReadUnit(datatype='DigitalRFReader',
83 path=path,
83 path=path,
84 startDate=start_date,
84 startDate=start_date,
85 endDate=end_date,
85 endDate=end_date,
86 startTime=start_time,
86 startTime=start_time,
87 endTime=end_time,
87 endTime=end_time,
88 delay=30,
88 delay=30,
89 online=args.online,
89 online=args.online,
90 walk=1,
90 walk=1,
91 ippKm = ipp_km,
91 ippKm = ipp_km,
92 getByBlock = 1,
92 getByBlock = 1,
93 nProfileBlocks = N,
93 nProfileBlocks = N,
94 )
94 )
95
95
96 if not conf['usrp_tx']['enable_2']: # One Pulse
96 if not conf['usrp_tx']['enable_2']: # One Pulse
97 n_pulses = 1
97 n_pulses = 1
98 pulse_1_width = conf['usrp_tx']['pulse_1']
98 pulse_1_width = conf['usrp_tx']['pulse_1']
99 pulse_1_repetitions = conf['usrp_tx']['repetitions_1']
99 pulse_1_repetitions = conf['usrp_tx']['repetitions_1']
100 pulse_2_width = 0
100 pulse_2_width = 0
101 pulse_2_repetitions = 0
101 pulse_2_repetitions = 0
102
102
103 voltage = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId())
103 voltage = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId())
104
104
105 if conf['usrp_tx']['code_type_1'] != 'None':
105 if conf['usrp_tx']['code_type_1'] != 'None':
106 codes = [ c.strip() for c in conf['usrp_tx']['code_1'].split(',')]
106 codes = [ c.strip() for c in conf['usrp_tx']['code_1'].split(',')]
107 code = []
107 code = []
108 for c in codes:
108 for c in codes:
109 code.append([int(x) for x in c])
109 code.append([int(x) for x in c])
110 op = voltage.addOperation(name='Decoder', optype='other')
110 op = voltage.addOperation(name='Decoder', optype='other')
111 op.addParameter(name='code', value=code)
111 op.addParameter(name='code', value=code)
112 op.addParameter(name='nCode', value=len(code), format='int')
112 op.addParameter(name='nCode', value=len(code), format='int')
113 op.addParameter(name='nBaud', value=len(code[0]), format='int')
113 op.addParameter(name='nBaud', value=len(code[0]), format='int')
114
114
115 op = voltage.addOperation(name='CohInt', optype='other') #Minimo integrar 2 perfiles por ser codigo complementario
115 op = voltage.addOperation(name='CohInt', optype='other') #Minimo integrar 2 perfiles por ser codigo complementario
116 op.addParameter(name='n', value=len(code), format='int')
116 op.addParameter(name='n', value=len(code), format='int')
117 ncode = len(code)
117 ncode = len(code)
118 else:
118 else:
119 ncode = 1
119 ncode = 1
120 code = ['0']
120 code = ['0']
121
121
122 op = voltage.addOperation(name='setH0')
122 op = voltage.addOperation(name='setH0')
123 op.addParameter(name='h0', value=H0)
123 op.addParameter(name='h0', value=H0)
124
124
125 if args.range > 0:
125 if args.range > 0:
126 op = voltage.addOperation(name='selectHeights')
126 op = voltage.addOperation(name='selectHeights')
127 op.addParameter(name='minIndex', value='0', format='int')
127 op.addParameter(name='minIndex', value='0', format='int')
128 op.addParameter(name='maxIndex', value=max_index(args.range, sample_rate, ipp, H0), format='int')
128 op.addParameter(name='maxIndex', value=max_index(args.range, sample_rate, ipp, H0), format='int')
129
129
130 op = voltage.addOperation(name='PulsePair_vRF', optype='other')
130 op = voltage.addOperation(name='PulsePair_vRF', optype='other')
131 op.addParameter(name='n', value=int(N)/ncode, format='int')
131 op.addParameter(name='n', value=int(N)/ncode, format='int')
132 if args.rmDC:
132 if args.rmDC:
133 op.addParameter(name='removeDC', value=1, format='int')
133 op.addParameter(name='removeDC', value=1, format='int')
134
134
135 proc = project.addProcUnit(datatype='ParametersProc', inputId=voltage.getId())
135 proc = project.addProcUnit(datatype='ParametersProc', inputId=voltage.getId())
136
136
137 opObj10 = proc.addOperation(name="WeatherRadar")
137 opObj10 = proc.addOperation(name="WeatherRadar")
138 opObj10.addParameter(name='tauW',value=(1e-6/sample_rate)*len(code[0]))
138 opObj10.addParameter(name='tauW',value=(1e-6/sample_rate)*len(code[0]))
139 # opObj10.addParameter(name='Pt',value=((1e-6/sample_rate)*len(code[0])/ipp)*200)
139 # opObj10.addParameter(name='Pt',value=((1e-6/sample_rate)*len(code[0])/ipp)*200)
140 opObj10.addParameter(name='Pt',value=200)
140 opObj10.addParameter(name='Pt',value=200)
141
141
142 op = proc.addOperation(name='PedestalInformation')
142 op = proc.addOperation(name='PedestalInformation')
143 op.addParameter(name='path', value=path_ped, format='str')
143 op.addParameter(name='path', value=path_ped, format='str')
144 op.addParameter(name='interval', value='0.04')
144 op.addParameter(name='interval', value='0.04')
145 op.addParameter(name='time_offset', value=time_offset)
145 op.addParameter(name='time_offset', value=time_offset)
146 op.addParameter(name='mode', value=args.mode)
146 op.addParameter(name='mode', value=args.mode)
147 op.addParameter(name='heading', value=conf['heading'])
147
148
148 for param in parameters:
149 for param in parameters:
149 op = proc.addOperation(name='Block360')
150 op = proc.addOperation(name='Block360')
150 op.addParameter(name='runNextOp', value=True)
151 op.addParameter(name='runNextOp', value=True)
151 op.addParameter(name='attr_data', value='data_param')
152 op.addParameter(name='attr_data', value='data_param')
152 op.addParameter(name='angles', value=angles)
153 op.addParameter(name='angles', value=angles)
153
154
154 op= proc.addOperation(name='WeatherParamsPlot')
155 op= proc.addOperation(name='WeatherParamsPlot')
155 if args.save: op.addParameter(name='save', value=path_plots, format='str')
156 if args.save: op.addParameter(name='save', value=path_plots, format='str')
156 op.addParameter(name='save_period', value=-1)
157 op.addParameter(name='save_period', value=-1)
157 op.addParameter(name='show', value=args.show)
158 op.addParameter(name='show', value=args.show)
158 op.addParameter(name='channels', value='0,')
159 op.addParameter(name='channels', value='0,')
159 op.addParameter(name='zmin', value=PARAM[param]['zmin'])
160 op.addParameter(name='zmin', value=PARAM[param]['zmin'])
160 op.addParameter(name='zmax', value=PARAM[param]['zmax'])
161 op.addParameter(name='zmax', value=PARAM[param]['zmax'])
161 op.addParameter(name='yrange', value=20, format='int')
162 op.addParameter(name='yrange', value=20, format='int')
162 op.addParameter(name='xrange', value=args.range, format='int')
163 op.addParameter(name='xrange', value=args.range, format='int')
163 op.addParameter(name='attr_data', value=param, format='str')
164 op.addParameter(name='attr_data', value=param, format='str')
164 op.addParameter(name='labels', value=[PARAM[param]['label'], PARAM[param]['label']])
165 op.addParameter(name='labels', value=[PARAM[param]['label'], PARAM[param]['label']])
165 op.addParameter(name='save_code', value=param)
166 op.addParameter(name='save_code', value=param)
166 op.addParameter(name='cb_label', value=PARAM[param]['cb_label'])
167 op.addParameter(name='cb_label', value=PARAM[param]['cb_label'])
167 op.addParameter(name='colormap', value=PARAM[param]['colormap'])
168 op.addParameter(name='colormap', value=PARAM[param]['colormap'])
168 op.addParameter(name='bgcolor', value='black')
169 op.addParameter(name='bgcolor', value='black')
169 op.addParameter(name='localtime', value=False)
170 op.addParameter(name='localtime', value=False)
170 op.addParameter(name='shapes', value='./shapes')
171 op.addParameter(name='shapes', value='./shapes')
171 op.addParameter(name='latitude', value=conf['latitude'], format='float')
172 op.addParameter(name='latitude', value=conf['latitude'], format='float')
172 op.addParameter(name='longitude', value=conf['longitude'], format='float')
173 op.addParameter(name='longitude', value=conf['longitude'], format='float')
173 op.addParameter(name='map', value=True)
174 op.addParameter(name='map', value=True)
174
175
175 if MASK: op.addParameter(name='mask', value=MASK, format='float')
176 if MASK: op.addParameter(name='mask', value=MASK, format='float')
176 if args.server:
177 if args.server:
177 op.addParameter(name='server', value='190.187.237.239:4444')
178 op.addParameter(name='server', value='190.187.237.239:4444')
178 op.addParameter(name='exp_code', value='400')
179 op.addParameter(name='exp_code', value='400')
179
180
180 desc = {
181 desc = {
181 'Data': {
182 'Data': {
182 'data_param': {PARAM[param]['wrname']: ['H', 'V']},
183 'data_param': {PARAM[param]['wrname']: ['H', 'V']},
183 'utctime': 'time'
184 'utctime': 'time'
184 },
185 },
185 'Metadata': {
186 'Metadata': {
186 'heightList': 'range',
187 'heightList': 'range',
187 'data_azi': 'azimuth',
188 'data_azi': 'azimuth',
188 'data_ele': 'elevation',
189 'data_ele': 'elevation',
189 'mode_op': 'scan_type',
190 'mode_op': 'scan_type',
190 'h0': 'range_correction',
191 'h0': 'range_correction',
191 'dataPP_NOISE': 'noise',
192 'dataPP_NOISE': 'noise',
192 }
193 }
193 }
194 }
194
195
195 if args.save:
196 if args.save:
196 writer = proc.addOperation(name='HDFWriter')
197 writer = proc.addOperation(name='HDFWriter')
197 writer.addParameter(name='path', value=path_save, format='str')
198 writer.addParameter(name='path', value=path_save, format='str')
198 writer.addParameter(name='Reset', value=True)
199 writer.addParameter(name='Reset', value=True)
199 writer.addParameter(name='setType', value='weather')
200 writer.addParameter(name='setType', value='weather')
200 writer.addParameter(name='description', value=json.dumps(desc))
201 writer.addParameter(name='description', value=json.dumps(desc))
201 writer.addParameter(name='blocksPerFile', value='1',format='int')
202 writer.addParameter(name='blocksPerFile', value='1',format='int')
202 writer.addParameter(name='metadataList', value=','.join(META))
203 writer.addParameter(name='metadataList', value=','.join(META))
203 writer.addParameter(name='dataList', value='data_param,utctime')
204 writer.addParameter(name='dataList', value='data_param,utctime')
204 writer.addParameter(name='weather_var', value=param)
205 writer.addParameter(name='weather_var', value=param)
205 writer.addParameter(name='mask', value=MASK, format='float')
206 writer.addParameter(name='mask', value=MASK, format='float')
206 writer.addParameter(name='localtime', value=False)
207 writer.addParameter(name='localtime', value=False)
207 # meta
208 # meta
208 writer.addParameter(name='latitude', value='-12.040436')
209 writer.addParameter(name='latitude', value='-12.040436')
209 writer.addParameter(name='longitude', value='-75.295893')
210 writer.addParameter(name='longitude', value='-75.295893')
210 writer.addParameter(name='altitude', value='3379.2147')
211 writer.addParameter(name='altitude', value='3379.2147')
211 writer.addParameter(name='heading', value='0')
212 writer.addParameter(name='heading', value='0')
212 writer.addParameter(name='radar_name', value='SOPHy')
213 writer.addParameter(name='radar_name', value='SOPHy')
213 writer.addParameter(name='institution', value='IGP')
214 writer.addParameter(name='institution', value='IGP')
214 writer.addParameter(name='contact', value='dscipion@igp.gob.pe')
215 writer.addParameter(name='contact', value='dscipion@igp.gob.pe')
215 writer.addParameter(name='created_by', value='Signal Chain (https://pypi.org/project/schainpy/)')
216 writer.addParameter(name='created_by', value='Signal Chain (https://pypi.org/project/schainpy/)')
216 writer.addParameter(name='range_unit', value='km')
217 writer.addParameter(name='range_unit', value='km')
217 writer.addParameter(name='prf', value=1/ipp)
218 writer.addParameter(name='prf', value=1/ipp)
218 writer.addParameter(name='prf_unit', value='hertz')
219 writer.addParameter(name='prf_unit', value='hertz')
219 writer.addParameter(name='variable', value=PARAM[param]['label'])
220 writer.addParameter(name='variable', value=PARAM[param]['label'])
220 writer.addParameter(name='variable_unit', value=PARAM[param]['cb_label'])
221 writer.addParameter(name='variable_unit', value=PARAM[param]['cb_label'])
221 writer.addParameter(name='n_pulses', value=n_pulses)
222 writer.addParameter(name='n_pulses', value=n_pulses)
222 writer.addParameter(name='pulse1_range', value=RMIX)
223 writer.addParameter(name='pulse1_range', value=RMIX)
223 writer.addParameter(name='pulse1_width', value=pulse_1_width)
224 writer.addParameter(name='pulse1_width', value=pulse_1_width)
224 writer.addParameter(name='pulse2_width', value=pulse_2_width)
225 writer.addParameter(name='pulse2_width', value=pulse_2_width)
225 writer.addParameter(name='pulse1_repetitions', value=pulse_1_repetitions)
226 writer.addParameter(name='pulse1_repetitions', value=pulse_1_repetitions)
226 writer.addParameter(name='pulse2_repetitions', value=pulse_2_repetitions)
227 writer.addParameter(name='pulse2_repetitions', value=pulse_2_repetitions)
227 writer.addParameter(name='pulse_width_unit', value='microseconds')
228 writer.addParameter(name='pulse_width_unit', value='microseconds')
228 writer.addParameter(name='snr_threshold', value=MASK)
229 writer.addParameter(name='snr_threshold', value=MASK)
229
230
230
231
231 else: #Two pulses
232 else: #Two pulses
232 n_pulses = 1
233 n_pulses = 1
233 pulse_1_width = conf['usrp_tx']['pulse_1']
234 pulse_1_width = conf['usrp_tx']['pulse_1']
234 pulse_1_repetitions = conf['usrp_tx']['repetitions_1']
235 pulse_1_repetitions = conf['usrp_tx']['repetitions_1']
235 pulse_2_width = conf['usrp_tx']['pulse_2']
236 pulse_2_width = conf['usrp_tx']['pulse_2']
236 pulse_2_repetitions = conf['usrp_tx']['repetitions_2']
237 pulse_2_repetitions = conf['usrp_tx']['repetitions_2']
237
238
238 voltage1 = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId())
239 voltage1 = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId())
239
240
240 op = voltage1.addOperation(name='ProfileSelector')
241 op = voltage1.addOperation(name='ProfileSelector')
241 op.addParameter(name='profileRangeList', value='0,{}'.format(conf['usrp_tx']['repetitions_1']-1))
242 op.addParameter(name='profileRangeList', value='0,{}'.format(conf['usrp_tx']['repetitions_1']-1))
242
243
243 if conf['usrp_tx']['code_type_1'] != 'None':
244 if conf['usrp_tx']['code_type_1'] != 'None':
244 codes = [ c.strip() for c in conf['usrp_tx']['code_1'].split(',')]
245 codes = [ c.strip() for c in conf['usrp_tx']['code_1'].split(',')]
245 code = []
246 code = []
246 for c in codes:
247 for c in codes:
247 code.append([int(x) for x in c])
248 code.append([int(x) for x in c])
248 op = voltage1.addOperation(name='Decoder', optype='other')
249 op = voltage1.addOperation(name='Decoder', optype='other')
249 op.addParameter(name='code', value=code)
250 op.addParameter(name='code', value=code)
250 op.addParameter(name='nCode', value=len(code), format='int')
251 op.addParameter(name='nCode', value=len(code), format='int')
251 op.addParameter(name='nBaud', value=len(code[0]), format='int')
252 op.addParameter(name='nBaud', value=len(code[0]), format='int')
252 ncode = len(code)
253 ncode = len(code)
253 else:
254 else:
254 ncode = 1
255 ncode = 1
255 code = ['0']
256 code = ['0']
256
257
257 op = voltage1.addOperation(name='CohInt', optype='other') #Minimo integrar 2 perfiles por ser codigo complementario
258 op = voltage1.addOperation(name='CohInt', optype='other') #Minimo integrar 2 perfiles por ser codigo complementario
258 op.addParameter(name='n', value=2, format='int')
259 op.addParameter(name='n', value=2, format='int')
259
260
260 op = voltage1.addOperation(name='setH0')
261 op = voltage1.addOperation(name='setH0')
261 op.addParameter(name='h0', value=H0, format='float')
262 op.addParameter(name='h0', value=H0, format='float')
262
263
263 if args.range > 0:
264 if args.range > 0:
264 op = voltage1.addOperation(name='selectHeights')
265 op = voltage1.addOperation(name='selectHeights')
265 op.addParameter(name='minIndex', value=max_index(0, sample_rate, ipp, H0), format='int')
266 op.addParameter(name='minIndex', value=max_index(0, sample_rate, ipp, H0), format='int')
266 op.addParameter(name='maxIndex', value=max_index(RMIX, sample_rate, ipp, H0), format='int')
267 op.addParameter(name='maxIndex', value=max_index(RMIX, sample_rate, ipp, H0), format='int')
267
268
268 #op = voltage1.addOperation(name='setH0')
269 #op = voltage1.addOperation(name='setH0')
269 #op.addParameter(name='h0', value=H0, format='float')
270 #op.addParameter(name='h0', value=H0, format='float')
270
271
271 op = voltage1.addOperation(name='PulsePair_vRF', optype='other')
272 op = voltage1.addOperation(name='PulsePair_vRF', optype='other')
272 op.addParameter(name='n', value=int(conf['usrp_tx']['repetitions_1'])/ncode, format='int')
273 op.addParameter(name='n', value=int(conf['usrp_tx']['repetitions_1'])/ncode, format='int')
273 if args.rmDC:
274 if args.rmDC:
274 op.addParameter(name='removeDC', value=1, format='int')
275 op.addParameter(name='removeDC', value=1, format='int')
275
276
276 proc1 = project.addProcUnit(datatype='ParametersProc', inputId=voltage1.getId())
277 proc1 = project.addProcUnit(datatype='ParametersProc', inputId=voltage1.getId())
277 proc1.addParameter(name='runNextUnit', value=True)
278 proc1.addParameter(name='runNextUnit', value=True)
278
279
279 opObj10 = proc1.addOperation(name="WeatherRadar")
280 opObj10 = proc1.addOperation(name="WeatherRadar")
280 opObj10.addParameter(name='CR_Flag',value=True)
281 opObj10.addParameter(name='CR_Flag',value=True)
281 opObj10.addParameter(name='tauW',value=(1e-6/sample_rate)*len(code[0]))
282 opObj10.addParameter(name='tauW',value=(1e-6/sample_rate)*len(code[0]))
282 #opObj10.addParameter(name='Pt',value=((1e-6/sample_rate)*len(code[0])/ipp)*200)
283 #opObj10.addParameter(name='Pt',value=((1e-6/sample_rate)*len(code[0])/ipp)*200)
283 opObj10.addParameter(name='Pt',value=200)
284 opObj10.addParameter(name='Pt',value=200)
284 #opObj10.addParameter(name='min_index',value=0)
285 #opObj10.addParameter(name='min_index',value=0)
285 opObj10.addParameter(name='min_index',value=max_index(0, sample_rate, ipp, H0))
286 opObj10.addParameter(name='min_index',value=max_index(0, sample_rate, ipp, H0))
286 #opObj10.addParameter(name='sesgoZD',value=7.73)
287 #opObj10.addParameter(name='sesgoZD',value=7.73)
287
288
288
289
289 op = proc1.addOperation(name='PedestalInformation')
290 op = proc1.addOperation(name='PedestalInformation')
290 op.addParameter(name='path', value=path_ped, format='str')
291 op.addParameter(name='path', value=path_ped, format='str')
291 op.addParameter(name='interval', value='0.04')
292 op.addParameter(name='interval', value='0.04')
292 op.addParameter(name='time_offset', value=time_offset)
293 op.addParameter(name='time_offset', value=time_offset)
293 op.addParameter(name='mode', value=args.mode)
294 op.addParameter(name='mode', value=args.mode)
294
295
295 op = proc1.addOperation(name='Block360')
296 op = proc1.addOperation(name='Block360')
296 op.addParameter(name='attr_data', value='data_param')
297 op.addParameter(name='attr_data', value='data_param')
297 op.addParameter(name='runNextOp', value=True)
298 op.addParameter(name='runNextOp', value=True)
298 op.addParameter(name='angles', value=angles)
299 op.addParameter(name='angles', value=angles)
299 #op.addParameter(name='horario',value=False)
300 #op.addParameter(name='horario',value=False)
300
301
301 voltage2 = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId())
302 voltage2 = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId())
302
303
303 op = voltage2.addOperation(name='ProfileSelector')
304 op = voltage2.addOperation(name='ProfileSelector')
304 op.addParameter(name='profileRangeList', value='{},{}'.format(conf['usrp_tx']['repetitions_1'], conf['usrp_tx']['repetitions_1']+conf['usrp_tx']['repetitions_2']-1))
305 op.addParameter(name='profileRangeList', value='{},{}'.format(conf['usrp_tx']['repetitions_1'], conf['usrp_tx']['repetitions_1']+conf['usrp_tx']['repetitions_2']-1))
305
306
306 if conf['usrp_tx']['code_type_2']:
307 if conf['usrp_tx']['code_type_2']:
307 codes = [ c.strip() for c in conf['usrp_tx']['code_2'].split(',')]
308 codes = [ c.strip() for c in conf['usrp_tx']['code_2'].split(',')]
308 code = []
309 code = []
309 for c in codes:
310 for c in codes:
310 code.append([int(x) for x in c])
311 code.append([int(x) for x in c])
311 op = voltage2.addOperation(name='Decoder', optype='other')
312 op = voltage2.addOperation(name='Decoder', optype='other')
312 op.addParameter(name='code', value=code)
313 op.addParameter(name='code', value=code)
313 op.addParameter(name='nCode', value=len(code), format='int')
314 op.addParameter(name='nCode', value=len(code), format='int')
314 op.addParameter(name='nBaud', value=len(code[0]), format='int')
315 op.addParameter(name='nBaud', value=len(code[0]), format='int')
315
316
316 op = voltage2.addOperation(name='CohInt', optype='other') #Minimo integrar 2 perfiles por ser codigo complementario
317 op = voltage2.addOperation(name='CohInt', optype='other') #Minimo integrar 2 perfiles por ser codigo complementario
317 op.addParameter(name='n', value=len(code), format='int')
318 op.addParameter(name='n', value=len(code), format='int')
318 ncode = len(code)
319 ncode = len(code)
319 else:
320 else:
320 ncode = 1
321 ncode = 1
321
322
322 op = voltage2.addOperation(name='setH0')
323 op = voltage2.addOperation(name='setH0')
323 op.addParameter(name='h0', value=H0, format='float')
324 op.addParameter(name='h0', value=H0, format='float')
324
325
325 if args.range > 0:
326 if args.range > 0:
326 print('voltage 2', flush=True)
327 print('voltage 2', flush=True)
327 op = voltage2.addOperation(name='selectHeights')
328 op = voltage2.addOperation(name='selectHeights')
328 op.addParameter(name='minIndex', value=max_index(RMIX, sample_rate, ipp, H0), format='int')
329 op.addParameter(name='minIndex', value=max_index(RMIX, sample_rate, ipp, H0), format='int')
329 op.addParameter(name='maxIndex', value=max_index(args.range, sample_rate, ipp, H0), format='int')
330 op.addParameter(name='maxIndex', value=max_index(args.range, sample_rate, ipp, H0), format='int')
330
331
331 #op = voltage2.addOperation(name='setH0')
332 #op = voltage2.addOperation(name='setH0')
332 #op.addParameter(name='h0', value=H0, format='float')
333 #op.addParameter(name='h0', value=H0, format='float')
333
334
334 op = voltage2.addOperation(name='PulsePair_vRF', optype='other')
335 op = voltage2.addOperation(name='PulsePair_vRF', optype='other')
335 op.addParameter(name='n', value=int(conf['usrp_tx']['repetitions_2'])/ncode, format='int')
336 op.addParameter(name='n', value=int(conf['usrp_tx']['repetitions_2'])/ncode, format='int')
336 if args.rmDC:
337 if args.rmDC:
337 op.addParameter(name='removeDC', value=1, format='int')
338 op.addParameter(name='removeDC', value=1, format='int')
338
339
339 proc2 = project.addProcUnit(datatype='ParametersProc', inputId=voltage2.getId())
340 proc2 = project.addProcUnit(datatype='ParametersProc', inputId=voltage2.getId())
340 proc2.addParameter(name='runNextUnit', value=True)
341 proc2.addParameter(name='runNextUnit', value=True)
341
342
342 opObj10 = proc2.addOperation(name="WeatherRadar")
343 opObj10 = proc2.addOperation(name="WeatherRadar")
343 opObj10.addParameter(name='CR_Flag',value=True,format='bool')
344 opObj10.addParameter(name='CR_Flag',value=True,format='bool')
344 opObj10.addParameter(name='tauW',value=(1e-6/sample_rate)*len(code[0]))
345 opObj10.addParameter(name='tauW',value=(1e-6/sample_rate)*len(code[0]))
345 #opObj10.addParameter(name='Pt',value=((1e-6/sample_rate)*len(code[0])/ipp)*200)
346 #opObj10.addParameter(name='Pt',value=((1e-6/sample_rate)*len(code[0])/ipp)*200)
346 opObj10.addParameter(name='Pt',value=200)
347 opObj10.addParameter(name='Pt',value=200)
347 opObj10.addParameter(name='min_index',value=max_index(RMIX, sample_rate, ipp, H0))
348 opObj10.addParameter(name='min_index',value=max_index(RMIX, sample_rate, ipp, H0))
348 #opObj10.addParameter(name='sesgoZD',value=7.73)
349 #opObj10.addParameter(name='sesgoZD',value=7.73)
349
350
350 op = proc2.addOperation(name='PedestalInformation')
351 op = proc2.addOperation(name='PedestalInformation')
351 op.addParameter(name='path', value=path_ped, format='str')
352 op.addParameter(name='path', value=path_ped, format='str')
352 op.addParameter(name='interval', value='0.04')
353 op.addParameter(name='interval', value='0.04')
353 op.addParameter(name='time_offset', value=time_offset)
354 op.addParameter(name='time_offset', value=time_offset)
354 op.addParameter(name='mode', value=args.mode)
355 op.addParameter(name='mode', value=args.mode)
356 op.addParameter(name='heading', value=conf['heading'])
355
357
356 op = proc2.addOperation(name='Block360')
358 op = proc2.addOperation(name='Block360')
357 op.addParameter(name='attr_data', value='data_param')
359 op.addParameter(name='attr_data', value='data_param')
358 op.addParameter(name='runNextOp', value=True)
360 op.addParameter(name='runNextOp', value=True)
359 op.addParameter(name='angles', value=angles)
361 op.addParameter(name='angles', value=angles)
360 #op.addParameter(name='horario',value=False)
362 #op.addParameter(name='horario',value=False)
361
363
362 merge = project.addProcUnit(datatype='MergeProc', inputId=[proc1.getId(), proc2.getId()])
364 merge = project.addProcUnit(datatype='MergeProc', inputId=[proc1.getId(), proc2.getId()])
363 merge.addParameter(name='attr_data', value='data_param')
365 merge.addParameter(name='attr_data', value='data_param')
364 merge.addParameter(name='mode', value='7') #RM
366 merge.addParameter(name='mode', value='7') #RM
365
367
366 for param in parameters:
368 for param in parameters:
367
369
368 if args.plot:
370 if args.plot:
369 op= merge.addOperation(name='WeatherParamsPlot')
371 op= merge.addOperation(name='WeatherParamsPlot')
370 if args.save:
372 if args.save:
371 op.addParameter(name='save', value=path_plots, format='str')
373 op.addParameter(name='save', value=path_plots, format='str')
372 op.addParameter(name='save_period', value=-1)
374 op.addParameter(name='save_period', value=-1)
373 op.addParameter(name='show', value=args.show)
375 op.addParameter(name='show', value=args.show)
374 #op.addParameter(name='channels', value='0,1')
376 #op.addParameter(name='channels', value='0,1')
375 op.addParameter(name='channels', value='0,')
377 op.addParameter(name='channels', value='0,')
376 op.addParameter(name='zmin', value=PARAM[param]['zmin'], format='int')
378 op.addParameter(name='zmin', value=PARAM[param]['zmin'], format='int')
377 op.addParameter(name='zmax', value=PARAM[param]['zmax'], format='int')
379 op.addParameter(name='zmax', value=PARAM[param]['zmax'], format='int')
378 op.addParameter(name='yrange', value=20, format='int')
380 op.addParameter(name='yrange', value=20, format='int')
379 op.addParameter(name='xrange', value=args.range, format='int')
381 op.addParameter(name='xrange', value=args.range, format='int')
380 op.addParameter(name='attr_data', value=param, format='str')
382 op.addParameter(name='attr_data', value=param, format='str')
381 op.addParameter(name='labels', value=[[PARAM[param]['label']], [PARAM[param]['label']]])
383 op.addParameter(name='labels', value=[[PARAM[param]['label']], [PARAM[param]['label']]])
382 op.addParameter(name='save_code', value=param)
384 op.addParameter(name='save_code', value=param)
383 op.addParameter(name='cb_label', value=PARAM[param]['cb_label'])
385 op.addParameter(name='cb_label', value=PARAM[param]['cb_label'])
384 op.addParameter(name='colormap', value=PARAM[param]['colormap'])
386 op.addParameter(name='colormap', value=PARAM[param]['colormap'])
385 op.addParameter(name='bgcolor', value='black')
387 op.addParameter(name='bgcolor', value='black')
386 op.addParameter(name='localtime', value=False)
388 op.addParameter(name='localtime', value=False)
387 op.addParameter(name='shapes', value='./shapes')
389 op.addParameter(name='shapes', value='./shapes')
388 op.addParameter(name='latitude', value=conf['latitude'], format='float')
390 op.addParameter(name='latitude', value=conf['latitude'], format='float')
389 op.addParameter(name='longitude', value=conf['longitude'], format='float')
391 op.addParameter(name='longitude', value=conf['longitude'], format='float')
390 op.addParameter(name='map', value=False)
392 op.addParameter(name='map', value=False)
391
393
392 if MASK: op.addParameter(name='mask', value=MASK, format='float')
394 if MASK: op.addParameter(name='mask', value=MASK, format='float')
393 if args.server:
395 if args.server:
394 op.addParameter(name='server', value='190.187.237.239:4444')
396 op.addParameter(name='server', value='190.187.237.239:4444')
395 op.addParameter(name='exp_code', value='400')
397 op.addParameter(name='exp_code', value='400')
396
398
397 desc = {
399 desc = {
398 'Data': {
400 'Data': {
399 'data_param': {PARAM[param]['wrname']: ['H', 'V']},
401 'data_param': {PARAM[param]['wrname']: ['H', 'V']},
400 'utctime': 'time'
402 'utctime': 'time'
401 },
403 },
402 'Metadata': {
404 'Metadata': {
403 'heightList': 'range',
405 'heightList': 'range',
404 'data_azi': 'azimuth',
406 'data_azi': 'azimuth',
405 'data_ele': 'elevation',
407 'data_ele': 'elevation',
406 'mode_op': 'scan_type',
408 'mode_op': 'scan_type',
407 'h0': 'range_correction',
409 'h0': 'range_correction',
408 'dataPP_NOISE': 'noise',
410 'dataPP_NOISE': 'noise',
409 }
411 }
410 }
412 }
411
413
412 if args.save:
414 if args.save:
413 writer = merge.addOperation(name='HDFWriter')
415 writer = merge.addOperation(name='HDFWriter')
414 writer.addParameter(name='path', value=path_save, format='str')
416 writer.addParameter(name='path', value=path_save, format='str')
415 writer.addParameter(name='Reset', value=True)
417 writer.addParameter(name='Reset', value=True)
416 writer.addParameter(name='setType', value='weather')
418 writer.addParameter(name='setType', value='weather')
417 writer.addParameter(name='setChannel', value='0') #new parameter choose ch 0 H or ch 1 V
419 writer.addParameter(name='setChannel', value='0') #new parameter choose ch 0 H or ch 1 V
418 writer.addParameter(name='description', value=json.dumps(desc))
420 writer.addParameter(name='description', value=json.dumps(desc))
419 writer.addParameter(name='blocksPerFile', value='1',format='int')
421 writer.addParameter(name='blocksPerFile', value='1',format='int')
420 writer.addParameter(name='metadataList', value=','.join(META))
422 writer.addParameter(name='metadataList', value=','.join(META))
421 writer.addParameter(name='dataList', value='data_param,utctime')
423 writer.addParameter(name='dataList', value='data_param,utctime')
422 writer.addParameter(name='weather_var', value=param)
424 writer.addParameter(name='weather_var', value=param)
423 writer.addParameter(name='mask', value=MASK, format='float')
425 writer.addParameter(name='mask', value=MASK, format='float')
424 writer.addParameter(name='localtime', value=False)
426 writer.addParameter(name='localtime', value=False)
425 # meta
427 # meta
426 writer.addParameter(name='latitude', value=conf['latitude'])
428 writer.addParameter(name='latitude', value=conf['latitude'])
427 writer.addParameter(name='longitude', value=conf['longitude'])
429 writer.addParameter(name='longitude', value=conf['longitude'])
428 writer.addParameter(name='altitude', value=conf['altitude'])
430 writer.addParameter(name='altitude', value=conf['altitude'])
429 writer.addParameter(name='heading', value=conf['heading'])
431 writer.addParameter(name='heading', value=conf['heading'])
430 writer.addParameter(name='radar_name', value='SOPHy')
432 writer.addParameter(name='radar_name', value='SOPHy')
431 writer.addParameter(name='institution', value='IGP')
433 writer.addParameter(name='institution', value='IGP')
432 writer.addParameter(name='contact', value='dscipion@igp.gob.pe')
434 writer.addParameter(name='contact', value='dscipion@igp.gob.pe')
433 writer.addParameter(name='created_by', value='Signal Chain (https://pypi.org/project/schainpy/)')
435 writer.addParameter(name='created_by', value='Signal Chain (https://pypi.org/project/schainpy/)')
434 writer.addParameter(name='range_unit', value='km')
436 writer.addParameter(name='range_unit', value='km')
435 writer.addParameter(name='prf', value=1/ipp)
437 writer.addParameter(name='prf', value=1/ipp)
436 writer.addParameter(name='prf_unit', value='hertz')
438 writer.addParameter(name='prf_unit', value='hertz')
437 writer.addParameter(name='variable', value=PARAM[param]['label'])
439 writer.addParameter(name='variable', value=PARAM[param]['label'])
438 writer.addParameter(name='variable_unit', value=PARAM[param]['cb_label'])
440 writer.addParameter(name='variable_unit', value=PARAM[param]['cb_label'])
439 writer.addParameter(name='n_pulses', value=n_pulses)
441 writer.addParameter(name='n_pulses', value=n_pulses)
440 writer.addParameter(name='pulse1_range', value=RMIX)
442 writer.addParameter(name='pulse1_range', value=RMIX)
441 writer.addParameter(name='pulse1_width', value=pulse_1_width)
443 writer.addParameter(name='pulse1_width', value=pulse_1_width)
442 writer.addParameter(name='pulse2_width', value=pulse_2_width)
444 writer.addParameter(name='pulse2_width', value=pulse_2_width)
443 writer.addParameter(name='pulse1_repetitions', value=pulse_1_repetitions)
445 writer.addParameter(name='pulse1_repetitions', value=pulse_1_repetitions)
444 writer.addParameter(name='pulse2_repetitions', value=pulse_2_repetitions)
446 writer.addParameter(name='pulse2_repetitions', value=pulse_2_repetitions)
445 writer.addParameter(name='pulse_width_unit', value='microseconds')
447 writer.addParameter(name='pulse_width_unit', value='microseconds')
446 writer.addParameter(name='snr_threshold', value=MASK)
448 writer.addParameter(name='snr_threshold', value=MASK)
447
449
448 return project
450 return project
449
451
450 if __name__ == '__main__':
452 if __name__ == '__main__':
451
453
452 parser = argparse.ArgumentParser(description='Script to process SOPHy data.')
454 parser = argparse.ArgumentParser(description='Script to process SOPHy data.')
453 parser.add_argument('experiment',
455 parser.add_argument('experiment',
454 help='Experiment name')
456 help='Experiment name')
455 parser.add_argument('--parameters', nargs='*', default=['S'],
457 parser.add_argument('--parameters', nargs='*', default=['S'],
456 help='Variables to process: P, Z, V')
458 help='Variables to process: P, Z, V')
457 parser.add_argument('--angles', nargs='*', default=[], type=int,
459 parser.add_argument('--angles', nargs='*', default=[], type=int,
458 help='Angles to process')
460 help='Angles to process')
459 parser.add_argument('--time_offset', default=0,
461 parser.add_argument('--time_offset', default=0,
460 help='Fix time offset')
462 help='Fix time offset')
461 parser.add_argument('--range', default=60, type=float,
463 parser.add_argument('--range', default=60, type=float,
462 help='Max range to plot')
464 help='Max range to plot')
463 parser.add_argument('--save', action='store_true',
465 parser.add_argument('--save', action='store_true',
464 help='Create output files')
466 help='Create output files')
465 parser.add_argument('--plot', action='store_true',
467 parser.add_argument('--plot', action='store_true',
466 help='Create plot files')
468 help='Create plot files')
467 parser.add_argument('--show', action='store_true',
469 parser.add_argument('--show', action='store_true',
468 help='Show matplotlib plot.')
470 help='Show matplotlib plot.')
469 parser.add_argument('--online', action='store_true',
471 parser.add_argument('--online', action='store_true',
470 help='Set online mode.')
472 help='Set online mode.')
471 parser.add_argument('--server', action='store_true',
473 parser.add_argument('--server', action='store_true',
472 help='Send to realtime')
474 help='Send to realtime')
473 parser.add_argument('--start_time', default='',
475 parser.add_argument('--start_time', default='',
474 help='Set start time.')
476 help='Set start time.')
475 parser.add_argument('--end_time', default='',
477 parser.add_argument('--end_time', default='',
476 help='Set end time.')
478 help='Set end time.')
477 parser.add_argument('--label', default='',
479 parser.add_argument('--label', default='',
478 help='Label for plot & param folder')
480 help='Label for plot & param folder')
479 parser.add_argument('--mode', default=None,
481 parser.add_argument('--mode', default=None,
480 help='Type of scan')
482 help='Type of scan')
481 parser.add_argument('--rmDC', action='store_true',
483 parser.add_argument('--rmDC', action='store_true',
482 help='Apply remove DC.')
484 help='Apply remove DC.')
483 args = parser.parse_args()
485 args = parser.parse_args()
484
486
485 project = main(args)
487 project = main(args)
486 project.start()
488 project.start()
487
489
488 #python sophy_A.py HYO_CC4_CC64_COMB@2022-12-27T00-00-32 --parameters Z --plot --save --show --rmDC --label Z_04 --range 60 --start_time "22:00:00"
490 #python sophy_A.py HYO_CC4_CC64_COMB@2022-12-27T00-00-32 --parameters Z --plot --save --show --rmDC --label Z_04 --range 60 --start_time "22:00:00"
489 # colocar siempre el range que asume 0 y no hace la seleccion de alturas
491 # colocar siempre el range que asume 0 y no hace la seleccion de alturas
General Comments 0
You need to be logged in to leave comments. Login now