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."