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 = |
|
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(ipp |
|
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 leer |
|
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