source: trunk/MESOSCALE/PLOT/PYTHON/pywrf.example.r39/wrf/pydown/utils.py @ 183

import sys
import os
from socket import gethostname

# VB the following is both host and user specific hence
hostname = gethostname()
user = os.getlogin()
if hostname == 'hn3.its.monash.edu.au':
    if user == 'vbisign':
        sys.path.append('/nfs/1/home/vbisign/wrf/pywrf')   
        import wrf.utils as wu
    elif user == 'tchubb':
        print 'Hey Thom where do you keep pywrf on this computer?'
        sys.exit()
        sys.path.append('/somewhere/pylib')
        import pywrf.wrf.utils as wu
elif hostname == 'linux450':
    # VB Sorry Thom if this is not correct ;)
    print 'Hey Thom where do you keep pywrf on this computer?'
    sys.exit()
    sys.path.append('/somewhere/pylib')
    import pywrf.wrf.utils as wu
elif hostname == 'val.maths.monash.edu.au' \
  or hostname == 'valerio-bisignanesis-computer.local':
    sys.path.append('/Users/val/Desktop/workspace/pywrf')
    import wrf.utils as wu
else:
    print 'Warning: since I do not know this user/'\
      + 'hostname combination, I am not sure of ' \
      + ' where to find pywrf.wrf.util, I will try\n' \
      + ' import pywrf.wrf.utils as wu'
    import pywrf.wrf.utils as wu



def generate_namelist_input_d01_real(namelist_input_master,run_dir='.'):
    """Generate a namelist file for real.exe for the outermost domain
    
    The namelist file for real.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.real')
    wu.write_namelist(namelist_dict,output_file)
    return output_file

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_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

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