@@ -0,0 +1,418 | |||||
|
1 | # SOPHY PROC script | |||
|
2 | import os, sys, json, argparse | |||
|
3 | import datetime | |||
|
4 | import time | |||
|
5 | ||||
|
6 | PATH = '/DATA_RM/DATA' | |||
|
7 | PATH = '/media/jespinoza/Elements' | |||
|
8 | PATH = '/media/jespinoza/data/SOPHY' | |||
|
9 | PATH = '/home/soporte/Documents/EVENTO/' | |||
|
10 | ||||
|
11 | # NOTA: EL GRABADO ESTA EN PARAM | |||
|
12 | PARAM = { | |||
|
13 | 'S': {'zmin': -45, 'zmax': -25, 'colormap': 'jet', 'label': 'Power', 'wrname': 'power','cb_label': 'dBm', 'ch':0}, | |||
|
14 | 'SNR': {'zmin': -40, 'zmax': -20, 'colormap': 'jet', 'label': 'SNR', 'wrname': 'snr','cb_label': 'dB', 'ch':0}, | |||
|
15 | 'V': {'zmin': -12, 'zmax': 12, 'colormap': 'sophy_v', 'label': 'Velocity', 'wrname': 'velocity', 'cb_label': 'm/s', 'ch':0}, | |||
|
16 | 'R': {'zmin': 0, 'zmax': 1, 'colormap': 'jet', 'label': 'RhoHV', 'wrname':'rhoHV', 'cb_label': '', 'ch':0}, | |||
|
17 | 'P': {'zmin': -180,'zmax': 180,'colormap': 'RdBu_r', 'label': 'PhiDP', 'wrname':'phiDP' , 'cb_label': 'degrees', 'ch':0}, | |||
|
18 | 'D': {'zmin': -30, 'zmax': 80, 'colormap': 'sophy_r','label': 'ZDR','wrname':'differential_reflectivity' , 'cb_label': 'dBz','ch':0}, | |||
|
19 | 'Z': {'zmin': -30, 'zmax': 80, 'colormap': 'sophy_r','label': 'Reflectivity ', 'wrname':'reflectivity', 'cb_label': 'dBz','ch':0}, | |||
|
20 | 'W': {'zmin': 0, 'zmax': 15, 'colormap': 'sophy_w','label': 'Spectral Width', 'wrname':'spectral_width', 'cb_label': 'm/s', 'ch':0} | |||
|
21 | } | |||
|
22 | ||||
|
23 | META = ['heightList', 'data_azi', 'data_ele', 'mode_op', 'latitude', 'longitude', 'altitude', 'heading', 'radar_name', | |||
|
24 | 'institution', 'contact', 'h0', 'range_unit', 'prf', 'prf_unit', 'variable', 'variable_unit', 'n_pulses', | |||
|
25 | 'pulse1_range', 'pulse1_width', 'pulse2_width', 'pulse1_repetitions', 'pulse2_repetitions', 'pulse_width_unit', | |||
|
26 | 'snr_threshold','dataPP_NOISE'] | |||
|
27 | ||||
|
28 | ||||
|
29 | def max_index(r, sample_rate, ipp): | |||
|
30 | ||||
|
31 | return int(sample_rate*ipp*1e6 * r / 60) + int(sample_rate*ipp*1e6 * 1.2 / 60) | |||
|
32 | ||||
|
33 | def main(args): | |||
|
34 | ||||
|
35 | experiment = args.experiment | |||
|
36 | fp = open(os.path.join(PATH, experiment, 'experiment.conf')) | |||
|
37 | conf = json.loads(fp.read()) | |||
|
38 | ||||
|
39 | ipp_km = conf['usrp_tx']['ipp'] | |||
|
40 | ipp = ipp_km * 2 /300000 | |||
|
41 | sample_rate = conf['usrp_rx']['sample_rate'] | |||
|
42 | axis = ['0' if x=='elevation' else '1' for x in conf['pedestal']['axis']] # AZIMUTH 1 ELEVACION 0 | |||
|
43 | speed_axis = conf['pedestal']['speed'] | |||
|
44 | steps = conf['pedestal']['table'] | |||
|
45 | time_offset = args.time_offset | |||
|
46 | parameters = args.parameters | |||
|
47 | start_date = experiment.split('@')[1].split('T')[0].replace('-', '/') | |||
|
48 | end_date = start_date | |||
|
49 | if args.start_time: | |||
|
50 | start_time = args.start_time | |||
|
51 | else: | |||
|
52 | start_time = experiment.split('@')[1].split('T')[1].replace('-', ':') | |||
|
53 | end_time = '23:59:59' | |||
|
54 | N = int(1/(speed_axis[0]*ipp)) # 1 GRADO DE RESOLUCION | |||
|
55 | path = os.path.join(PATH, experiment, 'rawdata') | |||
|
56 | path_ped = os.path.join(PATH, experiment, 'position') | |||
|
57 | if args.label: | |||
|
58 | label = '-{}'.format(args.label) | |||
|
59 | else: | |||
|
60 | label = '' | |||
|
61 | path_plots = os.path.join(PATH, experiment, 'plots{}'.format(label)) | |||
|
62 | path_save = os.path.join(PATH, experiment, 'param{}'.format(label)) | |||
|
63 | RMIX = 1.62 | |||
|
64 | H0 = -1.68 | |||
|
65 | MASK = 0.3 | |||
|
66 | ||||
|
67 | from schainpy.controller import Project | |||
|
68 | ||||
|
69 | project = Project() | |||
|
70 | project.setup(id='1', name='Sophy', description='sophy proc') | |||
|
71 | ||||
|
72 | reader = project.addReadUnit(datatype='DigitalRFReader', | |||
|
73 | path=path, | |||
|
74 | startDate=start_date, | |||
|
75 | endDate=end_date, | |||
|
76 | startTime=start_time, | |||
|
77 | endTime=end_time, | |||
|
78 | delay=30, | |||
|
79 | online=args.online, | |||
|
80 | walk=1, | |||
|
81 | ippKm = ipp_km, | |||
|
82 | getByBlock = 1, | |||
|
83 | nProfileBlocks = N, | |||
|
84 | ) | |||
|
85 | ||||
|
86 | if not conf['usrp_tx']['enable_2']: # One Pulse | |||
|
87 | n_pulses = 1 | |||
|
88 | pulse_1_width = conf['usrp_tx']['pulse_1'] | |||
|
89 | pulse_1_repetitions = conf['usrp_tx']['repetitions_1'] | |||
|
90 | pulse_2_width = 0 | |||
|
91 | pulse_2_repetitions = 0 | |||
|
92 | ||||
|
93 | voltage = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId()) | |||
|
94 | ||||
|
95 | if conf['usrp_tx']['code_type_1'] != 'None': | |||
|
96 | codes = [ c.strip() for c in conf['usrp_tx']['code_1'].split(',')] | |||
|
97 | code = [] | |||
|
98 | for c in codes: | |||
|
99 | code.append([int(x) for x in c]) | |||
|
100 | op = voltage.addOperation(name='Decoder', optype='other') | |||
|
101 | op.addParameter(name='code', value=code) | |||
|
102 | op.addParameter(name='nCode', value=len(code), format='int') | |||
|
103 | op.addParameter(name='nBaud', value=len(code[0]), format='int') | |||
|
104 | ||||
|
105 | op = voltage.addOperation(name='CohInt', optype='other') #Minimo integrar 2 perfiles por ser codigo complementario | |||
|
106 | op.addParameter(name='n', value=len(code), format='int') | |||
|
107 | ncode = len(code) | |||
|
108 | else: | |||
|
109 | ncode = 1 | |||
|
110 | code = ['0'] | |||
|
111 | ||||
|
112 | op = voltage.addOperation(name='setH0') | |||
|
113 | op.addParameter(name='h0', value=H0) | |||
|
114 | ||||
|
115 | if args.range > 0: | |||
|
116 | op = voltage.addOperation(name='selectHeights') | |||
|
117 | op.addParameter(name='minIndex', value='0', format='int') | |||
|
118 | op.addParameter(name='maxIndex', value=max_index(args.range, sample_rate, ipp), format='int') | |||
|
119 | ||||
|
120 | op = voltage.addOperation(name='PulsePair_vRF', optype='other') | |||
|
121 | op.addParameter(name='n', value=int(N)/ncode, format='int') | |||
|
122 | ||||
|
123 | proc = project.addProcUnit(datatype='ParametersProc', inputId=voltage.getId()) | |||
|
124 | ||||
|
125 | opObj10 = proc.addOperation(name="WeatherRadar") | |||
|
126 | opObj10.addParameter(name='tauW',value=(1e-6/sample_rate)*len(code[0])) | |||
|
127 | opObj10.addParameter(name='Pt',value=((1e-6/sample_rate)*len(code[0])/ipp)*200) | |||
|
128 | ||||
|
129 | op = proc.addOperation(name='PedestalInformation') | |||
|
130 | op.addParameter(name='path', value=path_ped, format='str') | |||
|
131 | op.addParameter(name='interval', value='0.04') | |||
|
132 | op.addParameter(name='time_offset', value=time_offset) | |||
|
133 | op.addParameter(name='mode', value='PPI') | |||
|
134 | ||||
|
135 | for param in parameters: | |||
|
136 | op = proc.addOperation(name='Block360') | |||
|
137 | op.addParameter(name='runNextOp', value=True) | |||
|
138 | ||||
|
139 | op= proc.addOperation(name='WeatherParamsPlot') | |||
|
140 | if args.save: op.addParameter(name='save', value=path_plots, format='str') | |||
|
141 | op.addParameter(name='save_period', value=-1) | |||
|
142 | op.addParameter(name='show', value=args.show) | |||
|
143 | op.addParameter(name='channels', value='1,') | |||
|
144 | op.addParameter(name='zmin', value=PARAM[param]['zmin']) | |||
|
145 | op.addParameter(name='zmax', value=PARAM[param]['zmax']) | |||
|
146 | op.addParameter(name='attr_data', value=param, format='str') | |||
|
147 | op.addParameter(name='labels', value=[PARAM[param]['label']]) | |||
|
148 | op.addParameter(name='save_code', value=param) | |||
|
149 | op.addParameter(name='cb_label', value=PARAM[param]['cb_label']) | |||
|
150 | op.addParameter(name='colormap', value=PARAM[param]['colormap']) | |||
|
151 | op.addParameter(name='bgcolor', value='black') | |||
|
152 | if MASK: op.addParameter(name='mask', value=MASK, format='float') | |||
|
153 | if args.server: | |||
|
154 | op.addParameter(name='server', value='0.0.0.0:4444') | |||
|
155 | op.addParameter(name='exp_code', value='400') | |||
|
156 | ||||
|
157 | desc = { | |||
|
158 | 'Data': { | |||
|
159 | param: PARAM[param]['wrname'], | |||
|
160 | 'utctime': 'time' | |||
|
161 | }, | |||
|
162 | 'Metadata': { | |||
|
163 | 'heightList': 'range', | |||
|
164 | 'data_azi': 'azimuth', | |||
|
165 | 'data_ele': 'elevation', | |||
|
166 | 'mode_op': 'scan_type', | |||
|
167 | 'h0': 'range_correction', | |||
|
168 | } | |||
|
169 | } | |||
|
170 | ||||
|
171 | if args.save: | |||
|
172 | writer = merge.addOperation(name='HDFWriter') | |||
|
173 | writer.addParameter(name='path', value=path_save, format='str') | |||
|
174 | writer.addParameter(name='Reset', value=True) | |||
|
175 | writer.addParameter(name='setType', value='weather') | |||
|
176 | writer.addParameter(name='description', value=json.dumps(desc)) | |||
|
177 | writer.addParameter(name='blocksPerFile', value='1',format='int') | |||
|
178 | writer.addParameter(name='metadataList', value=','.join(META)) | |||
|
179 | writer.addParameter(name='dataList', value='data_param,utctime') | |||
|
180 | writer.addParameter(name='weather_var', value=param) | |||
|
181 | writer.addParameter(name='mask', value=MASK, format='float') | |||
|
182 | # meta | |||
|
183 | writer.addParameter(name='latitude', value='-12.040436') | |||
|
184 | writer.addParameter(name='longitude', value='-75.295893') | |||
|
185 | writer.addParameter(name='altitude', value='3379.2147') | |||
|
186 | writer.addParameter(name='heading', value='0') | |||
|
187 | writer.addParameter(name='radar_name', value='SOPHy') | |||
|
188 | writer.addParameter(name='institution', value='IGP') | |||
|
189 | writer.addParameter(name='contact', value='dscipion@igp.gob.pe') | |||
|
190 | writer.addParameter(name='created_by', value='Signal Chain (https://pypi.org/project/schainpy/)') | |||
|
191 | writer.addParameter(name='range_unit', value='km') | |||
|
192 | writer.addParameter(name='prf', value=1/ipp) | |||
|
193 | writer.addParameter(name='prf_unit', value='hertz') | |||
|
194 | writer.addParameter(name='variable', value=PARAM[param]['label']) | |||
|
195 | writer.addParameter(name='variable_unit', value=PARAM[param]['cb_label']) | |||
|
196 | writer.addParameter(name='n_pulses', value=n_pulses) | |||
|
197 | writer.addParameter(name='pulse1_range', value=RMIX) | |||
|
198 | writer.addParameter(name='pulse1_width', value=pulse_1_width) | |||
|
199 | writer.addParameter(name='pulse2_width', value=pulse_2_width) | |||
|
200 | writer.addParameter(name='pulse1_repetitions', value=pulse_1_repetitions) | |||
|
201 | writer.addParameter(name='pulse2_repetitions', value=pulse_2_repetitions) | |||
|
202 | writer.addParameter(name='pulse_width_unit', value='microseconds') | |||
|
203 | writer.addParameter(name='snr_threshold', value=MASK) | |||
|
204 | ||||
|
205 | ||||
|
206 | else: #Two pulses | |||
|
207 | n_pulses = 1 | |||
|
208 | pulse_1_width = conf['usrp_tx']['pulse_1'] | |||
|
209 | pulse_1_repetitions = conf['usrp_tx']['repetitions_1'] | |||
|
210 | pulse_2_width = conf['usrp_tx']['pulse_2'] | |||
|
211 | pulse_2_repetitions = conf['usrp_tx']['repetitions_2'] | |||
|
212 | ||||
|
213 | voltage1 = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId()) | |||
|
214 | ||||
|
215 | op = voltage1.addOperation(name='ProfileSelector') | |||
|
216 | op.addParameter(name='profileRangeList', value='0,{}'.format(conf['usrp_tx']['repetitions_1']-1)) | |||
|
217 | ||||
|
218 | if conf['usrp_tx']['code_type_1'] != 'None': | |||
|
219 | codes = [ c.strip() for c in conf['usrp_tx']['code_1'].split(',')] | |||
|
220 | code = [] | |||
|
221 | for c in codes: | |||
|
222 | code.append([int(x) for x in c]) | |||
|
223 | op = voltage1.addOperation(name='Decoder', optype='other') | |||
|
224 | op.addParameter(name='code', value=code) | |||
|
225 | op.addParameter(name='nCode', value=len(code), format='int') | |||
|
226 | op.addParameter(name='nBaud', value=len(code[0]), format='int') | |||
|
227 | else: | |||
|
228 | code = ['0'] | |||
|
229 | ||||
|
230 | op = voltage1.addOperation(name='CohInt', optype='other') #Minimo integrar 2 perfiles por ser codigo complementario | |||
|
231 | op.addParameter(name='n', value=2, format='int') | |||
|
232 | ||||
|
233 | if args.range > 0: | |||
|
234 | op = voltage1.addOperation(name='selectHeights') | |||
|
235 | op.addParameter(name='minIndex', value='0', format='int') | |||
|
236 | op.addParameter(name='maxIndex', value=max_index(RMIX, sample_rate, ipp), format='int') | |||
|
237 | ||||
|
238 | op = voltage1.addOperation(name='setH0') | |||
|
239 | op.addParameter(name='h0', value=H0, format='float') | |||
|
240 | ||||
|
241 | op = voltage1.addOperation(name='PulsePair_vRF', optype='other') | |||
|
242 | op.addParameter(name='n', value=int(conf['usrp_tx']['repetitions_1'])/2, format='int') | |||
|
243 | ||||
|
244 | proc1 = project.addProcUnit(datatype='ParametersProc', inputId=voltage1.getId()) | |||
|
245 | proc1.addParameter(name='runNextUnit', value=True) | |||
|
246 | ||||
|
247 | opObj10 = proc1.addOperation(name="WeatherRadar") | |||
|
248 | opObj10.addParameter(name='tauW',value=(1e-6/sample_rate)*len(code[0])) | |||
|
249 | opObj10.addParameter(name='Pt',value=((1e-6/sample_rate)*len(code[0])/ipp)*200) | |||
|
250 | ||||
|
251 | op = proc1.addOperation(name='PedestalInformation') | |||
|
252 | op.addParameter(name='path', value=path_ped, format='str') | |||
|
253 | op.addParameter(name='interval', value='0.04') | |||
|
254 | op.addParameter(name='time_offset', value=time_offset) | |||
|
255 | op.addParameter(name='mode', value='PPI') | |||
|
256 | ||||
|
257 | op = proc1.addOperation(name='Block360') | |||
|
258 | op.addParameter(name='attr_data', value='data_param') | |||
|
259 | op.addParameter(name='runNextOp', value=True) | |||
|
260 | ||||
|
261 | ||||
|
262 | voltage2 = project.addProcUnit(datatype='VoltageProc', inputId=reader.getId()) | |||
|
263 | ||||
|
264 | op = voltage2.addOperation(name='ProfileSelector') | |||
|
265 | op.addParameter(name='profileRangeList', value='{},{}'.format(conf['usrp_tx']['repetitions_1'], conf['usrp_tx']['repetitions_1']+conf['usrp_tx']['repetitions_2']-1)) | |||
|
266 | ||||
|
267 | if conf['usrp_tx']['code_type_2']: | |||
|
268 | codes = [ c.strip() for c in conf['usrp_tx']['code_2'].split(',')] | |||
|
269 | code = [] | |||
|
270 | for c in codes: | |||
|
271 | code.append([int(x) for x in c]) | |||
|
272 | op = voltage2.addOperation(name='Decoder', optype='other') | |||
|
273 | op.addParameter(name='code', value=code) | |||
|
274 | op.addParameter(name='nCode', value=len(code), format='int') | |||
|
275 | op.addParameter(name='nBaud', value=len(code[0]), format='int') | |||
|
276 | ||||
|
277 | op = voltage2.addOperation(name='CohInt', optype='other') #Minimo integrar 2 perfiles por ser codigo complementario | |||
|
278 | op.addParameter(name='n', value=len(code), format='int') | |||
|
279 | ncode = len(code) | |||
|
280 | else: | |||
|
281 | ncode = 1 | |||
|
282 | ||||
|
283 | if args.range > 0: | |||
|
284 | op = voltage2.addOperation(name='selectHeights') | |||
|
285 | op.addParameter(name='minIndex', value=max_index(RMIX, sample_rate, ipp), format='int') | |||
|
286 | op.addParameter(name='maxIndex', value=max_index(args.range, sample_rate, ipp), format='int') | |||
|
287 | ||||
|
288 | op = voltage2.addOperation(name='setH0') | |||
|
289 | op.addParameter(name='h0', value=H0, format='float') | |||
|
290 | ||||
|
291 | op = voltage2.addOperation(name='PulsePair_vRF', optype='other') | |||
|
292 | op.addParameter(name='n', value=int(conf['usrp_tx']['repetitions_2'])/ncode, format='int') | |||
|
293 | ||||
|
294 | proc2 = project.addProcUnit(datatype='ParametersProc', inputId=voltage2.getId()) | |||
|
295 | proc2.addParameter(name='runNextUnit', value=True) | |||
|
296 | ||||
|
297 | opObj10 = proc2.addOperation(name="WeatherRadar") | |||
|
298 | opObj10.addParameter(name='tauW',value=(1e-6/sample_rate)*len(code[0])) | |||
|
299 | opObj10.addParameter(name='Pt',value=((1e-6/sample_rate)*len(code[0])/ipp)*200) | |||
|
300 | ||||
|
301 | op = proc2.addOperation(name='PedestalInformation') | |||
|
302 | op.addParameter(name='path', value=path_ped, format='str') | |||
|
303 | op.addParameter(name='interval', value='0.04') | |||
|
304 | op.addParameter(name='time_offset', value=time_offset) | |||
|
305 | op.addParameter(name='mode', value='PPI') | |||
|
306 | ||||
|
307 | op = proc2.addOperation(name='Block360') | |||
|
308 | op.addParameter(name='attr_data', value='data_param') | |||
|
309 | op.addParameter(name='runNextOp', value=True) | |||
|
310 | ||||
|
311 | merge = project.addProcUnit(datatype='MergeProc', inputId=[proc1.getId(), proc2.getId()]) | |||
|
312 | merge.addParameter(name='attr_data', value='data_param') | |||
|
313 | merge.addParameter(name='mode', value='7') #RM | |||
|
314 | ||||
|
315 | ||||
|
316 | for param in parameters: | |||
|
317 | ||||
|
318 | if args.plot: | |||
|
319 | op= merge.addOperation(name='WeatherParamsPlot') | |||
|
320 | if args.save: | |||
|
321 | op.addParameter(name='save', value=path_plots, format='str') | |||
|
322 | op.addParameter(name='save_period', value=-1) | |||
|
323 | op.addParameter(name='show', value=args.show) | |||
|
324 | op.addParameter(name='channels', value='0,') | |||
|
325 | op.addParameter(name='zmin', value=PARAM[param]['zmin'], format='int') | |||
|
326 | op.addParameter(name='zmax', value=PARAM[param]['zmax'], format='int') | |||
|
327 | op.addParameter(name='attr_data', value=param, format='str') | |||
|
328 | op.addParameter(name='labels', value=[PARAM[param]['label']]) | |||
|
329 | op.addParameter(name='save_code', value=param) | |||
|
330 | op.addParameter(name='cb_label', value=PARAM[param]['cb_label']) | |||
|
331 | op.addParameter(name='colormap', value=PARAM[param]['colormap']) | |||
|
332 | op.addParameter(name='bgcolor', value='black') | |||
|
333 | if MASK: op.addParameter(name='mask', value=MASK, format='float') | |||
|
334 | if args.server: | |||
|
335 | op.addParameter(name='server', value='0.0.0.0:4444') | |||
|
336 | op.addParameter(name='exp_code', value='400') | |||
|
337 | ||||
|
338 | desc = { | |||
|
339 | 'Data': { | |||
|
340 | 'data_param': {PARAM[param]['wrname']: ['H', 'V']}, | |||
|
341 | 'utctime': 'time' | |||
|
342 | }, | |||
|
343 | 'Metadata': { | |||
|
344 | 'heightList': 'range', | |||
|
345 | 'data_azi': 'azimuth', | |||
|
346 | 'data_ele': 'elevation', | |||
|
347 | 'mode_op': 'scan_type', | |||
|
348 | 'h0': 'range_correction', | |||
|
349 | 'dataPP_NOISE': 'noise', | |||
|
350 | } | |||
|
351 | } | |||
|
352 | ||||
|
353 | if args.save: | |||
|
354 | writer = merge.addOperation(name='HDFWriter') | |||
|
355 | writer.addParameter(name='path', value=path_save, format='str') | |||
|
356 | writer.addParameter(name='Reset', value=True) | |||
|
357 | writer.addParameter(name='setType', value='weather') | |||
|
358 | writer.addParameter(name='description', value=json.dumps(desc)) | |||
|
359 | writer.addParameter(name='blocksPerFile', value='1',format='int') | |||
|
360 | writer.addParameter(name='metadataList', value=','.join(META)) | |||
|
361 | writer.addParameter(name='dataList', value='data_param,utctime') | |||
|
362 | writer.addParameter(name='weather_var', value=param) | |||
|
363 | writer.addParameter(name='mask', value=MASK, format='float') | |||
|
364 | # meta | |||
|
365 | writer.addParameter(name='latitude', value='-12.040436') | |||
|
366 | writer.addParameter(name='longitude', value='-75.295893') | |||
|
367 | writer.addParameter(name='altitude', value='3379.2147') | |||
|
368 | writer.addParameter(name='heading', value='0') | |||
|
369 | writer.addParameter(name='radar_name', value='SOPHy') | |||
|
370 | writer.addParameter(name='institution', value='IGP') | |||
|
371 | writer.addParameter(name='contact', value='dscipion@igp.gob.pe') | |||
|
372 | writer.addParameter(name='created_by', value='Signal Chain (https://pypi.org/project/schainpy/)') | |||
|
373 | writer.addParameter(name='range_unit', value='km') | |||
|
374 | writer.addParameter(name='prf', value=1/ipp) | |||
|
375 | writer.addParameter(name='prf_unit', value='hertz') | |||
|
376 | writer.addParameter(name='variable', value=PARAM[param]['label']) | |||
|
377 | writer.addParameter(name='variable_unit', value=PARAM[param]['cb_label']) | |||
|
378 | writer.addParameter(name='n_pulses', value=n_pulses) | |||
|
379 | writer.addParameter(name='pulse1_range', value=RMIX) | |||
|
380 | writer.addParameter(name='pulse1_width', value=pulse_1_width) | |||
|
381 | writer.addParameter(name='pulse2_width', value=pulse_2_width) | |||
|
382 | writer.addParameter(name='pulse1_repetitions', value=pulse_1_repetitions) | |||
|
383 | writer.addParameter(name='pulse2_repetitions', value=pulse_2_repetitions) | |||
|
384 | writer.addParameter(name='pulse_width_unit', value='microseconds') | |||
|
385 | writer.addParameter(name='snr_threshold', value=MASK) | |||
|
386 | ||||
|
387 | ||||
|
388 | project.start() | |||
|
389 | ||||
|
390 | if __name__ == '__main__': | |||
|
391 | ||||
|
392 | parser = argparse.ArgumentParser(description='Script to process SOPHy data.') | |||
|
393 | parser.add_argument('experiment', | |||
|
394 | help='Experiment name') | |||
|
395 | parser.add_argument('--parameters', nargs='*', default=['S'], | |||
|
396 | help='Variables to process: P, Z, V') | |||
|
397 | parser.add_argument('--time_offset', default=0, | |||
|
398 | help='Fix time offset') | |||
|
399 | parser.add_argument('--range', default=0, type=float, | |||
|
400 | help='Max range to plot') | |||
|
401 | parser.add_argument('--save', action='store_true', | |||
|
402 | help='Create output files') | |||
|
403 | parser.add_argument('--plot', action='store_true', | |||
|
404 | help='Create plot files') | |||
|
405 | parser.add_argument('--show', action='store_true', | |||
|
406 | help='Show matplotlib plot.') | |||
|
407 | parser.add_argument('--online', action='store_true', | |||
|
408 | help='Set online mode.') | |||
|
409 | parser.add_argument('--server', action='store_true', | |||
|
410 | help='Send to realtime') | |||
|
411 | parser.add_argument('--start_time', default='', | |||
|
412 | help='Set start time.') | |||
|
413 | parser.add_argument('--label', default='', | |||
|
414 | help='Label for plot & param folder') | |||
|
415 | ||||
|
416 | args = parser.parse_args() | |||
|
417 | ||||
|
418 | main(args) |
@@ -0,0 +1,117 | |||||
|
1 | import os,sys,json | |||
|
2 | import datetime | |||
|
3 | import time | |||
|
4 | import argparse | |||
|
5 | ||||
|
6 | from schainpy.controller import Project | |||
|
7 | ''' | |||
|
8 | NOTA: | |||
|
9 | Este script de prueba. | |||
|
10 | - Unidad del lectura 'HDFReader'. | |||
|
11 | - Unidad de procesamiento ParametersProc | |||
|
12 | ''' | |||
|
13 | PATH = "/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/paramC0N36.0/2022-06-09T18-00-00/" | |||
|
14 | #PATH = "/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/param/2022-06-09T18-00-00/" | |||
|
15 | ||||
|
16 | #PATH = "/home/soporte/Documents/EVENTO/HYO_PM@2022-06-09T15-05-12/paramC0N36.0/2022-06-09T19-00-00/" | |||
|
17 | #PATH = "/home/soporte/Documents/EVENTO/HYO_PM@2022-05-31T12-00-17/paramC0N36.0/2022-05-31T16-00-00/" | |||
|
18 | path = PATH | |||
|
19 | PARAM = { | |||
|
20 | 'S': {'zmin': -45, 'zmax': -25, 'colormap': 'jet', 'label': 'Power', 'wrname': 'power','cb_label': 'dBm', 'ch':0}, | |||
|
21 | 'SNR': {'zmin': -40, 'zmax': -20, 'colormap': 'jet', 'label': 'SNR', 'wrname': 'snr','cb_label': 'dB', 'ch':0}, | |||
|
22 | 'V': {'zmin': -12, 'zmax': 12, 'colormap': 'sophy_v', 'label': 'Velocity', 'wrname': 'velocity', 'cb_label': 'm/s', 'ch':0}, | |||
|
23 | 'R': {'zmin': 0, 'zmax': 1, 'colormap': 'jet', 'label': 'RhoHV', 'wrname':'rhoHV', 'cb_label': '*', 'ch':0}, | |||
|
24 | 'P': {'zmin': -180,'zmax': 180,'colormap': 'RdBu_r', 'label': 'PhiDP', 'wrname':'phiDP' , 'cb_label': 'ΒΊ', 'ch':0}, | |||
|
25 | 'D': {'zmin': -30, 'zmax': 80, 'colormap': 'sophy_r','label': 'ZDR','wrname':'differential_reflectivity' , 'cb_label': 'dBz','ch':0}, | |||
|
26 | 'Z': {'zmin': -30, 'zmax': 80, 'colormap': 'sophy_r','label': 'Reflectivity ', 'wrname':'reflectivity', 'cb_label': 'dBz','ch':0}, | |||
|
27 | 'W': {'zmin': 0, 'zmax': 15, 'colormap': 'sophy_w','label': 'Spectral Width', 'wrname':'spectral_width', 'cb_label': 'm/s', 'ch':0} | |||
|
28 | } | |||
|
29 | ||||
|
30 | def main(args): | |||
|
31 | #filefmt="******%Y%m%d*%H%M%S*******" | |||
|
32 | #filefmt="SOPHY_20220609_184620_E8.0_Z" | |||
|
33 | parameters = args.parameters | |||
|
34 | grado = args.grado | |||
|
35 | MASK = None | |||
|
36 | ||||
|
37 | for param in parameters: | |||
|
38 | filefmt ="******%Y%m%d*%H%M%S*******" | |||
|
39 | filter= "_E"+str(grado)+".0_"+param | |||
|
40 | variable = 'Data/'+PARAM[param]['wrname']+'/H' | |||
|
41 | desc = { | |||
|
42 | 'Data': { | |||
|
43 | 'data_param': [variable], | |||
|
44 | 'utctime' : 'Data/time' | |||
|
45 | }, | |||
|
46 | 'Metadata': { | |||
|
47 | 'heightList': 'Metadata/range', | |||
|
48 | 'data_azi' : 'Metadata/azimuth', | |||
|
49 | 'data_ele' : 'Metadata/elevation', | |||
|
50 | 'mode_op' : 'Metadata/scan_type', | |||
|
51 | 'h0' : 'Metadata/range_correction', | |||
|
52 | } | |||
|
53 | } | |||
|
54 | ||||
|
55 | project = Project() | |||
|
56 | ||||
|
57 | project.setup(id='10',name='Test Simulator',description=desc) | |||
|
58 | ||||
|
59 | readUnitConfObj = project.addReadUnit(datatype='HDFReader', | |||
|
60 | path=path, | |||
|
61 | startDate="2022/01/01", #"2020/01/01",#today, | |||
|
62 | endDate= "2022/12/01", #"2020/12/30",#today, | |||
|
63 | startTime='00:00:00', | |||
|
64 | endTime='23:59:59', | |||
|
65 | delay=0, | |||
|
66 | #set=0, | |||
|
67 | online=0, | |||
|
68 | walk=0, | |||
|
69 | filefmt=filefmt, | |||
|
70 | filter=filter, | |||
|
71 | dparam= 1, | |||
|
72 | description= json.dumps(desc))#1 | |||
|
73 | ||||
|
74 | proc1 = project.addProcUnit(datatype='ParametersProc',inputId=readUnitConfObj.getId()) | |||
|
75 | ||||
|
76 | if args.plot: | |||
|
77 | print("plotea") | |||
|
78 | op= proc1.addOperation(name='WeatherParamsPlot') | |||
|
79 | if args.save: | |||
|
80 | op.addParameter(name='save', value=path_plots, format='str') | |||
|
81 | op.addParameter(name='save_period', value=-1) | |||
|
82 | op.addParameter(name='show', value=args.show) | |||
|
83 | op.addParameter(name='channels', value='0,') | |||
|
84 | op.addParameter(name='zmin', value=PARAM[param]['zmin'], format='int') | |||
|
85 | op.addParameter(name='zmax', value=PARAM[param]['zmax'], format='int') | |||
|
86 | op.addParameter(name='attr_data', value=param, format='str') | |||
|
87 | op.addParameter(name='labels', value=[PARAM[param]['label']]) | |||
|
88 | op.addParameter(name='save_code', value=param) | |||
|
89 | op.addParameter(name='cb_label', value=PARAM[param]['cb_label']) | |||
|
90 | op.addParameter(name='colormap', value=PARAM[param]['colormap']) | |||
|
91 | op.addParameter(name='bgcolor', value='black') | |||
|
92 | if MASK: op.addParameter(name='mask', value=MASK, format='float') | |||
|
93 | if args.server: | |||
|
94 | op.addParameter(name='server', value='0.0.0.0:4444') | |||
|
95 | op.addParameter(name='exp_code', value='400') | |||
|
96 | project.start() | |||
|
97 | ||||
|
98 | if __name__ == '__main__': | |||
|
99 | ||||
|
100 | parser = argparse.ArgumentParser(description='Script to process SOPHy data.') | |||
|
101 | parser.add_argument('--parameters', nargs='*', default=['S'], | |||
|
102 | help='Variables to process: P, Z, V ,W') | |||
|
103 | parser.add_argument('--grado', default=2, | |||
|
104 | help='Angle in Elev to plot') | |||
|
105 | parser.add_argument('--save', default=0, | |||
|
106 | help='Save plot') | |||
|
107 | parser.add_argument('--range', default=0, type=float, | |||
|
108 | help='Max range to plot') | |||
|
109 | parser.add_argument('--plot', action='store_true', | |||
|
110 | help='Create plot files') | |||
|
111 | parser.add_argument('--show', action='store_true', | |||
|
112 | help='Show matplotlib plot.') | |||
|
113 | parser.add_argument('--server', action='store_true', | |||
|
114 | help='Send to realtime') | |||
|
115 | args = parser.parse_args() | |||
|
116 | ||||
|
117 | main(args) |
General Comments 0
You need to be logged in to leave comments.
Login now