func.py

from faasmtools.compile_util import wasm_cmake, wasm_copy_upload
from invoke import task
from os.path import join
from tasks.env import PROJ_ROOT

FUNC_DIR = join(PROJ_ROOT, "func")
FUNC_BUILD_DIR = join(PROJ_ROOT, "build", "func")


def _copy_built_function(user, func):
    exe_name = "{}_{}.{}".format(user, func, "wasm")
    src_file = join(FUNC_BUILD_DIR, user, exe_name)
    wasm_copy_upload(user, func, src_file)


@task(default=True)
def compile(ctx, user, func, clean=False, debug=False):
    """
    Compile a function to test a sample library
    """
    # Keep track of which functions require the threads target
    is_threads = user in ["tf"]

    # Build the function (gets written to the build dir)
    wasm_cmake(
        FUNC_DIR,
        FUNC_BUILD_DIR,
        "{}_{}".format(user, func),
        clean,
        debug,
        is_threads=is_threads,
    )

    # Copy into place
    _copy_built_function(user, func)


@task
def tests(ctx, clean=False):
    """
    Build the functions used in the tests
    """
    funcs = [
        ["ffmpeg", "check"],
        ["opencv", "check"],
        ["opencv", "pca"],
        ["rabe", "test"],
        ["tf", "check"],
        # TODO: this two functions are not used in the tests, as they are used
        # to exercise migration, for what we need a distributed test setting
        ["lammps", "chain"],
        ["mpi", "migrate"],
    ]

    for user, func in funcs:
        compile(ctx, user, func, clean=clean)