def plot(self,va,lv,times,**kwargs):
for t in times:
fig = plt.figure()
data = self.get(va,lv,t)
m, x, y = self.basemap_setup()
if 'scale' in kwargs:
S = kwargs['scale']
f1 = m.contour(x,y,data,S,colors='k')
else:
f1 = m.contour(x,y,data,colors='k')
if self.C.plot_titles:
title = utils.string_from_time('title',t)
plt.title(title)
if 'wind_overlay' in kwargs:
jet = kwargs['wind_overlay']
wind = self.get('wind',lv,t)
windplot = m.contourf(x,y,wind,jet,alpha=0.6)
plt.colorbar(windplot)
elif 'W_overlay' in kwargs:
Wscale = kwargs['W_overlay']
W = self.get('W',lv,t)
windplot = m.contourf(x,y,W,alpha=0.6)
plt.colorbar(windplot)
# if self.C.colorbar:
# plt.colorbar(orientation='horizontal')
datestr = utils.string_from_time('output',t)
fname = '_'.join(('ECMWF',va,str(lv),datestr)) + '.png'
print("Plotting {0} at {1} for {2}".format(
va,lv,datestr))
plt.clabel(f1, inline=1, fmt='%4u', fontsize=12, colors='k')
utils.trycreate(self.C.output_root)
plt.savefig(os.path.join(self.C.output_root,fname))
plt.clf()
plt.close()
def plot(self, va, lv, times, **kwargs):
for t in times:
fig = plt.figure()
data = self.get(va, lv, t)
m, x, y = self.basemap_setup()
if 'scale' in kwargs:
S = kwargs['scale']
f1 = m.contour(x, y, data, S, colors='k')
else:
f1 = m.contour(x, y, data, colors='k')
if self.C.plot_titles:
title = utils.string_from_time('title', t)
plt.title(title)
if 'wind_overlay' in kwargs:
jet = kwargs['wind_overlay']
wind = self.get('wind', lv, t)
windplot = m.contourf(x, y, wind, jet, alpha=0.6)
plt.colorbar(windplot)
elif 'W_overlay' in kwargs:
Wscale = kwargs['W_overlay']
W = self.get('W', lv, t)
windplot = m.contourf(x, y, W, alpha=0.6)
plt.colorbar(windplot)
# if self.C.colorbar:
# plt.colorbar(orientation='horizontal')
datestr = utils.string_from_time('output', t)
fname = '_'.join(('ECMWF', va, str(lv), datestr)) + '.png'
print("Plotting {0} at {1} for {2}".format(va, lv, datestr))
plt.clabel(f1, inline=1, fmt='%4u', fontsize=12, colors='k')
utils.trycreate(self.C.output_root)
plt.savefig(os.path.join(self.C.output_root, fname))
plt.clf()
plt.close()
elif p == 1 and not verifplot and ex is not "ICBC":
img = M.image.imread(fpath)
ax.imshow(img)
elif (p == 2) & (ex != "ICBC"):
img = M.image.imread(fpath)
ax.imshow(img)
else:
try:
img = M.image.imread(fpath)
except:
print("Skipping ", panel)
else:
print("Plotting ", panel)
ax.imshow(img)
ax.axis("off")
plt.title(panel)
# pdb.set_trace()
if ex == "STCH":
outfiledir = os.path.join(config.output_root, case_str, IC, ens, MP)
else:
outfiledir = os.path.join(config.output_root, case_str, IC, ens)
outfilename = "_".join(("postage", ex, va_str, lv_str, time_str)) + ".png"
outfilepath = os.path.join(outfiledir, outfilename)
print(outfilepath)
utils.trycreate(outfiledir)
# plt.tight_layout()
fig.savefig(outfilepath)
fig.clf()
print("Verif skipped")
elif p == 1 and not verifplot and ex is not 'ICBC':
img = M.image.imread(fpath)
ax.imshow(img)
elif (p == 2) & (ex != 'ICBC'):
img = M.image.imread(fpath)
ax.imshow(img)
else:
try:
img = M.image.imread(fpath)
except:
print("Skipping ", panel)
else:
print("Plotting ", panel)
ax.imshow(img)
ax.axis('off')
plt.title(panel)
#pdb.set_trace()
if ex == 'STCH':
outfiledir = os.path.join(config.output_root, case_str, IC, ens, MP)
else:
outfiledir = os.path.join(config.output_root, case_str, IC, ens)
outfilename = '_'.join(('postage', ex, va_str, lv_str, time_str)) + '.png'
outfilepath = os.path.join(outfiledir, outfilename)
print(outfilepath)
utils.trycreate(outfiledir)
#plt.tight_layout()
fig.savefig(outfilepath)
fig.clf()