BaseGraph.py
605 lines
| 17.4 KiB
| text/x-python
|
PythonLexer
Miguel Valdez
|
r17 | """ | |
| Created on Feb 7, 2012 | |||
| @autor $Author$ | |||
| @version $Id$ | |||
| """ | |||
| import numpy | |||
| import plplot | |||
| class BaseGraph: | |||
| """ | |||
| """ | |||
| hasNotRange = True | |||
| 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 = False | |||
| #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 | |||
| self.hasNotRange = False | |||
| def setColormap(self, colormap=None): | |||
| if colormap == None: | |||
| colormap = self.__colormap | |||
| cmap1_init(colormap) | |||
| def setXAxisAsTime(self): | |||
| self.__xrangeIsTime = True | |||
| def plotBox(self): | |||
| """ | |||
| """ | |||
| plplot.plvpor(self.__xpos[0], self.__xpos[1], self.__ypos[0], self.__ypos[1]) | |||
| plplot.plwind(float(self.xrange[0]), | |||
| float(self.xrange[1]), | |||
| float(self.yrange[0]), | |||
| float(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.] | |||
| self.plotBox() | |||
| 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, 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) | |||
| self.plotBox() | |||
| 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) | |||
| class ColorPlot(): | |||
| 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) | |||
| self.graphObjDict[key] = self.m_BaseGraph | |||
| 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) | |||
| self.m_BaseGraph.setOpt("bcnts","bcnts") | |||
| 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 | |||
| self.setPos() | |||
| if XAxisAsTime: | |||
| self.m_BaseGraph.setXAxisAsTime() | |||
| #self.setPos(xi = 0.05, yi = 0.18, xw = 0.92, yw = 0.74, xcmapw = 0.015, xpoww = 0.14, deltaxcmap = 0.01, deltaxpow = 0.02) | |||
| def setPos(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): | |||
| 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) | |||
| class LinearPlot(): | |||
| __szchar = 0.7 | |||
| __xrange = None | |||
| __yrange = None | |||
| m_BaseGraph = BaseGraph() | |||
| def __init__(self): | |||
| key = "linearplot" | |||
| self.m_BaseGraph.setName(key) | |||
| self.graphObjDict[key] = self.m_BaseGraph | |||
| def setup(self, subpage, title="", xlabel="", ylabel="", colormap="jet", XAxisAsTime=False): | |||
| """ | |||
| """ | |||
| self.m_BaseGraph.setSubpage(subpage) | |||
| self.m_BaseGraph.setSzchar(self.__szchar) | |||
| self.m_BaseGraph.setOpt("bcnts","bcnts") | |||
| self.m_BaseGraph.setup(title, | |||
| xlabel, | |||
| ylabel, | |||
| colormap) | |||
| self.setPos() | |||
| if XAxisAsTime: | |||
| self.m_BaseGraph.setXAxisAsTime() | |||
| #self.setPos(xi = 0.05, yi = 0.18, xw = 0.92, yw = 0.74, xcmapw = 0.015, xpoww = 0.14, deltaxcmap = 0.01, deltaxpow = 0.02) | |||
| def setPos(self, xi = 0.12, yi = 0.14, xw = 0.78, yw = 0.80): | |||
| xf = xi + xw | |||
| yf = yi + yw | |||
| self.m_BaseGraph.setScreenPos([xi, xf], [yi, yf]) | |||
| def setRanges(self, xrange, yrange, zrange): | |||
| self.m_BaseGraph.setRanges(xrange, yrange, zrange) | |||
| def plotData(self, x, y): | |||
| """ | |||
| """ | |||
| xmin = x[0] | |||
| xmax = x[-1] | |||
| ymin = y[0] | |||
| ymax = y[-1] | |||
| if self.m_BaseGraph.hasNotRange: | |||
| self.setRanges([xmin, xmax], [ymin,ymax]) | |||
| self.m_BaseGraph.initSubpage() | |||
| self.m_BaseGraph.basicLineTimePlot(x, y) | |||
| 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 |