def generate_namelist_input_d01_wrf(namelist_input_master,run_dir='.'):
"""Generate a namelist file for wrf.exe for the outermost domain
The namelist file for wrf.exe for the outermost domain differs from the
'master' namelist file by the entry in max_dom (max_dom=1)
"""
namelist_dict=wu.read_namelist(namelist_input_master)
namelist_dict['&domains']['max_dom'][0]=1
output_file = os.path.join(run_dir,'namelist.input.d01.wrf')
wu.write_namelist(namelist_dict,output_file)
return output_file
def generate_namelist_input_dpp_real(pp,namelist_input_master,run_dir='.'):
"""Generate a namelist for real.exe for the pp'th domain
This namelist will contain only one column, with max_dom = 1 and
interval_seconds = grib interval
"""
namelist_dict=wu.read_namelist(namelist_input_master)
idx_discard=range(int(namelist_dict['&domains']['max_dom'][0]))
# We'll be using this list as the list of entries to 'pop' from each variable
# in the namelist, so we need ot reverse siht list to preserve only the 'pp'th
# datum
idx_discard.pop(idx_discard.index(pp-1))
idx_discard.reverse()
for group in namelist_dict.keys():
for variable in namelist_dict[group].keys():
dummy=namelist_dict[group][variable]
if len(dummy)==1:
pass
else:
for idx in idx_discard:
dummy.pop(idx)
# &domains
namelist_dict['&domains']['max_dom'][0]=1
# grid id
namelist_dict['&domains']['grid_id'][0]=1
namelist_dict['&domains']['parent_id'][0]=0
namelist_dict['&domains']['parent_grid_ratio'][0]=1
# j_parent_start of parent is 1
namelist_dict['&domains']['j_parent_start'][0]=0
# same for i_parent_start
namelist_dict['&domains']['i_parent_start'][0]=0
# &time_control
namelist_dict['&time_control']['input_from_file'][0]='.true.'
# &bdy_control
# parent grid is not nested as far as WRF is concerned...
namelist_dict['&bdy_control']['nested'][0]='.false.'
# parent grid will have specified boundary conditions
namelist_dict['&bdy_control']['specified'][0]='.true.'
# TODO
# Shorten the time extent to save on CPU resources (ref. step 15 in dia)
# Include explicit definition of eta_levels
output_file = os.path.join(run_dir,'namelist.input.d'+str(pp).zfill(2)+'.real')
wu.write_namelist(namelist_dict,output_file)
return output_file
def generate_namelist_input_dpp_wrf(pp,namelist_input_master,run_dir='.'):
"""Generate a namelist for wrf.exe for the pp'th domain
This namelist will contain only one column, with max_dom = 1 and
interval_seconds = grib interval
"""
namelist_dict=wu.read_namelist(namelist_input_master)
idx_discard=range(int(namelist_dict['&domains']['max_dom'][0]))
# We'll be using this list as the list of entries to 'pop' from each variable
# in the namelist, so we need ot reverse siht list to preserve only the 'pp'th
# datum
idx_discard.pop(idx_discard.index(pp-1))
idx_discard.reverse()
# multiplicative time step ratio
mult_time_step_ratio=1
# Must incluide CURRENT parent_time_step ratio as well, i.e. up to pp rather than pp-1.
# time_step_ratio_1*time_step_ratio_2*...*time_step_ratio_pp
for idx in range(pp):
mult_time_step_ratio*=namelist_dict['&domains']['parent_time_step_ratio'][idx]
for group in namelist_dict.keys():
for variable in namelist_dict[group].keys():
dummy=namelist_dict[group][variable]
if len(dummy)==1:
pass
else:
for idx in idx_discard:
dummy.pop(idx)
# &domains
# take care of time step stuff
# be smart... make sure that your INITIAL time step is an integer multiple of the
# product of the set of parent_time_step_ratio OR come here and modify this yourself...
# TODO: modify this myself to handle integer arithmetics. Need to consider
# time_step_fract_num and time_step_fract_den
namelist_dict['&domains']['parent_time_step_ratio'][0]=1
namelist_dict['&domains']['time_step'][0]/=mult_time_step_ratio
namelist_dict['&domains']['max_dom'][0]=1
# grid id
namelist_dict['&domains']['grid_id'][0]=1
namelist_dict['&domains']['parent_id'][0]=0
namelist_dict['&domains']['parent_grid_ratio'][0]=1
# j_parent_start of parent is 1
namelist_dict['&domains']['j_parent_start'][0]=0
# same for i_parent_start
namelist_dict['&domains']['i_parent_start'][0]=0
# &time_control
namelist_dict['&time_control']['input_from_file'][0]='.true.'
# &bdy_control
# parent grid is not nested as far as WRF is concerned...
namelist_dict['&bdy_control']['nested'][0]='.false.'
# parent grid will have specified boundary conditions
namelist_dict['&bdy_control']['specified'][0]='.true.'
# TODO
# Shorten the time extent to save on CPU resources (ref. step 15 in dia)
# Include explicit definition of eta_levels
output_file = os.path.join(run_dir,'namelist.input.d'+str(pp).zfill(2)+'.wrf')
wu.write_namelist(namelist_dict,output_file)
return output_file
def generate_namelist_input_dpp_ndown(pp,namelist_input_master,run_dir='.'):
"""Generate a namelist for ndown.exe for the pp'th domain
This namelist will contain only one column, with max_dom = 1 and
interval_seconds = grib interval
"""
namelist_dict=wu.read_namelist(namelist_input_master)
# Correcting to give BC's at time of history_interval for parent grid.
# history_interval given in minutes, interval_seconds, in seconds (duh??)
dom_idx = pp-1
history_interval = namelist_dict['&time_control']['history_interval'][dom_idx-1]
interval_seconds = history_interval*60
namelist_dict['&time_control']['interval_seconds'][0]=interval_seconds
idx_discard=range(int(namelist_dict['&domains']['max_dom'][0]))
# We'll be using this list as the list of entries to 'pop' from each variable
# in the namelist, so we need ot reverse siht list to preserve only the 'pp'th
# datum
idx_discard.pop(idx_discard.index(dom_idx))
idx_discard.pop(idx_discard.index(dom_idx-1))
idx_discard.reverse()
for group in namelist_dict.keys():
for variable in namelist_dict[group].keys():
dummy=namelist_dict[group][variable]
if len(dummy)==1:
pass
else:
for idx in idx_discard:
dummy.pop(idx)
# &domain corrections
namelist_dict['&domains']['max_dom'][0]=2
# grid id
namelist_dict['&domains']['grid_id']=[1,2]
# ndown reads 1 as parent and 2 as child hence
namelist_dict['&domains']['parent_id']=[0,1]
# parent grid ratio of 'parent' is 1:
namelist_dict['&domains']['parent_grid_ratio'][0]=1
# j_parent_start of parent is 1
namelist_dict['&domains']['j_parent_start'][0]=0
# same for i_parent_start
namelist_dict['&domains']['i_parent_start'][0]=0
# currently not changing parent_time_step_ratio, but watch this space
# &time control corrections
namelist_dict['&time_control']['input_from_file'][0]='.true.'
# &bdy_control
# parent grid is not nested as far as WRF is concerned...
namelist_dict['&bdy_control']['nested'][0]='.false.'
# parent grid will have specified boundary conditions
namelist_dict['&bdy_control']['specified'][0]='.true.'
output_file = \
os.path.join(run_dir,'namelist.input.d'+str(pp).zfill(2)+'.ndown')
wu.write_namelist(namelist_dict,output_file)
return output_file