pxproc_parameters.py
64 lines
| 1.6 KiB
| text/x-python
|
PythonLexer
| r1140 | ''' | |||
| Created on Oct 24, 2016 | ||||
| @author: roj- LouVD | ||||
| ''' | ||||
| import numpy | ||||
| import datetime | ||||
| import time | ||||
| from time import gmtime | ||||
| from numpy import transpose | ||||
George Yong
|
r1167 | from .jroproc_base import ProcessingUnit, Operation | ||
| r1140 | from schainpy.model.data.jrodata import Parameters | |||
| class PXParametersProc(ProcessingUnit): | ||||
| ''' | ||||
| Processing unit for PX parameters data | ||||
| ''' | ||||
| def __init__(self, **kwargs): | ||||
| """ | ||||
| Inputs: None | ||||
| """ | ||||
| ProcessingUnit.__init__(self, **kwargs) | ||||
| self.dataOut = Parameters() | ||||
| self.isConfig = False | ||||
| def setup(self, mode): | ||||
| """ | ||||
| """ | ||||
| self.dataOut.mode = mode | ||||
| def run(self, mode): | ||||
| """ | ||||
| Args: | ||||
| mode (str): select independent variable 'E' for elevation or 'A' for azimuth | ||||
| """ | ||||
| if not self.isConfig: | ||||
| self.setup(mode) | ||||
| self.isConfig = True | ||||
| if self.dataIn.type == 'Parameters': | ||||
| self.dataOut.copy(self.dataIn) | ||||
| self.dataOut.data_param = numpy.array([self.dataOut.data[var] for var in self.dataOut.parameters]) | ||||
| self.dataOut.data_param[self.dataOut.data_param == self.dataOut.missing] = numpy.nan | ||||
| if mode.upper()=='E': | ||||
| self.dataOut.heightList = self.dataOut.data['Azimuth'] | ||||
| else: | ||||
| self.dataOut.heightList = self.dataOut.data['Elevation'] | ||||
| r1145 | attrs = ['units', 'elevation', 'azimuth', 'max_range', 'latitude', 'longitude'] | |||
| r1140 | meta = {} | |||
| for attr in attrs: | ||||
| meta[attr] = getattr(self.dataOut, attr) | ||||
| meta['mode'] = mode | ||||
| r1377 | self.dataOut.meta = meta | |||
