dtetimes,
d_save=pickledir,
d_return=0,
d_fname=pfname)
if plot_1D_dte:
# Contour fixed at these values
V = list(range(250, 5250, 250))
VV = [
100,
] + V
ofname = pfname
p.plot_diff_energy('1D',
'DTE',
pickledir,
outdir,
dataf=pfname,
outprefix=ofname,
clvs=VV,
utc=False)
if plot_3D_dte:
SENS = {
'ICBC': ensnames,
'MXMP': experiments,
'STCH': 0,
'STCH5': 0,
'STMX': experiments
}
ylimits = [0, 2e8]
ofname = pfname
p.plot_error_growth(outdir,
for ex in experiments:
od,fpath = get_folders(en,ex)
print fpath
path_to_wrfouts.append(utils.netcdf_files_in(fpath))
if compute_dte:
p.compute_diff_energy('1D','DTE',path_to_wrfouts,dtetimes,
d_save=pickledir, d_return=0,d_fname=pfname)
if plot_1D_dte:
# Contour fixed at these values
# V = range(250,5250,250)
V = range(500,18000,500)
VV = [250,] + V
ofname = pfname
p.plot_diff_energy('1D','DTE',pickledir,outdir,dataf=pfname,outprefix=ofname,clvs=VV,utc=False,cb=True)
if plot_3D_dte:
SENS = {'ICBC':ensnames,'MXMP':experiments,'STCH':0,'STCH5':0,'STMX':experiments}
ylimits = [0,2e8]
ofname = pfname
p.plot_error_growth(outdir,pickledir,dataf=pfname,sensitivity=SENS[enstype],ylim=ylimits,f_prefix=enstype)
# if powerspectrum:
# listofncfiles =
# p.plot_diff_power_spectrum('DTE',pickledir,outdir,dataf=pfname,outprefix=ofname,utc=powertime,meanenergy=listofncfiles)
if all_3D_dte:
if case[:4] == '2006':
# EXS = {'GEFS-ICBC':{},'NAM-MXMP':{},'NAM-STMX':{},'NAM-STCH5':{},'GEFS-STCH':{},'NAM-STCH':{},'GEFS-MXMP':{}}
# EXS = {'ICBC (GEFS/R2)':{},'MXMP (NAM)':{},'STMX (NAM)':{},'STCH5 (NAM; WDM6 Grau)':{},'STCH (GEFS/R2 c00)':{},'STCH (NAM; WDM6 Grau)':{},'MXMP (GEFS/R2 c00)':{}}
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 = list(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."
outpath = os.path.join(output_root,case,IC,ens)
sensitivity = MPs
path_to_wrfouts = []
for mp in MPs:
fpath = os.path.join(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 = list(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."
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
p.plot_diff_energy('sum_z', 'total', times, picklefolder, pfname, 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."
sensitivity=ens_names
#fpath = os.path.join(config.wrfout_root,case,IC,)
#path_to_wrfouts = p.wrfout_files_in(fpath,dom=1)
elif experiment=='MXMP':
picklefolder = os.path.join(config.wrfout_root,case,IC,ens)
p.C.output_root = os.path.join(config.output_root,case,IC,ens)
sensitivity = MPs
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
p.plot_diff_energy('sum_z','total',times,picklefolder,pfname,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."
