|
|
'''
|
|
|
Created on Feb 7, 2012
|
|
|
|
|
|
@author $Author$
|
|
|
@version $Id$
|
|
|
'''
|
|
|
|
|
|
import os, sys
|
|
|
import numpy
|
|
|
|
|
|
path = os.path.split(os.getcwd())[0]
|
|
|
sys.path.append(path)
|
|
|
|
|
|
from Model.Correlation import Correlation
|
|
|
from IO.CorrelationIO import CorrelationWriter
|
|
|
#from Graphics.CorrelationPlot import Correlator
|
|
|
|
|
|
|
|
|
from Model.Voltage import Voltage
|
|
|
from Model.Spectra import Spectra
|
|
|
|
|
|
class CorrelationProcessor:
|
|
|
'''
|
|
|
classdocs
|
|
|
'''
|
|
|
|
|
|
integratorIndex = None
|
|
|
writerIndex = None
|
|
|
plotterIndex = None
|
|
|
|
|
|
lagsList = None
|
|
|
|
|
|
nLags = None
|
|
|
tauList = None
|
|
|
pairList = None
|
|
|
indexTau = None
|
|
|
|
|
|
|
|
|
def __init__(self,dataInObj, dataOutObj=None):
|
|
|
'''
|
|
|
Constructor
|
|
|
'''
|
|
|
self.dataInObj = dataInObj
|
|
|
|
|
|
if dataOutObj == None:
|
|
|
self.dataOutObj = Correlation()
|
|
|
else:
|
|
|
self.dataOutObj = dataOutObj
|
|
|
|
|
|
self.indexTau = 0
|
|
|
self.buffer = None
|
|
|
|
|
|
def init(self,pairList=None,tauList=None):
|
|
|
|
|
|
self.integratorIndex = 0
|
|
|
self.writerIndex = 0
|
|
|
self.plotterIndex = 0
|
|
|
|
|
|
self.pairList = pairList
|
|
|
self.tauList = tauList
|
|
|
|
|
|
if ( isinstance(self.dataInObj, Voltage) ):
|
|
|
self.__getCorrelation()
|
|
|
|
|
|
if ( isinstance(self.dataInObj, Spectra) ):
|
|
|
sys.exit(0)
|
|
|
|
|
|
if ( isinstance(self.dataInObj, Correlation) ):
|
|
|
sel.__getCopy()
|
|
|
|
|
|
def __getCorrelation(self):
|
|
|
if self.dataInObj.flagNoData:
|
|
|
return 0
|
|
|
|
|
|
if self.tauList == None: # se lee el tauList desde el archivo
|
|
|
flip = None
|
|
|
if self.dataInObj.m_RadarControllerHeader.flip1 != None:
|
|
|
flip = self.dataInObj.m_RadarControllerHeader.flip1
|
|
|
|
|
|
if self.dataInObj.m_RadarControllerHeader.flip2 != None:
|
|
|
flip = self.dataInObj.m_RadarControllerHeader.flip2
|
|
|
|
|
|
if flip == None:
|
|
|
flip = 2
|
|
|
print 'flip is None --> flip = %d '%flip
|
|
|
|
|
|
ntaus = self.dataInObj.m_RadarControllerHeader.numTaus
|
|
|
taus = self.dataInObj.m_RadarControllerHeader.Taus.reshape(ntaus/flip,flip)
|
|
|
|
|
|
index = 0
|
|
|
self.tauList = taus[:,index]
|
|
|
print 'tauList is None --> tauList = obj.m_RadarControllerHeader.Taus[:,%d]'%index
|
|
|
|
|
|
self.nLags = len(self.tauList)
|
|
|
|
|
|
if self.pairList == None:
|
|
|
self.pairList = [(0,0)] # por defecto calcula la AutoCorrelacion de una canal
|
|
|
|
|
|
self.dataOutObj.tauList = self.tauList
|
|
|
self.dataOutObj.nLags = self.nLags
|
|
|
self.dataOutObj.pairList = self.pairList
|
|
|
|
|
|
if self.buffer == None:
|
|
|
nhei = self.dataInObj.nHeights
|
|
|
npairList = len(self.pairList)
|
|
|
self.buffer = numpy.zeros((self.nLags,nhei,npairList),dtype='complex')
|
|
|
|
|
|
bufferZ = numpy.zeros((npairList,self.dataInObj.nHeights),dtype='complex')
|
|
|
|
|
|
indexHeight = self.tauList[self.indexTau] / self.dataInObj.m_ProcessingHeader.deltaHeight
|
|
|
|
|
|
countPair = 0
|
|
|
|
|
|
# make (signalA*signalB'), where signalA: channel without delay, signalB: channel with delay,
|
|
|
for pair in self.pairList:
|
|
|
bufferZ[countPair,0:self.dataInObj.nHeights-indexHeight] = self.dataInObj.data[pair[1],indexHeight:self.dataInObj.nHeights]
|
|
|
signalA = self.dataInObj.data[pair[0],:]
|
|
|
signalB = bufferZ[countPair,:]
|
|
|
data = signalA * numpy.conjugate(signalB)
|
|
|
self.buffer[self.indexTau,:,countPair] = data
|
|
|
countPair += 1
|
|
|
|
|
|
# change index Tau and lagCounter
|
|
|
self.indexTau += 1
|
|
|
if self.indexTau >= self.nLags:
|
|
|
self.indexTau = 0
|
|
|
self.dataOutObj.data = self.buffer
|
|
|
self.buffer = None
|
|
|
self.dataOutObj.flagNoData = False
|
|
|
else:
|
|
|
self.dataOutObj.flagNoData = True
|
|
|
|
|
|
|
|
|
def addIntegrator(self):
|
|
|
pass
|
|
|
|
|
|
def addWriter(self):
|
|
|
pass
|
|
|
|
|
|
def addPlotter(self):
|
|
|
pass
|
|
|
|
|
|
class Integrator():
|
|
|
def __init__(self):
|
|
|
pass
|
|
|
|
|
|
|
