import os import sys import datetime import numpy import matplotlib if 'BACKEND' in os.environ: matplotlib.use(os.environ['BACKEND']) elif 'linux' in sys.platform: matplotlib.use("TkAgg") elif 'darwin' in sys.platform: matplotlib.use('TkAgg') else: from schainpy.utils import log log.warning('Using default Backend="Agg"', 'INFO') matplotlib.use('Agg') # Qt4Agg', 'GTK', 'GTKAgg', 'ps', 'agg', 'cairo', 'MacOSX', 'GTKCairo', 'WXAgg', 'template', 'TkAgg', 'GTK3Cairo', 'GTK3Agg', 'svg', 'WebAgg', 'CocoaAgg', 'emf', 'gdk', 'WX' import matplotlib.pyplot from mpl_toolkits.axes_grid1 import make_axes_locatable from matplotlib.ticker import FuncFormatter, LinearLocator ########################################### # Actualizacion de las funciones del driver ########################################### # create jro colormap jet_values = matplotlib.pyplot.get_cmap("jet", 100)(numpy.arange(100))[10:90] blu_values = matplotlib.pyplot.get_cmap( "seismic_r", 20)(numpy.arange(20))[10:15] ncmap = matplotlib.colors.LinearSegmentedColormap.from_list( "jro", numpy.vstack((blu_values, jet_values))) matplotlib.pyplot.register_cmap(cmap=ncmap) def createFigure(id, wintitle, width, height, facecolor="w", show=True, dpi=80): matplotlib.pyplot.ioff() fig = matplotlib.pyplot.figure(num=id, facecolor=facecolor, figsize=( 1.0 * width / dpi, 1.0 * height / dpi)) fig.canvas.manager.set_window_title(wintitle) # fig.canvas.manager.resize(width, height) matplotlib.pyplot.ion() if show: matplotlib.pyplot.show() return fig def closeFigure(show=False, fig=None): # matplotlib.pyplot.ioff() # matplotlib.pyplot.pause(0) if show: matplotlib.pyplot.show() if fig != None: matplotlib.pyplot.close(fig) # matplotlib.pyplot.pause(0) # matplotlib.pyplot.ion() return matplotlib.pyplot.close("all") # matplotlib.pyplot.pause(0) # matplotlib.pyplot.ion() return def saveFigure(fig, filename): # matplotlib.pyplot.ioff() fig.savefig(filename, dpi=matplotlib.pyplot.gcf().dpi) # matplotlib.pyplot.ion() def clearFigure(fig): fig.clf() def setWinTitle(fig, title): fig.canvas.manager.set_window_title(title) def setTitle(fig, title): fig.suptitle(title) def createAxes(fig, nrow, ncol, xpos, ypos, colspan, rowspan, polar=False): matplotlib.pyplot.ioff() matplotlib.pyplot.figure(fig.number) axes = matplotlib.pyplot.subplot2grid((nrow, ncol), (xpos, ypos), colspan=colspan, rowspan=rowspan, polar=polar) matplotlib.pyplot.ion() return axes def setAxesText(ax, text): ax.annotate(text, xy=(.1, .99), xycoords='figure fraction', horizontalalignment='left', verticalalignment='top', fontsize=10) def printLabels(ax, xlabel, ylabel, title): ax.set_xlabel(xlabel, size=11) ax.set_ylabel(ylabel, size=11) ax.set_title(title, size=8) def createPline(ax, x, y, xmin, xmax, ymin, ymax, xlabel='', ylabel='', title='', ticksize=9, xtick_visible=True, ytick_visible=True, nxticks=4, nyticks=10, grid=None, color='blue'): """ Input: grid : None, 'both', 'x', 'y' """ matplotlib.pyplot.ioff() ax.set_xlim([xmin, xmax]) ax.set_ylim([ymin, ymax]) printLabels(ax, xlabel, ylabel, title) ###################################################### if (xmax - xmin) <= 1: xtickspos = numpy.linspace(xmin, xmax, nxticks) xtickspos = numpy.array([float("%.1f" % i) for i in xtickspos]) ax.set_xticks(xtickspos) else: xtickspos = numpy.arange(nxticks) * \ int((xmax - xmin) / (nxticks)) + int(xmin) # xtickspos = numpy.arange(nxticks)*float(xmax-xmin)/float(nxticks) + int(xmin) ax.set_xticks(xtickspos) for tick in ax.get_xticklabels(): tick.set_visible(xtick_visible) for tick in ax.xaxis.get_major_ticks(): tick.label.set_fontsize(ticksize) ###################################################### for tick in ax.get_yticklabels(): tick.set_visible(ytick_visible) for tick in ax.yaxis.get_major_ticks(): tick.label.set_fontsize(ticksize) ax.plot(x, y, color=color) iplot = ax.lines[-1] ###################################################### if '0.' in matplotlib.__version__[0:2]: print "The matplotlib version has to be updated to 1.1 or newer" return iplot if '1.0.' in matplotlib.__version__[0:4]: print "The matplotlib version has to be updated to 1.1 or newer" return iplot if grid != None: ax.grid(b=True, which='major', axis=grid) matplotlib.pyplot.tight_layout() matplotlib.pyplot.ion() return iplot def set_linedata(ax, x, y, idline): ax.lines[idline].set_data(x, y) def pline(iplot, x, y, xlabel='', ylabel='', title=''): ax = iplot.axes printLabels(ax, xlabel, ylabel, title) set_linedata(ax, x, y, idline=0) def addpline(ax, x, y, color, linestyle, lw): ax.plot(x, y, color=color, linestyle=linestyle, lw=lw) def createPcolor(ax, x, y, z, xmin, xmax, ymin, ymax, zmin, zmax, xlabel='', ylabel='', title='', ticksize=9, colormap='jet', cblabel='', cbsize="5%", XAxisAsTime=False): matplotlib.pyplot.ioff() divider = make_axes_locatable(ax) ax_cb = divider.new_horizontal(size=cbsize, pad=0.05) fig = ax.get_figure() fig.add_axes(ax_cb) ax.set_xlim([xmin, xmax]) ax.set_ylim([ymin, ymax]) printLabels(ax, xlabel, ylabel, title) z = numpy.ma.masked_invalid(z) cmap = matplotlib.pyplot.get_cmap(colormap) cmap.set_bad('black', 1.) imesh = ax.pcolormesh(x, y, z.T, vmin=zmin, vmax=zmax, cmap=cmap) cb = matplotlib.pyplot.colorbar(imesh, cax=ax_cb) cb.set_label(cblabel) # for tl in ax_cb.get_yticklabels(): # tl.set_visible(True) for tick in ax.yaxis.get_major_ticks(): tick.label.set_fontsize(ticksize) for tick in ax.xaxis.get_major_ticks(): tick.label.set_fontsize(ticksize) for tick in cb.ax.get_yticklabels(): tick.set_fontsize(ticksize) ax_cb.yaxis.tick_right() if '0.' in matplotlib.__version__[0:2]: print "The matplotlib version has to be updated to 1.1 or newer" return imesh if '1.0.' in matplotlib.__version__[0:4]: print "The matplotlib version has to be updated to 1.1 or newer" return imesh matplotlib.pyplot.tight_layout() if XAxisAsTime: def func(x, pos): return ('%s') % ( datetime.datetime.utcfromtimestamp(x).strftime("%H:%M:%S")) ax.xaxis.set_major_formatter(FuncFormatter(func)) ax.xaxis.set_major_locator(LinearLocator(7)) matplotlib.pyplot.ion() return imesh def pcolor(imesh, z, xlabel='', ylabel='', title=''): z = z.T ax = imesh.axes printLabels(ax, xlabel, ylabel, title) imesh.set_array(z.ravel()) def addpcolor(ax, x, y, z, zmin, zmax, xlabel='', ylabel='', title='', colormap='jet'): printLabels(ax, xlabel, ylabel, title) ax.pcolormesh(x, y, z.T, vmin=zmin, vmax=zmax, cmap=matplotlib.pyplot.get_cmap(colormap)) def addpcolorbuffer(ax, x, y, z, zmin, zmax, xlabel='', ylabel='', title='', colormap='jet'): printLabels(ax, xlabel, ylabel, title) ax.collections.remove(ax.collections[0]) z = numpy.ma.masked_invalid(z) cmap = matplotlib.pyplot.get_cmap(colormap) cmap.set_bad('black', 1.) ax.pcolormesh(x, y, z.T, vmin=zmin, vmax=zmax, cmap=cmap) def createPmultiline(ax, x, y, xmin, xmax, ymin, ymax, xlabel='', ylabel='', title='', legendlabels=None, ticksize=9, xtick_visible=True, ytick_visible=True, nxticks=4, nyticks=10, grid=None): """ Input: grid : None, 'both', 'x', 'y' """ matplotlib.pyplot.ioff() lines = ax.plot(x.T, y) leg = ax.legend(lines, legendlabels, loc='upper right') leg.get_frame().set_alpha(0.5) ax.set_xlim([xmin, xmax]) ax.set_ylim([ymin, ymax]) printLabels(ax, xlabel, ylabel, title) xtickspos = numpy.arange(nxticks) * \ int((xmax - xmin) / (nxticks)) + int(xmin) ax.set_xticks(xtickspos) for tick in ax.get_xticklabels(): tick.set_visible(xtick_visible) for tick in ax.xaxis.get_major_ticks(): tick.label.set_fontsize(ticksize) for tick in ax.get_yticklabels(): tick.set_visible(ytick_visible) for tick in ax.yaxis.get_major_ticks(): tick.label.set_fontsize(ticksize) iplot = ax.lines[-1] if '0.' in matplotlib.__version__[0:2]: print "The matplotlib version has to be updated to 1.1 or newer" return iplot if '1.0.' in matplotlib.__version__[0:4]: print "The matplotlib version has to be updated to 1.1 or newer" return iplot if grid != None: ax.grid(b=True, which='major', axis=grid) matplotlib.pyplot.tight_layout() matplotlib.pyplot.ion() return iplot def pmultiline(iplot, x, y, xlabel='', ylabel='', title=''): ax = iplot.axes printLabels(ax, xlabel, ylabel, title) for i in range(len(ax.lines)): line = ax.lines[i] line.set_data(x[i, :], y) def createPmultilineYAxis(ax, x, y, xmin, xmax, ymin, ymax, xlabel='', ylabel='', title='', legendlabels=None, ticksize=9, xtick_visible=True, ytick_visible=True, nxticks=4, nyticks=10, marker='.', markersize=10, linestyle="None", grid=None, XAxisAsTime=False): """ Input: grid : None, 'both', 'x', 'y' """ matplotlib.pyplot.ioff() # lines = ax.plot(x, y.T, marker=marker,markersize=markersize,linestyle=linestyle) lines = ax.plot(x, y.T) # leg = ax.legend(lines, legendlabels, loc=2, bbox_to_anchor=(1.01, 1.00), numpoints=1, handlelength=1.5, \ # handletextpad=0.5, borderpad=0.5, labelspacing=0.5, borderaxespad=0.) leg = ax.legend(lines, legendlabels, loc='upper right', bbox_to_anchor=(1.16, 1), borderaxespad=0) for label in leg.get_texts(): label.set_fontsize(9) ax.set_xlim([xmin, xmax]) ax.set_ylim([ymin, ymax]) printLabels(ax, xlabel, ylabel, title) # xtickspos = numpy.arange(nxticks)*int((xmax-xmin)/(nxticks)) + int(xmin) # ax.set_xticks(xtickspos) for tick in ax.get_xticklabels(): tick.set_visible(xtick_visible) for tick in ax.xaxis.get_major_ticks(): tick.label.set_fontsize(ticksize) for tick in ax.get_yticklabels(): tick.set_visible(ytick_visible) for tick in ax.yaxis.get_major_ticks(): tick.label.set_fontsize(ticksize) iplot = ax.lines[-1] if '0.' in matplotlib.__version__[0:2]: print "The matplotlib version has to be updated to 1.1 or newer" return iplot if '1.0.' in matplotlib.__version__[0:4]: print "The matplotlib version has to be updated to 1.1 or newer" return iplot if grid != None: ax.grid(b=True, which='major', axis=grid) matplotlib.pyplot.tight_layout() if XAxisAsTime: def func(x, pos): return ('%s') % ( datetime.datetime.utcfromtimestamp(x).strftime("%H:%M:%S")) ax.xaxis.set_major_formatter(FuncFormatter(func)) ax.xaxis.set_major_locator(LinearLocator(7)) matplotlib.pyplot.ion() return iplot def pmultilineyaxis(iplot, x, y, xlabel='', ylabel='', title=''): ax = iplot.axes printLabels(ax, xlabel, ylabel, title) for i in range(len(ax.lines)): line = ax.lines[i] line.set_data(x, y[i, :]) def createPolar(ax, x, y, xlabel='', ylabel='', title='', ticksize=9, colormap='jet', cblabel='', cbsize="5%", XAxisAsTime=False): matplotlib.pyplot.ioff() ax.plot(x, y, 'bo', markersize=5) # ax.set_rmax(90) ax.set_ylim(0, 90) ax.set_yticks(numpy.arange(0, 90, 20)) # ax.text(0, -110, ylabel, rotation='vertical', va ='center', ha = 'center' ,size='11') # ax.text(0, 50, ylabel, rotation='vertical', va ='center', ha = 'left' ,size='11') # ax.text(100, 100, 'example', ha='left', va='center', rotation='vertical') ax.yaxis.labelpad = 40 printLabels(ax, xlabel, ylabel, title) iplot = ax.lines[-1] if '0.' in matplotlib.__version__[0:2]: print "The matplotlib version has to be updated to 1.1 or newer" return iplot if '1.0.' in matplotlib.__version__[0:4]: print "The matplotlib version has to be updated to 1.1 or newer" return iplot # if grid != None: # ax.grid(b=True, which='major', axis=grid) matplotlib.pyplot.tight_layout() matplotlib.pyplot.ion() return iplot def polar(iplot, x, y, xlabel='', ylabel='', title=''): ax = iplot.axes # ax.text(0, -110, ylabel, rotation='vertical', va ='center', ha = 'center',size='11') printLabels(ax, xlabel, ylabel, title) set_linedata(ax, x, y, idline=0) def draw(fig): if type(fig) == 'int': raise ValueError, "Error drawing: Fig parameter should be a matplotlib figure object figure" fig.canvas.draw() def pause(interval=0.000001): matplotlib.pyplot.pause(interval)