# -*- python -*-

#

# Copyright (c) 2016 Stefan Seefeld

# All rights reserved.

#

# Distributed under the Boost Software License, Version 1.0.

# (See accompanying file LICENSE_1_0.txt or copy at

# http://www.boost.org/LICENSE_1_0.txt)

from faber.feature import set

from faber.types import cxx

from faber.tools.compiler import cxxflags, define, include

from faber.tools.python import python

from faber.config import report, cxx_checks

from faber.config.try_run import try_run

features += include('include')

features += define('BOOST_ALL_NO_LIB') # disable auto-linking

features += define('BOOST_NO_AUTO_PTR')

boost_include = options.get_with('boost-include')

if boost_include:

features += include(boost_include)

python = python.instance()

py_suffix = '{}{}'.format(*python.version.split('.')[:2])

features |= set(python.include, python.linkpath, python.libs)

class has_numpy(try_run):

src = r"""

// If defined, enforces linking against PythonXXd.lib, which

// is usually not included in Python environments.

#undef _DEBUG

#include "Python.h"

#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION

#include "numpy/arrayobject.h"

#if PY_VERSION_HEX >= 0x03000000

void *initialize() { import_array();}

#else

void initialize() { import_array();}

#endif

int main()

{

int result = 0;

Py_Initialize();

initialize();

if (PyErr_Occurred())

{

result = 1;

}

else

{

npy_intp dims = 2;

PyObject * a = PyArray_SimpleNew(1, &dims, NPY_INT);

if (!a) result = 1;

Py_DECREF(a);

}

Py_Finalize();

return result;

}

"""

def __init__(self, features=()):

inc = ''

try:

inc = python.check_python('import numpy; print(numpy.get_include())')

features |= include(inc)

except Exception:

# ignore errors, the check will fail during compilation...

pass

try_run.__init__(self, 'has_numpy', has_numpy.src, cxx, features,

if_=(include(inc), define('HAS_NUMPY')))

checks = [cxx_checks.has_cxx11(features, define('HAS_CXX11')),

has_numpy(features)]

config = report('config', checks)

src = module('src', features=features|config.use)

test = module('test', features=features|config.use)

doc = module('doc', features=features|config.use)

default = src.default