figure.py
656 lines
| 19.6 KiB
| text/x-python
|
PythonLexer
Miguel Valdez
|
r212 | import os | |
|
Miguel Valdez
|
r201 | import numpy | |
|
Miguel Valdez
|
r231 | import time, datetime | |
George Yong
|
r1167 | from schainpy.model.graphics import mpldriver | |
Daniel Valdez
|
r497 | ||
|
Miguel Valdez
|
r577 | from schainpy.model.proc.jroproc_base import Operation | |
|
Daniel Valdez
|
r435 | ||
|
Miguel Valdez
|
r760 | def isTimeInHourRange(datatime, xmin, xmax): | |
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r760 | if xmin == None or xmax == None: | |
| return 1 | |||
| hour = datatime.hour + datatime.minute/60.0 | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r760 | if xmin < (xmax % 24): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r760 | if hour >= xmin and hour <= xmax: | |
| return 1 | |||
| else: | |||
| return 0 | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r760 | else: | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r760 | if hour >= xmin or hour <= (xmax % 24): | |
| return 1 | |||
| else: | |||
| return 0 | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r760 | return 0 | |
|
Daniel Valdez
|
r487 | def isRealtime(utcdatatime): | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r487 | utcnow = time.mktime(time.localtime()) | |
| delta = abs(utcnow - utcdatatime) # abs | |||
| if delta >= 30.: | |||
| return False | |||
| return True | |||
|
Miguel Valdez
|
r577 | class Figure(Operation): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | __driver = mpldriver | |
|
Miguel Valdez
|
r206 | fig = None | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r395 | id = None | |
|
Miguel Valdez
|
r201 | wintitle = None | |
|
Daniel Valdez
|
r192 | width = None | |
| height = None | |||
|
Miguel Valdez
|
r201 | nplots = None | |
|
Miguel Valdez
|
r231 | timerange = None | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | axesObjList = [] | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r696 | WIDTH = 300 | |
| HEIGHT = 200 | |||
|
Miguel Valdez
|
r212 | PREFIX = 'fig' | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r471 | xmin = None | |
| xmax = None | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r573 | counter_imagwr = 0 | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r573 | figfile = None | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r760 | created = False | |
|
José Chávez
|
r954 | parameters = {} | |
|
José Chávez
|
r897 | def __init__(self, **kwargs): | |
| Operation.__init__(self, **kwargs) | |||
|
Miguel Valdez
|
r206 | def __del__(self): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r577 | self.__driver.closeFigure() | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r212 | def getFilename(self, name, ext='.png'): | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r395 | path = '%s%03d' %(self.PREFIX, self.id) | |
|
Miguel Valdez
|
r248 | filename = '%s_%s%s' %(self.PREFIX, name, ext) | |
| return os.path.join(path, filename) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | def getAxesObjList(self): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | return self.axesObjList | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | def getSubplots(self): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r696 | raise NotImplementedError | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r210 | def getScreenDim(self, widthplot, heightplot): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | nrow, ncol = self.getSubplots() | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r210 | widthscreen = widthplot*ncol | |
| heightscreen = heightplot*nrow | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r210 | return widthscreen, heightscreen | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r568 | def getTimeLim(self, x, xmin=None, xmax=None, timerange=None): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r760 | # if self.xmin != None and self.xmax != None: | |
| # if timerange == None: | |||
| # timerange = self.xmax - self.xmin | |||
| # xmin = self.xmin + timerange | |||
| # xmax = self.xmax + timerange | |||
|
José Chávez
|
r897 | # | |
|
Miguel Valdez
|
r760 | # return xmin, xmax | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r568 | if timerange == None and (xmin==None or xmax==None): | |
|
Miguel Valdez
|
r577 | timerange = 14400 #seconds | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r568 | if timerange != None: | |
Julio Valdez
|
r608 | txmin = x[0] #- x[0] % min(timerange/10, 10*60) | |
|
Daniel Valdez
|
r504 | else: | |
|
Julio Valdez
|
r608 | txmin = x[0] #- x[0] % 10*60 | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r506 | thisdatetime = datetime.datetime.utcfromtimestamp(txmin) | |
| thisdate = datetime.datetime.combine(thisdatetime.date(), datetime.time(0,0,0)) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r568 | if timerange != None: | |
|
Daniel Valdez
|
r471 | xmin = (thisdatetime - thisdate).seconds/(60*60.) | |
| xmax = xmin + timerange/(60*60.) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r814 | d1970 = datetime.datetime(1970,1,1) | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r814 | mindt = thisdate + datetime.timedelta(hours=xmin) #- datetime.timedelta(seconds=time.timezone) | |
| xmin_sec = (mindt - d1970).total_seconds() #time.mktime(mindt.timetuple()) - time.timezone | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r814 | maxdt = thisdate + datetime.timedelta(hours=xmax) #- datetime.timedelta(seconds=time.timezone) | |
| xmax_sec = (maxdt - d1970).total_seconds() #time.mktime(maxdt.timetuple()) - time.timezone | |||
|
Daniel Valdez
|
r471 | ||
| return xmin_sec, xmax_sec | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r395 | def init(self, id, nplots, wintitle): | |
|
José Chávez
|
r897 | ||
|
George Yong
|
r1167 | raise NotImplementedError("This method has been replaced by createFigure") | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r395 | def createFigure(self, id, wintitle, widthplot=None, heightplot=None, show=True): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | """ | |
|
Miguel Valdez
|
r204 | Crea la figura de acuerdo al driver y parametros seleccionados seleccionados. | |
| Las dimensiones de la pantalla es calculada a partir de los atributos self.WIDTH | |||
| y self.HEIGHT y el numero de subplots (nrow, ncol) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | Input: | |
|
José Chávez
|
r897 | id : Los parametros necesarios son | |
| wintitle : | |||
|
Miguel Valdez
|
r201 | """ | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r210 | if widthplot == None: | |
| widthplot = self.WIDTH | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r210 | if heightplot == None: | |
| heightplot = self.HEIGHT | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r395 | self.id = id | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | self.wintitle = wintitle | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r210 | self.widthscreen, self.heightscreen = self.getScreenDim(widthplot, heightplot) | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r760 | # if self.created: | |
| # self.__driver.closeFigure(self.fig) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r760 | if not self.created: | |
| self.fig = self.__driver.createFigure(id=self.id, | |||
| wintitle=self.wintitle, | |||
| width=self.widthscreen, | |||
| height=self.heightscreen, | |||
| show=show) | |||
| else: | |||
| self.__driver.clearFigure(self.fig) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | self.axesObjList = [] | |
|
Miguel Valdez
|
r573 | self.counter_imagwr = 0 | |
|
Daniel Valdez
|
r283 | ||
|
Miguel Valdez
|
r760 | self.created = True | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | def setDriver(self, driver=mpldriver): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | self.__driver = driver | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r190 | def setTitle(self, title): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | self.__driver.setTitle(self.fig, title) | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | def setWinTitle(self, title): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | self.__driver.setWinTitle(self.fig, title=title) | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | def setTextFromAxes(self, text): | |
|
José Chávez
|
r897 | ||
|
George Yong
|
r1167 | raise NotImplementedError("This method has been replaced with Axes.setText") | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r190 | def makeAxes(self, nrow, ncol, xpos, ypos, colspan, rowspan): | |
|
José Chávez
|
r897 | ||
|
George Yong
|
r1167 | raise NotImplementedError("This method has been replaced with Axes.addAxes") | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | def addAxes(self, *args): | |
| """ | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | Input: | |
| *args : Los parametros necesarios son | |||
| nrow, ncol, xpos, ypos, colspan, rowspan | |||
| """ | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | axesObj = Axes(self.fig, *args) | |
| self.axesObjList.append(axesObj) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r212 | def saveFigure(self, figpath, figfile, *args): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r212 | filename = os.path.join(figpath, figfile) | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r248 | fullpath = os.path.split(filename)[0] | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r248 | if not os.path.exists(fullpath): | |
|
Miguel Valdez
|
r326 | subpath = os.path.split(fullpath)[0] | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r326 | if not os.path.exists(subpath): | |
| os.mkdir(subpath) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r248 | os.mkdir(fullpath) | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r212 | self.__driver.saveFigure(self.fig, filename, *args) | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r573 | def save(self, figpath, figfile=None, save=True, ftp=False, wr_period=1, thisDatetime=None, update_figfile=True): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r577 | self.counter_imagwr += 1 | |
| if self.counter_imagwr < wr_period: | |||
|
Miguel Valdez
|
r573 | return | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r577 | self.counter_imagwr = 0 | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r577 | if save: | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r600 | if not figfile: | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r577 | if not thisDatetime: | |
|
George Yong
|
r1167 | raise ValueError("Saving figure: figfile or thisDatetime should be defined") | |
|
Miguel Valdez
|
r577 | return | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r577 | str_datetime = thisDatetime.strftime("%Y%m%d_%H%M%S") | |
| figfile = self.getFilename(name = str_datetime) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r577 | if self.figfile == None: | |
| self.figfile = figfile | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r577 | if update_figfile: | |
| self.figfile = figfile | |||
|
José Chávez
|
r897 | ||
| # store png plot to local folder | |||
|
Miguel Valdez
|
r577 | self.saveFigure(figpath, self.figfile) | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r573 | if not ftp: | |
| return | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r573 | if not thisDatetime: | |
| return | |||
|
José Chávez
|
r897 | ||
| # store png plot to FTP server according to RT-Web format | |||
|
Miguel Valdez
|
r596 | ftp_filename = self.getNameToFtp(thisDatetime, self.FTP_WEI, self.EXP_CODE, self.SUB_EXP_CODE, self.PLOT_CODE, self.PLOT_POS) | |
| # ftp_filename = os.path.join(figpath, name) | |||
|
José Chávez
|
r897 | self.saveFigure(figpath, ftp_filename) | |
|
Daniel Valdez
|
r401 | def getNameToFtp(self, thisDatetime, FTP_WEI, EXP_CODE, SUB_EXP_CODE, PLOT_CODE, PLOT_POS): | |
|
José Chávez
|
r897 | YEAR_STR = '%4.4d'%thisDatetime.timetuple().tm_year | |
|
Daniel Valdez
|
r401 | DOY_STR = '%3.3d'%thisDatetime.timetuple().tm_yday | |
| FTP_WEI = '%2.2d'%FTP_WEI | |||
| EXP_CODE = '%3.3d'%EXP_CODE | |||
| SUB_EXP_CODE = '%2.2d'%SUB_EXP_CODE | |||
| PLOT_CODE = '%2.2d'%PLOT_CODE | |||
| PLOT_POS = '%2.2d'%PLOT_POS | |||
| name = YEAR_STR + DOY_STR + FTP_WEI + EXP_CODE + SUB_EXP_CODE + PLOT_CODE + PLOT_POS | |||
| return name | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r190 | def draw(self): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | self.__driver.draw(self.fig) | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r190 | def run(self): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r696 | raise NotImplementedError | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r587 | def close(self, show=False): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r587 | self.__driver.closeFigure(show=show, fig=self.fig) | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | axesList = property(getAxesObjList) | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r190 | ||
| class Axes: | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | __driver = mpldriver | |
| fig = None | |||
|
Daniel Valdez
|
r190 | ax = None | |
|
Miguel Valdez
|
r201 | plot = None | |
|
Daniel Valdez
|
r318 | __missing = 1E30 | |
|
Miguel Valdez
|
r210 | __firsttime = None | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r204 | __showprofile = False | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r210 | xmin = None | |
| xmax = None | |||
| ymin = None | |||
| ymax = None | |||
| zmin = None | |||
| zmax = None | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r318 | x_buffer = None | |
| z_buffer = None | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r396 | decimationx = None | |
| decimationy = None | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r813 | __MAXNUMX = 200 | |
Erick Valdez
|
r869 | __MAXNUMY = 400 | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r813 | __MAXNUMTIME = 500 | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | def __init__(self, *args): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | """ | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | Input: | |
| *args : Los parametros necesarios son | |||
| fig, nrow, ncol, xpos, ypos, colspan, rowspan | |||
| """ | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | ax = self.__driver.createAxes(*args) | |
| self.fig = args[0] | |||
|
Daniel Valdez
|
r190 | self.ax = ax | |
|
Miguel Valdez
|
r201 | self.plot = None | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r210 | self.__firsttime = True | |
|
Daniel Valdez
|
r239 | self.idlineList = [] | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r318 | self.x_buffer = numpy.array([]) | |
| self.z_buffer = numpy.array([]) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | def setText(self, text): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | self.__driver.setAxesText(self.ax, text) | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r209 | def setXAxisAsTime(self): | |
| pass | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | def pline(self, x, y, | |
| xmin=None, xmax=None, | |||
| ymin=None, ymax=None, | |||
| xlabel='', ylabel='', | |||
| title='', | |||
| **kwargs): | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | """ | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | Input: | |
|
José Chávez
|
r897 | x : | |
| y : | |||
|
Miguel Valdez
|
r201 | xmin : | |
| xmax : | |||
| ymin : | |||
|
José Chávez
|
r897 | ymax : | |
|
Miguel Valdez
|
r201 | xlabel : | |
| ylabel : | |||
| title : | |||
| **kwargs : Los parametros aceptados son | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | ticksize | |
|
Miguel Valdez
|
r204 | ytick_visible | |
|
Miguel Valdez
|
r201 | """ | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r210 | if self.__firsttime: | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | 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) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | self.plot = self.__driver.createPline(self.ax, x, y, | |
| xmin, xmax, | |||
| ymin, ymax, | |||
| xlabel=xlabel, | |||
| ylabel=ylabel, | |||
| title=title, | |||
| **kwargs) | |||
|
Daniel Valdez
|
r239 | ||
| self.idlineList.append(0) | |||
|
Miguel Valdez
|
r210 | self.__firsttime = False | |
|
Miguel Valdez
|
r201 | return | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | self.__driver.pline(self.plot, x, y, xlabel=xlabel, | |
| ylabel=ylabel, | |||
| title=title) | |||
|
Daniel Valdez
|
r239 | ||
|
Miguel Valdez
|
r631 | # self.__driver.pause() | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r239 | def addpline(self, x, y, idline, **kwargs): | |
| lines = self.ax.lines | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r239 | if idline in self.idlineList: | |
| self.__driver.set_linedata(self.ax, x, y, idline) | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r239 | if idline not in(self.idlineList): | |
| self.__driver.addpline(self.ax, x, y, **kwargs) | |||
| self.idlineList.append(idline) | |||
| return | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r229 | def pmultiline(self, x, y, | |
| xmin=None, xmax=None, | |||
| ymin=None, ymax=None, | |||
| xlabel='', ylabel='', | |||
| title='', | |||
| **kwargs): | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r229 | if self.__firsttime: | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r229 | 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) | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r229 | self.plot = self.__driver.createPmultiline(self.ax, x, y, | |
| xmin, xmax, | |||
| ymin, ymax, | |||
| xlabel=xlabel, | |||
| ylabel=ylabel, | |||
| title=title, | |||
| **kwargs) | |||
| self.__firsttime = False | |||
| return | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r229 | self.__driver.pmultiline(self.plot, x, y, xlabel=xlabel, | |
| ylabel=ylabel, | |||
| title=title) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r631 | # self.__driver.pause() | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r240 | def pmultilineyaxis(self, x, y, | |
| xmin=None, xmax=None, | |||
| ymin=None, ymax=None, | |||
| xlabel='', ylabel='', | |||
| title='', | |||
| **kwargs): | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r240 | if self.__firsttime: | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r240 | 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) | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r240 | self.plot = self.__driver.createPmultilineYAxis(self.ax, x, y, | |
| xmin, xmax, | |||
| ymin, ymax, | |||
| xlabel=xlabel, | |||
| ylabel=ylabel, | |||
| title=title, | |||
| **kwargs) | |||
|
Daniel Valdez
|
r245 | if self.xmin == None: self.xmin = xmin | |
| if self.xmax == None: self.xmax = xmax | |||
| if self.ymin == None: self.ymin = ymin | |||
| if self.ymax == None: self.ymax = ymax | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r240 | self.__firsttime = False | |
| return | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r240 | self.__driver.pmultilineyaxis(self.plot, x, y, xlabel=xlabel, | |
| ylabel=ylabel, | |||
| title=title) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r631 | # self.__driver.pause() | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | def pcolor(self, x, y, z, | |
| xmin=None, xmax=None, | |||
| ymin=None, ymax=None, | |||
| zmin=None, zmax=None, | |||
| xlabel='', ylabel='', | |||
|
Miguel Valdez
|
r813 | title='', colormap='jet', | |
|
Miguel Valdez
|
r201 | **kwargs): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | """ | |
| Input: | |||
|
José Chávez
|
r897 | x : | |
| y : | |||
|
Miguel Valdez
|
r201 | x : | |
| xmin : | |||
| xmax : | |||
| ymin : | |||
| ymax : | |||
| zmin : | |||
| zmax : | |||
| xlabel : | |||
| ylabel : | |||
| title : | |||
| **kwargs : Los parametros aceptados son | |||
| ticksize=9, | |||
| cblabel='' | |||
| """ | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r813 | #Decimating data | |
| xlen = len(x) | |||
| ylen = len(y) | |||
|
José Chávez
|
r897 | ||
| decimationx = int(xlen/self.__MAXNUMX) + 1 | |||
Juan C. Valdez
|
r876 | decimationy = int(ylen/self.__MAXNUMY) + 1 | |
|
José Chávez
|
r897 | ||
|
Juan C. Valdez
|
r860 | x_buffer = x#[::decimationx] | |
|
José Chávez
|
r897 | y_buffer = y#[::decimationy] | |
|
Juan C. Valdez
|
r860 | z_buffer = z#[::decimationx, ::decimationy] | |
|
Miguel Valdez
|
r813 | #=================================================== | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r210 | if self.__firsttime: | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r201 | 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) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r813 | self.plot = self.__driver.createPcolor(self.ax, x_buffer, | |
| y_buffer, | |||
| z_buffer, | |||
|
Miguel Valdez
|
r201 | xmin, xmax, | |
| ymin, ymax, | |||
| zmin, zmax, | |||
| xlabel=xlabel, | |||
| ylabel=ylabel, | |||
| title=title, | |||
|
Daniel Valdez
|
r229 | colormap=colormap, | |
|
Miguel Valdez
|
r201 | **kwargs) | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r210 | if self.xmin == None: self.xmin = xmin | |
| if self.xmax == None: self.xmax = xmax | |||
| if self.ymin == None: self.ymin = ymin | |||
| if self.ymax == None: self.ymax = ymax | |||
| if self.zmin == None: self.zmin = zmin | |||
| if self.zmax == None: self.zmax = zmax | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r210 | self.__firsttime = False | |
|
Miguel Valdez
|
r201 | return | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r813 | self.__driver.pcolor(self.plot, | |
| z_buffer, | |||
|
Miguel Valdez
|
r210 | xlabel=xlabel, | |
| ylabel=ylabel, | |||
| title=title) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r631 | # self.__driver.pause() | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r318 | def pcolorbuffer(self, x, y, z, | |
| xmin=None, xmax=None, | |||
| ymin=None, ymax=None, | |||
| zmin=None, zmax=None, | |||
| xlabel='', ylabel='', | |||
|
Daniel Valdez
|
r396 | title='', rti = True, colormap='jet', | |
|
Daniel Valdez
|
r395 | maxNumX = None, maxNumY = None, | |
|
Daniel Valdez
|
r318 | **kwargs): | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r395 | if maxNumX == None: | |
|
Miguel Valdez
|
r813 | maxNumX = self.__MAXNUMTIME | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r395 | if maxNumY == None: | |
|
José Chávez
|
r897 | maxNumY = self.__MAXNUMY | |
| if self.__firsttime: | |||
|
Daniel Valdez
|
r318 | self.z_buffer = z | |
| self.x_buffer = numpy.hstack((self.x_buffer, x)) | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r318 | 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) | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r318 | self.plot = self.__driver.createPcolor(self.ax, self.x_buffer, y, z, | |
| xmin, xmax, | |||
| ymin, ymax, | |||
| zmin, zmax, | |||
| xlabel=xlabel, | |||
| ylabel=ylabel, | |||
| title=title, | |||
| colormap=colormap, | |||
| **kwargs) | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r318 | if self.xmin == None: self.xmin = xmin | |
| if self.xmax == None: self.xmax = xmax | |||
| if self.ymin == None: self.ymin = ymin | |||
| if self.ymax == None: self.ymax = ymax | |||
| if self.zmin == None: self.zmin = zmin | |||
| if self.zmax == None: self.zmax = zmax | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r396 | self.__firsttime = False | |
| return | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r779 | self.x_buffer = numpy.hstack((self.x_buffer[:-1], x[0], x[-1])) | |
|
Daniel Valdez
|
r396 | self.z_buffer = numpy.hstack((self.z_buffer, z)) | |
|
Miguel Valdez
|
r813 | z_buffer = self.z_buffer.reshape(-1,len(y)) | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r813 | #Decimating data | |
| xlen = len(self.x_buffer) | |||
| ylen = len(y) | |||
|
José Chávez
|
r897 | ||
|
Juan C. Valdez
|
r876 | decimationx = int(xlen/maxNumX) + 1 | |
| decimationy = int(ylen/maxNumY) + 1 | |||
|
José Chávez
|
r897 | ||
|
Juan C. Valdez
|
r860 | x_buffer = self.x_buffer#[::decimationx] | |
|
José Chávez
|
r897 | y_buffer = y#[::decimationy] | |
|
Juan C. Valdez
|
r860 | z_buffer = z_buffer#[::decimationx, ::decimationy] | |
|
Daniel Valdez
|
r396 | #=================================================== | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r396 | x_buffer, y_buffer, z_buffer = self.__fillGaps(x_buffer, y_buffer, z_buffer) | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r396 | self.__driver.addpcolorbuffer(self.ax, x_buffer, y_buffer, z_buffer, self.zmin, self.zmax, | |
| xlabel=xlabel, | |||
| ylabel=ylabel, | |||
| title=title, | |||
| colormap=colormap) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r631 | # self.__driver.pause() | |
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | def polar(self, x, y, | |
| title='', xlabel='',ylabel='',**kwargs): | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | if self.__firsttime: | |
| self.plot = self.__driver.createPolar(self.ax, x, y, title = title, xlabel = xlabel, ylabel = ylabel) | |||
| self.__firsttime = False | |||
| self.x_buffer = x | |||
| self.y_buffer = y | |||
| return | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | self.x_buffer = numpy.hstack((self.x_buffer,x)) | |
| self.y_buffer = numpy.hstack((self.y_buffer,y)) | |||
| self.__driver.polar(self.plot, self.x_buffer, self.y_buffer, xlabel=xlabel, | |||
| ylabel=ylabel, | |||
| title=title) | |||
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r631 | # self.__driver.pause() | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r396 | def __fillGaps(self, x_buffer, y_buffer, z_buffer): | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r657 | if x_buffer.shape[0] < 2: | |
| return x_buffer, y_buffer, z_buffer | |||
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r396 | deltas = x_buffer[1:] - x_buffer[0:-1] | |
| x_median = numpy.median(deltas) | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r663 | index = numpy.where(deltas > 5*x_median) | |
|
José Chávez
|
r897 | ||
|
Daniel Valdez
|
r396 | if len(index[0]) != 0: | |
| z_buffer[index[0],::] = self.__missing | |||
| z_buffer = numpy.ma.masked_inside(z_buffer,0.99*self.__missing,1.01*self.__missing) | |||
|
George Yong
|
r1167 | return x_buffer, y_buffer, z_buffer |