SpectraProcessor.py
858 lines
| 29.2 KiB
| text/x-python
|
PythonLexer
Daniel Valdez
|
r124 | ''' | |
|
Daniel Valdez
|
r126 | $Author$ | |
| $Id$ | |||
|
Daniel Valdez
|
r124 | ''' | |
|
Daniel Valdez
|
r126 | ||
|
Daniel Valdez
|
r124 | import os, sys | |
| import numpy | |||
| import time | |||
|
Daniel Valdez
|
r141 | import datetime | |
|
Daniel Valdez
|
r124 | path = os.path.split(os.getcwd())[0] | |
| sys.path.append(path) | |||
Miguel Valdez
|
r153 | from Data.JROData import Spectra, SpectraHeis | |
Alexander Valdez
|
r171 | from IO.SpectraIO import SpectraWriter,SpectraHeisWriter | |
|
Daniel Valdez
|
r162 | from Graphics.schainPlotTypes import ScopeFigure, SpcFigure, RTIFigure | |
|
Alexander Valdez
|
r171 | from Graphics.BaseGraph_mpl import LinearPlot | |
|
Daniel Valdez
|
r124 | #from JRONoise import Noise | |
| class SpectraProcessor: | |||
| ''' | |||
| classdocs | |||
| ''' | |||
| dataInObj = None | |||
| dataOutObj = None | |||
| noiseObj = None | |||
| integratorObjList = [] | |||
| writerObjList = [] | |||
| integratorObjIndex = None | |||
| writerObjIndex = None | |||
| profIndex = 0 # Se emplea cuando el objeto de entrada es un Voltage | |||
|
Daniel Valdez
|
r162 | firstdatatime = None | |
|
Daniel Valdez
|
r124 | ||
| def __init__(self): | |||
| ''' | |||
| Constructor | |||
| ''' | |||
| self.integratorObjIndex = None | |||
| self.writerObjIndex = None | |||
|
Daniel Valdez
|
r141 | self.plotObjIndex = None | |
|
Alexander Valdez
|
r145 | self.integratorOst = [] | |
|
Daniel Valdez
|
r141 | self.plotObjList = [] | |
|
Miguel Valdez
|
r153 | self.noiseObj = [] | |
| self.writerObjList = [] | |||
|
Daniel Valdez
|
r124 | self.buffer = None | |
|
Daniel Valdez
|
r162 | self.firstdatatime = None | |
|
Daniel Valdez
|
r124 | self.profIndex = 0 | |
|
Miguel Valdez
|
r170 | def setup(self, dataInObj=None, dataOutObj=None, nFFTPoints=None, pairsList=None, wavelength=6): | |
|
Daniel Valdez
|
r124 | ||
| if dataInObj == None: | |||
| raise ValueError, "This SpectraProcessor.setup() function needs dataInObj input variable" | |||
| if dataInObj.type == "Voltage": | |||
|
Miguel Valdez
|
r153 | if nFFTPoints == None: | |
|
Daniel Valdez
|
r124 | raise ValueError, "This SpectraProcessor.setup() function needs nFFTPoints input variable" | |
|
Miguel Valdez
|
r153 | ||
| if dataInObj.type == "Spectra": | |||
| if nFFTPoints != None: | |||
| raise ValueError, "The nFFTPoints cannot be selected to this object type" | |||
|
Daniel Valdez
|
r124 | nFFTPoints = dataInObj.nFFTPoints | |
|
Miguel Valdez
|
r153 | ||
|
Daniel Valdez
|
r156 | if pairsList == None: | |
| pairsList = dataInObj.pairsList | |||
|
Miguel Valdez
|
r153 | ||
|
Daniel Valdez
|
r156 | if pairsList == None: | |
|
Miguel Valdez
|
r153 | nPairs = 0 | |
| else: | |||
|
Daniel Valdez
|
r156 | nPairs = len(pairsList) | |
|
Daniel Valdez
|
r124 | ||
| self.dataInObj = dataInObj | |||
| if dataOutObj == None: | |||
| dataOutObj = Spectra() | |||
| self.dataOutObj = dataOutObj | |||
|
Miguel Valdez
|
r153 | self.dataOutObj.nFFTPoints = nFFTPoints | |
|
Daniel Valdez
|
r156 | self.dataOutObj.pairsList = pairsList | |
|
Miguel Valdez
|
r153 | self.dataOutObj.nPairs = nPairs | |
|
Miguel Valdez
|
r170 | self.dataOutObj.wavelength = wavelength | |
|
Daniel Valdez
|
r124 | ||
| return self.dataOutObj | |||
| def init(self): | |||
|
Miguel Valdez
|
r170 | """ | |
| Este metodo reinicia los contadores de los objetos integrator, writer y plotter | |||
| y actualiza los parametros del objeto de salida dataOutObj a partir del objeto de entrada. | |||
| """ | |||
|
Miguel Valdez
|
r153 | self.dataOutObj.flagNoData = True | |
| if self.dataInObj.flagNoData: | |||
| return 0 | |||
|
Daniel Valdez
|
r124 | self.integratorObjIndex = 0 | |
| self.writerObjIndex = 0 | |||
|
Daniel Valdez
|
r141 | self.plotObjIndex = 0 | |
|
Miguel Valdez
|
r153 | ||
| if self.dataInObj.type == "Spectra": | |||
| self.dataOutObj.copy(self.dataInObj) | |||
| self.dataOutObj.flagNoData = False | |||
| return | |||
|
Daniel Valdez
|
r124 | if self.dataInObj.type == "Voltage": | |
| if self.buffer == None: | |||
|
Miguel Valdez
|
r153 | self.buffer = numpy.zeros((self.dataInObj.nChannels, | |
| self.dataOutObj.nFFTPoints, | |||
|
Daniel Valdez
|
r124 | self.dataInObj.nHeights), | |
| dtype='complex') | |||
| self.buffer[:,self.profIndex,:] = self.dataInObj.data | |||
| self.profIndex += 1 | |||
|
Daniel Valdez
|
r162 | if self.firstdatatime == None: | |
| self.firstdatatime = self.dataInObj.utctime | |||
|
Miguel Valdez
|
r153 | if self.profIndex == self.dataOutObj.nFFTPoints: | |
| self.__updateObjFromInput() | |||
|
Daniel Valdez
|
r124 | self.__getFft() | |
|
Miguel Valdez
|
r153 | ||
|
Daniel Valdez
|
r124 | self.dataOutObj.flagNoData = False | |
| self.buffer = None | |||
|
Daniel Valdez
|
r162 | self.firstdatatime = None | |
|
Daniel Valdez
|
r124 | self.profIndex = 0 | |
| return | |||
|
Miguel Valdez
|
r153 | #Other kind of data | |
| raise ValueError, "The type object %(s) is not valid " %(self.dataOutObj.type) | |||
|
Daniel Valdez
|
r124 | ||
| def __getFft(self): | |||
| """ | |||
| Convierte valores de Voltaje a Spectra | |||
| Affected: | |||
| self.dataOutObj.data_spc | |||
| self.dataOutObj.data_cspc | |||
| self.dataOutObj.data_dc | |||
| self.dataOutObj.heightList | |||
| self.dataOutObj.m_BasicHeader | |||
| self.dataOutObj.m_ProcessingHeader | |||
|
Miguel Valdez
|
r148 | self.dataOutObj.radarControllerHeaderObj | |
| self.dataOutObj.systemHeaderObj | |||
|
Daniel Valdez
|
r124 | self.profIndex | |
| self.buffer | |||
| self.dataOutObj.flagNoData | |||
|
Miguel Valdez
|
r148 | self.dataOutObj.dtype | |
|
Daniel Valdez
|
r124 | self.dataOutObj.nPairs | |
| self.dataOutObj.nChannels | |||
| self.dataOutObj.nProfiles | |||
|
Miguel Valdez
|
r148 | self.dataOutObj.systemHeaderObj.numChannels | |
|
Daniel Valdez
|
r124 | self.dataOutObj.m_ProcessingHeader.totalSpectra | |
| self.dataOutObj.m_ProcessingHeader.profilesPerBlock | |||
| self.dataOutObj.m_ProcessingHeader.numHeights | |||
| self.dataOutObj.m_ProcessingHeader.spectraComb | |||
| self.dataOutObj.m_ProcessingHeader.shif_fft | |||
| """ | |||
| fft_volt = numpy.fft.fft(self.buffer,axis=1) | |||
| dc = fft_volt[:,0,:] | |||
| #calculo de self-spectra | |||
| fft_volt = numpy.fft.fftshift(fft_volt,axes=(1,)) | |||
| spc = fft_volt * numpy.conjugate(fft_volt) | |||
| spc = spc.real | |||
| blocksize = 0 | |||
| blocksize += dc.size | |||
| blocksize += spc.size | |||
| cspc = None | |||
| pairIndex = 0 | |||
|
Daniel Valdez
|
r156 | if self.dataOutObj.pairsList != None: | |
|
Daniel Valdez
|
r124 | #calculo de cross-spectra | |
|
Miguel Valdez
|
r153 | cspc = numpy.zeros((self.dataOutObj.nPairs, self.dataOutObj.nFFTPoints, self.dataOutObj.nHeights), dtype='complex') | |
|
Daniel Valdez
|
r156 | for pair in self.dataOutObj.pairsList: | |
|
Daniel Valdez
|
r124 | cspc[pairIndex,:,:] = numpy.abs(fft_volt[pair[0],:,:] * numpy.conjugate(fft_volt[pair[1],:,:])) | |
| pairIndex += 1 | |||
| blocksize += cspc.size | |||
| self.dataOutObj.data_spc = spc | |||
| self.dataOutObj.data_cspc = cspc | |||
| self.dataOutObj.data_dc = dc | |||
|
Miguel Valdez
|
r153 | self.dataOutObj.blockSize = blocksize | |
|
Daniel Valdez
|
r124 | ||
| # self.getNoise() | |||
|
Miguel Valdez
|
r153 | ||
| def __updateObjFromInput(self): | |||
| self.dataOutObj.radarControllerHeaderObj = self.dataInObj.radarControllerHeaderObj.copy() | |||
| self.dataOutObj.systemHeaderObj = self.dataInObj.systemHeaderObj.copy() | |||
| self.dataOutObj.channelList = self.dataInObj.channelList | |||
| self.dataOutObj.heightList = self.dataInObj.heightList | |||
| self.dataOutObj.dtype = self.dataInObj.dtype | |||
| self.dataOutObj.nHeights = self.dataInObj.nHeights | |||
| self.dataOutObj.nChannels = self.dataInObj.nChannels | |||
| self.dataOutObj.nBaud = self.dataInObj.nBaud | |||
| self.dataOutObj.nCode = self.dataInObj.nCode | |||
| self.dataOutObj.code = self.dataInObj.code | |||
| self.dataOutObj.nProfiles = self.dataOutObj.nFFTPoints | |||
| self.dataOutObj.channelIndexList = self.dataInObj.channelIndexList | |||
| self.dataOutObj.flagTimeBlock = self.dataInObj.flagTimeBlock | |||
|
Daniel Valdez
|
r162 | self.dataOutObj.utctime = self.firstdatatime | |
|
Miguel Valdez
|
r153 | self.dataOutObj.flagDecodeData = self.dataInObj.flagDecodeData #asumo q la data esta decodificada | |
| self.dataOutObj.flagDeflipData = self.dataInObj.flagDeflipData #asumo q la data esta sin flip | |||
| self.dataOutObj.flagShiftFFT = self.dataInObj.flagShiftFFT | |||
|
Daniel Valdez
|
r162 | self.dataOutObj.nCohInt = self.dataInObj.nCohInt | |
|
Miguel Valdez
|
r153 | self.dataOutObj.nIncohInt = 1 | |
|
Daniel Valdez
|
r159 | self.dataOutObj.ippSeconds = self.dataInObj.ippSeconds | |
|
Daniel Valdez
|
r162 | self.dataOutObj.timeInterval = self.dataInObj.timeInterval*self.dataOutObj.nFFTPoints | |
|
Daniel Valdez
|
r124 | ||
|
Daniel Valdez
|
r126 | def addWriter(self, wrpath, blocksPerFile): | |
|
Miguel Valdez
|
r153 | ||
|
Daniel Valdez
|
r125 | objWriter = SpectraWriter(self.dataOutObj) | |
|
Daniel Valdez
|
r126 | objWriter.setup(wrpath, blocksPerFile) | |
|
Daniel Valdez
|
r124 | self.writerObjList.append(objWriter) | |
| def addIntegrator(self,N,timeInterval): | |||
| objIncohInt = IncoherentIntegration(N,timeInterval) | |||
| self.integratorObjList.append(objIncohInt) | |||
|
Daniel Valdez
|
r156 | ||
| def addCrossSpc(self, idfigure, nframes, wintitle, driver, colormap, colorbar, showprofile): | |||
| crossSpcObj = CrossSpcFigure(idfigure, nframes, wintitle, driver, colormap, colorbar, showprofile) | |||
| self.plotObjList.append(crossSpcObj) | |||
| def plotCrossSpc(self, idfigure=None, | |||
| xmin=None, | |||
| xmax=None, | |||
| ymin=None, | |||
| ymax=None, | |||
| minvalue=None, | |||
| maxvalue=None, | |||
| wintitle='', | |||
| driver='plplot', | |||
| colormap='br_green', | |||
| colorbar=True, | |||
| showprofile=False, | |||
| save=False, | |||
| gpath=None, | |||
| pairsList = None): | |||
| if self.dataOutObj.flagNoData: | |||
| return 0 | |||
| if pairsList == None: | |||
| pairsList = self.dataOutObj.pairsList | |||
| nframes = len(pairsList) | |||
| x = numpy.arange(self.dataOutObj.nFFTPoints) | |||
| y = self.dataOutObj.heightList | |||
|
Daniel Valdez
|
r162 | data_spc = self.dataOutObj.data_spc | |
| data_cspc = self.dataOutObj.data_cspc | |||
|
Daniel Valdez
|
r156 | ||
|
Daniel Valdez
|
r162 | data = [] | |
|
Daniel Valdez
|
r156 | ||
| if len(self.plotObjList) <= self.plotObjIndex: | |||
| self.addSpc(idfigure, nframes, wintitle, driver, colormap, colorbar, showprofile) | |||
|
Daniel Valdez
|
r162 | ||
|
Daniel Valdez
|
r156 | ||
|
Daniel Valdez
|
r159 | def addRti(self, idfigure, nframes, wintitle, driver, colormap, colorbar, showprofile): | |
| rtiObj = RTIFigure(idfigure, nframes, wintitle, driver, colormap, colorbar, showprofile) | |||
| self.plotObjList.append(rtiObj) | |||
| def plotRti(self, idfigure=None, | |||
| starttime=None, | |||
| endtime=None, | |||
| rangemin=None, | |||
| rangemax=None, | |||
| minvalue=None, | |||
| maxvalue=None, | |||
| wintitle='', | |||
| driver='plplot', | |||
| colormap='br_greeen', | |||
| colorbar=True, | |||
| showprofile=False, | |||
| xrangestep=None, | |||
| save=False, | |||
| gpath=None, | |||
| ratio=1, | |||
| channelList=None): | |||
| if self.dataOutObj.flagNoData: | |||
| return 0 | |||
| if channelList == None: | |||
| channelList = self.dataOutObj.channelList | |||
| nframes = len(channelList) | |||
| if len(self.plotObjList) <= self.plotObjIndex: | |||
| self.addRti(idfigure, nframes, wintitle, driver, colormap, colorbar, showprofile) | |||
| data = 10.*numpy.log10(self.dataOutObj.data_spc[channelList,:,:]) | |||
|
Daniel Valdez
|
r162 | data = numpy.average(data, axis=1) | |
|
Daniel Valdez
|
r159 | currenttime = self.dataOutObj.utctime - time.timezone | |
| range = self.dataOutObj.heightList | |||
| figuretitle = "RTI Plot for Spectra Data" #+ date | |||
| cleardata = False | |||
| deltax = self.dataOutObj.timeInterval | |||
| plotObj = self.plotObjList[self.plotObjIndex] | |||
|
Daniel Valdez
|
r162 | plotObj.plotPcolor(data=data, | |
| x=currenttime, | |||
| y=range, | |||
| channelList=channelList, | |||
| xmin=starttime, | |||
| xmax=endtime, | |||
| ymin=rangemin, | |||
| ymax=rangemax, | |||
| minvalue=minvalue, | |||
| maxvalue=maxvalue, | |||
| figuretitle=figuretitle, | |||
| xrangestep=xrangestep, | |||
| deltax=deltax, | |||
| save=save, | |||
| gpath=gpath, | |||
| ratio=ratio, | |||
| cleardata=cleardata | |||
| ) | |||
|
Daniel Valdez
|
r159 | ||
| self.plotObjIndex += 1 | |||
|
Daniel Valdez
|
r156 | ||
|
Daniel Valdez
|
r124 | ||
|
Daniel Valdez
|
r141 | def addSpc(self, idfigure, nframes, wintitle, driver, colormap, colorbar, showprofile): | |
|
Miguel Valdez
|
r153 | ||
|
Daniel Valdez
|
r141 | spcObj = SpcFigure(idfigure, nframes, wintitle, driver, colormap, colorbar, showprofile) | |
| self.plotObjList.append(spcObj) | |||
| def plotSpc(self, idfigure=None, | |||
| xmin=None, | |||
| xmax=None, | |||
| ymin=None, | |||
| ymax=None, | |||
| minvalue=None, | |||
| maxvalue=None, | |||
| wintitle='', | |||
| driver='plplot', | |||
|
Miguel Valdez
|
r153 | colormap='br_green', | |
|
Daniel Valdez
|
r141 | colorbar=True, | |
| showprofile=False, | |||
| save=False, | |||
|
Miguel Valdez
|
r153 | gpath=None, | |
|
Daniel Valdez
|
r158 | ratio=1, | |
| channelList=None): | |||
|
Daniel Valdez
|
r141 | ||
| if self.dataOutObj.flagNoData: | |||
| return 0 | |||
|
Miguel Valdez
|
r153 | if channelList == None: | |
| channelList = self.dataOutObj.channelList | |||
| nframes = len(channelList) | |||
|
Daniel Valdez
|
r141 | ||
| if len(self.plotObjList) <= self.plotObjIndex: | |||
| self.addSpc(idfigure, nframes, wintitle, driver, colormap, colorbar, showprofile) | |||
| x = numpy.arange(self.dataOutObj.nFFTPoints) | |||
| y = self.dataOutObj.heightList | |||
| data = 10.*numpy.log10(self.dataOutObj.data_spc[channelList,:,:]) | |||
| # noisedB = 10.*numpy.log10(noise) | |||
| noisedB = numpy.arange(len(channelList)+1) | |||
| noisedB = noisedB *1.2 | |||
| titleList = [] | |||
| for i in range(len(noisedB)): | |||
| title = "%.2f"%noisedB[i] | |||
| titleList.append(title) | |||
|
Miguel Valdez
|
r153 | thisdatetime = datetime.datetime.fromtimestamp(self.dataOutObj.utctime) | |
|
Daniel Valdez
|
r141 | dateTime = "%s"%(thisdatetime.strftime("%d-%b-%Y %H:%M:%S")) | |
| figuretitle = "Spc Radar Data: %s"%dateTime | |||
| cleardata = True | |||
| plotObj = self.plotObjList[self.plotObjIndex] | |||
|
Daniel Valdez
|
r156 | plotObj.plotPcolor(data=data, | |
|
Miguel Valdez
|
r154 | x=x, | |
| y=y, | |||
| channelList=channelList, | |||
| xmin=xmin, | |||
| xmax=xmax, | |||
| ymin=ymin, | |||
| ymax=ymax, | |||
| minvalue=minvalue, | |||
| maxvalue=maxvalue, | |||
| figuretitle=figuretitle, | |||
| xrangestep=None, | |||
| deltax=None, | |||
|
Daniel Valdez
|
r156 | save=save, | |
| gpath=gpath, | |||
| cleardata=cleardata | |||
|
Miguel Valdez
|
r154 | ) | |
|
Daniel Valdez
|
r141 | ||
| self.plotObjIndex += 1 | |||
|
Daniel Valdez
|
r126 | def writeData(self, wrpath, blocksPerFile): | |
|
Miguel Valdez
|
r153 | ||
|
Daniel Valdez
|
r124 | if self.dataOutObj.flagNoData: | |
|
Miguel Valdez
|
r153 | return 0 | |
|
Daniel Valdez
|
r124 | ||
| if len(self.writerObjList) <= self.writerObjIndex: | |||
|
Daniel Valdez
|
r126 | self.addWriter(wrpath, blocksPerFile) | |
|
Daniel Valdez
|
r124 | ||
| self.writerObjList[self.writerObjIndex].putData() | |||
| self.writerObjIndex += 1 | |||
| def integrator(self, N=None, timeInterval=None): | |||
| if self.dataOutObj.flagNoData: | |||
| return 0 | |||
| if len(self.integratorObjList) <= self.integratorObjIndex: | |||
| self.addIntegrator(N,timeInterval) | |||
| myIncohIntObj = self.integratorObjList[self.integratorObjIndex] | |||
|
Daniel Valdez
|
r162 | myIncohIntObj.exe(data=self.dataOutObj.data_spc,datatime=self.dataOutObj.utctime) | |
|
Daniel Valdez
|
r124 | ||
| if myIncohIntObj.isReady: | |||
| self.dataOutObj.data_spc = myIncohIntObj.data | |||
|
Daniel Valdez
|
r162 | self.dataOutObj.timeInterval *= myCohIntObj.nIncohInt | |
| self.dataOutObj.nIncohInt = myIncohIntObj.navg * self.dataInObj.nIncohInt | |||
| self.dataOutObj.utctime = myIncohIntObj.firstdatatime | |||
|
Daniel Valdez
|
r124 | self.dataOutObj.flagNoData = False | |
| """Calcular el ruido""" | |||
| self.getNoise() | |||
| else: | |||
| self.dataOutObj.flagNoData = True | |||
| self.integratorObjIndex += 1 | |||
|
Alexander Valdez
|
r145 | ||
| class SpectraHeisProcessor: | |||
|
Miguel Valdez
|
r148 | ||
|
Alexander Valdez
|
r145 | def __init__(self): | |
|
Miguel Valdez
|
r148 | ||
|
Alexander Valdez
|
r145 | self.integratorObjIndex = None | |
| self.writerObjIndex = None | |||
|
Alexander Valdez
|
r151 | self.plotObjIndex = None | |
|
Alexander Valdez
|
r145 | self.integratorObjList = [] | |
| self.writerObjList = [] | |||
|
Alexander Valdez
|
r151 | self.plotObjList = [] | |
|
Alexander Valdez
|
r145 | #self.noiseObj = Noise() | |
|
Miguel Valdez
|
r148 | def setup(self, dataInObj, dataOutObj=None, nFFTPoints=None, pairList=None): | |
|
Alexander Valdez
|
r145 | ||
| if nFFTPoints == None: | |||
|
Miguel Valdez
|
r148 | nFFTPoints = self.dataInObj.nHeights | |
|
Alexander Valdez
|
r145 | ||
| self.dataInObj = dataInObj | |||
| if dataOutObj == None: | |||
| dataOutObj = SpectraHeis() | |||
| self.dataOutObj = dataOutObj | |||
| return self.dataOutObj | |||
| def init(self): | |||
|
Miguel Valdez
|
r153 | ||
| self.dataOutObj.flagNoData = True | |||
| if self.dataInObj.flagNoData: | |||
| return 0 | |||
|
Alexander Valdez
|
r145 | self.integratorObjIndex = 0 | |
| self.writerObjIndex = 0 | |||
|
Alexander Valdez
|
r151 | self.plotObjIndex = 0 | |
|
Daniel Valdez
|
r124 | ||
|
Alexander Valdez
|
r145 | if self.dataInObj.type == "Voltage": | |
|
Miguel Valdez
|
r153 | self.__updateObjFromInput() | |
|
Alexander Valdez
|
r145 | self.__getFft() | |
| self.dataOutObj.flagNoData = False | |||
| return | |||
| #Other kind of data | |||
| if self.dataInObj.type == "SpectraHeis": | |||
| self.dataOutObj.copy(self.dataInObj) | |||
| self.dataOutObj.flagNoData = False | |||
| return | |||
|
Miguel Valdez
|
r148 | raise ValueError, "The type is not valid" | |
|
Alexander Valdez
|
r145 | ||
|
Miguel Valdez
|
r153 | def __updateObjFromInput(self): | |
|
Alexander Valdez
|
r151 | self.dataOutObj.radarControllerHeaderObj = self.dataInObj.radarControllerHeaderObj.copy() | |
| self.dataOutObj.systemHeaderObj = self.dataInObj.systemHeaderObj.copy() | |||
| self.dataOutObj.channelList = self.dataInObj.channelList | |||
| self.dataOutObj.heightList = self.dataInObj.heightList | |||
| self.dataOutObj.dtype = self.dataInObj.dtype | |||
| self.dataOutObj.nHeights = self.dataInObj.nHeights | |||
| self.dataOutObj.nChannels = self.dataInObj.nChannels | |||
| self.dataOutObj.nBaud = self.dataInObj.nBaud | |||
| self.dataOutObj.nCode = self.dataInObj.nCode | |||
| self.dataOutObj.code = self.dataInObj.code | |||
| self.dataOutObj.nProfiles = 1 | |||
| self.dataOutObj.nFFTPoints = self.dataInObj.nHeights | |||
| self.dataOutObj.channelIndexList = self.dataInObj.channelIndexList | |||
| self.dataOutObj.flagNoData = self.dataInObj.flagNoData | |||
| self.dataOutObj.flagTimeBlock = self.dataInObj.flagTimeBlock | |||
|
Miguel Valdez
|
r153 | self.dataOutObj.utctime = self.dataInObj.utctime | |
|
Alexander Valdez
|
r151 | self.dataOutObj.flagDecodeData = self.dataInObj.flagDecodeData #asumo q la data esta decodificada | |
| self.dataOutObj.flagDeflipData = self.dataInObj.flagDeflipData #asumo q la data esta sin flip | |||
| self.dataOutObj.flagShiftFFT = self.dataInObj.flagShiftFFT | |||
| self.dataOutObj.nIncohInt = 1 | |||
|
Alexander Valdez
|
r163 | self.dataOutObj.ippSeconds= self.dataInObj.ippSeconds | |
|
Alexander Valdez
|
r171 | self.dataOutObj.set=self.dataInObj.set | |
| self.dataOutObj.deltaHeight=self.dataInObj.deltaHeight | |||
|
Alexander Valdez
|
r163 | # def addWriter(self,wrpath,blocksPerfile): | |
| def addWriter(self,wrpath): | |||
| objWriter=SpectraHeisWriter(self.dataOutObj) | |||
| objWriter.setup(wrpath) | |||
| #objWriter.setup(wrpath,blocksPerfile) | |||
| self.writerObjList.append(objWriter) | |||
| # def writedata(self,wrpath,blocksPerfile): | |||
| def writedata(self,wrpath): | |||
| if self.dataOutObj.flagNoData: | |||
| return 0 | |||
| if len(self.writerObjList) <= self.writerObjIndex: | |||
| #self.addWriter(wrpath, blocksPerFile) | |||
| self.addWriter(wrpath) | |||
| self.writerObjList[self.writerObjIndex].putData() | |||
| self.writerObjIndex += 1 | |||
|
Alexander Valdez
|
r145 | def __getFft(self): | |
| fft_volt = numpy.fft.fft(self.dataInObj.data, axis=1) | |||
| #print fft_volt | |||
| #calculo de self-spectra | |||
| fft_volt = numpy.fft.fftshift(fft_volt,axes=(1,)) | |||
| spc = numpy.abs(fft_volt * numpy.conjugate(fft_volt)) | |||
| self.dataOutObj.data_spc = spc | |||
| def getSpectra(self): | |||
| return self.dataOutObj.data_spc | |||
| def getFrecuencies(self): | |||
| print self.nFFTPoints | |||
| return numpy.arange(int(self.nFFTPoints)) | |||
| def addIntegrator(self,N,timeInterval): | |||
|
Miguel Valdez
|
r153 | ||
|
Alexander Valdez
|
r145 | objIncohInt = IncoherentIntegration(N,timeInterval) | |
| self.integratorObjList.append(objIncohInt) | |||
| def integrator(self, N=None, timeInterval=None): | |||
|
Miguel Valdez
|
r153 | ||
|
Alexander Valdez
|
r145 | if self.dataOutObj.flagNoData: | |
| return 0 | |||
| if len(self.integratorObjList) <= self.integratorObjIndex: | |||
| self.addIntegrator(N,timeInterval) | |||
| myIncohIntObj = self.integratorObjList[self.integratorObjIndex] | |||
|
Alexander Valdez
|
r171 | myIncohIntObj.exe(data=self.dataOutObj.data_spc,datatime=self.dataOutObj.utctime) | |
|
Alexander Valdez
|
r145 | ||
| if myIncohIntObj.isReady: | |||
| self.dataOutObj.data_spc = myIncohIntObj.data | |||
|
Alexander Valdez
|
r151 | self.dataOutObj.nIncohInt = self.dataOutObj.nIncohInt*myIncohIntObj.navg | |
|
Alexander Valdez
|
r145 | self.dataOutObj.flagNoData = False | |
| #self.getNoise(type="hildebrand",parm=myIncohIntObj.navg) | |||
| # self.getNoise(type="sort", parm=16) | |||
| else: | |||
| self.dataOutObj.flagNoData = True | |||
| self.integratorObjIndex += 1 | |||
|
Daniel Valdez
|
r124 | ||
|
Alexander Valdez
|
r151 | def addScope(self, idfigure, nframes, wintitle, driver): | |
|
Miguel Valdez
|
r153 | ||
|
Alexander Valdez
|
r151 | if idfigure==None: | |
| idfigure = self.plotObjIndex | |||
| scopeObj = ScopeFigure(idfigure, nframes, wintitle, driver) | |||
| self.plotObjList.append(scopeObj) | |||
| def plotScope(self, | |||
| idfigure=None, | |||
| minvalue=None, | |||
| maxvalue=None, | |||
| xmin=None, | |||
| xmax=None, | |||
| wintitle='', | |||
| driver='plplot', | |||
| save=False, | |||
| gpath=None, | |||
| titleList=None, | |||
| xlabelList=None, | |||
| ylabelList=None): | |||
| if self.dataOutObj.flagNoData: | |||
| return 0 | |||
| nframes = len(self.dataOutObj.channelList) | |||
| if len(self.plotObjList) <= self.plotObjIndex: | |||
| self.addScope(idfigure, nframes, wintitle, driver) | |||
| data1D = self.dataOutObj.data_spc | |||
| x = numpy.arange(self.dataOutObj.nHeights) | |||
|
Miguel Valdez
|
r153 | thisDatetime = datetime.datetime.fromtimestamp(self.dataOutObj.utctime) | |
|
Alexander Valdez
|
r151 | ||
| dateTime = "%s"%(thisDatetime.strftime("%d-%b-%Y %H:%M:%S")) | |||
| date = "%s"%(thisDatetime.strftime("%d-%b-%Y")) | |||
| figureTitle = "Scope Plot Radar Data: " + date | |||
| plotObj = self.plotObjList[self.plotObjIndex] | |||
| plotObj.plot1DArray(data1D, | |||
| x, | |||
| self.dataOutObj.channelList, | |||
| xmin, | |||
| xmax, | |||
| minvalue, | |||
| maxvalue, | |||
| figureTitle, | |||
| save, | |||
| gpath) | |||
| self.plotObjIndex += 1 | |||
|
Alexander Valdez
|
r171 | ||
| def addPlotMatScope(self, indexPlot,nsubplot,winTitle): | |||
| linearObj = LinearPlot(indexPlot,nsubplot,winTitle) | |||
| self.plotObjList.append(linearObj) | |||
| def plotMatScope(self, idfigure, | |||
| channelList=None, | |||
| wintitle="", | |||
| save=None, | |||
| gpath=None): | |||
| if self.dataOutObj.flagNoData: | |||
| return 0 | |||
| if channelList == None: | |||
| channelList = self.dataOutObj.channelList | |||
| nchannels = len(channelList) | |||
| thisDatetime = datetime.datetime.fromtimestamp(self.dataOutObj.utctime) | |||
| dateTime = "%s"%(thisDatetime.strftime("%d-%b-%Y %H:%M:%S")) | |||
| winTitle = "Spectra Profile" #+ dateTime | |||
| if len(self.plotObjList) <= self.plotObjIndex: | |||
| self.addPlotMatScope(idfigure,nchannels,winTitle) | |||
| c = 3E8 | |||
| x=numpy.arange(-1*self.dataOutObj.nHeights/2.,self.dataOutObj.nHeights/2.)*(c/(2*self.dataOutObj.deltaHeight*self.dataOutObj.nHeights*1000)) | |||
| x= x/(10000.0) | |||
| data = 10*numpy.log10(self.dataOutObj.data_spc) | |||
| subplotTitle = "Channel No." | |||
| xlabel = "Frecuencias(hz)*10K" | |||
| ylabel = "Potencia(dB)" | |||
| isPlotIni=False | |||
| isPlotConfig=False | |||
| ntimes=2 | |||
| nsubplot= nchannels | |||
| plotObj = self.plotObjList[self.plotObjIndex] | |||
| # for i in range(ntimes): | |||
| #Creacion de Subplots | |||
| if not(plotObj.isPlotIni): | |||
| for index in range(nsubplot): | |||
| indexplot = index + 1 | |||
| title = subplotTitle + '%d'%indexplot | |||
| xmin = numpy.min(x) | |||
| xmax = numpy.max(x) | |||
| ymin = numpy.min(data[index,:]) | |||
| ymax = numpy.max(data[index,:]) | |||
| plotObj.createObjects(indexplot,xmin,xmax,ymin,ymax,title,xlabel,ylabel) | |||
| plotObj.isPlotIni =True | |||
| # Inicializa el grafico en cada itearcion | |||
| plotObj.setFigure(idfigure) | |||
| plotObj.setFigure("") | |||
| for index in range(nsubplot): | |||
| subplot = index | |||
| plotObj.iniPlot(subplot+1) | |||
| #Ploteo de datos | |||
| for channel in range(nsubplot): | |||
| y=data[channel,:] | |||
| plotObj.plot(channel+1,x,y,type='simple') | |||
| plotObj.refresh() | |||
| #time.sleep(0.05) | |||
| # plotObj.show() | |||
| self.plotObjIndex += 1 | |||
| #print "Ploteo Finalizado" | |||
|
Daniel Valdez
|
r124 | ||
|
Alexander Valdez
|
r163 | def rti(self): | |
| if self.dataOutObj.flagNoData: | |||
| return 0 | |||
| data=numpy.average(self.dataOutObj.data_spc,axis=1) | |||
| data[0] | |||
| print data[0] | |||
| x = numpy.arange(100000) | |||
| print "test" | |||
| #print self.dataOutObj.data_spc.average(axis=1) | |||
|
Daniel Valdez
|
r124 | class IncoherentIntegration: | |
| integ_counter = None | |||
| data = None | |||
| navg = None | |||
| buffer = None | |||
| nIncohInt = None | |||
|
Daniel Valdez
|
r162 | firstdatatime = None | |
|
Daniel Valdez
|
r124 | ||
| def __init__(self, N = None, timeInterval = None): | |||
| """ | |||
| N | |||
| timeInterval - interval time [min], integer value | |||
| """ | |||
| self.data = None | |||
| self.navg = None | |||
| self.buffer = None | |||
| self.timeOut = None | |||
| self.exitCondition = False | |||
| self.isReady = False | |||
| self.nIncohInt = N | |||
| self.integ_counter = 0 | |||
|
Daniel Valdez
|
r162 | self.firstdatatime = None | |
|
Daniel Valdez
|
r124 | if timeInterval!=None: | |
| self.timeIntervalInSeconds = timeInterval * 60. #if (type(timeInterval)!=integer) -> change this line | |||
| if ((timeInterval==None) and (N==None)): | |||
| print 'N = None ; timeInterval = None' | |||
| sys.exit(0) | |||
| elif timeInterval == None: | |||
| self.timeFlag = False | |||
| else: | |||
| self.timeFlag = True | |||
|
Daniel Valdez
|
r162 | def exe(self,data,datatime): | |
|
Daniel Valdez
|
r124 | """ | |
| data | |||
|
Daniel Valdez
|
r162 | datatime [seconds] | |
|
Daniel Valdez
|
r124 | """ | |
|
Daniel Valdez
|
r162 | if self.firstdatatime == None or self.isReady: | |
| self.firstdatatime = datatime | |||
|
Daniel Valdez
|
r124 | ||
| if self.timeFlag: | |||
| if self.timeOut == None: | |||
|
Daniel Valdez
|
r162 | self.timeOut = datatime + self.timeIntervalInSeconds | |
|
Daniel Valdez
|
r124 | ||
|
Daniel Valdez
|
r162 | if datatime < self.timeOut: | |
|
Daniel Valdez
|
r124 | if self.buffer == None: | |
| self.buffer = data | |||
| else: | |||
| self.buffer = self.buffer + data | |||
| self.integ_counter += 1 | |||
| else: | |||
| self.exitCondition = True | |||
| else: | |||
| if self.integ_counter < self.nIncohInt: | |||
| if self.buffer == None: | |||
| self.buffer = data | |||
| else: | |||
| self.buffer = self.buffer + data | |||
| self.integ_counter += 1 | |||
| if self.integ_counter == self.nIncohInt: | |||
| self.exitCondition = True | |||
| if self.exitCondition: | |||
| self.data = self.buffer | |||
| self.navg = self.integ_counter | |||
| self.isReady = True | |||
| self.buffer = None | |||
| self.timeOut = None | |||
| self.integ_counter = 0 | |||
| self.exitCondition = False | |||
| if self.timeFlag: | |||
| self.buffer = data | |||
|
Daniel Valdez
|
r162 | self.timeOut = datatime + self.timeIntervalInSeconds | |
|
Daniel Valdez
|
r124 | else: | |
| self.isReady = False | |||