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


def options(opt):
    opt.load('nux_compiler')
    opt.load('test_base')
    opt.add_option("--dls-version",
                   choices=["2", "3"],
                   help="DLS version to use (2 or 3).")


def configure(conf):
    env = conf.env
    conf.setenv('nux')
    if not conf.options.dls_version:
        raise RuntimeError("Please specify DLS version to use (2 or 3).")
    conf.define("LIBNUX_DLS_VERSION", int(conf.options.dls_version))
    conf.load('nux_compiler')
    conf.load('objcopy')
    conf.load('test_base')
    # restore env
    conf.setenv('', env=env)

def build(bld):
    bld(
        target = 'nux_inc',
        export_includes = ['.'],
        env = bld.all_envs['nux'],
    )

    bld.stlib(
        target = 'nux',
        source = [
                'src/exp.c',
                'src/fxv.c',
                'src/mailbox.c',
                'src/unittest.c',
                'src/unittest_mailbox.c',
                ],
        use = ['nux_inc'],
        env = bld.all_envs['nux'],
    )

    bld(
        features = 'c',
        name = 'initdeinit_obj',
        source = 'src/initdeinit.c',
        env = bld.all_envs['nux'],
    )

    bld(
        features = 'c',
        name = 'cxa_pure_virtual_obj',
        source = 'src/cxa_pure_virtual.c',
        use = 'nux_inc',
        env = bld.all_envs['nux'],
    )

    bld(
        name = 'nux_runtime',
        target = 'crt.o',
        source = ['src/crt.s'],
        use = ['asm', 'initdeinit_obj'],
        env = bld.all_envs['nux'],
    )

    bld(
        name = 'nux_runtime_cpp',
        target = 'crt.o',
        source = ['src/crt.s'],
        use = ['asm', 'initdeinit_obj', 'cxa_pure_virtual_obj'],
        env = bld.all_envs['nux'],
    )

    bld.objects(
        features='c',
        target='spikes',
        name='spikes',
        source=['libnux/spikes.c'],
        use=['nux'],
        env = bld.all_envs['nux'],
    )

    bld.objects(
        features='c',
        target='random',
        name='random',
        source=['libnux/random.c'],
        use=['nux'],
        env = bld.all_envs['nux'],
    )

    bld.objects(
        features='c',
        target='time',
        name='time',
        source=['libnux/time.c'],
        use=['nux'],
        env = bld.all_envs['nux'],
    )

    bld.objects(
        features='c',
        target='counter',
        name='counter',
        source=['libnux/counter.c'],
        use=['nux'],
        env = bld.all_envs['nux'],
    )

    bld.objects(
        features='c',
        target='correlation',
        name='correlation',
        source=['libnux/correlation.c'],
        use=['nux'],
        env = bld.all_envs['nux'],
    )

    bld.program(
        features = 'c objcopy',
        objcopy_bfdname = 'binary',
        target = 'failing_test_returncode.bin',
        source = ['test/test_returncode.c'],
        use = ['nux', 'nux_runtime'],
        env = bld.all_envs['nux'],
    )

    bld.program(
        features = 'c objcopy',
        objcopy_bfdname = 'binary',
        target = 'failing_test_unittest.bin',
        source = ['test/test_unittest.c'],
        use = ['nux', 'nux_runtime'],
        env = bld.all_envs['nux'],
    )

    bld.program(
        features = 'c objcopy',
        objcopy_bfdname = 'binary',
        target = 'test_vector.bin',
        source = ['test/test_vector.c'],
        use = ['nux', 'nux_runtime'],
        env = bld.all_envs['nux'],
    )

    bld.program(
        features = 'cxx objcopy',
        objcopy_bfdname = 'binary',
        target = 'test_vector_cc.bin',
        source = ['test/test_vector.cc'],
        use = ['nux', 'nux_runtime_cpp'],
        env = bld.all_envs['nux'],
    )

    bld.program(
        features = 'c objcopy',
        objcopy_bfdname = 'binary',
        target = "test_fxvsel.bin",
        source = ["test/test_fxvsel.c"],
        use = ["nux", "nux_runtime"],
        env = bld.all_envs['nux'],
    )

    bld.program(
        features='c objcopy',
        objcopy_bfdname='binary',
        target="test_synram_rw.bin",
        source="test/test_synram_rw.c",
        use=["nux", "nux_runtime", "random"],
        env = bld.all_envs['nux'], )

    bld.program(
        features = 'c objcopy check_size',
        check_size_max = 348,
        objcopy_bfdname = 'binary',
        target = 'test_empty.bin',
        source = ['test/test_empty.c'],
        use = ['nux', 'nux_runtime'],
        env = bld.all_envs['nux'],
    )

    bld.program(
        features = 'c objcopy',
        objcopy_bfdname = 'binary',
        target = 'test_malloc.bin',
        source = ['test/test_malloc.c'],
        use = ['nux', 'nux_runtime_cpp'],
        env = bld.all_envs['nux'],
    )

    bld.program(
        features = 'cxx objcopy check_size',
        check_size_max = 400,
        objcopy_bfdname = 'binary',
        target = 'test_empty_cc.bin',
        source = ['test/test_empty.cc'],
        use = ['nux', 'nux_runtime_cpp'],
        env = bld.all_envs['nux'],
    )

    bld.program(
        features = 'c objcopy',
        objcopy_bfdname = 'binary',
        target = 'test_xorshift_vector.bin',
        source = ['test/test_xorshift_vector.c'],
        use = ['random','nux', 'nux_runtime'],
        env = bld.all_envs['nux'],
    )

    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/test_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)