##// END OF EJS Templates
schain
RSS
Miguel Valdez - - Load All Authors

File last commit:

r27:fb9ede38dd50
r98:e2f0a078c091
Show More
BaseGraph.py
863 lines | 25.8 KiB | text/x-python | PythonLexer
/ schainpy / Graphics / BaseGraph.py
History | Source | Raw |Copy content |Copy permalink
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17 """
Created on Feb 7, 2012
@autor $Author$
@version $Id$
"""
import numpy
import plplot
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 def cmap1_init(colormap="gray"):
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=="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
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17 class BaseGraph:
"""
"""
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
def __init__(self):
"""
"""
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 self.hasNotRange = True
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 self.xrange = None
self.yrange = None
self.zrange = None
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 self.xlabel = None
self.ylabel = None
self.title = None
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 self.legends = None
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 self.__name = None
self.__colormap = None
self.__colbox = None
self.__colleg = None
self.__xpos = None
self.__ypos = None
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 self.__xopt = None #"bcnst"
self.__yopt = None #"bcnstv"
self.__xlpos = None
self.__ylpos = None
self.__xrangeIsTime = False
#Advanced
self.__xg = None
self.__yg = None
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
def setName(self, name):
self.__name = name
def setScreenPos(self, xpos, ypos):
self.__xpos = xpos
self.__ypos = ypos
def setOpt(self, xopt, yopt):
self.__xopt = xopt
self.__yopt = yopt
def setXAxisAsTime(self):
self.__xrangeIsTime = True
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
def setup(self, title=None, xlabel=None, ylabel=None, colormap=None):
"""
"""
self.title = title
self.xlabel = xlabel
self.ylabel = ylabel
self.__colormap = colormap
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20
Miguel Valdez
Grafico de espectros actualizado
 r27 def plotBox(self, xmin, xmax, ymin, ymax, xopt=None, yopt=None, nolabels=False):
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 """
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 """
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17 if self.__xrangeIsTime:
plplot.pltimefmt("%H:%M")
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 plplot.plvpor(self.__xpos[0], self.__xpos[1], self.__ypos[0], self.__ypos[1])
plplot.plwind(float(xmin),
float(xmax),
float(ymin),
float(ymax)
)
Miguel Valdez
Desplazamiento automatico de los puntos de FFT en el modulo de lectura de espectros.
 r26
if xopt == None: xopt = self.__xopt
if yopt == None: yopt = self.__yopt
plplot.plbox(xopt, 0.0, 0, yopt, 0.0, 0)
Miguel Valdez
Grafico de espectros actualizado
 r27 if not(nolabels):
plplot.pllab(self.xlabel, self.ylabel, self.title)
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 def colorbarPlot(self, xmin=0., xmax=1., ymin=0., ymax=1.):
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17 data = numpy.arange(256)
data = numpy.reshape(data, (1,-1))
plplot.plimage(data,
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 float(xmin),
float(xmax),
float(ymin),
float(ymax),
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17 0.,
255.,
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 float(xmin),
float(xmax),
float(ymin),
float(ymax))
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]
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
plplot.plline(x, y)
def basicXYwithErrorPlot(self):
pass
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 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)
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
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)>1 and len(y)>1):
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
def advPcolorPlot(self, data, x, y, zmin=0., zmax=0.):
"""
"""
if self.__xg == None and self.__yg == None:
self.__getBoxpltr(x, y)
plplot.plimagefr(data, x[0], x[-1], y[0], y[-1], 0., 0., zmin, zmax, plplot.pltr2, self.__xg, self.__yg)
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 class LinearPlot():
__szchar = 1.0
__xrange = None
__yrange = None
m_BaseGraph = None
def __init__(self):
key = "linearplot"
self.m_BaseGraph = BaseGraph()
self.m_BaseGraph.setName(key)
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25
self.__subpage = 0
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 def setColormap(self, colormap="br_green"):
if colormap == None:
colormap = self.__colormap
cmap1_init(colormap)
def iniSubpage(self):
if plplot.plgdev() == '':
raise ValueError, "Plot device has not been initialize"
plplot.pladv(self.__subpage)
plplot.plschr(0.0, self.__szchar)
self.setColormap()
def setScreenPos(self, width='small'):
if width == 'small':
xi = 0.12; yi = 0.14; xw = 0.78; yw = 0.80
if width == 'medium':
xi = 0.07; yi = 0.10; xw = 0.90; yw = 0.60
xf = xi + xw
yf = yi + yw
self.m_BaseGraph.setScreenPos([xi, xf], [yi, yf])
def setup(self, subpage, title="", xlabel="", ylabel="", XAxisAsTime=False):
"""
"""
self.m_BaseGraph.setOpt("bcnts","bcntsv")
self.m_BaseGraph.setup(title,
xlabel,
ylabel
)
self.setScreenPos(width='medium')
if XAxisAsTime:
self.m_BaseGraph.setXAxisAsTime()
self.__subpage = subpage
# def setRanges(self, xrange, yrange, zrange):
#
# self.m_BaseGraph.setRanges(xrange, yrange, zrange)
def plotData(self, x, y=None, xmin=None, xmax=None, ymin=None, ymax=None, colline=1):
"""
Inputs:
x : Numpy array of dimension 1
y : Numpy array of dimension 1
"""
try:
nX = numpy.shape(x)
except:
raise ValueError, "x is not a numpy array"
if y == None: y = numpy.arange(nX)
if xmin == None: xmin = x[0]
if xmax == None: xmax = x[-1]
if ymin == None: ymin = y[0]
if ymax == None: ymax = y[-1]
self.m_BaseGraph.plotBox(xmin, xmax, ymin, ymax)
self.m_BaseGraph.basicLineTimePlot(x, y, xmin, xmax, ymin, ymax, colline)
def plotComplexData(self, x, y, xmin=None, xmax=None, ymin=None, ymax=None, colline=1, type='power'):
"""
Inputs:
x : Numpy array of dimension 1
y : Complex numpy array of dimension 1
"""
try:
nX = numpy.shape(x)
except:
raise ValueError, "x is not a numpy array"
try:
nY = numpy.shape(y)
except:
raise ValueError, "y is not a numpy array"
if xmin == None: xmin = x[0]
if xmax == None: xmax = x[-1]
if ymin == None: ymin = y[0]
if ymax == None: ymax = y[-1]
self.m_BaseGraph.plotBox(xmin, xmax, ymin, ymax)
if type.lower() == 'power':
self.m_BaseGraph.basicLineTimePlot(x, abs(y), xmin, xmax, ymin, ymax, colline)
if type.lower() == 'iq':
self.m_BaseGraph.basicLineTimePlot(x, y.real, xmin, xmax, ymin, ymax, colline)
self.m_BaseGraph.basicLineTimePlot(x, y.imag, xmin, xmax, ymin, ymax, colline+1)
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17 class ColorPlot():
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25
def __init__(self):
self.graphObjDict = {}
self.__subpage = 0
self.__showColorbar = False
self.__showPowerProfile = True
Miguel Valdez
Grafico de espectros actualizado
 r27 self.__szchar = 0.65
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 self.__xrange = None
self.__yrange = None
self.__zrange = None
key = "colorplot"
self.m_BaseGraph = BaseGraph()
self.m_BaseGraph.setName(key)
def setup(self, subpage, title="", xlabel="Frequency", ylabel="Range", colormap="jet", showColorbar=False, showPowerProfile=False, XAxisAsTime=False):
"""
"""
self.m_BaseGraph.setOpt("bcnts","bcntsv")
self.m_BaseGraph.setup(title,
xlabel,
ylabel
)
self.__subpage = subpage
self.__colormap = colormap
self.__showColorbar = showColorbar
self.__showPowerProfile = showPowerProfile
if showColorbar:
key = "colorbar"
cmapObj = BaseGraph()
cmapObj.setName(key)
Miguel Valdez
Grafico de espectros actualizado
 r27 cmapObj.setOpt("bc","bcmtv")
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 cmapObj.setup(title="dBs",
xlabel="",
ylabel="",
colormap=colormap)
self.graphObjDict[key] = cmapObj
if showPowerProfile:
key = "powerprof"
powObj = BaseGraph()
powObj.setName(key)
powObj.setOpt("bcntg","bc")
powObj.setup(title="Power Profile",
Miguel Valdez
Desplazamiento automatico de los puntos de FFT en el modulo de lectura de espectros.
 r26 xlabel="dB",
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 ylabel="")
self.graphObjDict[key] = powObj
self.setScreenPos(width='small')
if XAxisAsTime:
self.m_BaseGraph.setXAxisAsTime()
def setColormap(self, colormap="br_green"):
if colormap == None:
colormap = self.__colormap
cmap1_init(colormap)
def iniSubpage(self):
if plplot.plgdev() == '':
raise ValueError, "Plot device has not been initialize"
plplot.pladv(self.__subpage)
plplot.plschr(0.0, self.__szchar)
self.setColormap()
def setScreenPos(self, width='small'):
if width == 'small':
Miguel Valdez
Grafico de espectros actualizado
 r27 xi = 0.13; yi = 0.12; xw = 0.86; yw = 0.70; xcmapw = 0.04; xpoww = 0.25; deltaxcmap = 0.02; deltaxpow = 0.06
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25
if width == 'medium':
Miguel Valdez
Grafico de espectros actualizado
 r27 xi = 0.07; yi = 0.10; xw = 0.90; yw = 0.60; xcmapw = 0.04; xpoww = 0.24; deltaxcmap = 0.02; deltaxpow = 0.06
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25
if self.__showColorbar:
xw -= xcmapw + deltaxcmap
if self.__showPowerProfile:
xw -= xpoww + deltaxpow
xf = xi + xw
yf = yi + yw
xcmapf = xf
self.m_BaseGraph.setScreenPos([xi, xf], [yi, yf])
if self.__showColorbar:
xcmapi = xf + deltaxcmap
xcmapf = xcmapi + xcmapw
key = "colorbar"
cmapObj = self.graphObjDict[key]
cmapObj.setScreenPos([xcmapi, xcmapf], [yi, yf])
if self.__showPowerProfile:
xpowi = xcmapf + deltaxpow
xpowf = xpowi + xpoww
key = "powerprof"
powObj = self.graphObjDict[key]
powObj.setScreenPos([xpowi, xpowf], [yi, yf])
def plotData(self, data, x=None, y=None, xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None):
"""
Inputs:
x : Numpy array of dimension 1
y : Numpy array of dimension 1
"""
try:
nX, nY = numpy.shape(data)
except:
raise ValueError, "data is not a numpy array"
if x == None: x = numpy.arange(nX)
if y == None: y = numpy.arange(nY)
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)
Miguel Valdez
Grafico de espectros actualizado
 r27 plplot.plschr(0.0, self.__szchar)
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 self.m_BaseGraph.plotBox(xmin, xmax, ymin, ymax)
self.m_BaseGraph.basicPcolorPlot(data, x, y, xmin, xmax, ymin, ymax, zmin, zmax)
if self.__showColorbar:
Miguel Valdez
Grafico de espectros actualizado
 r27
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 key = "colorbar"
cmapObj = self.graphObjDict[key]
Miguel Valdez
Grafico de espectros actualizado
 r27
plplot.plschr(0.0, self.__szchar-0.05)
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 cmapObj.plotBox(0., 1., zmin, zmax)
cmapObj.colorbarPlot(0., 1., zmin, zmax)
if self.__showPowerProfile:
Miguel Valdez
Desplazamiento automatico de los puntos de FFT en el modulo de lectura de espectros.
 r26 power = numpy.max(data, axis=0)
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25
step = (ymax - ymin)/(nY-1)
heis = numpy.arange(ymin, ymax + step, step)
key = "powerprof"
powObj = self.graphObjDict[key]
plplot.pllsty(2)
Miguel Valdez
Grafico de espectros actualizado
 r27 plplot.plschr(0.0, self.__szchar-0.05)
powObj.plotBox(zmin, zmax, ymin, ymax, nolabels=True)
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 plplot.pllsty(1)
Miguel Valdez
Grafico de espectros actualizado
 r27 plplot.plschr(0.0, self.__szchar)
Miguel Valdez
Desplazamiento automatico de los puntos de FFT en el modulo de lectura de espectros.
 r26 powObj.plotBox(zmin, zmax, ymin, ymax, xopt='bc', yopt='bc')
Miguel Valdez
Grafico de espectros actualizado
 r27
Miguel Valdez
Desplazamiento automatico de los puntos de FFT en el modulo de lectura de espectros.
 r26 plplot.plcol0(9)
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 powObj.basicXYPlot(power, heis)
Miguel Valdez
Desplazamiento automatico de los puntos de FFT en el modulo de lectura de espectros.
 r26 plplot.plcol0(1)
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25
class ColorPlotX():
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
graphObjDict = {}
showColorbar = False
showPowerProfile = True
__szchar = 0.7
__xrange = None
__yrange = None
__zrange = None
m_BaseGraph = BaseGraph()
def __init__(self):
key = "colorplot"
self.m_BaseGraph.setName(key)
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 self.__subpage = 0
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17 self.graphObjDict[key] = self.m_BaseGraph
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25
def setColormap(self, colormap="br_green"):
if colormap == None:
colormap = self.__colormap
cmap1_init(colormap)
def iniSubpage(self):
if plplot.plgdev() == '':
raise ValueError, "Plot device has not been initialize"
plplot.pladv(self.__subpage)
plplot.plschr(0.0, self.__szchar)
self.setColormap()
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17 def setup(self, subpage, title="", xlabel="", ylabel="", colormap="jet", showColorbar=False, showPowerProfile=False, XAxisAsTime=False):
"""
"""
self.m_BaseGraph.setSubpage(subpage)
self.m_BaseGraph.setSzchar(self.__szchar)
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 self.m_BaseGraph.setOpt("bcnts","bcntsv")
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17 self.m_BaseGraph.setup(title,
xlabel,
ylabel,
colormap)
if showColorbar:
key = "colorbar"
cmapObj = BaseGraph()
cmapObj.setName(key)
cmapObj.setSubpage(subpage)
cmapObj.setSzchar(self.__szchar)
cmapObj.setOpt("bc","bcmt")
cmapObj.setup(title="dBs",
xlabel="",
ylabel="",
colormap=colormap)
self.graphObjDict[key] = cmapObj
if showPowerProfile:
key = "powerprof"
powObj = BaseGraph()
powObj.setName(key)
powObj.setSubpage(subpage)
powObj.setSzchar(self.__szchar)
plplot.pllsty(2)
powObj.setOpt("bcntg","bc")
plplot.pllsty(1)
powObj.setup(title="Power Profile",
xlabel="dBs",
ylabel="")
self.graphObjDict[key] = powObj
self.showColorbar = showColorbar
self.showPowerProfile = showPowerProfile
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 self.setScreenPos()
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
if XAxisAsTime:
self.m_BaseGraph.setXAxisAsTime()
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 #self.setScreenPos(xi = 0.05, yi = 0.18, xw = 0.92, yw = 0.74, xcmapw = 0.015, xpoww = 0.14, deltaxcmap = 0.01, deltaxpow = 0.02)
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Dentro del paquete IO se agrego un sufijo "IO" a todos los modulos
 r20 def setScreenPos(self, xi = 0.12, yi = 0.14, xw = 0.78, yw = 0.80, xcmapw = 0.05, xpoww = 0.24, deltaxcmap = 0.02, deltaxpow = 0.06):
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
if self.showColorbar:
xw -= xcmapw + deltaxcmap
if self.showPowerProfile:
xw -= xpoww + deltaxpow
xf = xi + xw
yf = yi + yw
xcmapf = xf
self.m_BaseGraph.setScreenPos([xi, xf], [yi, yf])
if self.showColorbar:
xcmapi = xf + deltaxcmap
xcmapf = xcmapi + xcmapw
key = "colorbar"
cmapObj = self.graphObjDict[key]
cmapObj.setScreenPos([xcmapi, xcmapf], [yi, yf])
if self.showPowerProfile:
xpowi = xcmapf + deltaxpow
xpowf = xpowi + xpoww
key = "powerprof"
powObj = self.graphObjDict[key]
powObj.setScreenPos([xpowi, xpowf], [yi, yf])
def setRanges(self, xrange, yrange, zrange):
self.m_BaseGraph.setRanges(xrange, yrange, zrange)
keyList = self.graphObjDict.keys()
key = "colorbar"
if key in keyList:
cmapObj = self.graphObjDict[key]
cmapObj.setRanges([0., 1.], zrange)
key = "powerprof"
if key in keyList:
powObj = self.graphObjDict[key]
powObj.setRanges(zrange, yrange)
def plotData(self, data, x=None, y=None, xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None):
"""
"""
try:
nX, nY = numpy.shape(data)
except:
raise ValueError, "data is not a numpy array"
if x == None: x = numpy.arange(nX)
if y == None: y = numpy.arange(nY)
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)
if self.m_BaseGraph.hasNotRange:
self.setRanges([xmin, xmax], [ymin,ymax], [zmin,zmax])
self.m_BaseGraph.initSubpage()
self.m_BaseGraph.basicPcolorPlot(data, x, y, xmin, xmax, ymin, ymax, self.m_BaseGraph.zrange[0], self.m_BaseGraph.zrange[1])
if self.showColorbar:
key = "colorbar"
cmapObj = self.graphObjDict[key]
cmapObj.colorbarPlot()
if self.showPowerProfile:
power = numpy.average(data, axis=1)
step = (ymax - ymin)/(nY-1)
heis = numpy.arange(ymin, ymax + step, step)
key = "powerprof"
powObj = self.graphObjDict[key]
powObj.basicXYPlot(power, heis)
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 if __name__ == '__main__':
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 import numpy
plplot.plsetopt("geometry", "%dx%d" %(350*2, 300*2))
plplot.plsdev("xwin")
plplot.plscolbg(255,255,255)
plplot.plscol0(1,0,0,0)
plplot.plspause(False)
plplot.plinit()
plplot.plssub(2, 2)
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 nx = 64
ny = 100
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 data = numpy.random.uniform(-50,50,(nx,ny))
baseObj = ColorPlot()
specObj = ColorPlot()
baseObj1 = ColorPlot()
specObj1 = ColorPlot()
baseObj.setup(1, "Spectrum", "Frequency", "Range", "br_green", True, True)
specObj.setup(2, "Spectrum", "Frequency", "Range", "br_green", False, True)
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 baseObj1.setup(3, "Spectrum", "Frequency", "Range", "br_green", False, True)
specObj1.setup(4, "Spectrum", "Frequency", "Range", "br_green", False, True)
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 data = numpy.random.uniform(-50,50,(nx,ny))
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 plplot.plbop()
baseObj.iniSubpage()
baseObj.plotData(data)
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 specObj.iniSubpage()
specObj.plotData(data)
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 baseObj1.iniSubpage()
baseObj1.plotData(data)
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 specObj1.iniSubpage()
specObj1.plotData(data)
Miguel Valdez
El módulo BasicGraph fue cambiado de nombre a BaseGraph.
 r17
Miguel Valdez
Modulo de lectura de espectros corregido y testeado....
 r25 plplot.plflush()
plplot.plspause(1)
plplot.plend()
exit(0)