setplot.py.html | |
Source file: setplot.py | |
Directory: /Users/rjl/Dropbox/ggsss-pages/claw-apps/advection-1d-3 | |
Converted: Sun Jul 29 2012 at 22:07:49 using clawcode2html | |
This documentation file will not reflect any later changes in the source file. |
""" Set up the plot figures, axes, and items to be done for each frame. This module is imported by the plotting routines and then the function setplot is called to set the plot parameters. """ #-------------------------- def setplot(plotdata): #-------------------------- """ Specify what is to be plotted at each frame. Input: plotdata, an instance of pyclaw.plotters.data.ClawPlotData. Output: a modified version of plotdata. """ plotdata.clearfigures() # clear any old figures,axes,items data # Figure for q[0] plotfigure = plotdata.new_plotfigure(name='q[0]', figno=1) # Set up for axes in this figure: plotaxes = plotfigure.new_plotaxes() plotaxes.title = 'Solution' plotaxes.xlimits = 'auto' plotaxes.ylimits = [-1.0, 1.5] # Set up for item on these axes: plotitem = plotaxes.new_plotitem(plot_type='1d_plot') plotitem.plot_var = 0 plotitem.plotstyle = '-o' plotitem.color = 'b' plotitem.show = True # show on plot? def qtrue(x,t): """ The true solution assuming periodic boundary conditions. Should agree with what is specified in the fortran code qinit.f. The advection velocity should agree with u specified in setrun.py. """ from numpy import mod,exp,where,pi,cos u = 1.0 xi = mod(x-u*t, 1.0) q1 = 0.5*cos(2*pi*xi) + exp(-1000.*(xi-0.3)**2) q = where((xi>0.7)&(xi<0.8), 1.0, q1) return q def plot_true(current_data): from pylab import plot,legend x = current_data.x t = current_data.t q = qtrue(x,t) plot(x,q,'r-',linewidth=2) legend(['computed','true'],'upper left') plotaxes.afteraxes = plot_true # Parameters used only when creating html and/or latex hardcopy # e.g., via pyclaw.plotters.frametools.printframes: plotdata.printfigs = True # print figures plotdata.print_format = 'png' # file format plotdata.print_framenos = 'all' # list of frames to print plotdata.print_fignos = 'all' # list of figures to print plotdata.html = True # create html files of plots? plotdata.html_homelink = '../README.html' # pointer for top of index plotdata.latex = True # create latex file of plots? plotdata.latex_figsperline = 2 # layout of plots plotdata.latex_framesperline = 1 # layout of plots plotdata.latex_makepdf = False # also run pdflatex? return plotdata