# vim:set syntax=python:

uselib = []
uselib += ['M']
uselib += ['FFTW3', 'FFTW3F']
uselib += ['INTEL_IPP']
uselib += ['SAMPLERATE']
uselib += ['SNDFILE']
uselib += ['AVCODEC']
uselib += ['AVFORMAT']
uselib += ['SWRESAMPLE']
uselib += ['AVRESAMPLE']
uselib += ['AVUTIL']
uselib += ['BLAS']

source = ctx.path.ant_glob('*.c **/*.c')

ctx(features = 'c',
        source = source,
        includes = ['.'],
        use = uselib,
        target = 'lib_objects')

# build libaubio.so (cshlib) and/or libaubio.a (cstlib)
if ctx.env['DEST_OS'] in ['ios', 'iosimulator']:
    build_features = ['cstlib', 'cshlib']
elif ctx.env['DEST_OS'] in ['win32', 'win64']:
    build_features = ['cstlib', 'cshlib gensyms']
elif ctx.env['DEST_OS'] in ['emscripten']:
    build_features = ['cstlib','cshlib']
elif '--static' in ctx.env['LDFLAGS'] or '--static' in ctx.env['LINKFLAGS']:
    # static in cflags, ...
    build_features = ['cstlib']
else:
    # linux, darwin, android, mingw, ...
    build_features = ['cstlib', 'cshlib']

# also install static lib
from waflib.Tools.c import cstlib
from waflib.Tools.fc import fcstlib
fcstlib.inst_to = cstlib.inst_to = '${LIBDIR}'

import re
from waflib import TaskGen, Task
from waflib.Context import STDOUT
from waflib.Utils import O644

class gen_sym_file(Task.Task):
    color = 'BLUE'
    inst_to = '${LIBDIR}'
    def run(self):
        syms = {}
        reg = getattr(self.generator, 'export_symbols_regex','.+?')
        if 'msvc' in self.env.CC_NAME:
            outputs = [x.abspath() for x in self.generator.link_task.outputs]
            binary_path = list(filter(lambda x: x.endswith('lib'), outputs))[0]
            reg_compiled = re.compile(r'External\s+\|\s+(?P<symbol>%s)\b' % reg)
            cmd =(self.env.LINK_CC) + ['/dump', '/symbols', binary_path]
        else: # using gcc? assume we have nm
            binary_path = self.generator.link_task.outputs[0].abspath()
            reg_compiled = re.compile(r'(T|D)\s+_(?P<symbol>%s)\b'%reg)
            cmd = (self.env.NM or ['nm']) + ['-g', binary_path]
        dump_output = self.generator.bld.cmd_and_log(cmd, quiet=STDOUT)
        syms = set([])
        for m in reg_compiled.finditer(dump_output):
            syms.add(m.group('symbol'))
        syms = list(syms)
        syms.sort()
        self.outputs[0].write('EXPORTS\n'+'\n'.join(syms))

@TaskGen.feature('gensyms')
@TaskGen.after_method('process_source','process_use','apply_link','process_uselib_local','propagate_uselib_vars')
def gen_symbols(self):
    #sym_file = self.path.find_or_declare(self.target + '.def')
    sym_file_name = os.path.splitext(self.link_task.outputs[0].abspath())[0] + '.def'
    sym_file = self.path.find_or_declare(sym_file_name)
    symtask = self.create_task('gen_sym_file', self.link_task.outputs, sym_file)
    self.add_install_files(install_to=self.link_task.inst_to, install_from=sym_file,
        chmod=O644, task=self.link_task)

for target in build_features:
    ctx(features = 'c ' + target,
            use = uselib + ['lib_objects'],
            target = 'aubio',
            export_symbols_regex=r'(?:.*aubio|fvec|lvec|cvec|fmat|new|del)_.*',
            vnum = ctx.env['LIB_VERSION'])

# install headers, except _priv.h ones
ctx.install_files('${INCLUDEDIR}/aubio/',
        ctx.path.ant_glob('**/*.h', excl = ['**_priv.h', 'config.h']),
        relative_trick=True)