# -*- python -*-

#

# Copyright (c) 2010-2012, Jim Bosch

# All rights reserved.

#

# ndarray is distributed under a simple BSD-like license;

# see the LICENSE file that should be present in the root

# of the source distribution, or alternately available at:

# https://github.com/ndarray/ndarray

#

import os

import sys

sys.path.insert(0, os.path.abspath('Boost.NumPy'))

from SConsChecks import AddLibOptions, GetLibChecks

setupOptions, makeEnvironment, setupTargets, checks, libs = SConscript("Boost.NumPy/SConscript")

libs.extend(['swig', 'eigen', 'boost.test', 'fftw', 'boost.preprocessor'])

checks = GetLibChecks(libs)

variables = setupOptions()

AddOption("--with-eigen", dest="eigen_prefix", type="string", nargs=1, action="store",

metavar="DIR", default=os.environ.get("EIGEN_DIR"),

help="prefix for Eigen libraries; should have an 'include' subdirectory")

AddOption("--with-eigen-include", dest="eigen_include", type="string", nargs=1, action="store",

metavar="DIR", help="location of Eigen header files")

AddOption("--with-fftw", dest="fftw_prefix", type="string", nargs=1, action="store",

metavar="DIR", default=os.environ.get("FFTW_DIR"),

help="prefix for FFTW libraries; should have 'include' and 'lib' subdirectories")

AddOption("--with-fftw-include", dest="fftw_include", type="string", nargs=1, action="store",

metavar="DIR", help="location of FFTW header files")

AddOption("--with-fftw-lib", dest="fftw_lib", type="string", nargs=1, action="store",

metavar="DIR", help="location of FFTW library")

building = not GetOption("help") and not GetOption("clean")

env = makeEnvironment(variables)

env.Tool('m4')

env.AppendUnique(CPPPATH=["#include"])

if building:

config = env.Configure(custom_tests=checks)

haveEigen = config.CheckEigen()

haveFFTW = config.CheckFFTW()

env = config.Finish()

else:

haveEigen = False

haveFFTW = False

env.haveEigen = haveEigen

env.haveFFTW = haveFFTW

testEnv = env.Clone()

if building:

config = testEnv.Configure(custom_tests=checks)

haveBoostTest = config.CheckBoostTest()

testEnv = config.Finish()

else:

haveBoostTest = False

testEnv.haveBoostTest = haveBoostTest

pyEnv = env.Clone()

if building:

config = pyEnv.Configure(custom_tests=checks)

havePython = config.CheckPython() and config.CheckNumPy()

haveSwig = havePython and config.CheckSwig()

if haveSwig:

pyEnv.AppendUnique(SWIGPATH = ["#include"])

config.CheckBoostPP()

pyEnv = config.Finish()

else:

havePython = False

haveSwig = False

pyEnv.havePython = havePython

pyEnv.haveSwig = haveSwig

bpEnv = pyEnv.Clone()

if building:

config = bpEnv.Configure(custom_tests=checks)

haveBoostPython = config.CheckBoostPython()

bpEnv = config.Finish()

else:

haveBoostPython = False

bpEnv.haveBoostPython = haveBoostPython

bpEnv.AppendUnique(CPPPATH=["#Boost.NumPy"])

if haveBoostPython:

setupTargets(bpEnv, root="Boost.NumPy")

elif building:

print("Not building Boost.NumPy component.")

headers = SConscript(os.path.join("include", "SConscript"), exports="env")

prefix = Dir(GetOption("prefix")).abspath

install_headers = GetOption("install_headers")

if not install_headers:

install_headers = os.path.join(prefix, "include")

for header in Flatten(headers):

relative = os.path.relpath(header.abspath, Dir("#include").abspath)

env.Alias("install", env.InstallAs(os.path.join(install_headers, relative), header))

tests = SConscript(os.path.join("tests", "SConscript"), exports=["env", "testEnv", "pyEnv", "bpEnv"])