mpldriver.py
249 lines
| 6.9 KiB
| text/x-python
|
PythonLexer
Miguel Valdez
|
r201 | import numpy | |
Daniel Valdez
|
r190 | import matplotlib | |
|
Miguel Valdez
|
r199 | matplotlib.use("TKAgg") | |
|
Daniel Valdez
|
r190 | import matplotlib.pyplot | |
|
Miguel Valdez
|
r197 | #import scitools.numpyutils | |
|
Daniel Valdez
|
r192 | from mpl_toolkits.axes_grid1 import make_axes_locatable | |
|
Daniel Valdez
|
r190 | ||
|
Miguel Valdez
|
r201 | def init(idfigure, wintitle, width, height, facecolor="w"): | |
|
Daniel Valdez
|
r190 | matplotlib.pyplot.ioff() | |
|
Miguel Valdez
|
r201 | fig = matplotlib.pyplot.matplotlib.pyplot.figure(num=idfigure, facecolor=facecolor) | |
|
Daniel Valdez
|
r190 | fig.canvas.manager.set_window_title(wintitle) | |
|
Miguel Valdez
|
r201 | fig.canvas.manager.resize(width, height) | |
|
Daniel Valdez
|
r190 | matplotlib.pyplot.ion() | |
|
Miguel Valdez
|
r201 | ||
|
Daniel Valdez
|
r196 | return fig | |
| def setWinTitle(fig, title): | |||
|
Miguel Valdez
|
r201 | ||
|
Daniel Valdez
|
r196 | fig.canvas.manager.set_window_title(title) | |
|
Daniel Valdez
|
r190 | def setTitle(idfigure, title): | |
| fig = matplotlib.pyplot.figure(idfigure) | |||
| fig.suptitle(title) | |||
| def makeAxes(idfigure, nrow, ncol, xpos, ypos, colspan, rowspan): | |||
| fig = matplotlib.pyplot.figure(idfigure) | |||
| ax = matplotlib.pyplot.subplot2grid((nrow, ncol), (xpos, ypos), colspan=colspan, rowspan=rowspan) | |||
| return ax | |||
|
Miguel Valdez
|
r201 | def setTextFromAxes(idfigure, ax, title): | |
| fig = matplotlib.pyplot.figure(idfigure) | |||
| ax.annotate(title, xy=(.1, .99), | |||
| xycoords='figure fraction', | |||
| horizontalalignment='left', verticalalignment='top', | |||
| fontsize=10) | |||
|
Daniel Valdez
|
r190 | def pline(ax, x, y, xmin, xmax, ymin, ymax, xlabel, ylabel, title, firsttime): | |
|
Miguel Valdez
|
r201 | ||
|
Daniel Valdez
|
r190 | if firsttime: | |
| ax.plot(x, y) | |||
| ax.set_xlim([xmin,xmax]) | |||
| ax.set_ylim([ymin,ymax]) | |||
| ax.set_xlabel(xlabel, size=8) | |||
| ax.set_ylabel(ylabel, size=8) | |||
| ax.set_title(title, size=10) | |||
| matplotlib.pyplot.tight_layout() | |||
| else: | |||
| ax.lines[0].set_data(x,y) | |||
| def draw(idfigure): | |||
|
Miguel Valdez
|
r201 | ||
|
Daniel Valdez
|
r190 | fig = matplotlib.pyplot.figure(idfigure) | |
| fig.canvas.draw() | |||
|
Daniel Valdez
|
r192 | def pcolor(ax, x, y, z, xmin, xmax, ymin, ymax, zmin, zmax, xlabel, ylabel, title, firsttime, mesh): | |
|
Miguel Valdez
|
r201 | ||
|
Daniel Valdez
|
r192 | if firsttime: | |
| divider = make_axes_locatable(ax) | |||
|
Miguel Valdez
|
r201 | ax_cb = divider.new_horizontal(size="4%", pad=0.05) | |
|
Daniel Valdez
|
r192 | fig1 = ax.get_figure() | |
| fig1.add_axes(ax_cb) | |||
| ax.set_xlim([xmin,xmax]) | |||
| ax.set_ylim([ymin,ymax]) | |||
| ax.set_xlabel(xlabel) | |||
| ax.set_ylabel(ylabel) | |||
| ax.set_title(title) | |||
|
Miguel Valdez
|
r201 | print x | |
|
Daniel Valdez
|
r196 | imesh=ax.pcolormesh(x,y,z.T,vmin=zmin,vmax=zmax) | |
|
Daniel Valdez
|
r192 | matplotlib.pyplot.colorbar(imesh, cax=ax_cb) | |
| ax_cb.yaxis.tick_right() | |||
| for tl in ax_cb.get_yticklabels(): | |||
| tl.set_visible(True) | |||
| ax_cb.yaxis.tick_right() | |||
| matplotlib.pyplot.tight_layout() | |||
| return imesh | |||
| else: | |||
|
Miguel Valdez
|
r201 | # ax.set_xlim([xmin,xmax]) | |
| # ax.set_ylim([ymin,ymax]) | |||
|
Miguel Valdez
|
r199 | ax.set_xlabel(xlabel) | |
| ax.set_ylabel(ylabel) | |||
| ax.set_title(title) | |||
|
Daniel Valdez
|
r196 | z = z.T | |
|
Miguel Valdez
|
r201 | # z = z[0:-1,0:-1] | |
|
Daniel Valdez
|
r196 | mesh.set_array(z.ravel()) | |
|
Daniel Valdez
|
r192 | ||
| return mesh | |||
|
Daniel Valdez
|
r190 | ||
|
Miguel Valdez
|
r201 | ########################################### | |
| #Actualizacion de las funciones del driver | |||
| ########################################### | |||
| def createFigure(idfigure, wintitle, width, height, facecolor="w"): | |||
| matplotlib.pyplot.ioff() | |||
| fig = matplotlib.pyplot.matplotlib.pyplot.figure(num=idfigure, facecolor=facecolor) | |||
| fig.canvas.manager.set_window_title(wintitle) | |||
| fig.canvas.manager.resize(width, height) | |||
| matplotlib.pyplot.ion() | |||
| return fig | |||
|
Miguel Valdez
|
r206 | def closeFigure(): | |
| matplotlib.pyplot.ioff() | |||
| matplotlib.pyplot.show() | |||
|
Daniel Valdez
|
r207 | return | |
|
Miguel Valdez
|
r206 | ||
|
Miguel Valdez
|
r201 | 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): | |||
| matplotlib.pyplot.figure(fig.number) | |||
| axes = matplotlib.pyplot.subplot2grid((nrow, ncol), | |||
| (xpos, ypos), | |||
| colspan=colspan, | |||
| rowspan=rowspan) | |||
| 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=12) | |||
|
Miguel Valdez
|
r204 | 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): | |||
|
Miguel Valdez
|
r201 | ||
|
Miguel Valdez
|
r204 | """ | |
| Input: | |||
| grid : None, 'both', 'x', 'y' | |||
| """ | |||
|
Miguel Valdez
|
r201 | ax.plot(x, y) | |
| ax.set_xlim([xmin,xmax]) | |||
| ax.set_ylim([ymin,ymax]) | |||
| printLabels(ax, xlabel, ylabel, title) | |||
|
Miguel Valdez
|
r204 | ###################################################### | |
| 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) | |||
|
Miguel Valdez
|
r201 | ||
| for tick in ax.xaxis.get_major_ticks(): | |||
| tick.label.set_fontsize(ticksize) | |||
|
Miguel Valdez
|
r204 | ||
| ###################################################### | |||
| for tick in ax.get_yticklabels(): | |||
| tick.set_visible(ytick_visible) | |||
| for tick in ax.yaxis.get_major_ticks(): | |||
| tick.label.set_fontsize(ticksize) | |||
| ###################################################### | |||
| if grid != None: | |||
| ax.grid(b=True, which='major', axis=grid) | |||
|
Miguel Valdez
|
r201 | matplotlib.pyplot.tight_layout() | |
| iplot = ax.lines[-1] | |||
| return iplot | |||
| def pline(iplot, x, y, xlabel='', ylabel='', title=''): | |||
| ax = iplot.get_axes() | |||
| printLabels(ax, xlabel, ylabel, title) | |||
| iplot.set_data(x, y) | |||
| def createPcolor(ax, x, y, z, xmin, xmax, ymin, ymax, zmin, zmax, xlabel='', ylabel='', title='', ticksize = 9, cblabel=''): | |||
| divider = make_axes_locatable(ax) | |||
| ax_cb = divider.new_horizontal(size="4%", 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) | |||
| imesh = ax.pcolormesh(x,y,z.T,vmin=zmin,vmax=zmax) | |||
| cb = matplotlib.pyplot.colorbar(imesh, cax=ax_cb) | |||
| cb.set_label(cblabel) | |||
| ax_cb.yaxis.tick_right() | |||
| 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() | |||
| matplotlib.pyplot.tight_layout() | |||
| return imesh | |||
|
Daniel Valdez
|
r190 | ||
|
Miguel Valdez
|
r201 | def pcolor(imesh, z, xlabel='', ylabel='', title=''): | |
| z = z.T | |||
| ax = imesh.get_axes() | |||
| printLabels(ax, xlabel, ylabel, title) | |||
| imesh.set_array(z.ravel()) | |||
|
Daniel Valdez
|
r190 | ||
|
Daniel Valdez
|
r207 | def addpcolor(ax, x, y, z, zmin, zmax): | |
| imesh = ax.pcolormesh(x,y,z.T,vmin=zmin,vmax=zmax) | |||
|
Miguel Valdez
|
r201 | def draw(fig): | |
| if type(fig) == 'int': | |||
| raise ValueError, "This parameter should be of tpye matplotlib figure" | |||
| fig.canvas.draw() |