|
|
import numpy
|
|
|
import plplot
|
|
|
|
|
|
class BasicGraph:
|
|
|
"""
|
|
|
|
|
|
"""
|
|
|
|
|
|
hasRange = False
|
|
|
|
|
|
xrange = None
|
|
|
yrange = None
|
|
|
zrange = None
|
|
|
|
|
|
xlabel = None
|
|
|
ylabel = None
|
|
|
title = None
|
|
|
|
|
|
legends = None
|
|
|
|
|
|
__name = None
|
|
|
__subpage = None
|
|
|
__szchar = None
|
|
|
|
|
|
__colormap = None
|
|
|
__colbox = None
|
|
|
__colleg = None
|
|
|
|
|
|
__xpos = None
|
|
|
__ypos = None
|
|
|
|
|
|
__xopt = None #"bcnst"
|
|
|
__yopt = None #"bcnstv"
|
|
|
|
|
|
__xlpos = None
|
|
|
__ylpos = None
|
|
|
|
|
|
__xrangeIsTime = None
|
|
|
|
|
|
#Advanced
|
|
|
__xg = None
|
|
|
__yg = None
|
|
|
|
|
|
def __init__(self):
|
|
|
"""
|
|
|
|
|
|
"""
|
|
|
pass
|
|
|
|
|
|
def hasNotXrange(self):
|
|
|
|
|
|
if self.xrange == None:
|
|
|
return 1
|
|
|
|
|
|
return 0
|
|
|
|
|
|
def hasNotYrange(self):
|
|
|
|
|
|
if self.yrange == None:
|
|
|
return 1
|
|
|
|
|
|
return 0
|
|
|
|
|
|
def hasNotZrange(self):
|
|
|
|
|
|
if self.zrange == None:
|
|
|
return 1
|
|
|
|
|
|
return 0
|
|
|
def setName(self, name):
|
|
|
self.__name = name
|
|
|
|
|
|
def setScreenPos(self, xpos, ypos):
|
|
|
self.__xpos = xpos
|
|
|
self.__ypos = ypos
|
|
|
|
|
|
def setScreenPosbyWidth(self, xoff, yoff, xw, yw):
|
|
|
self.__xpos = [xoff, xoff + xw]
|
|
|
self.__ypos = [yoff, yoff + yw]
|
|
|
|
|
|
def setSubpage(self, subpage):
|
|
|
self.__subpage = subpage
|
|
|
|
|
|
def setSzchar(self, szchar):
|
|
|
self.__szchar = szchar
|
|
|
|
|
|
def setOpt(self, xopt, yopt):
|
|
|
self.__xopt = xopt
|
|
|
self.__yopt = yopt
|
|
|
|
|
|
def setRanges(self, xrange, yrange, zrange=None):
|
|
|
"""
|
|
|
"""
|
|
|
self.xrange = xrange
|
|
|
|
|
|
self.yrange = yrange
|
|
|
|
|
|
if zrange != None:
|
|
|
self.zrange = zrange
|
|
|
|
|
|
def setColormap(self, colormap=None):
|
|
|
|
|
|
if colormap == None:
|
|
|
colormap = self.__colormap
|
|
|
|
|
|
cmap1_init(colormap)
|
|
|
|
|
|
def plotBox(self):
|
|
|
"""
|
|
|
|
|
|
"""
|
|
|
plplot.plvpor(self.__xpos[0], self.__xpos[1], self.__ypos[0], self.__ypos[1])
|
|
|
plplot.plwind(self.xrange[0], self.xrange[1], self.yrange[0], self.yrange[1])
|
|
|
plplot.plbox(self.__xopt, 0.0, 0, self.__yopt, 0.0, 0)
|
|
|
plplot.pllab(self.xlabel, self.ylabel, self.title)
|
|
|
|
|
|
def setup(self, title=None, xlabel=None, ylabel=None, colormap=None):
|
|
|
"""
|
|
|
"""
|
|
|
self.title = title
|
|
|
self.xlabel = xlabel
|
|
|
self.ylabel = ylabel
|
|
|
self.__colormap = colormap
|
|
|
|
|
|
def initSubpage(self):
|
|
|
|
|
|
if plplot.plgdev() == '':
|
|
|
raise ValueError, "Plot device has not been initialize"
|
|
|
|
|
|
plplot.pladv(self.__subpage)
|
|
|
plplot.plschr(0.0, self.__szchar)
|
|
|
|
|
|
if self.__xrangeIsTime:
|
|
|
plplot.pltimefmt("%H:%M")
|
|
|
|
|
|
self.setColormap()
|
|
|
self.initPlot()
|
|
|
|
|
|
def initPlot(self):
|
|
|
"""
|
|
|
|
|
|
"""
|
|
|
if plplot.plgdev() == '':
|
|
|
raise ValueError, "Plot device has not been initialize"
|
|
|
|
|
|
xrange = self.xrange
|
|
|
if xrange == None:
|
|
|
xrange = [0., 1.]
|
|
|
|
|
|
yrange = self.yrange
|
|
|
if yrange == None:
|
|
|
yrange = [0., 1.]
|
|
|
|
|
|
plplot.plvpor(self.__xpos[0], self.__xpos[1], self.__ypos[0], self.__ypos[1])
|
|
|
plplot.plwind(xrange[0], xrange[1], yrange[0], yrange[1])
|
|
|
plplot.plbox(self.__xopt, 0.0, 0, self.__yopt, 0.0, 0)
|
|
|
plplot.pllab(self.xlabel, self.ylabel, self.title)
|
|
|
|
|
|
def colorbarPlot(self):
|
|
|
data = numpy.arange(256)
|
|
|
data = numpy.reshape(data, (1,-1))
|
|
|
|
|
|
self.plotBox()
|
|
|
plplot.plimage(data,
|
|
|
self.xrange[0],
|
|
|
self.xrange[1],
|
|
|
self.yrange[0],
|
|
|
self.yrange[1],
|
|
|
0.,
|
|
|
255.,
|
|
|
self.xrange[0],
|
|
|
self.xrange[1],
|
|
|
self.yrange[0],
|
|
|
self.yrange[1],)
|
|
|
|
|
|
def basicXYPlot(self, x, y):
|
|
|
self.plotBox()
|
|
|
plplot.plline(x, y)
|
|
|
|
|
|
def basicXYwithErrorPlot(self):
|
|
|
pass
|
|
|
|
|
|
def basicLineTimePlot(self):
|
|
|
pass
|
|
|
|
|
|
def basicPcolorPlot(self, data, xmin, xmax, ymin, ymax, zmin, zmax):
|
|
|
"""
|
|
|
"""
|
|
|
|
|
|
self.plotBox()
|
|
|
plplot.plimage(data, xmin, xmax, ymin, ymax, zmin, zmax, xmin, xmax, ymin, 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)
|
|
|
|
|
|
|
|
|
|
