sensitivity = MPs
path_to_wrfouts = []
for mp in MPs:
fpath = os.path.join(config.wrfout_root, case, IC, ens, MP)
path_to_wrfouts.append(utils.wrfout_files_in(fpath, dom=1)[0])
else:
print "Typo!"
raise Exception
pfname = 'DTE_' + experiment
if compute:
p.compute_diff_energy('sum_z',
'total',
path_to_wrfouts,
times,
d_save=picklefolder,
d_return=0,
d_fname=pfname)
if plot_2D:
# Contour fixed at these values
V = range(250, 5250, 250)
VV = [
100,
] + V
ofname = pfname + '_2D'
for t in times:
p.plot_diff_energy('sum_z', 'total', t, picklefolder, pfname, outpath,
ofname, VV)
elif experiment=='MXMP':
picklefolder = os.path.join(config.wrfout_root,case,IC,ens)
outpath = os.path.join(config.output_root,case,IC,ens)
sensitivity = MPs
path_to_wrfouts = []
for mp in MPs:
fpath = os.path.join(config.wrfout_root,case,IC,ens,MP)
path_to_wrfouts.append(utils.wrfout_files_in(fpath,dom=1)[0])
else:
print "Typo!"
raise Exception
pfname = 'DTE_' + experiment
if compute:
p.compute_diff_energy('sum_z','total',path_to_wrfouts,times,
d_save=picklefolder, d_return=0,d_fname=pfname)
if plot_2D:
# Contour fixed at these values
V = range(250,5250,250)
VV = [100,] + V
ofname = pfname + '_2D'
for t in times:
p.plot_diff_energy('sum_z','total',t,picklefolder,pfname,outpath,ofname,VV)
if plot_1D:
ylimits = [0,2e8]
ofname = pfname
p.plot_error_growth(ofname,picklefolder,pfname,sensitivity=sensitivity,ylimits=ylimits)
#print "Script took", time.time()-scriptstart, "seconds."
