unset(NUMPY_VERSION)

unset(NUMPY_INCLUDE_DIR)

if(PYTHONINTERP_FOUND)

execute_process(COMMAND "${PYTHON_EXECUTABLE}" "-c"

"import numpy as n; print(n.__version__); print(n.get_include());"

RESULT_VARIABLE __result

OUTPUT_VARIABLE __output

OUTPUT_STRIP_TRAILING_WHITESPACE)

if(__result MATCHES 0)

string(REGEX REPLACE ";" "\\\\;" __values ${__output})

string(REGEX REPLACE "\r?\n" ";" __values ${__values})

list(GET __values 0 NUMPY_VERSION)

list(GET __values 1 NUMPY_INCLUDE_DIR)

string(REGEX MATCH "^([0-9])+\\.([0-9])+\\.([0-9])+" __ver_check "${NUMPY_VERSION}")

if(NOT "${__ver_check}" STREQUAL "")

set(NUMPY_VERSION_MAJOR ${CMAKE_MATCH_1})

set(NUMPY_VERSION_MINOR ${CMAKE_MATCH_2})

set(NUMPY_VERSION_PATCH ${CMAKE_MATCH_3})

math(EXPR NUMPY_VERSION_DECIMAL

"(${NUMPY_VERSION_MAJOR} * 10000) + (${NUMPY_VERSION_MINOR} * 100) + ${NUMPY_VERSION_PATCH}")

string(REGEX REPLACE "\\\\" "/" NUMPY_INCLUDE_DIR ${NUMPY_INCLUDE_DIR})

else()

unset(NUMPY_VERSION)

unset(NUMPY_INCLUDE_DIR)

message(STATUS "Requested NumPy version and include path, but got instead:\n${__output}\n")

endif()

endif()

else()

message(STATUS "To find NumPy Python interpretator is required to be found.")

endif()

include(FindPackageHandleStandardArgs)

find_package_handle_standard_args(NumPy REQUIRED_VARS NUMPY_INCLUDE_DIR NUMPY_VERSION

VERSION_VAR NUMPY_VERSION)

if(NUMPY_FOUND)

message(STATUS "NumPy ver. ${NUMPY_VERSION} found (include: ${NUMPY_INCLUDE_DIR})")

endif()

caffe_clear_vars(__result __output __error_value __values __ver_check __error_value)