wscript

import os

from waflib.TaskGen import feature, after_method
from waflib.extras import test_base
from waflib.extras.test_base import summary


def depends(dep):
    dep("hicann-dls-scripts", branch="v3testing")
    dep("haldls")


def options(opt):
    opt.load('nux_assembler')
    opt.load('nux_compiler')
    opt.load('test_base')
    opt.load('pytest')
    opt.add_option("--enable-stack-protector",
                   default=False,
                   action='store_true',
                   dest='stack_protector',
                   help="Enable stack frame overflow check.")
    opt.add_option("--enable-stack-redzone",
                   default=False,
                   action='store_true',
                   dest='stack_redzone',
                   help="Enable stack redzone check towards the program memory region.")
    opt.add_option("--disable-mailbox",
                   default=False,
                   action='store_true',
                   dest='disable_mailbox',
                   help="Disable mailbox memory region.")


def configure(conf):
    # host-based python stuff also needed for cross-env tests
    conf.load('pytest')

    # now configure for nux cross compiler
    env = conf.env
    conf.setenv('nux')
    conf.load('nux_assembler')
    conf.load('nux_compiler')
    conf.load('objcopy')
    conf.load('test_base')
    if(conf.options.stack_protector):
        conf.define("LIBNUX_STACK_PROTECTOR", True)
        conf.env.append_value('LIBNUX_STACK_PROTECTOR_ENABLED', 'True')
        conf.env.append_value('CFLAGS', '-fstack-protector-all')
        conf.env.append_value('CXXFLAGS', '-fstack-protector-all')
    else:
        conf.env.append_value('LIBNUX_STACK_PROTECTOR_ENABLED', 'False')
    if(conf.options.stack_redzone):
        conf.define("LIBNUX_STACK_REDZONE", True)
        conf.env.append_value('LIBNUX_STACK_REDZONE_ENABLED', 'True')
        conf.env.append_value('CFLAGS', '-fstack-limit-symbol=stack_redzone')
        conf.env.append_value('CXXFLAGS', '-fstack-limit-symbol=stack_redzone')
    else:
        conf.env.append_value('LIBNUX_STACK_REDZONE_ENABLED', 'False')
    if(not conf.options.disable_mailbox):
        conf.env.append_value('LINKFLAGS', '-Wl,--defsym=__mailbox__=1')
    # specialize for v2
    conf.setenv('nux_v2', env=conf.all_envs['nux'])
    conf.define('LIBNUX_DLS_VERSION_V2', True)

    # specialize for v3
    conf.setenv('nux_v3', env=conf.all_envs['nux'])
    conf.define('LIBNUX_DLS_VERSION_V3', True)

    # specialize for vx
    conf.setenv('nux_vx', env=conf.all_envs['nux'])
    conf.define('LIBNUX_DLS_VERSION_VX', True)

    # restore env
    conf.setenv('', env=env)

def build(bld):
    bld.env.dls_partition = "dls" == os.environ.get("SLURM_JOB_PARTITION")
    bld.env.cube_partition = "cube" == os.environ.get("SLURM_JOB_PARTITION")
    if (bld.env.dls_partition):
        # FIXME should come from env or hwdb, see Issue #3366
        bld.env.dls_test_version = ["2", "3"][(int("B291673" in os.environ.get("SLURM_FLYSPI_ID")))]

    for dls_version in ['v2', 'v3', 'vx']:
        env = bld.all_envs['nux_' + dls_version]

        bld(
            target = 'nux_inc_' + dls_version,
            export_includes = ['.'],
            env = env,
        )

        nux_sources = [
            'src/bitformatting.cpp',
            'src/correlation.cpp',
            'src/counter.cpp',
            'src/exp.cpp',
            'src/fxv.cpp',
            'src/mailbox.cpp',
            'src/random.cpp',
            'src/stack_guards.cpp',
            'src/spikes.cpp',
            'src/time.cpp',
            'src/unittest.cpp',
            'src/unittest_mailbox.cpp',
        ]

        bld.stlib(
            target = 'nux_' + dls_version,
            source = nux_sources,
            use = ['nux_inc_' + dls_version],
            env = env,
        )

        bld(
            features = 'cxx',
            name = 'nux_runtime_obj_' + dls_version,
            source = ['src/start.cpp',
                      'src/initdeinit.cpp',
                      'src/cxa_pure_virtual.cpp'],
            use = 'nux_inc_' + dls_version,
            env = env,
        )

        bld(
            name = 'nux_runtime_' + dls_version,
            target = 'crt.o',
            source = ['src/crt.s'],
            features = 'use asm',
            use = ['nux_runtime_obj_' + dls_version],
            env = env,
        )

        program_list = [
            'examples/stdp.cpp',
            'test/test_bitformatting.cpp',
            'test/test_bool.cpp',
            'test/test_helper.cpp',
            'test/test_malloc.cpp',
            'test/test_measure_time.cpp',
            'test/test_returncode.cpp',
            'test/test_stack_guard.cpp',
            'test/test_stack_redzone.cpp',
            'test/test_unittest.cpp',
        ]

        if dls_version != 'vx':
            # These tests don't work for HX, see Issue #3365
            program_list += [
                'test/test_fxvadd.cpp',
                'test/test_fxvsel.cpp',
                'test/test_inline_vector_argument.cpp',
                'test/test_many_vectors.cpp',
                'test/test_noinline_vector_argument.cpp',
                'test/test_return_vector.cpp',
                'test/test_synram_rw.cpp',
                'test/test_vector.cpp',
                'test/test_vector_alignment.cpp',
                'test/test_vector_cc.cpp',
                'test/test_vector_sync.cpp',
                'test/test_xorshift_vector.cpp',
            ]

        if dls_version != 'v3':
            # This test doesn't work on the v3 setup
            program_list += [
                'test/test_neuron_counter.cpp',
            ]

        for program in program_list:
            bld.program(
                features = 'cxx objcopy',
                objcopy_bfdname = 'binary',
                target = program.replace('.cpp', '') + '_' + dls_version + '.bin',
                source = [program],
                use = ['nux_' + dls_version, 'nux_runtime_' + dls_version],
                env = bld.all_envs['nux_' + dls_version],
            )

        def max_size_empty():
            stack_protector = env.LIBNUX_STACK_PROTECTOR_ENABLED[0].lower() == "true"
            stack_redzone = env.LIBNUX_STACK_REDZONE_ENABLED[0].lower() == "true"
            build_profile = bld.options.build_profile

            if not stack_protector and not stack_redzone:
                if build_profile == 'release':
                    return 400
                else:
                    return 512

            if stack_protector and not stack_redzone:
                if build_profile == 'release':
                    return 784
                else:
                    return 864

            if not stack_protector and stack_redzone:
                if build_profile == 'release':
                    return 496
                else:
                    return 608

            if stack_protector and stack_redzone:
                if build_profile == 'release':
                    return 928
                else:
                    return 976

        bld.program(
            features = 'cxx objcopy check_size',
            check_size_max = max_size_empty(),
            objcopy_bfdname = 'binary',
            target = 'test_empty_' + dls_version + '.bin',
            source = ['test/test_empty.cpp'],
            use = ['nux_' + dls_version, 'nux_runtime_' + dls_version],
            env = env,
        )

    bld(
        name='libnux_hwtests_vx',
        tests='test/test_hwtests.py',
        features='use pytest',
        use='run_ppu_program_vx_py',
        install_path='${PREFIX}/bin/tests',
        skip_run=not bld.env.cube_partition,
        env = bld.all_envs[''],
        test_environ = dict(STACK_PROTECTION=env.LIBNUX_STACK_PROTECTOR_ENABLED[0],
                            STACK_REDZONE=env.LIBNUX_STACK_REDZONE_ENABLED[0],
                            TEST_BINARY_PATH=os.path.join(bld.env.PREFIX, 'build', 'libnux'),
                            TEST_DLS_VERSION="vx")
    )

    bld(
        name='libnux_hwtests_v2',
        tests='test/test_hwtests.py',
        features='use pytest',
        use='run_ppu_program_v2_py',
        install_path='${PREFIX}/bin/tests',
        skip_run=not (bld.env.dls_partition and bld.env.dls_test_version == "2"),
        env = bld.all_envs[''],
        test_environ = dict(STACK_PROTECTION=env.LIBNUX_STACK_PROTECTOR_ENABLED[0],
                            STACK_REDZONE=env.LIBNUX_STACK_REDZONE_ENABLED[0],
                            TEST_BINARY_PATH=os.path.join(bld.env.PREFIX, 'build', 'libnux'),
                            TEST_DLS_VERSION="v2")
    )

    bld(
        name='libnux_hwtests_v3',
        tests='test/test_hwtests.py',
        features='use pytest',
        use='hdls-scripts_runprogram',
        install_path='${PREFIX}/bin/tests',
        skip_run=not (bld.env.dls_partition and bld.env.dls_test_version == "3"),
        env = bld.all_envs[''],
        test_environ = dict(STACK_PROTECTION=env.LIBNUX_STACK_PROTECTOR_ENABLED[0],
                            STACK_REDZONE=env.LIBNUX_STACK_REDZONE_ENABLED[0],
                            TEST_BINARY_PATH=os.path.join(bld.env.PREFIX, 'build', 'libnux'),
                            TEST_DLS_VERSION="v3")
    )

    bld.add_post_fun(summary)


class check_size(test_base.TestBase):
    def run(self):
        test = self.inputs[0]
        test_abspath = test.abspath()
        xmlfile_abspath = self.getXMLFile(test).abspath()
        max_size = self.generator.check_size_max
        cmd = ['python test/helpers/check_obj_size.py {} {} {}'.format(
            test_abspath, xmlfile_abspath, max_size)]
        self.runTest(test, cmd)


@feature('check_size')
@after_method('apply_link', 'process_use', 'propagate_uselib_vars')
def check_size_run_test(self):
    if self.testsDisabled():
        return
    if self.isTestExecutionEnabled() and getattr(self, 'link_task', None):
        t = self.create_task('check_size', self.link_task.outputs)
        self.bld.env = self.env
        t.init(self)