buildlib.mpi-serial

#!/usr/bin/env python3
import os, sys, logging, argparse

from standard_script_setup import *
from CIME import utils
from CIME.config import Config
from CIME.utils import copyifnewer, run_bld_cmd_ensure_logging
from CIME.case import Case
from CIME.build import get_standard_makefile_args
import glob

logger = logging.getLogger(__name__)


def parse_command_line(args, description):
    ###############################################################################
    parser = argparse.ArgumentParser(
        usage="""\n{0} [--debug]
OR
{0} --verbose
OR
{0} --help

\033[1mEXAMPLES:\033[0m
    \033[1;32m# Run \033[0m
    > {0}
""".format(
            os.path.basename(args[0])
        ),
        description=description,
        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
    )

    utils.setup_standard_logging_options(parser)

    parser.add_argument("buildroot", help="build path root")

    parser.add_argument("installpath", help="install path ")

    parser.add_argument(
        "caseroot", nargs="?", default=os.getcwd(), help="Case directory to build"
    )

    args = utils.parse_args_and_handle_standard_logging_options(args, parser)

    return args.buildroot, args.installpath, args.caseroot


def buildlib(bldroot, installpath, case):
    ###############################################################################
    caseroot = case.get_value("CASEROOT")
    srcroot = case.get_value("SRCROOT")

    customize_path = os.path.join(srcroot, "cime_config", "customize")

    config = Config.load(customize_path)

    mct_path = config.mct_path.format(srcroot=srcroot)

    for _file in glob.iglob(os.path.join(mct_path, "mpi-serial", "*.h")):
        copyifnewer(_file, os.path.join(bldroot, os.path.basename(_file)))

    gmake_opts = "-f {} ".format(os.path.join(caseroot, "Tools", "Makefile"))
    gmake_opts += " -C {} ".format(bldroot)
    gmake_opts += " {} ".format(get_standard_makefile_args(case, shared_lib=True))
    gmake_opts += "COMP_NAME=mpi-serial {}".format(
        os.path.join(bldroot, "Makefile.conf")
    )

    gmake_cmd = case.get_value("GMAKE")

    # This runs the mpi-serial configure command
    cmd = "{} {}".format(gmake_cmd, gmake_opts)
    run_bld_cmd_ensure_logging(cmd, logger)

    # Now we run the mpi-serial make command
    gmake_opts = "-f {} ".format(os.path.join(mct_path, "mpi-serial", "Makefile"))
    gmake_opts += " -C {} ".format(bldroot)
    gmake_opts += " -j {} ".format(case.get_value("GMAKE_J"))
    gmake_opts += " SRCDIR={} ".format(os.path.join(mct_path))

    cmd = "{} {}".format(gmake_cmd, gmake_opts)
    run_bld_cmd_ensure_logging(cmd, logger)

    copyifnewer(
        os.path.join(bldroot, "libmpi-serial.a"),
        os.path.join(installpath, "lib", "libmpi-serial.a"),
    )
    for _file in ("mpi.h", "mpif.h", "mpi.mod", "MPI.mod"):
        if os.path.isfile(os.path.join(bldroot, _file)):
            copyifnewer(
                os.path.join(bldroot, _file),
                os.path.join(installpath, "include", _file),
            )


def _main(argv, documentation):
    bldroot, installpath, caseroot = parse_command_line(argv, documentation)
    with Case(caseroot) as case:
        buildlib(bldroot, installpath, case)


if __name__ == "__main__":
    _main(sys.argv, __doc__)