|
|
import os
|
|
|
import numpy
|
|
|
import time, datetime
|
|
|
import mpldriver
|
|
|
|
|
|
from schainpy.model.proc.jroproc_base import Operation
|
|
|
|
|
|
def isRealtime(utcdatatime):
|
|
|
utcnow = time.mktime(time.localtime())
|
|
|
delta = abs(utcnow - utcdatatime) # abs
|
|
|
if delta >= 30.:
|
|
|
return False
|
|
|
return True
|
|
|
|
|
|
class Figure(Operation):
|
|
|
|
|
|
__driver = mpldriver
|
|
|
__isConfigThread = False
|
|
|
fig = None
|
|
|
|
|
|
id = None
|
|
|
wintitle = None
|
|
|
width = None
|
|
|
height = None
|
|
|
nplots = None
|
|
|
timerange = None
|
|
|
|
|
|
axesObjList = []
|
|
|
|
|
|
WIDTH = None
|
|
|
HEIGHT = None
|
|
|
PREFIX = 'fig'
|
|
|
|
|
|
xmin = None
|
|
|
xmax = None
|
|
|
|
|
|
counter_imagwr = 0
|
|
|
|
|
|
figfile = None
|
|
|
|
|
|
def __init__(self):
|
|
|
|
|
|
raise ValueError, "This method is not implemented"
|
|
|
|
|
|
def __del__(self):
|
|
|
|
|
|
self.__driver.closeFigure()
|
|
|
|
|
|
def getFilename(self, name, ext='.png'):
|
|
|
|
|
|
path = '%s%03d' %(self.PREFIX, self.id)
|
|
|
filename = '%s_%s%s' %(self.PREFIX, name, ext)
|
|
|
return os.path.join(path, filename)
|
|
|
|
|
|
def getAxesObjList(self):
|
|
|
|
|
|
return self.axesObjList
|
|
|
|
|
|
def getSubplots(self):
|
|
|
|
|
|
raise ValueError, "Abstract method: This method should be defined"
|
|
|
|
|
|
def getScreenDim(self, widthplot, heightplot):
|
|
|
|
|
|
nrow, ncol = self.getSubplots()
|
|
|
|
|
|
widthscreen = widthplot*ncol
|
|
|
heightscreen = heightplot*nrow
|
|
|
|
|
|
return widthscreen, heightscreen
|
|
|
|
|
|
def getTimeLim(self, x, xmin=None, xmax=None, timerange=None):
|
|
|
|
|
|
if self.xmin != None and self.xmax != None:
|
|
|
if timerange == None:
|
|
|
timerange = self.xmax - self.xmin
|
|
|
xmin = self.xmin + timerange
|
|
|
xmax = self.xmax + timerange
|
|
|
|
|
|
return xmin, xmax
|
|
|
|
|
|
if timerange == None and (xmin==None or xmax==None):
|
|
|
timerange = 14400 #seconds
|
|
|
#raise ValueError, "(timerange) or (xmin & xmax) should be defined"
|
|
|
|
|
|
if timerange != None:
|
|
|
txmin = x[0] - x[0] % min(timerange/10, 10*60)
|
|
|
else:
|
|
|
txmin = x[0] - x[0] % 10*60
|
|
|
|
|
|
thisdatetime = datetime.datetime.utcfromtimestamp(txmin)
|
|
|
thisdate = datetime.datetime.combine(thisdatetime.date(), datetime.time(0,0,0))
|
|
|
|
|
|
if timerange != None:
|
|
|
xmin = (thisdatetime - thisdate).seconds/(60*60.)
|
|
|
xmax = xmin + timerange/(60*60.)
|
|
|
|
|
|
mindt = thisdate + datetime.timedelta(hours=xmin) - datetime.timedelta(seconds=time.timezone)
|
|
|
xmin_sec = time.mktime(mindt.timetuple())
|
|
|
|
|
|
maxdt = thisdate + datetime.timedelta(hours=xmax) - datetime.timedelta(seconds=time.timezone)
|
|
|
xmax_sec = time.mktime(maxdt.timetuple())
|
|
|
|
|
|
return xmin_sec, xmax_sec
|
|
|
|
|
|
def init(self, id, nplots, wintitle):
|
|
|
|
|
|
raise ValueError, "This method has been replaced with createFigure"
|
|
|
|
|
|
def createFigure(self, id, wintitle, widthplot=None, heightplot=None, show=True):
|
|
|
|
|
|
"""
|
|
|
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)
|
|
|
|
|
|
Input:
|
|
|
id : Los parametros necesarios son
|
|
|
wintitle :
|
|
|
|
|
|
"""
|
|
|
|
|
|
if widthplot == None:
|
|
|
widthplot = self.WIDTH
|
|
|
|
|
|
if heightplot == None:
|
|
|
heightplot = self.HEIGHT
|
|
|
|
|
|
self.id = id
|
|
|
|
|
|
self.wintitle = wintitle
|
|
|
|
|
|
self.widthscreen, self.heightscreen = self.getScreenDim(widthplot, heightplot)
|
|
|
|
|
|
self.fig = self.__driver.createFigure(id=self.id,
|
|
|
wintitle=self.wintitle,
|
|
|
width=self.widthscreen,
|
|
|
height=self.heightscreen,
|
|
|
show=show)
|
|
|
|
|
|
self.axesObjList = []
|
|
|
self.counter_imagwr = 0
|
|
|
|
|
|
|
|
|
def setDriver(self, driver=mpldriver):
|
|
|
|
|
|
self.__driver = driver
|
|
|
|
|
|
def setTitle(self, title):
|
|
|
|
|
|
self.__driver.setTitle(self.fig, title)
|
|
|
|
|
|
def setWinTitle(self, title):
|
|
|
|
|
|
self.__driver.setWinTitle(self.fig, title=title)
|
|
|
|
|
|
def setTextFromAxes(self, text):
|
|
|
|
|
|
raise ValueError, "Este metodo ha sido reemplazaado con el metodo setText de la clase Axes"
|
|
|
|
|
|
def makeAxes(self, nrow, ncol, xpos, ypos, colspan, rowspan):
|
|
|
|
|
|
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)
|
|
|
|
|
|
def saveFigure(self, figpath, figfile, *args):
|
|
|
|
|
|
filename = os.path.join(figpath, figfile)
|
|
|
|
|
|
fullpath = os.path.split(filename)[0]
|
|
|
|
|
|
if not os.path.exists(fullpath):
|
|
|
subpath = os.path.split(fullpath)[0]
|
|
|
|
|
|
if not os.path.exists(subpath):
|
|
|
os.mkdir(subpath)
|
|
|
|
|
|
os.mkdir(fullpath)
|
|
|
|
|
|
self.__driver.saveFigure(self.fig, filename, *args)
|
|
|
|
|
|
def save(self, figpath, figfile=None, save=True, ftp=False, wr_period=1, thisDatetime=None, update_figfile=True):
|
|
|
|
|
|
self.counter_imagwr += 1
|
|
|
if self.counter_imagwr < wr_period:
|
|
|
return
|
|
|
|
|
|
self.counter_imagwr = 0
|
|
|
|
|
|
if save:
|
|
|
|
|
|
if figfile == None:
|
|
|
|
|
|
if not thisDatetime:
|
|
|
raise ValueError, "Saving figure: figfile or thisDatetime should be defined"
|
|
|
return
|
|
|
|
|
|
str_datetime = thisDatetime.strftime("%Y%m%d_%H%M%S")
|
|
|
figfile = self.getFilename(name = str_datetime)
|
|
|
|
|
|
if self.figfile == None:
|
|
|
self.figfile = figfile
|
|
|
|
|
|
if update_figfile:
|
|
|
self.figfile = figfile
|
|
|
|
|
|
# store png plot to local folder
|
|
|
self.saveFigure(figpath, self.figfile)
|
|
|
|
|
|
|
|
|
if not ftp:
|
|
|
return
|
|
|
|
|
|
if not thisDatetime:
|
|
|
return
|
|
|
|
|
|
# store png plot to FTP server according to RT-Web format
|
|
|
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)
|
|
|
self.saveFigure(figpath, ftp_filename)
|
|
|
|
|
|
def getNameToFtp(self, thisDatetime, FTP_WEI, EXP_CODE, SUB_EXP_CODE, PLOT_CODE, PLOT_POS):
|
|
|
YEAR_STR = '%4.4d'%thisDatetime.timetuple().tm_year
|
|
|
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
|
|
|
|
|
|
def draw(self):
|
|
|
|
|
|
self.__driver.draw(self.fig)
|
|
|
|
|
|
def run(self):
|
|
|
|
|
|
raise ValueError, "This method is not implemented"
|
|
|
|
|
|
def close(self, show=False):
|
|
|
|
|
|
self.__driver.closeFigure(show=show, fig=self.fig)
|
|
|
|
|
|
axesList = property(getAxesObjList)
|
|
|
|
|
|
|
|
|
class Axes:
|
|
|
|
|
|
__driver = mpldriver
|
|
|
fig = None
|
|
|
ax = None
|
|
|
plot = None
|
|
|
__missing = 1E30
|
|
|
__firsttime = None
|
|
|
|
|
|
__showprofile = False
|
|
|
|
|
|
xmin = None
|
|
|
xmax = None
|
|
|
ymin = None
|
|
|
ymax = None
|
|
|
zmin = None
|
|
|
zmax = None
|
|
|
|
|
|
x_buffer = None
|
|
|
z_buffer = None
|
|
|
|
|
|
decimationx = None
|
|
|
decimationy = None
|
|
|
|
|
|
__MAXNUMX = 300
|
|
|
__MAXNUMY = 150
|
|
|
|
|
|
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]
|
|
|
self.ax = ax
|
|
|
self.plot = None
|
|
|
|
|
|
self.__firsttime = True
|
|
|
self.idlineList = []
|
|
|
|
|
|
self.x_buffer = numpy.array([])
|
|
|
self.z_buffer = numpy.array([])
|
|
|
|
|
|
def setText(self, text):
|
|
|
|
|
|
self.__driver.setAxesText(self.ax, text)
|
|
|
|
|
|
def setXAxisAsTime(self):
|
|
|
pass
|
|
|
|
|
|
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
|
|
|
ytick_visible
|
|
|
"""
|
|
|
|
|
|
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.idlineList.append(0)
|
|
|
self.__firsttime = False
|
|
|
return
|
|
|
|
|
|
self.__driver.pline(self.plot, x, y, xlabel=xlabel,
|
|
|
ylabel=ylabel,
|
|
|
title=title)
|
|
|
|
|
|
def addpline(self, x, y, idline, **kwargs):
|
|
|
lines = self.ax.lines
|
|
|
|
|
|
if idline in self.idlineList:
|
|
|
self.__driver.set_linedata(self.ax, x, y, idline)
|
|
|
|
|
|
if idline not in(self.idlineList):
|
|
|
self.__driver.addpline(self.ax, x, y, **kwargs)
|
|
|
self.idlineList.append(idline)
|
|
|
|
|
|
return
|
|
|
|
|
|
def pmultiline(self, x, y,
|
|
|
xmin=None, xmax=None,
|
|
|
ymin=None, ymax=None,
|
|
|
xlabel='', ylabel='',
|
|
|
title='',
|
|
|
**kwargs):
|
|
|
|
|
|
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.createPmultiline(self.ax, x, y,
|
|
|
xmin, xmax,
|
|
|
ymin, ymax,
|
|
|
xlabel=xlabel,
|
|
|
ylabel=ylabel,
|
|
|
title=title,
|
|
|
**kwargs)
|
|
|
self.__firsttime = False
|
|
|
return
|
|
|
|
|
|
self.__driver.pmultiline(self.plot, x, y, xlabel=xlabel,
|
|
|
ylabel=ylabel,
|
|
|
title=title)
|
|
|
|
|
|
def pmultilineyaxis(self, x, y,
|
|
|
xmin=None, xmax=None,
|
|
|
ymin=None, ymax=None,
|
|
|
xlabel='', ylabel='',
|
|
|
title='',
|
|
|
**kwargs):
|
|
|
|
|
|
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.createPmultilineYAxis(self.ax, x, y,
|
|
|
xmin, xmax,
|
|
|
ymin, ymax,
|
|
|
xlabel=xlabel,
|
|
|
ylabel=ylabel,
|
|
|
title=title,
|
|
|
**kwargs)
|
|
|
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
|
|
|
|
|
|
self.__firsttime = False
|
|
|
return
|
|
|
|
|
|
self.__driver.pmultilineyaxis(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='', rti = False, colormap='jet',
|
|
|
**kwargs):
|
|
|
|
|
|
"""
|
|
|
Input:
|
|
|
x :
|
|
|
y :
|
|
|
x :
|
|
|
xmin :
|
|
|
xmax :
|
|
|
ymin :
|
|
|
ymax :
|
|
|
zmin :
|
|
|
zmax :
|
|
|
xlabel :
|
|
|
ylabel :
|
|
|
title :
|
|
|
**kwargs : Los parametros aceptados son
|
|
|
ticksize=9,
|
|
|
cblabel=''
|
|
|
rti = True or False
|
|
|
"""
|
|
|
|
|
|
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,
|
|
|
colormap=colormap,
|
|
|
**kwargs)
|
|
|
|
|
|
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
|
|
|
|
|
|
self.__firsttime = False
|
|
|
return
|
|
|
|
|
|
if rti:
|
|
|
self.__driver.addpcolor(self.ax, x, y, z, self.zmin, self.zmax,
|
|
|
xlabel=xlabel,
|
|
|
ylabel=ylabel,
|
|
|
title=title,
|
|
|
colormap=colormap)
|
|
|
return
|
|
|
|
|
|
self.__driver.pcolor(self.plot, z,
|
|
|
xlabel=xlabel,
|
|
|
ylabel=ylabel,
|
|
|
title=title)
|
|
|
|
|
|
def pcolorbuffer(self, x, y, z,
|
|
|
xmin=None, xmax=None,
|
|
|
ymin=None, ymax=None,
|
|
|
zmin=None, zmax=None,
|
|
|
xlabel='', ylabel='',
|
|
|
title='', rti = True, colormap='jet',
|
|
|
maxNumX = None, maxNumY = None,
|
|
|
**kwargs):
|
|
|
|
|
|
if maxNumX == None:
|
|
|
maxNumX = self.__MAXNUMX
|
|
|
|
|
|
if maxNumY == None:
|
|
|
maxNumY = self.__MAXNUMY
|
|
|
|
|
|
if self.__firsttime:
|
|
|
self.z_buffer = z
|
|
|
self.x_buffer = numpy.hstack((self.x_buffer, x))
|
|
|
|
|
|
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, self.x_buffer, y, z,
|
|
|
xmin, xmax,
|
|
|
ymin, ymax,
|
|
|
zmin, zmax,
|
|
|
xlabel=xlabel,
|
|
|
ylabel=ylabel,
|
|
|
title=title,
|
|
|
colormap=colormap,
|
|
|
**kwargs)
|
|
|
|
|
|
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
|
|
|
|
|
|
self.__firsttime = False
|
|
|
return
|
|
|
|
|
|
self.x_buffer = numpy.hstack((self.x_buffer, x[-1]))
|
|
|
self.z_buffer = numpy.hstack((self.z_buffer, z))
|
|
|
|
|
|
if self.decimationx == None:
|
|
|
deltax = float(self.xmax - self.xmin)/maxNumX
|
|
|
deltay = float(self.ymax - self.ymin)/maxNumY
|
|
|
|
|
|
resolutionx = self.x_buffer[2]-self.x_buffer[0]
|
|
|
resolutiony = y[1]-y[0]
|
|
|
|
|
|
self.decimationx = numpy.ceil(deltax / resolutionx)
|
|
|
self.decimationy = numpy.ceil(deltay / resolutiony)
|
|
|
|
|
|
z_buffer = self.z_buffer.reshape(-1,len(y))
|
|
|
|
|
|
x_buffer = self.x_buffer[::self.decimationx]
|
|
|
y_buffer = y[::self.decimationy]
|
|
|
z_buffer = z_buffer[::self.decimationx, ::self.decimationy]
|
|
|
#===================================================
|
|
|
|
|
|
x_buffer, y_buffer, z_buffer = self.__fillGaps(x_buffer, y_buffer, z_buffer)
|
|
|
|
|
|
self.__driver.addpcolorbuffer(self.ax, x_buffer, y_buffer, z_buffer, self.zmin, self.zmax,
|
|
|
xlabel=xlabel,
|
|
|
ylabel=ylabel,
|
|
|
title=title,
|
|
|
colormap=colormap)
|
|
|
|
|
|
def polar(self, x, y,
|
|
|
title='', xlabel='',ylabel='',**kwargs):
|
|
|
|
|
|
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
|
|
|
|
|
|
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)
|
|
|
|
|
|
def __fillGaps(self, x_buffer, y_buffer, z_buffer):
|
|
|
|
|
|
deltas = x_buffer[1:] - x_buffer[0:-1]
|
|
|
x_median = numpy.median(deltas)
|
|
|
|
|
|
index = numpy.where(deltas >= 2*x_median)
|
|
|
|
|
|
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)
|
|
|
|
|
|
return x_buffer, y_buffer, z_buffer
|
|
|
|
|
|
|
|
|
|
|
|
|
