forked from rbdl/rbdl
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathFindNumPy2.cmake
34 lines (28 loc) · 1.17 KB
/
FindNumPy2.cmake
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
# Find the Python NumPy package
# PYTHON_NUMPY_2_INCLUDE_DIR
# PYTHON_NUMPY_2_FOUND
# will be set by this script
#cmake_minimum_required(VERSION 2.6)
set(PYTHON_EXECUTABLE "python2")
if (PYTHON_EXECUTABLE)
# Find out the include path
execute_process(
COMMAND "${PYTHON_EXECUTABLE}" -c
"from __future__ import print_function\ntry: import numpy; print(numpy.get_include(), end='')\nexcept:pass\n"
OUTPUT_VARIABLE __numpy_path)
# And the version
execute_process(
COMMAND "${PYTHON_EXECUTABLE}" -c
"from __future__ import print_function\ntry: import numpy; print(numpy.__version__, end='')\nexcept:pass\n"
OUTPUT_VARIABLE __numpy_version)
elseif(__numpy_out)
message(STATUS "Python executable not found.")
endif(PYTHON_EXECUTABLE)
find_path(PYTHON_NUMPY_2_INCLUDE_DIR numpy/arrayobject.h
HINTS "${__numpy_path}" "${PYTHON_INCLUDE_PATH}" NO_DEFAULT_PATH)
if(PYTHON_NUMPY_2_INCLUDE_DIR)
set(PYTHON_NUMPY_2_FOUND 1 CACHE INTERNAL "Python numpy found")
endif(PYTHON_NUMPY_2_INCLUDE_DIR)
include(FindPackageHandleStandardArgs)
find_package_handle_standard_args(NumPy REQUIRED_VARS PYTHON_NUMPY_2_INCLUDE_DIR
VERSION_VAR __numpy_version)