wscript

#!/usr/bin/env python
# encoding: utf-8

import sys, glob
from os.path import join, exists, normpath
import os

def get_git_version():
    """ try grab the current version number from git"""
    version = "Undefined"
    if os.path.exists(".git"):
        try:
            version = os.popen("git describe --dirty --always").read().strip()
        except Exception as e:
            print(e)
    return version

APPNAME = 'gaia'
VERSION = open('VERSION', 'r').read().strip('\n')
GIT_SHA = get_git_version()

top = '.'
out = 'build'


def options(opt):
    opt.load('compiler_cxx compiler_c qt5')
    opt.recurse('src')

    # whether or not to have all the asserts working
    opt.add_option('--with-asserts', action = 'store_false',
                   dest = 'optimized', default = True,
                   help = 'whether to enable all asserts. Code can be much slower, but safer as well')

    # whether to use TBB for multithreading queries
    opt.add_option('--with-tbb', action = 'store_true',
                   dest = 'tbb', default = False,
                   help = 'whether to use TBB for multithreading queries')

    # whether to enable debug symbols and not optimize
    opt.add_option('--mode', action='store',
                   dest='MODE', default="release",
                   help='debug or release')

    opt.add_option('--cross-compile-mingw32', action='store_true',
                   dest='CROSS_COMPILE_MINGW32', default=False,
                   help='cross-compile for windows using mingw32 on linux')


def check_tbb(conf):
    tbb = conf.check_cxx(header_name = 'tbb/task_scheduler_init.h',
                         mandatory = True,
                         errmsg = 'Intel TBB cannot be found')

    if not tbb:
        return

    conf.env['CXXDEFINES_TBB'] = 'HAVE_TBB'
    conf.env['LIB_TBB'] = 'tbb'
    conf.env['USELIB'] += [ 'TBB' ]


def configure(conf):
    conf.load('compiler_cxx compiler_c qt5')
    #conf.load('compiler_cxx compiler_c qt4')
    
    if not sys.platform.startswith('linux') and sys.platform != 'darwin':
        print('Please use the QtCreator project for building Gaia in Windows...')
        sys.exit(1)

    print('- configuring the project in ' + conf.path.abspath())

    conf.env.APP_VERSION = VERSION

    # compiler flags
    conf.env.CXXFLAGS += [ '-std=c++11', '-Wall', '-fno-strict-aliasing', '-fPIC', '-fvisibility=hidden' ]

    # define this to be stricter, but sometimes some libraries can give problems...
    #conf.env.CXXFLAGS += [ '-Werror' ]

    if conf.options.MODE == 'debug':
        print('- Building in debug mode')
        conf.env.CXXFLAGS += [ '-g' ]

    elif conf.options.MODE == 'release':
        print('- Building in release mode')
        conf.env.CXXFLAGS += [ '-O2', '-msse2' ]
    else:
        raise ValueError('mode should be either "debug" or "release"')

    if conf.options.optimized:
        conf.env.CXXFLAGS += [ '-DNDEBUG', '-DQT_NO_DEBUG' ]

    # super optimized flags
    #conf.env['CXXFLAGS'] += [ '-Wall -Werror -O3 -fPIC -DNDEBUG -DQT_NO_DEBUG -ffast-math -fomit-frame-pointer -funroll-loops' ]

    conf.check_cfg(package='yaml-0.1', uselib_store='YAML',
                   args=['--cflags', '--libs'])

    conf.check_cfg(package='eigen3', uselib_store='EIGEN3',
                   args=['eigen3 >= 3.3.4', '--cflags'])

    conf.env['USELIB'] = [ 'QT5CORE', 'QT5CONCURRENT', 'YAML', 'EIGEN3' ]

    # optional dependency: tbb, if asked for it
    conf.env['WITH_TBB'] = conf.options.tbb
    if conf.env['WITH_TBB']:
        check_tbb(conf)

    conf.env.DEFINES = ['GAIA_VERSION="%s"' % VERSION, 'GAIA_GIT_SHA="%s"' % GIT_SHA]

    if sys.platform == 'darwin':
        # force the use of clang as compiler, we don't want gcc anymore on mac
        conf.env.CC = 'clang'
        conf.env.CXX = 'clang++'

        ###conf.env.DEFINES   += [ 'GTEST_HAS_TR1_TUPLE=0' ]
        conf.env.CXXFLAGS += [ '-stdlib=libc++', '-Wno-gnu', '-std=c++11']
        conf.env.LINKFLAGS = [ '-stdlib=libc++' ]
        # for defining static const variables in header
        conf.env.CXXFLAGS += [ '-Wno-static-float-init' ]

    if conf.options.CROSS_COMPILE_MINGW32:
        #print("- Cross-compiling for Windows with MinGW: search for pre-built dependencies in 'packaging/win32_3rdparty'")
        #os.environ["PKG_CONFIG_PATH"] = 'packaging/win32_3rdparty/lib/pkgconfig'
        #os.environ["PKG_CONFIG_LIBDIR"] = os.environ["PKG_CONFIG_PATH"]

        # locate MinGW compilers and use them
        conf.find_program('i686-w64-mingw32-gcc', var='CC')
        conf.find_program('i686-w64-mingw32-g++', var='CXX')
        conf.find_program('i686-w64-mingw32-ar', var='AR')

        # compile libgcc and libstd statically when using MinGW
        conf.env.CXXFLAGS = ['-static-libgcc', '-static-libstdc++']

    #conf.load('compiler_cxx compiler_c qt5')

    #conf.env['LINKFLAGS'] += [ '--as-needed' ] # TODO do we need this flag?

    # add this key otherwise gcc 4.8 will complain
    # conf.env['CXXFLAGS'] += [ '-Wno-unused-local-typedefs' ] #  --- outdated?

    # commented below as we don't care about centos for now:

    # big fat hack for centos, which is still in stone age...
    #if sys.version_info[1] > 4:
    #    # option not available in centos' gcc...
    #    conf.env['CXXFLAGS'] += [  '-Wno-unused-result' ]

    conf.recurse('src')

    if conf.options.tbb:
        tbblib = '-ltbb'
    else:
        tbblib = ''

    # write pkg-config file
    prefix = normpath(conf.options.prefix)

    if sys.platform.startswith('linux'):

        opts = { 'prefix': prefix,
             'qtlibdir': conf.env['LIB_QT5CORE'] or '/usr/lib',
             'qtincludedir': '-I' + ' -I'.join(conf.env['INCLUDES_QT5CORE']),
             'version': VERSION,
             'tbblib': tbblib,
             }

        pcfile = '''prefix=%(prefix)s
        libdir=${prefix}/lib
        includedir=${prefix}/include
        qtlibdir=%(qtlibdir)s
        qtincludes=%(qtincludedir)s

        Name: libgaia2
        Description: A library for doing similarity in semimetric spaces
        Version: %(version)s
        Libs: -L${libdir} -L${qtlibdir} -lgaia2 -lQt5Core -lQt5Concurrent -lyaml %(tbblib)s
        Cflags: -I${includedir} ${qtincludes}
        ''' % opts

    elif sys.platform == 'darwin':
        opts = { 'prefix': prefix,
             'qtlibdir': '-F' + conf.env['FRAMEWORKPATH_QT5CORE'][0] +
                         ' -framework ' + conf.env['FRAMEWORK_QT5CORE'][0] +
                         ' -framework ' + conf.env['FRAMEWORK_QT5CONCURRENT'][0],
             'qtincludedir': '-I' + ' -I'.join(conf.env['INCLUDES_QT5CORE']),
             'version': VERSION,
             'tbblib': tbblib,
             }

        pcfile = '''prefix=%(prefix)s
        libdir=${prefix}/lib
        includedir=${prefix}/include
        qtlibdir=%(qtlibdir)s
        qtincludes=%(qtincludedir)s

        Name: libgaia2
        Description: A library for doing similarity in semimetric spaces
        Version: %(version)s
        Requires.private: eigen3
        Libs: -L${libdir} ${qtlibdir} -lgaia2 -lyaml %(tbblib)s
        Cflags: -I${includedir} ${qtincludes}
        ''' % opts


    pcfile = '\n'.join([ l.strip() for l in pcfile.split('\n') ])
    conf.env.pcfile = pcfile
    #open('build/gaia2.pc', 'w').write(pcfile) # we'll do it later on the build stage


def build(bld):
    print('- building from ' + bld.path.abspath())
    bld.recurse('src')