The values set in the function setrun are then written out to data files
that will be read in by the Fortran code.
"""
import os, sys
import numpy as np
from clawpack.geoclaw_1d.nonuniform_grid_tools import make_mapc2p
# Read in nonuniform computational cell edges, which should have
# been created using make_celledges.py:
grid_type = 2
fname_celledges = 'celledges.txt'
mapc2p, mx_edge, xp_edge = make_mapc2p(fname_celledges)
mx = mx_edge - 1
print('Setting mx = %i, cell edges from %g to %g' %
(mx, xp_edge[0], xp_edge[-1]))
#------------------------------
def setrun(claw_pkg='geoclaw'):
#------------------------------
"""
Define the parameters used for running Clawpack.
INPUT:
claw_pkg expected to be "geoclaw" for this setrun.
def setplot(plotdata):
plotdata.clearfigures()
outdir1 = plotdata.outdir
mapc2p1, mx_edge, xp_edge = make_mapc2p(fname_celledges)
def fixticks1(current_data):
from pylab import ticklabel_format, grid, tight_layout
ticklabel_format(useOffset=False)
grid(True)
tight_layout()
def fixticks(current_data):
from pylab import ticklabel_format, plot, grid, gca
ticklabel_format(useOffset=False)
if xmax is not None:
plot(xmax, etamax, 'r')
grid(True)
def velocity(current_data):
from pylab import where, nan
q = current_data.q
hpos = where(q[0, :] < 1e-3, nan, q[0, :])
u = q[1, :] / hpos
return u
plotfigure = plotdata.new_plotfigure(name='domain', figno=0)
plotfigure.kwargs = {'figsize': (8, 7)}
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(311)'
plotaxes.xlimits = xlimits
plotaxes.ylimits = [-3, 10]
plotaxes.title = 'Surface displacement'
plotaxes.afteraxes = fixticks
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.surface
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
#plotitem.show = False
plotitem.plot_var = geoplot.topo
plotitem.color = 'g'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(312)'
plotaxes.xlimits = xlimits
plotaxes.ylimits = [-1.0, 2.0]
plotaxes.title = 'Velocity'
plotaxes.afteraxes = fixticks1
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = velocity
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(313)'
plotaxes.xlimits = xlimits
plotaxes.ylimits = [-5000, 500]
plotaxes.title = 'Full depth'
plotaxes.afteraxes = fixticks1
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_fill_between')
#plotitem.show = False
plotitem.plot_var = geoplot.surface
plotitem.plot_var2 = geoplot.topo
plotitem.color = [.5, .5, 1]
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
#plotitem.show = False
plotitem.plot_var = geoplot.topo
plotitem.color = 'g'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.surface
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
#----------
plotfigure = plotdata.new_plotfigure(name='shore', figno=1)
plotfigure.kwargs = {'figsize': (10, 4)}
#plotfigure.show = False
plotaxes = plotfigure.new_plotaxes()
plotaxes.xlimits = [-1000, 1000]
plotaxes.ylimits = [-20, 20]
plotaxes.title = 'Zoom around shore'
plotaxes.afteraxes = fixticks
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.show = False
plotitem.plot_var = geoplot.surface
plotitem = plotaxes.new_plotitem(plot_type='1d_fill_between')
plotitem.plot_var = geoplot.surface
plotitem.plot_var2 = geoplot.topo
plotitem.color = [.5, .5, 1]
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.surface
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.topo
plotitem.color = 'g'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
#-----------------------------------------
# Figures for gauges
#-----------------------------------------
plotfigure = plotdata.new_plotfigure(name='q', figno=300, \
type='each_gauge')
plotfigure.clf_each_gauge = True
plotaxes = plotfigure.new_plotaxes()
plotaxes.xlimits = 'auto'
plotaxes.ylimits = [-2, 2]
plotaxes.title = 'Surface elevation eta'
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = 2 # eta
plotitem.plotstyle = 'b-'
plotdata.printfigs = True # Whether to output figures
plotdata.print_format = 'png' # What type of output format
plotdata.print_framenos = 'all' # Which frames to output
plotdata.print_fignos = 'all' # Which figures to print
plotdata.html = True # Whether to create HTML files
plotdata.latex = False # Whether to make LaTeX output
plotdata.parallel = True
return plotdata
def setplot(plotdata):
plotdata.clearfigures()
outdir1 = plotdata.outdir
mapc2p1, mx_edge, xp_edge = make_mapc2p(
os.path.join(outdir1, 'celledges.data'))
from clawpack.amrclaw.data import GaugeData
setgauges = GaugeData()
setgauges.read(outdir1)
gauge_xc = {}
for k in range(len(setgauges.gauges)):
gauge = setgauges.gauges[k]
gaugeno = gauge[0]
gauge_xc[gaugeno] = gauge[1]
if outdir2:
mapc2p2, mx_edge, xp_edge = make_mapc2p(
os.path.join(outdir2, 'celledges.data'))
def fixticks1(current_data):
from pylab import ticklabel_format, grid, tight_layout
ticklabel_format(useOffset=False)
grid(True)
tight_layout()
#import pdb; pdb.set_trace()
def fixticks(current_data):
from pylab import ticklabel_format, plot, grid, gca
ticklabel_format(useOffset=False)
if xmax is not None:
plot(xmax, etamax, 'r')
grid(True)
def velocity(current_data):
from pylab import where
q = current_data.q
u = where(q[0, :] > 1e-3, q[1, :] / q[0, :], 0.)
return u
plotfigure = plotdata.new_plotfigure(name='domain', figno=0)
plotfigure.kwargs = {'figsize': (8, 7)}
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(311)'
plotaxes.xlimits = xlimits
plotaxes.ylimits = [-0.05, 0.2]
plotaxes.title = 'Surface displacement'
plotaxes.afteraxes = fixticks
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.surface
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
if outdir2:
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.outdir = outdir2
plotitem.plot_var = geoplot.surface
plotitem.color = 'k'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p2
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
#plotitem.show = False
plotitem.plot_var = geoplot.topo
plotitem.color = 'g'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(312)'
plotaxes.xlimits = xlimits
plotaxes.ylimits = [-0.6, 0.6]
plotaxes.title = 'Velocity'
plotaxes.afteraxes = fixticks1
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.velocity
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
if outdir2:
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.outdir = outdir2
plotitem.plot_var = geoplot.velocity
plotitem.color = 'k'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p2
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(313)'
plotaxes.xlimits = xlimits
plotaxes.ylimits = [-0.25, 0.1]
plotaxes.title = 'Full depth'
plotaxes.afteraxes = fixticks1
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_fill_between')
#plotitem.show = False
plotitem.plot_var = geoplot.surface
plotitem.plot_var2 = geoplot.topo
plotitem.color = [.5, .5, 1]
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
#plotitem.show = False
plotitem.plot_var = geoplot.topo
plotitem.color = 'g'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.surface
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
#----------
plotfigure = plotdata.new_plotfigure(name='shore', figno=1)
#plotfigure.kwargs = {'figsize':(9,11)}
plotfigure.show = False
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(211)'
plotaxes.xlimits = [0, 80e3]
plotaxes.ylimits = [-4, 4]
plotaxes.title = 'Zoom on shelf'
plotaxes.afteraxes = fixticks
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.surface
#plotitem = plotaxes.new_plotitem(plot_type='1d_fill_between')
#plotitem.plot_var = geoplot.surface
#plotitem.plot_var2 = geoplot.topo
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.topo
plotitem.color = 'k'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(212)'
#plotaxes.xlimits = [-2000,2000]
plotaxes.xlimits = [-1000, 1000]
#plotaxes.ylimits = [-10,40]
plotaxes.ylimits = [-20, 60]
plotaxes.title = 'Zoom around shore'
plotaxes.afteraxes = fixticks
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.show = False
plotitem.plot_var = geoplot.surface
plotitem = plotaxes.new_plotitem(plot_type='1d_fill_between')
plotitem.plot_var = geoplot.surface
plotitem.plot_var2 = geoplot.topo
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.topo
plotitem.color = 'k'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
#-----------------------------------------
# Figures for gauges
#-----------------------------------------
plotfigure = plotdata.new_plotfigure(name='q', figno=300, \
type='each_gauge')
plotfigure.clf_each_gauge = True
plotaxes = plotfigure.new_plotaxes()
def fixgauge(current_data):
from pylab import grid, title
grid(True)
gaugeno = current_data.gaugeno
xc = gauge_xc[gaugeno]
xp = mapc2p1(xc)
print('+++ xc,xp:', xc, xp)
title('Surface elevation at Gauge %i, x = %.0f m' \
% (gaugeno, xp))
plotaxes.afteraxes = fixgauge
plotaxes.xlimits = 'auto'
plotaxes.ylimits = [-0.02, 0.08]
plotaxes.title = 'Surface elevation eta'
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = 2 # eta
plotitem.plotstyle = 'b-'
plotdata.printfigs = True # Whether to output figures
plotdata.print_format = 'png' # What type of output format
plotdata.print_framenos = 'all' # Which frames to output
plotdata.print_fignos = 'all' # Which figures to print
plotdata.html = True # Whether to create HTML files
plotdata.latex = False # Whether to make LaTeX output
plotdata.parallel = True
return plotdata
def setplot(plotdata=None):
if plotdata is None:
from clawpack.visclaw.data import ClawPlotData
plotdata = ClawPlotData()
plotdata.clearfigures()
mapc2p, mx_edge, xp_edge = make_mapc2p(fname_celledges)
def mapc2p_km(xc):
x_m = mapc2p(xc)
x_km = x_m / 1000. # convert to km
return x_km
def fixticks1(current_data):
from pylab import ticklabel_format, grid
ticklabel_format(useOffset=False)
grid(True)
def fixticks(current_data):
from pylab import ticklabel_format, plot,grid,ones,sqrt, \
legend,title,ylabel,text
ticklabel_format(useOffset=False)
# to plot max elevation over entire computation:
#if xmax is not None:
# plot(xmax, etamax, 'r')
#grid(True)
hl = 3200.
hr = 200.
greens = (hl / hr)**(0.25)
print('greens = ', greens)
#plot(current_data.x, greens*ones(current_data.x.shape),'g--')
plot(xlimits, [greens, greens], 'g--', label='$C_g$, Greens Law')
ctrans = 2 * sqrt(hl) / (sqrt(hl) + sqrt(hr))
crefl = (sqrt(hl) - sqrt(hr)) / (sqrt(hl) + sqrt(hr))
print('ctrans = ', ctrans)
plot(xlimits, [ctrans, ctrans],
'r--',
label='$C_T$, Transmission coefficient')
legend(loc='upper left')
title('')
ylabel('meters', fontsize=14)
if current_data.frameno == 0:
text(-80, -0.4, '$\longrightarrow$', fontsize=20)
text(-80, -0.6, 'Incident')
h = current_data.q[0, :]
mx2 = int(round(len(h) / 2.))
etamax2 = (h[:mx2] - hl).max()
print('mx2 = %i, etamax2 = %g' % (mx2, etamax2))
plotfigure = plotdata.new_plotfigure(name='domain', figno=0)
plotfigure.kwargs = {'figsize': (7, 6.5)}
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'axes([.1,.4,.8,.5])' #'subplot(211)'
plotaxes.xlimits = xlimits
#plotaxes.xlimits = [-100e3,-20e3]
plotaxes.ylimits = [-1, 3]
plotaxes.title = 'Surface displacement'
plotaxes.afteraxes = fixticks
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.surface
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p_km
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.show = False
plotitem.plot_var = geoplot.topo
plotitem.color = 'k'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p_km
plotaxes = plotfigure.new_plotaxes()
plotaxes.show = False
plotaxes.axescmd = 'subplot(312)'
plotaxes.xlimits = xlimits
#plotaxes.xlimits = [-100e3,-20e3]
#plotaxes.ylimits = [-1000, 1000]
#plotaxes.title = 'Full depth'
plotaxes.title = 'momentum'
plotaxes.afteraxes = fixticks1
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p_km
plotitem = plotaxes.new_plotitem(plot_type='1d_fill_between')
plotitem.show = False
plotitem.plot_var = geoplot.surface
plotitem.plot_var2 = geoplot.topo
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p_km
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.show = False
plotitem.plot_var = geoplot.topo
plotitem.color = 'k'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p_km
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = 1
plotitem.color = 'k'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p_km
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'axes([.1,.1,.8,.2])' #'subplot(212)'
plotaxes.xlimits = xlimits
#plotaxes.xlimits = [-100e3,-20e3]
#plotaxes.ylimits = [-1000, 1000]
#plotaxes.title = 'Full depth'
#plotaxes.title = 'topography'
def fix_topo_plot(current_data):
from pylab import title, xlabel
title('')
xlabel('kilometers', fontsize=14)
plotaxes.afteraxes = fix_topo_plot
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p_km
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
#plotitem.show = False
plotitem.plot_var = geoplot.topo
plotitem.color = 'k'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p_km
#----------
plotfigure = plotdata.new_plotfigure(name='shore', figno=1)
#plotfigure.kwargs = {'figsize':(9,11)}
plotfigure.show = False
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(211)'
plotaxes.xlimits = [0, 80e3]
plotaxes.ylimits = [-4, 4]
plotaxes.title = 'Zoom on shelf'
plotaxes.afteraxes = fixticks
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.surface
#plotitem = plotaxes.new_plotitem(plot_type='1d_fill_between')
#plotitem.plot_var = geoplot.surface
#plotitem.plot_var2 = geoplot.topo
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p_km
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.topo
plotitem.color = 'k'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p_km
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(212)'
#plotaxes.xlimits = [-2000,2000]
plotaxes.xlimits = [-1000, 1000]
#plotaxes.ylimits = [-10,40]
plotaxes.ylimits = [-20, 60]
plotaxes.title = 'Zoom around shore'
plotaxes.afteraxes = fixticks
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.show = False
plotitem.plot_var = geoplot.surface
plotitem = plotaxes.new_plotitem(plot_type='1d_fill_between')
plotitem.plot_var = geoplot.surface
plotitem.plot_var2 = geoplot.topo
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p_km
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.topo
plotitem.color = 'k'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p_km
plotdata.printfigs = True # Whether to output figures
plotdata.print_format = 'png' # What type of output format
plotdata.print_framenos = 'all' # Which frames to output
plotdata.print_fignos = 'all' # Which figures to print
plotdata.html = True # Whether to create HTML files
plotdata.latex = False # Whether to make LaTeX output
plotdata.parallel = True
return plotdata
def setplot(plotdata):
plotdata.clearfigures()
def set_dry_tol(current_data):
current_data.user['dry_tolerance'] = 1e-6
plotdata.beforeframe = set_dry_tol
outdir1 = plotdata.outdir
mapc2p1, mx_edge, xp_edge = make_mapc2p(fname_celledges)
def fixticks1(current_data):
from pylab import ticklabel_format, grid,tight_layout
ticklabel_format(useOffset=False)
grid(True)
tight_layout()
#import pdb; pdb.set_trace()
def fixticks(current_data):
from pylab import ticklabel_format, plot,grid,gca
ticklabel_format(useOffset=False)
if xmax is not None:
plot(xmax, etamax, 'r')
grid(True)
plotfigure = plotdata.new_plotfigure(name='domain', figno=0)
plotfigure.kwargs = {'figsize':(8,7)}
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(311)'
plotaxes.xlimits = xlimits
plotaxes.ylimits = [-0.3,0.6]
plotaxes.title = 'Surface displacement'
plotaxes.afteraxes = fixticks
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.surface
plotitem.color = 'b'
#plotitem.plotstyle = '-+'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
#plotitem.show = False
plotitem.plot_var = geoplot.topo
plotitem.color = 'g'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(312)'
plotaxes.xlimits = xlimits
plotaxes.ylimits = [-1,1]
plotaxes.title = 'Velocity'
plotaxes.afteraxes = fixticks1
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.velocity
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(313)'
plotaxes.xlimits = xlimits
plotaxes.ylimits = [-5,2]
plotaxes.title = 'Full depth'
plotaxes.afteraxes = fixticks1
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_fill_between')
#plotitem.show = False
plotitem.plot_var = geoplot.surface
plotitem.plot_var2 = geoplot.topo
plotitem.color = [.5,.5,1]
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
#plotitem.show = False
plotitem.plot_var = geoplot.topo
plotitem.color = 'g'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.surface
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
#----------
plotfigure = plotdata.new_plotfigure(name='shore', figno=1)
#plotfigure.kwargs = {'figsize':(9,11)}
plotfigure.show = False
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(211)'
plotaxes.xlimits = [0,80e3]
plotaxes.ylimits = [-4,4]
plotaxes.title = 'Zoom on shelf'
plotaxes.afteraxes = fixticks
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.surface
#plotitem = plotaxes.new_plotitem(plot_type='1d_fill_between')
#plotitem.plot_var = geoplot.surface
#plotitem.plot_var2 = geoplot.topo
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.topo
plotitem.color = 'k'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotaxes = plotfigure.new_plotaxes()
plotaxes.axescmd = 'subplot(212)'
#plotaxes.xlimits = [-2000,2000]
plotaxes.xlimits = [-1000,1000]
#plotaxes.ylimits = [-10,40]
plotaxes.ylimits = [-20,60]
plotaxes.title = 'Zoom around shore'
plotaxes.afteraxes = fixticks
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.show = False
plotitem.plot_var = geoplot.surface
plotitem = plotaxes.new_plotitem(plot_type='1d_fill_between')
plotitem.plot_var = geoplot.surface
plotitem.plot_var2 = geoplot.topo
plotitem.color = 'b'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = geoplot.topo
plotitem.color = 'k'
plotitem.MappedGrid = True
plotitem.mapc2p = mapc2p1
#-----------------------------------------
# Figures for gauges
#-----------------------------------------
plotfigure = plotdata.new_plotfigure(name='q', figno=300, \
type='each_gauge')
plotfigure.kwargs = {'figsize':(12,3)}
plotfigure.clf_each_gauge = True
plotaxes = plotfigure.new_plotaxes()
plotaxes.xlimits = 'auto'
#plotaxes.ylimits = [-2,2]
plotaxes.title = 'Surface elevation eta'
def addgrid(current_data):
from pylab import grid, xlim
grid(True)
gno = current_data.gaugeno
if gno == 150:
xlim(0,60)
elif gno == 80:
xlim(10,60)
elif gno == 60:
xlim(20,70)
elif gno == 40:
xlim(40,90)
elif gno == 30:
xlim(40,90)
elif gno == 20:
xlim(50,100)
elif gno == 10:
xlim(60,110)
plotaxes.afteraxes = addgrid
plotitem = plotaxes.new_plotitem(plot_type='1d_plot')
plotitem.plot_var = 2 # eta
plotitem.plotstyle = 'b-'
plotdata.printfigs = True # Whether to output figures
plotdata.print_format = 'png' # What type of output format
plotdata.print_framenos = 'all' # Which frames to output
plotdata.print_fignos = 'all' # Which figures to print
plotdata.html = True # Whether to create HTML files
plotdata.latex = False # Whether to make LaTeX output
plotdata.parallel = True
return plotdata