-
Notifications
You must be signed in to change notification settings - Fork 202
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Adjust NumPy code to work with new directory / namespace structure.
- Loading branch information
1 parent
cbb3851
commit 8ccdcff
Showing
51 changed files
with
1,195 additions
and
480 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,86 @@ | ||
| # | ||
| # 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 . import ui | ||
| from contextlib import contextmanager | ||
|
|
||
| @contextmanager | ||
| def saved(context): | ||
| save_cpppath = context.env.get('CPPPATH', []) | ||
| save_libs = context.env.get('LIBS', []) | ||
| yield context | ||
| context.env.Replace(LIBS=save_libs) | ||
| context.env.Replace(CPPPATH=save_cpppath) | ||
|
|
||
|
|
||
| def add_options(vars): | ||
|
|
||
| pass | ||
|
|
||
|
|
||
| def check(context): | ||
|
|
||
| numpy_source_file = r""" | ||
| // If defined, enforces linking againg PythonXXd.lib, which | ||
| // is usually not included in Python environments. | ||
| #undef _DEBUG | ||
| #include "Python.h" | ||
| #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; | ||
| } | ||
| """ | ||
|
|
||
| import platform | ||
| import subprocess | ||
| import re, os | ||
|
|
||
| def check_python(cmd): | ||
| try: | ||
| return True, subprocess.check_output([python, '-c', cmd]).strip() | ||
| except subprocess.CalledProcessError as e: | ||
| return False, e | ||
|
|
||
| context.Message('Checking for NumPy...') | ||
| with saved(context): | ||
| python = context.env['PYTHON'] | ||
| result, numpy_incpath = check_python('import numpy; print(numpy.get_include())') | ||
| if result: | ||
| context.env.AppendUnique(CPPPATH=numpy_incpath) | ||
| context.env.AppendUnique(LIBS=context.env['PYTHONLIBS']) | ||
| result, output = context.TryRun(numpy_source_file,'.cpp') | ||
| if not result: | ||
| context.Result(0) | ||
| return False | ||
| context.env['NUMPY'] = True | ||
| context.env['NUMPY_CPPPATH'] = numpy_incpath | ||
| context.Result(1) | ||
| return True |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.