def default_levels(self, nlev):
""""""
loc = MaxNLocator(nlev + 1, symmetric=self.pm)
loc.create_dummy_axis()
loc.set_bounds(self.vmin, self.vmax)
loc.set_bounds(*loc.autoscale())
levs = loc()
return levs
def ps(dosave=True, fname='figures/domains.png', lont=None, latt=None, ht=None, dd=None):
'''
Plot Bathymetry of Puget Sound, Admiralty Inlet, and Admiralty Head
Inputs:
dosave Save figure
fname File name for figure
lont, latt transect points to plot if they are input
ht Depth along transect, if input
dd Distance in meters along transect
'''
# download bathymetry, which can be found at: http://figshare.com/preview/_preview/1165560 (27.3MB)
# Read in bathymetry
mat = scipy.io.loadmat('cascadia_gridded.mat')
# x and y limits for these plots
lonlimsPS = np.array([-124., -122.15]) #-123.21, -122.15])
latlimsPS = np.array([47.02, 48.82])
lonlimsAI = np.array([-122.85, -122.535])
latlimsAI = np.array([47.9665, 48.228])
lonlimsAH = np.array([-122.72, -122.64])
latlimsAH = np.array([48.12, 48.18])
# Functionality copied from https://github.com/clawpack/geoclaw/blob/master/src/python/geoclaw/topotools.py#L873
land_cmap = plt.get_cmap('Greens_r')
sea_cmap = plt.get_cmap('Blues_r')
cmapPS = colormaps.add_colormaps((land_cmap, sea_cmap), data_limits=[-375,2500], data_break=0.0)
cmapAI = 'Blues_r'
cmapAH = 'Blues_r'
# levels to plot
levsPS = np.concatenate((np.arange(-375, 0, 25), np.arange(0,3000,500)))
levsAI = np.arange(-200, 20, 20)
levsAH = np.arange(-120, 15, 15)
# use basemap
basemapPS = Basemap(llcrnrlon=lonlimsPS[0], llcrnrlat=latlimsPS[0],
urcrnrlon=lonlimsPS[1], urcrnrlat=latlimsPS[1],
lat_0=latlimsPS.mean(), lon_0=lonlimsPS.mean(),
projection='lcc', resolution='f',
area_thresh=0.)
xPS, yPS = basemapPS(mat['lon_topo'], mat['lat_topo'])
xlimsAI, ylimsAI = basemapPS(lonlimsAI, latlimsAI)
xlimsAH, ylimsAH = basemapPS(lonlimsAH, latlimsAH)
# Make Puget Sound plot
fig = plt.figure(figsize=(16,16))
axPS = fig.add_subplot(111)
basemapPS.drawcoastlines(ax=axPS)
mappablePS = axPS.contourf(xPS, yPS, mat['z_topo'], cmap=cmapPS, levels=levsPS, zorder=2)
locator = MaxNLocator(6) # if you want no more than 10 contours
locator.create_dummy_axis()
locator.set_bounds(lonlimsPS[0], lonlimsPS[1])
pars = locator()
locator = MaxNLocator(6) # if you want no more than 10 contours
locator.create_dummy_axis()
locator.set_bounds(latlimsPS[0], latlimsPS[1])
mers = locator()
basemapPS.drawparallels(mers, dashes=(1, 1), linewidth=0.15, labels=[1,0,0,0], ax=axPS)#, zorder=3)
basemapPS.drawmeridians(pars, dashes=(1, 1), linewidth=0.15, labels=[0,0,0,1], ax=axPS)#, zorder=3)
cbPS = fig.colorbar(mappablePS, pad=0.015, aspect=35)
cbPS.set_label('Height/depth [m]')
# Label
axPS.text(0.8, 0.025, 'Puget Sound', transform=axPS.transAxes, color='0.15')
# Inset magnified plot of Admiralty Inlet
axAI = zoomed_inset_axes(axPS, 2, loc=1)
basemapPS.drawcoastlines(ax=axAI)
basemapPS.fillcontinents('darkgreen', ax=axAI)
mappableAI = axAI.contourf(xPS, yPS, mat['z_topo'], cmap=cmapAI, levels=levsAI)
axAI.set_xlim(xlimsAI)
axAI.set_ylim(ylimsAI)
# Inlaid colorbar
caxAI = fig.add_axes([0.581, 0.665, 0.011, 0.1])
cbAI = plt.colorbar(mappableAI, cax=caxAI, orientation='vertical')
cbAI.ax.tick_params(labelsize=12)
# draw a bbox of the region of the inset axes in the parent axes and
# connecting lines between the bbox and the inset axes area
mark_inset(axPS, axAI, loc1=2, loc2=4, fc="none", ec="0.3", lw=1.5, zorder=5)
# Label
axAI.text(0.41, 0.83, 'Admiralty\n Inlet', transform=axAI.transAxes, color='0.15', fontsize=16)
# Inset magnified plot of Admiralty Head
axAH = zoomed_inset_axes(axPS, 9, loc=3)
basemapPS.drawcoastlines(ax=axAH)
basemapPS.fillcontinents('darkgreen', ax=axAH)
mappableAH = axAH.contourf(xPS, yPS, mat['z_topo'], cmap=cmapAH, levels=levsAH)
axAH.set_xlim(xlimsAH)
axAH.set_ylim(ylimsAH)
if plt.is_numlike(lont):
# add points if you have some
xt, yt = basemapPS(lont, latt)
axAH.plot(xt, yt, 'k', lw=3)
# Inlaid colorbar
caxAH = fig.add_axes([0.398, 0.116, 0.012, 0.15])
cbAH = plt.colorbar(mappableAH, cax=caxAH, orientation='vertical')
cbAH.ax.tick_params(labelsize=12)
# draw a bbox of the region of the inset axes in the parent axes and
# connecting lines between the bbox and the inset axes area
mark_inset(axPS, axAH, loc1=2, loc2=4, fc="none", ec="0.3", lw=1.5, zorder=5)
# Label
axAH.text(0.47, 0.92, 'Admiralty Head', transform=axAH.transAxes, color='0.15', fontsize=16)
# pdb.set_trace()
if plt.is_numlike(lont):
# Add axes to plot transect depths
axdepths = fig.add_axes([0.28, 0.39, 0.14, 0.075], zorder=11)
axdepths.plot((np.arange(lont.size)*dd)/1000., -ht, '0.2', lw=2, zorder=12)
axdepths.tick_params(axis='both', colors='0.1', top='off', right='off', width=2, length=4, labelsize=12, labelcolor='0.1')
axdepths.spines['bottom'].set_color('none')
axdepths.spines['top'].set_color('none')
axdepths.spines['left'].set_color('none')
axdepths.spines['right'].set_color('none')
axdepths.set_xlabel('Distance along transect [km]', fontsize=14, color='0.1')
axdepths.set_ylabel('Transect depth [m]', fontsize=14, color='0.1')
axdepths.patch.set_alpha(0.0) # make bg transparent
fig.show()
# Save figure
if dosave:
fig.savefig(fname, bbox_inches='tight')
fig.show()
data[key]['x'], data[key]['y'] = basemapPS(data[key]['lon'],
data[key]['lat'])
# Make Puget Sound plot
fig = plt.figure(figsize=(16, 16))
axPS = fig.add_subplot(111)
# basemapPS.drawcoastlines(ax=axPS)
mappablePS = axPS.contourf(xPS,
yPS,
mat['z_topo'],
cmap=cmapPS,
levels=levsPS,
zorder=2,
norm=norm)
locator = MaxNLocator(6) # if you want no more than 10 contours
locator.create_dummy_axis()
locator.set_bounds(lonlimsPS[0], lonlimsPS[1])
pars = locator()
locator = MaxNLocator(6) # if you want no more than 10 contours
locator.create_dummy_axis()
locator.set_bounds(latlimsPS[0], latlimsPS[1])
mers = locator()
basemapPS.drawparallels(mers,
dashes=(1, 1),
linewidth=0.15,
labels=[1, 0, 0, 0],
ax=axPS) #, zorder=3)
basemapPS.drawmeridians(pars,
dashes=(1, 1),
linewidth=0.15,
labels=[0, 0, 0, 1],