#! /usr/bin/env python

# script to build vamp-aubio-plugin with waf (https://waf.io)

import sys, os

local_aubio_include  = 'contrib/aubio-dist/include'
local_aubio_lib      = 'contrib/aubio-dist/lib'
local_vamp_include   = 'contrib/vamp-plugin-sdk-2.6'
local_vamp_lib_i686  = 'contrib/vamp-plugin-sdk-2.6-binaries-i686-linux'
local_vamp_lib_amd64 = 'contrib/vamp-plugin-sdk-2.6-binaries-amd64-linux'

def options(opt):
    opt.load('compiler_cxx')

def configure(conf):
    conf.load('compiler_cxx')

    if os.path.isdir(local_aubio_include):
        conf.env.append_value('CXXFLAGS', '-I../'+local_aubio_include)
        conf.env.append_value('LINKFLAGS', '-L../'+local_aubio_lib)
        conf.env.append_value('LINKFLAGS', '-laubio')
    else:
        conf.check_cfg (package='aubio', uselib_store='AUBIO',
                args=['--cflags', '--libs'], mandatory=True)

    if os.path.isdir(local_vamp_include):
        conf.env.append_value('CXXFLAGS', '-I../'+local_vamp_include)
        conf.env.append_value('SHLIB_MARKER', '-lvamp-sdk')
        if os.path.isdir(local_vamp_lib_amd64):
            conf.env.append_value('LINKFLAGS', '-L../'+local_vamp_lib_amd64)
        if os.path.isdir(local_vamp_lib_i686):
            conf.env.append_value('LINKFLAGS', '-L../'+local_vamp_lib_i686)
        conf.check(lib = 'vamp-sdk', mandatory = False)
    else:
        conf.check_cfg (package='vamp-sdk', uselib_store = 'VAMP',
                args=['--cflags','--libs'], mandatory=True)

    if sys.platform.startswith('linux'):
        conf.env['CXXFLAGS'] += ['-Wall', '-Wextra', '-O3', '-msse', '-msse2',
                '-mfpmath=sse', '-ftree-vectorize']
        conf.env.append_value('LINKFLAGS', '-Wl,-z,defs')
        # add plugin.map
        conf.env.append_value('LINKFLAGS', '-Wl,--version-script=../vamp-plugin.map')

def build(bld):
    # Host Library
    plugin_sources = bld.path.ant_glob('plugins/*.cpp')
    plugin_sources += bld.path.ant_glob('*.cpp')

    # rename libvamp-aubio to vamp-plugin binary name
    bld.env['cxxshlib_PATTERN'] = '%s.so'

    bld.program(source = plugin_sources,
               includes = '.',
               target = 'vamp-aubio',
               name = 'vamp-aubio',
               use = ['VAMP', 'AUBIO'],
               features = 'cxx cxxshlib'
               )

    #for k in bld.env.keys():
    #    print ("%s : %s", k, bld.env[k] )