schainPlot.py
293 lines
| 8.7 KiB
| text/x-python
|
PythonLexer
Daniel Valdez
|
r133 | import numpy | |
|
Daniel Valdez
|
r134 | from schainPlotLib import Driver | |
|
Daniel Valdez
|
r133 | ||
| class Figure: | |||
| __isDriverOpen = False | |||
| __isFigureOpen = False | |||
| __isConfig = False | |||
|
Daniel Valdez
|
r135 | drvObj = None | |
|
Daniel Valdez
|
r133 | idfigure = None | |
| nframes = None | |||
|
Daniel Valdez
|
r134 | wintitle = None | |
|
Daniel Valdez
|
r133 | colormap = None | |
| driver = None | |||
| overplot = None | |||
|
Daniel Valdez
|
r135 | xmin = None | |
| xmax = None | |||
| minvalue = None | |||
| maxvalue = None | |||
|
Daniel Valdez
|
r133 | frameObjList = [] | |
|
Daniel Valdez
|
r135 | figtitle = "" | |
|
Daniel Valdez
|
r133 | ||
|
Daniel Valdez
|
r135 | def __init__(self,idfigure, nframes, wintitle, xw=600, yw=800, overplot=0, driver='plplot', colorbar= True, colormap=None, *args): | |
|
Daniel Valdez
|
r133 | self.idfigure = idfigure | |
| self.nframes = nframes | |||
| self.wintitle = wintitle | |||
|
Daniel Valdez
|
r135 | self.xw = xw | |
| self.yw = yw | |||
|
Daniel Valdez
|
r133 | self.overplot = overplot | |
|
Daniel Valdez
|
r135 | self.colorbar = colorbar | |
|
Daniel Valdez
|
r133 | self.colormap = colormap | |
|
Daniel Valdez
|
r135 | #esta seccion deberia ser dinamica de acuerdo a len(args) | |
| self.showGraph1 = args[0] | |||
| self.showGraph2 = args[1] | |||
|
Daniel Valdez
|
r133 | ||
|
Daniel Valdez
|
r135 | self.drvObj = Driver(driver, idfigure, xw, yw, wintitle, overplot, colorbar, colormap) | |
|
Daniel Valdez
|
r133 | ||
|
Daniel Valdez
|
r135 | self.drvObj.driver.setFigure() | |
|
Daniel Valdez
|
r133 | ||
|
Daniel Valdez
|
r135 | ||
|
Daniel Valdez
|
r133 | def __openDriver(self): | |
|
Daniel Valdez
|
r135 | self.drvObj.driver.openDriver() | |
|
Daniel Valdez
|
r133 | ||
|
Daniel Valdez
|
r135 | # def __openFigure(self): | |
| # nrows, ncolumns = self.getSubplots() | |||
| # self.drvObj.driver.openFigure() | |||
| # self.drvObj.driver.setTitleFig(title) | |||
| # self.drvObj.driver.setSubPlots(nrows, ncolumns) | |||
|
Daniel Valdez
|
r133 | ||
|
Daniel Valdez
|
r135 | def __initFigure(self): | |
|
Daniel Valdez
|
r133 | nrows, ncolumns = self.getSubplots() | |
|
Daniel Valdez
|
r135 | self.drvObj.driver.openFigure() | |
| self.drvObj.driver.setFigTitle(self.figtitle) | |||
| self.drvObj.driver.setSubPlots(nrows, ncolumns) | |||
|
Daniel Valdez
|
r133 | ||
|
Daniel Valdez
|
r135 | def __isOutOfXRange(self,x): | |
| return 1 | |||
| def __refresh(self): | |||
| self.drvObj.driver.refresh() | |||
| def createFrames(self): | |||
| raise ValueError, "No implemented" | |||
|
Daniel Valdez
|
r138 | def plot1DArray(self, data1D, x=None, channelList=None, xmin=None, xmax=None, minvalue=None, maxvalue=None, figuretitle=None, save=False, gpath='./'): | |
|
Daniel Valdez
|
r133 | ||
| nx, ny = data1D.shape | |||
| if channelList == None: | |||
|
Daniel Valdez
|
r134 | channelList = range(nx) | |
|
Daniel Valdez
|
r133 | ||
| if x == None: | |||
| x = numpy.arange(data1D.size) | |||
|
Daniel Valdez
|
r135 | ||
|
Daniel Valdez
|
r138 | if figuretitle == None: | |
| self.figuretitle = "" | |||
|
Daniel Valdez
|
r135 | else: | |
|
Daniel Valdez
|
r138 | self.figuretitle = figuretitle | |
|
Daniel Valdez
|
r135 | ||
|
Daniel Valdez
|
r133 | if not(self.__isDriverOpen): | |
| self.__openDriver() | |||
| self.__isDriverOpen = True | |||
| if not(self.__isConfig): | |||
|
Daniel Valdez
|
r135 | self.xmin = xmin | |
| self.xmax = xmax | |||
| self.minvalue = minvalue | |||
| self.maxvalue = maxvalue | |||
| if self.xmin == None: self.xmin = numpy.min(x) | |||
| if self.xmax == None: self.xmax = numpy.max(x) | |||
| if self.minvalue == None: self.minvalue = numpy.min(data1D) | |||
| if self.maxvalue == None: self.maxvalue = numpy.max(data1D) | |||
|
Daniel Valdez
|
r133 | ||
|
Daniel Valdez
|
r135 | self.createFrames() | |
|
Daniel Valdez
|
r133 | self.__isConfig = True | |
| if not(self.__isOutOfXRange(x)): | |||
| self.__changeXRange(x) | |||
| if self.__isFigureOpen: | |||
|
Daniel Valdez
|
r135 | self.driverObj.closePage() | |
|
Daniel Valdez
|
r133 | self.__isFigureOpen = False | |
|
Daniel Valdez
|
r135 | self.__initFigure() | |
|
Daniel Valdez
|
r134 | ||
|
Daniel Valdez
|
r133 | for channel in channelList: | |
| frameObj = self.frameObjList[channel] | |||
|
Daniel Valdez
|
r135 | frameObj.init(xmin=self.xmin, | |
| xmax=self.xmax, | |||
| ymin=self.minvalue, | |||
| ymax=self.maxvalue, | |||
| minvalue=self.minvalue, | |||
| maxvalue=self.maxvalue) | |||
|
Daniel Valdez
|
r133 | ||
| for channel in channelList: | |||
|
Daniel Valdez
|
r135 | dataCh = data1D[channel,:] | |
| frameObj = self.frameObjList[channel] | |||
| # frameObj.clearData() | |||
| frameObj.plot(x, dataCh) | |||
| # frameObj.refresh() | |||
| self.__refresh() | |||
|
Daniel Valdez
|
r133 | ||
|
Daniel Valdez
|
r135 | # | |
| # if save: | |||
| # self.colorplotObj.setFigure(indexPlot) | |||
| # path = "/home/roj-idl71/tmp/" | |||
| # now = datetime.datetime.now().timetuple() | |||
| # file = "spc_img%02d_%03d_%02d%02d%02d.png"%(indexPlot,now[7],now[3],now[4],now[5]) | |||
| # filename = os.path.join(path,file) | |||
| # self.colorplotObj.savePlot(indexPlot, filename) | |||
| # | |||
| # self.colorplotObj.closePage() | |||
|
Daniel Valdez
|
r138 | ||
| def plotPcolor(self,data, | |||
| x=None, | |||
| y=None, | |||
| channelList=None, | |||
| xmin=None, | |||
| xmax=None, | |||
| ymin=None, | |||
| ymax=None, | |||
| minvalue=None, | |||
| maxvalue=None, | |||
| figuretitle=None, | |||
| deltax=None, | |||
| save=False, | |||
| gpath='./'): | |||
| # data, | |||
| # currenttime, | |||
| # range, | |||
| # starttime, | |||
| # endtime, | |||
| # minrange, | |||
| # maxrange, | |||
| # minvalue, | |||
| # maxvalue, | |||
| # figuretitle, | |||
| # interval, | |||
| # save, | |||
| # gpath): | |||
| if figuretitle == None: | |||
| self.figuretitle = "" | |||
| else: | |||
| self.figuretitle = figuretitle | |||
| if not(self.__isDriverOpen): | |||
| self.__openDriver() | |||
| self.__isDriverOpen = True | |||
| if not(self.__isConfig): | |||
| self.setParms(data,x,y,xmin,xmax,ymin,ymax,minvalue,maxvalue,deltax) | |||
| # if self.xmin == None: self.xmin = numpy.min(x) | |||
| # if self.xmax == None: self.xmax = numpy.max(x) | |||
| self.createFrames() | |||
| self.__isConfig = True | |||
| if not(self.__isOutOfXRange(x)): | |||
| self.__changeXRange(x) | |||
| if self.__isFigureOpen: | |||
| self.driverObj.closePage() | |||
| self.__isFigureOpen = False | |||
| self.__initFigure() | |||
|
Daniel Valdez
|
r133 | ||
|
Daniel Valdez
|
r135 | class Frame: | |
| nplots = None | |||
|
Daniel Valdez
|
r133 | plotObjList = [] | |
| title = "" | |||
|
Daniel Valdez
|
r135 | def __init__(self,drvObj, idframe): | |
| self.drvObj = drvObj | |||
|
Daniel Valdez
|
r133 | self.idframe = idframe | |
|
Daniel Valdez
|
r135 | self.createPlots() | |
|
Daniel Valdez
|
r133 | ||
|
Daniel Valdez
|
r135 | def createPlots(self): | |
| raise ValueError, "No implemented" | |||
|
Daniel Valdez
|
r133 | ||
|
Daniel Valdez
|
r135 | def getScreenPosMainPlot(self): | |
| raise ValueError, "No implemented" | |||
| def getScreenPos(self, nplot): | |||
|
Daniel Valdez
|
r133 | ||
| if nplot == 0: | |||
|
Daniel Valdez
|
r135 | xi, yi, xw, yw = self.getScreenPosMainPlot() | |
|
Daniel Valdez
|
r133 | return xi, yi, xw, yw | |
|
Daniel Valdez
|
r135 | ||
| def init(self, xmin, xmax, ymin, ymax, minvalue, maxvalue): | |||
|
Daniel Valdez
|
r133 | for plotObj in self.plotObjList: | |
|
Daniel Valdez
|
r135 | plotObj.plotBox(xmin, xmax, ymin, ymax, minvalue, maxvalue) | |
| # plotObj.setLabels() # ? evaluar si conviene colocarlo dentro del plotbox | |||
|
Daniel Valdez
|
r133 | ||
|
Daniel Valdez
|
r135 | def refresh(self): | |
| for plotObj in self.plotObjList: | |||
| plotObj.refresh() | |||
|
Daniel Valdez
|
r133 | ||
| class Plot: | |||
| title = "" | |||
|
Daniel Valdez
|
r135 | xlabel = "" | |
| ylabel = "" | |||
| xaxisastime = None | |||
| timefmt = None | |||
| xopt = "" | |||
| yopt = "" | |||
| xpos = None | |||
| ypos = None | |||
| szchar = None | |||
| idframe = None | |||
| idplot = None | |||
| def __init__(self, drvObj, idframe, idplot, xi, yi, xw, yw): | |||
| self.drvObj = drvObj | |||
| self.idframe = idframe | |||
| self.idplot = idplot | |||
|
Daniel Valdez
|
r133 | self.xi = xi | |
| self.yi = yi | |||
| self.xw = xw | |||
| self.yw = yw | |||
|
Daniel Valdez
|
r135 | # def setLabels(self): | |
| # self.drvObj.driver.setPlotLabels(self.xlabel, self.ylabel, self.title) | |||
| def plotBox(self, xmin, xmax, ymin, ymax, minvalue, maxvalue): | |||
| self.xmin = xmin | |||
| self.xmax = xmax | |||
| self.ymin = ymin | |||
| self.ymax = ymax | |||
| self.minvalue = minvalue | |||
| self.maxvalue = maxvalue | |||
| self.drvObj.driver.plotBox(self.idframe, | |||
| self.xpos, | |||
| self.ypos, | |||
| self.xmin, | |||
| self.xmax, | |||
| self.ymin, | |||
| self.ymax, | |||
| self.minvalue, | |||
| self.maxvalue, | |||
| self.xopt, | |||
| self.yopt, | |||
| self.szchar, | |||
| self.xaxisastime, | |||
| self.timefmt) | |||
| self.drvObj.driver.setPlotLabels(self.xlabel, self.ylabel, self.title) | |||
| def plotBasicLine(self,x, y, color): | |||
| """ | |||
| Inputs: | |||
| x: | |||
| y: | |||
| color: | |||
| """ | |||
| self.drvObj.driver.basicLine(x, y, self.xmin, self.xmax, self.ymin, self.ymax, color, self.idframe, self.xpos, self.ypos) | |||
|
Daniel Valdez
|
r133 | ||
|
Daniel Valdez
|
r135 |