ensnames = ['c00'] + ['p{0:02d}'.format(n) for n in range(1, 11)] #for rundate in ('25','27','29'): plot_all = 1 if plot_all: # for rundate in ['25','27','29']: for rundate in ['27', '29']: # print("Computing for {0} November".format(rundate)) foldername = '201111' + rundate + '00' runfolder = os.path.join(rootdir, foldername) path_to_wrfouts = utils.wrfout_files_in(runfolder, dom=1) itime = (2011, 11, int(rundate), 0, 0, 0) ftime = (2011, 12, 2, 12, 0, 0) times = p.generate_times(itime, ftime, 6 * 3600) path_to_plots = os.path.join(outdir, foldername) # for time in times: #pdb.set_trace() # Produce .npy data files with DKE data # print("Compute_diff_energy...") wrfouts = [ '/uufs/chpc.utah.edu/common/home/horel-group2/lawson2/201111{0}00/{1}/wrfout_d01_2011-11-{2}_00:00:00_PLEV' .format(rundate, e, rundate) for e in ensnames ] clvs = 0 plotfname = 'deltaDKE' # p.compute_diff_energy('sum_z','kinetic',path_to_wrfouts,times,upper=500, # d_save=runfolder, d_return=0,d_fname='DKE_500_sixhrly_'+foldername) p.delta_diff_energy(
ensnames = ['c00'] + ['p{0:02d}'.format(n) for n in range(1,11)] #for rundate in ('25','27','29'): plot_all = 1 if plot_all: # for rundate in ['25','27','29']: for rundate in ['27','29']: # print("Computing for {0} November".format(rundate)) foldername = '201111' + rundate + '00' runfolder = os.path.join(rootdir,foldername) path_to_wrfouts = utils.wrfout_files_in(runfolder,dom=1) itime = (2011,11,int(rundate),0,0,0) ftime = (2011,12,2,12,0,0) times = p.generate_times(itime,ftime,6*3600) path_to_plots = os.path.join(outdir,foldername) # for time in times: #pdb.set_trace() # Produce .npy data files with DKE data # print("Compute_diff_energy...") wrfouts = ['/uufs/chpc.utah.edu/common/home/horel-group2/lawson2/201111{0}00/{1}/wrfout_d01_2011-11-{2}_00:00:00_PLEV'.format(rundate,e,rundate) for e in ensnames] clvs = 0 plotfname = 'deltaDKE' # p.compute_diff_energy('sum_z','kinetic',path_to_wrfouts,times,upper=500, # d_save=runfolder, d_return=0,d_fname='DKE_500_sixhrly_'+foldername) p.delta_diff_energy('sum_z','kinetic',runfolder,'DKE_500_'+foldername,path_to_plots,plotfname, clvs,wrfouts,'GHT',)
import numpy as N sys.path.append('/path/to/WEM/') from WEM.postWRF import WRFEnviron p = WRFEnviron() # Time of initialisation nct = (2006,5,10,0,0,0) # Or name of wrfout file ncf = = 'wrfout_d01...' # Directories for output and netCDF files outdir = '/absolute/path/to/figures/' ncdir = '/absolute/path/to/wrfoutdata/' itime = (2006,5,10,12,0,0) ftime = (2006,5,11,12,0,0) hourly = 3 times = p.generate_times(itime,ftime,hourly*60*60) level = 2000 for t in times: p.plot2D('cref',t,level,outdir=outdir,ncdir=ncdir,ncf=ncf, nct=nct,legend=True) p.plot2D('Z',t,(300,850),outdir=outdir,ncdir=ncdir,ncf=ncf, nct=nct,legend=True) p.plot_streamlines()