compile.py

import argparse
import multiprocessing
import subprocess
import sys
import os

DEFAULT_CPU_COUNT = 4


def do_compile(args):
    try:
        cpu_count = multiprocessing.cpu_count()
    except NotImplementedError:
        cpu_count = DEFAULT_CPU_COUNT

    if args.build_parallelism:
        cpu_count = int(args.build_parallelism)

    # Get absolute path to source directory
    if args.src_dir:
      abs_src_dir = os.path.abspath(args.src_dir)
    else:
      abs_src_dir = os.path.abspath(os.path.join(__file__, "../.."))
    # Get absolute path to build directory
    if args.obj_dir:
      abs_obj_dir = os.path.abspath(args.obj_dir)
    else:
      abs_obj_dir = os.path.join(abs_src_dir, "build")

    cmake_cmd = [
        "cmake",
        "--build", abs_obj_dir,
        "--",
        args.build_target,
        "-j", str(cpu_count)]

    if args.verbose:
      cmake_cmd.append("--verbose")

    print("[Cmake Command]: {}".format(" ".join(cmake_cmd)))

    subprocess.check_call(cmake_cmd, cwd=abs_obj_dir)

    return True


def main():
    parser = argparse.ArgumentParser(prog="compile.py",
                                     description="script to do compile",
                                     formatter_class=argparse.RawTextHelpFormatter)
    parser.add_argument("-n", "--build-number", metavar="BUILD_NUM", help="build number")
    parser.add_argument("-b", "--branch", metavar="BRANCH", help="pull request branch")
    parser.add_argument("-d", "--base-branch", metavar="BASE_BRANCH", help="pull request base branch")
    parser.add_argument("-r", "--pr-number", metavar="PR_NUM", help="pull request number")
    parser.add_argument("-w", "--builder-dir", metavar="BUILDER_DIR",
                        help="builder directory, which is the directory containing source and build directories")
    parser.add_argument("-s", "--src-dir", metavar="SRC_DIR", help="source directory")
    parser.add_argument("-o", "--obj-dir", metavar="OBJ_DIR", help="build directory")
    parser.add_argument("-j", "--build-parallelism", metavar="BUILD_PARALLELISM", help="build parallelism")
    parser.add_argument("-v", "--verbose", action='store_true', help="verbose build output")
    parser.add_argument("-t", "--build-target", metavar="BUILD_TARGET", default="deploy-sycl-toolchain", help="set build target")

    args = parser.parse_args()

    print("args:{}".format(args))

    return do_compile(args)


if __name__ == "__main__":
    ret = main()
    exit_code = 0 if ret else 1
    sys.exit(exit_code)