|
|
'''
|
|
|
Created on Jul 31, 2012
|
|
|
|
|
|
@author $Author$
|
|
|
@version $Id$
|
|
|
'''
|
|
|
|
|
|
import os, sys
|
|
|
import time, datetime
|
|
|
#import pylab as pl
|
|
|
|
|
|
from Data.JROData import Voltage
|
|
|
from IO.VoltageIO import *
|
|
|
|
|
|
from Data.JROData import SpectraHeis
|
|
|
from IO.SpectraIO import *
|
|
|
|
|
|
from Processing.VoltageProcessor import *
|
|
|
from Processing.SpectraProcessor import *
|
|
|
|
|
|
from Graphics.BaseGraph_mpl import LinearPlot
|
|
|
|
|
|
class TestHeis():
|
|
|
i=None
|
|
|
def __init__(self):
|
|
|
self.setValues()
|
|
|
self.createObjects()
|
|
|
self.testSChain()
|
|
|
self.i=0
|
|
|
|
|
|
def setValues( self ):
|
|
|
|
|
|
self.path="C:\data2"
|
|
|
self.startDate = datetime.date(2012,4,1)
|
|
|
self.endDate = datetime.date(2012,6,30)
|
|
|
|
|
|
self.startTime = datetime.time(0,0,0)
|
|
|
self.endTime = datetime.time(23,0,0)
|
|
|
#self.N = 4
|
|
|
#self.npts = 8
|
|
|
|
|
|
def createObjects( self ):
|
|
|
|
|
|
self.readerObj = VoltageReader()
|
|
|
self.specProcObj = SpectraHeisProcessor()
|
|
|
|
|
|
self.voltObj1 = self.readerObj.setup(
|
|
|
path = self.path,
|
|
|
startDate = self.startDate,
|
|
|
endDate = self.endDate,
|
|
|
startTime = self.startTime,
|
|
|
endTime = self.endTime,
|
|
|
expLabel = '',
|
|
|
online = 0)
|
|
|
|
|
|
if not(self.voltObj1):
|
|
|
sys.exit(0)
|
|
|
|
|
|
print "voltaje o alturas: %03d"%(self.voltObj1.nHeights)
|
|
|
#print "perfiles: %03d"%(self.voltObj1.nProfiles)
|
|
|
self.specObj1 = self.specProcObj.setup(dataInObj = self.voltObj1,nFFTPoints=self.voltObj1.nHeights)
|
|
|
|
|
|
|
|
|
|
|
|
print "nChannels: %03d"%(self.specObj1.nChannels)
|
|
|
print "nHeights: %03d"%( self.specObj1.nHeights)
|
|
|
print "nFFTPoints: %03d"% int(self.specObj1.nFFTPoints)
|
|
|
|
|
|
|
|
|
def testSChain( self ):
|
|
|
|
|
|
ini = time.time()
|
|
|
|
|
|
while(True):
|
|
|
self.readerObj.getData()
|
|
|
#print self.readerObj.datablock
|
|
|
#self.voltProcObj.init()
|
|
|
self.specProcObj.init()
|
|
|
|
|
|
self.specProcObj.integrator(N=32) ## return self.dataOutObj
|
|
|
|
|
|
self.plot()
|
|
|
#print self.readerObj.dataOutObj.data
|
|
|
|
|
|
#CON TODO ESTO PLoTEO
|
|
|
#freq = self.specProcObj.getFrecuencies()
|
|
|
#spc = self.specProcObj.getSpectra()
|
|
|
# print freq
|
|
|
# print spc, spc.shape
|
|
|
# pl.plot(freq, spc[2])
|
|
|
# pl.show()
|
|
|
|
|
|
if self.readerObj.flagNoMoreFiles:
|
|
|
break
|
|
|
|
|
|
if self.readerObj.flagIsNewBlock:
|
|
|
print 'Block No %04d, Time: %s' %(self.readerObj.nTotalBlocks,
|
|
|
datetime.datetime.fromtimestamp(self.readerObj.m_BasicHeader.utc),)
|
|
|
|
|
|
|
|
|
def plot(self):
|
|
|
|
|
|
# x = self.specProcObj.getFrecuencies()
|
|
|
x = numpy.arange(self.specObj1.nFFTPoints)
|
|
|
spc = self.specObj1.data_spc
|
|
|
# spc = self.specProcObj.getSpectra()
|
|
|
nchannels = self.specObj1.nChannels
|
|
|
|
|
|
# y1=spc[0]
|
|
|
# y2=spc[1]
|
|
|
# y3=spc[2]
|
|
|
# y4=spc[3]
|
|
|
|
|
|
# signalList=[y1,y2,y3,y4]
|
|
|
# xmin = numpy.min(x)
|
|
|
# xmax = numpy.max(x)
|
|
|
|
|
|
#Creating Object
|
|
|
indexPlot = 1
|
|
|
nsubplot = nchannels
|
|
|
#nsubplot = 1
|
|
|
print "nsubplot:%d"%(nsubplot)
|
|
|
winTitle = "mi grafico de Espectrov1"
|
|
|
|
|
|
subplotTitle = "subplot-Espectro - No Channel."
|
|
|
xlabel = ""
|
|
|
ylabel = ""
|
|
|
|
|
|
linearObj = LinearPlot(indexPlot,nsubplot,winTitle)
|
|
|
|
|
|
#Config SubPlots
|
|
|
print "ConfigSubPlots"
|
|
|
print "nsubplot: %d "%nsubplot
|
|
|
for subplot in range(nsubplot):
|
|
|
indexplot = subplot + 1
|
|
|
name=subplot+1
|
|
|
print "Channel: %d"%(name)
|
|
|
title = subplotTitle + '%d'%name
|
|
|
xmin = numpy.min(x)
|
|
|
xmax = numpy.max(x)
|
|
|
ymin = numpy.min(spc[subplot])
|
|
|
ymax = numpy.max(spc[subplot])
|
|
|
linearObj.setup(indexplot, xmin, xmax, ymin, ymax, title, xlabel, ylabel)
|
|
|
|
|
|
#Plotting
|
|
|
type = "simple"
|
|
|
print "Ploteando"
|
|
|
print "nsubplot: %d "%nsubplot
|
|
|
for subplot in range(nsubplot):
|
|
|
indexplot = subplot + 1
|
|
|
y = spc[subplot,:]
|
|
|
linearObj.plot(indexplot, x, y, type)
|
|
|
|
|
|
linearObj.refresh()
|
|
|
linearObj.show()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
TestHeis()
|
|
|
|
|
|
|
