schain-jc Commit - r113:1c6a8ba2204a

Agregando funciones para guardar graficos del tipo Voltage

Daniel Valdez -

r113:1c6a8ba2204a

parent child

Context file:

r113:1c6a8ba2204a

Collapse all files

schainpy/Graphics/BaseGraph.py +9 -1

             """
             Created on Feb 7, 2012
             @autor $Author$
             @version $Id$
             """
             import os
             import numpy
             import sys
             import time
             import datetime
             import time
             import plplot
             def cmap1_init(colormap="gray"):
                 if colormap == None:
                     return
                 ncolor = None
                 rgb_lvl = None
                 # Routine for defining a specific color map 1 in HLS space.
                 # if gray is true, use basic grayscale variation from half-dark to light.
                 # otherwise use false color variation from blue (240 deg) to red (360 deg).
                 # Independent variable of control points.
                 i = numpy.array((0., 1.))
                 if colormap=="gray":
                     ncolor = 256
                     # Hue for control points.  Doesn't matter since saturation is zero.
                     h = numpy.array((0., 0.))
                     # Lightness ranging from half-dark (for interest) to light.
                     l = numpy.array((0.5, 1.))
                     # Gray scale has zero saturation
                     s = numpy.array((0., 0.))
                     # number of cmap1 colours is 256 in this case.
                     plplot.plscmap1n(ncolor)
                     # Interpolate between control points to set up cmap1.
                     plplot.plscmap1l(0, i, h, l, s)
                     return None
                 if colormap == 'jet':
                     ncolor = 256
                     pos = numpy.zeros((ncolor))
                     r = numpy.zeros((ncolor))
                     g = numpy.zeros((ncolor))
                     b = numpy.zeros((ncolor))
                     for i in range(ncolor):
                         if(i <= 35.0/100*(ncolor-1)): rf = 0.0
                         elif (i <= 66.0/100*(ncolor-1)): rf = (100.0/31)*i/(ncolor-1) - 35.0/31
                         elif (i <= 89.0/100*(ncolor-1)): rf = 1.0
                         else: rf = (-100.0/22)*i/(ncolor-1) + 111.0/22
                         if(i <= 12.0/100*(ncolor-1)): gf = 0.0
                         elif(i <= 38.0/100*(ncolor-1)): gf = (100.0/26)*i/(ncolor-1) - 12.0/26
                         elif(i <= 64.0/100*(ncolor-1)): gf = 1.0
                         elif(i <= 91.0/100*(ncolor-1)): gf = (-100.0/27)*i/(ncolor-1) + 91.0/27
                         else: gf = 0.0
                         if(i <= 11.0/100*(ncolor-1)): bf = (50.0/11)*i/(ncolor-1) + 0.5
                         elif(i <= 34.0/100*(ncolor-1)): bf = 1.0
                         elif(i <= 65.0/100*(ncolor-1)): bf = (-100.0/31)*i/(ncolor-1) + 65.0/31
                         else: bf = 0
                         r[i] = rf
                         g[i] = gf
                         b[i] = bf
                         pos[i] = float(i)/float(ncolor-1)
                     plplot.plscmap1n(ncolor)
                     plplot.plscmap1l(1, pos, r, g, b)
                 if colormap=="br_green":
                     ncolor = 256
                     # Hue ranges from blue (240 deg) to red (0 or 360 deg)
                     h = numpy.array((240., 0.))
                     # Lightness and saturation are constant (values taken from C example).
                     l = numpy.array((0.6, 0.6))
                     s = numpy.array((0.8, 0.8))
                     # number of cmap1 colours is 256 in this case.
                     plplot.plscmap1n(ncolor)
                     # Interpolate between control points to set up cmap1.
                     plplot.plscmap1l(0, i, h, l, s)
                     return None
                 if colormap=="tricolor":
                     ncolor = 3
                     # Hue ranges from blue (240 deg) to red (0 or 360 deg)
                     h = numpy.array((240., 0.))
                     # Lightness and saturation are constant (values taken from C example).
                     l = numpy.array((0.6, 0.6))
                     s = numpy.array((0.8, 0.8))
                     # number of cmap1 colours is 256 in this case.
                     plplot.plscmap1n(ncolor)
                     # Interpolate between control points to set up cmap1.
                     plplot.plscmap1l(0, i, h, l, s)
                     return None
                 if colormap == 'rgb' or colormap == 'rgb666':
                     color_sz = 6
                     ncolor = color_sz*color_sz*color_sz
                     pos = numpy.zeros((ncolor))
                     r = numpy.zeros((ncolor))
                     g = numpy.zeros((ncolor))
                     b = numpy.zeros((ncolor))
                     ind = 0
                     for ri in range(color_sz):
                         for gi in range(color_sz):
                             for bi in range(color_sz):
                                 r[ind] = ri/(color_sz-1.0)
                                 g[ind] = gi/(color_sz-1.0)
                                 b[ind] = bi/(color_sz-1.0)
                                 pos[ind] = ind/(ncolor-1.0)
                                 ind += 1
                     rgb_lvl = [6,6,6]  #Levels for RGB colors
                 if colormap == 'rgb676':
                     ncolor = 6*7*6
                     pos = numpy.zeros((ncolor))
                     r = numpy.zeros((ncolor))
                     g = numpy.zeros((ncolor))
                     b = numpy.zeros((ncolor))
                     ind = 0
                     for ri in range(8):
                         for gi in range(8):
                             for bi in range(4):
                                 r[ind] = ri/(6-1.0)
                                 g[ind] = gi/(7-1.0)
                                 b[ind] = bi/(6-1.0)
                                 pos[ind] = ind/(ncolor-1.0)
                                 ind += 1
                     rgb_lvl = [6,7,6]  #Levels for RGB colors
                 if colormap == 'rgb685':
                     ncolor = 6*8*5
                     pos = numpy.zeros((ncolor))
                     r = numpy.zeros((ncolor))
                     g = numpy.zeros((ncolor))
                     b = numpy.zeros((ncolor))
                     ind = 0
                     for ri in range(8):
                         for gi in range(8):
                             for bi in range(4):
                                 r[ind] = ri/(6-1.0)
                                 g[ind] = gi/(8-1.0)
                                 b[ind] = bi/(5-1.0)
                                 pos[ind] = ind/(ncolor-1.0)
                                 ind += 1
                     rgb_lvl = [6,8,5]  #Levels for RGB colors
                 if colormap == 'rgb884':
                     ncolor = 8*8*4
                     pos = numpy.zeros((ncolor))
                     r = numpy.zeros((ncolor))
                     g = numpy.zeros((ncolor))
                     b = numpy.zeros((ncolor))
                     ind = 0
                     for ri in range(8):
                         for gi in range(8):
                             for bi in range(4):
                                 r[ind] = ri/(8-1.0)
                                 g[ind] = gi/(8-1.0)
                                 b[ind] = bi/(4-1.0)
                                 pos[ind] = ind/(ncolor-1.0)
                                 ind += 1
                     rgb_lvl = [8,8,4]  #Levels for RGB colors
                 if ncolor == None:
                     raise ValueError, "The colormap selected is not valid"
                 plplot.plscmap1n(ncolor)
                 plplot.plscmap1l(1, pos, r, g, b)
                 return rgb_lvl
             def setColormap(colormap="jet"):
                 cmap1_init(colormap)
             def savePlplot(filename,width,height):
                 curr_strm = plplot.plgstrm()
                 save_strm = plplot.plmkstrm()
                 plplot.plsetopt("geometry", "%dx%d"%(width,height))
                 plplot.plsdev("png")
                 plplot.plsfnam(filename)
                 plplot.plcpstrm(curr_strm,0)
                 plplot.plreplot()
                 plplot.plend1()
                 plplot.plsstrm(curr_strm)
             def initPlplot(indexPlot,ncol,nrow,winTitle,width,height):
                 plplot.plsstrm(indexPlot)
                 plplot.plparseopts([winTitle],plplot.PL_PARSE_FULL)
                 plplot.plsetopt("geometry", "%dx%d"%(width*ncol,height*nrow))
                 plplot.plsdev("xwin")
                 plplot.plscolbg(255,255,255)
                 plplot.plscol0(1,0,0,0)
                 plplot.plinit()
                 plplot.plspause(False)
                 plplot.plssub(ncol,nrow)
             def setNewPage():
                 plplot.plbop()
                 plplot.pladv(0)
             def closePage():
                 plplot.pleop()
             def clearData(objGraph):
                 objGraph.plotBox(objGraph.xrange[0], objGraph.xrange[1], objGraph.yrange[0], objGraph.yrange[1], "bc", "bc")
                 objGraph.setColor(15) #Setting Line Color to White
                 if objGraph.datatype == "complex":
                     objGraph.basicXYPlot(objGraph.xdata,objGraph.ydata.real)
                     objGraph.basicXYPlot(objGraph.xdata,objGraph.ydata.imag)
                 if objGraph.datatype == "real":
                     objGraph.basicXYPlot(objGraph.xdata,objGraph.ydata)
                 objGraph.setColor(1) #Setting Line Color to Black
             #    objGraph.setLineStyle(2)
             #    objGraph.plotBox(objGraph.xrange[0], objGraph.xrange[1], objGraph.yrange[0], objGraph.yrange[1], "bcntg", "bc")
             #    objGraph.setLineStyle(1)
             def setStrm(indexPlot):
                 plplot.plsstrm(indexPlot)
             def plFlush():
                 plplot.plflush()
             def setPlTitle(pltitle,color, szchar=0.7):
                 setSubpages(1, 0)
                 plplot.pladv(0)
                 plplot.plvpor(0., 1., 0., 1.)
                 if color == "black":
                     plplot.plcol0(1)
                 if color == "white":
                     plplot.plcol0(15)
                 plplot.plschr(0.0,szchar)
                 plplot.plmtex("t",-1., 0.5, 0.5, pltitle)
             def setSubpages(ncol,nrow):
                 plplot.plssub(ncol,nrow)
             class BaseGraph:
                 __name = None
                 __xpos = None
                 __ypos = None
                 __subplot = None
                 __xg = None
                 __yg = None
                 xdata = None
                 ydata = None
                 getGrid = True
                 xaxisIsTime = False
                 deltax = None
                 xmin = None
                 xmax = None
                 def __init__(self,name,subplot,xpos,ypos,xlabel,ylabel,title,szchar,xrange,yrange,zrange=None,deltax=1.0):
                     self.setName(name)
                     self.setScreenPos(xpos, ypos)
                     self.setLabels(xlabel,ylabel,title)
                     self.setSubPlot(subplot)
                     self.setSizeOfChar(szchar)
                     self.setXYZrange(xrange,yrange,zrange)
                     self.getGrid = True
                     self.xaxisIsTime = False
                     self.deltax = deltax
                 def setXYZrange(self,xrange,yrange,zrange):
                     self.xrange = xrange
                     self.yrange = yrange
                     self.zrange = zrange
                 def setName(self, name):
                     self.__name = name
                 def setScreenPos(self,xpos,ypos):
                     self.__xpos = xpos
                     self.__ypos = ypos
                 def setXYData(self,xdata=None,ydata=None,datatype="real"):
                     if((xdata != None) and (ydata != None)):
                         self.xdata = xdata
                         self.ydata = ydata
                         self.datatype = datatype
                     if((self.xdata == None) and (self.ydata == None)):
                         return None
                     return 1
                 def setLabels(self,xlabel=None,ylabel=None,title=None):
                     if xlabel != None: self.xlabel = xlabel
                     if ylabel != None: self.ylabel = ylabel
                     if title != None: self.title = title
                 def setSubPlot(self,subplot):
                     self.__subplot = subplot
                 def setSizeOfChar(self,szchar):
                     self.__szchar = szchar
                 def setLineStyle(self,style):
                     plplot.pllsty(style)
                 def setColor(self,color):
                     plplot.plcol0(color)
                 def setXAxisAsTime(self,value=False):
                     self.xaxisIsTime = value
                 def basicLineTimePlot(self, x, y, xmin=None, xmax=None, ymin=None, ymax=None, colline=1):
                     if xmin == None: xmin = x[0]
                     if xmax == None: xmax = x[-1]
                     if ymin == None: ymin = y[0]
                     if ymax == None: ymax = y[-1]
                     plplot.plcol0(colline)
                     plplot.plline(x, y)
                     plplot.plcol0(1)
                 def basicXYPlot(self, x, y, xmin=None, xmax=None, ymin=None, ymax=None):
                     if xmin == None: xmin = x[0]
                     if xmax == None: xmax = x[-1]
                     if ymin == None: ymin = y[0]
                     if ymax == None: ymax = y[-1]
                     plplot.plline(x, y)
                 def basicPcolorPlot(self, data, x, y, xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None):
                     """
                     """
                     if xmin == None: xmin = x[0]
                     if xmax == None: xmax = x[-1]
                     if ymin == None: ymin = y[0]
                     if ymax == None: ymax = y[-1]
                     if zmin == None: zmin = numpy.nanmin(data)
                     if zmax == None: zmax = numpy.nanmax(data)
                     plplot.plimage(data,
                                    float(x[0]),
                                    float(x[-1]),
                                    float(y[0]),
                                    float(y[-1]),
                                    float(zmin),
                                    float(zmax),
                                    float(xmin),
                                    float(xmax),
                                    float(ymin),
                                    float(ymax)
                                    )
                 def __getBoxpltr(self, x, y, deltax=None, deltay=None):
                     if not(len(x)>0 and len(y)>0):
                         raise ValueError, "x axis and y axis are empty"
                     if deltax == None: deltax = x[-1] - x[-2]
                     if deltay == None: deltay = y[-1] - y[-2]
                     x1 = numpy.append(x, x[-1] + deltax)
                     y1 = numpy.append(y, y[-1] + deltay)
                     xg = (numpy.multiply.outer(x1, numpy.ones(len(y1))))
                     yg = (numpy.multiply.outer(numpy.ones(len(x1)), y1))
                     self.__xg = xg
                     self.__yg = yg
                     return xg, yg
                 def advPcolorPlot(self, data, x, y, xmin=None, xmax=None, ymin=None, ymax=None, zmin=0., zmax=0., deltax=1.0, deltay=None, getGrid = True):
                     if getGrid:
                         xg, yg = self.__getBoxpltr(x, y, deltax, deltay)
                     else:
                         xg = self.__xg
                         yg = self.__yg
                     plplot.plimagefr(data,
                                      float(xmin),
                                      float(xmax),
                                      float(ymin),
                                      float(ymax),
 .,
 .,
                                      float(zmin),
                                      float(zmax),
                                      plplot.pltr2,
                                      xg,
                                      yg)
                 def colorbarPlot(self, xmin=0., xmax=1., ymin=0., ymax=1.):
                     data = numpy.arange(256)
                     data = numpy.reshape(data, (1,-1))
                     plplot.plimage(data,
                                    float(xmin),
                                    float(xmax),
                                    float(ymin),
                                    float(ymax),
 .,
 .,
                                    float(xmin),
                                    float(xmax),
                                    float(ymin),
                                    float(ymax))
                 def plotBox(self, xmin, xmax, ymin, ymax, xopt, yopt, nolabels=False):
                     plplot.plschr(0.0,self.__szchar-0.05)
                     plplot.pladv(self.__subplot)
                     plplot.plvpor(self.__xpos[0], self.__xpos[1], self.__ypos[0], self.__ypos[1])
                     plplot.plwind(float(xmin), # self.xrange[0]
                                   float(xmax), # self.xrange[1]
                                   float(ymin), # self.yrange[0]
                                   float(ymax) # self.yrange[1]
                                   )
                     if self.xaxisIsTime:
                         plplot.pltimefmt("%H:%M")
                         timedelta = (xmax - xmin + 1)/8.
                         plplot.plbox(xopt, timedelta, 3, yopt, 0.0, 0)
                     else:
                         plplot.plbox(xopt, 0.0, 0, yopt, 0.0, 0)
                     if not(nolabels):
                         plplot.pllab(self.xlabel, self.ylabel, self.title)
                 def delLabels(self):
                     self.setColor(15) #Setting Line Color to White
                     plplot.pllab(self.xlabel, self.ylabel, self.title)
                     self.setColor(1) #Setting Line Color to Black
                 def plotImage(self,x,y,z,xrange,yrange,zrange):
                     xi = x[0]
                     xf = x[-1]
                     yi = y[0]
                     yf = y[-1]
                     plplot.plimage(z,
                                    float(xi),
                                    float(xf),
                                    float(yi),
                                    float(yf),
                                    float(zrange[0]),
                                    float(zrange[1]),
                                    float(xi),
                                    float(xf),
                                    float(yrange[0]),
                                    yrange[1])
             class LinearPlot:
                 linearObjDic = {}
                 __xpos = None
                 __ypos = None
                 def __init__(self,indexPlot,nsubplot,winTitle):
                     self.width = 700
                     self.height = 150
                     ncol = 1
                     nrow = nsubplot
                     initPlplot(indexPlot,ncol,nrow,winTitle,self.width,self.height)
                 def setFigure(self,indexPlot):
                     setStrm(indexPlot)
                 def setPosition(self):
                     xi = 0.07; xf = 0.9 #0.8,0.7,0.5
                     yi = 0.15; yf = 0.8
                     xpos = [xi,xf]
                     ypos = [yi,yf]
                     self.__xpos = xpos
                     self.__ypos = ypos
                     return xpos,ypos
                 def refresh(self):
                     plFlush()
                 def setup(self,subplot,xmin,xmax,ymin,ymax,title,xlabel,ylabel):
                     szchar = 1.10
                     name = "linear"
                     key = name + "%d"%subplot
                     xrange = [xmin,xmax]
                     yrange = [ymin,ymax]
                     xpos,ypos = self.setPosition()
                     linearObj = BaseGraph(name,subplot,xpos,ypos,xlabel,ylabel,title,szchar,xrange,yrange)
                     linearObj.plotBox(linearObj.xrange[0], linearObj.xrange[1], linearObj.yrange[0], linearObj.yrange[1], "bcnst", "bcnstv")
                     self.linearObjDic[key] = linearObj
                 def plot(self,subplot,x,y,type="power"):
                     name = "linear"
                     key = name + "%d"%subplot
                     linearObj = self.linearObjDic[key]
                     linearObj.plotBox(linearObj.xrange[0], linearObj.xrange[1], linearObj.yrange[0], linearObj.yrange[1], "bcst", "bcst")
                     if linearObj.setXYData() != None:
                             clearData(linearObj)
                     else:
                         if type.lower() == 'power':
                             linearObj.setXYData(x,abs(y),"real")
                         if type.lower() == 'iq':
                             linearObj.setXYData(x,y,"complex")
                     if type.lower() == 'power':
                         colline = 9
                         linearObj.basicLineTimePlot(x, abs(y), xmin, xmax, ymin, ymax, colline)
                         linearObj.setXYData(x,abs(y),"real")
                     if type.lower() == 'iq':
                         colline = 9
                         linearObj.basicLineTimePlot(x=x, y=y.real, colline=colline)
                         colline = 13
                         linearObj.basicLineTimePlot(x=x, y=y.imag, colline=colline)
                         linearObj.setXYData(x,y,"complex")
                     linearObj.plotBox(linearObj.xrange[0], linearObj.xrange[1], linearObj.yrange[0], linearObj.yrange[1], "bcst", "bcst")
             #        linearObj.plotBox(linearObj.xrange[0], linearObj.xrange[1], linearObj.yrange[0], linearObj.yrange[1], "bc", "bc")
             #        linearObj.basicXYPlot(data,y)
             #        linearObj.setXYData(data,y)
             class PcolorPlot:
                 pcolorObjDic = {}
                 colorbarObjDic = {}
                 pwprofileObjDic = {}
                 showColorbar = None
                 showPowerProfile = None
                 XAxisAsTime = None
                 width = None
                 height = None
                 __spcxpos = None
                 __spcypos = None
                 __cmapxpos = None
                 __cmapypos = None
                 __profxpos = None
                 __profypos = None
                 __lastTitle = None
                 def __init__(self,indexPlot,nsubplot,winTitle,colormap,showColorbar,showPowerProfile,XAxisAsTime):
                     self.width = 460
                     self.height = 300
                     self.showColorbar = showColorbar
                     self.showPowerProfile = showPowerProfile
                     self.XAxisAsTime = XAxisAsTime
                     ncol = int(numpy.sqrt(nsubplot)+0.9)
                     nrow = int(nsubplot*1./ncol + 0.9)
                     initPlplot(indexPlot,ncol,nrow,winTitle,self.width,self.height)
                     setColormap(colormap)
                     self.ncol = ncol
                     self.nrow = nrow
                 def setFigure(self,indexPlot):
                     setStrm(indexPlot)
                 def setSpectraPos(self): #modificar valores de acuerdo al colorbar y pwprofile
                     if self.showPowerProfile: xi = 0.09; xf = 0.6 #0.075
                     else:   xi = 0.15; xf = 0.8 #0.8,0.7,0.5
                     yi = 0.15; yf = 0.80
                     xpos = [xi,xf]
                     ypos = [yi,yf]
                     self.__spcxpos = xpos
                     self.__spcypos = ypos
                     return xpos,ypos
                 def setColorbarScreenPos(self):
                     xi = self.__spcxpos[1] + 0.03; xf = xi + 0.03
                     yi = self.__spcypos[0]; yf = self.__spcypos[1]
                     xpos = [xi,xf]
                     ypos = [yi,yf]
                     self.__cmapxpos = xpos
                     self.__cmapypos = ypos
                     return xpos,ypos
                 def setPowerprofileScreenPos(self):
                     xi = self.__cmapxpos[1] + 0.07; xf = xi + 0.25
                     yi = self.__spcypos[0]; yf = self.__spcypos[1]
                     xpos = [xi,xf]
                     ypos = [yi,yf]
                     self.__profxpos = [xi,xf]
                     self.__profypos = [yi,yf]
                     return xpos,ypos
                 def createObjects(self,subplot,xmin,xmax,ymin,ymax,zmin,zmax,title,xlabel,ylabel):
                     """
                     Crea los objetos necesarios para un subplot
                     """
                     # Config Spectra plot
                     szchar = 0.7
                     name = "spc"
                     key = name + "%d"%subplot
                     xrange = [xmin,xmax]
                     yrange = [ymin,ymax]
                     zrange = [zmin,zmax]
                     xpos,ypos = self.setSpectraPos()
                     pcolorObj = BaseGraph(name,subplot,xpos,ypos,xlabel,ylabel,title,szchar,xrange,yrange,zrange)
                     self.pcolorObjDic[key] = pcolorObj
                     # Config Colorbar
                     if self.showColorbar:
                         szchar = 0.65
                         name = "colorbar"
                         key = name + "%d"%subplot
                         xpos,ypos = self.setColorbarScreenPos()
                         xrange = [0.,1.]
                         yrange = [zmin,zmax]
                         cmapObj = BaseGraph(name,subplot,xpos,ypos,"","","dB",szchar,xrange,yrange)
                         self.colorbarObjDic[key] = cmapObj
                     # Config Power profile
                     if self.showPowerProfile:
                         szchar = 0.55
                         name = "pwprofile"
                         key = name + "%d"%subplot
                         xpos,ypos = self.setPowerprofileScreenPos()
                         xrange = [zmin,zmax]
                         yrange = [ymin,ymax]
                         powObj = BaseGraph(name,subplot,xpos,ypos,"dB","","Power Profile",szchar,xrange,yrange)
                         self.pwprofileObjDic[key] = powObj
                 def setNewPage(self, pltitle='No title'):
                     szchar = 0.7
                     setNewPage()
                     setPlTitle(pltitle,"black", szchar=szchar)
                     setSubpages(self.ncol, self.nrow)
                 def closePage(self):
                     closePage()
                 def iniPlot(self,subplot):
                     """
                     Inicializa los subplots con su frame, titulo, etc
                     """
                     # Config Spectra plot
                     name = "spc"
                     key = name + "%d"%subplot
                     pcolorObj = self.pcolorObjDic[key]
                     pcolorObj.plotBox(pcolorObj.xrange[0], pcolorObj.xrange[1], pcolorObj.yrange[0], pcolorObj.yrange[1], "bcnst", "bcnstv")
                     # Config Colorbar
                     if self.showColorbar:
                         name = "colorbar"
                         key = name + "%d"%subplot
                         cmapObj = self.colorbarObjDic[key]
                         cmapObj.plotBox(cmapObj.xrange[0], cmapObj.xrange[1], cmapObj.yrange[0], cmapObj.yrange[1], "bc", "bcmtsv")
                         cmapObj.colorbarPlot(cmapObj.xrange[0], cmapObj.xrange[1], cmapObj.yrange[0], cmapObj.yrange[1])
             #            cmapObj.plotBox(cmapObj.xrange[0], cmapObj.xrange[1], cmapObj.yrange[0], cmapObj.yrange[1], "bc", "bcmtsv")
                     # Config Power profile
                     if self.showPowerProfile:
                         name = "pwprofile"
                         key = name + "%d"%subplot
                         powObj = self.pwprofileObjDic[key]
                         powObj.setLineStyle(2)
                         powObj.plotBox(powObj.xrange[0], powObj.xrange[1], powObj.yrange[0], powObj.yrange[1], "bcntg", "bc")
                         powObj.setLineStyle(1)
                         powObj.plotBox(powObj.xrange[0], powObj.xrange[1], powObj.yrange[0], powObj.yrange[1], "bc", "bc")
                 def printTitle(self,pltitle):
             #        if self.__lastTitle != None:
             #            setPlTitle(self.__lastTitle,"white")
             #
             #        self.__lastTitle = pltitle
                     setPlTitle(pltitle,"black")
             #        setSubpages(self.ncol,self.nrow)
                 def plot(self,subplot,x,y,z,subtitle):
                     # Spectra plot
                     name = "spc"
                     key = name + "%d"%subplot
             #        newx = [x[0],x[-1]]
                     pcolorObj = self.pcolorObjDic[key]
                     pcolorObj.plotBox(pcolorObj.xrange[0], pcolorObj.xrange[1], pcolorObj.yrange[0], pcolorObj.yrange[1], "bcst", "bcst")
                     #pcolorObj.delLabels()
                     pcolorObj.setLabels(title=subtitle)
                     deltax = None; deltay = None
                     pcolorObj.advPcolorPlot(z,
                                             x,
                                             y,
                                             xmin=pcolorObj.xrange[0],
                                             xmax=pcolorObj.xrange[1],
                                             ymin=pcolorObj.yrange[0],
                                             ymax=pcolorObj.yrange[1],
                                             zmin=pcolorObj.zrange[0],
                                             zmax=pcolorObj.zrange[1],
                                             deltax=deltax,
                                             deltay=deltay,
                                             getGrid=pcolorObj.getGrid)
                     #Solo se calcula la primera vez que se ingresa a la funcion
                     pcolorObj.getGrid = False
                     pcolorObj.plotBox(pcolorObj.xrange[0], pcolorObj.xrange[1], pcolorObj.yrange[0], pcolorObj.yrange[1], "bcst", "bcst", nolabels=True)
                     # Power Profile
                     if self.showPowerProfile:
                         power = numpy.average(z, axis=0)
                         name = "pwprofile"
                         key = name + "%d"%subplot
                         powObj = self.pwprofileObjDic[key]
                         if powObj.setXYData() != None:
                             #clearData(powObj)
                             powObj.setLineStyle(2)
                             powObj.plotBox(powObj.xrange[0], powObj.xrange[1], powObj.yrange[0], powObj.yrange[1], "bcntg", "bc")
                             powObj.setLineStyle(1)
                         else:
                             powObj.setXYData(power,y)
                         powObj.plotBox(powObj.xrange[0], powObj.xrange[1], powObj.yrange[0], powObj.yrange[1], "bc", "bc")
                         powObj.basicXYPlot(power,y)
                         powObj.setXYData(power,y)
                 def savePlot(self,indexPlot,filename):
                     width = self.width*self.ncol
                     hei = self.height*self.nrow
                     savePlplot(filename,width,hei)
                 def refresh(self):
                     plFlush()
             class RtiPlot:
                 pcolorObjDic = {}
                 colorbarObjDic = {}
                 pwprofileObjDic = {}
                 showColorbar = None
                 showPowerProfile = None
                 XAxisAsTime = None
                 widht = None
                 height = None
                 __rtixpos = None
                 __rtiypos = None
                 __cmapxpos = None
                 __cmapypos = None
                 __profxpos = None
                 __profypos = None
                 def __init__(self,indexPlot,nsubplot,winTitle,colormap,showColorbar,showPowerProfile,XAxisAsTime):
                     self.width = 700
                     self.height = 150
                     self.showColorbar = showColorbar
                     self.showPowerProfile = showPowerProfile
                     self.XAxisAsTime = XAxisAsTime
                     ncol = 1
                     nrow = nsubplot
                     initPlplot(indexPlot,ncol,nrow,winTitle,self.width,self.height)
                     setColormap(colormap)
+                    self.ncol = ncol
+                    self.nrow = nrow
                 def setFigure(self,indexPlot):
                     setStrm(indexPlot)
                 def setRtiScreenPos(self):
                     if self.showPowerProfile: xi = 0.07; xf = 0.65
                     else:   xi = 0.07; xf = 0.9
                     yi = 0.15; yf = 0.80
                     xpos = [xi,xf]
                     ypos = [yi,yf]
                     self.__rtixpos = xpos
                     self.__rtiypos = ypos
                     return xpos,ypos
                 def setColorbarScreenPos(self):
                     xi = self.__rtixpos[1] + 0.03; xf = xi + 0.03
                     yi = self.__rtiypos[0]; yf = self.__rtiypos[1]
                     xpos = [xi,xf]
                     ypos = [yi,yf]
                     self.__cmapxpos = xpos
                     self.__cmapypos = ypos
                     return xpos,ypos
                 def setPowerprofileScreenPos(self):
                     xi = self.__cmapxpos[1] + 0.05; xf = xi + 0.20
                     yi = self.__rtiypos[0]; yf = self.__rtiypos[1]
                     xpos = [xi,xf]
                     ypos = [yi,yf]
                     self.__profxpos = [xi,xf]
                     self.__profypos = [yi,yf]
                     return xpos,ypos
                 def setup(self,subplot,xmin,xmax,ymin,ymax,zmin,zmax,title,xlabel,ylabel,timedata,timezone="lt",npoints=100):
                     # Config Rti plot
                     szchar = 1.10
                     name = "rti"
                     key = name + "%d"%subplot
                     # xmin, xmax --> minHour, max Hour : valores que definen el ejex x=[horaInicio,horaFinal]
                     thisDateTime = datetime.datetime.fromtimestamp(timedata)
                     startDateTime = datetime.datetime(thisDateTime.year,thisDateTime.month,thisDateTime.day,xmin,0,0)
                     endDateTime = datetime.datetime(thisDateTime.year,thisDateTime.month,thisDateTime.day,xmax,59,59)
                     deltaTime = 0
                     if timezone == "lt":
                         deltaTime = time.timezone
                     startTimeInSecs = time.mktime(startDateTime.timetuple()) - deltaTime
                     endTimeInSecs = time.mktime(endDateTime.timetuple()) - deltaTime
                     xrange = [startTimeInSecs,endTimeInSecs]
                     totalTimeInXrange = endTimeInSecs - startTimeInSecs + 1.
                     deltax = totalTimeInXrange / npoints
                     yrange = [ymin,ymax]
                     zrange = [zmin,zmax]
                     xpos,ypos = self.setRtiScreenPos()
                     pcolorObj = BaseGraph(name,subplot,xpos,ypos,xlabel,ylabel,title,szchar,xrange,yrange,zrange,deltax)
                     if self.XAxisAsTime:
                         pcolorObj.setXAxisAsTime(self.XAxisAsTime)
                         xopt = "bcnstd"
                         yopt = "bcnstv"
                     else:
                         xopt = "bcnst"
                         yopt = "bcnstv"
                     pcolorObj.plotBox(pcolorObj.xrange[0], pcolorObj.xrange[1], pcolorObj.yrange[0], pcolorObj.yrange[1], xopt, yopt)
                     self.pcolorObjDic[key] = pcolorObj
                     # Config Colorbar
                     if self.showColorbar:
                         szchar = 0.9
                         name = "colorbar"
                         key = name + "%d"%subplot
                         xpos,ypos = self.setColorbarScreenPos()
                         xrange = [0.,1.]
                         yrange = [zmin,zmax]
                         cmapObj = BaseGraph(name,subplot,xpos,ypos,"","","dB",szchar,xrange,yrange)
                         cmapObj.plotBox(cmapObj.xrange[0], cmapObj.xrange[1], cmapObj.yrange[0], cmapObj.yrange[1], "bc", "bcm")
                         cmapObj.colorbarPlot(cmapObj.xrange[0], cmapObj.xrange[1], cmapObj.yrange[0], cmapObj.yrange[1])
                         cmapObj.plotBox(cmapObj.xrange[0], cmapObj.xrange[1], cmapObj.yrange[0], cmapObj.yrange[1], "bc", "bcmtsv")
                         self.colorbarObjDic[key] = cmapObj
                     # Config Power profile
                     if self.showPowerProfile:
                         szchar = 0.8
                         name = "pwprofile"
                         key = name + "%d"%subplot
                         xpos,ypos = self.setPowerprofileScreenPos()
                         xrange = [zmin,zmax]
                         yrange = [ymin,ymax]
                         powObj = BaseGraph(name,subplot,xpos,ypos,"dB","","Power Profile",szchar,xrange,yrange)
                         powObj.setLineStyle(2)
                         powObj.plotBox(powObj.xrange[0], powObj.xrange[1], powObj.yrange[0], powObj.yrange[1], "bcntg", "bc")
                         powObj.setLineStyle(1)
                         powObj.plotBox(powObj.xrange[0], powObj.xrange[1], powObj.yrange[0], powObj.yrange[1], "bc", "bc")
                         self.pwprofileObjDic[key] = powObj
                 def plot(self,subplot,x,y,z):
                     # RTI plot
                     name = "rti"
                     key = name + "%d"%subplot
                     data = numpy.reshape(z, (1,-1))
                     data = numpy.abs(data)
                     data = 10*numpy.log10(data)
                     newx = [x,x+1]
                     pcolorObj = self.pcolorObjDic[key]
                     if pcolorObj.xaxisIsTime:
                         xopt = "bcstd"
                         yopt = "bcst"
                     else:
                         xopt = "bcst"
                         yopt = "bcst"
                     pcolorObj.plotBox(pcolorObj.xrange[0], pcolorObj.xrange[1], pcolorObj.yrange[0], pcolorObj.yrange[1], xopt, yopt)
                     deltax = pcolorObj.deltax
                     deltay = None
                     if pcolorObj.xmin == None and pcolorObj.xmax == None:
                         pcolorObj.xmin = x
                         pcolorObj.xmax = x
                     if x >= pcolorObj.xmax:
                         xmin = x
                         xmax = x + deltax
                         x = [x]
                         pcolorObj.advPcolorPlot(data,
                                                 x,
                                                 y,
                                                 xmin=xmin,
                                                 xmax=xmax,
                                                 ymin=pcolorObj.yrange[0],
                                                 ymax=pcolorObj.yrange[1],
                                                 zmin=pcolorObj.zrange[0],
                                                 zmax=pcolorObj.zrange[1],
                                                 deltax=deltax,
                                                 deltay=deltay,
                                                 getGrid=pcolorObj.getGrid)
                         pcolorObj.xmin = xmin
                         pcolorObj.xmax = xmax
                     # Power Profile
                     if self.showPowerProfile:
                         data = numpy.reshape(data,(numpy.size(data)))
                         name = "pwprofile"
                         key = name + "%d"%subplot
                         powObj = self.pwprofileObjDic[key]
                         if powObj.setXYData() != None:
                             clearData(powObj)
                             powObj.setLineStyle(2)
                             powObj.plotBox(powObj.xrange[0], powObj.xrange[1], powObj.yrange[0], powObj.yrange[1], "bcntg", "bc")
                             powObj.setLineStyle(1)
                         else:
                             powObj.setXYData(data,y)
                         powObj.plotBox(powObj.xrange[0], powObj.xrange[1], powObj.yrange[0], powObj.yrange[1], "bc", "bc")
                         powObj.basicXYPlot(data,y)
                         powObj.setXYData(data,y)
+                def savePlot(self,indexPlot,filename):
+                    width = self.width*self.ncol
+                    hei = self.height*self.nrow
+                    savePlplot(filename,width,hei)
                 def refresh(self):
                     plFlush()

schainpy/Graphics/VoltagePlot.py +10 -1

             '''
             Created on Feb 7, 2012
             @author $Author$
             @version $Id$
             '''
             import numpy
             import os
             import sys
             path = os.path.split(os.getcwd())[0]
             sys.path.append(path)
             from Graphics.BaseGraph import *
             from Model.Voltage import Voltage
             class Osciloscope:
                 linearplotObj = None
                 def __init__(self, Voltage, index):
                     self.__isPlotConfig = False
                     self.__isPlotIni = False
                     self.__xrange = None
                     self.__yrange = None
                     self.indexPlot = index
                     self.voltageObj = Voltage
                 def setup(self,indexPlot,nsubplot,winTitle=''):
                     self.linearplotObj = LinearPlot(indexPlot,nsubplot,winTitle)
                 def initPlot(self,xmin,xmax,ymin,ymax,titleList,xlabelList,ylabelList):
                     nsubplot = self.voltageObj.nChannels
                     for index in range(nsubplot):
                         title = titleList[index]
                         xlabel = xlabelList[index]
                         ylabel = ylabelList[index]
                         subplot = index
                         self.linearplotObj.setup(subplot+1,xmin,xmax,ymin,ymax,title,xlabel,ylabel)
                 def plotData(self,
                              xmin=None,
                              xmax=None,
                              ymin=None,
                              ymax=None,
                              titleList=None,
                              xlabelList=None,
                              ylabelList=None,
                              winTitle='',
                              type="power"):
                     databuffer = self.voltageObj.data
                     height = self.voltageObj.heightList
                     nsubplot = self.voltageObj.nChannels
                     indexPlot = self.indexPlot
                     if not(self.__isPlotConfig):
                         self.setup(indexPlot,nsubplot,winTitle)
                         self.__isPlotConfig = True
                     if not(self.__isPlotIni):
                         if titleList == None:
                             titleList = []
                             thisDatetime = datetime.datetime.fromtimestamp(self.voltageObj.m_BasicHeader.utc)
                             txtdate = "Date: %s" %(thisDatetime.strftime("%d-%b-%Y"))
                             for i in range(nsubplot):
                                 titleList.append("Channel: %d %s" %(i, txtdate))
                         if xlabelList == None:
                             xlabelList = []
                             for i in range(nsubplot):
                                 xlabelList.append("")
                         if ylabelList == None:
                             ylabelList = []
                             for i in range(nsubplot):
                                 ylabelList.append("")
                         if xmin == None: xmin = height[0]
                         if xmax == None: xmax = height[-1]
                         if ymin == None: ymin = numpy.nanmin(abs(databuffer))
                         if ymax == None: ymax = numpy.nanmax(abs(databuffer))
                         self.initPlot(xmin,xmax,ymin,ymax,titleList,xlabelList,ylabelList)
                         self.__isPlotIni = True
                     self.linearplotObj.setFigure(indexPlot)
                     for index in range(nsubplot):
                         data = databuffer[index,:]
                         self.linearplotObj.plot(subplot=index+1,x=height,y=data,type=type)
                     self.linearplotObj.refresh()
             class RTI:
                 colorplotObj = None
                 def __init__(self, Voltage, index):
                     self.__isPlotConfig = False
                     self.__isPlotIni = False
                     self.__xrange = None
                     self.__yrange = None
                     self.indexPlot = index
                     self.voltageObj = Voltage
                 def setup(self,indexPlot,nsubplot,winTitle='',colormap="br_green",showColorbar=False,showPowerProfile=False,XAxisAsTime=False):
                     self.colorplotObj = RtiPlot(indexPlot,nsubplot,winTitle,colormap,showColorbar,showPowerProfile,XAxisAsTime)
                 def initPlot(self,xmin,xmax,ymin,ymax,zmin,zmax,titleList,xlabelList,ylabelList,timezone,npoints):
                     nsubplot = self.voltageObj.nChannels
                     timedata = self.voltageObj.m_BasicHeader.utc
                     for index in range(nsubplot):
                         title = titleList[index]
                         xlabel = xlabelList[index]
                         ylabel = ylabelList[index]
                         subplot = index
                         self.colorplotObj.setup(subplot+1,xmin,xmax,ymin,ymax,zmin,zmax,title,xlabel,ylabel,timedata,timezone,npoints)
                 def plotData(self,
                              xmin=None,
                              xmax=None,
                              ymin=None,
                              ymax=None,
                              zmin=None,
                              zmax=None,
                              titleList=None,
                              xlabelList=None,
                              ylabelList=None,
                              winTitle='',
                              timezone='lt',
                              npoints=1000.0,
                              colormap="br_green",
                              showColorbar=True,
                              showPowerProfile=True,
-                             XAxisAsTime=True):
+                             XAxisAsTime=True,
+                             save = False):
                     databuffer = self.voltageObj.data
                     timedata = self.voltageObj.m_BasicHeader.utc
                     height = self.voltageObj.heightList
                     nsubplot = self.voltageObj.nChannels
                     indexPlot = self.indexPlot
                     if not(self.__isPlotConfig):
                         self.setup(indexPlot,nsubplot,winTitle,colormap,showColorbar,showPowerProfile,XAxisAsTime)
                         self.__isPlotConfig = True
                     if not(self.__isPlotIni):
                         if titleList == None:
                             titleList = []
                             thisDatetime = datetime.datetime.fromtimestamp(timedata)
                             txtdate = "Date: %s" %(thisDatetime.strftime("%d-%b-%Y"))
                             for i in range(nsubplot):
                                 titleList.append("Channel: %d %s" %(i, txtdate))
                         if xlabelList == None:
                             xlabelList = []
                             for i in range(nsubplot):
                                 xlabelList.append("")
                         if ylabelList == None:
                             ylabelList = []
                             for i in range(nsubplot):
                                 ylabelList.append("")
                         if xmin == None: xmin = 0
                         if xmax == None: xmax = 23
                         if ymin == None: ymin = min(self.voltageObj.heightList)
                         if ymax == None: ymax = max(self.voltageObj.heightList)
                         if zmin == None: zmin = 0
                         if zmax == None: zmax = 50
                         self.initPlot(xmin,xmax,ymin,ymax,zmin,zmax,titleList,xlabelList,ylabelList,timezone,npoints)
                         self.__isPlotIni = True
                     self.colorplotObj.setFigure(indexPlot)
                     if timezone == 'lt':
                         timedata = timedata - time.timezone
                     for index in range(nsubplot):
                         data = databuffer[index,:]
                         self.colorplotObj.plot(subplot=index+1,x=timedata,y=height,z=data)
                     self.colorplotObj.refresh()
+                    if save:
+                        self.colorplotObj.setFigure(indexPlot)
+                        path = "/Users/jro/Pictures"
+                        now = datetime.datetime.now().timetuple()
+                        file = "rti_img%02d_%03d_%02d%02d%02d.png"%(indexPlot,now[7],now[3],now[4],now[5])
+                        filename = os.path.join(path,file)
+                        self.colorplotObj.savePlot(indexPlot, filename)

schainpy/IO/JRODataIO.py +1 0

             '''
             Created on 23/01/2012
             @author $Author$
             @version $Id$
             @version $Id$
             '''
             import os, sys
             import glob
             import time
             import numpy
             import fnmatch
             import time, datetime
             path = os.path.split(os.getcwd())[0]
             sys.path.append(path)
             from Model.JROHeader import *
             from Model.JROData import JROData
             def checkForRealPath(path, year, doy, set, ext):
                 """
                 Por ser Linux Case Sensitive entonces checkForRealPath encuentra el nombre correcto de un path,
                 Prueba por varias combinaciones de nombres entre mayusculas y minusculas para determinar
                 el path exacto de un determinado file.
                 Example    :
                     nombre correcto del file es  .../.../D2009307/P2009307367.ext
                     Entonces la funcion prueba con las siguientes combinaciones
                         .../.../x2009307/y2009307367.ext
                         .../.../x2009307/Y2009307367.ext
                         .../.../X2009307/y2009307367.ext
                         .../.../X2009307/Y2009307367.ext
                     siendo para este caso, la ultima combinacion de letras, identica al file buscado
                 Return:
                     Si encuentra la cobinacion adecuada devuelve el path completo y el nombre del file
                     caso contrario devuelve None como path y el la ultima combinacion de nombre en mayusculas
                     para el filename
                 """
                 filepath = None
                 find_flag = False
                 filename = None
                 if ext.lower() == ".r": #voltage
                     header1 = "dD"
                     header2 = "dD"
                 elif ext.lower() == ".pdata": #spectra
                     header1 = "dD"
                     header2 = "pP"
                 else:
                     return None, filename
                 for dir in header1: #barrido por las dos combinaciones posibles de "D"
                     for fil in header2: #barrido por las dos combinaciones posibles de "D"
                         doypath = "%s%04d%03d" % ( dir, year, doy ) #formo el nombre del directorio xYYYYDDD (x=d o x=D)
                         filename = "%s%04d%03d%03d%s" % ( fil, year, doy, set, ext ) #formo el nombre del file xYYYYDDDSSS.ext
                         filepath = os.path.join( path, doypath, filename ) #formo el path completo
                         if os.path.exists( filepath ): #verifico que exista
                             find_flag = True
                             break
                     if find_flag:
                         break
                 if not(find_flag):
                     return None, filename
                 return filepath, filename
             def isNumber(str):
                 """
                 Chequea si el conjunto de caracteres que componen un string puede ser convertidos a un numero.
                 Excepciones:
                     Si un determinado string no puede ser convertido a numero
                 Input:
                     str, string al cual se le analiza para determinar si convertible a un numero o no
                 Return:
                     True    :    si el string es uno numerico
                     False   :    no es un string numerico
                 """
                 try:
                     float( str )
                     return True
                 except:
                     return False
             def isThisFileinRange(filename, startUTSeconds, endUTSeconds):
                 """
                 Esta funcion determina si un archivo de datos se encuentra o no dentro del rango de fecha especificado.
                 Inputs:
                     filename            :    nombre completo del archivo de datos en formato Jicamarca (.r)
                     startUTSeconds      :    fecha inicial del rango seleccionado. La fecha esta dada en
                                              segundos contados desde 01/01/1970.
                     endUTSeconds        :    fecha final del rango seleccionado. La fecha esta dada en
                                              segundos contados desde 01/01/1970.
                 Return:
                     Boolean    :    Retorna True si el archivo de datos contiene datos en el rango de
                                     fecha especificado, de lo contrario retorna False.
                 Excepciones:
                     Si el archivo no existe o no puede ser abierto
                     Si la cabecera no puede ser leida.
                 """
                 m_BasicHeader = BasicHeader()
                 try:
                     fp = open(filename,'rb')
                 except:
                     raise IOError, "The file %s can't be opened" %(filename)
                 sts = m_BasicHeader.read(fp)
                 fp.close()
                 if not(sts):
                     print "Skipping the file %s because it has not a valid header" %(filename)
                     return 0
                 if not ((startUTSeconds <= m_BasicHeader.utc) and (endUTSeconds > m_BasicHeader.utc)):
                     return 0
                 return 1
             def getlastFileFromPath(path, ext):
                 """
                 Depura el fileList dejando solo los que cumplan el formato de "PYYYYDDDSSS.ext"
                 al final de la depuracion devuelve el ultimo file de la lista que quedo.
                 Input:
                     fileList    :    lista conteniendo todos los files (sin path) que componen una determinada carpeta
                     ext         :    extension de los files contenidos en una carpeta
                 Return:
                     El ultimo file de una determinada carpeta, no se considera el path.
                 """
                 validFilelist = []
                 fileList = os.listdir(path)
                 # 0 1234 567 89A BCDE
                 # H YYYY DDD SSS .ext
                 for file in fileList:
                     try:
                         year = int(file[1:5])
                         doy  = int(file[5:8])
                         if (os.path.splitext(file)[-1].upper() != ext.upper()) : continue
                     except:
                         continue
                     validFilelist.append(file)
                 if validFilelist:
                     validFilelist = sorted( validFilelist, key=str.lower )
                     return validFilelist[-1]
                 return None
             class JRODataIO:
                 #speed of light
                 c = 3E8
                 m_BasicHeader = BasicHeader()
                 m_SystemHeader = SystemHeader()
                 m_RadarControllerHeader = RadarControllerHeader()
                 m_ProcessingHeader = ProcessingHeader()
                 dataOutObj = None
                 online = 0
                 fp = None
                 dataType = None
                 fileSizeByHeader = None
                 filenameList = []
                 filename = None
                 fileSize = None
                 firstHeaderSize = 0
                 basicHeaderSize = 24
                 nTotalBlocks = 0
                 ippSeconds = 0
                 blocksize = 0
                 set = 0
                 ext = None
                 path = None
                 maxTimeStep = 30
                 delay  = 3   #seconds
                 nTries  = 3  #quantity tries
                 nFiles = 3   #number of files for searching
                 flagNoMoreFiles = 0
                 flagIsNewFile = 1
                 flagResetProcessing = 0
                 flagIsNewBlock = 0
                 def __init__(self):
                     pass
             class JRODataReader(JRODataIO):
                 """
                 Esta clase es usada como la clase padre de las clases VoltageReader y SpectraReader.
                 Contiene todos lo metodos necesarios para leer datos desde archivos en formato
                 jicamarca o pdata (.r o .pdata). La lectura de los datos siempre se realiza por bloques. Los datos
                 leidos son array de 3 dimensiones:
                                   Para Voltajes  -  perfiles * alturas * canales
                                   Para Spectra   -  paresCanalesIguales    * alturas * perfiles  (Self Spectra)
                                                     paresCanalesDiferentes * alturas * perfiles  (Cross Spectra)
                                                     canales * alturas                            (DC Channels)
                 y son almacenados en su buffer respectivo.
                 Esta clase contiene instancias (objetos) de las clases BasicHeader, SystemHeader,
                 RadarControllerHeader y DataObj. Los tres primeros se usan para almacenar informacion de la
                 cabecera de datos (metadata), y el cuarto (DataObj) para obtener y almacenar los datos desde
                 el buffer de datos cada vez que se ejecute el metodo "getData".
                 """
                 nReadBlocks = 0
                 def __init__(self, dataOutObj=None):
                     raise ValueError, "This class can't be instanced"
                 def hasNotDataInBuffer(self):
                     raise ValueError, "Not implemented"
                 def getBlockDimension(self):
                     raise ValueError, "No implemented"
                 def readBlock(self):
                     self.nTotalBlocks += 1
                     self.nReadBlocks += 1
                     raise ValueError, "This method has not been implemented"
                 def getData( self ):
                     raise ValueError, "This method has not been implemented"
                 def createObjByDefault(self):
                     """
                     Los objetos creados por defecto por cada clase (Voltaje o Espectro) difieren en el tipo
                     raise ValueError, "This method has not been implemented
                     """
                     raise ValueError, "This method has not been implemented"
+            #    def setup(self, dataOutObj=None, path=None, startDateTime=None, endDateTime=None, set=0, expLabel = "", ext = None, online = 0):
                 def setup(self, dataOutObj=None, path=None, startDateTime=None, endDateTime=None, set=0, expLabel = "", ext = None, online = 0):
                     """
                     setup configura los parametros de lectura de la clase DataReader.
                     Si el modo de lectura es offline, primero se realiza una busqueda de todos los archivos
                     que coincidan con los parametros especificados; esta lista de archivos son almacenados en
                     self.filenameList.
                     Input:
                         path                :    Directorios donde se ubican los datos a leer. Dentro de este
                                                  directorio deberia de estar subdirectorios de la forma:
                                                  path/D[yyyy][ddd]/expLabel/P[yyyy][ddd][sss][ext]
                         startDateTime       :    Fecha inicial. Rechaza todos los archivos donde
                                                  file end time < startDatetime (obejto datetime.datetime)
                         endDateTime         :    Fecha final. Si no es None, rechaza todos los archivos donde
                                                  file end time < startDatetime (obejto datetime.datetime)
                         set                 :    Set del primer archivo a leer. Por defecto None
                         expLabel            :    Nombre del subdirectorio de datos.  Por defecto ""
                         ext                 :    Extension de los archivos a leer. Por defecto .r
                         online              :    Si es == a 0 entonces busca files que cumplan con las condiciones dadas
                     Return:
 :    Si no encuentra files que cumplan con las condiciones dadas
 :    Si encuentra files que cumplan con las condiciones dadas
                     Affected:
                         self.startUTCSeconds
                         self.endUTCSeconds
                         self.startYear
                         self.endYear
                         self.startDoy
                         self.endDoy
                         self.pathList
                         self.filenameList
                         self.online
                     """
                     if path == None:
                         raise ValueError, "The path is not valid"
                     if ext == None:
                         ext = self.ext
                     if dataOutObj == None:
                         dataOutObj = self.createObjByDefault()
                     self.dataOutObj = dataOutObj
                     if online:
                         print "Searching files ..."
                         doypath, file, year, doy, set = self.__searchFilesOnLine(path, startDateTime, endDateTime, expLabel, ext)
                         if not(doypath):
                             for nTries in range( self.nTries ):
                                 print '\tWaiting %0.2f sec for valid file in %s: try %02d ...' % (self.delay, path, nTries+1)
                                 time.sleep( self.delay )
                                 doypath, file, year, doy, set = self.__searchFilesOnLine(path, startDateTime, endDateTime, expLabel, ext)
                                 if doypath:
                                     break
                         if not(doypath):
                             print "There 'isn't valied files in %s" % path
                             return None
                         self.year = year
                         self.doy  = doy
                         self.set  = set - 1
                         self.path = path
                     else: # offline
                         pathList, filenameList = self.__searchFilesOffLine(path, startDateTime, endDateTime, set, expLabel, ext)
                         if not(pathList):
                             print "No files in range: %s - %s" %(startDateTime.ctime(), endDateTime.ctime())
                             return None
                         self.fileIndex = -1
                         self.pathList = pathList
                         self.filenameList = filenameList
                     self.online = online
                     self.ext = ext
                     ext = ext.lower()
                     if not( self.setNextFile() ):
                         if (startDateTime != None) and (endDateTime != None):
                             print "No files in range: %s - %s" %(startDateTime.ctime(), endDateTime.ctime())
                         elif startDateTime != None:
                             print "No files in : %s" % startDateTime.ctime()
                         else:
                             print "No files"
                         return None
                     if startDateTime != None:
                         self.startUTCSeconds = time.mktime(startDateTime.timetuple())
                         self.startYear = startDateTime.timetuple().tm_year
                         self.startDoy = startDateTime.timetuple().tm_yday
                     if endDateTime != None:
                         self.endUTCSeconds = time.mktime(endDateTime.timetuple())
                         self.endYear = endDateTime.timetuple().tm_year
                         self.endDoy = endDateTime.timetuple().tm_yday
                     #call fillHeaderValues() - to Data Object
                     self.updateDataHeader()
                     return self.dataOutObj
                 def __rdSystemHeader(self, fp=None):
                     if fp == None:
                         fp = self.fp
                     self.m_SystemHeader.read(fp)
                 def __rdRadarControllerHeader(self, fp=None):
                     if fp == None:
                         fp = self.fp
                     self.m_RadarControllerHeader.read(fp)
                 def __rdProcessingHeader(self, fp=None):
                     if fp == None:
                         fp = self.fp
                     self.m_ProcessingHeader.read(fp)
                 def __rdBasicHeader(self, fp=None):
                     if fp == None:
                         fp = self.fp
                     self.m_BasicHeader.read(fp)
                 def __readFirstHeader(self):
                     """
                     Lectura del First Header, es decir el Basic Header y el Long Header
                     Affected:
                         self.m_BasicHeader
                         self.m_SystemHeader
                         self.m_RadarControllerHeader
                         self.m_ProcessingHeader
                         self.firstHeaderSize
                         self.dataType
                         self.fileSizeByHeader
                         self.ippSeconds
                     Return:
                         None
                     """
                     self.__rdBasicHeader()
                     self.__rdSystemHeader()
                     self.__rdRadarControllerHeader()
                     self.__rdProcessingHeader()
                     self.firstHeaderSize = self.m_BasicHeader.size
                     datatype = int(numpy.log2((self.m_ProcessingHeader.processFlags & PROCFLAG.DATATYPE_MASK))-numpy.log2(PROCFLAG.DATATYPE_CHAR))
                     if datatype == 0:
                         datatype_str = numpy.dtype([('real','<i1'),('imag','<i1')])
                     elif datatype == 1:
                         datatype_str = numpy.dtype([('real','<i2'),('imag','<i2')])
                     elif datatype == 2:
                         datatype_str = numpy.dtype([('real','<i4'),('imag','<i4')])
                     elif datatype == 3:
                         datatype_str = numpy.dtype([('real','<i8'),('imag','<i8')])
                     elif datatype == 4:
                         datatype_str = numpy.dtype([('real','<f4'),('imag','<f4')])
                     elif datatype == 5:
                         datatype_str = numpy.dtype([('real','<f8'),('imag','<f8')])
                     else:
                         raise ValueError, 'Data type was not defined'
                     self.dataType = datatype_str
                     self.ippSeconds = 2 * 1000 * self.m_RadarControllerHeader.ipp / self.c
                     self.fileSizeByHeader = self.m_ProcessingHeader.dataBlocksPerFile * self.m_ProcessingHeader.blockSize + self.firstHeaderSize + self.basicHeaderSize*(self.m_ProcessingHeader.dataBlocksPerFile - 1)
                     self.dataOutObj.channelList = numpy.arange(self.m_SystemHeader.numChannels)
                     self.dataOutObj.channelIndexList = numpy.arange(self.m_SystemHeader.numChannels)
                     self.getBlockDimension()
                 def __setNewBlock(self):
                     """
                     Lee el Basic Header y posiciona le file pointer en la posicion inicial del bloque a leer
                     Affected:
                         self.m_BasicHeader
                         self.flagNoContinuousBlock
                         self.ns
                     Return:
 :    Si el file no tiene un Basic Header que pueda ser leido
 :    Si se pudo leer el Basic Header
                     """
                     if self.fp == None:
                         return 0
                     if self.flagIsNewFile:
                         return 1
                     self.lastUTTime = self.m_BasicHeader.utc
                     currentSize = self.fileSize - self.fp.tell()
                     neededSize = self.m_ProcessingHeader.blockSize + self.basicHeaderSize
                     #If there is enough data setting new data block
                     if ( currentSize >= neededSize ):
                         self.__rdBasicHeader()
                         return 1
                     #si es OnLine y ademas aun no se han leido un bloque completo entonces se espera por uno valido
                     if self.online and (self.nReadBlocks < self.m_ProcessingHeader.dataBlocksPerFile):
                         fpointer = self.fp.tell()
                         for nTries in range( self.nTries ):
                             #self.fp.close()
                             print "\tWaiting %0.2f seconds for the next block, try %03d ..." % (self.delay, nTries+1)
                             time.sleep( self.delay )
                             #self.fp = open( self.filename, 'rb' )
                             #self.fp.seek( fpointer )
                             self.fileSize = os.path.getsize( self.filename )
                             currentSize = self.fileSize - fpointer
                             if ( currentSize >= neededSize ):
                                 self.__rdBasicHeader()
                                 return 1
                     #Setting new file
                     if not( self.setNextFile() ):
                         return 0
                     deltaTime = self.m_BasicHeader.utc - self.lastUTTime # check this
                     self.flagResetProcessing = 0
                     if deltaTime > self.maxTimeStep:
                         self.flagResetProcessing = 1
                     return 1
                 def readNextBlock(self):
                     """
                     Establece un nuevo bloque de datos a leer y los lee, si es que no existiese
                     mas bloques disponibles en el archivo actual salta al siguiente.
                     Affected:
                         self.lastUTTime
                     Return: None
                     """
                     if not(self.__setNewBlock()):
                         return 0
                     if not(self.readBlock()):
                         return 0
                     return 1
                 def __setNextFileOnline(self):
                     """
                     Busca el siguiente file que tenga suficiente data para ser leida, dentro de un folder especifico, si
                     no encuentra un file valido espera un tiempo determinado y luego busca en los posibles n files
                     siguientes.
                     Affected:
                         self.flagIsNewFile
                         self.filename
                         self.fileSize
                         self.fp
                         self.set
                         self.flagNoMoreFiles
                     Return:
 : si luego de una busqueda del siguiente file valido este no pudo ser encontrado
 : si el file fue abierto con exito y esta listo a ser leido
                     Excepciones:
                         Si un determinado file no puede ser abierto
                     """
                     nFiles = 0
                     fileOk_flag = False
                     firstTime_flag = True
                     self.set += 1
                     #busca el 1er file disponible
                     file, filename = checkForRealPath( self.path, self.year, self.doy, self.set, self.ext )
                     if file:
                         if self.__verifyFile(file, False):
                             fileOk_flag = True
                     #si no encuentra un file entonces espera y vuelve a buscar
                     if not(fileOk_flag):
                         for nFiles in range(self.nFiles+1): #busco en los siguientes self.nFiles+1 files posibles
                             if firstTime_flag: #si es la 1era vez entonces hace el for self.nTries veces
                                 tries = self.nTries
                             else:
                                 tries = 1 #si no es la 1era vez entonces solo lo hace una vez
                             for nTries in range( tries ):
                                 if firstTime_flag:
                                     print "\tWaiting %0.2f sec for new \"%s\" file, try %03d ..." % ( self.delay, filename, nTries+1 )
                                     time.sleep( self.delay )
                                 else:
                                     print "\tSearching next \"%s%04d%03d%03d%s\" file ..." % (self.optchar, self.year, self.doy, self.set, self.ext)
                                 file, filename = checkForRealPath( self.path, self.year, self.doy, self.set, self.ext )
                                 if file:
                                     if self.__verifyFile(file):
                                         fileOk_flag = True
                                         break
                             if fileOk_flag:
                                 break
                             firstTime_flag = False
                             print "\tSkipping the file \"%s\" due to this file doesn't exist yet" % filename
                             self.set += 1
                             if nFiles == (self.nFiles-1): #si no encuentro el file buscado cambio de carpeta y busco en la siguiente carpeta
                                 self.set = 0
                                 self.doy += 1
                     if fileOk_flag:
                         self.fileSize = os.path.getsize( file )
                         self.filename = file
                         self.flagIsNewFile = 1
                         if self.fp != None: self.fp.close()
                         self.fp = open(file)
                         self.flagNoMoreFiles = 0
                         print 'Setting the file: %s' % file
                     else:
                         self.fileSize = 0
                         self.filename = None
                         self.flagIsNewFile = 0
                         self.fp = None
                         self.flagNoMoreFiles = 1
                         print 'No more Files'
                     return fileOk_flag
                 def __setNextFileOffline(self):
                     """
                     Busca el siguiente file dentro de un folder que tenga suficiente data para ser leida
                     Affected:
                         self.flagIsNewFile
                         self.fileIndex
                         self.filename
                         self.fileSize
                         self.fp
                     Return:
 : si un determinado file no puede ser abierto
 : si el file fue abierto con exito
                     Excepciones:
                         Si un determinado file no puede ser abierto
                     """
                     idFile = self.fileIndex
                     while(True):
                         idFile += 1
                         if not(idFile < len(self.filenameList)):
                             self.flagNoMoreFiles = 1
                             print 'No more Files'
                             return 0
                         filename = self.filenameList[idFile]
                         if not(self.__verifyFile(filename)):
                             continue
                         fileSize = os.path.getsize(filename)
                         fp = open(filename,'rb')
                         break
                     self.flagIsNewFile = 1
                     self.fileIndex = idFile
                     self.filename = filename
                     self.fileSize = fileSize
                     self.fp = fp
                     print 'Setting the file: %s'%self.filename
                     return 1
                 def setNextFile(self):
                     """
                     Determina el siguiente file a leer y si hay uno disponible lee el First Header
                     Affected:
                         self.m_BasicHeader
                         self.m_SystemHeader
                         self.m_RadarControllerHeader
                         self.m_ProcessingHeader
                         self.firstHeaderSize
                     Return:
 :    Si no hay files disponibles
 :    Si hay mas files disponibles
                     """
                     if self.fp != None:
                         self.fp.close()
                     if self.online:
                         newFile = self.__setNextFileOnline()
                     else:
                         newFile = self.__setNextFileOffline()
                     if not(newFile):
                         return 0
                     self.__readFirstHeader()
                     self.nReadBlocks = 0
                     return 1
                 def __searchFilesOnLine(self, path, startDateTime=None, endDateTime=None, expLabel = "", ext = None):
                     """
                     Busca el ultimo archivo de la ultima carpeta (determinada o no por startDateTime) y
                     devuelve el archivo encontrado ademas de otros datos.
                     Input:
                         path             :    carpeta donde estan contenidos los files que contiene data
                         startDateTime    :    punto especifico en el tiempo del cual se requiere la data
                         ext              :    extension de los files
                     Return:
                         year        :    el anho
                         doy         :    el numero de dia del anho
                         set         :    el set del archivo
                         filename    :    el ultimo file de una determinada carpeta
                         directory   :    eL directorio donde esta el file encontrado
                     """
                     dirList = []
                     pathList = []
                     directory = None
                     for thisPath in os.listdir(path):
                         if os.path.isdir(os.path.join(path,thisPath)):
                             dirList.append(thisPath)
                     if not(dirList):
                         return None, None, None, None, None
                     dirList = sorted( dirList, key=str.lower )
                     if startDateTime:
                         thisDateTime = startDateTime
                         if endDateTime == None: endDateTime = startDateTime
                         while(thisDateTime <= endDateTime):
                             year = thisDateTime.timetuple().tm_year
                             doy = thisDateTime.timetuple().tm_yday
                             match = fnmatch.filter(dirList, '?' + '%4.4d%3.3d' % (year,doy))
                             if len(match) == 0:
                                 thisDateTime += datetime.timedelta(1)
                                 continue
                             pathList.append(os.path.join(path,match[0], expLabel))
                             thisDateTime += datetime.timedelta(1)
                         if not(pathList):
                             print "\tNo files in range: %s - %s" %(startDateTime.ctime(), endDateTime.ctime())
                             return None, None, None, None, None
                         directory = pathList[0]
                     else:
                         directory = dirList[-1]
                         directory = os.path.join(path,directory)
                     filename = getlastFileFromPath(directory, ext)
                     if not(filename):
                         return None, None, None, None, None
                     if not(self.__verifyFile(os.path.join(directory, filename))):
                         return None, None, None, None, None
                     year = int( filename[1:5] )
                     doy  = int( filename[5:8] )
                     set  = int( filename[8:11] )
                     return directory, filename, year, doy, set
                 def __searchFilesOffLine(self, path, startDateTime, endDateTime, set=None, expLabel = "", ext = ".r"):
                     """
                     Realiza una busqueda de los archivos que coincidan con los parametros
                     especificados y se encuentren ubicados en el path indicado. Para realizar una busqueda
                     correcta la estructura de directorios debe ser la siguiente:
                     ...path/D[yyyy][ddd]/expLabel/D[yyyy][ddd][sss].ext
                     [yyyy]: anio
                     [ddd] : dia del anio
                     [sss] : set del archivo
                     Inputs:
                         path           :    Directorio de datos donde se realizara la busqueda. Todos los
                                             ficheros que concidan con el criterio de busqueda seran
                                             almacenados en una lista y luego retornados.
                         startDateTime  :    Fecha inicial. Rechaza todos los archivos donde
                                             file end time < startDateTime (obejto datetime.datetime)
                         endDateTime    :    Fecha final. Rechaza todos los archivos donde
                                             file start time > endDateTime (obejto datetime.datetime)
                         set            :    Set del primer archivo a leer. Por defecto None
                         expLabel       :    Nombre del subdirectorio de datos.  Por defecto ""
                         ext            :    Extension de los archivos a leer. Por defecto .r
                     Return:
                         (pathList, filenameList)
                         pathList        :    Lista de directorios donde se encontraron archivos dentro
                                              de los parametros especificados
                         filenameList    :    Lista de archivos (ruta completa) que coincidieron con los
                                              parametros especificados.
                     Variables afectadas:
                         self.filenameList:    Lista de archivos (ruta completa) que la clase utiliza
                                               como fuente para leer los bloque de datos, si se termina
                                               de leer todos los bloques de datos de un determinado
                                               archivo se pasa al siguiente archivo de la lista.
                     Excepciones:
                     """
                     print "Searching files ..."
                     dirList = []
                     for thisPath in os.listdir(path):
                         if os.path.isdir(os.path.join(path,thisPath)):
                             dirList.append(thisPath)
                     if not(dirList):
                         return None, None
                     pathList = []
                     thisDateTime = startDateTime
                     while(thisDateTime <= endDateTime):
                         year = thisDateTime.timetuple().tm_year
                         doy = thisDateTime.timetuple().tm_yday
                         match = fnmatch.filter(dirList, '?' + '%4.4d%3.3d' % (year,doy))
                         if len(match) == 0:
                             thisDateTime += datetime.timedelta(1)
                             continue
                         pathList.append(os.path.join(path,match[0],expLabel))
                         thisDateTime += datetime.timedelta(1)
                     startUtSeconds = time.mktime(startDateTime.timetuple())
                     endUtSeconds = time.mktime(endDateTime.timetuple())
                     filenameList = []
                     for thisPath in pathList:
                         fileList = glob.glob1(thisPath, "*%s" %ext)
                         fileList.sort()
                         for file in fileList:
                             filename = os.path.join(thisPath,file)
                             if isThisFileinRange(filename, startUtSeconds, endUtSeconds):
                                 filenameList.append(filename)
                     if not(filenameList):
                         return None, None
                     self.filenameList = filenameList
                     return pathList, filenameList
                 def __verifyFile(self, filename, msgFlag=True):
                     """
                     Verifica que el filename tenga data valida, para ello leo el FirstHeader del file
                     Return:
 :    file no valido para ser leido
 :    file valido para ser leido
                     """
                     msg = None
                     try:
                         fp = open( filename,'rb' ) #lectura binaria
                         currentPosition = fp.tell()
                     except:
                         if msgFlag:
                             print "The file %s can't be opened" % (filename)
                         return False
                     neededSize = self.m_ProcessingHeader.blockSize + self.firstHeaderSize
                     if neededSize == 0:
                         m_BasicHeader = BasicHeader()
                         m_SystemHeader = SystemHeader()
                         m_RadarControllerHeader = RadarControllerHeader()
                         m_ProcessingHeader = ProcessingHeader()
                         try:
                             if not( m_BasicHeader.read(fp) ): raise ValueError
                             if not( m_SystemHeader.read(fp) ): raise ValueError
                             if not( m_RadarControllerHeader.read(fp) ): raise ValueError
                             if not( m_ProcessingHeader.read(fp) ): raise ValueError
                             data_type = int(numpy.log2((m_ProcessingHeader.processFlags & PROCFLAG.DATATYPE_MASK))-numpy.log2(PROCFLAG.DATATYPE_CHAR))
                             neededSize = m_ProcessingHeader.blockSize + m_BasicHeader.size
                         except:
                             if msgFlag:
                                 print "\tThe file %s is empty or it hasn't enough data" % filename
                             fp.close()
                             return False
                     else:
                         msg = "\tSkipping the file %s due to it hasn't enough data" %filename
                     fp.close()
                     fileSize = os.path.getsize(filename)
                     currentSize = fileSize - currentPosition
                     if currentSize < neededSize:
                         if msgFlag and (msg != None):
                             print msg #print"\tSkipping the file %s due to it hasn't enough data" %filename
                         return False
                     return True
                 def updateDataHeader(self):
                     self.dataOutObj.m_BasicHeader = self.m_BasicHeader.copy()
                     self.dataOutObj.m_ProcessingHeader = self.m_ProcessingHeader.copy()
                     self.dataOutObj.m_RadarControllerHeader = self.m_RadarControllerHeader.copy()
                     self.dataOutObj.m_SystemHeader = self.m_SystemHeader.copy()
                     self.dataOutObj.dataType = self.dataType
                     self.dataOutObj.updateObjFromHeader()
             class JRODataWriter(JRODataIO):
                 """
                 Esta clase permite escribir datos a archivos procesados (.r o ,pdata). La escritura
                 de los datos siempre se realiza por bloques.
                 """
                 nWriteBlocks = 0
                 setFile = None
                 def __init__(self, dataOutObj=None):
                     raise ValueError, "Not implemented"
                 def hasAllDataInBuffer(self):
                     raise ValueError, "Not implemented"
                 def setBlockDimension(self):
                     raise ValueError, "Not implemented"
                 def writeBlock(self):
                     raise ValueError, "No implemented"
                 def putData(self):
                     raise ValueError, "No implemented"
                 def __writeFirstHeader(self):
                     """
                     Escribe el primer header del file es decir el Basic header y el Long header (SystemHeader, RadarControllerHeader, ProcessingHeader)
                     Affected:
                         __dataType
                     Return:
                         None
                     """
                     self.__writeBasicHeader()
                     self.__wrSystemHeader()
                     self.__wrRadarControllerHeader()
                     self.__wrProcessingHeader()
                     self.dataType = self.dataOutObj.dataType
                 def __writeBasicHeader(self, fp=None):
                     """
                     Escribe solo el Basic header en el file creado
                     Return:
                         None
                     """
                     if fp == None:
                         fp = self.fp
                     self.dataOutObj.m_BasicHeader.write(fp)
                 def __wrSystemHeader(self, fp=None):
                     """
                     Escribe solo el System header en el file creado
                     Return:
                         None
                     """
                     if fp == None:
                         fp = self.fp
                     self.dataOutObj.m_SystemHeader.write(fp)
                 def __wrRadarControllerHeader(self, fp=None):
                     """
                     Escribe solo el RadarController header en el file creado
                     Return:
                         None
                     """
                     if fp == None:
                         fp = self.fp
                     self.dataOutObj.m_RadarControllerHeader.write(fp)
                 def __wrProcessingHeader(self, fp=None):
                     """
                     Escribe solo el Processing header en el file creado
                     Return:
                         None
                     """
                     if fp == None:
                         fp = self.fp
                     self.dataOutObj.m_ProcessingHeader.write(fp)
                 def setNextFile(self):
                     """
                     Determina el siguiente file que sera escrito
                     Affected:
                         self.filename
                         self.subfolder
                         self.fp
                         self.setFile
                         self.flagIsNewFile
                     Return:
 :    Si el archivo no puede ser escrito
 :    Si el archivo esta listo para ser escrito
                     """
                     ext = self.ext
                     path = self.path
                     if self.fp != None:
                         self.fp.close()
                     timeTuple = time.localtime( self.dataOutObj.m_BasicHeader.utc )
                     subfolder = 'D%4.4d%3.3d' % (timeTuple.tm_year,timeTuple.tm_yday)
                     doypath = os.path.join( path, subfolder )
                     if not( os.path.exists(doypath) ):
                         os.mkdir(doypath)
                         self.setFile = -1 #inicializo mi contador de seteo
                     else:
                         filesList = os.listdir( doypath )
                         if len( filesList ) > 0:
                             filesList = sorted( filesList, key=str.lower )
                             filen = filesList[-1]
                             # el filename debera tener el siguiente formato
                             # 0 1234 567 89A BCDE (hex)
                             # x YYYY DDD SSS .ext
                             if isNumber( filen[8:11] ):
                                 self.setFile = int( filen[8:11] ) #inicializo mi contador de seteo al seteo del ultimo file
                             else:
                                 self.setFile = -1
                         else:
                             self.setFile = -1 #inicializo mi contador de seteo
                     setFile = self.setFile
                     setFile += 1
                     file = '%s%4.4d%3.3d%3.3d%s' % (self.optchar,
                                                     timeTuple.tm_year,
                                                     timeTuple.tm_yday,
                                                     setFile,
                                                     ext )
                     filename = os.path.join( path, subfolder, file )
                     fp = open( filename,'wb' )
                     self.nWriteBlocks = 0
                     #guardando atributos
                     self.filename = filename
                     self.subfolder = subfolder
                     self.fp = fp
                     self.setFile = setFile
                     self.flagIsNewFile = 1
                     print 'Writing the file: %s'%self.filename
                     self.__writeFirstHeader()
                     return 1
                 def __setNewBlock(self):
                     """
                     Si es un nuevo file escribe el First Header caso contrario escribe solo el Basic Header
                     Return:
 :    si no pudo escribir nada
 :    Si escribio el Basic el First Header
                     """
                     if self.fp == None:
                         self.setNextFile()
                     if self.flagIsNewFile:
                         return 1
                     if self.nWriteBlocks < self.m_ProcessingHeader.dataBlocksPerFile:
                         self.__writeBasicHeader()
                         return 1
                     if not( self.setNextFile() ):
                         return 0
                     return 1
                 def writeNextBlock(self):
                     """
                     Selecciona el bloque siguiente de datos y los escribe en un file
                     Return:
 :    Si no hizo pudo escribir el bloque de datos
 :    Si no pudo escribir el bloque de datos
                     """
                     if not( self.__setNewBlock() ):
                         return 0
                     self.writeBlock()
                     return 1
                 def getDataHeader(self):
                     """
                     Obtiene una copia del First Header
                     Affected:
                         self.m_BasicHeader
                         self.m_SystemHeader
                         self.m_RadarControllerHeader
                         self.m_ProcessingHeader
                         self.dataType
                     Return:
                         None
                     """
                     self.dataOutObj.updateHeaderFromObj()
                     self.m_BasicHeader = self.dataOutObj.m_BasicHeader.copy()
                     self.m_SystemHeader = self.dataOutObj.m_SystemHeader.copy()
                     self.m_RadarControllerHeader = self.dataOutObj.m_RadarControllerHeader.copy()
                     self.m_ProcessingHeader = self.dataOutObj.m_ProcessingHeader.copy()
                     self.dataType = self.dataOutObj.dataType
                 def setup(self, path, set=0, ext=None):
                     """
                     Setea el tipo de formato en la cual sera guardada la data y escribe el First Header
                     Inputs:
                         path      :    el path destino en el cual se escribiran los files a crear
                         format    :    formato en el cual sera salvado un file
                         set       :    el setebo del file
                     Return:
 :    Si no realizo un buen seteo
 :    Si realizo un buen seteo
                     """
                     if ext == None:
                         ext = self.ext
                     ext = ext.lower()
                     self.path = path
                     self.setFile = set - 1
                     self.ext = ext
                     #self.format = format
                     self.getDataHeader()
                     self.setBlockDimension()
                     if not( self.setNextFile() ):
                         print "There isn't a next file"
                         return 0
                     return 1

schainpy/Processing/VoltageProcessor.py +3 -1

             '''
             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.Voltage import Voltage
             from IO.VoltageIO import VoltageWriter
             from Graphics.VoltagePlot import Osciloscope
             from Graphics.VoltagePlot import RTI
             class VoltageProcessor:
                 '''
                 classdocs
                 '''
                 dataInObj = None
                 dataOutObj = None
                 integratorObjIndex = None
                 decoderObjIndex = None
                 profSelectorObjIndex = None
                 writerObjIndex = None
                 plotterObjIndex = None
                 flipIndex = None
                 integratorObjList = []
                 decoderObjList = []
                 profileSelectorObjList = []
                 writerObjList = []
                 plotterObjList = []
                 m_Voltage= Voltage()
                 def __init__(self):
                     '''
                     Constructor
                     '''
                     self.integratorObjIndex = None
                     self.decoderObjIndex = None
                     self.profSelectorObjIndex = None
                     self.writerObjIndex = None
                     self.plotterObjIndex = None
                     self.flipIndex = 1
                     self.integratorObjList = []
                     self.decoderObjList = []
                     self.profileSelectorObjList = []
                     self.writerObjList = []
                     self.plotterObjList = []
                 def setup(self, dataInObj=None, dataOutObj=None):
                     self.dataInObj = dataInObj
                     if dataOutObj == None:
                         dataOutObj = Voltage()
                     dataOutObj.copy(dataInObj)
                     self.dataOutObj = dataOutObj
                     return self.dataOutObj
                 def init(self):
                     self.integratorObjIndex = 0
                     self.decoderObjIndex = 0
                     self.profSelectorObjIndex = 0
                     self.writerObjIndex = 0
                     self.plotterObjIndex = 0
                     self.dataOutObj.copy(self.dataInObj)
                     if self.profSelectorObjIndex != None:
                         for profSelObj in self.profileSelectorObjList:
                             profSelObj.incIndex()
                 def addWriter(self, wrpath):
                     objWriter = VoltageWriter(self.dataOutObj)
                     objWriter.setup(wrpath)
                     self.writerObjList.append(objWriter)
                 def addRti(self,index=None):
                     if index==None:
                         index = self.plotterObjIndex
                     plotObj = RTI(self.dataOutObj, index)
                     self.plotterObjList.append(plotObj)
                 def addPlotter(self, index=None):
                     if index==None:
                         index = self.plotterObjIndex
                     plotObj = Osciloscope(self.dataOutObj, index)
                     self.plotterObjList.append(plotObj)
                 def addIntegrator(self, N,timeInterval):
                     objCohInt = CoherentIntegrator(N,timeInterval)
                     self.integratorObjList.append(objCohInt)
                 def addDecoder(self, code, ncode, nbaud):
                     objDecoder = Decoder(code,ncode,nbaud)
                     self.decoderObjList.append(objDecoder)
                 def addProfileSelector(self, nProfiles):
                     objProfSelector = ProfileSelector(nProfiles)
                     self.profileSelectorObjList.append(objProfSelector)
                 def writeData(self,wrpath):
                     if self.dataOutObj.flagNoData:
                         return 0
                     if len(self.writerObjList) <= self.writerObjIndex:
                         self.addWriter(wrpath)
                     self.writerObjList[self.writerObjIndex].putData()
                     self.writerObjIndex += 1
                 def addScope(self,index=None):
                     if index==None:
                         index = self.plotterObjIndex
                     plotObj = Osciloscope(self.dataOutObj, index)
                     self.plotterObjList.append(plotObj)
                 def plotScope(self,
                             xmin=None,
                             xmax=None,
                             ymin=None,
                             ymax=None,
                             titleList=None,
                             xlabelList=None,
                             ylabelList=None,
                             winTitle='',
                             type="power",
                             index=None):
                     if self.dataOutObj.flagNoData:
                         return 0
                     if len(self.plotterObjList) <= self.plotterObjIndex:
                         self.addScope(index)
                     self.plotterObjList[self.plotterObjIndex].plotData(xmin,
                                                                        xmax,
                                                                        ymin,
                                                                        ymax,
                                                                        titleList,
                                                                        xlabelList,
                                                                        ylabelList,
                                                                        winTitle,
                                                                        type)
                     self.plotterObjIndex += 1
                 def plotRti(self,
                             xmin=None,
                             xmax=None,
                             ymin=None,
                             ymax=None,
                             zmin=None,
                             zmax=None,
                             titleList=None,
                             xlabelList=None,
                             ylabelList=None,
                             winTitle='',
                             timezone='lt',
                             npoints=1000.0,
                             colormap="br_green",
                             showColorbar=True,
                             showPowerProfile=False,
                             XAxisAsTime=True,
+                            save=False,
                             index=None):
                     if self.dataOutObj.flagNoData:
                         return 0
                     if len(self.plotterObjList) <= self.plotterObjIndex:
                         self.addRti(index)
                     self.plotterObjList[self.plotterObjIndex].plotData(xmin,
                                                                        xmax,
                                                                        ymin,
                                                                        ymax,
                                                                        zmin,
                                                                        zmax,
                                                                        titleList,
                                                                        xlabelList,
                                                                        ylabelList,
                                                                        winTitle,
                                                                        timezone,
                                                                        npoints,
                                                                        colormap,
                                                                        showColorbar,
                                                                        showPowerProfile,
-                                                                       XAxisAsTime)
+                                                                       XAxisAsTime,
+                                                                       save)
                     self.plotterObjIndex += 1
                 def plotData(self,xmin=None, xmax=None, ymin=None, ymax=None, type='iq', winTitle='', index=None):
                     if self.dataOutObj.flagNoData:
                         return 0
                     if len(self.plotterObjList) <= self.plotterObjIndex:
                         self.addPlotter(index)
                     self.plotterObjList[self.plotterObjIndex].plotData(xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax,type=type, winTitle=winTitle)
                     self.plotterObjIndex += 1
                 def integrator(self, N=None, timeInterval=None):
                     if self.dataOutObj.flagNoData:
                         return 0
                     if len(self.integratorObjList) <= self.integratorObjIndex:
                         self.addIntegrator(N,timeInterval)
                     myCohIntObj = self.integratorObjList[self.integratorObjIndex]
                     myCohIntObj.exe(data=self.dataOutObj.data,timeOfData=self.dataOutObj.m_BasicHeader.utc)
                     if myCohIntObj.isReady:
                         self.dataOutObj.data = myCohIntObj.data
                         self.dataOutObj.nAvg = myCohIntObj.navg
                         self.dataOutObj.m_ProcessingHeader.coherentInt *= myCohIntObj.navg
                         #print "myCohIntObj.navg: ",myCohIntObj.navg
                         self.dataOutObj.flagNoData = False
                     else:
                         self.dataOutObj.flagNoData = True
                     self.integratorObjIndex += 1
                 def decoder(self,code=None,type = 0):
                     if self.dataOutObj.flagNoData:
                         return 0
                     if code == None:
                         code = self.dataOutObj.m_RadarControllerHeader.code
                     ncode, nbaud = code.shape
                     if len(self.decoderObjList) <= self.decoderObjIndex:
                         self.addDecoder(code,ncode,nbaud)
                     myDecodObj = self.decoderObjList[self.decoderObjIndex]
                     myDecodObj.exe(data=self.dataOutObj.data,type=type)
                     if myDecodObj.flag:
                         self.dataOutObj.data = myDecodObj.data
                         self.dataOutObj.flagNoData = False
                     else:
                         self.dataOutObj.flagNoData = True
                     self.decoderObjIndex += 1
                 def filterByHei(self, window):
                     if window == None:
                         window = self.dataOutObj.m_RadarControllerHeader.txA / self.dataOutObj.m_ProcessingHeader.deltaHeight[0]
                     newdelta = self.dataOutObj.m_ProcessingHeader.deltaHeight[0] * window
                     dim1 = self.dataOutObj.data.shape[0]
                     dim2 = self.dataOutObj.data.shape[1]
                     r = dim2 % window
                     buffer = self.dataOutObj.data[:,0:dim2-r]
                     buffer = buffer.reshape(dim1,dim2/window,window)
                     buffer = numpy.sum(buffer,2)
                     self.dataOutObj.data = buffer
                     self.dataOutObj.m_ProcessingHeader.deltaHeight = newdelta
                     self.dataOutObj.m_ProcessingHeader.numHeights = buffer.shape[1]
                     self.dataOutObj.nHeights = self.dataOutObj.m_ProcessingHeader.numHeights
                     #self.dataOutObj.heightList es un numpy.array
                     self.dataOutObj.heightList = numpy.arange(self.dataOutObj.m_ProcessingHeader.firstHeight[0],newdelta*self.dataOutObj.nHeights,newdelta)
                 def deFlip(self):
                     self.dataOutObj.data *= self.flipIndex
                     self.flipIndex *= -1.
                 def selectChannels(self, channelList):
                     pass
                 def selectChannelsByIndex(self, channelIndexList):
                     """
                     Selecciona un bloque de datos en base a canales segun el channelIndexList
                     Input:
                         channelIndexList    :    lista sencilla de canales a seleccionar por ej. [2,3,7]
                     Affected:
                         self.dataOutObj.data
                         self.dataOutObj.channelIndexList
                         self.dataOutObj.nChannels
                         self.dataOutObj.m_ProcessingHeader.totalSpectra
                         self.dataOutObj.m_SystemHeader.numChannels
                         self.dataOutObj.m_ProcessingHeader.blockSize
                     Return:
                         None
                     """
                     if self.dataOutObj.flagNoData:
                         return 0
                     for channel in channelIndexList:
                         if channel not in self.dataOutObj.channelIndexList:
                             raise ValueError, "The value %d in channelIndexList is not valid" %channel
                     nChannels = len(channelIndexList)
                     data = self.dataOutObj.data[channelIndexList,:]
                     self.dataOutObj.data = data
                     self.dataOutObj.channelIndexList = channelIndexList
                     self.dataOutObj.nChannels = nChannels
                     self.dataOutObj.m_ProcessingHeader.totalSpectra = 0
                     self.dataOutObj.m_SystemHeader.numChannels = nChannels
                     self.dataOutObj.m_ProcessingHeader.blockSize = data.size
                     return 1
                 def selectHeightsByValue(self, minHei, maxHei):
                     """
                     Selecciona un bloque de datos en base a un grupo de valores de alturas segun el rango
                     minHei <= height <= maxHei
                     Input:
                         minHei    :    valor minimo de altura a considerar
                         maxHei    :    valor maximo de altura a considerar
                     Affected:
                         Indirectamente son cambiados varios valores a travez del metodo selectHeightsByIndex
                     Return:
 si el metodo se ejecuto con exito caso contrario devuelve 0
                     """
                     if self.dataOutObj.flagNoData:
                         return 0
                     if (minHei < self.dataOutObj.heightList[0]) or (minHei > maxHei):
                         raise ValueError, "some value in (%d,%d) is not valid" % (minHei, maxHei)
                     if (maxHei > self.dataOutObj.heightList[-1]):
                         raise ValueError, "some value in (%d,%d) is not valid" % (minHei, maxHei)
                     minIndex = 0
                     maxIndex = 0
                     data = self.dataOutObj.heightList
                     for i,val in enumerate(data):
                         if val < minHei:
                             continue
                         else:
                             minIndex = i;
                             break
                     for i,val in enumerate(data):
                         if val <= maxHei:
                             maxIndex = i;
                         else:
                             break
                     self.selectHeightsByIndex(minIndex, maxIndex)
                     return 1
                 def selectHeightsByIndex(self, minIndex, maxIndex):
                     """
                     Selecciona un bloque de datos en base a un grupo indices de alturas segun el rango
                     minIndex <= index <= maxIndex
                     Input:
                         minIndex    :    valor de indice minimo de altura a considerar
                         maxIndex    :    valor de indice maximo de altura a considerar
                     Affected:
                         self.dataOutObj.data
                         self.dataOutObj.heightList
                         self.dataOutObj.nHeights
                         self.dataOutObj.m_ProcessingHeader.blockSize
                         self.dataOutObj.m_ProcessingHeader.numHeights
                         self.dataOutObj.m_ProcessingHeader.firstHeight
                         self.dataOutObj.m_RadarControllerHeader
                     Return:
 si el metodo se ejecuto con exito caso contrario devuelve 0
                     """
                     if self.dataOutObj.flagNoData:
                         return 0
                     if (minIndex < 0) or (minIndex > maxIndex):
                         raise ValueError, "some value in (%d,%d) is not valid" % (minIndex, maxIndex)
                     if (maxIndex >= self.dataOutObj.nHeights):
                         raise ValueError, "some value in (%d,%d) is not valid" % (minIndex, maxIndex)
                     nHeights = maxIndex - minIndex + 1
                     #voltage
                     data = self.dataOutObj.data[:,minIndex:maxIndex+1]
                     firstHeight = self.dataOutObj.heightList[minIndex]
                     self.dataOutObj.data = data
                     self.dataOutObj.heightList = self.dataOutObj.heightList[minIndex:maxIndex+1]
                     self.dataOutObj.nHeights = nHeights
                     self.dataOutObj.m_ProcessingHeader.blockSize = data.size
                     self.dataOutObj.m_ProcessingHeader.numHeights = nHeights
                     self.dataOutObj.m_ProcessingHeader.firstHeight = firstHeight
                     self.dataOutObj.m_RadarControllerHeader.numHeights = nHeights
                     return 1
                 def selectProfilesByValue(self,indexList, nProfiles):
                     if self.dataOutObj.flagNoData:
                         return 0
                     if self.profSelectorObjIndex >= len(self.profileSelectorObjList):
                         self.addProfileSelector(nProfiles)
                     profileSelectorObj = self.profileSelectorObjList[self.profSelectorObjIndex]
                     if not(profileSelectorObj.isProfileInList(indexList)):
                         self.dataOutObj.flagNoData = True
                         self.profSelectorObjIndex += 1
                         return 0
                     self.dataOutObj.flagNoData = False
                     self.profSelectorObjIndex += 1
                     return 1
                 def selectProfilesByIndex(self, minIndex, maxIndex, nProfiles):
                     """
                     Selecciona un bloque de datos en base a un grupo indices de perfiles segun el rango
                     minIndex <= index <= maxIndex
                     Input:
                         minIndex    :    valor de indice minimo de perfil a considerar
                         maxIndex    :    valor de indice maximo de perfil a considerar
                         nProfiles   :    numero de profiles
                     Affected:
                         self.dataOutObj.flagNoData
                         self.profSelectorObjIndex
                     Return:
 si el metodo se ejecuto con exito caso contrario devuelve 0
                     """
                     if self.dataOutObj.flagNoData:
                         return 0
                     if self.profSelectorObjIndex >= len(self.profileSelectorObjList):
                         self.addProfileSelector(nProfiles)
                     profileSelectorObj = self.profileSelectorObjList[self.profSelectorObjIndex]
                     if not(profileSelectorObj.isProfileInRange(minIndex, maxIndex)):
                         self.dataOutObj.flagNoData = True
                         self.profSelectorObjIndex += 1
                         return 0
                     self.dataOutObj.flagNoData = False
                     self.profSelectorObjIndex += 1
                     return 1
                 def selectNtxs(self, ntx):
                     pass
             class Decoder:
                 data = None
                 profCounter = 1
                 nCode = None
                 nBaud = None
                 codeIndex = 0
                 code = None
                 flag = False
                 def __init__(self,code, ncode, nbaud):
                     self.data = None
                     self.profCounter = 1
                     self.nCode = ncode
                     self.nBaud = nbaud
                     self.codeIndex = 0
                     self.code = code #this is a List
                     self.flag = False
                 def exe(self, data, ndata=None, type = 0):
                     if ndata == None: ndata = data.shape[1]
                     if type == 0:
                         self.convolutionInFreq(data,ndata)
                     if type == 1:
                         self.convolutionInTime(data, ndata)
                 def convolutionInFreq(self,data, ndata):
                     newcode = numpy.zeros(ndata)
                     newcode[0:self.nBaud] = self.code[self.codeIndex]
                     self.codeIndex += 1
                     fft_data = numpy.fft.fft(data, axis=1)
                     fft_code = numpy.conj(numpy.fft.fft(newcode))
                     fft_code = fft_code.reshape(1,len(fft_code))
                     conv = fft_data.copy()
                     conv.fill(0)
                     conv = fft_data*fft_code
                     self.data = numpy.fft.ifft(conv,axis=1)
                     self.flag = True
                     if self.profCounter == self.nCode:
                         self.profCounter = 0
                         self.codeIndex = 0
                     self.profCounter += 1
                 def convolutionInTime(self, data, ndata):
                     nchannel = data.shape[1]
                     newcode = self.code[self.codeIndex]
                     self.codeIndex += 1
                     conv = data.copy()
                     for i in range(nchannel):
                         conv[i,:] = numpy.correlate(data[i,:], newcode, 'same')
                     self.data = conv
                     self.flag = True
                     if self.profCounter == self.nCode:
                         self.profCounter = 0
                         self.codeIndex = 0
                     self.profCounter += 1
             class CoherentIntegrator:
                 integ_counter = None
                 data = None
                 navg = None
                 buffer = None
                 nCohInt = None
                 def __init__(self, N=None,timeInterval=None):
                     self.data = None
                     self.navg = None
                     self.buffer = None
                     self.timeOut = None
                     self.exitCondition = False
                     self.isReady = False
                     self.nCohInt = N
                     self.integ_counter = 0
                     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
                 def exe(self, data, timeOfData):
                     if self.timeFlag:
                         if self.timeOut == None:
                             self.timeOut = timeOfData + self.timeIntervalInSeconds
                         if timeOfData < self.timeOut:
                             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.nCohInt:
                             if self.buffer == None:
                                 self.buffer = data
                             else:
                                 self.buffer = self.buffer + data
                             self.integ_counter += 1
                         if self.integ_counter == self.nCohInt:
                             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
                             self.timeOut = timeOfData + self.timeIntervalInSeconds
                     else:
                         self.isReady = False
             class ProfileSelector:
                 profileIndex = None
                 # Tamanho total de los perfiles
                 nProfiles = None
                 def __init__(self, nProfiles):
                     self.profileIndex = 0
                     self.nProfiles = nProfiles
                 def incIndex(self):
                     self.profileIndex += 1
                     if self.profileIndex >= self.nProfiles:
                         self.profileIndex = 0
                 def isProfileInRange(self, minIndex, maxIndex):
                     if self.profileIndex < minIndex:
                         return False
                     if self.profileIndex > maxIndex:
                         return False
                     return True
                 def isProfileInList(self, profileList):
                     if self.profileIndex not in profileList:
                         return False
                     return True
   No newline at end of file

General Comments 0

Write
Preview

You need to be logged in to leave comments. Login now

No TODOs yet

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages