exhwrf_ocean_init.py

#! /usr/bin/env python

##@namespace scripts.exhwrf_ocean_init
# Runs the ocean initialization for the chosen ocean model and sets
# the ocean status file accordingly.  Does nothing if ocean is disabled
# via [config] run_ocean=no

import os, sys
import produtil.log, produtil.setup
import pom.exceptions
import hwrf.mpipomtc, hwrf.hycom, hwrf.coupling
from produtil.log import jlogger

def main():
    """!Runs the ocean initialization and sets the ocean status file."""
    import hwrf_expt
    hwrf_expt.init_module()
    conf=hwrf_expt.conf
    ocean=conf.getstr('config','ocean_model')
    logger=conf.log('exhwrf_ocean_init')
    ocstatus=hwrf_expt.ocstatus
    ocstatus.unset(logger)
    try:
        if not conf.getbool('config','run_ocean'):
            jlogger.info('Ocean is disabled.  This job need not be run.')
            ocstatus.set(False,logger)
            return
        if ocean=='POM':
            hwrf_expt.pominit.run()
            ocstatus.set(True,logger)
        elif ocean=='HYCOM':
            hi=hwrf_expt.hycominit
            hi.run()
            hi.fill_ocstatus(ocstatus,logger)
        else:
            jlogger.critical('Config file error: unsupported ocean model '
                             '%s.  Will run without ocean.'%(repr(ocean),))
            ocstatus.set(False,logger)
    except pom.exceptions.POMUnsupportedBasin as e:
        produtil.log.postmsg('Unsupported basin: will run without ocean.')
        ocstatus.set(False,logger)
        return
    except Exception as e:
        if conf.getbool('config','allow_fallbacks',False):
            logger.error('Could not run ocean init: will run without ocean.'
                         '  Unhandled exception: '+str(e),exc_info=True)
            ocstatus.set(False,logger)
            return
        raise

if __name__=='__main__':
    try:
        produtil.setup.setup()
        main()
    except Exception as e:
        jlogger.critical('HWRF Ocean Initialization is aborting: '
                         +str(e),exc_info=True)
        sys.exit(2)