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

File last commit:

r169:661bb8af254a
r261:1da5a2d48804
Show More
BaseGraph_mpl.py
1210 lines | 30.9 KiB | text/x-python | PythonLexer
/ schainpy / old / Graphics / BaseGraph_mpl.py
History | Source | Raw |Copy content |Copy permalink
Marco Valdez
Matplotlib version of BaseGraph
 r128 '''
Created on Feb 7, 2012
@autor $Author: dsuarez $
@version $Id: BaseGraph.py 117 2012-09-04 21:16:59Z dsuarez $
'''
#from __future__ import division
#import os
import numpy
#import sys
import time
import datetime
#import plplot
import matplotlib as mpl
mpl.use('TKAgg')
import matplotlib.pyplot as plt
import scitools.numpyutils as sn
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
pass
def setColormap(colormap='jet'):
# cmap1_init(colormap)
pass
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)
pass
def initMatplotlib(indexFig,ncol,nrow,winTitle,width,height):
plt.ioff()
fig = plt.figure(indexFig)
fig.canvas.manager.set_window_title(winTitle)
fig.canvas.manager.resize(width,height)
# fig.add_subplot(nrow,ncol,1)
plt.ion()
def setNewPage():
# plplot.plbop()
# plplot.pladv(0)
plt.clf()
def closePage():
# plplot.pleop()
pass
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 setFigure(indexFig):
# plplot.plsstrm(indexFig)
plt.figure(indexFig)
def refresh():
# plplot.refresh()
plt.draw()
def show():
# plplot.plspause(True)
# plplot.plend()
plt.ioff()
plt.show()
plt.ion()
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)
pass
def setSubpages(ncol,nrow):
# plplot.plssub(ncol,nrow)
pass
class BaseGraph:
__name = None
__xpos = None
__ypos = None
__subplot = None
__xg = None
__yg = None
__axesId = 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.setSubPlot(subplot)
self.setXYZrange(xrange,yrange,zrange)
self.setSizeOfChar(szchar)
self.setLabels(xlabel,ylabel,title)
self.getGrid = True
self.xaxisIsTime = False
self.deltax = deltax
pass
def makeAxes(self,indexFig,nrow,ncol,subplot):
fig = plt.figure(indexFig)
self.__axesId = fig.add_subplot(nrow,ncol,subplot)
def setParentAxesId(self,parent):
self.__axesId = parent.__axesId
def setName(self, name):
self.__name = name
def setScreenPos(self,xpos,ypos):
self.__xpos = xpos
self.__ypos = ypos
def setSubPlot(self,subplot):
self.__subplot = subplot
def setXYZrange(self,xrange,yrange,zrange):
self.xrange = xrange
self.yrange = yrange
self.zrange = zrange
def setSizeOfChar(self,szchar):
self.__szchar = szchar
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 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 setLineStyle(self, style):
# plplot.pllsty(style)
pass
def setColor(self, color):
# plplot.plcol0(color)
pass
def setXAxisAsTime(self, value=False):
self.xaxisIsTime = value
def basicLineTimePlot(self, x, y, colline=1):
# plplot.plcol0(colline)
# plplot.plline(x, y)
# plplot.plcol0(1)
ax = self.__axesId
if self.setXYData() == None:
ax.plot(x,y)
else:
ax.lines[0].set_data(x,y)
def basicXYPlot(self, x, y):
# plplot.plline(x, y)
ax = self.__axesId
if self.setXYData() == None:
ax.plot(x,y)
else:
ax.lines[0].set_data(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)
# )
pass
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
pass
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),
# 0.,
# 0.,
# float(zmin),
# float(zmax),
# plplot.pltr2,
# xg,
# yg)
ax = self.__axesId
# if self.setXYData() == None:
ax.pcolormesh(x,y,data.T,vmin=zmin,vmax=zmax)
# else:
# ax.lines[0].set_data(x,y)
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),
# 0.,
# 255.,
# float(xmin),
# float(xmax),
# float(ymin),
# float(ymax))
ax = self.__axesId
cax, kw = mpl.colorbar.make_axes(ax)
norm = mpl.colors.Normalize(vmin=ymin, vmax=ymax)
cb = mpl.colorbar.ColorbarBase(cax,norm=norm,**kw)
self.__colorbarId = cb
cb.set_label(self.title)
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)
ax = self.__axesId
ax.set_xlim([xmin,xmax])
ax.set_ylim([ymin,ymax])
if not(nolabels):
ax.set_xlabel(self.xlabel)
ax.set_ylabel(self.ylabel)
ax.set_title(self.title)
# fig = plt.gcf()
# nrows = self.
# fig.add_subplot
# print
pass
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
pass
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])
pass
class LinearPlot:
linearObjDic = {}
__xpos = None
__ypos = None
def __init__(self,indexFig,nsubplot,winTitle):
self.width = 700
self.height = 150
self.indexFig = indexFig
self.ncol = 1
self.nrow = nsubplot
initMatplotlib(indexFig,self.ncol,self.nrow,winTitle,self.width,self.height)
def setFigure(self,indexFig):
setFigure(indexFig)
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 show(self):
show()
def refresh(self):
refresh()
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.makeAxes(self.indexFig,self.nrow,self.ncol,subplot)
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='abs'):
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() == 'simple':
# linearObj.setXYData(x,y,'real')
# if type.lower() == 'power':
# linearObj.setXYData(x,abs(y),'real')
# if type.lower() == 'iq':
# linearObj.setXYData(x,y,'complex')
if type.lower() == 'simple':
colline = 9
linearObj.basicLineTimePlot(x, y)
linearObj.setXYData(x,y,'real')
if type.lower() == 'abs':
colline = 9
linearObj.basicLineTimePlot(x, abs(y), 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')
pass
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,indexFig,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)
self.ncol = ncol
self.nrow = nrow
self.indexFig = indexFig
initMatplotlib(indexFig,ncol,nrow,winTitle,self.width,self.height)
setColormap(colormap)
def setFigure(self,indexFig):
setFigure(indexFig)
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)
#pcolorObj.makeAxes(self.indexFig,self.nrow,self.ncol,subplot)
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)
#powObj.makeAxes(self.indexFig,self.nrow,self.ncol,subplot)
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 show(self):
show()
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.makeAxes(self.indexFig,self.nrow,self.ncol,subplot)
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.setParentAxesId(pcolorObj)
# 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,indexFig,filename):
width = self.width*self.ncol
hei = self.height*self.nrow
savePlplot(filename,width,hei)
def refresh(self):
refresh()
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,indexFig,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
initMatplotlib(indexFig,ncol,nrow,winTitle,self.width,self.height)
setColormap(colormap)
self.ncol = ncol
self.nrow = nrow
def setFigure(self,indexFig):
setFigure(indexFig)
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,indexFig,filename):
width = self.width*self.ncol
hei = self.height*self.nrow
savePlplot(filename,width,hei)
def refresh(self):
refresh()
if __name__ == '__main__':
"""
Ejemplo1
"""
#Setting the signal
fs = 8000
f0 = 200
f1 = 400
T = 1./fs
x = numpy.arange(160)
y1 = numpy.sin(2*numpy.pi*f0*x*T)
y2 = numpy.sin(2*numpy.pi*f1*x*T)
signalList = [y1,y2]
xmin = numpy.min(x)
xmax = numpy.max(x)
ymin = numpy.min(y1)
ymax = numpy.max(y1)
# Creating Object
indexPlot = 1
nsubplot = 2
winTitle = "mi grafico v1"
subplotTitle = "subplot - No."
xlabel = ""
ylabel = ""
linearObj = LinearPlot(indexPlot,nsubplot,winTitle)
#Config SubPlots
for subplot in range(nsubplot):
indexplot = subplot + 1
title = subplotTitle + '%d'%indexplot
linearObj.setup(indexplot, xmin, xmax, ymin, ymax, title, xlabel, ylabel)
#Plotting
type = "simple"
for subplot in range(nsubplot):
indexplot = subplot + 1
y = signalList[subplot]
linearObj.plot(indexplot, x, y, type)
linearObj.refresh()
# linearObj.show()
"""
Ejemplo2
"""
# make these smaller to increase the resolution
dx, dy = 0.05, 0.05
x = numpy.arange(-3.0, 3.0001, dx)
y = numpy.arange(-2.0, 2.0001, dy)
# X,Y = numpy.meshgrid(x, y)
X,Y = sn.ndgrid(x, y)
Z = (1- X/2 + X**5 + Y**3)*numpy.exp(-X**2-Y**2)
# Creating Object
indexPlot = 2
nsubplot = 1
winTitle = "mi grafico pcolor"
colormap = "br_green"
showColorbar = True
showPowerProfile = False
XAxisAsTime = False
subplotTitle = "subplot no. "
xlabel = ""
ylabel = ""
xmin = -3.0
xmax = 3.0
ymin = -2.0
ymax = 2.0
zmin = -0.3
zmax = 1.0
isPlotIni = False
isPlotConfig = False
ntimes = 10
for i in range(ntimes):
# Instancia del objeto
if not(isPlotConfig):
pcolorObj = PcolorPlot(indexPlot, nsubplot, winTitle, colormap, showColorbar, showPowerProfile, XAxisAsTime)
isPlotConfig = True
# Crea los subplots
if not(isPlotIni):
for index in range(nsubplot):
indexplot = index + 1
title = subplotTitle + '%d'%indexplot
subplot = index
pcolorObj.createObjects(indexplot,xmin,xmax,ymin,ymax,zmin,zmax,title,xlabel,ylabel)
isPlotIni = True
# Inicializa el grafico en cada iteracion
pcolorObj.setFigure(indexPlot)
pcolorObj.setNewPage("")
for index in range(nsubplot):
subplot = index
pcolorObj.iniPlot(subplot+1)
#plotea los datos
for channel in range(nsubplot):
data = Z+0.1*numpy.random.randn(len(x),len(y))
pcolorObj.plot(channel+1, x, y, data)
pcolorObj.refresh()
# pcolorObj.closePage() #descomentar esta linea para mas iteraciones
time.sleep(1)
pcolorObj.show()
print "end"