##// END OF EJS Templates
Modificación a kmamisr para ejecutarse en la versión 3, creación de scripts con terminación v3 para difereciarlos, se comentó la linea #720 de JroIO_param.py debido a que reiniciaba la lista de archivos, ocasionando la reescritura del archivo hdf5. Alguna otra modificación aparente es producto de algunas variaciones en espacios al usar la función print()
joabAM -
r1279:c53fe2a4a291
parent child
Show More

The requested changes are too big and content was truncated. Show full diff

@@ -1,1290 +1,1295
1 '''
1 '''
2 Updated on January , 2018, for multiprocessing purposes
2 Updated on January , 2018, for multiprocessing purposes
3 Author: Sergio Cortez
3 Author: Sergio Cortez
4 Created on September , 2012
4 Created on September , 2012
5 '''
5 '''
6 from platform import python_version
6 from platform import python_version
7 import sys
7 import sys
8 import ast
8 import ast
9 import datetime
9 import datetime
10 import traceback
10 import traceback
11 import math
11 import math
12 import time
12 import time
13 import zmq
13 import zmq
14 from multiprocessing import Process, Queue, Event, Value, cpu_count
14 from multiprocessing import Process, Queue, Event, Value, cpu_count
15 from threading import Thread
15 from threading import Thread
16 from xml.etree.ElementTree import ElementTree, Element, SubElement, tostring
16 from xml.etree.ElementTree import ElementTree, Element, SubElement, tostring
17 from xml.dom import minidom
17 from xml.dom import minidom
18
18
19
19
20 from schainpy.admin import Alarm, SchainWarning
20 from schainpy.admin import Alarm, SchainWarning
21 from schainpy.model import *
21 from schainpy.model import *
22 from schainpy.utils import log
22 from schainpy.utils import log
23
23
24
24
25 DTYPES = {
25 DTYPES = {
26 'Voltage': '.r',
26 'Voltage': '.r',
27 'Spectra': '.pdata'
27 'Spectra': '.pdata'
28 }
28 }
29
29
30
30
31 def MPProject(project, n=cpu_count()):
31 def MPProject(project, n=cpu_count()):
32 '''
32 '''
33 Project wrapper to run schain in n processes
33 Project wrapper to run schain in n processes
34 '''
34 '''
35
35
36 rconf = project.getReadUnitObj()
36 rconf = project.getReadUnitObj()
37 op = rconf.getOperationObj('run')
37 op = rconf.getOperationObj('run')
38 dt1 = op.getParameterValue('startDate')
38 dt1 = op.getParameterValue('startDate')
39 dt2 = op.getParameterValue('endDate')
39 dt2 = op.getParameterValue('endDate')
40 tm1 = op.getParameterValue('startTime')
40 tm1 = op.getParameterValue('startTime')
41 tm2 = op.getParameterValue('endTime')
41 tm2 = op.getParameterValue('endTime')
42 days = (dt2 - dt1).days
42 days = (dt2 - dt1).days
43
43
44 for day in range(days + 1):
44 for day in range(days + 1):
45 skip = 0
45 skip = 0
46 cursor = 0
46 cursor = 0
47 processes = []
47 processes = []
48 dt = dt1 + datetime.timedelta(day)
48 dt = dt1 + datetime.timedelta(day)
49 dt_str = dt.strftime('%Y/%m/%d')
49 dt_str = dt.strftime('%Y/%m/%d')
50 reader = JRODataReader()
50 reader = JRODataReader()
51 paths, files = reader.searchFilesOffLine(path=rconf.path,
51 paths, files = reader.searchFilesOffLine(path=rconf.path,
52 startDate=dt,
52 startDate=dt,
53 endDate=dt,
53 endDate=dt,
54 startTime=tm1,
54 startTime=tm1,
55 endTime=tm2,
55 endTime=tm2,
56 ext=DTYPES[rconf.datatype])
56 ext=DTYPES[rconf.datatype])
57 nFiles = len(files)
57 nFiles = len(files)
58 if nFiles == 0:
58 if nFiles == 0:
59 continue
59 continue
60 skip = int(math.ceil(nFiles / n))
60 skip = int(math.ceil(nFiles / n))
61 while nFiles > cursor * skip:
61 while nFiles > cursor * skip:
62 rconf.update(startDate=dt_str, endDate=dt_str, cursor=cursor,
62 rconf.update(startDate=dt_str, endDate=dt_str, cursor=cursor,
63 skip=skip)
63 skip=skip)
64 p = project.clone()
64 p = project.clone()
65 p.start()
65 p.start()
66 processes.append(p)
66 processes.append(p)
67 cursor += 1
67 cursor += 1
68
68
69 def beforeExit(exctype, value, trace):
69 def beforeExit(exctype, value, trace):
70 for process in processes:
70 for process in processes:
71 process.terminate()
71 process.terminate()
72 process.join()
72 process.join()
73 print(traceback.print_tb(trace))
73 print(traceback.print_tb(trace))
74
74
75 sys.excepthook = beforeExit
75 sys.excepthook = beforeExit
76
76
77 for process in processes:
77 for process in processes:
78 process.join()
78 process.join()
79 process.terminate()
79 process.terminate()
80
80
81 time.sleep(3)
81 time.sleep(3)
82
82
83 def wait(context):
83 def wait(context):
84
84
85 time.sleep(1)
85 time.sleep(1)
86 c = zmq.Context()
86 c = zmq.Context()
87 receiver = c.socket(zmq.SUB)
87 receiver = c.socket(zmq.SUB)
88 receiver.connect('ipc:///tmp/schain_{}_pub'.format(self.id))
88 receiver.connect('ipc:///tmp/schain_{}_pub'.format(self.id))
89 receiver.setsockopt(zmq.SUBSCRIBE, self.id.encode())
89 receiver.setsockopt(zmq.SUBSCRIBE, self.id.encode())
90 msg = receiver.recv_multipart()[1]
90 msg = receiver.recv_multipart()[1]
91 context.terminate()
91 context.terminate()
92
92
93 class ParameterConf():
93 class ParameterConf():
94
94
95 id = None
95 id = None
96 name = None
96 name = None
97 value = None
97 value = None
98 format = None
98 format = None
99
99
100 __formated_value = None
100 __formated_value = None
101
101
102 ELEMENTNAME = 'Parameter'
102 ELEMENTNAME = 'Parameter'
103
103
104 def __init__(self):
104 def __init__(self):
105
105
106 self.format = 'str'
106 self.format = 'str'
107
107
108 def getElementName(self):
108 def getElementName(self):
109
109
110 return self.ELEMENTNAME
110 return self.ELEMENTNAME
111
111
112 def getValue(self):
112 def getValue(self):
113
113
114 value = self.value
114 value = self.value
115 format = self.format
115 format = self.format
116
116
117 if self.__formated_value != None:
117 if self.__formated_value != None:
118
118
119 return self.__formated_value
119 return self.__formated_value
120
120
121 if format == 'obj':
121 if format == 'obj':
122 return value
122 return value
123
123
124 if format == 'str':
124 if format == 'str':
125 self.__formated_value = str(value)
125 self.__formated_value = str(value)
126 return self.__formated_value
126 return self.__formated_value
127
127
128 if value == '':
128 if value == '':
129 raise ValueError('%s: This parameter value is empty' % self.name)
129 raise ValueError('%s: This parameter value is empty' % self.name)
130
130
131 if format == 'list':
131 if format == 'list':
132 strList = [s.strip() for s in value.split(',')]
132 strList = [s.strip() for s in value.split(',')]
133 self.__formated_value = strList
133 self.__formated_value = strList
134
134
135 return self.__formated_value
135 return self.__formated_value
136
136
137 if format == 'intlist':
137 if format == 'intlist':
138 '''
138 '''
139 Example:
139 Example:
140 value = (0,1,2)
140 value = (0,1,2)
141 '''
141 '''
142
142
143 new_value = ast.literal_eval(value)
143 new_value = ast.literal_eval(value)
144
144
145 if type(new_value) not in (tuple, list):
145 if type(new_value) not in (tuple, list):
146 new_value = [int(new_value)]
146 new_value = [int(new_value)]
147
147
148 self.__formated_value = new_value
148 self.__formated_value = new_value
149
149
150 return self.__formated_value
150 return self.__formated_value
151
151
152 if format == 'floatlist':
152 if format == 'floatlist':
153 '''
153 '''
154 Example:
154 Example:
155 value = (0.5, 1.4, 2.7)
155 value = (0.5, 1.4, 2.7)
156 '''
156 '''
157
157
158 new_value = ast.literal_eval(value)
158 new_value = ast.literal_eval(value)
159
159
160 if type(new_value) not in (tuple, list):
160 if type(new_value) not in (tuple, list):
161 new_value = [float(new_value)]
161 new_value = [float(new_value)]
162
162
163 self.__formated_value = new_value
163 self.__formated_value = new_value
164
164
165 return self.__formated_value
165 return self.__formated_value
166
166
167 if format == 'date':
167 if format == 'date':
168 strList = value.split('/')
168 strList = value.split('/')
169 intList = [int(x) for x in strList]
169 intList = [int(x) for x in strList]
170 date = datetime.date(intList[0], intList[1], intList[2])
170 date = datetime.date(intList[0], intList[1], intList[2])
171
171
172 self.__formated_value = date
172 self.__formated_value = date
173
173
174 return self.__formated_value
174 return self.__formated_value
175
175
176 if format == 'time':
176 if format == 'time':
177 strList = value.split(':')
177 strList = value.split(':')
178 intList = [int(x) for x in strList]
178 intList = [int(x) for x in strList]
179 time = datetime.time(intList[0], intList[1], intList[2])
179 time = datetime.time(intList[0], intList[1], intList[2])
180
180
181 self.__formated_value = time
181 self.__formated_value = time
182
182
183 return self.__formated_value
183 return self.__formated_value
184
184
185 if format == 'pairslist':
185 if format == 'pairslist':
186 '''
186 '''
187 Example:
187 Example:
188 value = (0,1),(1,2)
188 value = (0,1),(1,2)
189 '''
189 '''
190
190
191 new_value = ast.literal_eval(value)
191 new_value = ast.literal_eval(value)
192
192
193 if type(new_value) not in (tuple, list):
193 if type(new_value) not in (tuple, list):
194 raise ValueError('%s has to be a tuple or list of pairs' % value)
194 raise ValueError('%s has to be a tuple or list of pairs' % value)
195
195
196 if type(new_value[0]) not in (tuple, list):
196 if type(new_value[0]) not in (tuple, list):
197 if len(new_value) != 2:
197 if len(new_value) != 2:
198 raise ValueError('%s has to be a tuple or list of pairs' % value)
198 raise ValueError('%s has to be a tuple or list of pairs' % value)
199 new_value = [new_value]
199 new_value = [new_value]
200
200
201 for thisPair in new_value:
201 for thisPair in new_value:
202 if len(thisPair) != 2:
202 if len(thisPair) != 2:
203 raise ValueError('%s has to be a tuple or list of pairs' % value)
203 raise ValueError('%s has to be a tuple or list of pairs' % value)
204
204
205 self.__formated_value = new_value
205 self.__formated_value = new_value
206
206
207 return self.__formated_value
207 return self.__formated_value
208
208
209 if format == 'multilist':
209 if format == 'multilist':
210 '''
210 '''
211 Example:
211 Example:
212 value = (0,1,2),(3,4,5)
212 value = (0,1,2),(3,4,5)
213 '''
213 '''
214 multiList = ast.literal_eval(value)
214 multiList = ast.literal_eval(value)
215
215
216 if type(multiList[0]) == int:
216 if type(multiList[0]) == int:
217 multiList = ast.literal_eval('(' + value + ')')
217 multiList = ast.literal_eval('(' + value + ')')
218
218
219 self.__formated_value = multiList
219 self.__formated_value = multiList
220
220
221 return self.__formated_value
221 return self.__formated_value
222
222
223 if format == 'bool':
223 if format == 'bool':
224 value = int(value)
224 value = int(value)
225
225
226 if format == 'int':
226 if format == 'int':
227 value = float(value)
227 value = float(value)
228
228
229 format_func = eval(format)
229 format_func = eval(format)
230
230
231 self.__formated_value = format_func(value)
231 self.__formated_value = format_func(value)
232
232
233 return self.__formated_value
233 return self.__formated_value
234
234
235 def updateId(self, new_id):
235 def updateId(self, new_id):
236
236
237 self.id = str(new_id)
237 self.id = str(new_id)
238
238
239 def setup(self, id, name, value, format='str'):
239 def setup(self, id, name, value, format='str'):
240 self.id = str(id)
240 self.id = str(id)
241 self.name = name
241 self.name = name
242 if format == 'obj':
242 if format == 'obj':
243 self.value = value
243 self.value = value
244 else:
244 else:
245 self.value = str(value)
245 self.value = str(value)
246 self.format = str.lower(format)
246 self.format = str.lower(format)
247
247
248 self.getValue()
248 self.getValue()
249
249
250 return 1
250 return 1
251
251
252 def update(self, name, value, format='str'):
252 def update(self, name, value, format='str'):
253
253
254 self.name = name
254 self.name = name
255 self.value = str(value)
255 self.value = str(value)
256 self.format = format
256 self.format = format
257
257
258 def makeXml(self, opElement):
258 def makeXml(self, opElement):
259 if self.name not in ('queue',):
259 if self.name not in ('queue',):
260 parmElement = SubElement(opElement, self.ELEMENTNAME)
260 parmElement = SubElement(opElement, self.ELEMENTNAME)
261 parmElement.set('id', str(self.id))
261 parmElement.set('id', str(self.id))
262 parmElement.set('name', self.name)
262 parmElement.set('name', self.name)
263 parmElement.set('value', self.value)
263 parmElement.set('value', self.value)
264 parmElement.set('format', self.format)
264 parmElement.set('format', self.format)
265
265
266 def readXml(self, parmElement):
266 def readXml(self, parmElement):
267
267
268 self.id = parmElement.get('id')
268 self.id = parmElement.get('id')
269 self.name = parmElement.get('name')
269 self.name = parmElement.get('name')
270 self.value = parmElement.get('value')
270 self.value = parmElement.get('value')
271 self.format = str.lower(parmElement.get('format'))
271 self.format = str.lower(parmElement.get('format'))
272
272
273 # Compatible with old signal chain version
273 # Compatible with old signal chain version
274 if self.format == 'int' and self.name == 'idfigure':
274 if self.format == 'int' and self.name == 'idfigure':
275 self.name = 'id'
275 self.name = 'id'
276
276
277 def printattr(self):
277 def printattr(self):
278
278
279 print('Parameter[%s]: name = %s, value = %s, format = %s, project_id = %s' % (self.id, self.name, self.value, self.format, self.project_id))
279 print('Parameter[%s]: name = %s, value = %s, format = %s, project_id = %s' % (self.id, self.name, self.value, self.format, self.project_id))
280
280
281 class OperationConf():
281 class OperationConf():
282
282
283 ELEMENTNAME = 'Operation'
283 ELEMENTNAME = 'Operation'
284
284
285 def __init__(self):
285 def __init__(self):
286
286
287 self.id = '0'
287 self.id = '0'
288 self.name = None
288 self.name = None
289 self.priority = None
289 self.priority = None
290 self.topic = None
290 self.topic = None
291
291
292 def __getNewId(self):
292 def __getNewId(self):
293
293
294 return int(self.id) * 10 + len(self.parmConfObjList) + 1
294 return int(self.id) * 10 + len(self.parmConfObjList) + 1
295
295
296 def getId(self):
296 def getId(self):
297 return self.id
297 return self.id
298
298
299 def updateId(self, new_id):
299 def updateId(self, new_id):
300
300
301 self.id = str(new_id)
301 self.id = str(new_id)
302
302
303 n = 1
303 n = 1
304 for parmObj in self.parmConfObjList:
304 for parmObj in self.parmConfObjList:
305
305
306 idParm = str(int(new_id) * 10 + n)
306 idParm = str(int(new_id) * 10 + n)
307 parmObj.updateId(idParm)
307 parmObj.updateId(idParm)
308
308
309 n += 1
309 n += 1
310
310
311 def getElementName(self):
311 def getElementName(self):
312
312
313 return self.ELEMENTNAME
313 return self.ELEMENTNAME
314
314
315 def getParameterObjList(self):
315 def getParameterObjList(self):
316
316
317 return self.parmConfObjList
317 return self.parmConfObjList
318
318
319 def getParameterObj(self, parameterName):
319 def getParameterObj(self, parameterName):
320
320
321 for parmConfObj in self.parmConfObjList:
321 for parmConfObj in self.parmConfObjList:
322
322
323 if parmConfObj.name != parameterName:
323 if parmConfObj.name != parameterName:
324 continue
324 continue
325
325
326 return parmConfObj
326 return parmConfObj
327
327
328 return None
328 return None
329
329
330 def getParameterObjfromValue(self, parameterValue):
330 def getParameterObjfromValue(self, parameterValue):
331
331
332 for parmConfObj in self.parmConfObjList:
332 for parmConfObj in self.parmConfObjList:
333
333
334 if parmConfObj.getValue() != parameterValue:
334 if parmConfObj.getValue() != parameterValue:
335 continue
335 continue
336
336
337 return parmConfObj.getValue()
337 return parmConfObj.getValue()
338
338
339 return None
339 return None
340
340
341 def getParameterValue(self, parameterName):
341 def getParameterValue(self, parameterName):
342
342
343 parameterObj = self.getParameterObj(parameterName)
343 parameterObj = self.getParameterObj(parameterName)
344
344
345 # if not parameterObj:
345 # if not parameterObj:
346 # return None
346 # return None
347
347
348 value = parameterObj.getValue()
348 value = parameterObj.getValue()
349
349
350 return value
350 return value
351
351
352 def getKwargs(self):
352 def getKwargs(self):
353
353
354 kwargs = {}
354 kwargs = {}
355
355
356 for parmConfObj in self.parmConfObjList:
356 for parmConfObj in self.parmConfObjList:
357 if self.name == 'run' and parmConfObj.name == 'datatype':
357 if self.name == 'run' and parmConfObj.name == 'datatype':
358 continue
358 continue
359
359
360 kwargs[parmConfObj.name] = parmConfObj.getValue()
360 kwargs[parmConfObj.name] = parmConfObj.getValue()
361
361
362 return kwargs
362 return kwargs
363
363
364 def setup(self, id, name, priority, type, project_id, err_queue, lock):
364 def setup(self, id, name, priority, type, project_id, err_queue, lock):
365
365
366 self.id = str(id)
366 self.id = str(id)
367 self.project_id = project_id
367 self.project_id = project_id
368 self.name = name
368 self.name = name
369 self.type = type
369 self.type = type
370 self.priority = priority
370 self.priority = priority
371 self.err_queue = err_queue
371 self.err_queue = err_queue
372 self.lock = lock
372 self.lock = lock
373 self.parmConfObjList = []
373 self.parmConfObjList = []
374
374
375 def removeParameters(self):
375 def removeParameters(self):
376
376
377 for obj in self.parmConfObjList:
377 for obj in self.parmConfObjList:
378 del obj
378 del obj
379
379
380 self.parmConfObjList = []
380 self.parmConfObjList = []
381
381
382 def addParameter(self, name, value, format='str'):
382 def addParameter(self, name, value, format='str'):
383
383
384 if value is None:
384 if value is None:
385 return None
385 return None
386 id = self.__getNewId()
386 id = self.__getNewId()
387
387
388 parmConfObj = ParameterConf()
388 parmConfObj = ParameterConf()
389 if not parmConfObj.setup(id, name, value, format):
389 if not parmConfObj.setup(id, name, value, format):
390 return None
390 return None
391
391
392 self.parmConfObjList.append(parmConfObj)
392 self.parmConfObjList.append(parmConfObj)
393
393
394 return parmConfObj
394 return parmConfObj
395
395
396 def changeParameter(self, name, value, format='str'):
396 def changeParameter(self, name, value, format='str'):
397
397
398 parmConfObj = self.getParameterObj(name)
398 parmConfObj = self.getParameterObj(name)
399 parmConfObj.update(name, value, format)
399 parmConfObj.update(name, value, format)
400
400
401 return parmConfObj
401 return parmConfObj
402
402
403 def makeXml(self, procUnitElement):
403 def makeXml(self, procUnitElement):
404
404
405 opElement = SubElement(procUnitElement, self.ELEMENTNAME)
405 opElement = SubElement(procUnitElement, self.ELEMENTNAME)
406 opElement.set('id', str(self.id))
406 opElement.set('id', str(self.id))
407 opElement.set('name', self.name)
407 opElement.set('name', self.name)
408 opElement.set('type', self.type)
408 opElement.set('type', self.type)
409 opElement.set('priority', str(self.priority))
409 opElement.set('priority', str(self.priority))
410
410
411 for parmConfObj in self.parmConfObjList:
411 for parmConfObj in self.parmConfObjList:
412 parmConfObj.makeXml(opElement)
412 parmConfObj.makeXml(opElement)
413
413
414 def readXml(self, opElement, project_id):
414 def readXml(self, opElement, project_id):
415
415
416 self.id = opElement.get('id')
416 self.id = opElement.get('id')
417 self.name = opElement.get('name')
417 self.name = opElement.get('name')
418 self.type = opElement.get('type')
418 self.type = opElement.get('type')
419 self.priority = opElement.get('priority')
419 self.priority = opElement.get('priority')
420 self.project_id = str(project_id)
420 self.project_id = str(project_id)
421
421
422 # Compatible with old signal chain version
422 # Compatible with old signal chain version
423 # Use of 'run' method instead 'init'
423 # Use of 'run' method instead 'init'
424 if self.type == 'self' and self.name == 'init':
424 if self.type == 'self' and self.name == 'init':
425 self.name = 'run'
425 self.name = 'run'
426
426
427 self.parmConfObjList = []
427 self.parmConfObjList = []
428
428
429 parmElementList = opElement.iter(ParameterConf().getElementName())
429 parmElementList = opElement.iter(ParameterConf().getElementName())
430
430
431 for parmElement in parmElementList:
431 for parmElement in parmElementList:
432 parmConfObj = ParameterConf()
432 parmConfObj = ParameterConf()
433 parmConfObj.readXml(parmElement)
433 parmConfObj.readXml(parmElement)
434
434
435 # Compatible with old signal chain version
435 # Compatible with old signal chain version
436 # If an 'plot' OPERATION is found, changes name operation by the value of its type PARAMETER
436 # If an 'plot' OPERATION is found, changes name operation by the value of its type PARAMETER
437 if self.type != 'self' and self.name == 'Plot':
437 if self.type != 'self' and self.name == 'Plot':
438 if parmConfObj.format == 'str' and parmConfObj.name == 'type':
438 if parmConfObj.format == 'str' and parmConfObj.name == 'type':
439 self.name = parmConfObj.value
439 self.name = parmConfObj.value
440 continue
440 continue
441
441
442 self.parmConfObjList.append(parmConfObj)
442 self.parmConfObjList.append(parmConfObj)
443
443
444 def printattr(self):
444 def printattr(self):
445
445
446 print('%s[%s]: name = %s, type = %s, priority = %s, project_id = %s' % (self.ELEMENTNAME,
446 print('%s[%s]: name = %s, type = %s, priority = %s, project_id = %s' % (self.ELEMENTNAME,
447 self.id,
447 self.id,
448 self.name,
448 self.name,
449 self.type,
449 self.type,
450 self.priority,
450 self.priority,
451 self.project_id))
451 self.project_id))
452
452
453 for parmConfObj in self.parmConfObjList:
453 for parmConfObj in self.parmConfObjList:
454 parmConfObj.printattr()
454 parmConfObj.printattr()
455
455
456 def createObject(self):
456 def createObject(self):
457
457
458 className = eval(self.name)
458 className = eval(self.name)
459
459
460 if self.type == 'other':
460 if self.type == 'other':
461 opObj = className()
461 opObj = className()
462 elif self.type == 'external':
462 elif self.type == 'external':
463 kwargs = self.getKwargs()
463 kwargs = self.getKwargs()
464 opObj = className(self.id, self.id, self.project_id, self.err_queue, self.lock, 'Operation', **kwargs)
464 opObj = className(self.id, self.id, self.project_id, self.err_queue, self.lock, 'Operation', **kwargs)
465 opObj.start()
465 opObj.start()
466 self.opObj = opObj
466 self.opObj = opObj
467
467
468 return opObj
468 return opObj
469
469
470 class ProcUnitConf():
470 class ProcUnitConf():
471
471
472 ELEMENTNAME = 'ProcUnit'
472 ELEMENTNAME = 'ProcUnit'
473
473
474 def __init__(self):
474 def __init__(self):
475
475
476 self.id = None
476 self.id = None
477 self.datatype = None
477 self.datatype = None
478 self.name = None
478 self.name = None
479 self.inputId = None
479 self.inputId = None
480 self.opConfObjList = []
480 self.opConfObjList = []
481 self.procUnitObj = None
481 self.procUnitObj = None
482 self.opObjDict = {}
482 self.opObjDict = {}
483
483
484 def __getPriority(self):
484 def __getPriority(self):
485
485
486 return len(self.opConfObjList) + 1
486 return len(self.opConfObjList) + 1
487
487
488 def __getNewId(self):
488 def __getNewId(self):
489
489
490 return int(self.id) * 10 + len(self.opConfObjList) + 1
490 return int(self.id) * 10 + len(self.opConfObjList) + 1
491
491
492 def getElementName(self):
492 def getElementName(self):
493
493
494 return self.ELEMENTNAME
494 return self.ELEMENTNAME
495
495
496 def getId(self):
496 def getId(self):
497
497
498 return self.id
498 return self.id
499
499
500 def updateId(self, new_id):
500 def updateId(self, new_id):
501 '''
501 '''
502 new_id = int(parentId) * 10 + (int(self.id) % 10)
502 new_id = int(parentId) * 10 + (int(self.id) % 10)
503 new_inputId = int(parentId) * 10 + (int(self.inputId) % 10)
503 new_inputId = int(parentId) * 10 + (int(self.inputId) % 10)
504
504
505 # If this proc unit has not inputs
505 # If this proc unit has not inputs
506 #if self.inputId == '0':
506 #if self.inputId == '0':
507 #new_inputId = 0
507 #new_inputId = 0
508
508
509 n = 1
509 n = 1
510 for opConfObj in self.opConfObjList:
510 for opConfObj in self.opConfObjList:
511
511
512 idOp = str(int(new_id) * 10 + n)
512 idOp = str(int(new_id) * 10 + n)
513 opConfObj.updateId(idOp)
513 opConfObj.updateId(idOp)
514
514
515 n += 1
515 n += 1
516
516
517 self.parentId = str(parentId)
517 self.parentId = str(parentId)
518 self.id = str(new_id)
518 self.id = str(new_id)
519 #self.inputId = str(new_inputId)
519 #self.inputId = str(new_inputId)
520 '''
520 '''
521 n = 1
521 n = 1
522
522
523 def getInputId(self):
523 def getInputId(self):
524
524
525 return self.inputId
525 return self.inputId
526
526
527 def getOperationObjList(self):
527 def getOperationObjList(self):
528
528
529 return self.opConfObjList
529 return self.opConfObjList
530
530
531 def getOperationObj(self, name=None):
531 def getOperationObj(self, name=None):
532
532
533 for opConfObj in self.opConfObjList:
533 for opConfObj in self.opConfObjList:
534
534
535 if opConfObj.name != name:
535 if opConfObj.name != name:
536 continue
536 continue
537
537
538 return opConfObj
538 return opConfObj
539
539
540 return None
540 return None
541
541
542 def getOpObjfromParamValue(self, value=None):
542 def getOpObjfromParamValue(self, value=None):
543
543
544 for opConfObj in self.opConfObjList:
544 for opConfObj in self.opConfObjList:
545 if opConfObj.getParameterObjfromValue(parameterValue=value) != value:
545 if opConfObj.getParameterObjfromValue(parameterValue=value) != value:
546 continue
546 continue
547 return opConfObj
547 return opConfObj
548 return None
548 return None
549
549
550 def getProcUnitObj(self):
550 def getProcUnitObj(self):
551
551
552 return self.procUnitObj
552 return self.procUnitObj
553
553
554 def setup(self, project_id, id, name, datatype, inputId, err_queue, lock):
554 def setup(self, project_id, id, name, datatype, inputId, err_queue, lock):
555 '''
555 '''
556 id sera el topico a publicar
556 id sera el topico a publicar
557 inputId sera el topico a subscribirse
557 inputId sera el topico a subscribirse
558 '''
558 '''
559
559
560 # Compatible with old signal chain version
560 # Compatible with old signal chain version
561 if datatype == None and name == None:
561 if datatype == None and name == None:
562 raise ValueError('datatype or name should be defined')
562 raise ValueError('datatype or name should be defined')
563
563
564 #Definir una condicion para inputId cuando sea 0
564 #Definir una condicion para inputId cuando sea 0
565
565
566 if name == None:
566 if name == None:
567 if 'Proc' in datatype:
567 if 'Proc' in datatype:
568 name = datatype
568 name = datatype
569 else:
569 else:
570 name = '%sProc' % (datatype)
570 name = '%sProc' % (datatype)
571
571
572 if datatype == None:
572 if datatype == None:
573 datatype = name.replace('Proc', '')
573 datatype = name.replace('Proc', '')
574
574
575 self.id = str(id)
575 self.id = str(id)
576 self.project_id = project_id
576 self.project_id = project_id
577 self.name = name
577 self.name = name
578 self.datatype = datatype
578 self.datatype = datatype
579 self.inputId = inputId
579 self.inputId = inputId
580 self.err_queue = err_queue
580 self.err_queue = err_queue
581 self.lock = lock
581 self.lock = lock
582 self.opConfObjList = []
582 self.opConfObjList = []
583
583
584 self.addOperation(name='run', optype='self')
584 self.addOperation(name='run', optype='self')
585
585
586 def removeOperations(self):
586 def removeOperations(self):
587
587
588 for obj in self.opConfObjList:
588 for obj in self.opConfObjList:
589 del obj
589 del obj
590
590
591 self.opConfObjList = []
591 self.opConfObjList = []
592 self.addOperation(name='run')
592 self.addOperation(name='run')
593
593
594 def addParameter(self, **kwargs):
594 def addParameter(self, **kwargs):
595 '''
595 '''
596 Add parameters to 'run' operation
596 Add parameters to 'run' operation
597 '''
597 '''
598 opObj = self.opConfObjList[0]
598 opObj = self.opConfObjList[0]
599
599
600 opObj.addParameter(**kwargs)
600 opObj.addParameter(**kwargs)
601
601
602 return opObj
602 return opObj
603
603
604 def addOperation(self, name, optype='self'):
604 def addOperation(self, name, optype='self'):
605 '''
605 '''
606 Actualizacion - > proceso comunicacion
606 Actualizacion - > proceso comunicacion
607 En el caso de optype='self', elminar. DEfinir comuncacion IPC -> Topic
607 En el caso de optype='self', elminar. DEfinir comuncacion IPC -> Topic
608 definir el tipoc de socket o comunicacion ipc++
608 definir el tipoc de socket o comunicacion ipc++
609
609
610 '''
610 '''
611
611
612 id = self.__getNewId()
612 id = self.__getNewId()
613 priority = self.__getPriority() # Sin mucho sentido, pero puede usarse
613 priority = self.__getPriority() # Sin mucho sentido, pero puede usarse
614 opConfObj = OperationConf()
614 opConfObj = OperationConf()
615 opConfObj.setup(id, name=name, priority=priority, type=optype, project_id=self.project_id, err_queue=self.err_queue, lock=self.lock)
615 opConfObj.setup(id, name=name, priority=priority, type=optype, project_id=self.project_id, err_queue=self.err_queue, lock=self.lock)
616 self.opConfObjList.append(opConfObj)
616 self.opConfObjList.append(opConfObj)
617
617
618 return opConfObj
618 return opConfObj
619
619
620 def makeXml(self, projectElement):
620 def makeXml(self, projectElement):
621
621
622 procUnitElement = SubElement(projectElement, self.ELEMENTNAME)
622 procUnitElement = SubElement(projectElement, self.ELEMENTNAME)
623 procUnitElement.set('id', str(self.id))
623 procUnitElement.set('id', str(self.id))
624 procUnitElement.set('name', self.name)
624 procUnitElement.set('name', self.name)
625 procUnitElement.set('datatype', self.datatype)
625 procUnitElement.set('datatype', self.datatype)
626 procUnitElement.set('inputId', str(self.inputId))
626 procUnitElement.set('inputId', str(self.inputId))
627
627
628 for opConfObj in self.opConfObjList:
628 for opConfObj in self.opConfObjList:
629 opConfObj.makeXml(procUnitElement)
629 opConfObj.makeXml(procUnitElement)
630
630
631 def readXml(self, upElement, project_id):
631 def readXml(self, upElement, project_id):
632
632
633 self.id = upElement.get('id')
633 self.id = upElement.get('id')
634 self.name = upElement.get('name')
634 self.name = upElement.get('name')
635 self.datatype = upElement.get('datatype')
635 self.datatype = upElement.get('datatype')
636 self.inputId = upElement.get('inputId')
636 self.inputId = upElement.get('inputId')
637 self.project_id = str(project_id)
637 self.project_id = str(project_id)
638
638
639 if self.ELEMENTNAME == 'ReadUnit':
639 if self.ELEMENTNAME == 'ReadUnit':
640 self.datatype = self.datatype.replace('Reader', '')
640 self.datatype = self.datatype.replace('Reader', '')
641
641
642 if self.ELEMENTNAME == 'ProcUnit':
642 if self.ELEMENTNAME == 'ProcUnit':
643 self.datatype = self.datatype.replace('Proc', '')
643 self.datatype = self.datatype.replace('Proc', '')
644
644
645 if self.inputId == 'None':
645 if self.inputId == 'None':
646 self.inputId = '0'
646 self.inputId = '0'
647
647
648 self.opConfObjList = []
648 self.opConfObjList = []
649
649
650 opElementList = upElement.iter(OperationConf().getElementName())
650 opElementList = upElement.iter(OperationConf().getElementName())
651
651
652 for opElement in opElementList:
652 for opElement in opElementList:
653 opConfObj = OperationConf()
653 opConfObj = OperationConf()
654 opConfObj.readXml(opElement, project_id)
654 opConfObj.readXml(opElement, project_id)
655 self.opConfObjList.append(opConfObj)
655 self.opConfObjList.append(opConfObj)
656
656
657 def printattr(self):
657 def printattr(self):
658
658
659 print('%s[%s]: name = %s, datatype = %s, inputId = %s, project_id = %s' % (self.ELEMENTNAME,
659 print('%s[%s]: name = %s, datatype = %s, inputId = %s, project_id = %s' % (self.ELEMENTNAME,
660 self.id,
660 self.id,
661 self.name,
661 self.name,
662 self.datatype,
662 self.datatype,
663 self.inputId,
663 self.inputId,
664 self.project_id))
664 self.project_id))
665
665
666 for opConfObj in self.opConfObjList:
666 for opConfObj in self.opConfObjList:
667 opConfObj.printattr()
667 opConfObj.printattr()
668
668
669 def getKwargs(self):
669 def getKwargs(self):
670
670
671 opObj = self.opConfObjList[0]
671 opObj = self.opConfObjList[0]
672 kwargs = opObj.getKwargs()
672 kwargs = opObj.getKwargs()
673
673
674 return kwargs
674 return kwargs
675
675
676 def createObjects(self):
676 def createObjects(self):
677 '''
677 '''
678 Instancia de unidades de procesamiento.
678 Instancia de unidades de procesamiento.
679 '''
679 '''
680
680
681 className = eval(self.name)
681 className = eval(self.name)
682 #print(self.name)
682 kwargs = self.getKwargs()
683 kwargs = self.getKwargs()
684 #print(kwargs)
685 #print("mark_a")
683 procUnitObj = className(self.id, self.inputId, self.project_id, self.err_queue, self.lock, 'ProcUnit', **kwargs)
686 procUnitObj = className(self.id, self.inputId, self.project_id, self.err_queue, self.lock, 'ProcUnit', **kwargs)
687 #print("mark_b")
684 log.success('creating process...', self.name)
688 log.success('creating process...', self.name)
685
689
686 for opConfObj in self.opConfObjList:
690 for opConfObj in self.opConfObjList:
687
691
688 if opConfObj.type == 'self' and opConfObj.name == 'run':
692 if opConfObj.type == 'self' and opConfObj.name == 'run':
689 continue
693 continue
690 elif opConfObj.type == 'self':
694 elif opConfObj.type == 'self':
691 opObj = getattr(procUnitObj, opConfObj.name)
695 opObj = getattr(procUnitObj, opConfObj.name)
692 else:
696 else:
693 opObj = opConfObj.createObject()
697 opObj = opConfObj.createObject()
694
698
695 log.success('adding operation: {}, type:{}'.format(
699 log.success('adding operation: {}, type:{}'.format(
696 opConfObj.name,
700 opConfObj.name,
697 opConfObj.type), self.name)
701 opConfObj.type), self.name)
698
702
699 procUnitObj.addOperation(opConfObj, opObj)
703 procUnitObj.addOperation(opConfObj, opObj)
700
704
701 procUnitObj.start()
705 procUnitObj.start()
702 self.procUnitObj = procUnitObj
706 self.procUnitObj = procUnitObj
703
707
704 def close(self):
708 def close(self):
705
709
706 for opConfObj in self.opConfObjList:
710 for opConfObj in self.opConfObjList:
707 if opConfObj.type == 'self':
711 if opConfObj.type == 'self':
708 continue
712 continue
709
713
710 opObj = self.procUnitObj.getOperationObj(opConfObj.id)
714 opObj = self.procUnitObj.getOperationObj(opConfObj.id)
711 opObj.close()
715 opObj.close()
712
716
713 self.procUnitObj.close()
717 self.procUnitObj.close()
714
718
715 return
719 return
716
720
717
721
718 class ReadUnitConf(ProcUnitConf):
722 class ReadUnitConf(ProcUnitConf):
719
723
720 ELEMENTNAME = 'ReadUnit'
724 ELEMENTNAME = 'ReadUnit'
721
725
722 def __init__(self):
726 def __init__(self):
723
727
724 self.id = None
728 self.id = None
725 self.datatype = None
729 self.datatype = None
726 self.name = None
730 self.name = None
727 self.inputId = None
731 self.inputId = None
728 self.opConfObjList = []
732 self.opConfObjList = []
729 self.lock = Event()
733 self.lock = Event()
730 self.lock.set()
734 self.lock.set()
731 self.lock.n = Value('d', 0)
735 self.lock.n = Value('d', 0)
732
736
733 def getElementName(self):
737 def getElementName(self):
734
738
735 return self.ELEMENTNAME
739 return self.ELEMENTNAME
736
740
737 def setup(self, project_id, id, name, datatype, err_queue, path='', startDate='', endDate='',
741 def setup(self, project_id, id, name, datatype, err_queue, path='', startDate='', endDate='',
738 startTime='', endTime='', server=None, **kwargs):
742 startTime='', endTime='', server=None, **kwargs):
739
743
740
744
741 '''
745 '''
742 *****el id del proceso sera el Topico
746 *****el id del proceso sera el Topico
743
747
744 Adicion de {topic}, si no esta presente -> error
748 Adicion de {topic}, si no esta presente -> error
745 kwargs deben ser trasmitidos en la instanciacion
749 kwargs deben ser trasmitidos en la instanciacion
746
750
747 '''
751 '''
748
752
749 # Compatible with old signal chain version
753 # Compatible with old signal chain version
750 if datatype == None and name == None:
754 if datatype == None and name == None:
751 raise ValueError('datatype or name should be defined')
755 raise ValueError('datatype or name should be defined')
752 if name == None:
756 if name == None:
753 if 'Reader' in datatype:
757 if 'Reader' in datatype:
754 name = datatype
758 name = datatype
755 datatype = name.replace('Reader','')
759 datatype = name.replace('Reader','')
756 else:
760 else:
757 name = '{}Reader'.format(datatype)
761 name = '{}Reader'.format(datatype)
758 if datatype == None:
762 if datatype == None:
759 if 'Reader' in name:
763 if 'Reader' in name:
760 datatype = name.replace('Reader','')
764 datatype = name.replace('Reader','')
761 else:
765 else:
762 datatype = name
766 datatype = name
763 name = '{}Reader'.format(name)
767 name = '{}Reader'.format(name)
764
768
765 self.id = id
769 self.id = id
766 self.project_id = project_id
770 self.project_id = project_id
767 self.name = name
771 self.name = name
768 self.datatype = datatype
772 self.datatype = datatype
769 if path != '':
773 if path != '':
770 self.path = os.path.abspath(path)
774 self.path = os.path.abspath(path)
775 print (self.path)
771 self.startDate = startDate
776 self.startDate = startDate
772 self.endDate = endDate
777 self.endDate = endDate
773 self.startTime = startTime
778 self.startTime = startTime
774 self.endTime = endTime
779 self.endTime = endTime
775 self.server = server
780 self.server = server
776 self.err_queue = err_queue
781 self.err_queue = err_queue
777 self.addRunOperation(**kwargs)
782 self.addRunOperation(**kwargs)
778
783
779 def update(self, **kwargs):
784 def update(self, **kwargs):
780
785
781 if 'datatype' in kwargs:
786 if 'datatype' in kwargs:
782 datatype = kwargs.pop('datatype')
787 datatype = kwargs.pop('datatype')
783 if 'Reader' in datatype:
788 if 'Reader' in datatype:
784 self.name = datatype
789 self.name = datatype
785 else:
790 else:
786 self.name = '%sReader' % (datatype)
791 self.name = '%sReader' % (datatype)
787 self.datatype = self.name.replace('Reader', '')
792 self.datatype = self.name.replace('Reader', '')
788
793
789 attrs = ('path', 'startDate', 'endDate',
794 attrs = ('path', 'startDate', 'endDate',
790 'startTime', 'endTime')
795 'startTime', 'endTime')
791
796
792 for attr in attrs:
797 for attr in attrs:
793 if attr in kwargs:
798 if attr in kwargs:
794 setattr(self, attr, kwargs.pop(attr))
799 setattr(self, attr, kwargs.pop(attr))
795
800
796 self.updateRunOperation(**kwargs)
801 self.updateRunOperation(**kwargs)
797
802
798 def removeOperations(self):
803 def removeOperations(self):
799
804
800 for obj in self.opConfObjList:
805 for obj in self.opConfObjList:
801 del obj
806 del obj
802
807
803 self.opConfObjList = []
808 self.opConfObjList = []
804
809
805 def addRunOperation(self, **kwargs):
810 def addRunOperation(self, **kwargs):
806
811
807 opObj = self.addOperation(name='run', optype='self')
812 opObj = self.addOperation(name='run', optype='self')
808
813
809 if self.server is None:
814 if self.server is None:
810 opObj.addParameter(
815 opObj.addParameter(
811 name='datatype', value=self.datatype, format='str')
816 name='datatype', value=self.datatype, format='str')
812 opObj.addParameter(name='path', value=self.path, format='str')
817 opObj.addParameter(name='path', value=self.path, format='str')
813 opObj.addParameter(
818 opObj.addParameter(
814 name='startDate', value=self.startDate, format='date')
819 name='startDate', value=self.startDate, format='date')
815 opObj.addParameter(
820 opObj.addParameter(
816 name='endDate', value=self.endDate, format='date')
821 name='endDate', value=self.endDate, format='date')
817 opObj.addParameter(
822 opObj.addParameter(
818 name='startTime', value=self.startTime, format='time')
823 name='startTime', value=self.startTime, format='time')
819 opObj.addParameter(
824 opObj.addParameter(
820 name='endTime', value=self.endTime, format='time')
825 name='endTime', value=self.endTime, format='time')
821
826
822 for key, value in list(kwargs.items()):
827 for key, value in list(kwargs.items()):
823 opObj.addParameter(name=key, value=value,
828 opObj.addParameter(name=key, value=value,
824 format=type(value).__name__)
829 format=type(value).__name__)
825 else:
830 else:
826 opObj.addParameter(name='server', value=self.server, format='str')
831 opObj.addParameter(name='server', value=self.server, format='str')
827
832
828 return opObj
833 return opObj
829
834
830 def updateRunOperation(self, **kwargs):
835 def updateRunOperation(self, **kwargs):
831
836
832 opObj = self.getOperationObj(name='run')
837 opObj = self.getOperationObj(name='run')
833 opObj.removeParameters()
838 opObj.removeParameters()
834
839
835 opObj.addParameter(name='datatype', value=self.datatype, format='str')
840 opObj.addParameter(name='datatype', value=self.datatype, format='str')
836 opObj.addParameter(name='path', value=self.path, format='str')
841 opObj.addParameter(name='path', value=self.path, format='str')
837 opObj.addParameter(
842 opObj.addParameter(
838 name='startDate', value=self.startDate, format='date')
843 name='startDate', value=self.startDate, format='date')
839 opObj.addParameter(name='endDate', value=self.endDate, format='date')
844 opObj.addParameter(name='endDate', value=self.endDate, format='date')
840 opObj.addParameter(
845 opObj.addParameter(
841 name='startTime', value=self.startTime, format='time')
846 name='startTime', value=self.startTime, format='time')
842 opObj.addParameter(name='endTime', value=self.endTime, format='time')
847 opObj.addParameter(name='endTime', value=self.endTime, format='time')
843
848
844 for key, value in list(kwargs.items()):
849 for key, value in list(kwargs.items()):
845 opObj.addParameter(name=key, value=value,
850 opObj.addParameter(name=key, value=value,
846 format=type(value).__name__)
851 format=type(value).__name__)
847
852
848 return opObj
853 return opObj
849
854
850 def readXml(self, upElement, project_id):
855 def readXml(self, upElement, project_id):
851
856
852 self.id = upElement.get('id')
857 self.id = upElement.get('id')
853 self.name = upElement.get('name')
858 self.name = upElement.get('name')
854 self.datatype = upElement.get('datatype')
859 self.datatype = upElement.get('datatype')
855 self.project_id = str(project_id) #yong
860 self.project_id = str(project_id) #yong
856
861
857 if self.ELEMENTNAME == 'ReadUnit':
862 if self.ELEMENTNAME == 'ReadUnit':
858 self.datatype = self.datatype.replace('Reader', '')
863 self.datatype = self.datatype.replace('Reader', '')
859
864
860 self.opConfObjList = []
865 self.opConfObjList = []
861
866
862 opElementList = upElement.iter(OperationConf().getElementName())
867 opElementList = upElement.iter(OperationConf().getElementName())
863
868
864 for opElement in opElementList:
869 for opElement in opElementList:
865 opConfObj = OperationConf()
870 opConfObj = OperationConf()
866 opConfObj.readXml(opElement, project_id)
871 opConfObj.readXml(opElement, project_id)
867 self.opConfObjList.append(opConfObj)
872 self.opConfObjList.append(opConfObj)
868
873
869 if opConfObj.name == 'run':
874 if opConfObj.name == 'run':
870 self.path = opConfObj.getParameterValue('path')
875 self.path = opConfObj.getParameterValue('path')
871 self.startDate = opConfObj.getParameterValue('startDate')
876 self.startDate = opConfObj.getParameterValue('startDate')
872 self.endDate = opConfObj.getParameterValue('endDate')
877 self.endDate = opConfObj.getParameterValue('endDate')
873 self.startTime = opConfObj.getParameterValue('startTime')
878 self.startTime = opConfObj.getParameterValue('startTime')
874 self.endTime = opConfObj.getParameterValue('endTime')
879 self.endTime = opConfObj.getParameterValue('endTime')
875
880
876
881
877 class Project(Process):
882 class Project(Process):
878
883
879 ELEMENTNAME = 'Project'
884 ELEMENTNAME = 'Project'
880
885
881 def __init__(self):
886 def __init__(self):
882
887
883 Process.__init__(self)
888 Process.__init__(self)
884 self.id = None
889 self.id = None
885 self.filename = None
890 self.filename = None
886 self.description = None
891 self.description = None
887 self.email = None
892 self.email = None
888 self.alarm = None
893 self.alarm = None
889 self.procUnitConfObjDict = {}
894 self.procUnitConfObjDict = {}
890 self.err_queue = Queue()
895 self.err_queue = Queue()
891
896
892 def __getNewId(self):
897 def __getNewId(self):
893
898
894 idList = list(self.procUnitConfObjDict.keys())
899 idList = list(self.procUnitConfObjDict.keys())
895 id = int(self.id) * 10
900 id = int(self.id) * 10
896
901
897 while True:
902 while True:
898 id += 1
903 id += 1
899
904
900 if str(id) in idList:
905 if str(id) in idList:
901 continue
906 continue
902
907
903 break
908 break
904
909
905 return str(id)
910 return str(id)
906
911
907 def getElementName(self):
912 def getElementName(self):
908
913
909 return self.ELEMENTNAME
914 return self.ELEMENTNAME
910
915
911 def getId(self):
916 def getId(self):
912
917
913 return self.id
918 return self.id
914
919
915 def updateId(self, new_id):
920 def updateId(self, new_id):
916
921
917 self.id = str(new_id)
922 self.id = str(new_id)
918
923
919 keyList = list(self.procUnitConfObjDict.keys())
924 keyList = list(self.procUnitConfObjDict.keys())
920 keyList.sort()
925 keyList.sort()
921
926
922 n = 1
927 n = 1
923 newProcUnitConfObjDict = {}
928 newProcUnitConfObjDict = {}
924
929
925 for procKey in keyList:
930 for procKey in keyList:
926
931
927 procUnitConfObj = self.procUnitConfObjDict[procKey]
932 procUnitConfObj = self.procUnitConfObjDict[procKey]
928 idProcUnit = str(int(self.id) * 10 + n)
933 idProcUnit = str(int(self.id) * 10 + n)
929 procUnitConfObj.updateId(idProcUnit)
934 procUnitConfObj.updateId(idProcUnit)
930 newProcUnitConfObjDict[idProcUnit] = procUnitConfObj
935 newProcUnitConfObjDict[idProcUnit] = procUnitConfObj
931 n += 1
936 n += 1
932
937
933 self.procUnitConfObjDict = newProcUnitConfObjDict
938 self.procUnitConfObjDict = newProcUnitConfObjDict
934
939
935 def setup(self, id=1, name='', description='', email=None, alarm=[]):
940 def setup(self, id=1, name='', description='', email=None, alarm=[]):
936
941
937 print(' ')
942 print(' ')
938 print('*' * 60)
943 print('*' * 60)
939 print('* Starting SIGNAL CHAIN PROCESSING (Multiprocessing) v%s *' % schainpy.__version__)
944 print('* Starting SIGNAL CHAIN PROCESSING (Multiprocessing) v%s *' % schainpy.__version__)
940 print('*' * 60)
945 print('*' * 60)
941 print("* Python " + python_version() + " *")
946 print("* Python " + python_version() + " *")
942 print('*' * 19)
947 print('*' * 19)
943 print(' ')
948 print(' ')
944 self.id = str(id)
949 self.id = str(id)
945 self.description = description
950 self.description = description
946 self.email = email
951 self.email = email
947 self.alarm = alarm
952 self.alarm = alarm
948 if name:
953 if name:
949 self.name = '{} ({})'.format(Process.__name__, name)
954 self.name = '{} ({})'.format(Process.__name__, name)
950
955
951 def update(self, **kwargs):
956 def update(self, **kwargs):
952
957
953 for key, value in list(kwargs.items()):
958 for key, value in list(kwargs.items()):
954 setattr(self, key, value)
959 setattr(self, key, value)
955
960
956 def clone(self):
961 def clone(self):
957
962
958 p = Project()
963 p = Project()
959 p.procUnitConfObjDict = self.procUnitConfObjDict
964 p.procUnitConfObjDict = self.procUnitConfObjDict
960 return p
965 return p
961
966
962 def addReadUnit(self, id=None, datatype=None, name=None, **kwargs):
967 def addReadUnit(self, id=None, datatype=None, name=None, **kwargs):
963
968
964 '''
969 '''
965 Actualizacion:
970 Actualizacion:
966 Se agrego un nuevo argumento: topic -relativo a la forma de comunicar los procesos simultaneos
971 Se agrego un nuevo argumento: topic -relativo a la forma de comunicar los procesos simultaneos
967
972
968 * El id del proceso sera el topico al que se deben subscribir los procUnits para recibir la informacion(data)
973 * El id del proceso sera el topico al que se deben subscribir los procUnits para recibir la informacion(data)
969
974
970 '''
975 '''
971
976
972 if id is None:
977 if id is None:
973 idReadUnit = self.__getNewId()
978 idReadUnit = self.__getNewId()
974 else:
979 else:
975 idReadUnit = str(id)
980 idReadUnit = str(id)
976
981
977 readUnitConfObj = ReadUnitConf()
982 readUnitConfObj = ReadUnitConf()
978 readUnitConfObj.setup(self.id, idReadUnit, name, datatype, self.err_queue, **kwargs)
983 readUnitConfObj.setup(self.id, idReadUnit, name, datatype, self.err_queue, **kwargs)
979 self.procUnitConfObjDict[readUnitConfObj.getId()] = readUnitConfObj
984 self.procUnitConfObjDict[readUnitConfObj.getId()] = readUnitConfObj
980
985
981 return readUnitConfObj
986 return readUnitConfObj
982
987
983 def addProcUnit(self, inputId='0', datatype=None, name=None):
988 def addProcUnit(self, inputId='0', datatype=None, name=None):
984
989
985 '''
990 '''
986 Actualizacion:
991 Actualizacion:
987 Se agrego dos nuevos argumentos: topic_read (lee data de otro procUnit) y topic_write(escribe o envia data a otro procUnit)
992 Se agrego dos nuevos argumentos: topic_read (lee data de otro procUnit) y topic_write(escribe o envia data a otro procUnit)
988 Deberia reemplazar a "inputId"
993 Deberia reemplazar a "inputId"
989
994
990 ** A fin de mantener el inputID, este sera la representaacion del topicoal que deben subscribirse. El ID propio de la intancia
995 ** A fin de mantener el inputID, este sera la representaacion del topicoal que deben subscribirse. El ID propio de la intancia
991 (proceso) sera el topico de la publicacion, todo sera asignado de manera dinamica.
996 (proceso) sera el topico de la publicacion, todo sera asignado de manera dinamica.
992
997
993 '''
998 '''
994
999
995 idProcUnit = self.__getNewId()
1000 idProcUnit = self.__getNewId()
996 procUnitConfObj = ProcUnitConf()
1001 procUnitConfObj = ProcUnitConf()
997 input_proc = self.procUnitConfObjDict[inputId]
1002 input_proc = self.procUnitConfObjDict[inputId]
998 procUnitConfObj.setup(self.id, idProcUnit, name, datatype, inputId, self.err_queue, input_proc.lock)
1003 procUnitConfObj.setup(self.id, idProcUnit, name, datatype, inputId, self.err_queue, input_proc.lock)
999 self.procUnitConfObjDict[procUnitConfObj.getId()] = procUnitConfObj
1004 self.procUnitConfObjDict[procUnitConfObj.getId()] = procUnitConfObj
1000
1005
1001 return procUnitConfObj
1006 return procUnitConfObj
1002
1007
1003 def removeProcUnit(self, id):
1008 def removeProcUnit(self, id):
1004
1009
1005 if id in list(self.procUnitConfObjDict.keys()):
1010 if id in list(self.procUnitConfObjDict.keys()):
1006 self.procUnitConfObjDict.pop(id)
1011 self.procUnitConfObjDict.pop(id)
1007
1012
1008 def getReadUnitId(self):
1013 def getReadUnitId(self):
1009
1014
1010 readUnitConfObj = self.getReadUnitObj()
1015 readUnitConfObj = self.getReadUnitObj()
1011
1016
1012 return readUnitConfObj.id
1017 return readUnitConfObj.id
1013
1018
1014 def getReadUnitObj(self):
1019 def getReadUnitObj(self):
1015
1020
1016 for obj in list(self.procUnitConfObjDict.values()):
1021 for obj in list(self.procUnitConfObjDict.values()):
1017 if obj.getElementName() == 'ReadUnit':
1022 if obj.getElementName() == 'ReadUnit':
1018 return obj
1023 return obj
1019
1024
1020 return None
1025 return None
1021
1026
1022 def getProcUnitObj(self, id=None, name=None):
1027 def getProcUnitObj(self, id=None, name=None):
1023
1028
1024 if id != None:
1029 if id != None:
1025 return self.procUnitConfObjDict[id]
1030 return self.procUnitConfObjDict[id]
1026
1031
1027 if name != None:
1032 if name != None:
1028 return self.getProcUnitObjByName(name)
1033 return self.getProcUnitObjByName(name)
1029
1034
1030 return None
1035 return None
1031
1036
1032 def getProcUnitObjByName(self, name):
1037 def getProcUnitObjByName(self, name):
1033
1038
1034 for obj in list(self.procUnitConfObjDict.values()):
1039 for obj in list(self.procUnitConfObjDict.values()):
1035 if obj.name == name:
1040 if obj.name == name:
1036 return obj
1041 return obj
1037
1042
1038 return None
1043 return None
1039
1044
1040 def procUnitItems(self):
1045 def procUnitItems(self):
1041
1046
1042 return list(self.procUnitConfObjDict.items())
1047 return list(self.procUnitConfObjDict.items())
1043
1048
1044 def makeXml(self):
1049 def makeXml(self):
1045
1050
1046 projectElement = Element('Project')
1051 projectElement = Element('Project')
1047 projectElement.set('id', str(self.id))
1052 projectElement.set('id', str(self.id))
1048 projectElement.set('name', self.name)
1053 projectElement.set('name', self.name)
1049 projectElement.set('description', self.description)
1054 projectElement.set('description', self.description)
1050
1055
1051 for procUnitConfObj in list(self.procUnitConfObjDict.values()):
1056 for procUnitConfObj in list(self.procUnitConfObjDict.values()):
1052 procUnitConfObj.makeXml(projectElement)
1057 procUnitConfObj.makeXml(projectElement)
1053
1058
1054 self.projectElement = projectElement
1059 self.projectElement = projectElement
1055
1060
1056 def writeXml(self, filename=None):
1061 def writeXml(self, filename=None):
1057
1062
1058 if filename == None:
1063 if filename == None:
1059 if self.filename:
1064 if self.filename:
1060 filename = self.filename
1065 filename = self.filename
1061 else:
1066 else:
1062 filename = 'schain.xml'
1067 filename = 'schain.xml'
1063
1068
1064 if not filename:
1069 if not filename:
1065 print('filename has not been defined. Use setFilename(filename) for do it.')
1070 print('filename has not been defined. Use setFilename(filename) for do it.')
1066 return 0
1071 return 0
1067
1072
1068 abs_file = os.path.abspath(filename)
1073 abs_file = os.path.abspath(filename)
1069
1074
1070 if not os.access(os.path.dirname(abs_file), os.W_OK):
1075 if not os.access(os.path.dirname(abs_file), os.W_OK):
1071 print('No write permission on %s' % os.path.dirname(abs_file))
1076 print('No write permission on %s' % os.path.dirname(abs_file))
1072 return 0
1077 return 0
1073
1078
1074 if os.path.isfile(abs_file) and not(os.access(abs_file, os.W_OK)):
1079 if os.path.isfile(abs_file) and not(os.access(abs_file, os.W_OK)):
1075 print('File %s already exists and it could not be overwriten' % abs_file)
1080 print('File %s already exists and it could not be overwriten' % abs_file)
1076 return 0
1081 return 0
1077
1082
1078 self.makeXml()
1083 self.makeXml()
1079
1084
1080 ElementTree(self.projectElement).write(abs_file, method='xml')
1085 ElementTree(self.projectElement).write(abs_file, method='xml')
1081
1086
1082 self.filename = abs_file
1087 self.filename = abs_file
1083
1088
1084 return 1
1089 return 1
1085
1090
1086 def readXml(self, filename=None):
1091 def readXml(self, filename=None):
1087
1092
1088 if not filename:
1093 if not filename:
1089 print('filename is not defined')
1094 print('filename is not defined')
1090 return 0
1095 return 0
1091
1096
1092 abs_file = os.path.abspath(filename)
1097 abs_file = os.path.abspath(filename)
1093
1098
1094 if not os.path.isfile(abs_file):
1099 if not os.path.isfile(abs_file):
1095 print('%s file does not exist' % abs_file)
1100 print('%s file does not exist' % abs_file)
1096 return 0
1101 return 0
1097
1102
1098 self.projectElement = None
1103 self.projectElement = None
1099 self.procUnitConfObjDict = {}
1104 self.procUnitConfObjDict = {}
1100
1105
1101 try:
1106 try:
1102 self.projectElement = ElementTree().parse(abs_file)
1107 self.projectElement = ElementTree().parse(abs_file)
1103 except:
1108 except:
1104 print('Error reading %s, verify file format' % filename)
1109 print('Error reading %s, verify file format' % filename)
1105 return 0
1110 return 0
1106
1111
1107 self.project = self.projectElement.tag
1112 self.project = self.projectElement.tag
1108
1113
1109 self.id = self.projectElement.get('id')
1114 self.id = self.projectElement.get('id')
1110 self.name = self.projectElement.get('name')
1115 self.name = self.projectElement.get('name')
1111 self.description = self.projectElement.get('description')
1116 self.description = self.projectElement.get('description')
1112
1117
1113 readUnitElementList = self.projectElement.iter(
1118 readUnitElementList = self.projectElement.iter(
1114 ReadUnitConf().getElementName())
1119 ReadUnitConf().getElementName())
1115
1120
1116 for readUnitElement in readUnitElementList:
1121 for readUnitElement in readUnitElementList:
1117 readUnitConfObj = ReadUnitConf()
1122 readUnitConfObj = ReadUnitConf()
1118 readUnitConfObj.readXml(readUnitElement, self.id)
1123 readUnitConfObj.readXml(readUnitElement, self.id)
1119 self.procUnitConfObjDict[readUnitConfObj.getId()] = readUnitConfObj
1124 self.procUnitConfObjDict[readUnitConfObj.getId()] = readUnitConfObj
1120
1125
1121 procUnitElementList = self.projectElement.iter(
1126 procUnitElementList = self.projectElement.iter(
1122 ProcUnitConf().getElementName())
1127 ProcUnitConf().getElementName())
1123
1128
1124 for procUnitElement in procUnitElementList:
1129 for procUnitElement in procUnitElementList:
1125 procUnitConfObj = ProcUnitConf()
1130 procUnitConfObj = ProcUnitConf()
1126 procUnitConfObj.readXml(procUnitElement, self.id)
1131 procUnitConfObj.readXml(procUnitElement, self.id)
1127 self.procUnitConfObjDict[procUnitConfObj.getId()] = procUnitConfObj
1132 self.procUnitConfObjDict[procUnitConfObj.getId()] = procUnitConfObj
1128
1133
1129 self.filename = abs_file
1134 self.filename = abs_file
1130
1135
1131 return 1
1136 return 1
1132
1137
1133 def __str__(self):
1138 def __str__(self):
1134
1139
1135 print('Project: name = %s, description = %s, id = %s' % (
1140 print('Project: name = %s, description = %s, id = %s' % (
1136 self.name,
1141 self.name,
1137 self.description,
1142 self.description,
1138 self.id))
1143 self.id))
1139
1144
1140 for procUnitConfObj in self.procUnitConfObjDict.values():
1145 for procUnitConfObj in self.procUnitConfObjDict.values():
1141 print(procUnitConfObj)
1146 print(procUnitConfObj)
1142
1147
1143 def createObjects(self):
1148 def createObjects(self):
1144
1149
1145
1150
1146 keys = list(self.procUnitConfObjDict.keys())
1151 keys = list(self.procUnitConfObjDict.keys())
1147 keys.sort()
1152 keys.sort()
1148 for key in keys:
1153 for key in keys:
1149 self.procUnitConfObjDict[key].createObjects()
1154 self.procUnitConfObjDict[key].createObjects()
1150
1155
1151 def monitor(self):
1156 def monitor(self):
1152
1157
1153 t = Thread(target=self.__monitor, args=(self.err_queue, self.ctx))
1158 t = Thread(target=self.__monitor, args=(self.err_queue, self.ctx))
1154 t.start()
1159 t.start()
1155
1160
1156 def __monitor(self, queue, ctx):
1161 def __monitor(self, queue, ctx):
1157
1162
1158 import socket
1163 import socket
1159
1164
1160 procs = 0
1165 procs = 0
1161 err_msg = ''
1166 err_msg = ''
1162
1167
1163 while True:
1168 while True:
1164 msg = queue.get()
1169 msg = queue.get()
1165 if '#_start_#' in msg:
1170 if '#_start_#' in msg:
1166 procs += 1
1171 procs += 1
1167 elif '#_end_#' in msg:
1172 elif '#_end_#' in msg:
1168 procs -=1
1173 procs -=1
1169 else:
1174 else:
1170 err_msg = msg
1175 err_msg = msg
1171
1176
1172 if procs == 0 or 'Traceback' in err_msg:
1177 if procs == 0 or 'Traceback' in err_msg:
1173 break
1178 break
1174 time.sleep(0.1)
1179 time.sleep(0.1)
1175
1180
1176 if '|' in err_msg:
1181 if '|' in err_msg:
1177 name, err = err_msg.split('|')
1182 name, err = err_msg.split('|')
1178 if 'SchainWarning' in err:
1183 if 'SchainWarning' in err:
1179 log.warning(err.split('SchainWarning:')[-1].split('\n')[0].strip(), name)
1184 log.warning(err.split('SchainWarning:')[-1].split('\n')[0].strip(), name)
1180 elif 'SchainError' in err:
1185 elif 'SchainError' in err:
1181 log.error(err.split('SchainError:')[-1].split('\n')[0].strip(), name)
1186 log.error(err.split('SchainError:')[-1].split('\n')[0].strip(), name)
1182 else:
1187 else:
1183 log.error(err, name)
1188 log.error(err, name)
1184 else:
1189 else:
1185 name, err = self.name, err_msg
1190 name, err = self.name, err_msg
1186
1191
1187 time.sleep(2)
1192 time.sleep(2)
1188
1193
1189 for conf in self.procUnitConfObjDict.values():
1194 for conf in self.procUnitConfObjDict.values():
1190 for confop in conf.opConfObjList:
1195 for confop in conf.opConfObjList:
1191 if confop.type == 'external':
1196 if confop.type == 'external':
1192 confop.opObj.terminate()
1197 confop.opObj.terminate()
1193 conf.procUnitObj.terminate()
1198 conf.procUnitObj.terminate()
1194
1199
1195 ctx.term()
1200 ctx.term()
1196
1201
1197 message = ''.join(err)
1202 message = ''.join(err)
1198
1203
1199 if err_msg:
1204 if err_msg:
1200 subject = 'SChain v%s: Error running %s\n' % (
1205 subject = 'SChain v%s: Error running %s\n' % (
1201 schainpy.__version__, self.name)
1206 schainpy.__version__, self.name)
1202
1207
1203 subtitle = 'Hostname: %s\n' % socket.gethostbyname(
1208 subtitle = 'Hostname: %s\n' % socket.gethostbyname(
1204 socket.gethostname())
1209 socket.gethostname())
1205 subtitle += 'Working directory: %s\n' % os.path.abspath('./')
1210 subtitle += 'Working directory: %s\n' % os.path.abspath('./')
1206 subtitle += 'Configuration file: %s\n' % self.filename
1211 subtitle += 'Configuration file: %s\n' % self.filename
1207 subtitle += 'Time: %s\n' % str(datetime.datetime.now())
1212 subtitle += 'Time: %s\n' % str(datetime.datetime.now())
1208
1213
1209 readUnitConfObj = self.getReadUnitObj()
1214 readUnitConfObj = self.getReadUnitObj()
1210 if readUnitConfObj:
1215 if readUnitConfObj:
1211 subtitle += '\nInput parameters:\n'
1216 subtitle += '\nInput parameters:\n'
1212 subtitle += '[Data path = %s]\n' % readUnitConfObj.path
1217 subtitle += '[Data path = %s]\n' % readUnitConfObj.path
1213 subtitle += '[Data type = %s]\n' % readUnitConfObj.datatype
1218 subtitle += '[Data type = %s]\n' % readUnitConfObj.datatype
1214 subtitle += '[Start date = %s]\n' % readUnitConfObj.startDate
1219 subtitle += '[Start date = %s]\n' % readUnitConfObj.startDate
1215 subtitle += '[End date = %s]\n' % readUnitConfObj.endDate
1220 subtitle += '[End date = %s]\n' % readUnitConfObj.endDate
1216 subtitle += '[Start time = %s]\n' % readUnitConfObj.startTime
1221 subtitle += '[Start time = %s]\n' % readUnitConfObj.startTime
1217 subtitle += '[End time = %s]\n' % readUnitConfObj.endTime
1222 subtitle += '[End time = %s]\n' % readUnitConfObj.endTime
1218
1223
1219 a = Alarm(
1224 a = Alarm(
1220 modes=self.alarm,
1225 modes=self.alarm,
1221 email=self.email,
1226 email=self.email,
1222 message=message,
1227 message=message,
1223 subject=subject,
1228 subject=subject,
1224 subtitle=subtitle,
1229 subtitle=subtitle,
1225 filename=self.filename
1230 filename=self.filename
1226 )
1231 )
1227
1232
1228 a.start()
1233 a.start()
1229
1234
1230 def isPaused(self):
1235 def isPaused(self):
1231 return 0
1236 return 0
1232
1237
1233 def isStopped(self):
1238 def isStopped(self):
1234 return 0
1239 return 0
1235
1240
1236 def runController(self):
1241 def runController(self):
1237 '''
1242 '''
1238 returns 0 when this process has been stopped, 1 otherwise
1243 returns 0 when this process has been stopped, 1 otherwise
1239 '''
1244 '''
1240
1245
1241 if self.isPaused():
1246 if self.isPaused():
1242 print('Process suspended')
1247 print('Process suspended')
1243
1248
1244 while True:
1249 while True:
1245 time.sleep(0.1)
1250 time.sleep(0.1)
1246
1251
1247 if not self.isPaused():
1252 if not self.isPaused():
1248 break
1253 break
1249
1254
1250 if self.isStopped():
1255 if self.isStopped():
1251 break
1256 break
1252
1257
1253 print('Process reinitialized')
1258 print('Process reinitialized')
1254
1259
1255 if self.isStopped():
1260 if self.isStopped():
1256 print('Process stopped')
1261 print('Process stopped')
1257 return 0
1262 return 0
1258
1263
1259 return 1
1264 return 1
1260
1265
1261 def setFilename(self, filename):
1266 def setFilename(self, filename):
1262
1267
1263 self.filename = filename
1268 self.filename = filename
1264
1269
1265 def setProxy(self):
1270 def setProxy(self):
1266
1271
1267 if not os.path.exists('/tmp/schain'):
1272 if not os.path.exists('/tmp/schain'):
1268 os.mkdir('/tmp/schain')
1273 os.mkdir('/tmp/schain')
1269
1274
1270 self.ctx = zmq.Context()
1275 self.ctx = zmq.Context()
1271 xpub = self.ctx.socket(zmq.XPUB)
1276 xpub = self.ctx.socket(zmq.XPUB)
1272 xpub.bind('ipc:///tmp/schain/{}_pub'.format(self.id))
1277 xpub.bind('ipc:///tmp/schain/{}_pub'.format(self.id))
1273 xsub = self.ctx.socket(zmq.XSUB)
1278 xsub = self.ctx.socket(zmq.XSUB)
1274 xsub.bind('ipc:///tmp/schain/{}_sub'.format(self.id))
1279 xsub.bind('ipc:///tmp/schain/{}_sub'.format(self.id))
1275 self.monitor()
1280 self.monitor()
1276 try:
1281 try:
1277 zmq.proxy(xpub, xsub)
1282 zmq.proxy(xpub, xsub)
1278 except zmq.ContextTerminated:
1283 except zmq.ContextTerminated:
1279 xpub.close()
1284 xpub.close()
1280 xsub.close()
1285 xsub.close()
1281
1286
1282 def run(self):
1287 def run(self):
1283
1288
1284 log.success('Starting {}: {}'.format(self.name, self.id), tag='')
1289 log.success('Starting {}: {}'.format(self.name, self.id), tag='')
1285 self.start_time = time.time()
1290 self.start_time = time.time()
1286 self.createObjects()
1291 self.createObjects()
1287 self.setProxy()
1292 self.setProxy()
1288 log.success('{} Done (Time: {}s)'.format(
1293 log.success('{} Done (Time: {}s)'.format(
1289 self.name,
1294 self.name,
1290 time.time()-self.start_time), '')
1295 time.time()-self.start_time), '')
1 NO CONTENT: modified file
NO CONTENT: modified file
1 NO CONTENT: modified file
NO CONTENT: modified file
@@ -1,810 +1,808
1
1
2 import os
2 import os
3 import sys
3 import sys
4 import zmq
4 import zmq
5 import time
5 import time
6 import numpy
6 import numpy
7 import datetime
7 import datetime
8 from functools import wraps
8 from functools import wraps
9 from threading import Thread
9 from threading import Thread
10 import matplotlib
10 import matplotlib
11
11
12 if 'BACKEND' in os.environ:
12 if 'BACKEND' in os.environ:
13 matplotlib.use(os.environ['BACKEND'])
13 matplotlib.use(os.environ['BACKEND'])
14 elif 'linux' in sys.platform:
14 elif 'linux' in sys.platform:
15 matplotlib.use("TkAgg")
15 matplotlib.use("TkAgg")
16 elif 'darwin' in sys.platform:
16 elif 'darwin' in sys.platform:
17 matplotlib.use('WxAgg')
17 matplotlib.use('WxAgg')
18 else:
18 else:
19 from schainpy.utils import log
19 from schainpy.utils import log
20 log.warning('Using default Backend="Agg"', 'INFO')
20 log.warning('Using default Backend="Agg"', 'INFO')
21 matplotlib.use('Agg')
21 matplotlib.use('Agg')
22
22
23 import matplotlib.pyplot as plt
23 import matplotlib.pyplot as plt
24 from matplotlib.patches import Polygon
24 from matplotlib.patches import Polygon
25 from mpl_toolkits.axes_grid1 import make_axes_locatable
25 from mpl_toolkits.axes_grid1 import make_axes_locatable
26 from matplotlib.ticker import FuncFormatter, LinearLocator, MultipleLocator
26 from matplotlib.ticker import FuncFormatter, LinearLocator, MultipleLocator
27
27
28 from schainpy.model.data.jrodata import PlotterData
28 from schainpy.model.data.jrodata import PlotterData
29 from schainpy.model.proc.jroproc_base import ProcessingUnit, Operation, MPDecorator
29 from schainpy.model.proc.jroproc_base import ProcessingUnit, Operation, MPDecorator
30 from schainpy.utils import log
30 from schainpy.utils import log
31
31
32 jet_values = matplotlib.pyplot.get_cmap('jet', 100)(numpy.arange(100))[10:90]
32 jet_values = matplotlib.pyplot.get_cmap('jet', 100)(numpy.arange(100))[10:90]
33 blu_values = matplotlib.pyplot.get_cmap(
33 blu_values = matplotlib.pyplot.get_cmap(
34 'seismic_r', 20)(numpy.arange(20))[10:15]
34 'seismic_r', 20)(numpy.arange(20))[10:15]
35 ncmap = matplotlib.colors.LinearSegmentedColormap.from_list(
35 ncmap = matplotlib.colors.LinearSegmentedColormap.from_list(
36 'jro', numpy.vstack((blu_values, jet_values)))
36 'jro', numpy.vstack((blu_values, jet_values)))
37 matplotlib.pyplot.register_cmap(cmap=ncmap)
37 matplotlib.pyplot.register_cmap(cmap=ncmap)
38
38
39 CMAPS = [plt.get_cmap(s) for s in ('jro', 'jet', 'viridis',
39 CMAPS = [plt.get_cmap(s) for s in ('jro', 'jet', 'viridis',
40 'plasma', 'inferno', 'Greys', 'seismic', 'bwr', 'coolwarm')]
40 'plasma', 'inferno', 'Greys', 'seismic', 'bwr', 'coolwarm')]
41
41
42 EARTH_RADIUS = 6.3710e3
42 EARTH_RADIUS = 6.3710e3
43
43
44 def ll2xy(lat1, lon1, lat2, lon2):
44 def ll2xy(lat1, lon1, lat2, lon2):
45
45
46 p = 0.017453292519943295
46 p = 0.017453292519943295
47 a = 0.5 - numpy.cos((lat2 - lat1) * p)/2 + numpy.cos(lat1 * p) * \
47 a = 0.5 - numpy.cos((lat2 - lat1) * p)/2 + numpy.cos(lat1 * p) * \
48 numpy.cos(lat2 * p) * (1 - numpy.cos((lon2 - lon1) * p)) / 2
48 numpy.cos(lat2 * p) * (1 - numpy.cos((lon2 - lon1) * p)) / 2
49 r = 12742 * numpy.arcsin(numpy.sqrt(a))
49 r = 12742 * numpy.arcsin(numpy.sqrt(a))
50 theta = numpy.arctan2(numpy.sin((lon2-lon1)*p)*numpy.cos(lat2*p), numpy.cos(lat1*p)
50 theta = numpy.arctan2(numpy.sin((lon2-lon1)*p)*numpy.cos(lat2*p), numpy.cos(lat1*p)
51 * numpy.sin(lat2*p)-numpy.sin(lat1*p)*numpy.cos(lat2*p)*numpy.cos((lon2-lon1)*p))
51 * numpy.sin(lat2*p)-numpy.sin(lat1*p)*numpy.cos(lat2*p)*numpy.cos((lon2-lon1)*p))
52 theta = -theta + numpy.pi/2
52 theta = -theta + numpy.pi/2
53 return r*numpy.cos(theta), r*numpy.sin(theta)
53 return r*numpy.cos(theta), r*numpy.sin(theta)
54
54
55
55
56 def km2deg(km):
56 def km2deg(km):
57 '''
57 '''
58 Convert distance in km to degrees
58 Convert distance in km to degrees
59 '''
59 '''
60
60
61 return numpy.rad2deg(km/EARTH_RADIUS)
61 return numpy.rad2deg(km/EARTH_RADIUS)
62
62
63
63
64 def figpause(interval):
64 def figpause(interval):
65 backend = plt.rcParams['backend']
65 backend = plt.rcParams['backend']
66 if backend in matplotlib.rcsetup.interactive_bk:
66 if backend in matplotlib.rcsetup.interactive_bk:
67 figManager = matplotlib._pylab_helpers.Gcf.get_active()
67 figManager = matplotlib._pylab_helpers.Gcf.get_active()
68 if figManager is not None:
68 if figManager is not None:
69 canvas = figManager.canvas
69 canvas = figManager.canvas
70 if canvas.figure.stale:
70 if canvas.figure.stale:
71 canvas.draw()
71 canvas.draw()
72 try:
72 try:
73 canvas.start_event_loop(interval)
73 canvas.start_event_loop(interval)
74 except:
74 except:
75 pass
75 pass
76 return
76 return
77
77
78
78
79 def popup(message):
79 def popup(message):
80 '''
80 '''
81 '''
81 '''
82
82
83 fig = plt.figure(figsize=(12, 8), facecolor='r')
83 fig = plt.figure(figsize=(12, 8), facecolor='r')
84 text = '\n'.join([s.strip() for s in message.split(':')])
84 text = '\n'.join([s.strip() for s in message.split(':')])
85 fig.text(0.01, 0.5, text, ha='left', va='center',
85 fig.text(0.01, 0.5, text, ha='left', va='center',
86 size='20', weight='heavy', color='w')
86 size='20', weight='heavy', color='w')
87 fig.show()
87 fig.show()
88 figpause(1000)
88 figpause(1000)
89
89
90
90
91 class Throttle(object):
91 class Throttle(object):
92 '''
92 '''
93 Decorator that prevents a function from being called more than once every
93 Decorator that prevents a function from being called more than once every
94 time period.
94 time period.
95 To create a function that cannot be called more than once a minute, but
95 To create a function that cannot be called more than once a minute, but
96 will sleep until it can be called:
96 will sleep until it can be called:
97 @Throttle(minutes=1)
97 @Throttle(minutes=1)
98 def foo():
98 def foo():
99 pass
99 pass
100
100
101 for i in range(10):
101 for i in range(10):
102 foo()
102 foo()
103 print "This function has run %s times." % i
103 print "This function has run %s times." % i
104 '''
104 '''
105
105
106 def __init__(self, seconds=0, minutes=0, hours=0):
106 def __init__(self, seconds=0, minutes=0, hours=0):
107 self.throttle_period = datetime.timedelta(
107 self.throttle_period = datetime.timedelta(
108 seconds=seconds, minutes=minutes, hours=hours
108 seconds=seconds, minutes=minutes, hours=hours
109 )
109 )
110
110
111 self.time_of_last_call = datetime.datetime.min
111 self.time_of_last_call = datetime.datetime.min
112
112
113 def __call__(self, fn):
113 def __call__(self, fn):
114 @wraps(fn)
114 @wraps(fn)
115 def wrapper(*args, **kwargs):
115 def wrapper(*args, **kwargs):
116 coerce = kwargs.pop('coerce', None)
116 coerce = kwargs.pop('coerce', None)
117 if coerce:
117 if coerce:
118 self.time_of_last_call = datetime.datetime.now()
118 self.time_of_last_call = datetime.datetime.now()
119 return fn(*args, **kwargs)
119 return fn(*args, **kwargs)
120 else:
120 else:
121 now = datetime.datetime.now()
121 now = datetime.datetime.now()
122 time_since_last_call = now - self.time_of_last_call
122 time_since_last_call = now - self.time_of_last_call
123 time_left = self.throttle_period - time_since_last_call
123 time_left = self.throttle_period - time_since_last_call
124
124
125 if time_left > datetime.timedelta(seconds=0):
125 if time_left > datetime.timedelta(seconds=0):
126 return
126 return
127
127
128 self.time_of_last_call = datetime.datetime.now()
128 self.time_of_last_call = datetime.datetime.now()
129 return fn(*args, **kwargs)
129 return fn(*args, **kwargs)
130
130
131 return wrapper
131 return wrapper
132
132
133 def apply_throttle(value):
133 def apply_throttle(value):
134
134
135 @Throttle(seconds=value)
135 @Throttle(seconds=value)
136 def fnThrottled(fn):
136 def fnThrottled(fn):
137 fn()
137 fn()
138
138
139 return fnThrottled
139 return fnThrottled
140
140
141
141
142 @MPDecorator
142 @MPDecorator
143 class Plot(Operation):
143 class Plot(Operation):
144 '''
144 '''
145 Base class for Schain plotting operations
145 Base class for Schain plotting operations
146 '''
146 '''
147
147
148 CODE = 'Figure'
148 CODE = 'Figure'
149 colormap = 'jet'
149 colormap = 'jet'
150 bgcolor = 'white'
150 bgcolor = 'white'
151 __missing = 1E30
151 __missing = 1E30
152
152
153 __attrs__ = ['show', 'save', 'xmin', 'xmax', 'ymin', 'ymax', 'zmin', 'zmax',
153 __attrs__ = ['show', 'save', 'xmin', 'xmax', 'ymin', 'ymax', 'zmin', 'zmax',
154 'zlimits', 'xlabel', 'ylabel', 'xaxis', 'cb_label', 'title',
154 'zlimits', 'xlabel', 'ylabel', 'xaxis', 'cb_label', 'title',
155 'colorbar', 'bgcolor', 'width', 'height', 'localtime', 'oneFigure',
155 'colorbar', 'bgcolor', 'width', 'height', 'localtime', 'oneFigure',
156 'showprofile', 'decimation', 'pause']
156 'showprofile', 'decimation', 'pause']
157
157
158 def __init__(self):
158 def __init__(self):
159
159
160 Operation.__init__(self)
160 Operation.__init__(self)
161 self.isConfig = False
161 self.isConfig = False
162 self.isPlotConfig = False
162 self.isPlotConfig = False
163 self.save_counter = 1
163 self.save_counter = 1
164 self.sender_counter = 1
164 self.sender_counter = 1
165 self.data = None
165 self.data = None
166
166
167 def __fmtTime(self, x, pos):
167 def __fmtTime(self, x, pos):
168 '''
168 '''
169 '''
169 '''
170
170
171 return '{}'.format(self.getDateTime(x).strftime('%H:%M'))
171 return '{}'.format(self.getDateTime(x).strftime('%H:%M'))
172
172
173 def __setup(self, **kwargs):
173 def __setup(self, **kwargs):
174 '''
174 '''
175 Initialize variables
175 Initialize variables
176 '''
176 '''
177
177
178 self.figures = []
178 self.figures = []
179 self.axes = []
179 self.axes = []
180 self.cb_axes = []
180 self.cb_axes = []
181 self.localtime = kwargs.pop('localtime', True)
181 self.localtime = kwargs.pop('localtime', True)
182 self.show = kwargs.get('show', True)
182 self.show = kwargs.get('show', True)
183 self.save = kwargs.get('save', False)
183 self.save = kwargs.get('save', False)
184 self.save_period = kwargs.get('save_period', 1)
184 self.save_period = kwargs.get('save_period', 1)
185 self.ftp = kwargs.get('ftp', False)
185 self.ftp = kwargs.get('ftp', False)
186 self.colormap = kwargs.get('colormap', self.colormap)
186 self.colormap = kwargs.get('colormap', self.colormap)
187 self.colormap_coh = kwargs.get('colormap_coh', 'jet')
187 self.colormap_coh = kwargs.get('colormap_coh', 'jet')
188 self.colormap_phase = kwargs.get('colormap_phase', 'RdBu_r')
188 self.colormap_phase = kwargs.get('colormap_phase', 'RdBu_r')
189 self.colormaps = kwargs.get('colormaps', None)
189 self.colormaps = kwargs.get('colormaps', None)
190 self.bgcolor = kwargs.get('bgcolor', self.bgcolor)
190 self.bgcolor = kwargs.get('bgcolor', self.bgcolor)
191 self.showprofile = kwargs.get('showprofile', False)
191 self.showprofile = kwargs.get('showprofile', False)
192 self.title = kwargs.get('wintitle', self.CODE.upper())
192 self.title = kwargs.get('wintitle', self.CODE.upper())
193 self.cb_label = kwargs.get('cb_label', None)
193 self.cb_label = kwargs.get('cb_label', None)
194 self.cb_labels = kwargs.get('cb_labels', None)
194 self.cb_labels = kwargs.get('cb_labels', None)
195 self.labels = kwargs.get('labels', None)
195 self.labels = kwargs.get('labels', None)
196 self.xaxis = kwargs.get('xaxis', 'frequency')
196 self.xaxis = kwargs.get('xaxis', 'frequency')
197 self.zmin = kwargs.get('zmin', None)
197 self.zmin = kwargs.get('zmin', None)
198 self.zmax = kwargs.get('zmax', None)
198 self.zmax = kwargs.get('zmax', None)
199 self.zlimits = kwargs.get('zlimits', None)
199 self.zlimits = kwargs.get('zlimits', None)
200 self.xmin = kwargs.get('xmin', None)
200 self.xmin = kwargs.get('xmin', None)
201 self.xmax = kwargs.get('xmax', None)
201 self.xmax = kwargs.get('xmax', None)
202 self.xrange = kwargs.get('xrange', 24)
202 self.xrange = kwargs.get('xrange', 24)
203 self.xscale = kwargs.get('xscale', None)
203 self.xscale = kwargs.get('xscale', None)
204 self.ymin = kwargs.get('ymin', None)
204 self.ymin = kwargs.get('ymin', None)
205 self.ymax = kwargs.get('ymax', None)
205 self.ymax = kwargs.get('ymax', None)
206 self.yscale = kwargs.get('yscale', None)
206 self.yscale = kwargs.get('yscale', None)
207 self.xlabel = kwargs.get('xlabel', None)
207 self.xlabel = kwargs.get('xlabel', None)
208 self.decimation = kwargs.get('decimation', None)
208 self.decimation = kwargs.get('decimation', None)
209 self.showSNR = kwargs.get('showSNR', False)
209 self.showSNR = kwargs.get('showSNR', False)
210 self.oneFigure = kwargs.get('oneFigure', True)
210 self.oneFigure = kwargs.get('oneFigure', True)
211 self.width = kwargs.get('width', None)
211 self.width = kwargs.get('width', None)
212 self.height = kwargs.get('height', None)
212 self.height = kwargs.get('height', None)
213 self.colorbar = kwargs.get('colorbar', True)
213 self.colorbar = kwargs.get('colorbar', True)
214 self.factors = kwargs.get('factors', [1, 1, 1, 1, 1, 1, 1, 1])
214 self.factors = kwargs.get('factors', [1, 1, 1, 1, 1, 1, 1, 1])
215 self.channels = kwargs.get('channels', None)
215 self.channels = kwargs.get('channels', None)
216 self.titles = kwargs.get('titles', [])
216 self.titles = kwargs.get('titles', [])
217 self.polar = False
217 self.polar = False
218 self.type = kwargs.get('type', 'iq')
218 self.type = kwargs.get('type', 'iq')
219 self.grid = kwargs.get('grid', False)
219 self.grid = kwargs.get('grid', False)
220 self.pause = kwargs.get('pause', False)
220 self.pause = kwargs.get('pause', False)
221 self.save_labels = kwargs.get('save_labels', None)
221 self.save_labels = kwargs.get('save_labels', None)
222 self.realtime = kwargs.get('realtime', True)
222 self.realtime = kwargs.get('realtime', True)
223 self.buffering = kwargs.get('buffering', True)
223 self.buffering = kwargs.get('buffering', True)
224 self.throttle = kwargs.get('throttle', 2)
224 self.throttle = kwargs.get('throttle', 2)
225 self.exp_code = kwargs.get('exp_code', None)
225 self.exp_code = kwargs.get('exp_code', None)
226 self.plot_server = kwargs.get('plot_server', False)
226 self.plot_server = kwargs.get('plot_server', False)
227 self.sender_period = kwargs.get('sender_period', 1)
227 self.sender_period = kwargs.get('sender_period', 1)
228 self.__throttle_plot = apply_throttle(self.throttle)
228 self.__throttle_plot = apply_throttle(self.throttle)
229 self.data = PlotterData(
229 self.data = PlotterData(
230 self.CODE, self.throttle, self.exp_code, self.buffering, snr=self.showSNR)
230 self.CODE, self.throttle, self.exp_code, self.buffering, snr=self.showSNR)
231
231
232 if self.plot_server:
232 if self.plot_server:
233 if not self.plot_server.startswith('tcp://'):
233 if not self.plot_server.startswith('tcp://'):
234 self.plot_server = 'tcp://{}'.format(self.plot_server)
234 self.plot_server = 'tcp://{}'.format(self.plot_server)
235 log.success(
235 log.success(
236 'Sending to server: {}'.format(self.plot_server),
236 'Sending to server: {}'.format(self.plot_server),
237 self.name
237 self.name
238 )
238 )
239 if 'plot_name' in kwargs:
239 if 'plot_name' in kwargs:
240 self.plot_name = kwargs['plot_name']
240 self.plot_name = kwargs['plot_name']
241
241
242 def __setup_plot(self):
242 def __setup_plot(self):
243 '''
243 '''
244 Common setup for all figures, here figures and axes are created
244 Common setup for all figures, here figures and axes are created
245 '''
245 '''
246
246
247 self.setup()
247 self.setup()
248
248
249 self.time_label = 'LT' if self.localtime else 'UTC'
249 self.time_label = 'LT' if self.localtime else 'UTC'
250
250
251 if self.width is None:
251 if self.width is None:
252 self.width = 8
252 self.width = 8
253
253
254 self.figures = []
254 self.figures = []
255 self.axes = []
255 self.axes = []
256 self.cb_axes = []
256 self.cb_axes = []
257 self.pf_axes = []
257 self.pf_axes = []
258 self.cmaps = []
258 self.cmaps = []
259
259
260 size = '15%' if self.ncols == 1 else '30%'
260 size = '15%' if self.ncols == 1 else '30%'
261 pad = '4%' if self.ncols == 1 else '8%'
261 pad = '4%' if self.ncols == 1 else '8%'
262
262
263 if self.oneFigure:
263 if self.oneFigure:
264 if self.height is None:
264 if self.height is None:
265 self.height = 1.4 * self.nrows + 1
265 self.height = 1.4 * self.nrows + 1
266 fig = plt.figure(figsize=(self.width, self.height),
266 fig = plt.figure(figsize=(self.width, self.height),
267 edgecolor='k',
267 edgecolor='k',
268 facecolor='w')
268 facecolor='w')
269 self.figures.append(fig)
269 self.figures.append(fig)
270 for n in range(self.nplots):
270 for n in range(self.nplots):
271 ax = fig.add_subplot(self.nrows, self.ncols,
271 ax = fig.add_subplot(self.nrows, self.ncols,
272 n + 1, polar=self.polar)
272 n + 1, polar=self.polar)
273 ax.tick_params(labelsize=8)
273 ax.tick_params(labelsize=8)
274 ax.firsttime = True
274 ax.firsttime = True
275 ax.index = 0
275 ax.index = 0
276 ax.press = None
276 ax.press = None
277 self.axes.append(ax)
277 self.axes.append(ax)
278 if self.showprofile:
278 if self.showprofile:
279 cax = self.__add_axes(ax, size=size, pad=pad)
279 cax = self.__add_axes(ax, size=size, pad=pad)
280 cax.tick_params(labelsize=8)
280 cax.tick_params(labelsize=8)
281 self.pf_axes.append(cax)
281 self.pf_axes.append(cax)
282 else:
282 else:
283 if self.height is None:
283 if self.height is None:
284 self.height = 3
284 self.height = 3
285 for n in range(self.nplots):
285 for n in range(self.nplots):
286 fig = plt.figure(figsize=(self.width, self.height),
286 fig = plt.figure(figsize=(self.width, self.height),
287 edgecolor='k',
287 edgecolor='k',
288 facecolor='w')
288 facecolor='w')
289 ax = fig.add_subplot(1, 1, 1, polar=self.polar)
289 ax = fig.add_subplot(1, 1, 1, polar=self.polar)
290 ax.tick_params(labelsize=8)
290 ax.tick_params(labelsize=8)
291 ax.firsttime = True
291 ax.firsttime = True
292 ax.index = 0
292 ax.index = 0
293 ax.press = None
293 ax.press = None
294 self.figures.append(fig)
294 self.figures.append(fig)
295 self.axes.append(ax)
295 self.axes.append(ax)
296 if self.showprofile:
296 if self.showprofile:
297 cax = self.__add_axes(ax, size=size, pad=pad)
297 cax = self.__add_axes(ax, size=size, pad=pad)
298 cax.tick_params(labelsize=8)
298 cax.tick_params(labelsize=8)
299 self.pf_axes.append(cax)
299 self.pf_axes.append(cax)
300
300
301 for n in range(self.nrows):
301 for n in range(self.nrows):
302 if self.colormaps is not None:
302 if self.colormaps is not None:
303 cmap = plt.get_cmap(self.colormaps[n])
303 cmap = plt.get_cmap(self.colormaps[n])
304 else:
304 else:
305 cmap = plt.get_cmap(self.colormap)
305 cmap = plt.get_cmap(self.colormap)
306 cmap.set_bad(self.bgcolor, 1.)
306 cmap.set_bad(self.bgcolor, 1.)
307 self.cmaps.append(cmap)
307 self.cmaps.append(cmap)
308
308
309 for fig in self.figures:
309 for fig in self.figures:
310 fig.canvas.mpl_connect('key_press_event', self.OnKeyPress)
310 fig.canvas.mpl_connect('key_press_event', self.OnKeyPress)
311 fig.canvas.mpl_connect('scroll_event', self.OnBtnScroll)
311 fig.canvas.mpl_connect('scroll_event', self.OnBtnScroll)
312 fig.canvas.mpl_connect('button_press_event', self.onBtnPress)
312 fig.canvas.mpl_connect('button_press_event', self.onBtnPress)
313 fig.canvas.mpl_connect('motion_notify_event', self.onMotion)
313 fig.canvas.mpl_connect('motion_notify_event', self.onMotion)
314 fig.canvas.mpl_connect('button_release_event', self.onBtnRelease)
314 fig.canvas.mpl_connect('button_release_event', self.onBtnRelease)
315
315
316 def OnKeyPress(self, event):
316 def OnKeyPress(self, event):
317 '''
317 '''
318 Event for pressing keys (up, down) change colormap
318 Event for pressing keys (up, down) change colormap
319 '''
319 '''
320 ax = event.inaxes
320 ax = event.inaxes
321 if ax in self.axes:
321 if ax in self.axes:
322 if event.key == 'down':
322 if event.key == 'down':
323 ax.index += 1
323 ax.index += 1
324 elif event.key == 'up':
324 elif event.key == 'up':
325 ax.index -= 1
325 ax.index -= 1
326 if ax.index < 0:
326 if ax.index < 0:
327 ax.index = len(CMAPS) - 1
327 ax.index = len(CMAPS) - 1
328 elif ax.index == len(CMAPS):
328 elif ax.index == len(CMAPS):
329 ax.index = 0
329 ax.index = 0
330 cmap = CMAPS[ax.index]
330 cmap = CMAPS[ax.index]
331 ax.cbar.set_cmap(cmap)
331 ax.cbar.set_cmap(cmap)
332 ax.cbar.draw_all()
332 ax.cbar.draw_all()
333 ax.plt.set_cmap(cmap)
333 ax.plt.set_cmap(cmap)
334 ax.cbar.patch.figure.canvas.draw()
334 ax.cbar.patch.figure.canvas.draw()
335 self.colormap = cmap.name
335 self.colormap = cmap.name
336
336
337 def OnBtnScroll(self, event):
337 def OnBtnScroll(self, event):
338 '''
338 '''
339 Event for scrolling, scale figure
339 Event for scrolling, scale figure
340 '''
340 '''
341 cb_ax = event.inaxes
341 cb_ax = event.inaxes
342 if cb_ax in [ax.cbar.ax for ax in self.axes if ax.cbar]:
342 if cb_ax in [ax.cbar.ax for ax in self.axes if ax.cbar]:
343 ax = [ax for ax in self.axes if cb_ax == ax.cbar.ax][0]
343 ax = [ax for ax in self.axes if cb_ax == ax.cbar.ax][0]
344 pt = ax.cbar.ax.bbox.get_points()[:, 1]
344 pt = ax.cbar.ax.bbox.get_points()[:, 1]
345 nrm = ax.cbar.norm
345 nrm = ax.cbar.norm
346 vmin, vmax, p0, p1, pS = (
346 vmin, vmax, p0, p1, pS = (
347 nrm.vmin, nrm.vmax, pt[0], pt[1], event.y)
347 nrm.vmin, nrm.vmax, pt[0], pt[1], event.y)
348 scale = 2 if event.step == 1 else 0.5
348 scale = 2 if event.step == 1 else 0.5
349 point = vmin + (vmax - vmin) / (p1 - p0) * (pS - p0)
349 point = vmin + (vmax - vmin) / (p1 - p0) * (pS - p0)
350 ax.cbar.norm.vmin = point - scale * (point - vmin)
350 ax.cbar.norm.vmin = point - scale * (point - vmin)
351 ax.cbar.norm.vmax = point - scale * (point - vmax)
351 ax.cbar.norm.vmax = point - scale * (point - vmax)
352 ax.plt.set_norm(ax.cbar.norm)
352 ax.plt.set_norm(ax.cbar.norm)
353 ax.cbar.draw_all()
353 ax.cbar.draw_all()
354 ax.cbar.patch.figure.canvas.draw()
354 ax.cbar.patch.figure.canvas.draw()
355
355
356 def onBtnPress(self, event):
356 def onBtnPress(self, event):
357 '''
357 '''
358 Event for mouse button press
358 Event for mouse button press
359 '''
359 '''
360 cb_ax = event.inaxes
360 cb_ax = event.inaxes
361 if cb_ax is None:
361 if cb_ax is None:
362 return
362 return
363
363
364 if cb_ax in [ax.cbar.ax for ax in self.axes if ax.cbar]:
364 if cb_ax in [ax.cbar.ax for ax in self.axes if ax.cbar]:
365 cb_ax.press = event.x, event.y
365 cb_ax.press = event.x, event.y
366 else:
366 else:
367 cb_ax.press = None
367 cb_ax.press = None
368
368
369 def onMotion(self, event):
369 def onMotion(self, event):
370 '''
370 '''
371 Event for move inside colorbar
371 Event for move inside colorbar
372 '''
372 '''
373 cb_ax = event.inaxes
373 cb_ax = event.inaxes
374 if cb_ax is None:
374 if cb_ax is None:
375 return
375 return
376 if cb_ax not in [ax.cbar.ax for ax in self.axes if ax.cbar]:
376 if cb_ax not in [ax.cbar.ax for ax in self.axes if ax.cbar]:
377 return
377 return
378 if cb_ax.press is None:
378 if cb_ax.press is None:
379 return
379 return
380
380
381 ax = [ax for ax in self.axes if cb_ax == ax.cbar.ax][0]
381 ax = [ax for ax in self.axes if cb_ax == ax.cbar.ax][0]
382 xprev, yprev = cb_ax.press
382 xprev, yprev = cb_ax.press
383 dx = event.x - xprev
383 dx = event.x - xprev
384 dy = event.y - yprev
384 dy = event.y - yprev
385 cb_ax.press = event.x, event.y
385 cb_ax.press = event.x, event.y
386 scale = ax.cbar.norm.vmax - ax.cbar.norm.vmin
386 scale = ax.cbar.norm.vmax - ax.cbar.norm.vmin
387 perc = 0.03
387 perc = 0.03
388
388
389 if event.button == 1:
389 if event.button == 1:
390 ax.cbar.norm.vmin -= (perc * scale) * numpy.sign(dy)
390 ax.cbar.norm.vmin -= (perc * scale) * numpy.sign(dy)
391 ax.cbar.norm.vmax -= (perc * scale) * numpy.sign(dy)
391 ax.cbar.norm.vmax -= (perc * scale) * numpy.sign(dy)
392 elif event.button == 3:
392 elif event.button == 3:
393 ax.cbar.norm.vmin -= (perc * scale) * numpy.sign(dy)
393 ax.cbar.norm.vmin -= (perc * scale) * numpy.sign(dy)
394 ax.cbar.norm.vmax += (perc * scale) * numpy.sign(dy)
394 ax.cbar.norm.vmax += (perc * scale) * numpy.sign(dy)
395
395
396 ax.cbar.draw_all()
396 ax.cbar.draw_all()
397 ax.plt.set_norm(ax.cbar.norm)
397 ax.plt.set_norm(ax.cbar.norm)
398 ax.cbar.patch.figure.canvas.draw()
398 ax.cbar.patch.figure.canvas.draw()
399
399
400 def onBtnRelease(self, event):
400 def onBtnRelease(self, event):
401 '''
401 '''
402 Event for mouse button release
402 Event for mouse button release
403 '''
403 '''
404 cb_ax = event.inaxes
404 cb_ax = event.inaxes
405 if cb_ax is not None:
405 if cb_ax is not None:
406 cb_ax.press = None
406 cb_ax.press = None
407
407
408 def __add_axes(self, ax, size='30%', pad='8%'):
408 def __add_axes(self, ax, size='30%', pad='8%'):
409 '''
409 '''
410 Add new axes to the given figure
410 Add new axes to the given figure
411 '''
411 '''
412 divider = make_axes_locatable(ax)
412 divider = make_axes_locatable(ax)
413 nax = divider.new_horizontal(size=size, pad=pad)
413 nax = divider.new_horizontal(size=size, pad=pad)
414 ax.figure.add_axes(nax)
414 ax.figure.add_axes(nax)
415 return nax
415 return nax
416
416
417 def fill_gaps(self, x_buffer, y_buffer, z_buffer):
417 def fill_gaps(self, x_buffer, y_buffer, z_buffer):
418 '''
418 '''
419 Create a masked array for missing data
419 Create a masked array for missing data
420 '''
420 '''
421 if x_buffer.shape[0] < 2:
421 if x_buffer.shape[0] < 2:
422 return x_buffer, y_buffer, z_buffer
422 return x_buffer, y_buffer, z_buffer
423
423
424 deltas = x_buffer[1:] - x_buffer[0:-1]
424 deltas = x_buffer[1:] - x_buffer[0:-1]
425 x_median = numpy.median(deltas)
425 x_median = numpy.median(deltas)
426
426
427 index = numpy.where(deltas > 5 * x_median)
427 index = numpy.where(deltas > 5 * x_median)
428
428
429 if len(index[0]) != 0:
429 if len(index[0]) != 0:
430 z_buffer[::, index[0], ::] = self.__missing
430 z_buffer[::, index[0], ::] = self.__missing
431 z_buffer = numpy.ma.masked_inside(z_buffer,
431 z_buffer = numpy.ma.masked_inside(z_buffer,
432 0.99 * self.__missing,
432 0.99 * self.__missing,
433 1.01 * self.__missing)
433 1.01 * self.__missing)
434
434
435 return x_buffer, y_buffer, z_buffer
435 return x_buffer, y_buffer, z_buffer
436
436
437 def decimate(self):
437 def decimate(self):
438
438
439 # dx = int(len(self.x)/self.__MAXNUMX) + 1
439 # dx = int(len(self.x)/self.__MAXNUMX) + 1
440 dy = int(len(self.y) / self.decimation) + 1
440 dy = int(len(self.y) / self.decimation) + 1
441
441
442 # x = self.x[::dx]
442 # x = self.x[::dx]
443 x = self.x
443 x = self.x
444 y = self.y[::dy]
444 y = self.y[::dy]
445 z = self.z[::, ::, ::dy]
445 z = self.z[::, ::, ::dy]
446
446
447 return x, y, z
447 return x, y, z
448
448
449 def format(self):
449 def format(self):
450 '''
450 '''
451 Set min and max values, labels, ticks and titles
451 Set min and max values, labels, ticks and titles
452 '''
452 '''
453
453
454 if self.xmin is None:
454 if self.xmin is None:
455 xmin = self.data.min_time
455 xmin = self.data.min_time
456 else:
456 else:
457 if self.xaxis is 'time':
457 if self.xaxis is 'time':
458 dt = self.getDateTime(self.data.min_time)
458 dt = self.getDateTime(self.data.min_time)
459 xmin = (dt.replace(hour=int(self.xmin), minute=0, second=0) -
459 xmin = (dt.replace(hour=int(self.xmin), minute=0, second=0) -
460 datetime.datetime(1970, 1, 1)).total_seconds()
460 datetime.datetime(1970, 1, 1)).total_seconds()
461 if self.data.localtime:
461 if self.data.localtime:
462 xmin += time.timezone
462 xmin += time.timezone
463 else:
463 else:
464 xmin = self.xmin
464 xmin = self.xmin
465
465
466 if self.xmax is None:
466 if self.xmax is None:
467 xmax = xmin + self.xrange * 60 * 60
467 xmax = xmin + self.xrange * 60 * 60
468 else:
468 else:
469 if self.xaxis is 'time':
469 if self.xaxis is 'time':
470 dt = self.getDateTime(self.data.max_time)
470 dt = self.getDateTime(self.data.max_time)
471 xmax = (dt.replace(hour=int(self.xmax), minute=59, second=59) -
471 xmax = (dt.replace(hour=int(self.xmax), minute=59, second=59) -
472 datetime.datetime(1970, 1, 1) + datetime.timedelta(seconds=1)).total_seconds()
472 datetime.datetime(1970, 1, 1) + datetime.timedelta(seconds=1)).total_seconds()
473 if self.data.localtime:
473 if self.data.localtime:
474 xmax += time.timezone
474 xmax += time.timezone
475 else:
475 else:
476 xmax = self.xmax
476 xmax = self.xmax
477
477
478 ymin = self.ymin if self.ymin else numpy.nanmin(self.y)
478 ymin = self.ymin if self.ymin else numpy.nanmin(self.y)
479 ymax = self.ymax if self.ymax else numpy.nanmax(self.y)
479 ymax = self.ymax if self.ymax else numpy.nanmax(self.y)
480 #Y = numpy.array([1, 2, 5, 10, 20, 50, 100, 200, 500, 1000, 2000, 5000, 10000, 20000, 50000])
480 #Y = numpy.array([1, 2, 5, 10, 20, 50, 100, 200, 500, 1000, 2000, 5000, 10000, 20000, 50000])
481
481
482 #i = 1 if numpy.where(
482 #i = 1 if numpy.where(
483 # abs(ymax-ymin) <= Y)[0][0] < 0 else numpy.where(abs(ymax-ymin) <= Y)[0][0]
483 # abs(ymax-ymin) <= Y)[0][0] < 0 else numpy.where(abs(ymax-ymin) <= Y)[0][0]
484 #ystep = Y[i] / 10.
484 #ystep = Y[i] / 10.
485 dig = int(numpy.log10(ymax))
485 dig = int(numpy.log10(ymax))
486 if dig == 0:
486 if dig == 0:
487 digD = len(str(ymax)) - 2
487 digD = len(str(ymax)) - 2
488 ydec = ymax*(10**digD)
488 ydec = ymax*(10**digD)
489
489
490 dig = int(numpy.log10(ydec))
490 dig = int(numpy.log10(ydec))
491 ystep = ((ydec + (10**(dig)))//10**(dig))*(10**(dig))
491 ystep = ((ydec + (10**(dig)))//10**(dig))*(10**(dig))
492 ystep = ystep/5
492 ystep = ystep/5
493 ystep = ystep/(10**digD)
493 ystep = ystep/(10**digD)
494
494
495 else:
495 else:
496 ystep = ((ymax + (10**(dig)))//10**(dig))*(10**(dig))
496 ystep = ((ymax + (10**(dig)))//10**(dig))*(10**(dig))
497 ystep = ystep/5
497 ystep = ystep/5
498
498
499 if self.xaxis is not 'time':
499 if self.xaxis is not 'time':
500
500
501 dig = int(numpy.log10(xmax))
501 dig = int(numpy.log10(xmax))
502
502
503 if dig <= 0:
503 if dig <= 0:
504 digD = len(str(xmax)) - 2
504 digD = len(str(xmax)) - 2
505 xdec = xmax*(10**digD)
505 xdec = xmax*(10**digD)
506
506
507 dig = int(numpy.log10(xdec))
507 dig = int(numpy.log10(xdec))
508 xstep = ((xdec + (10**(dig)))//10**(dig))*(10**(dig))
508 xstep = ((xdec + (10**(dig)))//10**(dig))*(10**(dig))
509 xstep = xstep*0.5
509 xstep = xstep*0.5
510 xstep = xstep/(10**digD)
510 xstep = xstep/(10**digD)
511
511
512 else:
512 else:
513 xstep = ((xmax + (10**(dig)))//10**(dig))*(10**(dig))
513 xstep = ((xmax + (10**(dig)))//10**(dig))*(10**(dig))
514 xstep = xstep/5
514 xstep = xstep/5
515
515
516 for n, ax in enumerate(self.axes):
516 for n, ax in enumerate(self.axes):
517 if ax.firsttime:
517 if ax.firsttime:
518 ax.set_facecolor(self.bgcolor)
518 ax.set_facecolor(self.bgcolor)
519 ax.yaxis.set_major_locator(MultipleLocator(ystep))
519 ax.yaxis.set_major_locator(MultipleLocator(ystep))
520 if self.xscale:
520 if self.xscale:
521 ax.xaxis.set_major_formatter(FuncFormatter(
521 ax.xaxis.set_major_formatter(FuncFormatter(
522 lambda x, pos: '{0:g}'.format(x*self.xscale)))
522 lambda x, pos: '{0:g}'.format(x*self.xscale)))
523 if self.xscale:
523 if self.xscale:
524 ax.yaxis.set_major_formatter(FuncFormatter(
524 ax.yaxis.set_major_formatter(FuncFormatter(
525 lambda x, pos: '{0:g}'.format(x*self.yscale)))
525 lambda x, pos: '{0:g}'.format(x*self.yscale)))
526 if self.xaxis is 'time':
526 if self.xaxis is 'time':
527 ax.xaxis.set_major_formatter(FuncFormatter(self.__fmtTime))
527 ax.xaxis.set_major_formatter(FuncFormatter(self.__fmtTime))
528 ax.xaxis.set_major_locator(LinearLocator(9))
528 ax.xaxis.set_major_locator(LinearLocator(9))
529 else:
529 else:
530 ax.xaxis.set_major_locator(MultipleLocator(xstep))
530 ax.xaxis.set_major_locator(MultipleLocator(xstep))
531 if self.xlabel is not None:
531 if self.xlabel is not None:
532 ax.set_xlabel(self.xlabel)
532 ax.set_xlabel(self.xlabel)
533 ax.set_ylabel(self.ylabel)
533 ax.set_ylabel(self.ylabel)
534 ax.firsttime = False
534 ax.firsttime = False
535 if self.showprofile:
535 if self.showprofile:
536 self.pf_axes[n].set_ylim(ymin, ymax)
536 self.pf_axes[n].set_ylim(ymin, ymax)
537 self.pf_axes[n].set_xlim(self.zmin, self.zmax)
537 self.pf_axes[n].set_xlim(self.zmin, self.zmax)
538 self.pf_axes[n].set_xlabel('dB')
538 self.pf_axes[n].set_xlabel('dB')
539 self.pf_axes[n].grid(b=True, axis='x')
539 self.pf_axes[n].grid(b=True, axis='x')
540 [tick.set_visible(False)
540 [tick.set_visible(False)
541 for tick in self.pf_axes[n].get_yticklabels()]
541 for tick in self.pf_axes[n].get_yticklabels()]
542 if self.colorbar:
542 if self.colorbar:
543 ax.cbar = plt.colorbar(
543 ax.cbar = plt.colorbar(
544 ax.plt, ax=ax, fraction=0.05, pad=0.02, aspect=10)
544 ax.plt, ax=ax, fraction=0.05, pad=0.02, aspect=10)
545 ax.cbar.ax.tick_params(labelsize=8)
545 ax.cbar.ax.tick_params(labelsize=8)
546 ax.cbar.ax.press = None
546 ax.cbar.ax.press = None
547 if self.cb_label:
547 if self.cb_label:
548 ax.cbar.set_label(self.cb_label, size=8)
548 ax.cbar.set_label(self.cb_label, size=8)
549 elif self.cb_labels:
549 elif self.cb_labels:
550 ax.cbar.set_label(self.cb_labels[n], size=8)
550 ax.cbar.set_label(self.cb_labels[n], size=8)
551 else:
551 else:
552 ax.cbar = None
552 ax.cbar = None
553 if self.grid:
553 if self.grid:
554 ax.grid(True)
554 ax.grid(True)
555
555
556 if not self.polar:
556 if not self.polar:
557 ax.set_xlim(xmin, xmax)
557 ax.set_xlim(xmin, xmax)
558 ax.set_ylim(ymin, ymax)
558 ax.set_ylim(ymin, ymax)
559 ax.set_title('{} {} {}'.format(
559 ax.set_title('{} {} {}'.format(
560 self.titles[n],
560 self.titles[n],
561 self.getDateTime(self.data.max_time).strftime(
561 self.getDateTime(self.data.max_time).strftime(
562 '%Y-%m-%d %H:%M:%S'),
562 '%Y-%m-%d %H:%M:%S'),
563 self.time_label),
563 self.time_label),
564 size=8)
564 size=8)
565 else:
565 else:
566 ax.set_title('{}'.format(self.titles[n]), size=8)
566 ax.set_title('{}'.format(self.titles[n]), size=8)
567 ax.set_ylim(0, 90)
567 ax.set_ylim(0, 90)
568 ax.set_yticks(numpy.arange(0, 90, 20))
568 ax.set_yticks(numpy.arange(0, 90, 20))
569 ax.yaxis.labelpad = 40
569 ax.yaxis.labelpad = 40
570
570
571 def clear_figures(self):
571 def clear_figures(self):
572 '''
572 '''
573 Reset axes for redraw plots
573 Reset axes for redraw plots
574 '''
574 '''
575
575
576 for ax in self.axes:
576 for ax in self.axes:
577 ax.clear()
577 ax.clear()
578 ax.firsttime = True
578 ax.firsttime = True
579 if ax.cbar:
579 if ax.cbar:
580 ax.cbar.remove()
580 ax.cbar.remove()
581
581
582 def __plot(self):
582 def __plot(self):
583 '''
583 '''
584 Main function to plot, format and save figures
584 Main function to plot, format and save figures
585 '''
585 '''
586
586
587 try:
587 try:
588 self.plot()
588 self.plot()
589 self.format()
589 self.format()
590 except Exception as e:
590 except Exception as e:
591 log.warning('{} Plot could not be updated... check data'.format(
591 log.warning('{} Plot could not be updated... check data'.format(
592 self.CODE), self.name)
592 self.CODE), self.name)
593 log.error(str(e), '')
593 log.error(str(e), '')
594 return
594 return
595
595
596 for n, fig in enumerate(self.figures):
596 for n, fig in enumerate(self.figures):
597 if self.nrows == 0 or self.nplots == 0:
597 if self.nrows == 0 or self.nplots == 0:
598 log.warning('No data', self.name)
598 log.warning('No data', self.name)
599 fig.text(0.5, 0.5, 'No Data', fontsize='large', ha='center')
599 fig.text(0.5, 0.5, 'No Data', fontsize='large', ha='center')
600 fig.canvas.manager.set_window_title(self.CODE)
600 fig.canvas.manager.set_window_title(self.CODE)
601 continue
601 continue
602
602
603 fig.tight_layout()
603 fig.tight_layout()
604 fig.canvas.manager.set_window_title('{} - {}'.format(self.title,
604 fig.canvas.manager.set_window_title('{} - {}'.format(self.title,
605 self.getDateTime(self.data.max_time).strftime('%Y/%m/%d')))
605 self.getDateTime(self.data.max_time).strftime('%Y/%m/%d')))
606 fig.canvas.draw()
606 fig.canvas.draw()
607 if self.show:
607 if self.show:
608 fig.show()
608 fig.show()
609 figpause(0.1)
609 figpause(0.1)
610
610
611 if self.save:
611 if self.save:
612 self.save_figure(n)
612 self.save_figure(n)
613
613
614 if self.plot_server:
614 if self.plot_server:
615 self.send_to_server()
615 self.send_to_server()
616 # t = Thread(target=self.send_to_server)
616 # t = Thread(target=self.send_to_server)
617 # t.start()
617 # t.start()
618
618
619 def save_figure(self, n):
619 def save_figure(self, n):
620 '''
620 '''
621 '''
621 '''
622
622
623 if self.save_counter < self.save_period:
623 if self.save_counter < self.save_period:
624 self.save_counter += 1
624 self.save_counter += 1
625 return
625 return
626
626
627 self.save_counter = 1
627 self.save_counter = 1
628
628
629 fig = self.figures[n]
629 fig = self.figures[n]
630
630
631 if self.save_labels:
631 if self.save_labels:
632 labels = self.save_labels
632 labels = self.save_labels
633 else:
633 else:
634 labels = list(range(self.nrows))
634 labels = list(range(self.nrows))
635
635
636 if self.oneFigure:
636 if self.oneFigure:
637 label = ''
637 label = ''
638 else:
638 else:
639 label = '-{}'.format(labels[n])
639 label = '-{}'.format(labels[n])
640 figname = os.path.join(
640 figname = os.path.join(
641 self.save,
641 self.save,
642 self.CODE,
642 self.CODE,
643 '{}{}_{}.png'.format(
643 '{}{}_{}.png'.format(
644 self.CODE,
644 self.CODE,
645 label,
645 label,
646 self.getDateTime(self.data.max_time).strftime(
646 self.getDateTime(self.data.max_time).strftime('%Y%m%d_%H%M%S'),
647 '%Y%m%d_%H%M%S'
648 ),
649 )
647 )
650 )
648 )
649
651 log.log('Saving figure: {}'.format(figname), self.name)
650 log.log('Saving figure: {}'.format(figname), self.name)
652 if not os.path.isdir(os.path.dirname(figname)):
651 if not os.path.isdir(os.path.dirname(figname)):
653 os.makedirs(os.path.dirname(figname))
652 os.makedirs(os.path.dirname(figname))
654 fig.savefig(figname)
653 fig.savefig(figname)
655
654
656 if self.realtime:
655 if self.realtime:
657 figname = os.path.join(
656 figname = os.path.join(
658 self.save,
657 self.save,
659 '{}{}_{}.png'.format(
658 '{}{}_{}.png'.format(
660 self.CODE,
659 self.CODE,
661 label,
660 label,
662 self.getDateTime(self.data.min_time).strftime(
661 self.getDateTime(self.data.min_time).strftime(
663 '%Y%m%d'
662 '%Y%m%d'
664 ),
663 ),
665 )
664 )
666 )
665 )
667 fig.savefig(figname)
666 fig.savefig(figname)
668
667
669 def send_to_server(self):
668 def send_to_server(self):
670 '''
669 '''
671 '''
670 '''
672
671
673 if self.sender_counter < self.sender_period:
672 if self.sender_counter < self.sender_period:
674 self.sender_counter += 1
673 self.sender_counter += 1
675 return
674 return
676
675
677 self.sender_counter = 1
676 self.sender_counter = 1
678 self.data.meta['titles'] = self.titles
677 self.data.meta['titles'] = self.titles
679 retries = 2
678 retries = 2
680 while True:
679 while True:
681 self.socket.send_string(self.data.jsonify(self.plot_name, self.plot_type))
680 self.socket.send_string(self.data.jsonify(self.plot_name, self.plot_type))
682 socks = dict(self.poll.poll(5000))
681 socks = dict(self.poll.poll(5000))
683 if socks.get(self.socket) == zmq.POLLIN:
682 if socks.get(self.socket) == zmq.POLLIN:
684 reply = self.socket.recv_string()
683 reply = self.socket.recv_string()
685 if reply == 'ok':
684 if reply == 'ok':
686 log.log("Response from server ok", self.name)
685 log.log("Response from server ok", self.name)
687 break
686 break
688 else:
687 else:
689 log.warning(
688 log.warning(
690 "Malformed reply from server: {}".format(reply), self.name)
689 "Malformed reply from server: {}".format(reply), self.name)
691
690
692 else:
691 else:
693 log.warning(
692 log.warning(
694 "No response from server, retrying...", self.name)
693 "No response from server, retrying...", self.name)
695 self.socket.setsockopt(zmq.LINGER, 0)
694 self.socket.setsockopt(zmq.LINGER, 0)
696 self.socket.close()
695 self.socket.close()
697 self.poll.unregister(self.socket)
696 self.poll.unregister(self.socket)
698 retries -= 1
697 retries -= 1
699 if retries == 0:
698 if retries == 0:
700 log.error(
699 log.error(
701 "Server seems to be offline, abandoning", self.name)
700 "Server seems to be offline, abandoning", self.name)
702 self.socket = self.context.socket(zmq.REQ)
701 self.socket = self.context.socket(zmq.REQ)
703 self.socket.connect(self.plot_server)
702 self.socket.connect(self.plot_server)
704 self.poll.register(self.socket, zmq.POLLIN)
703 self.poll.register(self.socket, zmq.POLLIN)
705 time.sleep(1)
704 time.sleep(1)
706 break
705 break
707 self.socket = self.context.socket(zmq.REQ)
706 self.socket = self.context.socket(zmq.REQ)
708 self.socket.connect(self.plot_server)
707 self.socket.connect(self.plot_server)
709 self.poll.register(self.socket, zmq.POLLIN)
708 self.poll.register(self.socket, zmq.POLLIN)
710 time.sleep(0.5)
709 time.sleep(0.5)
711
710
712 def setup(self):
711 def setup(self):
713 '''
712 '''
714 This method should be implemented in the child class, the following
713 This method should be implemented in the child class, the following
715 attributes should be set:
714 attributes should be set:
716
715
717 self.nrows: number of rows
716 self.nrows: number of rows
718 self.ncols: number of cols
717 self.ncols: number of cols
719 self.nplots: number of plots (channels or pairs)
718 self.nplots: number of plots (channels or pairs)
720 self.ylabel: label for Y axes
719 self.ylabel: label for Y axes
721 self.titles: list of axes title
720 self.titles: list of axes title
722
721
723 '''
722 '''
724 raise NotImplementedError
723 raise NotImplementedError
725
724
726 def plot(self):
725 def plot(self):
727 '''
726 '''
728 Must be defined in the child class
727 Must be defined in the child class
729 '''
728 '''
730 raise NotImplementedError
729 raise NotImplementedError
731
730
732 def run(self, dataOut, **kwargs):
731 def run(self, dataOut, **kwargs):
733 '''
732 '''
734 Main plotting routine
733 Main plotting routine
735 '''
734 '''
736
735
737 if self.isConfig is False:
736 if self.isConfig is False:
738 self.__setup(**kwargs)
737 self.__setup(**kwargs)
739 if dataOut.type == 'Parameters':
738 if dataOut.type == 'Parameters':
740 t = dataOut.utctimeInit
739 t = dataOut.utctimeInit
741 else:
740 else:
742 t = dataOut.utctime
741 t = dataOut.utctime
743
742
744 if dataOut.useLocalTime:
743 if dataOut.useLocalTime:
745 self.getDateTime = datetime.datetime.fromtimestamp
744 self.getDateTime = datetime.datetime.fromtimestamp
746 if not self.localtime:
745 if not self.localtime:
747 t += time.timezone
746 t += time.timezone
748 else:
747 else:
749 self.getDateTime = datetime.datetime.utcfromtimestamp
748 self.getDateTime = datetime.datetime.utcfromtimestamp
750 if self.localtime:
749 if self.localtime:
751 t -= time.timezone
750 t -= time.timezone
752
751
753 if 'buffer' in self.plot_type:
752 if 'buffer' in self.plot_type:
754 if self.xmin is None:
753 if self.xmin is None:
755 self.tmin = t
754 self.tmin = t
756 else:
755 else:
757 self.tmin = (
756 self.tmin = (
758 self.getDateTime(t).replace(
757 self.getDateTime(t).replace(
759 hour=self.xmin,
758 hour=self.xmin,
760 minute=0,
759 minute=0,
761 second=0) - self.getDateTime(0)).total_seconds()
760 second=0) - self.getDateTime(0)).total_seconds()
762
761
763 self.data.setup()
762 self.data.setup()
764 self.isConfig = True
763 self.isConfig = True
765 if self.plot_server:
764 if self.plot_server:
766 self.context = zmq.Context()
765 self.context = zmq.Context()
767 self.socket = self.context.socket(zmq.REQ)
766 self.socket = self.context.socket(zmq.REQ)
768 self.socket.connect(self.plot_server)
767 self.socket.connect(self.plot_server)
769 self.poll = zmq.Poller()
768 self.poll = zmq.Poller()
770 self.poll.register(self.socket, zmq.POLLIN)
769 self.poll.register(self.socket, zmq.POLLIN)
771
770
772 if dataOut.type == 'Parameters':
771 if dataOut.type == 'Parameters':
773 tm = dataOut.utctimeInit
772 tm = dataOut.utctimeInit
774 else:
773 else:
775 tm = dataOut.utctime
774 tm = dataOut.utctime
776
775
777 if not dataOut.useLocalTime and self.localtime:
776 if not dataOut.useLocalTime and self.localtime:
778 tm -= time.timezone
777 tm -= time.timezone
779 if dataOut.useLocalTime and not self.localtime:
778 if dataOut.useLocalTime and not self.localtime:
780 tm += time.timezone
779 tm += time.timezone
781
780
782 if self.xaxis is 'time' and self.data and (tm - self.tmin) >= self.xrange*60*60:
781 if self.xaxis is 'time' and self.data and (tm - self.tmin) >= self.xrange*60*60:
783 self.save_counter = self.save_period
782 self.save_counter = self.save_period
784 self.__plot()
783 self.__plot()
785 self.xmin += self.xrange
784 self.xmin += self.xrange
786 if self.xmin >= 24:
785 if self.xmin >= 24:
787 self.xmin -= 24
786 self.xmin -= 24
788 self.tmin += self.xrange*60*60
787 self.tmin += self.xrange*60*60
789 self.data.setup()
788 self.data.setup()
790 self.clear_figures()
789 self.clear_figures()
791
790
792 self.data.update(dataOut, tm)
791 self.data.update(dataOut, tm)
793
792
794 if self.isPlotConfig is False:
793 if self.isPlotConfig is False:
795 self.__setup_plot()
794 self.__setup_plot()
796 self.isPlotConfig = True
795 self.isPlotConfig = True
797
796
798 if self.realtime:
797 if self.realtime:
799 self.__plot()
798 self.__plot()
800 else:
799 else:
801 self.__throttle_plot(self.__plot)#, coerce=coerce)
800 self.__throttle_plot(self.__plot)#, coerce=coerce)
802
801
803 def close(self):
802 def close(self):
804
803
805 if self.data:
804 if self.data:
806 self.save_counter = self.save_period
805 self.save_counter = self.save_period
807 self.__plot()
806 self.__plot()
808 if self.data and self.pause:
807 if self.data and self.pause:
809 figpause(10)
808 figpause(10)
810
@@ -1,629 +1,649
1 '''
1 '''
2 Created on Set 9, 2015
2 Created on Set 9, 2015
3
3
4 @author: roj-idl71 Karim Kuyeng
4 @author: roj-idl71 Karim Kuyeng
5 '''
5 '''
6
6
7 import os
7 import os
8 import sys
8 import sys
9 import glob
9 import glob
10 import fnmatch
10 import fnmatch
11 import datetime
11 import datetime
12 import time
12 import time
13 import re
13 import re
14 import h5py
14 import h5py
15 import numpy
15 import numpy
16
16
17 try:
17 try:
18 from gevent import sleep
18 from gevent import sleep
19 except:
19 except:
20 from time import sleep
20 from time import sleep
21
21
22 from schainpy.model.data.jroheaderIO import RadarControllerHeader, SystemHeader
22 from schainpy.model.data.jroheaderIO import RadarControllerHeader, SystemHeader
23 from schainpy.model.data.jrodata import Voltage
23 from schainpy.model.data.jrodata import Voltage
24 from schainpy.model.proc.jroproc_base import ProcessingUnit, Operation
24 from schainpy.model.proc.jroproc_base import ProcessingUnit, Operation, MPDecorator
25 from numpy import imag
25 from numpy import imag
26
26
27 @MPDecorator
27 class AMISRReader(ProcessingUnit):
28 class AMISRReader(ProcessingUnit):
28 '''
29 '''
29 classdocs
30 classdocs
30 '''
31 '''
31
32
32 def __init__(self):
33 def __init__(self):
33 '''
34 '''
34 Constructor
35 Constructor
35 '''
36 '''
36
37
37 ProcessingUnit.__init__(self)
38 ProcessingUnit.__init__(self)
38
39
39 self.set = None
40 self.set = None
40 self.subset = None
41 self.subset = None
41 self.extension_file = '.h5'
42 self.extension_file = '.h5'
42 self.dtc_str = 'dtc'
43 self.dtc_str = 'dtc'
43 self.dtc_id = 0
44 self.dtc_id = 0
44 self.status = True
45 self.status = True
45 self.isConfig = False
46 self.isConfig = False
46 self.dirnameList = []
47 self.dirnameList = []
47 self.filenameList = []
48 self.filenameList = []
48 self.fileIndex = None
49 self.fileIndex = None
49 self.flagNoMoreFiles = False
50 self.flagNoMoreFiles = False
50 self.flagIsNewFile = 0
51 self.flagIsNewFile = 0
51 self.filename = ''
52 self.filename = ''
52 self.amisrFilePointer = None
53 self.amisrFilePointer = None
53
54
54
55
55 self.dataset = None
56 #self.dataset = None
56
57
57
58
58
59
59
60
60 self.profileIndex = 0
61 self.profileIndex = 0
61
62
62
63
63 self.beamCodeByFrame = None
64 self.beamCodeByFrame = None
64 self.radacTimeByFrame = None
65 self.radacTimeByFrame = None
65
66
66 self.dataset = None
67 self.dataset = None
67
68
68
69
69
70
70
71
71 self.__firstFile = True
72 self.__firstFile = True
72
73
73 self.buffer = None
74 self.buffer = None
74
75
75
76
76 self.timezone = 'ut'
77 self.timezone = 'ut'
77
78
78 self.__waitForNewFile = 20
79 self.__waitForNewFile = 20
79 self.__filename_online = None
80 self.__filename_online = None
80 #Is really necessary create the output object in the initializer
81 #Is really necessary create the output object in the initializer
81 self.dataOut = Voltage()
82 self.dataOut = Voltage()
83 self.dataOut.error=False
82
84
83 def setup(self,path=None,
85 def setup(self,path=None,
84 startDate=None,
86 startDate=None,
85 endDate=None,
87 endDate=None,
86 startTime=None,
88 startTime=None,
87 endTime=None,
89 endTime=None,
88 walk=True,
90 walk=True,
89 timezone='ut',
91 timezone='ut',
90 all=0,
92 all=0,
91 code = None,
93 code = None,
92 nCode = 0,
94 nCode = 0,
93 nBaud = 0,
95 nBaud = 0,
94 online=False):
96 online=False):
95
97
98 #print ("T",path)
99
96 self.timezone = timezone
100 self.timezone = timezone
97 self.all = all
101 self.all = all
98 self.online = online
102 self.online = online
99
103
100 self.code = code
104 self.code = code
101 self.nCode = int(nCode)
105 self.nCode = int(nCode)
102 self.nBaud = int(nBaud)
106 self.nBaud = int(nBaud)
103
107
104
108
105
109
106 #self.findFiles()
110 #self.findFiles()
107 if not(online):
111 if not(online):
108 #Busqueda de archivos offline
112 #Busqueda de archivos offline
109 self.searchFilesOffLine(path, startDate, endDate, startTime, endTime, walk)
113 self.searchFilesOffLine(path, startDate, endDate, startTime, endTime, walk)
110 else:
114 else:
111 self.searchFilesOnLine(path, startDate, endDate, startTime,endTime,walk)
115 self.searchFilesOnLine(path, startDate, endDate, startTime,endTime,walk)
112
116
113 if not(self.filenameList):
117 if not(self.filenameList):
114 print("There is no files into the folder: %s"%(path))
118 print("There is no files into the folder: %s"%(path))
115
116 sys.exit(-1)
119 sys.exit(-1)
117
120
118 self.fileIndex = -1
121 self.fileIndex = -1
119
122
120 self.readNextFile(online)
123 self.readNextFile(online)
121
124
122 '''
125 '''
123 Add code
126 Add code
124 '''
127 '''
125 self.isConfig = True
128 self.isConfig = True
126
129
127 pass
130 pass
128
131
129
132
130 def readAMISRHeader(self,fp):
133 def readAMISRHeader(self,fp):
131 header = 'Raw11/Data/RadacHeader'
134 header = 'Raw11/Data/RadacHeader'
132 self.beamCodeByPulse = fp.get(header+'/BeamCode') # LIST OF BEAMS PER PROFILE, TO BE USED ON REARRANGE
135 self.beamCodeByPulse = fp.get(header+'/BeamCode') # LIST OF BEAMS PER PROFILE, TO BE USED ON REARRANGE
133 self.beamCode = fp.get('Raw11/Data/Beamcodes') # NUMBER OF CHANNELS AND IDENTIFY POSITION TO CREATE A FILE WITH THAT INFO
136 self.beamCode = fp.get('Raw11/Data/Beamcodes') # NUMBER OF CHANNELS AND IDENTIFY POSITION TO CREATE A FILE WITH THAT INFO
134 #self.code = fp.get(header+'/Code') # NOT USE FOR THIS
137 #self.code = fp.get(header+'/Code') # NOT USE FOR THIS
135 self.frameCount = fp.get(header+'/FrameCount')# NOT USE FOR THIS
138 self.frameCount = fp.get(header+'/FrameCount')# NOT USE FOR THIS
136 self.modeGroup = fp.get(header+'/ModeGroup')# NOT USE FOR THIS
139 self.modeGroup = fp.get(header+'/ModeGroup')# NOT USE FOR THIS
137 self.nsamplesPulse = fp.get(header+'/NSamplesPulse')# TO GET NSA OR USING DATA FOR THAT
140 self.nsamplesPulse = fp.get(header+'/NSamplesPulse')# TO GET NSA OR USING DATA FOR THAT
138 self.pulseCount = fp.get(header+'/PulseCount')# NOT USE FOR THIS
141 self.pulseCount = fp.get(header+'/PulseCount')# NOT USE FOR THIS
139 self.radacTime = fp.get(header+'/RadacTime')# 1st TIME ON FILE ANDE CALCULATE THE REST WITH IPP*nindexprofile
142 self.radacTime = fp.get(header+'/RadacTime')# 1st TIME ON FILE ANDE CALCULATE THE REST WITH IPP*nindexprofile
140 self.timeCount = fp.get(header+'/TimeCount')# NOT USE FOR THIS
143 self.timeCount = fp.get(header+'/TimeCount')# NOT USE FOR THIS
141 self.timeStatus = fp.get(header+'/TimeStatus')# NOT USE FOR THIS
144 self.timeStatus = fp.get(header+'/TimeStatus')# NOT USE FOR THIS
142 self.rangeFromFile = fp.get('Raw11/Data/Samples/Range')
145 self.rangeFromFile = fp.get('Raw11/Data/Samples/Range')
143 self.frequency = fp.get('Rx/Frequency')
146 self.frequency = fp.get('Rx/Frequency')
144 txAus = fp.get('Raw11/Data/Pulsewidth')
147 txAus = fp.get('Raw11/Data/Pulsewidth')
145
148
146
149
147 self.nblocks = self.pulseCount.shape[0] #nblocks
150 self.nblocks = self.pulseCount.shape[0] #nblocks
148
151
149 self.nprofiles = self.pulseCount.shape[1] #nprofile
152 self.nprofiles = self.pulseCount.shape[1] #nprofile
150 self.nsa = self.nsamplesPulse[0,0] #ngates
153 self.nsa = self.nsamplesPulse[0,0] #ngates
151 self.nchannels = self.beamCode.shape[1]
154 self.nchannels = self.beamCode.shape[1]
152 self.ippSeconds = (self.radacTime[0][1] -self.radacTime[0][0]) #Ipp in seconds
155 self.ippSeconds = (self.radacTime[0][1] -self.radacTime[0][0]) #Ipp in seconds
153 #self.__waitForNewFile = self.nblocks # wait depending on the number of blocks since each block is 1 sec
156 #self.__waitForNewFile = self.nblocks # wait depending on the number of blocks since each block is 1 sec
154 self.__waitForNewFile = self.nblocks * self.nprofiles * self.ippSeconds # wait until new file is created
157 self.__waitForNewFile = self.nblocks * self.nprofiles * self.ippSeconds # wait until new file is created
155
158
156 #filling radar controller header parameters
159 #filling radar controller header parameters
157 self.__ippKm = self.ippSeconds *.15*1e6 # in km
160 self.__ippKm = self.ippSeconds *.15*1e6 # in km
158 self.__txA = (txAus.value)*.15 #(ipp[us]*.15km/1us) in km
161 self.__txA = (txAus.value)*.15 #(ipp[us]*.15km/1us) in km
159 self.__txB = 0
162 self.__txB = 0
160 nWindows=1
163 nWindows=1
161 self.__nSamples = self.nsa
164 self.__nSamples = self.nsa
162 self.__firstHeight = self.rangeFromFile[0][0]/1000 #in km
165 self.__firstHeight = self.rangeFromFile[0][0]/1000 #in km
163 self.__deltaHeight = (self.rangeFromFile[0][1] - self.rangeFromFile[0][0])/1000
166 self.__deltaHeight = (self.rangeFromFile[0][1] - self.rangeFromFile[0][0])/1000
164
167
165 #for now until understand why the code saved is different (code included even though code not in tuf file)
168 #for now until understand why the code saved is different (code included even though code not in tuf file)
166 #self.__codeType = 0
169 #self.__codeType = 0
167 # self.__nCode = None
170 # self.__nCode = None
168 # self.__nBaud = None
171 # self.__nBaud = None
169 self.__code = self.code
172 self.__code = self.code
170 self.__codeType = 0
173 self.__codeType = 0
171 if self.code != None:
174 if self.code != None:
172 self.__codeType = 1
175 self.__codeType = 1
173 self.__nCode = self.nCode
176 self.__nCode = self.nCode
174 self.__nBaud = self.nBaud
177 self.__nBaud = self.nBaud
175 #self.__code = 0
178 #self.__code = 0
176
179
177 #filling system header parameters
180 #filling system header parameters
178 self.__nSamples = self.nsa
181 self.__nSamples = self.nsa
179 self.newProfiles = self.nprofiles/self.nchannels
182 self.newProfiles = self.nprofiles/self.nchannels
180 self.__channelList = list(range(self.nchannels))
183 self.__channelList = list(range(self.nchannels))
181
184
182 self.__frequency = self.frequency[0][0]
185 self.__frequency = self.frequency[0][0]
183
186
184
187
185
188
186 def createBuffers(self):
189 def createBuffers(self):
187
190
188 pass
191 pass
189
192
190 def __setParameters(self,path='', startDate='',endDate='',startTime='', endTime='', walk=''):
193 def __setParameters(self,path='', startDate='',endDate='',startTime='', endTime='', walk=''):
191 self.path = path
194 self.path = path
192 self.startDate = startDate
195 self.startDate = startDate
193 self.endDate = endDate
196 self.endDate = endDate
194 self.startTime = startTime
197 self.startTime = startTime
195 self.endTime = endTime
198 self.endTime = endTime
196 self.walk = walk
199 self.walk = walk
197
200
198 def __checkPath(self):
201 def __checkPath(self):
199 if os.path.exists(self.path):
202 if os.path.exists(self.path):
200 self.status = 1
203 self.status = 1
201 else:
204 else:
202 self.status = 0
205 self.status = 0
203 print('Path:%s does not exists'%self.path)
206 print('Path:%s does not exists'%self.path)
204
207
205 return
208 return
206
209
207
210
208 def __selDates(self, amisr_dirname_format):
211 def __selDates(self, amisr_dirname_format):
209 try:
212 try:
210 year = int(amisr_dirname_format[0:4])
213 year = int(amisr_dirname_format[0:4])
211 month = int(amisr_dirname_format[4:6])
214 month = int(amisr_dirname_format[4:6])
212 dom = int(amisr_dirname_format[6:8])
215 dom = int(amisr_dirname_format[6:8])
213 thisDate = datetime.date(year,month,dom)
216 thisDate = datetime.date(year,month,dom)
214
217
215 if (thisDate>=self.startDate and thisDate <= self.endDate):
218 if (thisDate>=self.startDate and thisDate <= self.endDate):
216 return amisr_dirname_format
219 return amisr_dirname_format
217 except:
220 except:
218 return None
221 return None
219
222
220
223
221 def __findDataForDates(self,online=False):
224 def __findDataForDates(self,online=False):
222
225
223 if not(self.status):
226 if not(self.status):
224 return None
227 return None
225
228
226 pat = '\d+.\d+'
229 pat = '\d+.\d+'
227 dirnameList = [re.search(pat,x) for x in os.listdir(self.path)]
230 dirnameList = [re.search(pat,x) for x in os.listdir(self.path)]
228 dirnameList = [x for x in dirnameList if x!=None]
231 dirnameList = [x for x in dirnameList if x!=None]
229 dirnameList = [x.string for x in dirnameList]
232 dirnameList = [x.string for x in dirnameList]
230 if not(online):
233 if not(online):
231 dirnameList = [self.__selDates(x) for x in dirnameList]
234 dirnameList = [self.__selDates(x) for x in dirnameList]
232 dirnameList = [x for x in dirnameList if x!=None]
235 dirnameList = [x for x in dirnameList if x!=None]
233 if len(dirnameList)>0:
236 if len(dirnameList)>0:
234 self.status = 1
237 self.status = 1
235 self.dirnameList = dirnameList
238 self.dirnameList = dirnameList
236 self.dirnameList.sort()
239 self.dirnameList.sort()
237 else:
240 else:
238 self.status = 0
241 self.status = 0
239 return None
242 return None
240
243
241 def __getTimeFromData(self):
244 def __getTimeFromData(self):
242 startDateTime_Reader = datetime.datetime.combine(self.startDate,self.startTime)
245 startDateTime_Reader = datetime.datetime.combine(self.startDate,self.startTime)
243 endDateTime_Reader = datetime.datetime.combine(self.endDate,self.endTime)
246 endDateTime_Reader = datetime.datetime.combine(self.endDate,self.endTime)
244
247
245 print('Filtering Files from %s to %s'%(startDateTime_Reader, endDateTime_Reader))
248 print('Filtering Files from %s to %s'%(startDateTime_Reader, endDateTime_Reader))
246 print('........................................')
249 print('........................................')
247 filter_filenameList = []
250 filter_filenameList = []
248 self.filenameList.sort()
251 self.filenameList.sort()
249 #for i in range(len(self.filenameList)-1):
252 #for i in range(len(self.filenameList)-1):
250 for i in range(len(self.filenameList)):
253 for i in range(len(self.filenameList)):
251 filename = self.filenameList[i]
254 filename = self.filenameList[i]
252 fp = h5py.File(filename,'r')
255 fp = h5py.File(filename,'r')
253 time_str = fp.get('Time/RadacTimeString')
256 time_str = fp.get('Time/RadacTimeString')
254
257
255 startDateTimeStr_File = time_str[0][0].split('.')[0]
258 startDateTimeStr_File = time_str[0][0].decode('UTF-8').split('.')[0]
259 #startDateTimeStr_File = "2019-12-16 09:21:11"
256 junk = time.strptime(startDateTimeStr_File, '%Y-%m-%d %H:%M:%S')
260 junk = time.strptime(startDateTimeStr_File, '%Y-%m-%d %H:%M:%S')
257 startDateTime_File = datetime.datetime(junk.tm_year,junk.tm_mon,junk.tm_mday,junk.tm_hour, junk.tm_min, junk.tm_sec)
261 startDateTime_File = datetime.datetime(junk.tm_year,junk.tm_mon,junk.tm_mday,junk.tm_hour, junk.tm_min, junk.tm_sec)
258
262
259 endDateTimeStr_File = time_str[-1][-1].split('.')[0]
263 #endDateTimeStr_File = "2019-12-16 11:10:11"
264 endDateTimeStr_File = time_str[-1][-1].decode('UTF-8').split('.')[0]
260 junk = time.strptime(endDateTimeStr_File, '%Y-%m-%d %H:%M:%S')
265 junk = time.strptime(endDateTimeStr_File, '%Y-%m-%d %H:%M:%S')
261 endDateTime_File = datetime.datetime(junk.tm_year,junk.tm_mon,junk.tm_mday,junk.tm_hour, junk.tm_min, junk.tm_sec)
266 endDateTime_File = datetime.datetime(junk.tm_year,junk.tm_mon,junk.tm_mday,junk.tm_hour, junk.tm_min, junk.tm_sec)
262
267
263 fp.close()
268 fp.close()
264
269
270 #print("check time", startDateTime_File)
265 if self.timezone == 'lt':
271 if self.timezone == 'lt':
266 startDateTime_File = startDateTime_File - datetime.timedelta(minutes = 300)
272 startDateTime_File = startDateTime_File - datetime.timedelta(minutes = 300)
267 endDateTime_File = endDateTime_File - datetime.timedelta(minutes = 300)
273 endDateTime_File = endDateTime_File - datetime.timedelta(minutes = 300)
268
269 if (endDateTime_File>=startDateTime_Reader and endDateTime_File<endDateTime_Reader):
274 if (endDateTime_File>=startDateTime_Reader and endDateTime_File<endDateTime_Reader):
270 #self.filenameList.remove(filename)
275 #self.filenameList.remove(filename)
271 filter_filenameList.append(filename)
276 filter_filenameList.append(filename)
272
277
273 if (endDateTime_File>=endDateTime_Reader):
278 if (endDateTime_File>=endDateTime_Reader):
274 break
279 break
275
280
276
281
277 filter_filenameList.sort()
282 filter_filenameList.sort()
278 self.filenameList = filter_filenameList
283 self.filenameList = filter_filenameList
279 return 1
284 return 1
280
285
281 def __filterByGlob1(self, dirName):
286 def __filterByGlob1(self, dirName):
282 filter_files = glob.glob1(dirName, '*.*%s'%self.extension_file)
287 filter_files = glob.glob1(dirName, '*.*%s'%self.extension_file)
283 filter_files.sort()
288 filter_files.sort()
284 filterDict = {}
289 filterDict = {}
285 filterDict.setdefault(dirName)
290 filterDict.setdefault(dirName)
286 filterDict[dirName] = filter_files
291 filterDict[dirName] = filter_files
287 return filterDict
292 return filterDict
288
293
289 def __getFilenameList(self, fileListInKeys, dirList):
294 def __getFilenameList(self, fileListInKeys, dirList):
290 for value in fileListInKeys:
295 for value in fileListInKeys:
291 dirName = list(value.keys())[0]
296 dirName = list(value.keys())[0]
292 for file in value[dirName]:
297 for file in value[dirName]:
293 filename = os.path.join(dirName, file)
298 filename = os.path.join(dirName, file)
294 self.filenameList.append(filename)
299 self.filenameList.append(filename)
295
300
296
301
297 def __selectDataForTimes(self, online=False):
302 def __selectDataForTimes(self, online=False):
298 #aun no esta implementado el filtro for tiempo
303 #aun no esta implementado el filtro for tiempo
299 if not(self.status):
304 if not(self.status):
300 return None
305 return None
301
306
302 dirList = [os.path.join(self.path,x) for x in self.dirnameList]
307 dirList = [os.path.join(self.path,x) for x in self.dirnameList]
303
308
304 fileListInKeys = [self.__filterByGlob1(x) for x in dirList]
309 fileListInKeys = [self.__filterByGlob1(x) for x in dirList]
305
310
306 self.__getFilenameList(fileListInKeys, dirList)
311 self.__getFilenameList(fileListInKeys, dirList)
307 if not(online):
312 if not(online):
308 #filtro por tiempo
313 #filtro por tiempo
309 if not(self.all):
314 if not(self.all):
310 self.__getTimeFromData()
315 self.__getTimeFromData()
311
316
312 if len(self.filenameList)>0:
317 if len(self.filenameList)>0:
313 self.status = 1
318 self.status = 1
314 self.filenameList.sort()
319 self.filenameList.sort()
315 else:
320 else:
316 self.status = 0
321 self.status = 0
317 return None
322 return None
318
323
319 else:
324 else:
320 #get the last file - 1
325 #get the last file - 1
321 self.filenameList = [self.filenameList[-2]]
326 self.filenameList = [self.filenameList[-2]]
322
327
323 new_dirnameList = []
328 new_dirnameList = []
324 for dirname in self.dirnameList:
329 for dirname in self.dirnameList:
325 junk = numpy.array([dirname in x for x in self.filenameList])
330 junk = numpy.array([dirname in x for x in self.filenameList])
326 junk_sum = junk.sum()
331 junk_sum = junk.sum()
327 if junk_sum > 0:
332 if junk_sum > 0:
328 new_dirnameList.append(dirname)
333 new_dirnameList.append(dirname)
329 self.dirnameList = new_dirnameList
334 self.dirnameList = new_dirnameList
330 return 1
335 return 1
331
336
332 def searchFilesOnLine(self, path, startDate, endDate, startTime=datetime.time(0,0,0),
337 def searchFilesOnLine(self, path, startDate, endDate, startTime=datetime.time(0,0,0),
333 endTime=datetime.time(23,59,59),walk=True):
338 endTime=datetime.time(23,59,59),walk=True):
334
339
335 if endDate ==None:
340 if endDate ==None:
336 startDate = datetime.datetime.utcnow().date()
341 startDate = datetime.datetime.utcnow().date()
337 endDate = datetime.datetime.utcnow().date()
342 endDate = datetime.datetime.utcnow().date()
338
343
339 self.__setParameters(path=path, startDate=startDate, endDate=endDate,startTime = startTime,endTime=endTime, walk=walk)
344 self.__setParameters(path=path, startDate=startDate, endDate=endDate,startTime = startTime,endTime=endTime, walk=walk)
340
345
341 self.__checkPath()
346 self.__checkPath()
342
347
343 self.__findDataForDates(online=True)
348 self.__findDataForDates(online=True)
344
349
345 self.dirnameList = [self.dirnameList[-1]]
350 self.dirnameList = [self.dirnameList[-1]]
346
351
347 self.__selectDataForTimes(online=True)
352 self.__selectDataForTimes(online=True)
348
353
349 return
354 return
350
355
351
356
352 def searchFilesOffLine(self,
357 def searchFilesOffLine(self,
353 path,
358 path,
354 startDate,
359 startDate,
355 endDate,
360 endDate,
356 startTime=datetime.time(0,0,0),
361 startTime=datetime.time(0,0,0),
357 endTime=datetime.time(23,59,59),
362 endTime=datetime.time(23,59,59),
358 walk=True):
363 walk=True):
359
364
360 self.__setParameters(path, startDate, endDate, startTime, endTime, walk)
365 self.__setParameters(path, startDate, endDate, startTime, endTime, walk)
361
366
362 self.__checkPath()
367 self.__checkPath()
363
368
364 self.__findDataForDates()
369 self.__findDataForDates()
365
370
366 self.__selectDataForTimes()
371 self.__selectDataForTimes()
367
372
368 for i in range(len(self.filenameList)):
373 for i in range(len(self.filenameList)):
369 print("%s" %(self.filenameList[i]))
374 print("%s" %(self.filenameList[i]))
370
375
371 return
376 return
372
377
373 def __setNextFileOffline(self):
378 def __setNextFileOffline(self):
374 idFile = self.fileIndex
379 idFile = self.fileIndex
375
380
376 while (True):
381 while (True):
377 idFile += 1
382 idFile += 1
378 if not(idFile < len(self.filenameList)):
383 if not(idFile < len(self.filenameList)):
379 self.flagNoMoreFiles = 1
384 self.flagNoMoreFiles = 1
380 print("No more Files")
385 print("No more Files")
386 self.dataOut.error = True
381 return 0
387 return 0
382
388
383 filename = self.filenameList[idFile]
389 filename = self.filenameList[idFile]
384
390
385 amisrFilePointer = h5py.File(filename,'r')
391 amisrFilePointer = h5py.File(filename,'r')
386
392
387 break
393 break
388
394
389 self.flagIsNewFile = 1
395 self.flagIsNewFile = 1
390 self.fileIndex = idFile
396 self.fileIndex = idFile
391 self.filename = filename
397 self.filename = filename
392
398
393 self.amisrFilePointer = amisrFilePointer
399 self.amisrFilePointer = amisrFilePointer
394
400
395 print("Setting the file: %s"%self.filename)
401 print("Setting the file: %s"%self.filename)
396
402
397 return 1
403 return 1
398
404
399
405
400 def __setNextFileOnline(self):
406 def __setNextFileOnline(self):
401 filename = self.filenameList[0]
407 filename = self.filenameList[0]
402 if self.__filename_online != None:
408 if self.__filename_online != None:
403 self.__selectDataForTimes(online=True)
409 self.__selectDataForTimes(online=True)
404 filename = self.filenameList[0]
410 filename = self.filenameList[0]
405 wait = 0
411 wait = 0
406 while self.__filename_online == filename:
412 while self.__filename_online == filename:
407 print('waiting %d seconds to get a new file...'%(self.__waitForNewFile))
413 print('waiting %d seconds to get a new file...'%(self.__waitForNewFile))
408 if wait == 5:
414 if wait == 5:
409 return 0
415 return 0
410 sleep(self.__waitForNewFile)
416 sleep(self.__waitForNewFile)
411 self.__selectDataForTimes(online=True)
417 self.__selectDataForTimes(online=True)
412 filename = self.filenameList[0]
418 filename = self.filenameList[0]
413 wait += 1
419 wait += 1
414
420
415 self.__filename_online = filename
421 self.__filename_online = filename
416
422
417 self.amisrFilePointer = h5py.File(filename,'r')
423 self.amisrFilePointer = h5py.File(filename,'r')
418 self.flagIsNewFile = 1
424 self.flagIsNewFile = 1
419 self.filename = filename
425 self.filename = filename
420 print("Setting the file: %s"%self.filename)
426 print("Setting the file: %s"%self.filename)
421 return 1
427 return 1
422
428
423
429
424 def readData(self):
430 def readData(self):
425 buffer = self.amisrFilePointer.get('Raw11/Data/Samples/Data')
431 buffer = self.amisrFilePointer.get('Raw11/Data/Samples/Data')
426 re = buffer[:,:,:,0]
432 re = buffer[:,:,:,0]
427 im = buffer[:,:,:,1]
433 im = buffer[:,:,:,1]
428 dataset = re + im*1j
434 dataset = re + im*1j
435
429 self.radacTime = self.amisrFilePointer.get('Raw11/Data/RadacHeader/RadacTime')
436 self.radacTime = self.amisrFilePointer.get('Raw11/Data/RadacHeader/RadacTime')
430 timeset = self.radacTime[:,0]
437 timeset = self.radacTime[:,0]
438
431 return dataset,timeset
439 return dataset,timeset
432
440
433 def reshapeData(self):
441 def reshapeData(self):
434 #self.beamCodeByPulse, self.beamCode, self.nblocks, self.nprofiles, self.nsa,
442 #self.beamCodeByPulse, self.beamCode, self.nblocks, self.nprofiles, self.nsa,
435 channels = self.beamCodeByPulse[0,:]
443 channels = self.beamCodeByPulse[0,:]
436 nchan = self.nchannels
444 nchan = self.nchannels
437 #self.newProfiles = self.nprofiles/nchan #must be defined on filljroheader
445 #self.newProfiles = self.nprofiles/nchan #must be defined on filljroheader
438 nblocks = self.nblocks
446 nblocks = self.nblocks
439 nsamples = self.nsa
447 nsamples = self.nsa
440
448
441 #Dimensions : nChannels, nProfiles, nSamples
449 #Dimensions : nChannels, nProfiles, nSamples
442 new_block = numpy.empty((nblocks, nchan, self.newProfiles, nsamples), dtype="complex64")
450 new_block = numpy.empty((nblocks, nchan, numpy.int_(self.newProfiles), nsamples), dtype="complex64")
443 ############################################
451 ############################################
444
452
445 for thisChannel in range(nchan):
453 for thisChannel in range(nchan):
446 new_block[:,thisChannel,:,:] = self.dataset[:,numpy.where(channels==self.beamCode[0][thisChannel])[0],:]
454 new_block[:,thisChannel,:,:] = self.dataset[:,numpy.where(channels==self.beamCode[0][thisChannel])[0],:]
447
455
448
456
449 new_block = numpy.transpose(new_block, (1,0,2,3))
457 new_block = numpy.transpose(new_block, (1,0,2,3))
450 new_block = numpy.reshape(new_block, (nchan,-1, nsamples))
458 new_block = numpy.reshape(new_block, (nchan,-1, nsamples))
451
459
452 return new_block
460 return new_block
453
461
454 def updateIndexes(self):
462 def updateIndexes(self):
455
463
456 pass
464 pass
457
465
458 def fillJROHeader(self):
466 def fillJROHeader(self):
459
467
460 #fill radar controller header
468 #fill radar controller header
461 self.dataOut.radarControllerHeaderObj = RadarControllerHeader(ippKm=self.__ippKm,
469 self.dataOut.radarControllerHeaderObj = RadarControllerHeader(ipp=self.__ippKm,
462 txA=self.__txA,
470 txA=self.__txA,
463 txB=0,
471 txB=0,
464 nWindows=1,
472 nWindows=1,
465 nHeights=self.__nSamples,
473 nHeights=self.__nSamples,
466 firstHeight=self.__firstHeight,
474 firstHeight=self.__firstHeight,
467 deltaHeight=self.__deltaHeight,
475 deltaHeight=self.__deltaHeight,
468 codeType=self.__codeType,
476 codeType=self.__codeType,
469 nCode=self.__nCode, nBaud=self.__nBaud,
477 nCode=self.__nCode, nBaud=self.__nBaud,
470 code = self.__code,
478 code = self.__code,
471 fClock=1)
479 fClock=1)
472
480
473
474
475 #fill system header
481 #fill system header
476 self.dataOut.systemHeaderObj = SystemHeader(nSamples=self.__nSamples,
482 self.dataOut.systemHeaderObj = SystemHeader(nSamples=self.__nSamples,
477 nProfiles=self.newProfiles,
483 nProfiles=self.newProfiles,
478 nChannels=len(self.__channelList),
484 nChannels=len(self.__channelList),
479 adcResolution=14,
485 adcResolution=14,
480 pciDioBusWith=32)
486 pciDioBusWidth=32)
481
487
482 self.dataOut.type = "Voltage"
488 self.dataOut.type = "Voltage"
483
489
484 self.dataOut.data = None
490 self.dataOut.data = None
485
491
486 self.dataOut.dtype = numpy.dtype([('real','<i8'),('imag','<i8')])
492 self.dataOut.dtype = numpy.dtype([('real','<i8'),('imag','<i8')])
487
493
488 # self.dataOut.nChannels = 0
494 # self.dataOut.nChannels = 0
489
495
490 # self.dataOut.nHeights = 0
496 # self.dataOut.nHeights = 0
491
497
492 self.dataOut.nProfiles = self.newProfiles*self.nblocks
498 self.dataOut.nProfiles = self.newProfiles*self.nblocks
493
499
494 #self.dataOut.heightList = self.__firstHeigth + numpy.arange(self.__nSamples, dtype = numpy.float)*self.__deltaHeigth
500 #self.dataOut.heightList = self.__firstHeigth + numpy.arange(self.__nSamples, dtype = numpy.float)*self.__deltaHeigth
495 ranges = numpy.reshape(self.rangeFromFile.value,(-1))
501 ranges = numpy.reshape(self.rangeFromFile.value,(-1))
496 self.dataOut.heightList = ranges/1000.0 #km
502 self.dataOut.heightList = ranges/1000.0 #km
497
503
498
504
499 self.dataOut.channelList = self.__channelList
505 self.dataOut.channelList = self.__channelList
500
506
501 self.dataOut.blocksize = self.dataOut.getNChannels() * self.dataOut.getNHeights()
507 self.dataOut.blocksize = self.dataOut.getNChannels() * self.dataOut.getNHeights()
502
508
503 # self.dataOut.channelIndexList = None
509 # self.dataOut.channelIndexList = None
504
510
505 self.dataOut.flagNoData = True
511 self.dataOut.flagNoData = True
506
512
507 #Set to TRUE if the data is discontinuous
513 #Set to TRUE if the data is discontinuous
508 self.dataOut.flagDiscontinuousBlock = False
514 self.dataOut.flagDiscontinuousBlock = False
509
515
510 self.dataOut.utctime = None
516 self.dataOut.utctime = None
511
517
512 #self.dataOut.timeZone = -5 #self.__timezone/60 #timezone like jroheader, difference in minutes between UTC and localtime
518 #self.dataOut.timeZone = -5 #self.__timezone/60 #timezone like jroheader, difference in minutes between UTC and localtime
513 if self.timezone == 'lt':
519 if self.timezone == 'lt':
514 self.dataOut.timeZone = time.timezone / 60. #get the timezone in minutes
520 self.dataOut.timeZone = time.timezone / 60. #get the timezone in minutes
515 else:
521 else:
516 self.dataOut.timeZone = 0 #by default time is UTC
522 self.dataOut.timeZone = 0 #by default time is UTC
517
523
518 self.dataOut.dstFlag = 0
524 self.dataOut.dstFlag = 0
519
525
520 self.dataOut.errorCount = 0
526 self.dataOut.errorCount = 0
521
527
522 self.dataOut.nCohInt = 1
528 self.dataOut.nCohInt = 1
523
529
524 self.dataOut.flagDecodeData = False #asumo que la data esta decodificada
530 self.dataOut.flagDecodeData = False #asumo que la data esta decodificada
525
531
526 self.dataOut.flagDeflipData = False #asumo que la data esta sin flip
532 self.dataOut.flagDeflipData = False #asumo que la data esta sin flip
527
533
528 self.dataOut.flagShiftFFT = False
534 self.dataOut.flagShiftFFT = False
529
535
530 self.dataOut.ippSeconds = self.ippSeconds
536 self.dataOut.ippSeconds = self.ippSeconds
531
537
532 #Time interval between profiles
538 #Time interval between profiles
533 #self.dataOut.timeInterval = self.dataOut.ippSeconds * self.dataOut.nCohInt
539 #self.dataOut.timeInterval = self.dataOut.ippSeconds * self.dataOut.nCohInt
534
540
535 self.dataOut.frequency = self.__frequency
541 self.dataOut.frequency = self.__frequency
536
537 self.dataOut.realtime = self.online
542 self.dataOut.realtime = self.online
538 pass
543 pass
539
544
540 def readNextFile(self,online=False):
545 def readNextFile(self,online=False):
541
546
542 if not(online):
547 if not(online):
543 newFile = self.__setNextFileOffline()
548 newFile = self.__setNextFileOffline()
544 else:
549 else:
545 newFile = self.__setNextFileOnline()
550 newFile = self.__setNextFileOnline()
546
551
547 if not(newFile):
552 if not(newFile):
548 return 0
553 return 0
549
550 #if self.__firstFile:
554 #if self.__firstFile:
551 self.readAMISRHeader(self.amisrFilePointer)
555 self.readAMISRHeader(self.amisrFilePointer)
556
552 self.createBuffers()
557 self.createBuffers()
558
553 self.fillJROHeader()
559 self.fillJROHeader()
560
554 #self.__firstFile = False
561 #self.__firstFile = False
555
562
556
563
557
564
558 self.dataset,self.timeset = self.readData()
565 self.dataset,self.timeset = self.readData()
559
566
560 if self.endDate!=None:
567 if self.endDate!=None:
561 endDateTime_Reader = datetime.datetime.combine(self.endDate,self.endTime)
568 endDateTime_Reader = datetime.datetime.combine(self.endDate,self.endTime)
562 time_str = self.amisrFilePointer.get('Time/RadacTimeString')
569 time_str = self.amisrFilePointer.get('Time/RadacTimeString')
563 startDateTimeStr_File = time_str[0][0].split('.')[0]
570 startDateTimeStr_File = time_str[0][0].decode('UTF-8').split('.')[0]
564 junk = time.strptime(startDateTimeStr_File, '%Y-%m-%d %H:%M:%S')
571 junk = time.strptime(startDateTimeStr_File, '%Y-%m-%d %H:%M:%S')
565 startDateTime_File = datetime.datetime(junk.tm_year,junk.tm_mon,junk.tm_mday,junk.tm_hour, junk.tm_min, junk.tm_sec)
572 startDateTime_File = datetime.datetime(junk.tm_year,junk.tm_mon,junk.tm_mday,junk.tm_hour, junk.tm_min, junk.tm_sec)
566 if self.timezone == 'lt':
573 if self.timezone == 'lt':
567 startDateTime_File = startDateTime_File - datetime.timedelta(minutes = 300)
574 startDateTime_File = startDateTime_File - datetime.timedelta(minutes = 300)
568 if (startDateTime_File>endDateTime_Reader):
575 if (startDateTime_File>endDateTime_Reader):
569 return 0
576 return 0
570
577
571 self.jrodataset = self.reshapeData()
578 self.jrodataset = self.reshapeData()
572 #----self.updateIndexes()
579 #----self.updateIndexes()
573 self.profileIndex = 0
580 self.profileIndex = 0
574
581
575 return 1
582 return 1
576
583
577
584
578 def __hasNotDataInBuffer(self):
585 def __hasNotDataInBuffer(self):
579 if self.profileIndex >= (self.newProfiles*self.nblocks):
586 if self.profileIndex >= (self.newProfiles*self.nblocks):
580 return 1
587 return 1
581 return 0
588 return 0
582
589
583
590
584 def getData(self):
591 def getData(self):
585
592
586 if self.flagNoMoreFiles:
593 if self.flagNoMoreFiles:
587 self.dataOut.flagNoData = True
594 self.dataOut.flagNoData = True
588 return 0
595 return 0
589
596
590 if self.__hasNotDataInBuffer():
597 if self.__hasNotDataInBuffer():
591 if not (self.readNextFile(self.online)):
598 if not (self.readNextFile(self.online)):
592 return 0
599 return 0
593
600
594
601
595 if self.dataset is None: # setear esta condicion cuando no hayan datos por leers
602 if self.dataset is None: # setear esta condicion cuando no hayan datos por leer
596 self.dataOut.flagNoData = True
603 self.dataOut.flagNoData = True
597 return 0
604 return 0
598
605
599 #self.dataOut.data = numpy.reshape(self.jrodataset[self.profileIndex,:],(1,-1))
606 #self.dataOut.data = numpy.reshape(self.jrodataset[self.profileIndex,:],(1,-1))
600
607
601 self.dataOut.data = self.jrodataset[:,self.profileIndex,:]
608 self.dataOut.data = self.jrodataset[:,self.profileIndex,:]
602
609
610 #print("R_t",self.timeset)
611
603 #self.dataOut.utctime = self.jrotimeset[self.profileIndex]
612 #self.dataOut.utctime = self.jrotimeset[self.profileIndex]
604 #verificar basic header de jro data y ver si es compatible con este valor
613 #verificar basic header de jro data y ver si es compatible con este valor
605 #self.dataOut.utctime = self.timeset + (self.profileIndex * self.ippSeconds * self.nchannels)
614 #self.dataOut.utctime = self.timeset + (self.profileIndex * self.ippSeconds * self.nchannels)
606 indexprof = numpy.mod(self.profileIndex, self.newProfiles)
615 indexprof = numpy.mod(self.profileIndex, self.newProfiles)
607 indexblock = self.profileIndex/self.newProfiles
616 indexblock = self.profileIndex/self.newProfiles
608 #print indexblock, indexprof
617 #print (indexblock, indexprof)
609 self.dataOut.utctime = self.timeset[indexblock] + (indexprof * self.ippSeconds * self.nchannels)
618 diffUTC = 1.8e4 #UTC diference from peru in seconds --Joab
619 diffUTC = 0
620 t_comp = (indexprof * self.ippSeconds * self.nchannels) + diffUTC #
621 #cambio posible 18/02/2020
622
623
624
625 #print("utc :",indexblock," __ ",t_comp)
626 #print(numpy.shape(self.timeset))
627 self.dataOut.utctime = self.timeset[numpy.int_(indexblock)] + t_comp
628 #self.dataOut.utctime = self.timeset[self.profileIndex] + t_comp
629 #print(self.dataOut.utctime)
610 self.dataOut.profileIndex = self.profileIndex
630 self.dataOut.profileIndex = self.profileIndex
611 self.dataOut.flagNoData = False
631 self.dataOut.flagNoData = False
612 # if indexprof == 0:
632 # if indexprof == 0:
613 # print self.dataOut.utctime
633 # print self.dataOut.utctime
614
634
615 self.profileIndex += 1
635 self.profileIndex += 1
616
636
617 return self.dataOut.data
637 return self.dataOut.data
618
638
619
639
620 def run(self, **kwargs):
640 def run(self, **kwargs):
621 '''
641 '''
622 This method will be called many times so here you should put all your code
642 This method will be called many times so here you should put all your code
623 '''
643 '''
624
644
625 if not self.isConfig:
645 if not self.isConfig:
626 self.setup(**kwargs)
646 self.setup(**kwargs)
627 self.isConfig = True
647 self.isConfig = True
628
648
629 self.getData()
649 self.getData()
@@ -1,1435 +1,1435
1 import numpy
1 import numpy
2 import time
2 import time
3 import os
3 import os
4 import h5py
4 import h5py
5 import re
5 import re
6 import datetime
6 import datetime
7
7
8 import schainpy.admin
8 import schainpy.admin
9 from schainpy.model.data.jrodata import *
9 from schainpy.model.data.jrodata import *
10 from schainpy.model.proc.jroproc_base import ProcessingUnit, Operation, MPDecorator
10 from schainpy.model.proc.jroproc_base import ProcessingUnit, Operation, MPDecorator
11 from schainpy.model.io.jroIO_base import *
11 from schainpy.model.io.jroIO_base import *
12 from schainpy.utils import log
12 from schainpy.utils import log
13
13
14 @MPDecorator
14 @MPDecorator
15 class ParamReader(JRODataReader,ProcessingUnit):
15 class ParamReader(JRODataReader,ProcessingUnit):
16 '''
16 '''
17 Reads HDF5 format files
17 Reads HDF5 format files
18 path
18 path
19 startDate
19 startDate
20 endDate
20 endDate
21 startTime
21 startTime
22 endTime
22 endTime
23 '''
23 '''
24
24
25 ext = ".hdf5"
25 ext = ".hdf5"
26 optchar = "D"
26 optchar = "D"
27 timezone = None
27 timezone = None
28 startTime = None
28 startTime = None
29 endTime = None
29 endTime = None
30 fileIndex = None
30 fileIndex = None
31 utcList = None #To select data in the utctime list
31 utcList = None #To select data in the utctime list
32 blockList = None #List to blocks to be read from the file
32 blockList = None #List to blocks to be read from the file
33 blocksPerFile = None #Number of blocks to be read
33 blocksPerFile = None #Number of blocks to be read
34 blockIndex = None
34 blockIndex = None
35 path = None
35 path = None
36 #List of Files
36 #List of Files
37 filenameList = None
37 filenameList = None
38 datetimeList = None
38 datetimeList = None
39 #Hdf5 File
39 #Hdf5 File
40 listMetaname = None
40 listMetaname = None
41 listMeta = None
41 listMeta = None
42 listDataname = None
42 listDataname = None
43 listData = None
43 listData = None
44 listShapes = None
44 listShapes = None
45 fp = None
45 fp = None
46 #dataOut reconstruction
46 #dataOut reconstruction
47 dataOut = None
47 dataOut = None
48
48
49 def __init__(self):#, **kwargs):
49 def __init__(self):#, **kwargs):
50 ProcessingUnit.__init__(self) #, **kwargs)
50 ProcessingUnit.__init__(self) #, **kwargs)
51 self.dataOut = Parameters()
51 self.dataOut = Parameters()
52 return
52 return
53
53
54 def setup(self, **kwargs):
54 def setup(self, **kwargs):
55
55
56 path = kwargs['path']
56 path = kwargs['path']
57 startDate = kwargs['startDate']
57 startDate = kwargs['startDate']
58 endDate = kwargs['endDate']
58 endDate = kwargs['endDate']
59 startTime = kwargs['startTime']
59 startTime = kwargs['startTime']
60 endTime = kwargs['endTime']
60 endTime = kwargs['endTime']
61 walk = kwargs['walk']
61 walk = kwargs['walk']
62 if 'ext' in kwargs:
62 if 'ext' in kwargs:
63 ext = kwargs['ext']
63 ext = kwargs['ext']
64 else:
64 else:
65 ext = '.hdf5'
65 ext = '.hdf5'
66 if 'timezone' in kwargs:
66 if 'timezone' in kwargs:
67 self.timezone = kwargs['timezone']
67 self.timezone = kwargs['timezone']
68 else:
68 else:
69 self.timezone = 'lt'
69 self.timezone = 'lt'
70
70
71 print("[Reading] Searching files in offline mode ...")
71 print("[Reading] Searching files in offline mode ...")
72 pathList, filenameList = self.searchFilesOffLine(path, startDate=startDate, endDate=endDate,
72 pathList, filenameList = self.searchFilesOffLine(path, startDate=startDate, endDate=endDate,
73 startTime=startTime, endTime=endTime,
73 startTime=startTime, endTime=endTime,
74 ext=ext, walk=walk)
74 ext=ext, walk=walk)
75
75
76 if not(filenameList):
76 if not(filenameList):
77 print("There is no files into the folder: %s"%(path))
77 print("There is no files into the folder: %s"%(path))
78 sys.exit(-1)
78 sys.exit(-1)
79
79
80 self.fileIndex = -1
80 self.fileIndex = -1
81 self.startTime = startTime
81 self.startTime = startTime
82 self.endTime = endTime
82 self.endTime = endTime
83
83
84 self.__readMetadata()
84 self.__readMetadata()
85
85
86 self.__setNextFileOffline()
86 self.__setNextFileOffline()
87
87
88 return
88 return
89
89
90 def searchFilesOffLine(self,
90 def searchFilesOffLine(self,
91 path,
91 path,
92 startDate=None,
92 startDate=None,
93 endDate=None,
93 endDate=None,
94 startTime=datetime.time(0,0,0),
94 startTime=datetime.time(0,0,0),
95 endTime=datetime.time(23,59,59),
95 endTime=datetime.time(23,59,59),
96 ext='.hdf5',
96 ext='.hdf5',
97 walk=True):
97 walk=True):
98
98
99 expLabel = ''
99 expLabel = ''
100 self.filenameList = []
100 self.filenameList = []
101 self.datetimeList = []
101 self.datetimeList = []
102
102
103 pathList = []
103 pathList = []
104
104
105 JRODataObj = JRODataReader()
105 JRODataObj = JRODataReader()
106 dateList, pathList = JRODataObj.findDatafiles(path, startDate, endDate, expLabel, ext, walk, include_path=True)
106 dateList, pathList = JRODataObj.findDatafiles(path, startDate, endDate, expLabel, ext, walk, include_path=True)
107
107
108 if dateList == []:
108 if dateList == []:
109 print("[Reading] No *%s files in %s from %s to %s)"%(ext, path,
109 print("[Reading] No *%s files in %s from %s to %s)"%(ext, path,
110 datetime.datetime.combine(startDate,startTime).ctime(),
110 datetime.datetime.combine(startDate,startTime).ctime(),
111 datetime.datetime.combine(endDate,endTime).ctime()))
111 datetime.datetime.combine(endDate,endTime).ctime()))
112
112
113 return None, None
113 return None, None
114
114
115 if len(dateList) > 1:
115 if len(dateList) > 1:
116 print("[Reading] %d days were found in date range: %s - %s" %(len(dateList), startDate, endDate))
116 print("[Reading] %d days were found in date range: %s - %s" %(len(dateList), startDate, endDate))
117 else:
117 else:
118 print("[Reading] data was found for the date %s" %(dateList[0]))
118 print("[Reading] data was found for the date %s" %(dateList[0]))
119
119
120 filenameList = []
120 filenameList = []
121 datetimeList = []
121 datetimeList = []
122
122
123 #----------------------------------------------------------------------------------
123 #----------------------------------------------------------------------------------
124
124
125 for thisPath in pathList:
125 for thisPath in pathList:
126
126
127 fileList = glob.glob1(thisPath, "*%s" %ext)
127 fileList = glob.glob1(thisPath, "*%s" %ext)
128 fileList.sort()
128 fileList.sort()
129
129
130 for file in fileList:
130 for file in fileList:
131
131
132 filename = os.path.join(thisPath,file)
132 filename = os.path.join(thisPath,file)
133
133
134 if not isFileInDateRange(filename, startDate, endDate):
134 if not isFileInDateRange(filename, startDate, endDate):
135 continue
135 continue
136
136
137 thisDatetime = self.__isFileInTimeRange(filename, startDate, endDate, startTime, endTime)
137 thisDatetime = self.__isFileInTimeRange(filename, startDate, endDate, startTime, endTime)
138
138
139 if not(thisDatetime):
139 if not(thisDatetime):
140 continue
140 continue
141
141
142 filenameList.append(filename)
142 filenameList.append(filename)
143 datetimeList.append(thisDatetime)
143 datetimeList.append(thisDatetime)
144
144
145 if not(filenameList):
145 if not(filenameList):
146 print("[Reading] Any file was found int time range %s - %s" %(datetime.datetime.combine(startDate,startTime).ctime(), datetime.datetime.combine(endDate,endTime).ctime()))
146 print("[Reading] Any file was found int time range %s - %s" %(datetime.datetime.combine(startDate,startTime).ctime(), datetime.datetime.combine(endDate,endTime).ctime()))
147 return None, None
147 return None, None
148
148
149 print("[Reading] %d file(s) was(were) found in time range: %s - %s" %(len(filenameList), startTime, endTime))
149 print("[Reading] %d file(s) was(were) found in time range: %s - %s" %(len(filenameList), startTime, endTime))
150 print()
150 print()
151
151
152 self.filenameList = filenameList
152 self.filenameList = filenameList
153 self.datetimeList = datetimeList
153 self.datetimeList = datetimeList
154
154
155 return pathList, filenameList
155 return pathList, filenameList
156
156
157 def __isFileInTimeRange(self,filename, startDate, endDate, startTime, endTime):
157 def __isFileInTimeRange(self,filename, startDate, endDate, startTime, endTime):
158
158
159 """
159 """
160 Retorna 1 si el archivo de datos se encuentra dentro del rango de horas especificado.
160 Retorna 1 si el archivo de datos se encuentra dentro del rango de horas especificado.
161
161
162 Inputs:
162 Inputs:
163 filename : nombre completo del archivo de datos en formato Jicamarca (.r)
163 filename : nombre completo del archivo de datos en formato Jicamarca (.r)
164 startDate : fecha inicial del rango seleccionado en formato datetime.date
164 startDate : fecha inicial del rango seleccionado en formato datetime.date
165 endDate : fecha final del rango seleccionado en formato datetime.date
165 endDate : fecha final del rango seleccionado en formato datetime.date
166 startTime : tiempo inicial del rango seleccionado en formato datetime.time
166 startTime : tiempo inicial del rango seleccionado en formato datetime.time
167 endTime : tiempo final del rango seleccionado en formato datetime.time
167 endTime : tiempo final del rango seleccionado en formato datetime.time
168
168
169 Return:
169 Return:
170 Boolean : Retorna True si el archivo de datos contiene datos en el rango de
170 Boolean : Retorna True si el archivo de datos contiene datos en el rango de
171 fecha especificado, de lo contrario retorna False.
171 fecha especificado, de lo contrario retorna False.
172
172
173 Excepciones:
173 Excepciones:
174 Si el archivo no existe o no puede ser abierto
174 Si el archivo no existe o no puede ser abierto
175 Si la cabecera no puede ser leida.
175 Si la cabecera no puede ser leida.
176
176
177 """
177 """
178
178
179 try:
179 try:
180 fp = h5py.File(filename,'r')
180 fp = h5py.File(filename,'r')
181 grp1 = fp['Data']
181 grp1 = fp['Data']
182
182
183 except IOError:
183 except IOError:
184 traceback.print_exc()
184 traceback.print_exc()
185 raise IOError("The file %s can't be opened" %(filename))
185 raise IOError("The file %s can't be opened" %(filename))
186
186
187 #In case has utctime attribute
187 #In case has utctime attribute
188 grp2 = grp1['utctime']
188 grp2 = grp1['utctime']
189 # thisUtcTime = grp2.value[0] - 5*3600 #To convert to local time
189 # thisUtcTime = grp2.value[0] - 5*3600 #To convert to local time
190 thisUtcTime = grp2.value[0]
190 thisUtcTime = grp2.value[0]
191
191
192 fp.close()
192 fp.close()
193
193
194 if self.timezone == 'lt':
194 if self.timezone == 'lt':
195 thisUtcTime -= 5*3600
195 thisUtcTime -= 5*3600
196
196
197 thisDatetime = datetime.datetime.fromtimestamp(thisUtcTime[0] + 5*3600)
197 thisDatetime = datetime.datetime.fromtimestamp(thisUtcTime[0] + 5*3600)
198 thisDate = thisDatetime.date()
198 thisDate = thisDatetime.date()
199 thisTime = thisDatetime.time()
199 thisTime = thisDatetime.time()
200
200
201 startUtcTime = (datetime.datetime.combine(thisDate,startTime)- datetime.datetime(1970, 1, 1)).total_seconds()
201 startUtcTime = (datetime.datetime.combine(thisDate,startTime)- datetime.datetime(1970, 1, 1)).total_seconds()
202 endUtcTime = (datetime.datetime.combine(thisDate,endTime)- datetime.datetime(1970, 1, 1)).total_seconds()
202 endUtcTime = (datetime.datetime.combine(thisDate,endTime)- datetime.datetime(1970, 1, 1)).total_seconds()
203
203
204 #General case
204 #General case
205 # o>>>>>>>>>>>>>><<<<<<<<<<<<<<o
205 # o>>>>>>>>>>>>>><<<<<<<<<<<<<<o
206 #-----------o----------------------------o-----------
206 #-----------o----------------------------o-----------
207 # startTime endTime
207 # startTime endTime
208
208
209 if endTime >= startTime:
209 if endTime >= startTime:
210 thisUtcLog = numpy.logical_and(thisUtcTime > startUtcTime, thisUtcTime < endUtcTime)
210 thisUtcLog = numpy.logical_and(thisUtcTime > startUtcTime, thisUtcTime < endUtcTime)
211 if numpy.any(thisUtcLog): #If there is one block between the hours mentioned
211 if numpy.any(thisUtcLog): #If there is one block between the hours mentioned
212 return thisDatetime
212 return thisDatetime
213 return None
213 return None
214
214
215 #If endTime < startTime then endTime belongs to the next day
215 #If endTime < startTime then endTime belongs to the next day
216 #<<<<<<<<<<<o o>>>>>>>>>>>
216 #<<<<<<<<<<<o o>>>>>>>>>>>
217 #-----------o----------------------------o-----------
217 #-----------o----------------------------o-----------
218 # endTime startTime
218 # endTime startTime
219
219
220 if (thisDate == startDate) and numpy.all(thisUtcTime < startUtcTime):
220 if (thisDate == startDate) and numpy.all(thisUtcTime < startUtcTime):
221 return None
221 return None
222
222
223 if (thisDate == endDate) and numpy.all(thisUtcTime > endUtcTime):
223 if (thisDate == endDate) and numpy.all(thisUtcTime > endUtcTime):
224 return None
224 return None
225
225
226 if numpy.all(thisUtcTime < startUtcTime) and numpy.all(thisUtcTime > endUtcTime):
226 if numpy.all(thisUtcTime < startUtcTime) and numpy.all(thisUtcTime > endUtcTime):
227 return None
227 return None
228
228
229 return thisDatetime
229 return thisDatetime
230
230
231 def __setNextFileOffline(self):
231 def __setNextFileOffline(self):
232
232
233 self.fileIndex += 1
233 self.fileIndex += 1
234 idFile = self.fileIndex
234 idFile = self.fileIndex
235
235
236 if not(idFile < len(self.filenameList)):
236 if not(idFile < len(self.filenameList)):
237 raise schainpy.admin.SchainError("No more Files")
237 raise schainpy.admin.SchainError("No more Files")
238 return 0
238 return 0
239
239
240 filename = self.filenameList[idFile]
240 filename = self.filenameList[idFile]
241 filePointer = h5py.File(filename,'r')
241 filePointer = h5py.File(filename,'r')
242 self.filename = filename
242 self.filename = filename
243 self.fp = filePointer
243 self.fp = filePointer
244
244
245 print("Setting the file: %s"%self.filename)
245 print("Setting the file: %s"%self.filename)
246
246
247 self.__setBlockList()
247 self.__setBlockList()
248 self.__readData()
248 self.__readData()
249 self.blockIndex = 0
249 self.blockIndex = 0
250 return 1
250 return 1
251
251
252 def __setBlockList(self):
252 def __setBlockList(self):
253 '''
253 '''
254 Selects the data within the times defined
254 Selects the data within the times defined
255
255
256 self.fp
256 self.fp
257 self.startTime
257 self.startTime
258 self.endTime
258 self.endTime
259
259
260 self.blockList
260 self.blockList
261 self.blocksPerFile
261 self.blocksPerFile
262
262
263 '''
263 '''
264 fp = self.fp
264 fp = self.fp
265 startTime = self.startTime
265 startTime = self.startTime
266 endTime = self.endTime
266 endTime = self.endTime
267
267
268 grp = fp['Data']
268 grp = fp['Data']
269 thisUtcTime = grp['utctime'].value.astype(numpy.float)[0]
269 thisUtcTime = grp['utctime'].value.astype(numpy.float)[0]
270
270
271 #ERROOOOR
271 #ERROOOOR
272 if self.timezone == 'lt':
272 if self.timezone == 'lt':
273 thisUtcTime -= 5*3600
273 thisUtcTime -= 5*3600
274
274
275 thisDatetime = datetime.datetime.fromtimestamp(thisUtcTime[0] + 5*3600)
275 thisDatetime = datetime.datetime.fromtimestamp(thisUtcTime[0] + 5*3600)
276
276
277 thisDate = thisDatetime.date()
277 thisDate = thisDatetime.date()
278 thisTime = thisDatetime.time()
278 thisTime = thisDatetime.time()
279
279
280 startUtcTime = (datetime.datetime.combine(thisDate,startTime) - datetime.datetime(1970, 1, 1)).total_seconds()
280 startUtcTime = (datetime.datetime.combine(thisDate,startTime) - datetime.datetime(1970, 1, 1)).total_seconds()
281 endUtcTime = (datetime.datetime.combine(thisDate,endTime) - datetime.datetime(1970, 1, 1)).total_seconds()
281 endUtcTime = (datetime.datetime.combine(thisDate,endTime) - datetime.datetime(1970, 1, 1)).total_seconds()
282
282
283 ind = numpy.where(numpy.logical_and(thisUtcTime >= startUtcTime, thisUtcTime < endUtcTime))[0]
283 ind = numpy.where(numpy.logical_and(thisUtcTime >= startUtcTime, thisUtcTime < endUtcTime))[0]
284
284
285 self.blockList = ind
285 self.blockList = ind
286 self.blocksPerFile = len(ind)
286 self.blocksPerFile = len(ind)
287
287
288 return
288 return
289
289
290 def __readMetadata(self):
290 def __readMetadata(self):
291 '''
291 '''
292 Reads Metadata
292 Reads Metadata
293
293
294 self.pathMeta
294 self.pathMeta
295 self.listShapes
295 self.listShapes
296 self.listMetaname
296 self.listMetaname
297 self.listMeta
297 self.listMeta
298
298
299 '''
299 '''
300
300
301 filename = self.filenameList[0]
301 filename = self.filenameList[0]
302 fp = h5py.File(filename,'r')
302 fp = h5py.File(filename,'r')
303 gp = fp['Metadata']
303 gp = fp['Metadata']
304
304
305 listMetaname = []
305 listMetaname = []
306 listMetadata = []
306 listMetadata = []
307 for item in list(gp.items()):
307 for item in list(gp.items()):
308 name = item[0]
308 name = item[0]
309
309
310 if name=='array dimensions':
310 if name=='array dimensions':
311 table = gp[name][:]
311 table = gp[name][:]
312 listShapes = {}
312 listShapes = {}
313 for shapes in table:
313 for shapes in table:
314 listShapes[shapes[0]] = numpy.array([shapes[1],shapes[2],shapes[3],shapes[4],shapes[5]])
314 listShapes[shapes[0]] = numpy.array([shapes[1],shapes[2],shapes[3],shapes[4],shapes[5]])
315 else:
315 else:
316 data = gp[name].value
316 data = gp[name].value
317 listMetaname.append(name)
317 listMetaname.append(name)
318 listMetadata.append(data)
318 listMetadata.append(data)
319
319
320 self.listShapes = listShapes
320 self.listShapes = listShapes
321 self.listMetaname = listMetaname
321 self.listMetaname = listMetaname
322 self.listMeta = listMetadata
322 self.listMeta = listMetadata
323
323
324 fp.close()
324 fp.close()
325 return
325 return
326
326
327 def __readData(self):
327 def __readData(self):
328 grp = self.fp['Data']
328 grp = self.fp['Data']
329 listdataname = []
329 listdataname = []
330 listdata = []
330 listdata = []
331
331
332 for item in list(grp.items()):
332 for item in list(grp.items()):
333 name = item[0]
333 name = item[0]
334 listdataname.append(name)
334 listdataname.append(name)
335
335
336 array = self.__setDataArray(grp[name],self.listShapes[name])
336 array = self.__setDataArray(grp[name],self.listShapes[name])
337 listdata.append(array)
337 listdata.append(array)
338
338
339 self.listDataname = listdataname
339 self.listDataname = listdataname
340 self.listData = listdata
340 self.listData = listdata
341 return
341 return
342
342
343 def __setDataArray(self, dataset, shapes):
343 def __setDataArray(self, dataset, shapes):
344
344
345 nDims = shapes[0]
345 nDims = shapes[0]
346 nDim2 = shapes[1] #Dimension 0
346 nDim2 = shapes[1] #Dimension 0
347 nDim1 = shapes[2] #Dimension 1, number of Points or Parameters
347 nDim1 = shapes[2] #Dimension 1, number of Points or Parameters
348 nDim0 = shapes[3] #Dimension 2, number of samples or ranges
348 nDim0 = shapes[3] #Dimension 2, number of samples or ranges
349 mode = shapes[4] #Mode of storing
349 mode = shapes[4] #Mode of storing
350 blockList = self.blockList
350 blockList = self.blockList
351 blocksPerFile = self.blocksPerFile
351 blocksPerFile = self.blocksPerFile
352
352
353 #Depending on what mode the data was stored
353 #Depending on what mode the data was stored
354 if mode == 0: #Divided in channels
354 if mode == 0: #Divided in channels
355 arrayData = dataset.value.astype(numpy.float)[0][blockList]
355 arrayData = dataset.value.astype(numpy.float)[0][blockList]
356 if mode == 1: #Divided in parameter
356 if mode == 1: #Divided in parameter
357 strds = 'table'
357 strds = 'table'
358 nDatas = nDim1
358 nDatas = nDim1
359 newShapes = (blocksPerFile,nDim2,nDim0)
359 newShapes = (blocksPerFile,nDim2,nDim0)
360 elif mode==2: #Concatenated in a table
360 elif mode==2: #Concatenated in a table
361 strds = 'table0'
361 strds = 'table0'
362 arrayData = dataset[strds].value
362 arrayData = dataset[strds].value
363 #Selecting part of the dataset
363 #Selecting part of the dataset
364 utctime = arrayData[:,0]
364 utctime = arrayData[:,0]
365 u, indices = numpy.unique(utctime, return_index=True)
365 u, indices = numpy.unique(utctime, return_index=True)
366
366
367 if blockList.size != indices.size:
367 if blockList.size != indices.size:
368 indMin = indices[blockList[0]]
368 indMin = indices[blockList[0]]
369 if blockList[1] + 1 >= indices.size:
369 if blockList[1] + 1 >= indices.size:
370 arrayData = arrayData[indMin:,:]
370 arrayData = arrayData[indMin:,:]
371 else:
371 else:
372 indMax = indices[blockList[1] + 1]
372 indMax = indices[blockList[1] + 1]
373 arrayData = arrayData[indMin:indMax,:]
373 arrayData = arrayData[indMin:indMax,:]
374 return arrayData
374 return arrayData
375
375
376 # One dimension
376 # One dimension
377 if nDims == 0:
377 if nDims == 0:
378 arrayData = dataset.value.astype(numpy.float)[0][blockList]
378 arrayData = dataset.value.astype(numpy.float)[0][blockList]
379
379
380 # Two dimensions
380 # Two dimensions
381 elif nDims == 2:
381 elif nDims == 2:
382 arrayData = numpy.zeros((blocksPerFile,nDim1,nDim0))
382 arrayData = numpy.zeros((blocksPerFile,nDim1,nDim0))
383 newShapes = (blocksPerFile,nDim0)
383 newShapes = (blocksPerFile,nDim0)
384 nDatas = nDim1
384 nDatas = nDim1
385
385
386 for i in range(nDatas):
386 for i in range(nDatas):
387 data = dataset[strds + str(i)].value
387 data = dataset[strds + str(i)].value
388 arrayData[:,i,:] = data[blockList,:]
388 arrayData[:,i,:] = data[blockList,:]
389
389
390 # Three dimensions
390 # Three dimensions
391 else:
391 else:
392 arrayData = numpy.zeros((blocksPerFile,nDim2,nDim1,nDim0))
392 arrayData = numpy.zeros((blocksPerFile,nDim2,nDim1,nDim0))
393 for i in range(nDatas):
393 for i in range(nDatas):
394
394
395 data = dataset[strds + str(i)].value
395 data = dataset[strds + str(i)].value
396
396
397 for b in range(blockList.size):
397 for b in range(blockList.size):
398 arrayData[b,:,i,:] = data[:,:,blockList[b]]
398 arrayData[b,:,i,:] = data[:,:,blockList[b]]
399
399
400 return arrayData
400 return arrayData
401
401
402 def __setDataOut(self):
402 def __setDataOut(self):
403 listMeta = self.listMeta
403 listMeta = self.listMeta
404 listMetaname = self.listMetaname
404 listMetaname = self.listMetaname
405 listDataname = self.listDataname
405 listDataname = self.listDataname
406 listData = self.listData
406 listData = self.listData
407 listShapes = self.listShapes
407 listShapes = self.listShapes
408
408
409 blockIndex = self.blockIndex
409 blockIndex = self.blockIndex
410 # blockList = self.blockList
410 # blockList = self.blockList
411
411
412 for i in range(len(listMeta)):
412 for i in range(len(listMeta)):
413 setattr(self.dataOut,listMetaname[i],listMeta[i])
413 setattr(self.dataOut,listMetaname[i],listMeta[i])
414
414
415 for j in range(len(listData)):
415 for j in range(len(listData)):
416 nShapes = listShapes[listDataname[j]][0]
416 nShapes = listShapes[listDataname[j]][0]
417 mode = listShapes[listDataname[j]][4]
417 mode = listShapes[listDataname[j]][4]
418 if nShapes == 1:
418 if nShapes == 1:
419 setattr(self.dataOut,listDataname[j],listData[j][blockIndex])
419 setattr(self.dataOut,listDataname[j],listData[j][blockIndex])
420 elif nShapes > 1:
420 elif nShapes > 1:
421 setattr(self.dataOut,listDataname[j],listData[j][blockIndex,:])
421 setattr(self.dataOut,listDataname[j],listData[j][blockIndex,:])
422 elif mode==0:
422 elif mode==0:
423 setattr(self.dataOut,listDataname[j],listData[j][blockIndex])
423 setattr(self.dataOut,listDataname[j],listData[j][blockIndex])
424 #Mode Meteors
424 #Mode Meteors
425 elif mode ==2:
425 elif mode ==2:
426 selectedData = self.__selectDataMode2(listData[j], blockIndex)
426 selectedData = self.__selectDataMode2(listData[j], blockIndex)
427 setattr(self.dataOut, listDataname[j], selectedData)
427 setattr(self.dataOut, listDataname[j], selectedData)
428 return
428 return
429
429
430 def __selectDataMode2(self, data, blockIndex):
430 def __selectDataMode2(self, data, blockIndex):
431 utctime = data[:,0]
431 utctime = data[:,0]
432 aux, indices = numpy.unique(utctime, return_inverse=True)
432 aux, indices = numpy.unique(utctime, return_inverse=True)
433 selInd = numpy.where(indices == blockIndex)[0]
433 selInd = numpy.where(indices == blockIndex)[0]
434 selData = data[selInd,:]
434 selData = data[selInd,:]
435
435
436 return selData
436 return selData
437
437
438 def getData(self):
438 def getData(self):
439
439
440 if self.blockIndex==self.blocksPerFile:
440 if self.blockIndex==self.blocksPerFile:
441 if not( self.__setNextFileOffline() ):
441 if not( self.__setNextFileOffline() ):
442 self.dataOut.flagNoData = True
442 self.dataOut.flagNoData = True
443 return 0
443 return 0
444
444
445 self.__setDataOut()
445 self.__setDataOut()
446 self.dataOut.flagNoData = False
446 self.dataOut.flagNoData = False
447
447
448 self.blockIndex += 1
448 self.blockIndex += 1
449
449
450 return
450 return
451
451
452 def run(self, **kwargs):
452 def run(self, **kwargs):
453
453
454 if not(self.isConfig):
454 if not(self.isConfig):
455 self.setup(**kwargs)
455 self.setup(**kwargs)
456 self.isConfig = True
456 self.isConfig = True
457
457
458 self.getData()
458 self.getData()
459
459
460 return
460 return
461
461
462 @MPDecorator
462 @MPDecorator
463 class ParamWriter(Operation):
463 class ParamWriter(Operation):
464 '''
464 '''
465 HDF5 Writer, stores parameters data in HDF5 format files
465 HDF5 Writer, stores parameters data in HDF5 format files
466
466
467 path: path where the files will be stored
467 path: path where the files will be stored
468 blocksPerFile: number of blocks that will be saved in per HDF5 format file
468 blocksPerFile: number of blocks that will be saved in per HDF5 format file
469 mode: selects the data stacking mode: '0' channels, '1' parameters, '3' table (for meteors)
469 mode: selects the data stacking mode: '0' channels, '1' parameters, '3' table (for meteors)
470 metadataList: list of attributes that will be stored as metadata
470 metadataList: list of attributes that will be stored as metadata
471 dataList: list of attributes that will be stores as data
471 dataList: list of attributes that will be stores as data
472 '''
472 '''
473
473
474 ext = ".hdf5"
474 ext = ".hdf5"
475 optchar = "D"
475 optchar = "D"
476 metaoptchar = "M"
476 metaoptchar = "M"
477 metaFile = None
477 metaFile = None
478 filename = None
478 filename = None
479 path = None
479 path = None
480 setFile = None
480 setFile = None
481 fp = None
481 fp = None
482 grp = None
482 grp = None
483 ds = None
483 ds = None
484 firsttime = True
484 firsttime = True
485 #Configurations
485 #Configurations
486 blocksPerFile = None
486 blocksPerFile = None
487 blockIndex = None
487 blockIndex = None
488 dataOut = None
488 dataOut = None
489 #Data Arrays
489 #Data Arrays
490 dataList = None
490 dataList = None
491 metadataList = None
491 metadataList = None
492 dsList = None #List of dictionaries with dataset properties
492 dsList = None #List of dictionaries with dataset properties
493 tableDim = None
493 tableDim = None
494 dtype = [('arrayName', 'S20'),('nDimensions', 'i'), ('dim2', 'i'), ('dim1', 'i'),('dim0', 'i'),('mode', 'b')]
494 dtype = [('arrayName', 'S20'),('nDimensions', 'i'), ('dim2', 'i'), ('dim1', 'i'),('dim0', 'i'),('mode', 'b')]
495 currentDay = None
495 currentDay = None
496 lastTime = None
496 lastTime = None
497 setType = None
497 setType = None
498
498
499 def __init__(self):
499 def __init__(self):
500
500
501 Operation.__init__(self)
501 Operation.__init__(self)
502 return
502 return
503
503
504 def setup(self, dataOut, path=None, blocksPerFile=10, metadataList=None, dataList=None, mode=None, setType=None):
504 def setup(self, dataOut, path=None, blocksPerFile=10, metadataList=None, dataList=None, mode=None, setType=None):
505 self.path = path
505 self.path = path
506 self.blocksPerFile = blocksPerFile
506 self.blocksPerFile = blocksPerFile
507 self.metadataList = metadataList
507 self.metadataList = metadataList
508 self.dataList = dataList
508 self.dataList = dataList
509 self.dataOut = dataOut
509 self.dataOut = dataOut
510 self.mode = mode
510 self.mode = mode
511 if self.mode is not None:
511 if self.mode is not None:
512 self.mode = numpy.zeros(len(self.dataList)) + mode
512 self.mode = numpy.zeros(len(self.dataList)) + mode
513 else:
513 else:
514 self.mode = numpy.ones(len(self.dataList))
514 self.mode = numpy.ones(len(self.dataList))
515
515
516 self.setType = setType
516 self.setType = setType
517
517
518 arrayDim = numpy.zeros((len(self.dataList),5))
518 arrayDim = numpy.zeros((len(self.dataList),5))
519
519
520 #Table dimensions
520 #Table dimensions
521 dtype0 = self.dtype
521 dtype0 = self.dtype
522 tableList = []
522 tableList = []
523
523
524 #Dictionary and list of tables
524 #Dictionary and list of tables
525 dsList = []
525 dsList = []
526
526
527 for i in range(len(self.dataList)):
527 for i in range(len(self.dataList)):
528 dsDict = {}
528 dsDict = {}
529 dataAux = getattr(self.dataOut, self.dataList[i])
529 dataAux = getattr(self.dataOut, self.dataList[i])
530 dsDict['variable'] = self.dataList[i]
530 dsDict['variable'] = self.dataList[i]
531 #--------------------- Conditionals ------------------------
531 #--------------------- Conditionals ------------------------
532 #There is no data
532 #There is no data
533
533
534 if dataAux is None:
534 if dataAux is None:
535
535
536 return 0
536 return 0
537
537
538 if isinstance(dataAux, (int, float, numpy.integer, numpy.float)):
538 if isinstance(dataAux, (int, float, numpy.integer, numpy.float)):
539 dsDict['mode'] = 0
539 dsDict['mode'] = 0
540 dsDict['nDim'] = 0
540 dsDict['nDim'] = 0
541 arrayDim[i,0] = 0
541 arrayDim[i,0] = 0
542 dsList.append(dsDict)
542 dsList.append(dsDict)
543
543
544 #Mode 2: meteors
544 #Mode 2: meteors
545 elif self.mode[i] == 2:
545 elif self.mode[i] == 2:
546 dsDict['dsName'] = 'table0'
546 dsDict['dsName'] = 'table0'
547 dsDict['mode'] = 2 # Mode meteors
547 dsDict['mode'] = 2 # Mode meteors
548 dsDict['shape'] = dataAux.shape[-1]
548 dsDict['shape'] = dataAux.shape[-1]
549 dsDict['nDim'] = 0
549 dsDict['nDim'] = 0
550 dsDict['dsNumber'] = 1
550 dsDict['dsNumber'] = 1
551 arrayDim[i,3] = dataAux.shape[-1]
551 arrayDim[i,3] = dataAux.shape[-1]
552 arrayDim[i,4] = self.mode[i] #Mode the data was stored
552 arrayDim[i,4] = self.mode[i] #Mode the data was stored
553 dsList.append(dsDict)
553 dsList.append(dsDict)
554
554
555 #Mode 1
555 #Mode 1
556 else:
556 else:
557 arrayDim0 = dataAux.shape #Data dimensions
557 arrayDim0 = dataAux.shape #Data dimensions
558 arrayDim[i,0] = len(arrayDim0) #Number of array dimensions
558 arrayDim[i,0] = len(arrayDim0) #Number of array dimensions
559 arrayDim[i,4] = self.mode[i] #Mode the data was stored
559 arrayDim[i,4] = self.mode[i] #Mode the data was stored
560 strtable = 'table'
560 strtable = 'table'
561 dsDict['mode'] = 1 # Mode parameters
561 dsDict['mode'] = 1 # Mode parameters
562
562
563 # Three-dimension arrays
563 # Three-dimension arrays
564 if len(arrayDim0) == 3:
564 if len(arrayDim0) == 3:
565 arrayDim[i,1:-1] = numpy.array(arrayDim0)
565 arrayDim[i,1:-1] = numpy.array(arrayDim0)
566 nTables = int(arrayDim[i,2])
566 nTables = int(arrayDim[i,2])
567 dsDict['dsNumber'] = nTables
567 dsDict['dsNumber'] = nTables
568 dsDict['shape'] = arrayDim[i,2:4]
568 dsDict['shape'] = arrayDim[i,2:4]
569 dsDict['nDim'] = 3
569 dsDict['nDim'] = 3
570
570
571 for j in range(nTables):
571 for j in range(nTables):
572 dsDict = dsDict.copy()
572 dsDict = dsDict.copy()
573 dsDict['dsName'] = strtable + str(j)
573 dsDict['dsName'] = strtable + str(j)
574 dsList.append(dsDict)
574 dsList.append(dsDict)
575
575
576 # Two-dimension arrays
576 # Two-dimension arrays
577 elif len(arrayDim0) == 2:
577 elif len(arrayDim0) == 2:
578 arrayDim[i,2:-1] = numpy.array(arrayDim0)
578 arrayDim[i,2:-1] = numpy.array(arrayDim0)
579 nTables = int(arrayDim[i,2])
579 nTables = int(arrayDim[i,2])
580 dsDict['dsNumber'] = nTables
580 dsDict['dsNumber'] = nTables
581 dsDict['shape'] = arrayDim[i,3]
581 dsDict['shape'] = arrayDim[i,3]
582 dsDict['nDim'] = 2
582 dsDict['nDim'] = 2
583
583
584 for j in range(nTables):
584 for j in range(nTables):
585 dsDict = dsDict.copy()
585 dsDict = dsDict.copy()
586 dsDict['dsName'] = strtable + str(j)
586 dsDict['dsName'] = strtable + str(j)
587 dsList.append(dsDict)
587 dsList.append(dsDict)
588
588
589 # One-dimension arrays
589 # One-dimension arrays
590 elif len(arrayDim0) == 1:
590 elif len(arrayDim0) == 1:
591 arrayDim[i,3] = arrayDim0[0]
591 arrayDim[i,3] = arrayDim0[0]
592 dsDict['shape'] = arrayDim0[0]
592 dsDict['shape'] = arrayDim0[0]
593 dsDict['dsNumber'] = 1
593 dsDict['dsNumber'] = 1
594 dsDict['dsName'] = strtable + str(0)
594 dsDict['dsName'] = strtable + str(0)
595 dsDict['nDim'] = 1
595 dsDict['nDim'] = 1
596 dsList.append(dsDict)
596 dsList.append(dsDict)
597
597
598 table = numpy.array((self.dataList[i],) + tuple(arrayDim[i,:]),dtype = dtype0)
598 table = numpy.array((self.dataList[i],) + tuple(arrayDim[i,:]),dtype = dtype0)
599 tableList.append(table)
599 tableList.append(table)
600
600
601 self.dsList = dsList
601 self.dsList = dsList
602 self.tableDim = numpy.array(tableList, dtype = dtype0)
602 self.tableDim = numpy.array(tableList, dtype = dtype0)
603 self.blockIndex = 0
603 self.blockIndex = 0
604 timeTuple = time.localtime(dataOut.utctime)
604 timeTuple = time.localtime(dataOut.utctime)
605 self.currentDay = timeTuple.tm_yday
605 self.currentDay = timeTuple.tm_yday
606
606
607 def putMetadata(self):
607 def putMetadata(self):
608
608
609 fp = self.createMetadataFile()
609 fp = self.createMetadataFile()
610 self.writeMetadata(fp)
610 self.writeMetadata(fp)
611 fp.close()
611 fp.close()
612 return
612 return
613
613
614 def createMetadataFile(self):
614 def createMetadataFile(self):
615 ext = self.ext
615 ext = self.ext
616 path = self.path
616 path = self.path
617 setFile = self.setFile
617 setFile = self.setFile
618
618
619 timeTuple = time.localtime(self.dataOut.utctime)
619 timeTuple = time.localtime(self.dataOut.utctime)
620
620
621 subfolder = ''
621 subfolder = ''
622 fullpath = os.path.join( path, subfolder )
622 fullpath = os.path.join( path, subfolder )
623
623
624 if not( os.path.exists(fullpath) ):
624 if not( os.path.exists(fullpath) ):
625 os.mkdir(fullpath)
625 os.mkdir(fullpath)
626 setFile = -1 #inicializo mi contador de seteo
626 setFile = -1 #inicializo mi contador de seteo
627
627
628 subfolder = 'd%4.4d%3.3d' % (timeTuple.tm_year,timeTuple.tm_yday)
628 subfolder = 'd%4.4d%3.3d' % (timeTuple.tm_year,timeTuple.tm_yday)
629 fullpath = os.path.join( path, subfolder )
629 fullpath = os.path.join( path, subfolder )
630
630
631 if not( os.path.exists(fullpath) ):
631 if not( os.path.exists(fullpath) ):
632 os.mkdir(fullpath)
632 os.mkdir(fullpath)
633 setFile = -1 #inicializo mi contador de seteo
633 setFile = -1 #inicializo mi contador de seteo
634
634
635 else:
635 else:
636 filesList = os.listdir( fullpath )
636 filesList = os.listdir( fullpath )
637 filesList = sorted( filesList, key=str.lower )
637 filesList = sorted( filesList, key=str.lower )
638 if len( filesList ) > 0:
638 if len( filesList ) > 0:
639 filesList = [k for k in filesList if k.startswith(self.metaoptchar)]
639 filesList = [k for k in filesList if k.startswith(self.metaoptchar)]
640 filen = filesList[-1]
640 filen = filesList[-1]
641 # el filename debera tener el siguiente formato
641 # el filename debera tener el siguiente formato
642 # 0 1234 567 89A BCDE (hex)
642 # 0 1234 567 89A BCDE (hex)
643 # x YYYY DDD SSS .ext
643 # x YYYY DDD SSS .ext
644 if isNumber( filen[8:11] ):
644 if isNumber( filen[8:11] ):
645 setFile = int( filen[8:11] ) #inicializo mi contador de seteo al seteo del ultimo file
645 setFile = int( filen[8:11] ) #inicializo mi contador de seteo al seteo del ultimo file
646 else:
646 else:
647 setFile = -1
647 setFile = -1
648 else:
648 else:
649 setFile = -1 #inicializo mi contador de seteo
649 setFile = -1 #inicializo mi contador de seteo
650
650
651 if self.setType is None:
651 if self.setType is None:
652 setFile += 1
652 setFile += 1
653 file = '%s%4.4d%3.3d%03d%s' % (self.metaoptchar,
653 file = '%s%4.4d%3.3d%03d%s' % (self.metaoptchar,
654 timeTuple.tm_year,
654 timeTuple.tm_year,
655 timeTuple.tm_yday,
655 timeTuple.tm_yday,
656 setFile,
656 setFile,
657 ext )
657 ext )
658 else:
658 else:
659 setFile = timeTuple.tm_hour*60+timeTuple.tm_min
659 setFile = timeTuple.tm_hour*60+timeTuple.tm_min
660 file = '%s%4.4d%3.3d%04d%s' % (self.metaoptchar,
660 file = '%s%4.4d%3.3d%04d%s' % (self.metaoptchar,
661 timeTuple.tm_year,
661 timeTuple.tm_year,
662 timeTuple.tm_yday,
662 timeTuple.tm_yday,
663 setFile,
663 setFile,
664 ext )
664 ext )
665
665
666 filename = os.path.join( path, subfolder, file )
666 filename = os.path.join( path, subfolder, file )
667 self.metaFile = file
667 self.metaFile = file
668 #Setting HDF5 File
668 #Setting HDF5 File
669 fp = h5py.File(filename,'w')
669 fp = h5py.File(filename,'w')
670
670
671 return fp
671 return fp
672
672
673 def writeMetadata(self, fp):
673 def writeMetadata(self, fp):
674
674
675 grp = fp.create_group("Metadata")
675 grp = fp.create_group("Metadata")
676 grp.create_dataset('array dimensions', data = self.tableDim, dtype = self.dtype)
676 grp.create_dataset('array dimensions', data = self.tableDim, dtype = self.dtype)
677
677
678 for i in range(len(self.metadataList)):
678 for i in range(len(self.metadataList)):
679 grp.create_dataset(self.metadataList[i], data=getattr(self.dataOut, self.metadataList[i]))
679 grp.create_dataset(self.metadataList[i], data=getattr(self.dataOut, self.metadataList[i]))
680 return
680 return
681
681
682 def timeFlag(self):
682 def timeFlag(self):
683 currentTime = self.dataOut.utctime
683 currentTime = self.dataOut.utctime
684
684
685 if self.lastTime is None:
685 if self.lastTime is None:
686 self.lastTime = currentTime
686 self.lastTime = currentTime
687
687
688 #Day
688 #Day
689 timeTuple = time.localtime(currentTime)
689 timeTuple = time.localtime(currentTime)
690 dataDay = timeTuple.tm_yday
690 dataDay = timeTuple.tm_yday
691
691
692 #Time
692 #Time
693 timeDiff = currentTime - self.lastTime
693 timeDiff = currentTime - self.lastTime
694
694
695 #Si el dia es diferente o si la diferencia entre un dato y otro supera la hora
695 #Si el dia es diferente o si la diferencia entre un dato y otro supera la hora
696 if dataDay != self.currentDay:
696 if dataDay != self.currentDay:
697 self.currentDay = dataDay
697 self.currentDay = dataDay
698 return True
698 return True
699 elif timeDiff > 3*60*60:
699 elif timeDiff > 3*60*60:
700 self.lastTime = currentTime
700 self.lastTime = currentTime
701 return True
701 return True
702 else:
702 else:
703 self.lastTime = currentTime
703 self.lastTime = currentTime
704 return False
704 return False
705
705
706 def setNextFile(self):
706 def setNextFile(self):
707
707
708 ext = self.ext
708 ext = self.ext
709 path = self.path
709 path = self.path
710 setFile = self.setFile
710 setFile = self.setFile
711 mode = self.mode
711 mode = self.mode
712
712
713 timeTuple = time.localtime(self.dataOut.utctime)
713 timeTuple = time.localtime(self.dataOut.utctime)
714 subfolder = 'd%4.4d%3.3d' % (timeTuple.tm_year,timeTuple.tm_yday)
714 subfolder = 'd%4.4d%3.3d' % (timeTuple.tm_year,timeTuple.tm_yday)
715
715
716 fullpath = os.path.join( path, subfolder )
716 fullpath = os.path.join( path, subfolder )
717
717
718 if os.path.exists(fullpath):
718 if os.path.exists(fullpath):
719 filesList = os.listdir( fullpath )
719 filesList = os.listdir( fullpath )
720 filesList = [k for k in filesList if 'M' in k]
720 ##filesList = [k for k in filesList if 'M' in k]
721 if len( filesList ) > 0:
721 if len( filesList ) > 0:
722 filesList = sorted( filesList, key=str.lower )
722 filesList = sorted( filesList, key=str.lower )
723 filen = filesList[-1]
723 filen = filesList[-1]
724 # el filename debera tener el siguiente formato
724 # el filename debera tener el siguiente formato
725 # 0 1234 567 89A BCDE (hex)
725 # 0 1234 567 89A BCDE (hex)
726 # x YYYY DDD SSS .ext
726 # x YYYY DDD SSS .ext
727 if isNumber( filen[8:11] ):
727 if isNumber( filen[8:11] ):
728 setFile = int( filen[8:11] ) #inicializo mi contador de seteo al seteo del ultimo file
728 setFile = int( filen[8:11] ) #inicializo mi contador de seteo al seteo del ultimo file
729 else:
729 else:
730 setFile = -1
730 setFile = -1
731 else:
731 else:
732 setFile = -1 #inicializo mi contador de seteo
732 setFile = -1 #inicializo mi contador de seteo
733 else:
733 else:
734 os.makedirs(fullpath)
734 os.makedirs(fullpath)
735 setFile = -1 #inicializo mi contador de seteo
735 setFile = -1 #inicializo mi contador de seteo
736
736
737 if self.setType is None:
737 if self.setType is None:
738 setFile += 1
738 setFile += 1
739 file = '%s%4.4d%3.3d%03d%s' % (self.optchar,
739 file = '%s%4.4d%3.3d%03d%s' % (self.optchar,
740 timeTuple.tm_year,
740 timeTuple.tm_year,
741 timeTuple.tm_yday,
741 timeTuple.tm_yday,
742 setFile,
742 setFile,
743 ext )
743 ext )
744 else:
744 else:
745 setFile = timeTuple.tm_hour*60+timeTuple.tm_min
745 setFile = timeTuple.tm_hour*60+timeTuple.tm_min
746 file = '%s%4.4d%3.3d%04d%s' % (self.optchar,
746 file = '%s%4.4d%3.3d%04d%s' % (self.optchar,
747 timeTuple.tm_year,
747 timeTuple.tm_year,
748 timeTuple.tm_yday,
748 timeTuple.tm_yday,
749 setFile,
749 setFile,
750 ext )
750 ext )
751
751
752 filename = os.path.join( path, subfolder, file )
752 filename = os.path.join( path, subfolder, file )
753
753
754 #Setting HDF5 File
754 #Setting HDF5 File
755 fp = h5py.File(filename,'w')
755 fp = h5py.File(filename,'w')
756 #write metadata
756 #write metadata
757 self.writeMetadata(fp)
757 self.writeMetadata(fp)
758 #Write data
758 #Write data
759 grp = fp.create_group("Data")
759 grp = fp.create_group("Data")
760 ds = []
760 ds = []
761 data = []
761 data = []
762 dsList = self.dsList
762 dsList = self.dsList
763 i = 0
763 i = 0
764 while i < len(dsList):
764 while i < len(dsList):
765 dsInfo = dsList[i]
765 dsInfo = dsList[i]
766 #One-dimension data
766 #One-dimension data
767 if dsInfo['mode'] == 0:
767 if dsInfo['mode'] == 0:
768 ds0 = grp.create_dataset(dsInfo['variable'], (1,1), maxshape=(1,self.blocksPerFile) , chunks = True, dtype=numpy.float64)
768 ds0 = grp.create_dataset(dsInfo['variable'], (1,1), maxshape=(1,self.blocksPerFile) , chunks = True, dtype=numpy.float64)
769 ds.append(ds0)
769 ds.append(ds0)
770 data.append([])
770 data.append([])
771 i += 1
771 i += 1
772 continue
772 continue
773
773
774 elif dsInfo['mode'] == 2:
774 elif dsInfo['mode'] == 2:
775 grp0 = grp.create_group(dsInfo['variable'])
775 grp0 = grp.create_group(dsInfo['variable'])
776 ds0 = grp0.create_dataset(dsInfo['dsName'], (1,dsInfo['shape']), data = numpy.zeros((1,dsInfo['shape'])) , maxshape=(None,dsInfo['shape']), chunks=True)
776 ds0 = grp0.create_dataset(dsInfo['dsName'], (1,dsInfo['shape']), data = numpy.zeros((1,dsInfo['shape'])) , maxshape=(None,dsInfo['shape']), chunks=True)
777 ds.append(ds0)
777 ds.append(ds0)
778 data.append([])
778 data.append([])
779 i += 1
779 i += 1
780 continue
780 continue
781
781
782 elif dsInfo['mode'] == 1:
782 elif dsInfo['mode'] == 1:
783 grp0 = grp.create_group(dsInfo['variable'])
783 grp0 = grp.create_group(dsInfo['variable'])
784
784
785 for j in range(dsInfo['dsNumber']):
785 for j in range(dsInfo['dsNumber']):
786 dsInfo = dsList[i]
786 dsInfo = dsList[i]
787 tableName = dsInfo['dsName']
787 tableName = dsInfo['dsName']
788
788
789
789
790 if dsInfo['nDim'] == 3:
790 if dsInfo['nDim'] == 3:
791 shape = dsInfo['shape'].astype(int)
791 shape = dsInfo['shape'].astype(int)
792 ds0 = grp0.create_dataset(tableName, (shape[0],shape[1],1) , data = numpy.zeros((shape[0],shape[1],1)), maxshape = (None,shape[1],None), chunks=True)
792 ds0 = grp0.create_dataset(tableName, (shape[0],shape[1],1) , data = numpy.zeros((shape[0],shape[1],1)), maxshape = (None,shape[1],None), chunks=True)
793 else:
793 else:
794 shape = int(dsInfo['shape'])
794 shape = int(dsInfo['shape'])
795 ds0 = grp0.create_dataset(tableName, (1,shape), data = numpy.zeros((1,shape)) , maxshape=(None,shape), chunks=True)
795 ds0 = grp0.create_dataset(tableName, (1,shape), data = numpy.zeros((1,shape)) , maxshape=(None,shape), chunks=True)
796
796
797 ds.append(ds0)
797 ds.append(ds0)
798 data.append([])
798 data.append([])
799 i += 1
799 i += 1
800
800
801 fp.flush()
801 fp.flush()
802 fp.close()
802 fp.close()
803
803
804 log.log('creating file: {}'.format(filename), 'Writing')
804 log.log('creating file: {}'.format(filename), 'Writing')
805 self.filename = filename
805 self.filename = filename
806 self.ds = ds
806 self.ds = ds
807 self.data = data
807 self.data = data
808 self.firsttime = True
808 self.firsttime = True
809 self.blockIndex = 0
809 self.blockIndex = 0
810 return
810 return
811
811
812 def putData(self):
812 def putData(self):
813
813
814 if self.blockIndex == self.blocksPerFile or self.timeFlag():
814 if self.blockIndex == self.blocksPerFile or self.timeFlag():
815 self.setNextFile()
815 self.setNextFile()
816
816
817 self.readBlock()
817 self.readBlock()
818 self.setBlock() #Prepare data to be written
818 self.setBlock() #Prepare data to be written
819 self.writeBlock() #Write data
819 self.writeBlock() #Write data
820
820
821 return
821 return
822
822
823 def readBlock(self):
823 def readBlock(self):
824
824
825 '''
825 '''
826 data Array configured
826 data Array configured
827
827
828
828
829 self.data
829 self.data
830 '''
830 '''
831 dsList = self.dsList
831 dsList = self.dsList
832 ds = self.ds
832 ds = self.ds
833 #Setting HDF5 File
833 #Setting HDF5 File
834 fp = h5py.File(self.filename,'r+')
834 fp = h5py.File(self.filename,'r+')
835 grp = fp["Data"]
835 grp = fp["Data"]
836 ind = 0
836 ind = 0
837
837
838 while ind < len(dsList):
838 while ind < len(dsList):
839 dsInfo = dsList[ind]
839 dsInfo = dsList[ind]
840
840
841 if dsInfo['mode'] == 0:
841 if dsInfo['mode'] == 0:
842 ds0 = grp[dsInfo['variable']]
842 ds0 = grp[dsInfo['variable']]
843 ds[ind] = ds0
843 ds[ind] = ds0
844 ind += 1
844 ind += 1
845 else:
845 else:
846
846
847 grp0 = grp[dsInfo['variable']]
847 grp0 = grp[dsInfo['variable']]
848
848
849 for j in range(dsInfo['dsNumber']):
849 for j in range(dsInfo['dsNumber']):
850 dsInfo = dsList[ind]
850 dsInfo = dsList[ind]
851 ds0 = grp0[dsInfo['dsName']]
851 ds0 = grp0[dsInfo['dsName']]
852 ds[ind] = ds0
852 ds[ind] = ds0
853 ind += 1
853 ind += 1
854
854
855 self.fp = fp
855 self.fp = fp
856 self.grp = grp
856 self.grp = grp
857 self.ds = ds
857 self.ds = ds
858
858
859 return
859 return
860
860
861 def setBlock(self):
861 def setBlock(self):
862 '''
862 '''
863 data Array configured
863 data Array configured
864
864
865
865
866 self.data
866 self.data
867 '''
867 '''
868 #Creating Arrays
868 #Creating Arrays
869 dsList = self.dsList
869 dsList = self.dsList
870 data = self.data
870 data = self.data
871 ind = 0
871 ind = 0
872
872
873 while ind < len(dsList):
873 while ind < len(dsList):
874 dsInfo = dsList[ind]
874 dsInfo = dsList[ind]
875 dataAux = getattr(self.dataOut, dsInfo['variable'])
875 dataAux = getattr(self.dataOut, dsInfo['variable'])
876
876
877 mode = dsInfo['mode']
877 mode = dsInfo['mode']
878 nDim = dsInfo['nDim']
878 nDim = dsInfo['nDim']
879
879
880 if mode == 0 or mode == 2 or nDim == 1:
880 if mode == 0 or mode == 2 or nDim == 1:
881 data[ind] = dataAux
881 data[ind] = dataAux
882 ind += 1
882 ind += 1
883 # elif nDim == 1:
883 # elif nDim == 1:
884 # data[ind] = numpy.reshape(dataAux,(numpy.size(dataAux),1))
884 # data[ind] = numpy.reshape(dataAux,(numpy.size(dataAux),1))
885 # ind += 1
885 # ind += 1
886 elif nDim == 2:
886 elif nDim == 2:
887 for j in range(dsInfo['dsNumber']):
887 for j in range(dsInfo['dsNumber']):
888 data[ind] = dataAux[j,:]
888 data[ind] = dataAux[j,:]
889 ind += 1
889 ind += 1
890 elif nDim == 3:
890 elif nDim == 3:
891 for j in range(dsInfo['dsNumber']):
891 for j in range(dsInfo['dsNumber']):
892 data[ind] = dataAux[:,j,:]
892 data[ind] = dataAux[:,j,:]
893 ind += 1
893 ind += 1
894
894
895 self.data = data
895 self.data = data
896 return
896 return
897
897
898 def writeBlock(self):
898 def writeBlock(self):
899 '''
899 '''
900 Saves the block in the HDF5 file
900 Saves the block in the HDF5 file
901 '''
901 '''
902 dsList = self.dsList
902 dsList = self.dsList
903
903
904 for i in range(len(self.ds)):
904 for i in range(len(self.ds)):
905 dsInfo = dsList[i]
905 dsInfo = dsList[i]
906 nDim = dsInfo['nDim']
906 nDim = dsInfo['nDim']
907 mode = dsInfo['mode']
907 mode = dsInfo['mode']
908
908
909 # First time
909 # First time
910 if self.firsttime:
910 if self.firsttime:
911 if type(self.data[i]) == numpy.ndarray:
911 if type(self.data[i]) == numpy.ndarray:
912
912
913 if nDim == 3:
913 if nDim == 3:
914 self.data[i] = self.data[i].reshape((self.data[i].shape[0],self.data[i].shape[1],1))
914 self.data[i] = self.data[i].reshape((self.data[i].shape[0],self.data[i].shape[1],1))
915 self.ds[i].resize(self.data[i].shape)
915 self.ds[i].resize(self.data[i].shape)
916 if mode == 2:
916 if mode == 2:
917 self.ds[i].resize(self.data[i].shape)
917 self.ds[i].resize(self.data[i].shape)
918 self.ds[i][:] = self.data[i]
918 self.ds[i][:] = self.data[i]
919 else:
919 else:
920
920
921 # From second time
921 # From second time
922 # Meteors!
922 # Meteors!
923 if mode == 2:
923 if mode == 2:
924 dataShape = self.data[i].shape
924 dataShape = self.data[i].shape
925 dsShape = self.ds[i].shape
925 dsShape = self.ds[i].shape
926 self.ds[i].resize((self.ds[i].shape[0] + dataShape[0],self.ds[i].shape[1]))
926 self.ds[i].resize((self.ds[i].shape[0] + dataShape[0],self.ds[i].shape[1]))
927 self.ds[i][dsShape[0]:,:] = self.data[i]
927 self.ds[i][dsShape[0]:,:] = self.data[i]
928 # No dimension
928 # No dimension
929 elif mode == 0:
929 elif mode == 0:
930 self.ds[i].resize((self.ds[i].shape[0], self.ds[i].shape[1] + 1))
930 self.ds[i].resize((self.ds[i].shape[0], self.ds[i].shape[1] + 1))
931 self.ds[i][0,-1] = self.data[i]
931 self.ds[i][0,-1] = self.data[i]
932 # One dimension
932 # One dimension
933 elif nDim == 1:
933 elif nDim == 1:
934 self.ds[i].resize((self.ds[i].shape[0] + 1, self.ds[i].shape[1]))
934 self.ds[i].resize((self.ds[i].shape[0] + 1, self.ds[i].shape[1]))
935 self.ds[i][-1,:] = self.data[i]
935 self.ds[i][-1,:] = self.data[i]
936 # Two dimension
936 # Two dimension
937 elif nDim == 2:
937 elif nDim == 2:
938 self.ds[i].resize((self.ds[i].shape[0] + 1,self.ds[i].shape[1]))
938 self.ds[i].resize((self.ds[i].shape[0] + 1,self.ds[i].shape[1]))
939 self.ds[i][self.blockIndex,:] = self.data[i]
939 self.ds[i][self.blockIndex,:] = self.data[i]
940 # Three dimensions
940 # Three dimensions
941 elif nDim == 3:
941 elif nDim == 3:
942 self.ds[i].resize((self.ds[i].shape[0],self.ds[i].shape[1],self.ds[i].shape[2]+1))
942 self.ds[i].resize((self.ds[i].shape[0],self.ds[i].shape[1],self.ds[i].shape[2]+1))
943 self.ds[i][:,:,-1] = self.data[i]
943 self.ds[i][:,:,-1] = self.data[i]
944
944
945 self.firsttime = False
945 self.firsttime = False
946 self.blockIndex += 1
946 self.blockIndex += 1
947
947
948 #Close to save changes
948 #Close to save changes
949 self.fp.flush()
949 self.fp.flush()
950 self.fp.close()
950 self.fp.close()
951 return
951 return
952
952
953 def run(self, dataOut, path, blocksPerFile=10, metadataList=None, dataList=None, mode=None, setType=None):
953 def run(self, dataOut, path, blocksPerFile=10, metadataList=None, dataList=None, mode=None, setType=None):
954
954
955 self.dataOut = dataOut
955 self.dataOut = dataOut
956 if not(self.isConfig):
956 if not(self.isConfig):
957 self.setup(dataOut, path=path, blocksPerFile=blocksPerFile,
957 self.setup(dataOut, path=path, blocksPerFile=blocksPerFile,
958 metadataList=metadataList, dataList=dataList, mode=mode,
958 metadataList=metadataList, dataList=dataList, mode=mode,
959 setType=setType)
959 setType=setType)
960
960
961 self.isConfig = True
961 self.isConfig = True
962 self.setNextFile()
962 self.setNextFile()
963
963
964 self.putData()
964 self.putData()
965 return
965 return
966
966
967
967
968 @MPDecorator
968 @MPDecorator
969 class ParameterReader(Reader, ProcessingUnit):
969 class ParameterReader(Reader, ProcessingUnit):
970 '''
970 '''
971 Reads HDF5 format files
971 Reads HDF5 format files
972 '''
972 '''
973
973
974 def __init__(self):
974 def __init__(self):
975 ProcessingUnit.__init__(self)
975 ProcessingUnit.__init__(self)
976 self.dataOut = Parameters()
976 self.dataOut = Parameters()
977 self.ext = ".hdf5"
977 self.ext = ".hdf5"
978 self.optchar = "D"
978 self.optchar = "D"
979 self.timezone = "lt"
979 self.timezone = "lt"
980 self.listMetaname = []
980 self.listMetaname = []
981 self.listMeta = []
981 self.listMeta = []
982 self.listDataname = []
982 self.listDataname = []
983 self.listData = []
983 self.listData = []
984 self.listShapes = []
984 self.listShapes = []
985 self.open_file = h5py.File
985 self.open_file = h5py.File
986 self.open_mode = 'r'
986 self.open_mode = 'r'
987 self.metadata = False
987 self.metadata = False
988 self.filefmt = "*%Y%j***"
988 self.filefmt = "*%Y%j***"
989 self.folderfmt = "*%Y%j"
989 self.folderfmt = "*%Y%j"
990
990
991 def setup(self, **kwargs):
991 def setup(self, **kwargs):
992
992
993 self.set_kwargs(**kwargs)
993 self.set_kwargs(**kwargs)
994 if not self.ext.startswith('.'):
994 if not self.ext.startswith('.'):
995 self.ext = '.{}'.format(self.ext)
995 self.ext = '.{}'.format(self.ext)
996
996
997 if self.online:
997 if self.online:
998 log.log("Searching files in online mode...", self.name)
998 log.log("Searching files in online mode...", self.name)
999
999
1000 for nTries in range(self.nTries):
1000 for nTries in range(self.nTries):
1001 fullpath = self.searchFilesOnLine(self.path, self.startDate,
1001 fullpath = self.searchFilesOnLine(self.path, self.startDate,
1002 self.endDate, self.expLabel, self.ext, self.walk,
1002 self.endDate, self.expLabel, self.ext, self.walk,
1003 self.filefmt, self.folderfmt)
1003 self.filefmt, self.folderfmt)
1004
1004
1005 try:
1005 try:
1006 fullpath = next(fullpath)
1006 fullpath = next(fullpath)
1007 except:
1007 except:
1008 fullpath = None
1008 fullpath = None
1009
1009
1010 if fullpath:
1010 if fullpath:
1011 break
1011 break
1012
1012
1013 log.warning(
1013 log.warning(
1014 'Waiting {} sec for a valid file in {}: try {} ...'.format(
1014 'Waiting {} sec for a valid file in {}: try {} ...'.format(
1015 self.delay, self.path, nTries + 1),
1015 self.delay, self.path, nTries + 1),
1016 self.name)
1016 self.name)
1017 time.sleep(self.delay)
1017 time.sleep(self.delay)
1018
1018
1019 if not(fullpath):
1019 if not(fullpath):
1020 raise schainpy.admin.SchainError(
1020 raise schainpy.admin.SchainError(
1021 'There isn\'t any valid file in {}'.format(self.path))
1021 'There isn\'t any valid file in {}'.format(self.path))
1022
1022
1023 pathname, filename = os.path.split(fullpath)
1023 pathname, filename = os.path.split(fullpath)
1024 self.year = int(filename[1:5])
1024 self.year = int(filename[1:5])
1025 self.doy = int(filename[5:8])
1025 self.doy = int(filename[5:8])
1026 self.set = int(filename[8:11]) - 1
1026 self.set = int(filename[8:11]) - 1
1027 else:
1027 else:
1028 log.log("Searching files in {}".format(self.path), self.name)
1028 log.log("Searching files in {}".format(self.path), self.name)
1029 self.filenameList = self.searchFilesOffLine(self.path, self.startDate,
1029 self.filenameList = self.searchFilesOffLine(self.path, self.startDate,
1030 self.endDate, self.expLabel, self.ext, self.walk, self.filefmt, self.folderfmt)
1030 self.endDate, self.expLabel, self.ext, self.walk, self.filefmt, self.folderfmt)
1031
1031
1032 self.setNextFile()
1032 self.setNextFile()
1033
1033
1034 return
1034 return
1035
1035
1036 def readFirstHeader(self):
1036 def readFirstHeader(self):
1037 '''Read metadata and data'''
1037 '''Read metadata and data'''
1038
1038
1039 self.__readMetadata()
1039 self.__readMetadata()
1040 self.__readData()
1040 self.__readData()
1041 self.__setBlockList()
1041 self.__setBlockList()
1042 self.blockIndex = 0
1042 self.blockIndex = 0
1043
1043
1044 return
1044 return
1045
1045
1046 def __setBlockList(self):
1046 def __setBlockList(self):
1047 '''
1047 '''
1048 Selects the data within the times defined
1048 Selects the data within the times defined
1049
1049
1050 self.fp
1050 self.fp
1051 self.startTime
1051 self.startTime
1052 self.endTime
1052 self.endTime
1053 self.blockList
1053 self.blockList
1054 self.blocksPerFile
1054 self.blocksPerFile
1055
1055
1056 '''
1056 '''
1057
1057
1058 startTime = self.startTime
1058 startTime = self.startTime
1059 endTime = self.endTime
1059 endTime = self.endTime
1060
1060
1061 index = self.listDataname.index('utctime')
1061 index = self.listDataname.index('utctime')
1062 thisUtcTime = self.listData[index]
1062 thisUtcTime = self.listData[index]
1063 self.interval = numpy.min(thisUtcTime[1:] - thisUtcTime[:-1])
1063 self.interval = numpy.min(thisUtcTime[1:] - thisUtcTime[:-1])
1064
1064
1065 if self.timezone == 'lt':
1065 if self.timezone == 'lt':
1066 thisUtcTime -= 5*3600
1066 thisUtcTime -= 5*3600
1067
1067
1068 thisDatetime = datetime.datetime.fromtimestamp(thisUtcTime[0] + 5*3600)
1068 thisDatetime = datetime.datetime.fromtimestamp(thisUtcTime[0] + 5*3600)
1069
1069
1070 thisDate = thisDatetime.date()
1070 thisDate = thisDatetime.date()
1071 thisTime = thisDatetime.time()
1071 thisTime = thisDatetime.time()
1072
1072
1073 startUtcTime = (datetime.datetime.combine(thisDate,startTime) - datetime.datetime(1970, 1, 1)).total_seconds()
1073 startUtcTime = (datetime.datetime.combine(thisDate,startTime) - datetime.datetime(1970, 1, 1)).total_seconds()
1074 endUtcTime = (datetime.datetime.combine(thisDate,endTime) - datetime.datetime(1970, 1, 1)).total_seconds()
1074 endUtcTime = (datetime.datetime.combine(thisDate,endTime) - datetime.datetime(1970, 1, 1)).total_seconds()
1075
1075
1076 ind = numpy.where(numpy.logical_and(thisUtcTime >= startUtcTime, thisUtcTime < endUtcTime))[0]
1076 ind = numpy.where(numpy.logical_and(thisUtcTime >= startUtcTime, thisUtcTime < endUtcTime))[0]
1077
1077
1078 self.blockList = ind
1078 self.blockList = ind
1079 self.blocksPerFile = len(ind)
1079 self.blocksPerFile = len(ind)
1080 return
1080 return
1081
1081
1082 def __readMetadata(self):
1082 def __readMetadata(self):
1083 '''
1083 '''
1084 Reads Metadata
1084 Reads Metadata
1085 '''
1085 '''
1086
1086
1087 listMetaname = []
1087 listMetaname = []
1088 listMetadata = []
1088 listMetadata = []
1089 if 'Metadata' in self.fp:
1089 if 'Metadata' in self.fp:
1090 gp = self.fp['Metadata']
1090 gp = self.fp['Metadata']
1091 for item in list(gp.items()):
1091 for item in list(gp.items()):
1092 name = item[0]
1092 name = item[0]
1093
1093
1094 if name=='variables':
1094 if name=='variables':
1095 table = gp[name][:]
1095 table = gp[name][:]
1096 listShapes = {}
1096 listShapes = {}
1097 for shapes in table:
1097 for shapes in table:
1098 listShapes[shapes[0].decode()] = numpy.array([shapes[1]])
1098 listShapes[shapes[0].decode()] = numpy.array([shapes[1]])
1099 else:
1099 else:
1100 data = gp[name].value
1100 data = gp[name].value
1101 listMetaname.append(name)
1101 listMetaname.append(name)
1102 listMetadata.append(data)
1102 listMetadata.append(data)
1103 elif self.metadata:
1103 elif self.metadata:
1104 metadata = json.loads(self.metadata)
1104 metadata = json.loads(self.metadata)
1105 listShapes = {}
1105 listShapes = {}
1106 for tup in metadata:
1106 for tup in metadata:
1107 name, values, dim = tup
1107 name, values, dim = tup
1108 if dim == -1:
1108 if dim == -1:
1109 listMetaname.append(name)
1109 listMetaname.append(name)
1110 listMetadata.append(self.fp[values].value)
1110 listMetadata.append(self.fp[values].value)
1111 else:
1111 else:
1112 listShapes[name] = numpy.array([dim])
1112 listShapes[name] = numpy.array([dim])
1113 else:
1113 else:
1114 raise IOError('Missing Metadata group in file or metadata info')
1114 raise IOError('Missing Metadata group in file or metadata info')
1115
1115
1116 self.listShapes = listShapes
1116 self.listShapes = listShapes
1117 self.listMetaname = listMetaname
1117 self.listMetaname = listMetaname
1118 self.listMeta = listMetadata
1118 self.listMeta = listMetadata
1119
1119
1120 return
1120 return
1121
1121
1122 def __readData(self):
1122 def __readData(self):
1123
1123
1124 listdataname = []
1124 listdataname = []
1125 listdata = []
1125 listdata = []
1126
1126
1127 if 'Data' in self.fp:
1127 if 'Data' in self.fp:
1128 grp = self.fp['Data']
1128 grp = self.fp['Data']
1129 for item in list(grp.items()):
1129 for item in list(grp.items()):
1130 name = item[0]
1130 name = item[0]
1131 listdataname.append(name)
1131 listdataname.append(name)
1132 dim = self.listShapes[name][0]
1132 dim = self.listShapes[name][0]
1133 if dim == 0:
1133 if dim == 0:
1134 array = grp[name].value
1134 array = grp[name].value
1135 else:
1135 else:
1136 array = []
1136 array = []
1137 for i in range(dim):
1137 for i in range(dim):
1138 array.append(grp[name]['table{:02d}'.format(i)].value)
1138 array.append(grp[name]['table{:02d}'.format(i)].value)
1139 array = numpy.array(array)
1139 array = numpy.array(array)
1140
1140
1141 listdata.append(array)
1141 listdata.append(array)
1142 elif self.metadata:
1142 elif self.metadata:
1143 metadata = json.loads(self.metadata)
1143 metadata = json.loads(self.metadata)
1144 for tup in metadata:
1144 for tup in metadata:
1145 name, values, dim = tup
1145 name, values, dim = tup
1146 listdataname.append(name)
1146 listdataname.append(name)
1147 if dim == -1:
1147 if dim == -1:
1148 continue
1148 continue
1149 elif dim == 0:
1149 elif dim == 0:
1150 array = self.fp[values].value
1150 array = self.fp[values].value
1151 else:
1151 else:
1152 array = []
1152 array = []
1153 for var in values:
1153 for var in values:
1154 array.append(self.fp[var].value)
1154 array.append(self.fp[var].value)
1155 array = numpy.array(array)
1155 array = numpy.array(array)
1156 listdata.append(array)
1156 listdata.append(array)
1157 else:
1157 else:
1158 raise IOError('Missing Data group in file or metadata info')
1158 raise IOError('Missing Data group in file or metadata info')
1159
1159
1160 self.listDataname = listdataname
1160 self.listDataname = listdataname
1161 self.listData = listdata
1161 self.listData = listdata
1162 return
1162 return
1163
1163
1164 def getData(self):
1164 def getData(self):
1165
1165
1166 for i in range(len(self.listMeta)):
1166 for i in range(len(self.listMeta)):
1167 setattr(self.dataOut, self.listMetaname[i], self.listMeta[i])
1167 setattr(self.dataOut, self.listMetaname[i], self.listMeta[i])
1168
1168
1169 for j in range(len(self.listData)):
1169 for j in range(len(self.listData)):
1170 dim = self.listShapes[self.listDataname[j]][0]
1170 dim = self.listShapes[self.listDataname[j]][0]
1171 if dim == 0:
1171 if dim == 0:
1172 setattr(self.dataOut, self.listDataname[j], self.listData[j][self.blockIndex])
1172 setattr(self.dataOut, self.listDataname[j], self.listData[j][self.blockIndex])
1173 else:
1173 else:
1174 setattr(self.dataOut, self.listDataname[j], self.listData[j][:,self.blockIndex])
1174 setattr(self.dataOut, self.listDataname[j], self.listData[j][:,self.blockIndex])
1175
1175
1176 self.dataOut.paramInterval = self.interval
1176 self.dataOut.paramInterval = self.interval
1177 self.dataOut.flagNoData = False
1177 self.dataOut.flagNoData = False
1178 self.blockIndex += 1
1178 self.blockIndex += 1
1179
1179
1180 return
1180 return
1181
1181
1182 def run(self, **kwargs):
1182 def run(self, **kwargs):
1183
1183
1184 if not(self.isConfig):
1184 if not(self.isConfig):
1185 self.setup(**kwargs)
1185 self.setup(**kwargs)
1186 self.isConfig = True
1186 self.isConfig = True
1187
1187
1188 if self.blockIndex == self.blocksPerFile:
1188 if self.blockIndex == self.blocksPerFile:
1189 self.setNextFile()
1189 self.setNextFile()
1190
1190
1191 self.getData()
1191 self.getData()
1192
1192
1193 return
1193 return
1194
1194
1195 @MPDecorator
1195 @MPDecorator
1196 class ParameterWriter(Operation):
1196 class ParameterWriter(Operation):
1197 '''
1197 '''
1198 HDF5 Writer, stores parameters data in HDF5 format files
1198 HDF5 Writer, stores parameters data in HDF5 format files
1199
1199
1200 path: path where the files will be stored
1200 path: path where the files will be stored
1201 blocksPerFile: number of blocks that will be saved in per HDF5 format file
1201 blocksPerFile: number of blocks that will be saved in per HDF5 format file
1202 mode: selects the data stacking mode: '0' channels, '1' parameters, '3' table (for meteors)
1202 mode: selects the data stacking mode: '0' channels, '1' parameters, '3' table (for meteors)
1203 metadataList: list of attributes that will be stored as metadata
1203 metadataList: list of attributes that will be stored as metadata
1204 dataList: list of attributes that will be stores as data
1204 dataList: list of attributes that will be stores as data
1205 '''
1205 '''
1206
1206
1207
1207
1208 ext = ".hdf5"
1208 ext = ".hdf5"
1209 optchar = "D"
1209 optchar = "D"
1210 metaoptchar = "M"
1210 metaoptchar = "M"
1211 metaFile = None
1211 metaFile = None
1212 filename = None
1212 filename = None
1213 path = None
1213 path = None
1214 setFile = None
1214 setFile = None
1215 fp = None
1215 fp = None
1216 grp = None
1216 grp = None
1217 ds = None
1217 ds = None
1218 firsttime = True
1218 firsttime = True
1219 #Configurations
1219 #Configurations
1220 blocksPerFile = None
1220 blocksPerFile = None
1221 blockIndex = None
1221 blockIndex = None
1222 dataOut = None
1222 dataOut = None
1223 #Data Arrays
1223 #Data Arrays
1224 dataList = None
1224 dataList = None
1225 metadataList = None
1225 metadataList = None
1226 dsList = None #List of dictionaries with dataset properties
1226 dsList = None #List of dictionaries with dataset properties
1227 tableDim = None
1227 tableDim = None
1228 dtype = [('name', 'S20'),('nDim', 'i')]
1228 dtype = [('name', 'S20'),('nDim', 'i')]
1229 currentDay = None
1229 currentDay = None
1230 lastTime = None
1230 lastTime = None
1231
1231
1232 def __init__(self):
1232 def __init__(self):
1233
1233
1234 Operation.__init__(self)
1234 Operation.__init__(self)
1235 return
1235 return
1236
1236
1237 def setup(self, path=None, blocksPerFile=10, metadataList=None, dataList=None, setType=None):
1237 def setup(self, path=None, blocksPerFile=10, metadataList=None, dataList=None, setType=None):
1238 self.path = path
1238 self.path = path
1239 self.blocksPerFile = blocksPerFile
1239 self.blocksPerFile = blocksPerFile
1240 self.metadataList = metadataList
1240 self.metadataList = metadataList
1241 self.dataList = dataList
1241 self.dataList = dataList
1242 self.setType = setType
1242 self.setType = setType
1243
1243
1244 tableList = []
1244 tableList = []
1245 dsList = []
1245 dsList = []
1246
1246
1247 for i in range(len(self.dataList)):
1247 for i in range(len(self.dataList)):
1248 dsDict = {}
1248 dsDict = {}
1249 dataAux = getattr(self.dataOut, self.dataList[i])
1249 dataAux = getattr(self.dataOut, self.dataList[i])
1250 dsDict['variable'] = self.dataList[i]
1250 dsDict['variable'] = self.dataList[i]
1251
1251
1252 if dataAux is None:
1252 if dataAux is None:
1253 continue
1253 continue
1254 elif isinstance(dataAux, (int, float, numpy.integer, numpy.float)):
1254 elif isinstance(dataAux, (int, float, numpy.integer, numpy.float)):
1255 dsDict['nDim'] = 0
1255 dsDict['nDim'] = 0
1256 else:
1256 else:
1257 dsDict['nDim'] = len(dataAux.shape)
1257 dsDict['nDim'] = len(dataAux.shape)
1258 dsDict['shape'] = dataAux.shape
1258 dsDict['shape'] = dataAux.shape
1259 dsDict['dsNumber'] = dataAux.shape[0]
1259 dsDict['dsNumber'] = dataAux.shape[0]
1260
1260
1261 dsList.append(dsDict)
1261 dsList.append(dsDict)
1262 tableList.append((self.dataList[i], dsDict['nDim']))
1262 tableList.append((self.dataList[i], dsDict['nDim']))
1263
1263
1264 self.dsList = dsList
1264 self.dsList = dsList
1265 self.tableDim = numpy.array(tableList, dtype=self.dtype)
1265 self.tableDim = numpy.array(tableList, dtype=self.dtype)
1266 self.currentDay = self.dataOut.datatime.date()
1266 self.currentDay = self.dataOut.datatime.date()
1267
1267
1268 def timeFlag(self):
1268 def timeFlag(self):
1269 currentTime = self.dataOut.utctime
1269 currentTime = self.dataOut.utctime
1270 timeTuple = time.localtime(currentTime)
1270 timeTuple = time.localtime(currentTime)
1271 dataDay = timeTuple.tm_yday
1271 dataDay = timeTuple.tm_yday
1272
1272
1273 if self.lastTime is None:
1273 if self.lastTime is None:
1274 self.lastTime = currentTime
1274 self.lastTime = currentTime
1275 self.currentDay = dataDay
1275 self.currentDay = dataDay
1276 return False
1276 return False
1277
1277
1278 timeDiff = currentTime - self.lastTime
1278 timeDiff = currentTime - self.lastTime
1279
1279
1280 #Si el dia es diferente o si la diferencia entre un dato y otro supera la hora
1280 #Si el dia es diferente o si la diferencia entre un dato y otro supera la hora
1281 if dataDay != self.currentDay:
1281 if dataDay != self.currentDay:
1282 self.currentDay = dataDay
1282 self.currentDay = dataDay
1283 return True
1283 return True
1284 elif timeDiff > 3*60*60:
1284 elif timeDiff > 3*60*60:
1285 self.lastTime = currentTime
1285 self.lastTime = currentTime
1286 return True
1286 return True
1287 else:
1287 else:
1288 self.lastTime = currentTime
1288 self.lastTime = currentTime
1289 return False
1289 return False
1290
1290
1291 def run(self, dataOut, path, blocksPerFile=10, metadataList=None, dataList=None, setType=None):
1291 def run(self, dataOut, path, blocksPerFile=10, metadataList=None, dataList=None, setType=None):
1292
1292
1293 self.dataOut = dataOut
1293 self.dataOut = dataOut
1294 if not(self.isConfig):
1294 if not(self.isConfig):
1295 self.setup(path=path, blocksPerFile=blocksPerFile,
1295 self.setup(path=path, blocksPerFile=blocksPerFile,
1296 metadataList=metadataList, dataList=dataList,
1296 metadataList=metadataList, dataList=dataList,
1297 setType=setType)
1297 setType=setType)
1298
1298
1299 self.isConfig = True
1299 self.isConfig = True
1300 self.setNextFile()
1300 self.setNextFile()
1301
1301
1302 self.putData()
1302 self.putData()
1303 return
1303 return
1304
1304
1305 def setNextFile(self):
1305 def setNextFile(self):
1306
1306
1307 ext = self.ext
1307 ext = self.ext
1308 path = self.path
1308 path = self.path
1309 setFile = self.setFile
1309 setFile = self.setFile
1310
1310
1311 timeTuple = time.localtime(self.dataOut.utctime)
1311 timeTuple = time.localtime(self.dataOut.utctime)
1312 subfolder = 'd%4.4d%3.3d' % (timeTuple.tm_year,timeTuple.tm_yday)
1312 subfolder = 'd%4.4d%3.3d' % (timeTuple.tm_year,timeTuple.tm_yday)
1313 fullpath = os.path.join(path, subfolder)
1313 fullpath = os.path.join(path, subfolder)
1314
1314
1315 if os.path.exists(fullpath):
1315 if os.path.exists(fullpath):
1316 filesList = os.listdir(fullpath)
1316 filesList = os.listdir(fullpath)
1317 filesList = [k for k in filesList if k.startswith(self.optchar)]
1317 filesList = [k for k in filesList if k.startswith(self.optchar)]
1318 if len( filesList ) > 0:
1318 if len( filesList ) > 0:
1319 filesList = sorted(filesList, key=str.lower)
1319 filesList = sorted(filesList, key=str.lower)
1320 filen = filesList[-1]
1320 filen = filesList[-1]
1321 # el filename debera tener el siguiente formato
1321 # el filename debera tener el siguiente formato
1322 # 0 1234 567 89A BCDE (hex)
1322 # 0 1234 567 89A BCDE (hex)
1323 # x YYYY DDD SSS .ext
1323 # x YYYY DDD SSS .ext
1324 if isNumber(filen[8:11]):
1324 if isNumber(filen[8:11]):
1325 setFile = int(filen[8:11]) #inicializo mi contador de seteo al seteo del ultimo file
1325 setFile = int(filen[8:11]) #inicializo mi contador de seteo al seteo del ultimo file
1326 else:
1326 else:
1327 setFile = -1
1327 setFile = -1
1328 else:
1328 else:
1329 setFile = -1 #inicializo mi contador de seteo
1329 setFile = -1 #inicializo mi contador de seteo
1330 else:
1330 else:
1331 os.makedirs(fullpath)
1331 os.makedirs(fullpath)
1332 setFile = -1 #inicializo mi contador de seteo
1332 setFile = -1 #inicializo mi contador de seteo
1333
1333
1334 if self.setType is None:
1334 if self.setType is None:
1335 setFile += 1
1335 setFile += 1
1336 file = '%s%4.4d%3.3d%03d%s' % (self.optchar,
1336 file = '%s%4.4d%3.3d%03d%s' % (self.optchar,
1337 timeTuple.tm_year,
1337 timeTuple.tm_year,
1338 timeTuple.tm_yday,
1338 timeTuple.tm_yday,
1339 setFile,
1339 setFile,
1340 ext )
1340 ext )
1341 else:
1341 else:
1342 setFile = timeTuple.tm_hour*60+timeTuple.tm_min
1342 setFile = timeTuple.tm_hour*60+timeTuple.tm_min
1343 file = '%s%4.4d%3.3d%04d%s' % (self.optchar,
1343 file = '%s%4.4d%3.3d%04d%s' % (self.optchar,
1344 timeTuple.tm_year,
1344 timeTuple.tm_year,
1345 timeTuple.tm_yday,
1345 timeTuple.tm_yday,
1346 setFile,
1346 setFile,
1347 ext )
1347 ext )
1348
1348
1349 self.filename = os.path.join( path, subfolder, file )
1349 self.filename = os.path.join( path, subfolder, file )
1350
1350
1351 #Setting HDF5 File
1351 #Setting HDF5 File
1352 self.fp = h5py.File(self.filename, 'w')
1352 self.fp = h5py.File(self.filename, 'w')
1353 #write metadata
1353 #write metadata
1354 self.writeMetadata(self.fp)
1354 self.writeMetadata(self.fp)
1355 #Write data
1355 #Write data
1356 self.writeData(self.fp)
1356 self.writeData(self.fp)
1357
1357
1358 def writeMetadata(self, fp):
1358 def writeMetadata(self, fp):
1359
1359
1360 grp = fp.create_group("Metadata")
1360 grp = fp.create_group("Metadata")
1361 grp.create_dataset('variables', data=self.tableDim, dtype=self.dtype)
1361 grp.create_dataset('variables', data=self.tableDim, dtype=self.dtype)
1362
1362
1363 for i in range(len(self.metadataList)):
1363 for i in range(len(self.metadataList)):
1364 if not hasattr(self.dataOut, self.metadataList[i]):
1364 if not hasattr(self.dataOut, self.metadataList[i]):
1365 log.warning('Metadata: `{}` not found'.format(self.metadataList[i]), self.name)
1365 log.warning('Metadata: `{}` not found'.format(self.metadataList[i]), self.name)
1366 continue
1366 continue
1367 value = getattr(self.dataOut, self.metadataList[i])
1367 value = getattr(self.dataOut, self.metadataList[i])
1368 grp.create_dataset(self.metadataList[i], data=value)
1368 grp.create_dataset(self.metadataList[i], data=value)
1369 return
1369 return
1370
1370
1371 def writeData(self, fp):
1371 def writeData(self, fp):
1372
1372
1373 grp = fp.create_group("Data")
1373 grp = fp.create_group("Data")
1374 dtsets = []
1374 dtsets = []
1375 data = []
1375 data = []
1376
1376
1377 for dsInfo in self.dsList:
1377 for dsInfo in self.dsList:
1378 if dsInfo['nDim'] == 0:
1378 if dsInfo['nDim'] == 0:
1379 ds = grp.create_dataset(
1379 ds = grp.create_dataset(
1380 dsInfo['variable'],
1380 dsInfo['variable'],
1381 (self.blocksPerFile, ),
1381 (self.blocksPerFile, ),
1382 chunks=True,
1382 chunks=True,
1383 dtype=numpy.float64)
1383 dtype=numpy.float64)
1384 dtsets.append(ds)
1384 dtsets.append(ds)
1385 data.append((dsInfo['variable'], -1))
1385 data.append((dsInfo['variable'], -1))
1386 else:
1386 else:
1387 sgrp = grp.create_group(dsInfo['variable'])
1387 sgrp = grp.create_group(dsInfo['variable'])
1388 for i in range(dsInfo['dsNumber']):
1388 for i in range(dsInfo['dsNumber']):
1389 ds = sgrp.create_dataset(
1389 ds = sgrp.create_dataset(
1390 'table{:02d}'.format(i),
1390 'table{:02d}'.format(i),
1391 (self.blocksPerFile, ) + dsInfo['shape'][1:],
1391 (self.blocksPerFile, ) + dsInfo['shape'][1:],
1392 chunks=True)
1392 chunks=True)
1393 dtsets.append(ds)
1393 dtsets.append(ds)
1394 data.append((dsInfo['variable'], i))
1394 data.append((dsInfo['variable'], i))
1395 fp.flush()
1395 fp.flush()
1396
1396
1397 log.log('Creating file: {}'.format(fp.filename), self.name)
1397 log.log('Creating file: {}'.format(fp.filename), self.name)
1398
1398
1399 self.ds = dtsets
1399 self.ds = dtsets
1400 self.data = data
1400 self.data = data
1401 self.firsttime = True
1401 self.firsttime = True
1402 self.blockIndex = 0
1402 self.blockIndex = 0
1403 return
1403 return
1404
1404
1405 def putData(self):
1405 def putData(self):
1406
1406
1407 if (self.blockIndex == self.blocksPerFile) or self.timeFlag():
1407 if (self.blockIndex == self.blocksPerFile) or self.timeFlag():
1408 self.closeFile()
1408 self.closeFile()
1409 self.setNextFile()
1409 self.setNextFile()
1410
1410
1411 for i, ds in enumerate(self.ds):
1411 for i, ds in enumerate(self.ds):
1412 attr, ch = self.data[i]
1412 attr, ch = self.data[i]
1413 if ch == -1:
1413 if ch == -1:
1414 ds[self.blockIndex] = getattr(self.dataOut, attr)
1414 ds[self.blockIndex] = getattr(self.dataOut, attr)
1415 else:
1415 else:
1416 ds[self.blockIndex] = getattr(self.dataOut, attr)[ch]
1416 ds[self.blockIndex] = getattr(self.dataOut, attr)[ch]
1417
1417
1418 self.fp.flush()
1418 self.fp.flush()
1419 self.blockIndex += 1
1419 self.blockIndex += 1
1420 log.log('Block No. {}/{}'.format(self.blockIndex, self.blocksPerFile), self.name)
1420 log.log('Block No. {}/{}'.format(self.blockIndex, self.blocksPerFile), self.name)
1421
1421
1422 return
1422 return
1423
1423
1424 def closeFile(self):
1424 def closeFile(self):
1425
1425
1426 if self.blockIndex != self.blocksPerFile:
1426 if self.blockIndex != self.blocksPerFile:
1427 for ds in self.ds:
1427 for ds in self.ds:
1428 ds.resize(self.blockIndex, axis=0)
1428 ds.resize(self.blockIndex, axis=0)
1429
1429
1430 self.fp.flush()
1430 self.fp.flush()
1431 self.fp.close()
1431 self.fp.close()
1432
1432
1433 def close(self):
1433 def close(self):
1434
1434
1435 self.closeFile()
1435 self.closeFile()
@@ -1,429 +1,426
1 '''
1 '''
2 Updated for multiprocessing
2 Updated for multiprocessing
3 Author : Sergio Cortez
3 Author : Sergio Cortez
4 Jan 2018
4 Jan 2018
5 Abstract:
5 Abstract:
6 Base class for processing units and operations. A decorator provides multiprocessing features and interconnect the processes created.
6 Base class for processing units and operations. A decorator provides multiprocessing features and interconnect the processes created.
7 The argument (kwargs) sent from the controller is parsed and filtered via the decorator for each processing unit or operation instantiated.
7 The argument (kwargs) sent from the controller is parsed and filtered via the decorator for each processing unit or operation instantiated.
8 The decorator handle also the methods inside the processing unit to be called from the main script (not as operations) (OPERATION -> type ='self').
8 The decorator handle also the methods inside the processing unit to be called from the main script (not as operations) (OPERATION -> type ='self').
9
9
10 Based on:
10 Based on:
11 $Author: murco $
11 $Author: murco $
12 $Id: jroproc_base.py 1 2012-11-12 18:56:07Z murco $
12 $Id: jroproc_base.py 1 2012-11-12 18:56:07Z murco $
13 '''
13 '''
14
14
15 import os
15 import os
16 import sys
16 import sys
17 import inspect
17 import inspect
18 import zmq
18 import zmq
19 import time
19 import time
20 import pickle
20 import pickle
21 import traceback
21 import traceback
22 try:
22 try:
23 from queue import Queue
23 from queue import Queue
24 except:
24 except:
25 from Queue import Queue
25 from Queue import Queue
26 from threading import Thread
26 from threading import Thread
27 from multiprocessing import Process
27 from multiprocessing import Process
28
28
29 from schainpy.utils import log
29 from schainpy.utils import log
30
30
31
31
32 class ProcessingUnit(object):
32 class ProcessingUnit(object):
33
33
34 """
34 """
35 Update - Jan 2018 - MULTIPROCESSING
35 Update - Jan 2018 - MULTIPROCESSING
36 All the "call" methods present in the previous base were removed.
36 All the "call" methods present in the previous base were removed.
37 The majority of operations are independant processes, thus
37 The majority of operations are independant processes, thus
38 the decorator is in charge of communicate the operation processes
38 the decorator is in charge of communicate the operation processes
39 with the proccessing unit via IPC.
39 with the proccessing unit via IPC.
40
40
41 The constructor does not receive any argument. The remaining methods
41 The constructor does not receive any argument. The remaining methods
42 are related with the operations to execute.
42 are related with the operations to execute.
43
43
44
44
45 """
45 """
46 proc_type = 'processing'
46 proc_type = 'processing'
47 __attrs__ = []
47 __attrs__ = []
48
48
49 def __init__(self):
49 def __init__(self):
50
50
51 self.dataIn = None
51 self.dataIn = None
52 self.dataOut = None
52 self.dataOut = None
53 self.isConfig = False
53 self.isConfig = False
54 self.operations = []
54 self.operations = []
55 self.plots = []
55 self.plots = []
56
56
57 def getAllowedArgs(self):
57 def getAllowedArgs(self):
58 if hasattr(self, '__attrs__'):
58 if hasattr(self, '__attrs__'):
59 return self.__attrs__
59 return self.__attrs__
60 else:
60 else:
61 return inspect.getargspec(self.run).args
61 return inspect.getargspec(self.run).args
62
62
63 def addOperation(self, conf, operation):
63 def addOperation(self, conf, operation):
64 """
64 """
65 This method is used in the controller, and update the dictionary containing the operations to execute. The dict
65 This method is used in the controller, and update the dictionary containing the operations to execute. The dict
66 posses the id of the operation process (IPC purposes)
66 posses the id of the operation process (IPC purposes)
67
67
68 Agrega un objeto del tipo "Operation" (opObj) a la lista de objetos "self.objectList" y retorna el
68 Agrega un objeto del tipo "Operation" (opObj) a la lista de objetos "self.objectList" y retorna el
69 identificador asociado a este objeto.
69 identificador asociado a este objeto.
70
70
71 Input:
71 Input:
72
72
73 object : objeto de la clase "Operation"
73 object : objeto de la clase "Operation"
74
74
75 Return:
75 Return:
76
76
77 objId : identificador del objeto, necesario para comunicar con master(procUnit)
77 objId : identificador del objeto, necesario para comunicar con master(procUnit)
78 """
78 """
79
79
80 self.operations.append(
80 self.operations.append(
81 (operation, conf.type, conf.id, conf.getKwargs()))
81 (operation, conf.type, conf.id, conf.getKwargs()))
82
82
83 if 'plot' in self.name.lower():
83 if 'plot' in self.name.lower():
84 self.plots.append(operation.CODE)
84 self.plots.append(operation.CODE)
85
85
86 def getOperationObj(self, objId):
86 def getOperationObj(self, objId):
87
87
88 if objId not in list(self.operations.keys()):
88 if objId not in list(self.operations.keys()):
89 return None
89 return None
90
90
91 return self.operations[objId]
91 return self.operations[objId]
92
92
93 def operation(self, **kwargs):
93 def operation(self, **kwargs):
94 """
94 """
95 Operacion directa sobre la data (dataOut.data). Es necesario actualizar los valores de los
95 Operacion directa sobre la data (dataOut.data). Es necesario actualizar los valores de los
96 atributos del objeto dataOut
96 atributos del objeto dataOut
97
97
98 Input:
98 Input:
99
99
100 **kwargs : Diccionario de argumentos de la funcion a ejecutar
100 **kwargs : Diccionario de argumentos de la funcion a ejecutar
101 """
101 """
102
102
103 raise NotImplementedError
103 raise NotImplementedError
104
104
105 def setup(self):
105 def setup(self):
106
106
107 raise NotImplementedError
107 raise NotImplementedError
108
108
109 def run(self):
109 def run(self):
110
110
111 raise NotImplementedError
111 raise NotImplementedError
112
112
113 def close(self):
113 def close(self):
114
114
115 return
115 return
116
116
117
117
118 class Operation(object):
118 class Operation(object):
119
119
120 """
120 """
121 Update - Jan 2018 - MULTIPROCESSING
121 Update - Jan 2018 - MULTIPROCESSING
122
122
123 Most of the methods remained the same. The decorator parse the arguments and executed the run() method for each process.
123 Most of the methods remained the same. The decorator parse the arguments and executed the run() method for each process.
124 The constructor doe snot receive any argument, neither the baseclass.
124 The constructor doe snot receive any argument, neither the baseclass.
125
125
126
126
127 Clase base para definir las operaciones adicionales que se pueden agregar a la clase ProcessingUnit
127 Clase base para definir las operaciones adicionales que se pueden agregar a la clase ProcessingUnit
128 y necesiten acumular informacion previa de los datos a procesar. De preferencia usar un buffer de
128 y necesiten acumular informacion previa de los datos a procesar. De preferencia usar un buffer de
129 acumulacion dentro de esta clase
129 acumulacion dentro de esta clase
130
130
131 Ejemplo: Integraciones coherentes, necesita la informacion previa de los n perfiles anteriores (bufffer)
131 Ejemplo: Integraciones coherentes, necesita la informacion previa de los n perfiles anteriores (bufffer)
132
132
133 """
133 """
134 proc_type = 'operation'
134 proc_type = 'operation'
135 __attrs__ = []
135 __attrs__ = []
136
136
137 def __init__(self):
137 def __init__(self):
138
138
139 self.id = None
139 self.id = None
140 self.isConfig = False
140 self.isConfig = False
141
141
142 if not hasattr(self, 'name'):
142 if not hasattr(self, 'name'):
143 self.name = self.__class__.__name__
143 self.name = self.__class__.__name__
144
144
145 def getAllowedArgs(self):
145 def getAllowedArgs(self):
146 if hasattr(self, '__attrs__'):
146 if hasattr(self, '__attrs__'):
147 return self.__attrs__
147 return self.__attrs__
148 else:
148 else:
149 return inspect.getargspec(self.run).args
149 return inspect.getargspec(self.run).args
150
150
151 def setup(self):
151 def setup(self):
152
152
153 self.isConfig = True
153 self.isConfig = True
154
154
155 raise NotImplementedError
155 raise NotImplementedError
156
156
157 def run(self, dataIn, **kwargs):
157 def run(self, dataIn, **kwargs):
158 """
158 """
159 Realiza las operaciones necesarias sobre la dataIn.data y actualiza los
159 Realiza las operaciones necesarias sobre la dataIn.data y actualiza los
160 atributos del objeto dataIn.
160 atributos del objeto dataIn.
161
161
162 Input:
162 Input:
163
163
164 dataIn : objeto del tipo JROData
164 dataIn : objeto del tipo JROData
165
165
166 Return:
166 Return:
167
167
168 None
168 None
169
169
170 Affected:
170 Affected:
171 __buffer : buffer de recepcion de datos.
171 __buffer : buffer de recepcion de datos.
172
172
173 """
173 """
174 if not self.isConfig:
174 if not self.isConfig:
175 self.setup(**kwargs)
175 self.setup(**kwargs)
176
176
177 raise NotImplementedError
177 raise NotImplementedError
178
178
179 def close(self):
179 def close(self):
180
180
181 return
181 return
182
182
183 class InputQueue(Thread):
183 class InputQueue(Thread):
184
184
185 '''
185 '''
186 Class to hold input data for Proccessing Units and external Operations,
186 Class to hold input data for Proccessing Units and external Operations,
187 '''
187 '''
188
188
189 def __init__(self, project_id, inputId, lock=None):
189 def __init__(self, project_id, inputId, lock=None):
190
190
191 Thread.__init__(self)
191 Thread.__init__(self)
192 self.queue = Queue()
192 self.queue = Queue()
193 self.project_id = project_id
193 self.project_id = project_id
194 self.inputId = inputId
194 self.inputId = inputId
195 self.lock = lock
195 self.lock = lock
196 self.islocked = False
196 self.islocked = False
197 self.size = 0
197 self.size = 0
198
198
199 def run(self):
199 def run(self):
200
200
201 c = zmq.Context()
201 c = zmq.Context()
202 self.receiver = c.socket(zmq.SUB)
202 self.receiver = c.socket(zmq.SUB)
203 self.receiver.connect(
203 self.receiver.connect(
204 'ipc:///tmp/schain/{}_pub'.format(self.project_id))
204 'ipc:///tmp/schain/{}_pub'.format(self.project_id))
205 self.receiver.setsockopt(zmq.SUBSCRIBE, self.inputId.encode())
205 self.receiver.setsockopt(zmq.SUBSCRIBE, self.inputId.encode())
206
206
207 while True:
207 while True:
208 obj = self.receiver.recv_multipart()[1]
208 obj = self.receiver.recv_multipart()[1]
209 self.size += sys.getsizeof(obj)
209 self.size += sys.getsizeof(obj)
210 self.queue.put(obj)
210 self.queue.put(obj)
211
211
212 def get(self):
212 def get(self):
213
213
214 if not self.islocked and self.size/1000000 > 512:
214 if not self.islocked and self.size/1000000 > 512:
215 self.lock.n.value += 1
215 self.lock.n.value += 1
216 self.islocked = True
216 self.islocked = True
217 self.lock.clear()
217 self.lock.clear()
218 elif self.islocked and self.size/1000000 <= 512:
218 elif self.islocked and self.size/1000000 <= 512:
219 self.islocked = False
219 self.islocked = False
220 self.lock.n.value -= 1
220 self.lock.n.value -= 1
221 if self.lock.n.value == 0:
221 if self.lock.n.value == 0:
222 self.lock.set()
222 self.lock.set()
223
223
224 obj = self.queue.get()
224 obj = self.queue.get()
225 self.size -= sys.getsizeof(obj)
225 self.size -= sys.getsizeof(obj)
226 return pickle.loads(obj)
226 return pickle.loads(obj)
227
227
228
228
229 def MPDecorator(BaseClass):
229 def MPDecorator(BaseClass):
230 """
230 """
231 Multiprocessing class decorator
231 Multiprocessing class decorator
232
232
233 This function add multiprocessing features to a BaseClass. Also, it handle
233 This function add multiprocessing features to a BaseClass. Also, it handle
234 the communication beetween processes (readers, procUnits and operations).
234 the communication beetween processes (readers, procUnits and operations).
235 """
235 """
236
236
237 class MPClass(BaseClass, Process):
237 class MPClass(BaseClass, Process):
238
238
239 def __init__(self, *args, **kwargs):
239 def __init__(self, *args, **kwargs):
240 super(MPClass, self).__init__()
240 super(MPClass, self).__init__()
241 Process.__init__(self)
241 Process.__init__(self)
242 self.operationKwargs = {}
242 self.operationKwargs = {}
243 self.args = args
243 self.args = args
244 self.kwargs = kwargs
244 self.kwargs = kwargs
245 self.sender = None
245 self.sender = None
246 self.receiver = None
246 self.receiver = None
247 self.i = 0
247 self.i = 0
248 self.t = time.time()
248 self.t = time.time()
249 self.name = BaseClass.__name__
249 self.name = BaseClass.__name__
250 self.__doc__ = BaseClass.__doc__
250 self.__doc__ = BaseClass.__doc__
251
251
252 if 'plot' in self.name.lower() and not self.name.endswith('_'):
252 if 'plot' in self.name.lower() and not self.name.endswith('_'):
253 self.name = '{}{}'.format(self.CODE.upper(), 'Plot')
253 self.name = '{}{}'.format(self.CODE.upper(), 'Plot')
254
254
255 self.start_time = time.time()
255 self.start_time = time.time()
256 self.id = args[0]
256 self.id = args[0]
257 self.inputId = args[1]
257 self.inputId = args[1]
258 self.project_id = args[2]
258 self.project_id = args[2]
259 self.err_queue = args[3]
259 self.err_queue = args[3]
260 self.lock = args[4]
260 self.lock = args[4]
261 self.typeProc = args[5]
261 self.typeProc = args[5]
262 self.err_queue.put('#_start_#')
262 self.err_queue.put('#_start_#')
263 if self.inputId is not None:
263 if self.inputId is not None:
264 self.queue = InputQueue(self.project_id, self.inputId, self.lock)
264 self.queue = InputQueue(self.project_id, self.inputId, self.lock)
265
265
266 def subscribe(self):
266 def subscribe(self):
267 '''
267 '''
268 Start the zmq socket receiver and subcribe to input ID.
268 Start the zmq socket receiver and subcribe to input ID.
269 '''
269 '''
270
270
271 self.queue.start()
271 self.queue.start()
272
272
273 def listen(self):
273 def listen(self):
274 '''
274 '''
275 This function waits for objects
275 This function waits for objects
276 '''
276 '''
277
277
278 return self.queue.get()
278 return self.queue.get()
279
279
280 def set_publisher(self):
280 def set_publisher(self):
281 '''
281 '''
282 This function create a zmq socket for publishing objects.
282 This function create a zmq socket for publishing objects.
283 '''
283 '''
284
284
285 time.sleep(0.5)
285 time.sleep(0.5)
286
286
287 c = zmq.Context()
287 c = zmq.Context()
288 self.sender = c.socket(zmq.PUB)
288 self.sender = c.socket(zmq.PUB)
289 self.sender.connect(
289 self.sender.connect(
290 'ipc:///tmp/schain/{}_sub'.format(self.project_id))
290 'ipc:///tmp/schain/{}_sub'.format(self.project_id))
291
291
292 def publish(self, data, id):
292 def publish(self, data, id):
293 '''
293 '''
294 This function publish an object, to an specific topic.
294 This function publish an object, to an specific topic.
295 It blocks publishing when receiver queue is full to avoid data loss
295 It blocks publishing when receiver queue is full to avoid data loss
296 '''
296 '''
297
297
298 if self.inputId is None:
298 if self.inputId is None:
299 self.lock.wait()
299 self.lock.wait()
300 self.sender.send_multipart([str(id).encode(), pickle.dumps(data)])
300 self.sender.send_multipart([str(id).encode(), pickle.dumps(data)])
301
302 def runReader(self):
301 def runReader(self):
303 '''
302 '''
304 Run fuction for read units
303 Run fuction for read units
305 '''
304 '''
306 while True:
305 while True:
307
306
308 try:
307 try:
309 BaseClass.run(self, **self.kwargs)
308 BaseClass.run(self, **self.kwargs)
310 except:
309 except:
311 err = traceback.format_exc()
310 err = traceback.format_exc()
312 if 'No more files' in err:
311 if 'No more files' in err:
313 log.warning('No more files to read', self.name)
312 log.warning('No more files to read', self.name)
314 else:
313 else:
315 self.err_queue.put('{}|{}'.format(self.name, err))
314 self.err_queue.put('{}|{}'.format(self.name, err))
316 self.dataOut.error = True
315 self.dataOut.error = True
317
316
318 for op, optype, opId, kwargs in self.operations:
317 for op, optype, opId, kwargs in self.operations:
319 if optype == 'self' and not self.dataOut.flagNoData:
318 if optype == 'self' and not self.dataOut.flagNoData:
320 op(**kwargs)
319 op(**kwargs)
321 elif optype == 'other' and not self.dataOut.flagNoData:
320 elif optype == 'other' and not self.dataOut.flagNoData:
322 self.dataOut = op.run(self.dataOut, **self.kwargs)
321 self.dataOut = op.run(self.dataOut, **self.kwargs)
323 elif optype == 'external':
322 elif optype == 'external':
324 self.publish(self.dataOut, opId)
323 self.publish(self.dataOut, opId)
325
324
326 if self.dataOut.flagNoData and not self.dataOut.error:
325 if self.dataOut.flagNoData and not self.dataOut.error:
327 continue
326 continue
328
327
329 self.publish(self.dataOut, self.id)
328 self.publish(self.dataOut, self.id)
330
331 if self.dataOut.error:
329 if self.dataOut.error:
332 break
330 break
333
331
334 time.sleep(0.5)
332 time.sleep(0.5)
335
333
336 def runProc(self):
334 def runProc(self):
337 '''
335 '''
338 Run function for proccessing units
336 Run function for proccessing units
339 '''
337 '''
340
338
341 while True:
339 while True:
342 self.dataIn = self.listen()
340 self.dataIn = self.listen()
343
341
344 if self.dataIn.flagNoData and self.dataIn.error is None:
342 if self.dataIn.flagNoData and self.dataIn.error is None:
345 continue
343 continue
346 elif not self.dataIn.error:
344 elif not self.dataIn.error:
347 try:
345 try:
348 BaseClass.run(self, **self.kwargs)
346 BaseClass.run(self, **self.kwargs)
349 except:
347 except:
350 self.err_queue.put('{}|{}'.format(self.name, traceback.format_exc()))
348 self.err_queue.put('{}|{}'.format(self.name, traceback.format_exc()))
351 self.dataOut.error = True
349 self.dataOut.error = True
352 elif self.dataIn.error:
350 elif self.dataIn.error:
353 self.dataOut.error = self.dataIn.error
351 self.dataOut.error = self.dataIn.error
354 self.dataOut.flagNoData = True
352 self.dataOut.flagNoData = True
355
353
356 for op, optype, opId, kwargs in self.operations:
354 for op, optype, opId, kwargs in self.operations:
357 if optype == 'self' and not self.dataOut.flagNoData:
355 if optype == 'self' and not self.dataOut.flagNoData:
358 op(**kwargs)
356 op(**kwargs)
359 elif optype == 'other' and not self.dataOut.flagNoData:
357 elif optype == 'other' and not self.dataOut.flagNoData:
360 self.dataOut = op.run(self.dataOut, **kwargs)
358 self.dataOut = op.run(self.dataOut, **kwargs)
361 elif optype == 'external' and not self.dataOut.flagNoData:
359 elif optype == 'external' and not self.dataOut.flagNoData:
362 self.publish(self.dataOut, opId)
360 self.publish(self.dataOut, opId)
363
361
364 self.publish(self.dataOut, self.id)
362 self.publish(self.dataOut, self.id)
365 for op, optype, opId, kwargs in self.operations:
363 for op, optype, opId, kwargs in self.operations:
366 if optype == 'external' and self.dataOut.error:
364 if optype == 'external' and self.dataOut.error:
367 self.publish(self.dataOut, opId)
365 self.publish(self.dataOut, opId)
368
366
369 if self.dataOut.error:
367 if self.dataOut.error:
370 break
368 break
371
369
372 time.sleep(0.5)
370 time.sleep(0.5)
373
371
374 def runOp(self):
372 def runOp(self):
375 '''
373 '''
376 Run function for external operations (this operations just receive data
374 Run function for external operations (this operations just receive data
377 ex: plots, writers, publishers)
375 ex: plots, writers, publishers)
378 '''
376 '''
379
377
380 while True:
378 while True:
381
379
382 dataOut = self.listen()
380 dataOut = self.listen()
383
381
384 if not dataOut.error:
382 if not dataOut.error:
385 try:
383 try:
386 BaseClass.run(self, dataOut, **self.kwargs)
384 BaseClass.run(self, dataOut, **self.kwargs)
387 except:
385 except:
388 self.err_queue.put('{}|{}'.format(self.name, traceback.format_exc()))
386 self.err_queue.put('{}|{}'.format(self.name, traceback.format_exc()))
389 dataOut.error = True
387 dataOut.error = True
390 else:
388 else:
391 break
389 break
392
390
393 def run(self):
391 def run(self):
394 if self.typeProc is "ProcUnit":
392 if self.typeProc is "ProcUnit":
395
393
396 if self.inputId is not None:
394 if self.inputId is not None:
397 self.subscribe()
395 self.subscribe()
398
396
399 self.set_publisher()
397 self.set_publisher()
400
398
401 if 'Reader' not in BaseClass.__name__:
399 if 'Reader' not in BaseClass.__name__:
402 self.runProc()
400 self.runProc()
403 else:
401 else:
404 self.runReader()
402 self.runReader()
405
406 elif self.typeProc is "Operation":
403 elif self.typeProc is "Operation":
407
404
408 self.subscribe()
405 self.subscribe()
409 self.runOp()
406 self.runOp()
410
407
411 else:
408 else:
412 raise ValueError("Unknown type")
409 raise ValueError("Unknown type")
413
410
414 self.close()
411 self.close()
415
412
416 def close(self):
413 def close(self):
417
414
418 BaseClass.close(self)
415 BaseClass.close(self)
419 self.err_queue.put('#_end_#')
416 self.err_queue.put('#_end_#')
420
417
421 if self.sender:
418 if self.sender:
422 self.sender.close()
419 self.sender.close()
423
420
424 if self.receiver:
421 if self.receiver:
425 self.receiver.close()
422 self.receiver.close()
426
423
427 log.success('Done...(Time:{:4.2f} secs)'.format(time.time()-self.start_time), self.name)
424 log.success('Done...(Time:{:4.2f} secs)'.format(time.time()-self.start_time), self.name)
428
425
429 return MPClass
426 return MPClass
1 NO CONTENT: modified file
NO CONTENT: modified file
The requested commit or file is too big and content was truncated. Show full diff
1 NO CONTENT: modified file
NO CONTENT: modified file
@@ -1,1328 +1,1327
1 import sys
1 import sys
2 import numpy
2 import numpy
3 from scipy import interpolate
3 from scipy import interpolate
4 from schainpy.model.proc.jroproc_base import ProcessingUnit, Operation, MPDecorator
4 from schainpy.model.proc.jroproc_base import ProcessingUnit, Operation, MPDecorator
5 from schainpy.model.data.jrodata import Voltage
5 from schainpy.model.data.jrodata import Voltage
6 from schainpy.utils import log
6 from schainpy.utils import log
7 from time import time
7 from time import time
8
8
9
9
10 @MPDecorator
10 @MPDecorator
11 class VoltageProc(ProcessingUnit):
11 class VoltageProc(ProcessingUnit):
12
12
13 def __init__(self):
13 def __init__(self):
14
14
15 ProcessingUnit.__init__(self)
15 ProcessingUnit.__init__(self)
16
16
17 self.dataOut = Voltage()
17 self.dataOut = Voltage()
18 self.flip = 1
18 self.flip = 1
19 self.setupReq = False
19 self.setupReq = False
20
20
21 def run(self):
21 def run(self):
22
22
23 if self.dataIn.type == 'AMISR':
23 if self.dataIn.type == 'AMISR':
24 self.__updateObjFromAmisrInput()
24 self.__updateObjFromAmisrInput()
25
25
26 if self.dataIn.type == 'Voltage':
26 if self.dataIn.type == 'Voltage':
27 self.dataOut.copy(self.dataIn)
27 self.dataOut.copy(self.dataIn)
28
28
29 # self.dataOut.copy(self.dataIn)
29 # self.dataOut.copy(self.dataIn)
30
30
31 def __updateObjFromAmisrInput(self):
31 def __updateObjFromAmisrInput(self):
32
32
33 self.dataOut.timeZone = self.dataIn.timeZone
33 self.dataOut.timeZone = self.dataIn.timeZone
34 self.dataOut.dstFlag = self.dataIn.dstFlag
34 self.dataOut.dstFlag = self.dataIn.dstFlag
35 self.dataOut.errorCount = self.dataIn.errorCount
35 self.dataOut.errorCount = self.dataIn.errorCount
36 self.dataOut.useLocalTime = self.dataIn.useLocalTime
36 self.dataOut.useLocalTime = self.dataIn.useLocalTime
37
37
38 self.dataOut.flagNoData = self.dataIn.flagNoData
38 self.dataOut.flagNoData = self.dataIn.flagNoData
39 self.dataOut.data = self.dataIn.data
39 self.dataOut.data = self.dataIn.data
40 self.dataOut.utctime = self.dataIn.utctime
40 self.dataOut.utctime = self.dataIn.utctime
41 self.dataOut.channelList = self.dataIn.channelList
41 self.dataOut.channelList = self.dataIn.channelList
42 #self.dataOut.timeInterval = self.dataIn.timeInterval
42 #self.dataOut.timeInterval = self.dataIn.timeInterval
43 self.dataOut.heightList = self.dataIn.heightList
43 self.dataOut.heightList = self.dataIn.heightList
44 self.dataOut.nProfiles = self.dataIn.nProfiles
44 self.dataOut.nProfiles = self.dataIn.nProfiles
45
45
46 self.dataOut.nCohInt = self.dataIn.nCohInt
46 self.dataOut.nCohInt = self.dataIn.nCohInt
47 self.dataOut.ippSeconds = self.dataIn.ippSeconds
47 self.dataOut.ippSeconds = self.dataIn.ippSeconds
48 self.dataOut.frequency = self.dataIn.frequency
48 self.dataOut.frequency = self.dataIn.frequency
49
49
50 self.dataOut.azimuth = self.dataIn.azimuth
50 self.dataOut.azimuth = self.dataIn.azimuth
51 self.dataOut.zenith = self.dataIn.zenith
51 self.dataOut.zenith = self.dataIn.zenith
52
52
53 self.dataOut.beam.codeList = self.dataIn.beam.codeList
53 self.dataOut.beam.codeList = self.dataIn.beam.codeList
54 self.dataOut.beam.azimuthList = self.dataIn.beam.azimuthList
54 self.dataOut.beam.azimuthList = self.dataIn.beam.azimuthList
55 self.dataOut.beam.zenithList = self.dataIn.beam.zenithList
55 self.dataOut.beam.zenithList = self.dataIn.beam.zenithList
56 #
56 #
57 # pass#
57 # pass#
58 #
58 #
59 # def init(self):
59 # def init(self):
60 #
60 #
61 #
61 #
62 # if self.dataIn.type == 'AMISR':
62 # if self.dataIn.type == 'AMISR':
63 # self.__updateObjFromAmisrInput()
63 # self.__updateObjFromAmisrInput()
64 #
64 #
65 # if self.dataIn.type == 'Voltage':
65 # if self.dataIn.type == 'Voltage':
66 # self.dataOut.copy(self.dataIn)
66 # self.dataOut.copy(self.dataIn)
67 # # No necesita copiar en cada init() los atributos de dataIn
67 # # No necesita copiar en cada init() los atributos de dataIn
68 # # la copia deberia hacerse por cada nuevo bloque de datos
68 # # la copia deberia hacerse por cada nuevo bloque de datos
69
69
70 def selectChannels(self, channelList):
70 def selectChannels(self, channelList):
71
71
72 channelIndexList = []
72 channelIndexList = []
73
73
74 for channel in channelList:
74 for channel in channelList:
75 if channel not in self.dataOut.channelList:
75 if channel not in self.dataOut.channelList:
76 raise ValueError("Channel %d is not in %s" %(channel, str(self.dataOut.channelList)))
76 raise ValueError("Channel %d is not in %s" %(channel, str(self.dataOut.channelList)))
77
77
78 index = self.dataOut.channelList.index(channel)
78 index = self.dataOut.channelList.index(channel)
79 channelIndexList.append(index)
79 channelIndexList.append(index)
80
80
81 self.selectChannelsByIndex(channelIndexList)
81 self.selectChannelsByIndex(channelIndexList)
82
82
83 def selectChannelsByIndex(self, channelIndexList):
83 def selectChannelsByIndex(self, channelIndexList):
84 """
84 """
85 Selecciona un bloque de datos en base a canales segun el channelIndexList
85 Selecciona un bloque de datos en base a canales segun el channelIndexList
86
86
87 Input:
87 Input:
88 channelIndexList : lista sencilla de canales a seleccionar por ej. [2,3,7]
88 channelIndexList : lista sencilla de canales a seleccionar por ej. [2,3,7]
89
89
90 Affected:
90 Affected:
91 self.dataOut.data
91 self.dataOut.data
92 self.dataOut.channelIndexList
92 self.dataOut.channelIndexList
93 self.dataOut.nChannels
93 self.dataOut.nChannels
94 self.dataOut.m_ProcessingHeader.totalSpectra
94 self.dataOut.m_ProcessingHeader.totalSpectra
95 self.dataOut.systemHeaderObj.numChannels
95 self.dataOut.systemHeaderObj.numChannels
96 self.dataOut.m_ProcessingHeader.blockSize
96 self.dataOut.m_ProcessingHeader.blockSize
97
97
98 Return:
98 Return:
99 None
99 None
100 """
100 """
101
101
102 for channelIndex in channelIndexList:
102 for channelIndex in channelIndexList:
103 if channelIndex not in self.dataOut.channelIndexList:
103 if channelIndex not in self.dataOut.channelIndexList:
104 print(channelIndexList)
104 print(channelIndexList)
105 raise ValueError("The value %d in channelIndexList is not valid" %channelIndex)
105 raise ValueError("The value %d in channelIndexList is not valid" %channelIndex)
106
106
107 if self.dataOut.flagDataAsBlock:
107 if self.dataOut.flagDataAsBlock:
108 """
108 """
109 Si la data es obtenida por bloques, dimension = [nChannels, nProfiles, nHeis]
109 Si la data es obtenida por bloques, dimension = [nChannels, nProfiles, nHeis]
110 """
110 """
111 data = self.dataOut.data[channelIndexList,:,:]
111 data = self.dataOut.data[channelIndexList,:,:]
112 else:
112 else:
113 data = self.dataOut.data[channelIndexList,:]
113 data = self.dataOut.data[channelIndexList,:]
114
114
115 self.dataOut.data = data
115 self.dataOut.data = data
116 # self.dataOut.channelList = [self.dataOut.channelList[i] for i in channelIndexList]
116 # self.dataOut.channelList = [self.dataOut.channelList[i] for i in channelIndexList]
117 self.dataOut.channelList = range(len(channelIndexList))
117 self.dataOut.channelList = range(len(channelIndexList))
118
118
119 return 1
119 return 1
120
120
121 def selectHeights(self, minHei=None, maxHei=None):
121 def selectHeights(self, minHei=None, maxHei=None):
122 """
122 """
123 Selecciona un bloque de datos en base a un grupo de valores de alturas segun el rango
123 Selecciona un bloque de datos en base a un grupo de valores de alturas segun el rango
124 minHei <= height <= maxHei
124 minHei <= height <= maxHei
125
125
126 Input:
126 Input:
127 minHei : valor minimo de altura a considerar
127 minHei : valor minimo de altura a considerar
128 maxHei : valor maximo de altura a considerar
128 maxHei : valor maximo de altura a considerar
129
129
130 Affected:
130 Affected:
131 Indirectamente son cambiados varios valores a travez del metodo selectHeightsByIndex
131 Indirectamente son cambiados varios valores a travez del metodo selectHeightsByIndex
132
132
133 Return:
133 Return:
134 1 si el metodo se ejecuto con exito caso contrario devuelve 0
134 1 si el metodo se ejecuto con exito caso contrario devuelve 0
135 """
135 """
136
136
137 if minHei == None:
137 if minHei == None:
138 minHei = self.dataOut.heightList[0]
138 minHei = self.dataOut.heightList[0]
139
139
140 if maxHei == None:
140 if maxHei == None:
141 maxHei = self.dataOut.heightList[-1]
141 maxHei = self.dataOut.heightList[-1]
142
142
143 if (minHei < self.dataOut.heightList[0]):
143 if (minHei < self.dataOut.heightList[0]):
144 minHei = self.dataOut.heightList[0]
144 minHei = self.dataOut.heightList[0]
145
145
146 if (maxHei > self.dataOut.heightList[-1]):
146 if (maxHei > self.dataOut.heightList[-1]):
147 maxHei = self.dataOut.heightList[-1]
147 maxHei = self.dataOut.heightList[-1]
148
148
149 minIndex = 0
149 minIndex = 0
150 maxIndex = 0
150 maxIndex = 0
151 heights = self.dataOut.heightList
151 heights = self.dataOut.heightList
152
152
153 inda = numpy.where(heights >= minHei)
153 inda = numpy.where(heights >= minHei)
154 indb = numpy.where(heights <= maxHei)
154 indb = numpy.where(heights <= maxHei)
155
155
156 try:
156 try:
157 minIndex = inda[0][0]
157 minIndex = inda[0][0]
158 except:
158 except:
159 minIndex = 0
159 minIndex = 0
160
160
161 try:
161 try:
162 maxIndex = indb[0][-1]
162 maxIndex = indb[0][-1]
163 except:
163 except:
164 maxIndex = len(heights)
164 maxIndex = len(heights)
165
165
166 self.selectHeightsByIndex(minIndex, maxIndex)
166 self.selectHeightsByIndex(minIndex, maxIndex)
167
167
168 return 1
168 return 1
169
169
170
170
171 def selectHeightsByIndex(self, minIndex, maxIndex):
171 def selectHeightsByIndex(self, minIndex, maxIndex):
172 """
172 """
173 Selecciona un bloque de datos en base a un grupo indices de alturas segun el rango
173 Selecciona un bloque de datos en base a un grupo indices de alturas segun el rango
174 minIndex <= index <= maxIndex
174 minIndex <= index <= maxIndex
175
175
176 Input:
176 Input:
177 minIndex : valor de indice minimo de altura a considerar
177 minIndex : valor de indice minimo de altura a considerar
178 maxIndex : valor de indice maximo de altura a considerar
178 maxIndex : valor de indice maximo de altura a considerar
179
179
180 Affected:
180 Affected:
181 self.dataOut.data
181 self.dataOut.data
182 self.dataOut.heightList
182 self.dataOut.heightList
183
183
184 Return:
184 Return:
185 1 si el metodo se ejecuto con exito caso contrario devuelve 0
185 1 si el metodo se ejecuto con exito caso contrario devuelve 0
186 """
186 """
187
187
188 if (minIndex < 0) or (minIndex > maxIndex):
188 if (minIndex < 0) or (minIndex > maxIndex):
189 raise ValueError("Height index range (%d,%d) is not valid" % (minIndex, maxIndex))
189 raise ValueError("Height index range (%d,%d) is not valid" % (minIndex, maxIndex))
190
190
191 if (maxIndex >= self.dataOut.nHeights):
191 if (maxIndex >= self.dataOut.nHeights):
192 maxIndex = self.dataOut.nHeights
192 maxIndex = self.dataOut.nHeights
193
193
194 #voltage
194 #voltage
195 if self.dataOut.flagDataAsBlock:
195 if self.dataOut.flagDataAsBlock:
196 """
196 """
197 Si la data es obtenida por bloques, dimension = [nChannels, nProfiles, nHeis]
197 Si la data es obtenida por bloques, dimension = [nChannels, nProfiles, nHeis]
198 """
198 """
199 data = self.dataOut.data[:,:, minIndex:maxIndex]
199 data = self.dataOut.data[:,:, minIndex:maxIndex]
200 else:
200 else:
201 data = self.dataOut.data[:, minIndex:maxIndex]
201 data = self.dataOut.data[:, minIndex:maxIndex]
202
202
203 # firstHeight = self.dataOut.heightList[minIndex]
203 # firstHeight = self.dataOut.heightList[minIndex]
204
204
205 self.dataOut.data = data
205 self.dataOut.data = data
206 self.dataOut.heightList = self.dataOut.heightList[minIndex:maxIndex]
206 self.dataOut.heightList = self.dataOut.heightList[minIndex:maxIndex]
207
207
208 if self.dataOut.nHeights <= 1:
208 if self.dataOut.nHeights <= 1:
209 raise ValueError("selectHeights: Too few heights. Current number of heights is %d" %(self.dataOut.nHeights))
209 raise ValueError("selectHeights: Too few heights. Current number of heights is %d" %(self.dataOut.nHeights))
210
210
211 return 1
211 return 1
212
212
213
213
214 def filterByHeights(self, window):
214 def filterByHeights(self, window):
215
215
216 deltaHeight = self.dataOut.heightList[1] - self.dataOut.heightList[0]
216 deltaHeight = self.dataOut.heightList[1] - self.dataOut.heightList[0]
217
217
218 if window == None:
218 if window == None:
219 window = (self.dataOut.radarControllerHeaderObj.txA/self.dataOut.radarControllerHeaderObj.nBaud) / deltaHeight
219 window = (self.dataOut.radarControllerHeaderObj.txA/self.dataOut.radarControllerHeaderObj.nBaud) / deltaHeight
220
220
221 newdelta = deltaHeight * window
221 newdelta = deltaHeight * window
222 r = self.dataOut.nHeights % window
222 r = self.dataOut.nHeights % window
223 newheights = (self.dataOut.nHeights-r)/window
223 newheights = (self.dataOut.nHeights-r)/window
224
224
225 if newheights <= 1:
225 if newheights <= 1:
226 raise ValueError("filterByHeights: Too few heights. Current number of heights is %d and window is %d" %(self.dataOut.nHeights, window))
226 raise ValueError("filterByHeights: Too few heights. Current number of heights is %d and window is %d" %(self.dataOut.nHeights, window))
227
227
228 if self.dataOut.flagDataAsBlock:
228 if self.dataOut.flagDataAsBlock:
229 """
229 """
230 Si la data es obtenida por bloques, dimension = [nChannels, nProfiles, nHeis]
230 Si la data es obtenida por bloques, dimension = [nChannels, nProfiles, nHeis]
231 """
231 """
232 buffer = self.dataOut.data[:, :, 0:int(self.dataOut.nHeights-r)]
232 buffer = self.dataOut.data[:, :, 0:int(self.dataOut.nHeights-r)]
233 buffer = buffer.reshape(self.dataOut.nChannels, self.dataOut.nProfiles, int(self.dataOut.nHeights/window), window)
233 buffer = buffer.reshape(self.dataOut.nChannels, self.dataOut.nProfiles, int(self.dataOut.nHeights/window), window)
234 buffer = numpy.sum(buffer,3)
234 buffer = numpy.sum(buffer,3)
235
235
236 else:
236 else:
237 buffer = self.dataOut.data[:,0:int(self.dataOut.nHeights-r)]
237 buffer = self.dataOut.data[:,0:int(self.dataOut.nHeights-r)]
238 buffer = buffer.reshape(self.dataOut.nChannels,int(self.dataOut.nHeights/window),int(window))
238 buffer = buffer.reshape(self.dataOut.nChannels,int(self.dataOut.nHeights/window),int(window))
239 buffer = numpy.sum(buffer,2)
239 buffer = numpy.sum(buffer,2)
240
240
241 self.dataOut.data = buffer
241 self.dataOut.data = buffer
242 self.dataOut.heightList = self.dataOut.heightList[0] + numpy.arange( newheights )*newdelta
242 self.dataOut.heightList = self.dataOut.heightList[0] + numpy.arange( newheights )*newdelta
243 self.dataOut.windowOfFilter = window
243 self.dataOut.windowOfFilter = window
244
244
245 def setH0(self, h0, deltaHeight = None):
245 def setH0(self, h0, deltaHeight = None):
246
246
247 if not deltaHeight:
247 if not deltaHeight:
248 deltaHeight = self.dataOut.heightList[1] - self.dataOut.heightList[0]
248 deltaHeight = self.dataOut.heightList[1] - self.dataOut.heightList[0]
249
249
250 nHeights = self.dataOut.nHeights
250 nHeights = self.dataOut.nHeights
251
251
252 newHeiRange = h0 + numpy.arange(nHeights)*deltaHeight
252 newHeiRange = h0 + numpy.arange(nHeights)*deltaHeight
253
253
254 self.dataOut.heightList = newHeiRange
254 self.dataOut.heightList = newHeiRange
255
255
256 def deFlip(self, channelList = []):
256 def deFlip(self, channelList = []):
257
257
258 data = self.dataOut.data.copy()
258 data = self.dataOut.data.copy()
259
259
260 if self.dataOut.flagDataAsBlock:
260 if self.dataOut.flagDataAsBlock:
261 flip = self.flip
261 flip = self.flip
262 profileList = list(range(self.dataOut.nProfiles))
262 profileList = list(range(self.dataOut.nProfiles))
263
263
264 if not channelList:
264 if not channelList:
265 for thisProfile in profileList:
265 for thisProfile in profileList:
266 data[:,thisProfile,:] = data[:,thisProfile,:]*flip
266 data[:,thisProfile,:] = data[:,thisProfile,:]*flip
267 flip *= -1.0
267 flip *= -1.0
268 else:
268 else:
269 for thisChannel in channelList:
269 for thisChannel in channelList:
270 if thisChannel not in self.dataOut.channelList:
270 if thisChannel not in self.dataOut.channelList:
271 continue
271 continue
272
272
273 for thisProfile in profileList:
273 for thisProfile in profileList:
274 data[thisChannel,thisProfile,:] = data[thisChannel,thisProfile,:]*flip
274 data[thisChannel,thisProfile,:] = data[thisChannel,thisProfile,:]*flip
275 flip *= -1.0
275 flip *= -1.0
276
276
277 self.flip = flip
277 self.flip = flip
278
278
279 else:
279 else:
280 if not channelList:
280 if not channelList:
281 data[:,:] = data[:,:]*self.flip
281 data[:,:] = data[:,:]*self.flip
282 else:
282 else:
283 for thisChannel in channelList:
283 for thisChannel in channelList:
284 if thisChannel not in self.dataOut.channelList:
284 if thisChannel not in self.dataOut.channelList:
285 continue
285 continue
286
286
287 data[thisChannel,:] = data[thisChannel,:]*self.flip
287 data[thisChannel,:] = data[thisChannel,:]*self.flip
288
288
289 self.flip *= -1.
289 self.flip *= -1.
290
290
291 self.dataOut.data = data
291 self.dataOut.data = data
292
292
293 def setRadarFrequency(self, frequency=None):
293 def setRadarFrequency(self, frequency=None):
294
294
295 if frequency != None:
295 if frequency != None:
296 self.dataOut.frequency = frequency
296 self.dataOut.frequency = frequency
297
297
298 return 1
298 return 1
299
299
300 def interpolateHeights(self, topLim, botLim):
300 def interpolateHeights(self, topLim, botLim):
301 #69 al 72 para julia
301 #69 al 72 para julia
302 #82-84 para meteoros
302 #82-84 para meteoros
303 if len(numpy.shape(self.dataOut.data))==2:
303 if len(numpy.shape(self.dataOut.data))==2:
304 sampInterp = (self.dataOut.data[:,botLim-1] + self.dataOut.data[:,topLim+1])/2
304 sampInterp = (self.dataOut.data[:,botLim-1] + self.dataOut.data[:,topLim+1])/2
305 sampInterp = numpy.transpose(numpy.tile(sampInterp,(topLim-botLim + 1,1)))
305 sampInterp = numpy.transpose(numpy.tile(sampInterp,(topLim-botLim + 1,1)))
306 #self.dataOut.data[:,botLim:limSup+1] = sampInterp
306 #self.dataOut.data[:,botLim:limSup+1] = sampInterp
307 self.dataOut.data[:,botLim:topLim+1] = sampInterp
307 self.dataOut.data[:,botLim:topLim+1] = sampInterp
308 else:
308 else:
309 nHeights = self.dataOut.data.shape[2]
309 nHeights = self.dataOut.data.shape[2]
310 x = numpy.hstack((numpy.arange(botLim),numpy.arange(topLim+1,nHeights)))
310 x = numpy.hstack((numpy.arange(botLim),numpy.arange(topLim+1,nHeights)))
311 y = self.dataOut.data[:,:,list(range(botLim))+list(range(topLim+1,nHeights))]
311 y = self.dataOut.data[:,:,list(range(botLim))+list(range(topLim+1,nHeights))]
312 f = interpolate.interp1d(x, y, axis = 2)
312 f = interpolate.interp1d(x, y, axis = 2)
313 xnew = numpy.arange(botLim,topLim+1)
313 xnew = numpy.arange(botLim,topLim+1)
314 ynew = f(xnew)
314 ynew = f(xnew)
315
315
316 self.dataOut.data[:,:,botLim:topLim+1] = ynew
316 self.dataOut.data[:,:,botLim:topLim+1] = ynew
317
317
318 # import collections
318 # import collections
319
319
320 class CohInt(Operation):
320 class CohInt(Operation):
321
321
322 isConfig = False
322 isConfig = False
323 __profIndex = 0
323 __profIndex = 0
324 __byTime = False
324 __byTime = False
325 __initime = None
325 __initime = None
326 __lastdatatime = None
326 __lastdatatime = None
327 __integrationtime = None
327 __integrationtime = None
328 __buffer = None
328 __buffer = None
329 __bufferStride = []
329 __bufferStride = []
330 __dataReady = False
330 __dataReady = False
331 __profIndexStride = 0
331 __profIndexStride = 0
332 __dataToPutStride = False
332 __dataToPutStride = False
333 n = None
333 n = None
334
334
335 def __init__(self, **kwargs):
335 def __init__(self, **kwargs):
336
336
337 Operation.__init__(self, **kwargs)
337 Operation.__init__(self, **kwargs)
338
338
339 # self.isConfig = False
339 # self.isConfig = False
340
340
341 def setup(self, n=None, timeInterval=None, stride=None, overlapping=False, byblock=False):
341 def setup(self, n=None, timeInterval=None, stride=None, overlapping=False, byblock=False):
342 """
342 """
343 Set the parameters of the integration class.
343 Set the parameters of the integration class.
344
344
345 Inputs:
345 Inputs:
346
346
347 n : Number of coherent integrations
347 n : Number of coherent integrations
348 timeInterval : Time of integration. If the parameter "n" is selected this one does not work
348 timeInterval : Time of integration. If the parameter "n" is selected this one does not work
349 overlapping :
349 overlapping :
350 """
350 """
351
351
352 self.__initime = None
352 self.__initime = None
353 self.__lastdatatime = 0
353 self.__lastdatatime = 0
354 self.__buffer = None
354 self.__buffer = None
355 self.__dataReady = False
355 self.__dataReady = False
356 self.byblock = byblock
356 self.byblock = byblock
357 self.stride = stride
357 self.stride = stride
358
358
359 if n == None and timeInterval == None:
359 if n == None and timeInterval == None:
360 raise ValueError("n or timeInterval should be specified ...")
360 raise ValueError("n or timeInterval should be specified ...")
361
361
362 if n != None:
362 if n != None:
363 self.n = n
363 self.n = n
364 self.__byTime = False
364 self.__byTime = False
365 else:
365 else:
366 self.__integrationtime = timeInterval #* 60. #if (type(timeInterval)!=integer) -> change this line
366 self.__integrationtime = timeInterval #* 60. #if (type(timeInterval)!=integer) -> change this line
367 self.n = 9999
367 self.n = 9999
368 self.__byTime = True
368 self.__byTime = True
369
369
370 if overlapping:
370 if overlapping:
371 self.__withOverlapping = True
371 self.__withOverlapping = True
372 self.__buffer = None
372 self.__buffer = None
373 else:
373 else:
374 self.__withOverlapping = False
374 self.__withOverlapping = False
375 self.__buffer = 0
375 self.__buffer = 0
376
376
377 self.__profIndex = 0
377 self.__profIndex = 0
378
378
379 def putData(self, data):
379 def putData(self, data):
380
380
381 """
381 """
382 Add a profile to the __buffer and increase in one the __profileIndex
382 Add a profile to the __buffer and increase in one the __profileIndex
383
383
384 """
384 """
385
385
386 if not self.__withOverlapping:
386 if not self.__withOverlapping:
387 self.__buffer += data.copy()
387 self.__buffer += data.copy()
388 self.__profIndex += 1
388 self.__profIndex += 1
389 return
389 return
390
390
391 #Overlapping data
391 #Overlapping data
392 nChannels, nHeis = data.shape
392 nChannels, nHeis = data.shape
393 data = numpy.reshape(data, (1, nChannels, nHeis))
393 data = numpy.reshape(data, (1, nChannels, nHeis))
394
394
395 #If the buffer is empty then it takes the data value
395 #If the buffer is empty then it takes the data value
396 if self.__buffer is None:
396 if self.__buffer is None:
397 self.__buffer = data
397 self.__buffer = data
398 self.__profIndex += 1
398 self.__profIndex += 1
399 return
399 return
400
400
401 #If the buffer length is lower than n then stakcing the data value
401 #If the buffer length is lower than n then stakcing the data value
402 if self.__profIndex < self.n:
402 if self.__profIndex < self.n:
403 self.__buffer = numpy.vstack((self.__buffer, data))
403 self.__buffer = numpy.vstack((self.__buffer, data))
404 self.__profIndex += 1
404 self.__profIndex += 1
405 return
405 return
406
406
407 #If the buffer length is equal to n then replacing the last buffer value with the data value
407 #If the buffer length is equal to n then replacing the last buffer value with the data value
408 self.__buffer = numpy.roll(self.__buffer, -1, axis=0)
408 self.__buffer = numpy.roll(self.__buffer, -1, axis=0)
409 self.__buffer[self.n-1] = data
409 self.__buffer[self.n-1] = data
410 self.__profIndex = self.n
410 self.__profIndex = self.n
411 return
411 return
412
412
413
413
414 def pushData(self):
414 def pushData(self):
415 """
415 """
416 Return the sum of the last profiles and the profiles used in the sum.
416 Return the sum of the last profiles and the profiles used in the sum.
417
417
418 Affected:
418 Affected:
419
419
420 self.__profileIndex
420 self.__profileIndex
421
421
422 """
422 """
423
423
424 if not self.__withOverlapping:
424 if not self.__withOverlapping:
425 data = self.__buffer
425 data = self.__buffer
426 n = self.__profIndex
426 n = self.__profIndex
427
427
428 self.__buffer = 0
428 self.__buffer = 0
429 self.__profIndex = 0
429 self.__profIndex = 0
430
430
431 return data, n
431 return data, n
432
432
433 #Integration with Overlapping
433 #Integration with Overlapping
434 data = numpy.sum(self.__buffer, axis=0)
434 data = numpy.sum(self.__buffer, axis=0)
435 # print data
435 # print data
436 # raise
436 # raise
437 n = self.__profIndex
437 n = self.__profIndex
438
438
439 return data, n
439 return data, n
440
440
441 def byProfiles(self, data):
441 def byProfiles(self, data):
442
442
443 self.__dataReady = False
443 self.__dataReady = False
444 avgdata = None
444 avgdata = None
445 # n = None
445 # n = None
446 # print data
446 # print data
447 # raise
447 # raise
448 self.putData(data)
448 self.putData(data)
449
449
450 if self.__profIndex == self.n:
450 if self.__profIndex == self.n:
451 avgdata, n = self.pushData()
451 avgdata, n = self.pushData()
452 self.__dataReady = True
452 self.__dataReady = True
453
453
454 return avgdata
454 return avgdata
455
455
456 def byTime(self, data, datatime):
456 def byTime(self, data, datatime):
457
457
458 self.__dataReady = False
458 self.__dataReady = False
459 avgdata = None
459 avgdata = None
460 n = None
460 n = None
461
461
462 self.putData(data)
462 self.putData(data)
463
463
464 if (datatime - self.__initime) >= self.__integrationtime:
464 if (datatime - self.__initime) >= self.__integrationtime:
465 avgdata, n = self.pushData()
465 avgdata, n = self.pushData()
466 self.n = n
466 self.n = n
467 self.__dataReady = True
467 self.__dataReady = True
468
468
469 return avgdata
469 return avgdata
470
470
471 def integrateByStride(self, data, datatime):
471 def integrateByStride(self, data, datatime):
472 # print data
472 # print data
473 if self.__profIndex == 0:
473 if self.__profIndex == 0:
474 self.__buffer = [[data.copy(), datatime]]
474 self.__buffer = [[data.copy(), datatime]]
475 else:
475 else:
476 self.__buffer.append([data.copy(),datatime])
476 self.__buffer.append([data.copy(),datatime])
477 self.__profIndex += 1
477 self.__profIndex += 1
478 self.__dataReady = False
478 self.__dataReady = False
479
479
480 if self.__profIndex == self.n * self.stride :
480 if self.__profIndex == self.n * self.stride :
481 self.__dataToPutStride = True
481 self.__dataToPutStride = True
482 self.__profIndexStride = 0
482 self.__profIndexStride = 0
483 self.__profIndex = 0
483 self.__profIndex = 0
484 self.__bufferStride = []
484 self.__bufferStride = []
485 for i in range(self.stride):
485 for i in range(self.stride):
486 current = self.__buffer[i::self.stride]
486 current = self.__buffer[i::self.stride]
487 data = numpy.sum([t[0] for t in current], axis=0)
487 data = numpy.sum([t[0] for t in current], axis=0)
488 avgdatatime = numpy.average([t[1] for t in current])
488 avgdatatime = numpy.average([t[1] for t in current])
489 # print data
489 # print data
490 self.__bufferStride.append((data, avgdatatime))
490 self.__bufferStride.append((data, avgdatatime))
491
491
492 if self.__dataToPutStride:
492 if self.__dataToPutStride:
493 self.__dataReady = True
493 self.__dataReady = True
494 self.__profIndexStride += 1
494 self.__profIndexStride += 1
495 if self.__profIndexStride == self.stride:
495 if self.__profIndexStride == self.stride:
496 self.__dataToPutStride = False
496 self.__dataToPutStride = False
497 # print self.__bufferStride[self.__profIndexStride - 1]
497 # print self.__bufferStride[self.__profIndexStride - 1]
498 # raise
498 # raise
499 return self.__bufferStride[self.__profIndexStride - 1]
499 return self.__bufferStride[self.__profIndexStride - 1]
500
500
501
501
502 return None, None
502 return None, None
503
503
504 def integrate(self, data, datatime=None):
504 def integrate(self, data, datatime=None):
505
505
506 if self.__initime == None:
506 if self.__initime == None:
507 self.__initime = datatime
507 self.__initime = datatime
508
508
509 if self.__byTime:
509 if self.__byTime:
510 avgdata = self.byTime(data, datatime)
510 avgdata = self.byTime(data, datatime)
511 else:
511 else:
512 avgdata = self.byProfiles(data)
512 avgdata = self.byProfiles(data)
513
513
514
514
515 self.__lastdatatime = datatime
515 self.__lastdatatime = datatime
516
516
517 if avgdata is None:
517 if avgdata is None:
518 return None, None
518 return None, None
519
519
520 avgdatatime = self.__initime
520 avgdatatime = self.__initime
521
521
522 deltatime = datatime - self.__lastdatatime
522 deltatime = datatime - self.__lastdatatime
523
523
524 if not self.__withOverlapping:
524 if not self.__withOverlapping:
525 self.__initime = datatime
525 self.__initime = datatime
526 else:
526 else:
527 self.__initime += deltatime
527 self.__initime += deltatime
528
528
529 return avgdata, avgdatatime
529 return avgdata, avgdatatime
530
530
531 def integrateByBlock(self, dataOut):
531 def integrateByBlock(self, dataOut):
532
532
533 times = int(dataOut.data.shape[1]/self.n)
533 times = int(dataOut.data.shape[1]/self.n)
534 avgdata = numpy.zeros((dataOut.nChannels, times, dataOut.nHeights), dtype=numpy.complex)
534 avgdata = numpy.zeros((dataOut.nChannels, times, dataOut.nHeights), dtype=numpy.complex)
535
535
536 id_min = 0
536 id_min = 0
537 id_max = self.n
537 id_max = self.n
538
538
539 for i in range(times):
539 for i in range(times):
540 junk = dataOut.data[:,id_min:id_max,:]
540 junk = dataOut.data[:,id_min:id_max,:]
541 avgdata[:,i,:] = junk.sum(axis=1)
541 avgdata[:,i,:] = junk.sum(axis=1)
542 id_min += self.n
542 id_min += self.n
543 id_max += self.n
543 id_max += self.n
544
544
545 timeInterval = dataOut.ippSeconds*self.n
545 timeInterval = dataOut.ippSeconds*self.n
546 avgdatatime = (times - 1) * timeInterval + dataOut.utctime
546 avgdatatime = (times - 1) * timeInterval + dataOut.utctime
547 self.__dataReady = True
547 self.__dataReady = True
548 return avgdata, avgdatatime
548 return avgdata, avgdatatime
549
549
550 def run(self, dataOut, n=None, timeInterval=None, stride=None, overlapping=False, byblock=False, **kwargs):
550 def run(self, dataOut, n=None, timeInterval=None, stride=None, overlapping=False, byblock=False, **kwargs):
551
551
552 if not self.isConfig:
552 if not self.isConfig:
553 self.setup(n=n, stride=stride, timeInterval=timeInterval, overlapping=overlapping, byblock=byblock, **kwargs)
553 self.setup(n=n, stride=stride, timeInterval=timeInterval, overlapping=overlapping, byblock=byblock, **kwargs)
554 self.isConfig = True
554 self.isConfig = True
555
555
556 if dataOut.flagDataAsBlock:
556 if dataOut.flagDataAsBlock:
557 """
557 """
558 Si la data es leida por bloques, dimension = [nChannels, nProfiles, nHeis]
558 Si la data es leida por bloques, dimension = [nChannels, nProfiles, nHeis]
559 """
559 """
560 avgdata, avgdatatime = self.integrateByBlock(dataOut)
560 avgdata, avgdatatime = self.integrateByBlock(dataOut)
561 dataOut.nProfiles /= self.n
561 dataOut.nProfiles /= self.n
562 else:
562 else:
563 if stride is None:
563 if stride is None:
564 avgdata, avgdatatime = self.integrate(dataOut.data, dataOut.utctime)
564 avgdata, avgdatatime = self.integrate(dataOut.data, dataOut.utctime)
565 else:
565 else:
566 avgdata, avgdatatime = self.integrateByStride(dataOut.data, dataOut.utctime)
566 avgdata, avgdatatime = self.integrateByStride(dataOut.data, dataOut.utctime)
567
567
568
568
569 # dataOut.timeInterval *= n
569 # dataOut.timeInterval *= n
570 dataOut.flagNoData = True
570 dataOut.flagNoData = True
571
571
572 if self.__dataReady:
572 if self.__dataReady:
573 dataOut.data = avgdata
573 dataOut.data = avgdata
574 dataOut.nCohInt *= self.n
574 dataOut.nCohInt *= self.n
575 dataOut.utctime = avgdatatime
575 dataOut.utctime = avgdatatime
576 # print avgdata, avgdatatime
576 # print avgdata, avgdatatime
577 # raise
577 # raise
578 # dataOut.timeInterval = dataOut.ippSeconds * dataOut.nCohInt
578 # dataOut.timeInterval = dataOut.ippSeconds * dataOut.nCohInt
579 dataOut.flagNoData = False
579 dataOut.flagNoData = False
580 return dataOut
580 return dataOut
581
581
582 class Decoder(Operation):
582 class Decoder(Operation):
583
583
584 isConfig = False
584 isConfig = False
585 __profIndex = 0
585 __profIndex = 0
586
586
587 code = None
587 code = None
588
588
589 nCode = None
589 nCode = None
590 nBaud = None
590 nBaud = None
591
591
592 def __init__(self, **kwargs):
592 def __init__(self, **kwargs):
593
593
594 Operation.__init__(self, **kwargs)
594 Operation.__init__(self, **kwargs)
595
595
596 self.times = None
596 self.times = None
597 self.osamp = None
597 self.osamp = None
598 # self.__setValues = False
598 # self.__setValues = False
599 self.isConfig = False
599 self.isConfig = False
600 self.setupReq = False
600 self.setupReq = False
601 def setup(self, code, osamp, dataOut):
601 def setup(self, code, osamp, dataOut):
602
602
603 self.__profIndex = 0
603 self.__profIndex = 0
604
604
605 self.code = code
605 self.code = code
606
606
607 self.nCode = len(code)
607 self.nCode = len(code)
608 self.nBaud = len(code[0])
608 self.nBaud = len(code[0])
609
610 if (osamp != None) and (osamp >1):
609 if (osamp != None) and (osamp >1):
611 self.osamp = osamp
610 self.osamp = osamp
612 self.code = numpy.repeat(code, repeats=self.osamp, axis=1)
611 self.code = numpy.repeat(code, repeats=self.osamp, axis=1)
613 self.nBaud = self.nBaud*self.osamp
612 self.nBaud = self.nBaud*self.osamp
614
613
615 self.__nChannels = dataOut.nChannels
614 self.__nChannels = dataOut.nChannels
616 self.__nProfiles = dataOut.nProfiles
615 self.__nProfiles = dataOut.nProfiles
617 self.__nHeis = dataOut.nHeights
616 self.__nHeis = dataOut.nHeights
618
617
619 if self.__nHeis < self.nBaud:
618 if self.__nHeis < self.nBaud:
620 raise ValueError('Number of heights (%d) should be greater than number of bauds (%d)' %(self.__nHeis, self.nBaud))
619 raise ValueError('Number of heights (%d) should be greater than number of bauds (%d)' %(self.__nHeis, self.nBaud))
621
620
622 #Frequency
621 #Frequency
623 __codeBuffer = numpy.zeros((self.nCode, self.__nHeis), dtype=numpy.complex)
622 __codeBuffer = numpy.zeros((self.nCode, self.__nHeis), dtype=numpy.complex)
624
623
625 __codeBuffer[:,0:self.nBaud] = self.code
624 __codeBuffer[:,0:self.nBaud] = self.code
626
625
627 self.fft_code = numpy.conj(numpy.fft.fft(__codeBuffer, axis=1))
626 self.fft_code = numpy.conj(numpy.fft.fft(__codeBuffer, axis=1))
628
627
629 if dataOut.flagDataAsBlock:
628 if dataOut.flagDataAsBlock:
630
629
631 self.ndatadec = self.__nHeis #- self.nBaud + 1
630 self.ndatadec = self.__nHeis #- self.nBaud + 1
632
631
633 self.datadecTime = numpy.zeros((self.__nChannels, self.__nProfiles, self.ndatadec), dtype=numpy.complex)
632 self.datadecTime = numpy.zeros((self.__nChannels, self.__nProfiles, self.ndatadec), dtype=numpy.complex)
634
633
635 else:
634 else:
636
635
637 #Time
636 #Time
638 self.ndatadec = self.__nHeis #- self.nBaud + 1
637 self.ndatadec = self.__nHeis #- self.nBaud + 1
639
638
640 self.datadecTime = numpy.zeros((self.__nChannels, self.ndatadec), dtype=numpy.complex)
639 self.datadecTime = numpy.zeros((self.__nChannels, self.ndatadec), dtype=numpy.complex)
641
640
642 def __convolutionInFreq(self, data):
641 def __convolutionInFreq(self, data):
643
642
644 fft_code = self.fft_code[self.__profIndex].reshape(1,-1)
643 fft_code = self.fft_code[self.__profIndex].reshape(1,-1)
645
644
646 fft_data = numpy.fft.fft(data, axis=1)
645 fft_data = numpy.fft.fft(data, axis=1)
647
646
648 conv = fft_data*fft_code
647 conv = fft_data*fft_code
649
648
650 data = numpy.fft.ifft(conv,axis=1)
649 data = numpy.fft.ifft(conv,axis=1)
651
650
652 return data
651 return data
653
652
654 def __convolutionInFreqOpt(self, data):
653 def __convolutionInFreqOpt(self, data):
655
654
656 raise NotImplementedError
655 raise NotImplementedError
657
656
658 def __convolutionInTime(self, data):
657 def __convolutionInTime(self, data):
659
658
660 code = self.code[self.__profIndex]
659 code = self.code[self.__profIndex]
661 for i in range(self.__nChannels):
660 for i in range(self.__nChannels):
662 self.datadecTime[i,:] = numpy.correlate(data[i,:], code, mode='full')[self.nBaud-1:]
661 self.datadecTime[i,:] = numpy.correlate(data[i,:], code, mode='full')[self.nBaud-1:]
663
662
664 return self.datadecTime
663 return self.datadecTime
665
664
666 def __convolutionByBlockInTime(self, data):
665 def __convolutionByBlockInTime(self, data):
667
666
668 repetitions = int(self.__nProfiles / self.nCode)
667 repetitions = int(self.__nProfiles / self.nCode)
669 junk = numpy.lib.stride_tricks.as_strided(self.code, (repetitions, self.code.size), (0, self.code.itemsize))
668 junk = numpy.lib.stride_tricks.as_strided(self.code, (repetitions, self.code.size), (0, self.code.itemsize))
670 junk = junk.flatten()
669 junk = junk.flatten()
671 code_block = numpy.reshape(junk, (self.nCode*repetitions, self.nBaud))
670 code_block = numpy.reshape(junk, (self.nCode*repetitions, self.nBaud))
672 profilesList = range(self.__nProfiles)
671 profilesList = range(self.__nProfiles)
673
672
674 for i in range(self.__nChannels):
673 for i in range(self.__nChannels):
675 for j in profilesList:
674 for j in profilesList:
676 self.datadecTime[i,j,:] = numpy.correlate(data[i,j,:], code_block[j,:], mode='full')[self.nBaud-1:]
675 self.datadecTime[i,j,:] = numpy.correlate(data[i,j,:], code_block[j,:], mode='full')[self.nBaud-1:]
677 return self.datadecTime
676 return self.datadecTime
678
677
679 def __convolutionByBlockInFreq(self, data):
678 def __convolutionByBlockInFreq(self, data):
680
679
681 raise NotImplementedError("Decoder by frequency fro Blocks not implemented")
680 raise NotImplementedError("Decoder by frequency fro Blocks not implemented")
682
681
683
682
684 fft_code = self.fft_code[self.__profIndex].reshape(1,-1)
683 fft_code = self.fft_code[self.__profIndex].reshape(1,-1)
685
684
686 fft_data = numpy.fft.fft(data, axis=2)
685 fft_data = numpy.fft.fft(data, axis=2)
687
686
688 conv = fft_data*fft_code
687 conv = fft_data*fft_code
689
688
690 data = numpy.fft.ifft(conv,axis=2)
689 data = numpy.fft.ifft(conv,axis=2)
691
690
692 return data
691 return data
693
692
694
693
695 def run(self, dataOut, code=None, nCode=None, nBaud=None, mode = 0, osamp=None, times=None):
694 def run(self, dataOut, code=None, nCode=None, nBaud=None, mode = 0, osamp=None, times=None):
696
695
697 if dataOut.flagDecodeData:
696 if dataOut.flagDecodeData:
698 print("This data is already decoded, recoding again ...")
697 print("This data is already decoded, recoding again ...")
699
698
700 if not self.isConfig:
699 if not self.isConfig:
701
700
702 if code is None:
701 if code is None:
703 if dataOut.code is None:
702 if dataOut.code is None:
704 raise ValueError("Code could not be read from %s instance. Enter a value in Code parameter" %dataOut.type)
703 raise ValueError("Code could not be read from %s instance. Enter a value in Code parameter" %dataOut.type)
705
704
706 code = dataOut.code
705 code = dataOut.code
707 else:
706 else:
708 code = numpy.array(code).reshape(nCode,nBaud)
707 code = numpy.array(code).reshape(nCode,nBaud)
709 self.setup(code, osamp, dataOut)
708 self.setup(code, osamp, dataOut)
710
709
711 self.isConfig = True
710 self.isConfig = True
712
711
713 if mode == 3:
712 if mode == 3:
714 sys.stderr.write("Decoder Warning: mode=%d is not valid, using mode=0\n" %mode)
713 sys.stderr.write("Decoder Warning: mode=%d is not valid, using mode=0\n" %mode)
715
714
716 if times != None:
715 if times != None:
717 sys.stderr.write("Decoder Warning: Argument 'times' in not used anymore\n")
716 sys.stderr.write("Decoder Warning: Argument 'times' in not used anymore\n")
718
717
719 if self.code is None:
718 if self.code is None:
720 print("Fail decoding: Code is not defined.")
719 print("Fail decoding: Code is not defined.")
721 return
720 return
722
721
723 self.__nProfiles = dataOut.nProfiles
722 self.__nProfiles = dataOut.nProfiles
724 datadec = None
723 datadec = None
725
724
726 if mode == 3:
725 if mode == 3:
727 mode = 0
726 mode = 0
728
727
729 if dataOut.flagDataAsBlock:
728 if dataOut.flagDataAsBlock:
730 """
729 """
731 Decoding when data have been read as block,
730 Decoding when data have been read as block,
732 """
731 """
733
732
734 if mode == 0:
733 if mode == 0:
735 datadec = self.__convolutionByBlockInTime(dataOut.data)
734 datadec = self.__convolutionByBlockInTime(dataOut.data)
736 if mode == 1:
735 if mode == 1:
737 datadec = self.__convolutionByBlockInFreq(dataOut.data)
736 datadec = self.__convolutionByBlockInFreq(dataOut.data)
738 else:
737 else:
739 """
738 """
740 Decoding when data have been read profile by profile
739 Decoding when data have been read profile by profile
741 """
740 """
742 if mode == 0:
741 if mode == 0:
743 datadec = self.__convolutionInTime(dataOut.data)
742 datadec = self.__convolutionInTime(dataOut.data)
744
743
745 if mode == 1:
744 if mode == 1:
746 datadec = self.__convolutionInFreq(dataOut.data)
745 datadec = self.__convolutionInFreq(dataOut.data)
747
746
748 if mode == 2:
747 if mode == 2:
749 datadec = self.__convolutionInFreqOpt(dataOut.data)
748 datadec = self.__convolutionInFreqOpt(dataOut.data)
750
749
751 if datadec is None:
750 if datadec is None:
752 raise ValueError("Codification mode selected is not valid: mode=%d. Try selecting 0 or 1" %mode)
751 raise ValueError("Codification mode selected is not valid: mode=%d. Try selecting 0 or 1" %mode)
753
752
754 dataOut.code = self.code
753 dataOut.code = self.code
755 dataOut.nCode = self.nCode
754 dataOut.nCode = self.nCode
756 dataOut.nBaud = self.nBaud
755 dataOut.nBaud = self.nBaud
757
756
758 dataOut.data = datadec
757 dataOut.data = datadec
759
758
760 dataOut.heightList = dataOut.heightList[0:datadec.shape[-1]]
759 dataOut.heightList = dataOut.heightList[0:datadec.shape[-1]]
761
760
762 dataOut.flagDecodeData = True #asumo q la data esta decodificada
761 dataOut.flagDecodeData = True #asumo q la data esta decodificada
763
762
764 if self.__profIndex == self.nCode-1:
763 if self.__profIndex == self.nCode-1:
765 self.__profIndex = 0
764 self.__profIndex = 0
766 return dataOut
765 return dataOut
767
766
768 self.__profIndex += 1
767 self.__profIndex += 1
769
768
770 return dataOut
769 return dataOut
771 # dataOut.flagDeflipData = True #asumo q la data no esta sin flip
770 # dataOut.flagDeflipData = True #asumo q la data no esta sin flip
772
771
773
772
774 class ProfileConcat(Operation):
773 class ProfileConcat(Operation):
775
774
776 isConfig = False
775 isConfig = False
777 buffer = None
776 buffer = None
778
777
779 def __init__(self, **kwargs):
778 def __init__(self, **kwargs):
780
779
781 Operation.__init__(self, **kwargs)
780 Operation.__init__(self, **kwargs)
782 self.profileIndex = 0
781 self.profileIndex = 0
783
782
784 def reset(self):
783 def reset(self):
785 self.buffer = numpy.zeros_like(self.buffer)
784 self.buffer = numpy.zeros_like(self.buffer)
786 self.start_index = 0
785 self.start_index = 0
787 self.times = 1
786 self.times = 1
788
787
789 def setup(self, data, m, n=1):
788 def setup(self, data, m, n=1):
790 self.buffer = numpy.zeros((data.shape[0],data.shape[1]*m),dtype=type(data[0,0]))
789 self.buffer = numpy.zeros((data.shape[0],data.shape[1]*m),dtype=type(data[0,0]))
791 self.nHeights = data.shape[1]#.nHeights
790 self.nHeights = data.shape[1]#.nHeights
792 self.start_index = 0
791 self.start_index = 0
793 self.times = 1
792 self.times = 1
794
793
795 def concat(self, data):
794 def concat(self, data):
796
795
797 self.buffer[:,self.start_index:self.nHeights*self.times] = data.copy()
796 self.buffer[:,self.start_index:self.nHeights*self.times] = data.copy()
798 self.start_index = self.start_index + self.nHeights
797 self.start_index = self.start_index + self.nHeights
799
798
800 def run(self, dataOut, m):
799 def run(self, dataOut, m):
801 dataOut.flagNoData = True
800 dataOut.flagNoData = True
802
801
803 if not self.isConfig:
802 if not self.isConfig:
804 self.setup(dataOut.data, m, 1)
803 self.setup(dataOut.data, m, 1)
805 self.isConfig = True
804 self.isConfig = True
806
805
807 if dataOut.flagDataAsBlock:
806 if dataOut.flagDataAsBlock:
808 raise ValueError("ProfileConcat can only be used when voltage have been read profile by profile, getBlock = False")
807 raise ValueError("ProfileConcat can only be used when voltage have been read profile by profile, getBlock = False")
809
808
810 else:
809 else:
811 self.concat(dataOut.data)
810 self.concat(dataOut.data)
812 self.times += 1
811 self.times += 1
813 if self.times > m:
812 if self.times > m:
814 dataOut.data = self.buffer
813 dataOut.data = self.buffer
815 self.reset()
814 self.reset()
816 dataOut.flagNoData = False
815 dataOut.flagNoData = False
817 # se deben actualizar mas propiedades del header y del objeto dataOut, por ejemplo, las alturas
816 # se deben actualizar mas propiedades del header y del objeto dataOut, por ejemplo, las alturas
818 deltaHeight = dataOut.heightList[1] - dataOut.heightList[0]
817 deltaHeight = dataOut.heightList[1] - dataOut.heightList[0]
819 xf = dataOut.heightList[0] + dataOut.nHeights * deltaHeight * m
818 xf = dataOut.heightList[0] + dataOut.nHeights * deltaHeight * m
820 dataOut.heightList = numpy.arange(dataOut.heightList[0], xf, deltaHeight)
819 dataOut.heightList = numpy.arange(dataOut.heightList[0], xf, deltaHeight)
821 dataOut.ippSeconds *= m
820 dataOut.ippSeconds *= m
822 return dataOut
821 return dataOut
823
822
824 class ProfileSelector(Operation):
823 class ProfileSelector(Operation):
825
824
826 profileIndex = None
825 profileIndex = None
827 # Tamanho total de los perfiles
826 # Tamanho total de los perfiles
828 nProfiles = None
827 nProfiles = None
829
828
830 def __init__(self, **kwargs):
829 def __init__(self, **kwargs):
831
830
832 Operation.__init__(self, **kwargs)
831 Operation.__init__(self, **kwargs)
833 self.profileIndex = 0
832 self.profileIndex = 0
834
833
835 def incProfileIndex(self):
834 def incProfileIndex(self):
836
835
837 self.profileIndex += 1
836 self.profileIndex += 1
838
837
839 if self.profileIndex >= self.nProfiles:
838 if self.profileIndex >= self.nProfiles:
840 self.profileIndex = 0
839 self.profileIndex = 0
841
840
842 def isThisProfileInRange(self, profileIndex, minIndex, maxIndex):
841 def isThisProfileInRange(self, profileIndex, minIndex, maxIndex):
843
842
844 if profileIndex < minIndex:
843 if profileIndex < minIndex:
845 return False
844 return False
846
845
847 if profileIndex > maxIndex:
846 if profileIndex > maxIndex:
848 return False
847 return False
849
848
850 return True
849 return True
851
850
852 def isThisProfileInList(self, profileIndex, profileList):
851 def isThisProfileInList(self, profileIndex, profileList):
853
852
854 if profileIndex not in profileList:
853 if profileIndex not in profileList:
855 return False
854 return False
856
855
857 return True
856 return True
858
857
859 def run(self, dataOut, profileList=None, profileRangeList=None, beam=None, byblock=False, rangeList = None, nProfiles=None):
858 def run(self, dataOut, profileList=None, profileRangeList=None, beam=None, byblock=False, rangeList = None, nProfiles=None):
860
859
861 """
860 """
862 ProfileSelector:
861 ProfileSelector:
863
862
864 Inputs:
863 Inputs:
865 profileList : Index of profiles selected. Example: profileList = (0,1,2,7,8)
864 profileList : Index of profiles selected. Example: profileList = (0,1,2,7,8)
866
865
867 profileRangeList : Minimum and maximum profile indexes. Example: profileRangeList = (4, 30)
866 profileRangeList : Minimum and maximum profile indexes. Example: profileRangeList = (4, 30)
868
867
869 rangeList : List of profile ranges. Example: rangeList = ((4, 30), (32, 64), (128, 256))
868 rangeList : List of profile ranges. Example: rangeList = ((4, 30), (32, 64), (128, 256))
870
869
871 """
870 """
872
871
873 if rangeList is not None:
872 if rangeList is not None:
874 if type(rangeList[0]) not in (tuple, list):
873 if type(rangeList[0]) not in (tuple, list):
875 rangeList = [rangeList]
874 rangeList = [rangeList]
876
875
877 dataOut.flagNoData = True
876 dataOut.flagNoData = True
878
877
879 if dataOut.flagDataAsBlock:
878 if dataOut.flagDataAsBlock:
880 """
879 """
881 data dimension = [nChannels, nProfiles, nHeis]
880 data dimension = [nChannels, nProfiles, nHeis]
882 """
881 """
883 if profileList != None:
882 if profileList != None:
884 dataOut.data = dataOut.data[:,profileList,:]
883 dataOut.data = dataOut.data[:,profileList,:]
885
884
886 if profileRangeList != None:
885 if profileRangeList != None:
887 minIndex = profileRangeList[0]
886 minIndex = profileRangeList[0]
888 maxIndex = profileRangeList[1]
887 maxIndex = profileRangeList[1]
889 profileList = list(range(minIndex, maxIndex+1))
888 profileList = list(range(minIndex, maxIndex+1))
890
889
891 dataOut.data = dataOut.data[:,minIndex:maxIndex+1,:]
890 dataOut.data = dataOut.data[:,minIndex:maxIndex+1,:]
892
891
893 if rangeList != None:
892 if rangeList != None:
894
893
895 profileList = []
894 profileList = []
896
895
897 for thisRange in rangeList:
896 for thisRange in rangeList:
898 minIndex = thisRange[0]
897 minIndex = thisRange[0]
899 maxIndex = thisRange[1]
898 maxIndex = thisRange[1]
900
899
901 profileList.extend(list(range(minIndex, maxIndex+1)))
900 profileList.extend(list(range(minIndex, maxIndex+1)))
902
901
903 dataOut.data = dataOut.data[:,profileList,:]
902 dataOut.data = dataOut.data[:,profileList,:]
904
903
905 dataOut.nProfiles = len(profileList)
904 dataOut.nProfiles = len(profileList)
906 dataOut.profileIndex = dataOut.nProfiles - 1
905 dataOut.profileIndex = dataOut.nProfiles - 1
907 dataOut.flagNoData = False
906 dataOut.flagNoData = False
908
907
909 return dataOut
908 return dataOut
910
909
911 """
910 """
912 data dimension = [nChannels, nHeis]
911 data dimension = [nChannels, nHeis]
913 """
912 """
914
913
915 if profileList != None:
914 if profileList != None:
916
915
917 if self.isThisProfileInList(dataOut.profileIndex, profileList):
916 if self.isThisProfileInList(dataOut.profileIndex, profileList):
918
917
919 self.nProfiles = len(profileList)
918 self.nProfiles = len(profileList)
920 dataOut.nProfiles = self.nProfiles
919 dataOut.nProfiles = self.nProfiles
921 dataOut.profileIndex = self.profileIndex
920 dataOut.profileIndex = self.profileIndex
922 dataOut.flagNoData = False
921 dataOut.flagNoData = False
923
922
924 self.incProfileIndex()
923 self.incProfileIndex()
925 return dataOut
924 return dataOut
926
925
927 if profileRangeList != None:
926 if profileRangeList != None:
928
927
929 minIndex = profileRangeList[0]
928 minIndex = profileRangeList[0]
930 maxIndex = profileRangeList[1]
929 maxIndex = profileRangeList[1]
931
930
932 if self.isThisProfileInRange(dataOut.profileIndex, minIndex, maxIndex):
931 if self.isThisProfileInRange(dataOut.profileIndex, minIndex, maxIndex):
933
932
934 self.nProfiles = maxIndex - minIndex + 1
933 self.nProfiles = maxIndex - minIndex + 1
935 dataOut.nProfiles = self.nProfiles
934 dataOut.nProfiles = self.nProfiles
936 dataOut.profileIndex = self.profileIndex
935 dataOut.profileIndex = self.profileIndex
937 dataOut.flagNoData = False
936 dataOut.flagNoData = False
938
937
939 self.incProfileIndex()
938 self.incProfileIndex()
940 return dataOut
939 return dataOut
941
940
942 if rangeList != None:
941 if rangeList != None:
943
942
944 nProfiles = 0
943 nProfiles = 0
945
944
946 for thisRange in rangeList:
945 for thisRange in rangeList:
947 minIndex = thisRange[0]
946 minIndex = thisRange[0]
948 maxIndex = thisRange[1]
947 maxIndex = thisRange[1]
949
948
950 nProfiles += maxIndex - minIndex + 1
949 nProfiles += maxIndex - minIndex + 1
951
950
952 for thisRange in rangeList:
951 for thisRange in rangeList:
953
952
954 minIndex = thisRange[0]
953 minIndex = thisRange[0]
955 maxIndex = thisRange[1]
954 maxIndex = thisRange[1]
956
955
957 if self.isThisProfileInRange(dataOut.profileIndex, minIndex, maxIndex):
956 if self.isThisProfileInRange(dataOut.profileIndex, minIndex, maxIndex):
958
957
959 self.nProfiles = nProfiles
958 self.nProfiles = nProfiles
960 dataOut.nProfiles = self.nProfiles
959 dataOut.nProfiles = self.nProfiles
961 dataOut.profileIndex = self.profileIndex
960 dataOut.profileIndex = self.profileIndex
962 dataOut.flagNoData = False
961 dataOut.flagNoData = False
963
962
964 self.incProfileIndex()
963 self.incProfileIndex()
965
964
966 break
965 break
967
966
968 return dataOut
967 return dataOut
969
968
970
969
971 if beam != None: #beam is only for AMISR data
970 if beam != None: #beam is only for AMISR data
972 if self.isThisProfileInList(dataOut.profileIndex, dataOut.beamRangeDict[beam]):
971 if self.isThisProfileInList(dataOut.profileIndex, dataOut.beamRangeDict[beam]):
973 dataOut.flagNoData = False
972 dataOut.flagNoData = False
974 dataOut.profileIndex = self.profileIndex
973 dataOut.profileIndex = self.profileIndex
975
974
976 self.incProfileIndex()
975 self.incProfileIndex()
977
976
978 return dataOut
977 return dataOut
979
978
980 raise ValueError("ProfileSelector needs profileList, profileRangeList or rangeList parameter")
979 raise ValueError("ProfileSelector needs profileList, profileRangeList or rangeList parameter")
981
980
982 #return False
981 #return False
983 return dataOut
982 return dataOut
984
983
985 class Reshaper(Operation):
984 class Reshaper(Operation):
986
985
987 def __init__(self, **kwargs):
986 def __init__(self, **kwargs):
988
987
989 Operation.__init__(self, **kwargs)
988 Operation.__init__(self, **kwargs)
990
989
991 self.__buffer = None
990 self.__buffer = None
992 self.__nitems = 0
991 self.__nitems = 0
993
992
994 def __appendProfile(self, dataOut, nTxs):
993 def __appendProfile(self, dataOut, nTxs):
995
994
996 if self.__buffer is None:
995 if self.__buffer is None:
997 shape = (dataOut.nChannels, int(dataOut.nHeights/nTxs) )
996 shape = (dataOut.nChannels, int(dataOut.nHeights/nTxs) )
998 self.__buffer = numpy.empty(shape, dtype = dataOut.data.dtype)
997 self.__buffer = numpy.empty(shape, dtype = dataOut.data.dtype)
999
998
1000 ini = dataOut.nHeights * self.__nitems
999 ini = dataOut.nHeights * self.__nitems
1001 end = ini + dataOut.nHeights
1000 end = ini + dataOut.nHeights
1002
1001
1003 self.__buffer[:, ini:end] = dataOut.data
1002 self.__buffer[:, ini:end] = dataOut.data
1004
1003
1005 self.__nitems += 1
1004 self.__nitems += 1
1006
1005
1007 return int(self.__nitems*nTxs)
1006 return int(self.__nitems*nTxs)
1008
1007
1009 def __getBuffer(self):
1008 def __getBuffer(self):
1010
1009
1011 if self.__nitems == int(1./self.__nTxs):
1010 if self.__nitems == int(1./self.__nTxs):
1012
1011
1013 self.__nitems = 0
1012 self.__nitems = 0
1014
1013
1015 return self.__buffer.copy()
1014 return self.__buffer.copy()
1016
1015
1017 return None
1016 return None
1018
1017
1019 def __checkInputs(self, dataOut, shape, nTxs):
1018 def __checkInputs(self, dataOut, shape, nTxs):
1020
1019
1021 if shape is None and nTxs is None:
1020 if shape is None and nTxs is None:
1022 raise ValueError("Reshaper: shape of factor should be defined")
1021 raise ValueError("Reshaper: shape of factor should be defined")
1023
1022
1024 if nTxs:
1023 if nTxs:
1025 if nTxs < 0:
1024 if nTxs < 0:
1026 raise ValueError("nTxs should be greater than 0")
1025 raise ValueError("nTxs should be greater than 0")
1027
1026
1028 if nTxs < 1 and dataOut.nProfiles % (1./nTxs) != 0:
1027 if nTxs < 1 and dataOut.nProfiles % (1./nTxs) != 0:
1029 raise ValueError("nProfiles= %d is not divisibled by (1./nTxs) = %f" %(dataOut.nProfiles, (1./nTxs)))
1028 raise ValueError("nProfiles= %d is not divisibled by (1./nTxs) = %f" %(dataOut.nProfiles, (1./nTxs)))
1030
1029
1031 shape = [dataOut.nChannels, dataOut.nProfiles*nTxs, dataOut.nHeights/nTxs]
1030 shape = [dataOut.nChannels, dataOut.nProfiles*nTxs, dataOut.nHeights/nTxs]
1032
1031
1033 return shape, nTxs
1032 return shape, nTxs
1034
1033
1035 if len(shape) != 2 and len(shape) != 3:
1034 if len(shape) != 2 and len(shape) != 3:
1036 raise ValueError("shape dimension should be equal to 2 or 3. shape = (nProfiles, nHeis) or (nChannels, nProfiles, nHeis). Actually shape = (%d, %d, %d)" %(dataOut.nChannels, dataOut.nProfiles, dataOut.nHeights))
1035 raise ValueError("shape dimension should be equal to 2 or 3. shape = (nProfiles, nHeis) or (nChannels, nProfiles, nHeis). Actually shape = (%d, %d, %d)" %(dataOut.nChannels, dataOut.nProfiles, dataOut.nHeights))
1037
1036
1038 if len(shape) == 2:
1037 if len(shape) == 2:
1039 shape_tuple = [dataOut.nChannels]
1038 shape_tuple = [dataOut.nChannels]
1040 shape_tuple.extend(shape)
1039 shape_tuple.extend(shape)
1041 else:
1040 else:
1042 shape_tuple = list(shape)
1041 shape_tuple = list(shape)
1043
1042
1044 nTxs = 1.0*shape_tuple[1]/dataOut.nProfiles
1043 nTxs = 1.0*shape_tuple[1]/dataOut.nProfiles
1045
1044
1046 return shape_tuple, nTxs
1045 return shape_tuple, nTxs
1047
1046
1048 def run(self, dataOut, shape=None, nTxs=None):
1047 def run(self, dataOut, shape=None, nTxs=None):
1049
1048
1050 shape_tuple, self.__nTxs = self.__checkInputs(dataOut, shape, nTxs)
1049 shape_tuple, self.__nTxs = self.__checkInputs(dataOut, shape, nTxs)
1051
1050
1052 dataOut.flagNoData = True
1051 dataOut.flagNoData = True
1053 profileIndex = None
1052 profileIndex = None
1054
1053
1055 if dataOut.flagDataAsBlock:
1054 if dataOut.flagDataAsBlock:
1056
1055
1057 dataOut.data = numpy.reshape(dataOut.data, shape_tuple)
1056 dataOut.data = numpy.reshape(dataOut.data, shape_tuple)
1058 dataOut.flagNoData = False
1057 dataOut.flagNoData = False
1059
1058
1060 profileIndex = int(dataOut.nProfiles*self.__nTxs) - 1
1059 profileIndex = int(dataOut.nProfiles*self.__nTxs) - 1
1061
1060
1062 else:
1061 else:
1063
1062
1064 if self.__nTxs < 1:
1063 if self.__nTxs < 1:
1065
1064
1066 self.__appendProfile(dataOut, self.__nTxs)
1065 self.__appendProfile(dataOut, self.__nTxs)
1067 new_data = self.__getBuffer()
1066 new_data = self.__getBuffer()
1068
1067
1069 if new_data is not None:
1068 if new_data is not None:
1070 dataOut.data = new_data
1069 dataOut.data = new_data
1071 dataOut.flagNoData = False
1070 dataOut.flagNoData = False
1072
1071
1073 profileIndex = dataOut.profileIndex*nTxs
1072 profileIndex = dataOut.profileIndex*nTxs
1074
1073
1075 else:
1074 else:
1076 raise ValueError("nTxs should be greater than 0 and lower than 1, or use VoltageReader(..., getblock=True)")
1075 raise ValueError("nTxs should be greater than 0 and lower than 1, or use VoltageReader(..., getblock=True)")
1077
1076
1078 deltaHeight = dataOut.heightList[1] - dataOut.heightList[0]
1077 deltaHeight = dataOut.heightList[1] - dataOut.heightList[0]
1079
1078
1080 dataOut.heightList = numpy.arange(dataOut.nHeights/self.__nTxs) * deltaHeight + dataOut.heightList[0]
1079 dataOut.heightList = numpy.arange(dataOut.nHeights/self.__nTxs) * deltaHeight + dataOut.heightList[0]
1081
1080
1082 dataOut.nProfiles = int(dataOut.nProfiles*self.__nTxs)
1081 dataOut.nProfiles = int(dataOut.nProfiles*self.__nTxs)
1083
1082
1084 dataOut.profileIndex = profileIndex
1083 dataOut.profileIndex = profileIndex
1085
1084
1086 dataOut.ippSeconds /= self.__nTxs
1085 dataOut.ippSeconds /= self.__nTxs
1087
1086
1088 return dataOut
1087 return dataOut
1089
1088
1090 class SplitProfiles(Operation):
1089 class SplitProfiles(Operation):
1091
1090
1092 def __init__(self, **kwargs):
1091 def __init__(self, **kwargs):
1093
1092
1094 Operation.__init__(self, **kwargs)
1093 Operation.__init__(self, **kwargs)
1095
1094
1096 def run(self, dataOut, n):
1095 def run(self, dataOut, n):
1097
1096
1098 dataOut.flagNoData = True
1097 dataOut.flagNoData = True
1099 profileIndex = None
1098 profileIndex = None
1100
1099
1101 if dataOut.flagDataAsBlock:
1100 if dataOut.flagDataAsBlock:
1102
1101
1103 #nchannels, nprofiles, nsamples
1102 #nchannels, nprofiles, nsamples
1104 shape = dataOut.data.shape
1103 shape = dataOut.data.shape
1105
1104
1106 if shape[2] % n != 0:
1105 if shape[2] % n != 0:
1107 raise ValueError("Could not split the data, n=%d has to be multiple of %d" %(n, shape[2]))
1106 raise ValueError("Could not split the data, n=%d has to be multiple of %d" %(n, shape[2]))
1108
1107
1109 new_shape = shape[0], shape[1]*n, int(shape[2]/n)
1108 new_shape = shape[0], shape[1]*n, int(shape[2]/n)
1110
1109
1111 dataOut.data = numpy.reshape(dataOut.data, new_shape)
1110 dataOut.data = numpy.reshape(dataOut.data, new_shape)
1112 dataOut.flagNoData = False
1111 dataOut.flagNoData = False
1113
1112
1114 profileIndex = int(dataOut.nProfiles/n) - 1
1113 profileIndex = int(dataOut.nProfiles/n) - 1
1115
1114
1116 else:
1115 else:
1117
1116
1118 raise ValueError("Could not split the data when is read Profile by Profile. Use VoltageReader(..., getblock=True)")
1117 raise ValueError("Could not split the data when is read Profile by Profile. Use VoltageReader(..., getblock=True)")
1119
1118
1120 deltaHeight = dataOut.heightList[1] - dataOut.heightList[0]
1119 deltaHeight = dataOut.heightList[1] - dataOut.heightList[0]
1121
1120
1122 dataOut.heightList = numpy.arange(dataOut.nHeights/n) * deltaHeight + dataOut.heightList[0]
1121 dataOut.heightList = numpy.arange(dataOut.nHeights/n) * deltaHeight + dataOut.heightList[0]
1123
1122
1124 dataOut.nProfiles = int(dataOut.nProfiles*n)
1123 dataOut.nProfiles = int(dataOut.nProfiles*n)
1125
1124
1126 dataOut.profileIndex = profileIndex
1125 dataOut.profileIndex = profileIndex
1127
1126
1128 dataOut.ippSeconds /= n
1127 dataOut.ippSeconds /= n
1129
1128
1130 return dataOut
1129 return dataOut
1131
1130
1132 class CombineProfiles(Operation):
1131 class CombineProfiles(Operation):
1133 def __init__(self, **kwargs):
1132 def __init__(self, **kwargs):
1134
1133
1135 Operation.__init__(self, **kwargs)
1134 Operation.__init__(self, **kwargs)
1136
1135
1137 self.__remData = None
1136 self.__remData = None
1138 self.__profileIndex = 0
1137 self.__profileIndex = 0
1139
1138
1140 def run(self, dataOut, n):
1139 def run(self, dataOut, n):
1141
1140
1142 dataOut.flagNoData = True
1141 dataOut.flagNoData = True
1143 profileIndex = None
1142 profileIndex = None
1144
1143
1145 if dataOut.flagDataAsBlock:
1144 if dataOut.flagDataAsBlock:
1146
1145
1147 #nchannels, nprofiles, nsamples
1146 #nchannels, nprofiles, nsamples
1148 shape = dataOut.data.shape
1147 shape = dataOut.data.shape
1149 new_shape = shape[0], shape[1]/n, shape[2]*n
1148 new_shape = shape[0], shape[1]/n, shape[2]*n
1150
1149
1151 if shape[1] % n != 0:
1150 if shape[1] % n != 0:
1152 raise ValueError("Could not split the data, n=%d has to be multiple of %d" %(n, shape[1]))
1151 raise ValueError("Could not split the data, n=%d has to be multiple of %d" %(n, shape[1]))
1153
1152
1154 dataOut.data = numpy.reshape(dataOut.data, new_shape)
1153 dataOut.data = numpy.reshape(dataOut.data, new_shape)
1155 dataOut.flagNoData = False
1154 dataOut.flagNoData = False
1156
1155
1157 profileIndex = int(dataOut.nProfiles*n) - 1
1156 profileIndex = int(dataOut.nProfiles*n) - 1
1158
1157
1159 else:
1158 else:
1160
1159
1161 #nchannels, nsamples
1160 #nchannels, nsamples
1162 if self.__remData is None:
1161 if self.__remData is None:
1163 newData = dataOut.data
1162 newData = dataOut.data
1164 else:
1163 else:
1165 newData = numpy.concatenate((self.__remData, dataOut.data), axis=1)
1164 newData = numpy.concatenate((self.__remData, dataOut.data), axis=1)
1166
1165
1167 self.__profileIndex += 1
1166 self.__profileIndex += 1
1168
1167
1169 if self.__profileIndex < n:
1168 if self.__profileIndex < n:
1170 self.__remData = newData
1169 self.__remData = newData
1171 #continue
1170 #continue
1172 return
1171 return
1173
1172
1174 self.__profileIndex = 0
1173 self.__profileIndex = 0
1175 self.__remData = None
1174 self.__remData = None
1176
1175
1177 dataOut.data = newData
1176 dataOut.data = newData
1178 dataOut.flagNoData = False
1177 dataOut.flagNoData = False
1179
1178
1180 profileIndex = dataOut.profileIndex/n
1179 profileIndex = dataOut.profileIndex/n
1181
1180
1182
1181
1183 deltaHeight = dataOut.heightList[1] - dataOut.heightList[0]
1182 deltaHeight = dataOut.heightList[1] - dataOut.heightList[0]
1184
1183
1185 dataOut.heightList = numpy.arange(dataOut.nHeights*n) * deltaHeight + dataOut.heightList[0]
1184 dataOut.heightList = numpy.arange(dataOut.nHeights*n) * deltaHeight + dataOut.heightList[0]
1186
1185
1187 dataOut.nProfiles = int(dataOut.nProfiles/n)
1186 dataOut.nProfiles = int(dataOut.nProfiles/n)
1188
1187
1189 dataOut.profileIndex = profileIndex
1188 dataOut.profileIndex = profileIndex
1190
1189
1191 dataOut.ippSeconds *= n
1190 dataOut.ippSeconds *= n
1192
1191
1193 return dataOut
1192 return dataOut
1194 # import collections
1193 # import collections
1195 # from scipy.stats import mode
1194 # from scipy.stats import mode
1196 #
1195 #
1197 # class Synchronize(Operation):
1196 # class Synchronize(Operation):
1198 #
1197 #
1199 # isConfig = False
1198 # isConfig = False
1200 # __profIndex = 0
1199 # __profIndex = 0
1201 #
1200 #
1202 # def __init__(self, **kwargs):
1201 # def __init__(self, **kwargs):
1203 #
1202 #
1204 # Operation.__init__(self, **kwargs)
1203 # Operation.__init__(self, **kwargs)
1205 # # self.isConfig = False
1204 # # self.isConfig = False
1206 # self.__powBuffer = None
1205 # self.__powBuffer = None
1207 # self.__startIndex = 0
1206 # self.__startIndex = 0
1208 # self.__pulseFound = False
1207 # self.__pulseFound = False
1209 #
1208 #
1210 # def __findTxPulse(self, dataOut, channel=0, pulse_with = None):
1209 # def __findTxPulse(self, dataOut, channel=0, pulse_with = None):
1211 #
1210 #
1212 # #Read data
1211 # #Read data
1213 #
1212 #
1214 # powerdB = dataOut.getPower(channel = channel)
1213 # powerdB = dataOut.getPower(channel = channel)
1215 # noisedB = dataOut.getNoise(channel = channel)[0]
1214 # noisedB = dataOut.getNoise(channel = channel)[0]
1216 #
1215 #
1217 # self.__powBuffer.extend(powerdB.flatten())
1216 # self.__powBuffer.extend(powerdB.flatten())
1218 #
1217 #
1219 # dataArray = numpy.array(self.__powBuffer)
1218 # dataArray = numpy.array(self.__powBuffer)
1220 #
1219 #
1221 # filteredPower = numpy.correlate(dataArray, dataArray[0:self.__nSamples], "same")
1220 # filteredPower = numpy.correlate(dataArray, dataArray[0:self.__nSamples], "same")
1222 #
1221 #
1223 # maxValue = numpy.nanmax(filteredPower)
1222 # maxValue = numpy.nanmax(filteredPower)
1224 #
1223 #
1225 # if maxValue < noisedB + 10:
1224 # if maxValue < noisedB + 10:
1226 # #No se encuentra ningun pulso de transmision
1225 # #No se encuentra ningun pulso de transmision
1227 # return None
1226 # return None
1228 #
1227 #
1229 # maxValuesIndex = numpy.where(filteredPower > maxValue - 0.1*abs(maxValue))[0]
1228 # maxValuesIndex = numpy.where(filteredPower > maxValue - 0.1*abs(maxValue))[0]
1230 #
1229 #
1231 # if len(maxValuesIndex) < 2:
1230 # if len(maxValuesIndex) < 2:
1232 # #Solo se encontro un solo pulso de transmision de un baudio, esperando por el siguiente TX
1231 # #Solo se encontro un solo pulso de transmision de un baudio, esperando por el siguiente TX
1233 # return None
1232 # return None
1234 #
1233 #
1235 # phasedMaxValuesIndex = maxValuesIndex - self.__nSamples
1234 # phasedMaxValuesIndex = maxValuesIndex - self.__nSamples
1236 #
1235 #
1237 # #Seleccionar solo valores con un espaciamiento de nSamples
1236 # #Seleccionar solo valores con un espaciamiento de nSamples
1238 # pulseIndex = numpy.intersect1d(maxValuesIndex, phasedMaxValuesIndex)
1237 # pulseIndex = numpy.intersect1d(maxValuesIndex, phasedMaxValuesIndex)
1239 #
1238 #
1240 # if len(pulseIndex) < 2:
1239 # if len(pulseIndex) < 2:
1241 # #Solo se encontro un pulso de transmision con ancho mayor a 1
1240 # #Solo se encontro un pulso de transmision con ancho mayor a 1
1242 # return None
1241 # return None
1243 #
1242 #
1244 # spacing = pulseIndex[1:] - pulseIndex[:-1]
1243 # spacing = pulseIndex[1:] - pulseIndex[:-1]
1245 #
1244 #
1246 # #remover senales que se distancien menos de 10 unidades o muestras
1245 # #remover senales que se distancien menos de 10 unidades o muestras
1247 # #(No deberian existir IPP menor a 10 unidades)
1246 # #(No deberian existir IPP menor a 10 unidades)
1248 #
1247 #
1249 # realIndex = numpy.where(spacing > 10 )[0]
1248 # realIndex = numpy.where(spacing > 10 )[0]
1250 #
1249 #
1251 # if len(realIndex) < 2:
1250 # if len(realIndex) < 2:
1252 # #Solo se encontro un pulso de transmision con ancho mayor a 1
1251 # #Solo se encontro un pulso de transmision con ancho mayor a 1
1253 # return None
1252 # return None
1254 #
1253 #
1255 # #Eliminar pulsos anchos (deja solo la diferencia entre IPPs)
1254 # #Eliminar pulsos anchos (deja solo la diferencia entre IPPs)
1256 # realPulseIndex = pulseIndex[realIndex]
1255 # realPulseIndex = pulseIndex[realIndex]
1257 #
1256 #
1258 # period = mode(realPulseIndex[1:] - realPulseIndex[:-1])[0][0]
1257 # period = mode(realPulseIndex[1:] - realPulseIndex[:-1])[0][0]
1259 #
1258 #
1260 # print "IPP = %d samples" %period
1259 # print "IPP = %d samples" %period
1261 #
1260 #
1262 # self.__newNSamples = dataOut.nHeights #int(period)
1261 # self.__newNSamples = dataOut.nHeights #int(period)
1263 # self.__startIndex = int(realPulseIndex[0])
1262 # self.__startIndex = int(realPulseIndex[0])
1264 #
1263 #
1265 # return 1
1264 # return 1
1266 #
1265 #
1267 #
1266 #
1268 # def setup(self, nSamples, nChannels, buffer_size = 4):
1267 # def setup(self, nSamples, nChannels, buffer_size = 4):
1269 #
1268 #
1270 # self.__powBuffer = collections.deque(numpy.zeros( buffer_size*nSamples,dtype=numpy.float),
1269 # self.__powBuffer = collections.deque(numpy.zeros( buffer_size*nSamples,dtype=numpy.float),
1271 # maxlen = buffer_size*nSamples)
1270 # maxlen = buffer_size*nSamples)
1272 #
1271 #
1273 # bufferList = []
1272 # bufferList = []
1274 #
1273 #
1275 # for i in range(nChannels):
1274 # for i in range(nChannels):
1276 # bufferByChannel = collections.deque(numpy.zeros( buffer_size*nSamples, dtype=numpy.complex) + numpy.NAN,
1275 # bufferByChannel = collections.deque(numpy.zeros( buffer_size*nSamples, dtype=numpy.complex) + numpy.NAN,
1277 # maxlen = buffer_size*nSamples)
1276 # maxlen = buffer_size*nSamples)
1278 #
1277 #
1279 # bufferList.append(bufferByChannel)
1278 # bufferList.append(bufferByChannel)
1280 #
1279 #
1281 # self.__nSamples = nSamples
1280 # self.__nSamples = nSamples
1282 # self.__nChannels = nChannels
1281 # self.__nChannels = nChannels
1283 # self.__bufferList = bufferList
1282 # self.__bufferList = bufferList
1284 #
1283 #
1285 # def run(self, dataOut, channel = 0):
1284 # def run(self, dataOut, channel = 0):
1286 #
1285 #
1287 # if not self.isConfig:
1286 # if not self.isConfig:
1288 # nSamples = dataOut.nHeights
1287 # nSamples = dataOut.nHeights
1289 # nChannels = dataOut.nChannels
1288 # nChannels = dataOut.nChannels
1290 # self.setup(nSamples, nChannels)
1289 # self.setup(nSamples, nChannels)
1291 # self.isConfig = True
1290 # self.isConfig = True
1292 #
1291 #
1293 # #Append new data to internal buffer
1292 # #Append new data to internal buffer
1294 # for thisChannel in range(self.__nChannels):
1293 # for thisChannel in range(self.__nChannels):
1295 # bufferByChannel = self.__bufferList[thisChannel]
1294 # bufferByChannel = self.__bufferList[thisChannel]
1296 # bufferByChannel.extend(dataOut.data[thisChannel])
1295 # bufferByChannel.extend(dataOut.data[thisChannel])
1297 #
1296 #
1298 # if self.__pulseFound:
1297 # if self.__pulseFound:
1299 # self.__startIndex -= self.__nSamples
1298 # self.__startIndex -= self.__nSamples
1300 #
1299 #
1301 # #Finding Tx Pulse
1300 # #Finding Tx Pulse
1302 # if not self.__pulseFound:
1301 # if not self.__pulseFound:
1303 # indexFound = self.__findTxPulse(dataOut, channel)
1302 # indexFound = self.__findTxPulse(dataOut, channel)
1304 #
1303 #
1305 # if indexFound == None:
1304 # if indexFound == None:
1306 # dataOut.flagNoData = True
1305 # dataOut.flagNoData = True
1307 # return
1306 # return
1308 #
1307 #
1309 # self.__arrayBuffer = numpy.zeros((self.__nChannels, self.__newNSamples), dtype = numpy.complex)
1308 # self.__arrayBuffer = numpy.zeros((self.__nChannels, self.__newNSamples), dtype = numpy.complex)
1310 # self.__pulseFound = True
1309 # self.__pulseFound = True
1311 # self.__startIndex = indexFound
1310 # self.__startIndex = indexFound
1312 #
1311 #
1313 # #If pulse was found ...
1312 # #If pulse was found ...
1314 # for thisChannel in range(self.__nChannels):
1313 # for thisChannel in range(self.__nChannels):
1315 # bufferByChannel = self.__bufferList[thisChannel]
1314 # bufferByChannel = self.__bufferList[thisChannel]
1316 # #print self.__startIndex
1315 # #print self.__startIndex
1317 # x = numpy.array(bufferByChannel)
1316 # x = numpy.array(bufferByChannel)
1318 # self.__arrayBuffer[thisChannel] = x[self.__startIndex:self.__startIndex+self.__newNSamples]
1317 # self.__arrayBuffer[thisChannel] = x[self.__startIndex:self.__startIndex+self.__newNSamples]
1319 #
1318 #
1320 # deltaHeight = dataOut.heightList[1] - dataOut.heightList[0]
1319 # deltaHeight = dataOut.heightList[1] - dataOut.heightList[0]
1321 # dataOut.heightList = numpy.arange(self.__newNSamples)*deltaHeight
1320 # dataOut.heightList = numpy.arange(self.__newNSamples)*deltaHeight
1322 # # dataOut.ippSeconds = (self.__newNSamples / deltaHeight)/1e6
1321 # # dataOut.ippSeconds = (self.__newNSamples / deltaHeight)/1e6
1323 #
1322 #
1324 # dataOut.data = self.__arrayBuffer
1323 # dataOut.data = self.__arrayBuffer
1325 #
1324 #
1326 # self.__startIndex += self.__newNSamples
1325 # self.__startIndex += self.__newNSamples
1327 #
1326 #
1328 # return
1327 # return
1 NO CONTENT: modified file
NO CONTENT: modified file
1 NO CONTENT: modified file
NO CONTENT: modified file
General Comments 0
You need to be logged in to leave comments. Login now