schainPlotLib.py
680 lines
| 20.5 KiB
| text/x-python
|
PythonLexer
Miguel Valdez
|
r129 | import plplot | ||
| import numpy | ||||
| import sys | ||||
Daniel Valdez
|
r132 | import plplot #condicional | ||
| class Driver: | ||||
| def __init__(self,driver, idfigure, xw, yw, wintitle, overplot, colormap, *showGraphs): | ||||
| if driver == "plplot": | ||||
| self.driver = PlplotDriver(idfigure, xw, yw, wintitle, overplot, colormap, *showGraphs) | ||||
| elif driver == "mpl": | ||||
| self.driver = MplDriver(idfigure, xw, yw, wintitle, overplot, colormap, *showGraphs) | ||||
| else: | ||||
| raise ValueError, "The driver: %s is not defined"%driver | ||||
|
Miguel Valdez
|
r129 | |||
|
Daniel Valdez
|
r131 | class PlplotDriver: | ||
|
Miguel Valdez
|
r129 | |||
| __isDriverOpen = False | ||||
|
Daniel Valdez
|
r132 | pldriver = None | ||
| def __init__(self, idfigure=None, xw, yw, wintitle, overplot, colormap, *showGraphs): | ||||
|
Miguel Valdez
|
r129 | |||
|
Daniel Valdez
|
r132 | if idfigure == None: | ||
| raise ValueError, 'idfigure input must be defined' | ||||
|
Daniel Valdez
|
r131 | |||
|
Daniel Valdez
|
r132 | self.idfigure = idfigure | ||
| self.xw = xw | ||||
| self.yw = yw | ||||
|
Daniel Valdez
|
r131 | self.wintitle = wintitle | ||
|
Miguel Valdez
|
r129 | self.overplot = overplot | ||
|
Daniel Valdez
|
r132 | self.colormap = colormap | ||
| self.showGraph1 = showGraphs[0] | ||||
| self.showGraph2 = showGraphs[1] | ||||
|
Miguel Valdez
|
r129 | |||
| def configDriver(self): | ||||
|
Daniel Valdez
|
r132 | """ | ||
| previous configuration to open(init) the plplot driver | ||||
| """ | ||||
| plplot.plsstrm(self.idfigure) | ||||
| plplot.plparseopts([self.wintitle],plplot.PL_PARSE_FULL) | ||||
| plplot.plsetopt("geometry", "%dx%d"%(self.xw, self.yw)) | ||||
| plplot.plscolbg(255,255,255) | ||||
| plplot.plscol0(1,0,0,0) | ||||
| def openDriver(self, pldriver=None): | ||||
| if pldriver == None: | ||||
| if sys.platform == "linux": | ||||
| pldriver = "xcairo" | ||||
| elif sys.platform == "darwin": | ||||
| pldriver = "xwin" | ||||
| else: | ||||
| pldriver = "" | ||||
| plplot.plsdev(pldriver) | ||||
| plplot.plinit() | ||||
| plplot.plspause(False) | ||||
| self.pldriver = pldriver | ||||
|
Miguel Valdez
|
r129 | |||
| def closeDriver(self): | ||||
| pass | ||||
| def openPage(self): | ||||
|
Daniel Valdez
|
r132 | plplot.plbop() | ||
| plplot.pladv(0) | ||||
|
Miguel Valdez
|
r129 | |||
| def closePage(self): | ||||
|
Daniel Valdez
|
r132 | plplot.pleop() | ||
| def openFigure(self): | ||||
| plplot.plbop() | ||||
| plplot.pladv(0) | ||||
| def closeFigure(self): | ||||
| plplot.pleop() | ||||
| def setSubPlots(self,nrows, ncolumns): | ||||
| plplot.plssub(nrows, ncolumns) | ||||
| def setColorMap(self,colormap): | ||||
|
Miguel Valdez
|
r129 | |||
|
Daniel Valdez
|
r132 | 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). | ||||
|
Miguel Valdez
|
r129 | |||
|
Daniel Valdez
|
r132 | # Independent variable of control points. | ||
|
Miguel Valdez
|
r129 | |||
|
Daniel Valdez
|
r132 | 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 | ||||
|
Miguel Valdez
|
r129 | |||
| def setBox(self): | ||||
| pass | ||||
| def refreshBox(self): | ||||
| pass | ||||
| def save(self): | ||||
| pass | ||||
| def show(self): | ||||
| pass | ||||
| def colorbarPlot(self): | ||||
| pass | ||||
| def linePlot(self): | ||||
| pass | ||||
| def pcolorPlot(self): | ||||
| pass | ||||
| def setLabels(self): | ||||
| pass | ||||
|
Daniel Valdez
|
r132 | |||
|
Miguel Valdez
|
r129 | |||
|
Daniel Valdez
|
r131 | |||
| class MplDriver: | ||||
| def __init__(self): | ||||
| pass | ||||
|
Daniel Valdez
|
r132 | def config_driver(idfigure, wintitle, width, height): | ||
| plplot.plsstrm(idfigure) | ||||
|
Miguel Valdez
|
r129 | plplot.plparseopts([wintitle],plplot.PL_PARSE_FULL) | ||
| plplot.plsetopt("geometry", "%dx%d"%(width,height)) | ||||
| def ini_driver(driver): | ||||
| if sys.platform == "darwin": | ||||
| plplot.plsdev("xwin") | ||||
| if sys.platform == "linux": | ||||
| plplot.plsdev("xcairo") | ||||
| plplot.plscolbg(255,255,255) | ||||
| plplot.plscol0(1,0,0,0) | ||||
| plplot.plinit() | ||||
| plplot.plspause(False) | ||||
| def set_subpages(ncol,nrow): | ||||
| plplot.plssub(ncol,nrow) | ||||
| 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 set_colormap(colormap="jet"): | ||||
| cmap1_init(colormap) | ||||
| def save_figure(filename,width,height): | ||||
| curr_strm = plplot.plgstrm() | ||||
| save_strm = plplot.plmkstrm() | ||||
| plplot.plsetopt("geometry", "%dx%d"%(width,height)) | ||||
| if sys.platform == "darwin": | ||||
| plplot.plsdev("png") | ||||
| if sys.platform == "linux": | ||||
| plplot.plsdev("pngcairo") | ||||
| plplot.plsfnam(filename) | ||||
| plplot.plcpstrm(curr_strm,0) | ||||
| plplot.plreplot() | ||||
| plplot.plend1() | ||||
| plplot.plsstrm(curr_strm) | ||||
| def set_new_figure(): | ||||
| plplot.plbop() | ||||
| plplot.pladv(0) | ||||
| def close_figure(): | ||||
| plplot.pleop() | ||||
| def set_strm(indexPlot): | ||||
| plplot.plsstrm(indexPlot) | ||||
| def refresh(): | ||||
| plplot.plflush() | ||||
| def show(): | ||||
| plplot.plspause(True) | ||||
| plplot.plend() | ||||
| def set_title(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 set_line_style(style): | ||||
| plplot.pllsty(style) | ||||
| def set_color(color): | ||||
| plplot.plcol0(color) | ||||
| def set_labels(xlabel, ylabel, title): | ||||
| plplot.pllab(xlabel, ylabel, title) | ||||
| def box(subplot, xpos, ypos, xmin, xmax, ymin, ymax, xopt, yopt, szchar, xaxisastime, timefmt="%H:%M"): | ||||
| plplot.pladv(subplot) | ||||
| plplot.plschr(0.0,szchar-0.05) | ||||
| plplot.plvpor(xpos[0], xpos[1], ypos[0], ypos[1]) | ||||
| plplot.plwind(float(xmin), | ||||
| float(xmax), | ||||
| float(ymin), | ||||
| float(ymax) | ||||
| ) | ||||
| if xaxisastime: | ||||
| plplot.pltimefmt(timefmt) | ||||
| 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) | ||||
| def colorbar(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)) | ||||
| def basicline_timeplot(x, y,colline=1): | ||||
| plplot.plcol0(colline) | ||||
| plplot.plline(x, y) | ||||
| plplot.plcol0(1) | ||||
| def basic_xy_plot(x, y): | ||||
| plplot.plline(x, y) | ||||
| def basic_pcolor_plot(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 image_plot(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]) | ||||
| def adv_pcolor_plot(data, x, y, xg, yg, xmin=None, xmax=None, ymin=None, ymax=None, zmin=0., zmax=0.): | ||||
| plplot.plimagefr(data, | ||||
| float(xmin), | ||||
| float(xmax), | ||||
| float(ymin), | ||||
| float(ymax), | ||||
| 0., | ||||
| 0., | ||||
| float(zmin), | ||||
| float(zmax), | ||||
| plplot.pltr2, | ||||
| xg, | ||||
| yg) | ||||
| #------------------------------------ | ||||
| #def get_grid(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(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) | ||||