figure.py
231 lines
| 6.5 KiB
| text/x-python
|
PythonLexer
Miguel Valdez
|
r201 | import numpy | ||
Daniel Valdez
|
r190 | import mpldriver | ||
|
Miguel Valdez
|
r201 | |||
|
Daniel Valdez
|
r190 | class Figure: | ||
|
Miguel Valdez
|
r201 | |||
| __driver = mpldriver | ||||
| idfigure = None | ||||
| wintitle = None | ||||
|
Daniel Valdez
|
r192 | width = None | ||
| height = None | ||||
|
Miguel Valdez
|
r201 | nplots = None | ||
| axesObjList = [] | ||||
| WIDTH = None | ||||
| HEIGHT = None | ||||
| def __init__(self): | ||||
| raise ValueError, "This method is not implemented" | ||||
| def getAxesObjList(self): | ||||
| return self.axesObjList | ||||
| def getSubplots(self): | ||||
| raise ValueError, "Abstract method: This method should be defined" | ||||
| def getScreenDim(self): | ||||
| nrow, ncol = self.getSubplots() | ||||
| width = self.WIDTH*ncol | ||||
| height = self.HEIGHT*nrow | ||||
| return width, height | ||||
| def init(self, idfigure, nplots, wintitle): | ||||
| """ | ||||
| Inicializa la figura de acuerdo al driver seleccionado | ||||
| Input: | ||||
| *args : Los parametros necesarios son | ||||
| idfigure, wintitle, width, height | ||||
| """ | ||||
|
Daniel Valdez
|
r190 | |||
| self.idfigure = idfigure | ||||
|
Miguel Valdez
|
r201 | |||
|
Daniel Valdez
|
r190 | self.nplots = nplots | ||
|
Miguel Valdez
|
r201 | self.wintitle = wintitle | ||
| self.width, self.height = self.getScreenDim() | ||||
| self.fig = self.__driver.createFigure(self.idfigure, | ||||
| self.wintitle, | ||||
| self.width, | ||||
| self.height) | ||||
| self.axesObjList = [] | ||||
|
Daniel Valdez
|
r190 | |||
|
Miguel Valdez
|
r201 | def setDriver(self, driver=mpldriver): | ||
| self.__driver = driver | ||||
|
Daniel Valdez
|
r190 | def setTitle(self, title): | ||
|
Miguel Valdez
|
r201 | |||
| self.__driver.setTitle(self.fig, title) | ||||
|
Daniel Valdez
|
r190 | |||
|
Miguel Valdez
|
r201 | def setWinTitle(self, title): | ||
| self.__driver.setWinTitle(self.fig, title=title) | ||||
|
Daniel Valdez
|
r196 | |||
|
Miguel Valdez
|
r201 | def setTextFromAxes(self, text): | ||
| raise ValueError, "Este metodo ha sido reemplazaado con el metodo setText de la clase Axes" | ||||
|
Daniel Valdez
|
r190 | |||
| def makeAxes(self, nrow, ncol, xpos, ypos, colspan, rowspan): | ||||
|
Miguel Valdez
|
r201 | |||
| raise ValueError, "Este metodo ha sido reemplazaado con el metodo addAxes" | ||||
| def addAxes(self, *args): | ||||
| """ | ||||
| Input: | ||||
| *args : Los parametros necesarios son | ||||
| nrow, ncol, xpos, ypos, colspan, rowspan | ||||
| """ | ||||
| axesObj = Axes(self.fig, *args) | ||||
| self.axesObjList.append(axesObj) | ||||
|
Daniel Valdez
|
r190 | |||
| def draw(self): | ||||
|
Miguel Valdez
|
r201 | |||
| self.__driver.draw(self.fig) | ||||
|
Daniel Valdez
|
r190 | |||
| def run(self): | ||||
|
Miguel Valdez
|
r201 | |||
| raise ValueError, "This method is not implemented" | ||||
| axesList = property(getAxesObjList) | ||||
|
Daniel Valdez
|
r190 | |||
| class Axes: | ||||
|
Miguel Valdez
|
r201 | |||
| __driver = mpldriver | ||||
| fig = None | ||||
|
Daniel Valdez
|
r190 | ax = None | ||
|
Miguel Valdez
|
r201 | plot = None | ||
|
Daniel Valdez
|
r190 | |||
|
Miguel Valdez
|
r201 | firsttime = None | ||
| def __init__(self, *args): | ||||
| """ | ||||
| Input: | ||||
| *args : Los parametros necesarios son | ||||
| fig, nrow, ncol, xpos, ypos, colspan, rowspan | ||||
| """ | ||||
| ax = self.__driver.createAxes(*args) | ||||
| self.fig = args[0] | ||||
|
Daniel Valdez
|
r190 | self.ax = ax | ||
|
Miguel Valdez
|
r201 | self.plot = None | ||
| self.firsttime = True | ||||
| def setText(self, text): | ||||
| self.__driver.setAxesText(self.ax, text) | ||||
| def pline(self, x, y, | ||||
| xmin=None, xmax=None, | ||||
| ymin=None, ymax=None, | ||||
| xlabel='', ylabel='', | ||||
| title='', | ||||
| **kwargs): | ||||
| """ | ||||
| Input: | ||||
| x : | ||||
| y : | ||||
| xmin : | ||||
| xmax : | ||||
| ymin : | ||||
| ymax : | ||||
| xlabel : | ||||
| ylabel : | ||||
| title : | ||||
| **kwargs : Los parametros aceptados son | ||||
| ticksize | ||||
| """ | ||||
| if self.firsttime: | ||||
| if xmin == None: xmin = numpy.nanmin(x) | ||||
| if xmax == None: xmax = numpy.nanmax(x) | ||||
| if ymin == None: ymin = numpy.nanmin(y) | ||||
| if ymax == None: ymax = numpy.nanmax(y) | ||||
| self.plot = self.__driver.createPline(self.ax, x, y, | ||||
| xmin, xmax, | ||||
| ymin, ymax, | ||||
| xlabel=xlabel, | ||||
| ylabel=ylabel, | ||||
| title=title, | ||||
| **kwargs) | ||||
| self.firsttime = False | ||||
| return | ||||
| self.__driver.pline(self.plot, x, y, xlabel=xlabel, | ||||
| ylabel=ylabel, | ||||
| title=title) | ||||
| def pcolor(self, x, y, z, | ||||
| xmin=None, xmax=None, | ||||
| ymin=None, ymax=None, | ||||
| zmin=None, zmax=None, | ||||
| xlabel='', ylabel='', | ||||
| title='', | ||||
| **kwargs): | ||||
| """ | ||||
| Input: | ||||
| x : | ||||
| y : | ||||
| x : | ||||
| xmin : | ||||
| xmax : | ||||
| ymin : | ||||
| ymax : | ||||
| zmin : | ||||
| zmax : | ||||
| xlabel : | ||||
| ylabel : | ||||
| title : | ||||
| **kwargs : Los parametros aceptados son | ||||
| ticksize=9, | ||||
| cblabel='' | ||||
| """ | ||||
| if self.firsttime: | ||||
| if xmin == None: xmin = numpy.nanmin(x) | ||||
| if xmax == None: xmax = numpy.nanmax(x) | ||||
| if ymin == None: ymin = numpy.nanmin(y) | ||||
| if ymax == None: ymax = numpy.nanmax(y) | ||||
| if zmin == None: zmin = numpy.nanmin(z) | ||||
| if zmax == None: zmax = numpy.nanmax(z) | ||||
| self.plot = self.__driver.createPcolor(self.ax, x, y, z, | ||||
| xmin, xmax, | ||||
| ymin, ymax, | ||||
| zmin, zmax, | ||||
| xlabel=xlabel, | ||||
| ylabel=ylabel, | ||||
| title=title, | ||||
| **kwargs) | ||||
| self.firsttime = False | ||||
| return | ||||
| mesh = self.__driver.pcolor(self.plot, z, xlabel=xlabel, | ||||
| ylabel=ylabel, | ||||
| title=title) | ||||