def plotgauge(gaugeno, plotdata, verbose=False):
#==========================================
"""
Plot all requested plots for a single gauge from the computation.
The plots are requested by setting attributes of plotdata
to ClawPlotFigure objects with plot_type="each_gauge".
"""
if verbose:
gaugesoln = plotdata.getgauge(gaugeno)
print ' Plotting gauge %s at x = %s, y = %s ... ' \
% (gaugeno, gaugesoln.location[0], gaugesoln.location[1])
if plotdata.mode() == 'iplotclaw':
pylab.ion()
try:
plotfigure_dict = plotdata.plotfigure_dict
except:
print '*** Error in plotgauge: plotdata missing plotfigure_dict'
print '*** This should not happen'
return None
if len(plotfigure_dict) == 0:
print '*** Warning in plotgauge: plotdata has empty plotfigure_dict'
print '*** Apparently no figures to plot'
# initialize current_data containing data that will be passed
# to beforegauge, aftergauge, afteraxes commands
current_data = clawdata.ClawData()
current_data.add_attribute("user", {}) # for user specified attributes
# to avoid potential conflicts
current_data.add_attribute('plotdata', plotdata)
current_data.add_attribute('gaugeno', gaugeno)
# call beforegauge if present, which might define additional
# attributes in current_data or otherwise set up plotting for this
# gauge.
beforegauge = getattr(plotdata, 'beforegauge', None)
if beforegauge:
if isinstance(beforegauge, str):
# a string to be executed
exec(beforegauge)
else:
# assume it's a function
try:
output = beforegauge(current_data)
if output: current_data = output
except:
print '*** Error in beforegauge ***'
raise
# iterate over each single plot that makes up this gauge:
# -------------------------------------------------------
if plotdata._mode == 'iplotclaw':
gaugesoln = plotdata.getgauge(gaugeno)
print ' Plotting Gauge %s at x = %s, y = %s ... ' \
% (gaugeno, gaugesoln.location[0], gaugesoln.location[1])
requested_fignos = plotdata.iplotclaw_fignos
else:
requested_fignos = plotdata.print_fignos
plotted_fignos = []
plotdata = set_show(plotdata) # set _show attributes for which figures
# and axes should be shown.
# loop over figures to appear for this gauge:
# -------------------------------------------
for figname in plotdata._fignames:
plotfigure = plotdata.plotfigure_dict[figname]
if (not plotfigure._show) or (plotfigure.type != 'each_gauge'):
continue # skip to next figure
figno = plotfigure.figno
if requested_fignos != 'all':
if figno not in requested_fignos:
continue # skip to next figure
plotted_fignos.append(figno)
if not plotfigure.kwargs.has_key('facecolor'):
# use Clawpack's default bg color (tan)
plotfigure.kwargs['facecolor'] = '#ffeebb'
# create figure and set handle:
plotfigure._handle = pylab.figure(num=figno, **plotfigure.kwargs)
pylab.ioff()
if plotfigure.clf_each_gauge:
pylab.clf()
try:
plotaxes_dict = plotfigure.plotaxes_dict
except:
print '*** Error in plotgauge: plotdata missing plotaxes_dict'
print '*** This should not happen'
return None
if (len(plotaxes_dict) == 0) or (len(plotfigure._axesnames) == 0):
print '*** Warning in plotgauge: plotdata has empty plotaxes_dict'
print '*** Apparently no axes to plot in figno ', figno
# loop over axes to appear on this figure:
# ----------------------------------------
for axesname in plotfigure._axesnames:
plotaxes = plotaxes_dict[axesname]
if not plotaxes._show:
continue # skip this axes if no items show
# create the axes:
axescmd = getattr(plotaxes, 'axescmd', 'subplot(1,1,1)')
axescmd = 'plotaxes._handle = pylab.%s' % axescmd
exec(axescmd)
pylab.hold(True)
# loop over items:
# ----------------
for itemname in plotaxes._itemnames:
plotitem = plotaxes.plotitem_dict[itemname]
outdir = plotitem.outdir
if outdir is None:
outdir = plotdata.outdir
gaugesoln = plotdata.getgauge(gaugeno, outdir)
current_data.add_attribute('gaugesoln', gaugesoln)
current_data.add_attribute('q', gaugesoln.q)
current_data.add_attribute('t', gaugesoln.t)
if plotitem._show:
try:
output = plotgauge1(gaugesoln, plotitem, current_data)
if output: current_data = output
if verbose:
print ' Plotted plotitem ', itemname
except:
print '*** Error in plotgauge: problem calling plotgauge1'
traceback.print_exc()
return None
# end of loop over plotitems
for itemname in plotaxes._itemnames:
plotitem = plotaxes.plotitem_dict[itemname]
pylab.title("%s at gauge %s" % (plotaxes.title, gaugeno))
# call an afteraxes function if present:
afteraxes = getattr(plotaxes, 'afteraxes', None)
if afteraxes:
if isinstance(afteraxes, str):
# a string to be executed
exec(afteraxes)
else:
# assume it's a function
try:
current_data.add_attribute("plotaxes", plotaxes)
current_data.add_attribute("plotfigure",
plotaxes._plotfigure)
output = afteraxes(current_data)
if output: current_data = output
except:
print '*** Error in afteraxes ***'
raise
if plotaxes.scaled:
pylab.axis('scaled')
# set axes limits:
if (plotaxes.xlimits is not None) & (type(plotaxes.xlimits)
is not str):
try:
pylab.xlim(plotaxes.xlimits[0], plotaxes.xlimits[1])
except:
pass # let axis be set automatically
if (plotaxes.ylimits is not None) & (type(plotaxes.ylimits)
is not str):
try:
pylab.ylim(plotaxes.ylimits[0], plotaxes.ylimits[1])
except:
pass # let axis be set automatically
# end of loop over plotaxes
# end of loop over plotfigures
# call an aftergauge function if present:
aftergauge = getattr(plotdata, 'aftergauge', None)
if aftergauge:
if isinstance(aftergauge, str):
# a string to be executed
exec(aftergauge)
else:
# assume it's a function
try:
output = aftergauge(current_data)
if output: current_data = output
except:
print '*** Error in aftergauge ***'
raise
if plotdata.mode() == 'iplotclaw':
pylab.ion()
for figno in plotted_fignos:
pylab.figure(figno)
pylab.draw()
if verbose:
print ' Done with plotgauge for gauge %i' % (gaugeno)
# print the figure(s) to file(s) if requested:
if (plotdata.mode() != 'iplotclaw') & plotdata.printfigs:
# iterate over all figures that are to be printed:
for figno in plotted_fignos:
printfig(gaugeno=gaugeno, figno=figno, \
format=plotdata.print_format, plotdir=plotdata.plotdir,\
verbose=verbose)
return current_data
def setplot(plotdata):
#--------------------------
"""
Specify what is to be plotted at each frame.
Input: plotdata, an instance of clawpack.visclaw.data.ClawPlotData.
Output: a modified version of plotdata.
"""
from clawpack.visclaw import colormaps, geoplot
from numpy import linspace
plotdata.clearfigures() # clear any old figures,axes,items data
plotdata.format = 'binary'
#plotdata.format = 'ascii'
# Load data from output
clawdata = clawutil.ClawInputData(2)
clawdata.read(os.path.join(plotdata.outdir, 'claw.data'))
ocean_xlimits = [clawdata.lower[0], clawdata.upper[0]]
ocean_ylimits = [clawdata.lower[1], clawdata.upper[1]]
amrdata = amrclaw.AmrclawInputData(clawdata)
amrdata.read(os.path.join(plotdata.outdir, 'amr.data'))
physics = geodata.GeoClawData()
physics.read(os.path.join(plotdata.outdir, 'geoclaw.data'))
surge_data = geodata.SurgeData()
surge_data.read(os.path.join(plotdata.outdir, 'surge.data'))
probdata = clawutil.ClawData()
#probdata.read('setprob.data', force=True)
#theta_island = probdata.theta_island
# To plot gauge locations on pcolor or contour plot, use this as
# an afteraxis function:
def addgauges(current_data):
from clawpack.visclaw import gaugetools
gaugetools.plot_gauge_locations(current_data.plotdata, \
gaugenos='all', format_string='ko', add_labels=True)
def mynewafteraxes(current_data):
addgauges(current_data)
bigfont(current_data)
#-----------------------------------------
# Some global KML settings
#-----------------------------------------
#plotdata.kml_publish = 'http://math.boisestate.edu/~calhoun/visclaw/GoogleEarth/kmz/' #public html
plotdata.kml_publish = None
plotdata.kml_name = "NYC Asteroid" # name appears in Google Earth display only
plotdata.kml_index_fname = "NYC_Asteroid" # name for .kmz and .kml files ["_GoogleEarth"]
# specify beginning of slider time. if not used, assumes Jan. 1 1970
plotdata.kml_starttime = [2115, 4, 29, 7, 32,
0] # Asteroid hits at 1:32AM, 4/29/2115 (UTC)
plotdata.kml_tz_offset = 6 # off set to UTC
# for todays date as the default use
kml_cmin = -0.005 #colorbar min and max
kml_cmax = 0.005
kml_dpi = 400 # only used if individual figures dpi not set
kml_cmap = geoplot.googleearth_lightblue
# kml_cmap = geoplot.googleearth_darkblue
# kml_cmap = geoplot.googleearth_transparent
# kml_cmap = geoplot.googleearth_white
def kml_colorbar(filename):
#cmin = -0.01
#cmax = 0.01
geoplot.kml_build_colorbar(filename, kml_cmap, kml_cmin, kml_cmax)
#-----------------------------------------
# Figure for pcolor plot for surface
#-----------------------------------------
plotfigure = plotdata.new_plotfigure(name='pcolor', figno=0)
# Set up for axes in this figure:
plotaxes = plotfigure.new_plotaxes('pcolor')
plotaxes.title = 'Surface'
plotaxes.scaled = True
#plotaxes.afteraxes = addgauges
# Water
plotitem = plotaxes.new_plotitem(plot_type='2d_pcolor')
plotitem.show = True
#plotitem.plot_var = geoplot.surface
plotitem.plot_var = geoplot.surface_or_depth
#plotitem.plot_var = 0/1/2 or plot that entry into q instead of a function
plotitem.pcolor_cmap = geoplot.tsunami_colormap
plotitem.pcolor_cmin = -.8
plotitem.pcolor_cmax = .8
plotitem.add_colorbar = True
plotitem.amr_celledges_show = [0, 0, 0]
plotitem.amr_patchedges_show = [0, 0, 0]
# Land
plotitem = plotaxes.new_plotitem(plot_type='2d_pcolor')
plotitem.plot_var = geoplot.land
plotitem.show = True
# plotitem.pcolor_cmap = colormaps.all_white # to print better in B&W
plotitem.pcolor_cmap = geoplot.land_colors
#plotitem.pcolor_cmap = geoplot.googleearth_transparent
plotitem.pcolor_cmin = 0.0
plotitem.pcolor_cmax = 100.0
plotitem.add_colorbar = False
plotitem.amr_celledges_show = [0, 0, 0, 0, 0]
#plotitem.patchedges_show = 1
plotitem.amr_patchedges_show = [0, 0, 0]
#plotaxes.xlimits = [-85,-55]
#plotaxes.ylimits = [25,45]
plotaxes.xlimits = ocean_xlimits
plotaxes.ylimits = ocean_ylimits
#plotaxes.xlimits = [-75,-70]
#plotaxes.ylimits = [38,43]
# add contour lines of bathy if desired:
plotitem = plotaxes.new_plotitem(plot_type='2d_contour')
plotitem.show = True
plotitem.plot_var = geoplot.topo
#plotitem.contour_levels = [-500,-250,-100, -50, 0, 50]
plotitem.contour_levels = linspace(-1000, -400, 3)
plotitem.amr_contour_colors = ['g'] # color on each level
#plotitem.kwargs = {'linestyles':'dashed','linewidths':2,'colors' : 'red' }
plotitem.kwargs = {
'linestyles': 'dashed',
'linewidths': 1,
'colors': 'magenta'
}
plotitem.amr_contour_show = [1, 1, 1]
plotitem.celledges_show = 0
plotitem.patchedges_show = 0
#-----------------------------------------------------------
# Figure for KML files - Sea Surface Height
#----------------------------------------------------------
#plotfigure = plotdata.new_plotfigure(name='Sea Surface',figno=1)
#plotfigure.show = True
#plotfigure.use_for_kml = True
#plotfigure.kml_use_for_initial_view = True
# These overide any values set in the plotitems below
#plotfigure.kml_xlimits = [-80,-55]
#plotfigure.kml_ylimits = [25, 45]
# Resolution
#plotfigure.kml_dpi = 300
#plotfigure.kml_tile_images = False
def kml_colorbar_transparent(filename):
#cmin = -0.01
#cmax = 0.01
geoplot.kml_build_colorbar(filename, geoplot.googleearth_transparent,
kml_cmin, kml_cmax)
#plotfigure.kml_colorbar = kml_colorbar_transparent
# Set up for axes in this figure:
#plotaxes = plotfigure.new_plotaxes('kml')
#plotaxes.scaled = True
# Water
#plotitem = plotaxes.new_plotitem(plot_type='2d_pcolor')
#plotitem.show = True
#plotitem.plot_var = geoplot.surface_or_depth
#plotitem.pcolor_cmap = geoplot.googleearth_transparent
#plotitem.pcolor_cmin = kml_cmin
#plotitem.pcolor_cmax = kml_cmax
#plotitem.amr_celledges_show = [0,0,0]
#plotitem.patchedges_show = 0
#plotfigure.kml_colorbar = kml_colorbar
#-----------------------------------------
# Figure for speeds
#-----------------------------------------
plotfigure = plotdata.new_plotfigure(name='Speeds', figno=10)
# Set up for axes in this figure:
plotaxes = plotfigure.new_plotaxes('pcolor')
plotaxes.title = 'Speed'
plotaxes.scaled = True
#def fixup(current_data):
# import pylab
# addgauges(current_data)
# t = current_data.t
# t = t / 3600. # hours
# pylab.title('Speed at %4.2f hours' % t, fontsize=20)
# pylab.xticks(fontsize=15)
# pylab.yticks(fontsize=15)
#plotaxes.afteraxes = fixup
def speed(current_data):
from pylab import where, sqrt
q = current_data.q
h = q[0, :]
dry_tol = 0.001
u = q[1, :] # this is just to initialize
v = q[2, :] # to correct size
s = 0 * q[2, :] # to correct size
nq = len(q[1, :])
[n, m] = h.shape
for ii in range(0, n):
for jj in range(0, m):
if h[ii, jj] > dry_tol:
u[ii, jj] = q[1, ii, jj] / h[ii, jj]
v[ii, jj] = q[2, ii, jj] / h[ii, jj]
s[ii,
jj] = sqrt(u[ii, jj] * u[ii, jj] + v[ii, jj] * v[ii, jj])
else:
u[ii, jj] = 0.
v[ii, jj] = 0.
s[ii, jj] = 0
#print("max of u = " + str(max(u)))
#u = where(h>dry_tol, q[1,:]/h, 0.)
#v = where(h>dry_tol, q[2,:]/h, 0.)
#s = sqrt(u**2 + v**2)
#s = sqrt(u*2+v*v) #try not dividing or using where
return s
# Water
plotitem = plotaxes.new_plotitem(plot_type='2d_pcolor')
plotitem.show = False
plotitem.plot_var = speed
plotitem.pcolor_cmap = geoplot.tsunami_colormap
#plotitem.pcolor_cmap = \
# colormaps.make_colormap({0:[1,1,1],0.5:[0.5,0.5,1],1:[1,0.3,0.3]})
plotitem.pcolor_cmin = 0.
plotitem.pcolor_cmax = .10
plotitem.add_colorbar = True
plotitem.amr_celledges_show = [0, 0, 0]
#plotitem.patchedges_show = 1
plotitem.amr_patchedges_show = [0, 0, 0]
# Land
plotitem = plotaxes.new_plotitem(plot_type='2d_pcolor')
plotitem.plot_var = geoplot.land
plotitem.pcolor_cmap = geoplot.land_colors
plotitem.pcolor_cmin = 0.0
plotitem.pcolor_cmax = 100.0
plotitem.add_colorbar = False
plotitem.amr_celledges_show = [0, 0, 0]
plotitem.patchedges_show = 0
plotaxes.xlimits = ocean_xlimits
plotaxes.ylimits = ocean_ylimits
#-----------------------------------------
# Figure for zoom around NYC
#-----------------------------------------
zoomWanted = True
if zoomWanted:
plotfigure = plotdata.new_plotfigure(name='Zoom1', figno=7)
# add another figure
#plotfigure.use_for_kml = True
#plotfigure.kml_use_for_initial_view = False
# These overide any values set in the plotitems below
#plotfigure.kml_xlimits = [-74.5, -73.5]
#plotfigure.kml_ylimits = [40.4,40.9]
# Resolution
#plotfigure.kml_dpi = 300
#plotfigure.kml_tile_images = True
#plotfigure.kml_colorbar = kml_colorbar # defined above
# Set up for axes in this figure:
plotaxes = plotfigure.new_plotaxes('zoom on nyc')
plotaxes.title = 'Surface elevation'
plotaxes.scaled = True
manhattan_island = -73.5
xisland, yisland = latlong(1600e3, manhattan_island, 40., Rearth)
#plotaxes.xlimits = [xisland-0.6, xisland+0.6]
#plotaxes.xlimits = [manhattan_island-1, manhattan_island+1]
#plotaxes.ylimits = [40.15,41.5]
plotaxes.xlimits = [-74.5, -72.5] # really zoom in on lower manhattan]
plotaxes.ylimits = [40.25, 41.5]
plotaxes.afteraxes = addgauges
def bigfont(current_data):
import pylab
t = current_data.t
pylab.title("Surface at t = %8.1f" % t, fontsize=20)
pylab.xticks(fontsize=15)
pylab.yticks(fontsize=15)
#plotaxes.afteraxes = bigfont
#plotaxes.afteraxes = mynewafteraxes # after axes functions mess with GE plots
# Water
plotitem = plotaxes.new_plotitem(plot_type='2d_pcolor')
plotitem.show = False
#plotitem.plot_var = geoplot.surface
plotitem.plot_var = geoplot.surface_or_depth
#plotitem.pcolor_cmap = geoplot.tsunami_colormap
plotitem.pcolor_cmap = kml_cmap
#plotitem.pcolor_cmin = kml_cmin # same as above
#plotitem.pcolor_cmax = kml_cmax
plotitem.pcolor_cmin = -.2
plotitem.pcolor_cmax = .2
plotitem.add_colorbar = True
plotitem.amr_celledges_show = [0, 0, 0]
plotitem.patchedges_show = 0
# Land
plotitem = plotaxes.new_plotitem(plot_type='2d_pcolor')
plotitem.show = False
plotitem.plot_var = geoplot.land
# plotitem.pcolor_cmap = colormaps.all_white # to print better in B&W
plotitem.pcolor_cmap = geoplot.land_colors
plotitem.pcolor_cmin = 0.0
plotitem.pcolor_cmax = 100.0
plotitem.add_colorbar = False
plotitem.amr_celledges_show = [0, 0, 0, 0, 0]
plotitem.patchedges_show = 0
# contour lines:
plotitem = plotaxes.new_plotitem(plot_type='2d_contour')
plotitem.show = False
plotitem.plot_var = geoplot.surface
plotitem.contour_levels = [-0.8, -0.4, 0.4, 0.8]
plotitem.amr_contour_colors = ['k'] # color on each level
plotitem.kwargs = {'linewidths': 1}
plotitem.amr_contour_show = [0, 0, 0, 1, 1]
plotitem.celledges_show = 0
plotitem.patchedges_show = 0
# add contour lines of bathy if desired:
plotitem = plotaxes.new_plotitem(plot_type='2d_contour')
plotitem.show = False
plotitem.plot_var = geoplot.topo
plotitem.contour_levels = linspace(-30, -10, 3)
plotitem.amr_contour_colors = ['m'] # color on each level
plotitem.kwargs = {'linestyles': 'dashed', 'linewidths': 1}
plotitem.amr_contour_show = [1, 1, 1]
plotitem.celledges_show = 0
plotitem.patchedges_show = 0
#-----------------------------------------
# Figures for gauges
#-----------------------------------------
plotfigure = plotdata.new_plotfigure(name='Surface & topo', figno=300, \
type='each_gauge')
plotfigure.clf_each_gauge = True
# Set up for axes in this figure:
plotaxes = plotfigure.new_plotaxes()
#plotaxes.xlimits = [0000, 6500]
#plotaxes.ylimits = [-.10, .15]
plotaxes.xlimits = 'auto'
plotaxes.ylimits = 'auto'
plotaxes.title = 'Surface'
# Plot surface as blue curve:
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = 3
plotitem.plotstyle = 'b-'
# Plot topo as green curve:
#plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
def gaugetopo(current_data):
q = current_data.q
h = q[0, :]
eta = q[3, :]
topo = eta - h
return topo
def gaugedpress(current_data):
q = current_data.q
#dpress = (q[4,:] - 101300)/101300
dpress = q[4, :] # already output as relative: dp/amb_pr
return dpress
def gs(current_data):
q = current_data.q
# different than speed function because q is function of time, not
# x,y at the gauges.
from numpy import where, sqrt
h = q[0, :]
#print('shape of h ' + str(h.shape))
dry_tol = 0.001
u = where(h > dry_tol, q[1, :] / h, 0.)
v = where(h > dry_tol, q[2, :] / h, 0.)
ssq = sqrt(u * u + v * v)
#s = sqrt(u**2 + v**2)
s = sqrt(ssq)
return ssq
#plotitem.plot_var = gaugetopo
#plotitem.plotstyle = 'g-'
# Plot relative delta pressure as red curve:
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = gaugedpress
plotitem.plotstyle = 'r-'
# add speed to this plot since cant get new one going
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = gs
plotitem.plotstyle = 'g-'
def add_zeroline(current_data):
from pylab import plot, legend
t = current_data.t
# legend(('surface','topography','dp'),loc='lower left')
legend(('surface', 'dp', 'speed'), loc='upper right')
#plot(t, 0*t, 'k')
plotaxes.afteraxes = add_zeroline
# -------------------------------
# Figure for speed at gauges
#--------------------------------
# plotfigure = plotdata.new_plotfigure(name='Speed at gauges', figno=310, \
# type='each_gauge')
#
#
# # Set up for axes in this figure:
# plotaxes = plotfigure.new_plotaxes()
# plotaxes.xlimits = 'auto'
# plotaxes.ylimits = 'auto'
# #plotaxes.xlimits = [0000, 7000]
# #plotaxes.ylimits = 'auto'
# plotaxes.title = 'Speed'
#
# # Plot surface as blue curve:
# plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
# #plotitem.plot_var = speed
# plotitem.plot_var = gs #gauge_speed
# plotitem.plotstyle = 'b-'
#
# plotfigure.show = True
#
#-----------------------------------------
# Figure for bathy alone
#-----------------------------------------
def bathy(current_data):
return current_data.aux[0, :, :]
plotfigure = plotdata.new_plotfigure(name='bathymetry', figno=3)
plotaxes = plotfigure.new_plotaxes('pcolor')
plotaxes.title = 'Bathymetry'
plotaxes.scaled = True
plotitem = plotaxes.new_plotitem(plot_type='2d_pcolor')
plotitem.plot_var = bathy
plotitem.pcolor_cmin = -3000.00
plotitem.pcolor_cmax = 500
plotitem.add_colorbar = True
plotitem = plotaxes.new_plotitem(plot_type='2d_contour')
plotitem.show = False
plotitem.plot_var = geoplot.topo
plotitem.contour_levels = linspace(-2000, 2000, 21)
plotitem.amr_contour_colors = ['k'] # color on each level
#plotitem.contour_levels = [-1000 -500,-250,-100, -5, 5]
#-----------------------------------------
# Figure for grids alone
#-----------------------------------------
#plotfigure = plotdata.new_plotfigure(name='grids', figno=2)
#plotfigure.show = False
# Set up for axes in this figure:
#plotaxes = plotfigure.new_plotaxes()
#plotaxes.xlimits = [0,1]
#plotaxes.ylimits = [0,1]
#plotaxes.title = 'grids'
#plotaxes.scaled = True
# Set up for item on these axes:
#plotitem = plotaxes.new_plotitem(plot_type='2d_patch')
#plotitem.amr_patch_bgcolor = ['#ffeeee', '#eeeeff', '#eeffee']
#plotitem.amr_celledges_show = [0,0,0]
#plotitem.amr_patchedges_show = [0,0,0]
def normalized_pressure(current_data):
pressure_index = 6 #7 in fortran, but python is 0 based
return current_data.aux[
pressure_index, :, :] / surge_data.ambient_pressure
# Pressure field
plotfigure = plotdata.new_plotfigure(name='Pressure', figno=33)
plotfigure.show = True
plotaxes = plotfigure.new_plotaxes('normalized_pressure')
plotaxes.xlimits = ocean_xlimits
plotaxes.ylimits = ocean_ylimits
plotaxes.title = 'Pressure Field'
plotaxes.scaled = True
plotaxes.afteraxes = addgauges
pressure_limits = [.99, 1.10]
#pressure_limits = [.999*surge_data.ambient_pressure / 100.0,
# 1.001 * surge_data.ambient_pressure / 100.0]
#pressure_limits = [-.000001*surge_data.ambient_pressure,
# .000001 * surge_data.ambient_pressure]
surgeplot.add_pressure(plotaxes, bounds=pressure_limits)
surgeplot.add_land(plotaxes)
#plotitem = plotaxes.new_plotitem(plot_type='2d_patch')
plotitem = plotaxes.plotitem_dict['pressure']
plotitem.show = False
plotitem.plot_var = normalized_pressure
plotitem.amr_celledges_show = [0, 0, 0]
plotitem.amr_patchedges_show = [0, 0, 0]
plotaxes.plotitem_dict['land'].amr_celledges_show = [0, 0, 0]
plotaxes.plotitem_dict['land'].amr_patchedges_show = [0, 0, 0]
#-----------------------------------------
# Parameters used only when creating html and/or latex hardcopy
# e.g., via clawpack.visclaw.frametools.printframes:
plotdata.printfigs = True # print figures
plotdata.print_format = 'png' # file format
plotdata.print_framenos = 'all' # list of frames to print
#plotdata.print_framenos = [30,50,70] # list of frames to print
#plotdata.print_framenos = [1] # list of frames to print
plotdata.print_gaugenos = 'all' # list of gauges to print
#plotdata.print_fignos = [0,7,10,33,300] # list of figures to print
#plotdata.print_fignos = [0,7,10,300] # list of figures to print
plotdata.print_fignos = [0, 300] # 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?
plotdata.kml = False
return plotdata