print 'Plotting part (a): spatiofreq'
plt.subplot(yplots, xplots, 1)
if res == 'native':
lon4sf = lon
lat4sf = lat
elif res == 'make':
lt1 = lat[0]
lt2 = lat[-1]
ln1 = lon[0]
ln2 = lon[-1]
lat4sf = np.arange(lt2, lt1 + extent, gsize)
lat4sf = lat4sf[::
-1] # latitude has to be made the other way because of the negative numbers
lon4sf = np.arange(ln1, ln2 + extent, gsize)
m = blb.AfrBasemap2(lat4sf, lon4sf, drawstuff=True, prj='cyl', rsltn='l')
allmask, img = stats.spatiofreq5(m, s, name, lat4sf, lon4sf, yrs, thesekeys, per=rate, clim=nos4cbar, \
savefig=False, flagonly=True, \
col='bw', frm_event=from_event,cbar='none',title='')
m.drawcountries(color='k')
m.drawcoastlines(color='k')
if testingoutput:
plt.savefig('tmpfig_a.png', dpi=150)
print 'Plotting part (b): CB outlines'
plt.subplot(yplots, xplots, 2)
m = blb.AfrBasemap2(lat, lon, drawstuff=True, prj='cyl', rsltn='l')
if plotshow == 'col5':
nch = 5
cols = ['r', 'b', 'c', 'm', 'g']
if choose_cb == 'random':
# Count ndays in composite with same sign as mean
mask_zeros=np.zeros((nlat,nlon),dtype=np.float32)
for i in range(nlat):
for j in range(nlon):
count=len(np.where(anoms_signs[:,i,j]==comp_signs[i,j])[0])
perc=(float(count)/float(nsamp))*100
if perc>=perc_ag:
mask_zeros[i,j]=1
else:
mask_zeros[i,j]=0
# Plot
print "Plotting for model "+name2
plt.subplot(yplots,xplots,cnt)
if wplotdraw=='all':
m = blb.AfrBasemap2(lat, lon, latsp,lonsp, drawstuff=True, prj='cyl', rsltn='l',\
fontdict={'fontsize':8,'fontweight':'normal'})
elif wplotdraw=='first':
if cnt==1:
m = blb.AfrBasemap2(lat, lon, latsp, lonsp, drawstuff=True, prj='cyl', rsltn='l', \
fontdict={'fontsize': 8, 'fontweight': 'demibold'})
else:
m = blb.AfrBasemap2(lat, lon, latsp, lonsp, drawstuff=False, prj='cyl', rsltn='l', \
fontdict={'fontsize': 8, 'fontweight': 'demibold'})
elif wplotdraw=='edges':
x_remain= cnt % xplots
if x_remain==1:
m = blb.AfrBasemap2(lat, lon, latsp, lonsp, drawstuff=True, prj='cyl', rsltn='l', \
fontdict={'fontsize': 8, 'fontweight': 'normal'})
else:
m = blb.AfrBasemap2(lat, lon, latsp, lonsp, drawstuff=True, prj='cyl', rsltn='l', \
fontdict={'fontsize': 8, 'fontweight': 'normal'},onlyedge='lon')
lat = np.arange(-70.0, 20.0, 10.0)
lon = np.arange(-2.5, 120.0, 10.0)
latsp = 20. # lat spacing
lonsp = 25. # lon spacing
### Get directories
bkdir = cwd + "/../../../../CTdata/"
botdir = bkdir + "metbot_multi_dset/"
figdir = botdir + "/histpaper_v2_figs/outlines_map/"
my.mkdir_p(figdir)
# Set up plot
print "Setting up plot..."
g, ax = plt.subplots(figsize=figdim)
m = blb.AfrBasemap2(lat, lon, latsp, lonsp, drawstuff=True, prj='cyl', rsltn='l', \
fontdict={'fontsize': 12, 'fontweight': 'normal'})
# Redraw map
m.drawcountries()
m.drawcoastlines()
# Plot MetBot full domain
fulldom_lats = [0, 0, -60, -60]
fulldom_lons = [7.5, 100.0, 100.0, 7.5]
draw_screen_poly(fulldom_lats,
fulldom_lons,
m,
ls='--',
lw=3,
edgecol='dodgerblue',
zord=10)
