ldas_setup

#!/usr/bin/env python3

import sys
import argparse
import resource
from setup_utils import *
from ldas        import *

def parseCmdLine():
    """
    parse command line arguments and return a dict of options
    """

    #print 'in: parseCmdLine'
    p = argparse.ArgumentParser(
        description= \
            "Script to setup a GEOSldas experiment. "\
            "To set up an *offline* GEOSldas experiment, the script requires "\
            "two (2) input files, one for the experiment inputs ('exeinp') and the "\
            "other for the (SLURM) resource manager ('batinp'). To create sample input "\
            "files use './ldas_setup sample -h'. "\
            "When used to set up a coupled land-atm DAS experiment, the script "\
            "*ignores* the input files, but dummy strings must still be specified. ",
        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
        )
    p_sub = p.add_subparsers(help='sub-command help')

    
    # subparser: "sample" command
    p_sample = p_sub.add_parser(
        'sample',
        help='write sample input files',
        description='Print sample input files - either for the '\
            'experiment inputs or the resource manager (SLURM)',
        )
    group = p_sample.add_mutually_exclusive_group(required=True)
    group.add_argument(
        '--exeinp',
        help='print sample input file used to generate RC files for GEOSldas App.',
        action='store_true',
        )
    group.add_argument(
        '--batinp',
        help='print sample input file for the resource manager (SLURM)',
        action='store_true',
        )

    
    # subparser: "setup" command
    p_setup = p_sub.add_parser(
        'setup',
        help='setup LDAS experiment',
        description="The 'setup' sub-command is used to set up a GEOSldas " \
            "experiment. The positional argument 'exphome' is used to create " \
            "work_path (=exphome+/output) and run_path (=exphome+/run)."
        )
    p_setup.add_argument(
        '-v',
        '--verbose',
        help='verbose output',
        action='store_true',
        )
    p_setup.add_argument('exphome', help='experiment location')
    p_setup.add_argument(
        'exeinpfile',
        help='input file with arguments used to generate RC files for GEOSldas App',
        )
    p_setup.add_argument(
        'batinpfile',
        help='input file with arguments for SLURM',
        )

    # the following command line arguments, if present, take precedence over what
    #   is specified in the exeinp and batinp files
    ladas_cpl_help='land-atm DAS coupling mode: default=0 (not coupled); LDAS coupled with (1) ADAS deterministic member or (2) ADAS ensemble. Required when exeinp is dummy'
    p_setup.add_argument(
        '--ladas_cpl',
        help=ladas_cpl_help,
        type=str
        )
    p_setup.add_argument(
        '--account',
        help='overwrites computing/sponsor account from batinp file',
        type=str
        )
    p_setup.add_argument(
        '--nymdb',
        help='overwrites date in BEG_DATE from exeinp file: yyyymmdd. Required when exeinp is dummy',
        type=str
        )
    p_setup.add_argument(
        '--nhmsb',
        help='overwrites time in BEG_DATE from exeinp file: hhmmss. Required when exeinp is dummy',
        type=str
        ) 
    p_setup.add_argument(
        '--agcm_res',
        help='AGCM resolution associated with boundary conditions: nnnn (4 digits). Required when exeinp is dummy',
        type=str
        )
    p_setup.add_argument(
        '--bcs_version',
        help='boundary conditions version. Required when exeinp is dummy',
        type=str
        )
    p_setup.add_argument(
        '--rstloc',
        help='location of LDAS restarts (restart_path/restart_id). Required when exeinp is dummy',
        type=str
        )
    p_setup.add_argument(
        '--varwindow',
        help='ADAS analysis window (minutes). Required when exeinp is dummy',
        type=str
        )
    p_setup.add_argument(
        '--nens',
        help='number of ensemble members. Required when exeinp is dummy',          
        type=str
        )  

    # obsolete command line args
    p_setup.add_argument(
        '--runmodel',
        help='Obsolete.',
        action='store_true',
        )
    spltgrp = p_setup.add_mutually_exclusive_group()
    spltgrp.add_argument(
        '--daysperjob',
        type=int,
        metavar='N',
        help='Obsolete.  Use NUM_SGMT and JOB_SGMT in exeinp file.',
        )
    spltgrp.add_argument(
        '--monthsperjob',
        type=int,
        metavar='N',
        help='Obsolete.  Use NUM_SGMT and JOB_SGMT in exeinp file.',
        )

    return p.parse_args()

if __name__=='__main__':

    resource.setrlimit(resource.RLIMIT_STACK, (resource.RLIM_INFINITY, resource.RLIM_INFINITY))
    #convert command line to dictionary
    args = vars(parseCmdLine())
   
    #./ldas_setup sample sub-command
    # print input sample file then exit 
    if 'exeinp' in args:
      printInputSampleFile(args)
      sys.exit(0)

    # start ./ldas_setup setup sub-command

    ldasObj = ldas(args)

    print ("creating dir structure")
    status = ldasObj.createDirStructure()
    assert(status)

    print ("creating links to restarts, BCs, met forcing, ...")
    status = ldasObj.createLnRstBc()
    assert(status)

    print ("creating RC Files")
    status = ldasObj.createRCFiles()
    assert(status)

    print ("creating gcm style batch Run scripts lenkf.j")
    status = ldasObj.createRunScripts()
    assert(status)
    
    print ("creating batch Run scripts")
    status = ldasObj.createBatchRun()
    assert (status)