-
Notifications
You must be signed in to change notification settings - Fork 71
/
ChoosePython.cmake
105 lines (95 loc) · 3.8 KB
/
ChoosePython.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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
##
## Copyright (c) 2012-2015 Cajal Blue Brain, BBP/EPFL
## All rights reserved. Do not distribute without permission.
##
## Responsible Author: Juan Hernando Vieites (JHV)
## contact: jhernando@fi.upm.es
# This script provides an option to choose which Python version should be
# requested to FindPythonInterp and FindPythonLibs. The user can choose
# between version 2, 3 or autodetection (the default). If
# CHOOSE_PYTHON_IGNORE_BOOST is set, autodetection will choose the Python
# version without checking if Boost.Python is available and which version it
# supports.
#
# Input Variables:
# * CHOOSE_PYTHON_IGNORE_BOOST
#
# Output Variables:
# * USE_PYTHON3: Set to 1 if and only if Python 3 is chosen.
# * USE_BOOST_PYTHON_VERSION: Equals 3 if Python 3 is chosen, empty string
# otherwise
# * PYTHON_LIBRARY_SUFFIX: The suffix path where Python site packages are
# to be installed for the chosen Python version.
# * PythonLibs_FIND_VERSION
# PythonInter_FIND_VERSION: Variables used by Python find scripts as input.
# Set to 2 or 3 depending on the version chosen.
#
# Input/output varaibles:
# * USE_PYTHON_VERSION: (input/output), the Python version chosen. At return it
# can only be 2 or 3.
#
# Defines:
# * -DUSE_PYTHON3=1: If Python 3 is chosen.
if(CHOOSE_PYTHON_DONE)
return()
endif()
set(CHOOSE_PYTHON_DONE ON)
include(FindBoostConfig) # Including the workarounds for boost finders.
set(USE_PYTHON_VERSION "auto" CACHE STRING
"Choose the required Python version (2, 3 or auto).")
set_property(CACHE USE_PYTHON_VERSION PROPERTY STRINGS 2 3 auto)
if(PYTHONINTERP_FOUND)
message(WARNING "This script must be included before trying to find Python.")
return()
endif()
if(${USE_PYTHON_VERSION} STREQUAL auto)
# Finding Boost first if needed because if the Python3 interpreter is found
# first there's no way back.
if(NOT CHOOSE_PYTHON_IGNORE_BOOST)
foreach(__suffix 3 38 -py38 37 -py37 36 -py36 -py35 -py34 )
find_package(Boost COMPONENTS python${__suffix} QUIET)
if(Boost_FOUND)
string(TOUPPER ${__suffix} __boost_python_library_suffix)
break()
endif()
endforeach()
endif()
if(Boost_FOUND OR CHOOSE_PYTHON_IGNORE_BOOST)
find_package(PythonInterp 3 QUIET)
endif()
if(PYTHONINTERP_FOUND AND (CHOOSE_PYTHON_IGNORE_BOOST OR Boost_FOUND))
set(USE_PYTHON_VERSION 3)
else()
set(USE_PYTHON_VERSION 2)
endif()
endif()
if(${USE_PYTHON_VERSION} STREQUAL 3)
set(USE_PYTHON3 ON)
# Enforcing a Python version to be searched by scripts included by
# Common.cmake that search for Python (e.g. cpplint)
set(PYTHON_ADDITIONAL_VERSIONS 3.9 3.8 3.7 3.6 3.5 3.4 3.3 3.2)
set(Python_ADDITIONAL_VERSIONS 3.9 3.8 3.7 3.6 3.5 3.4 3.3 3.2)
set(PythonLibs_FIND_VERSION 3)
add_definitions(-DUSE_PYTHON3=1)
set(USE_BOOST_PYTHON_VERSION ${__boost_python_library_suffix})
# This shouldn't be necessary but helps detecting the Python libs
# provided by the module python/3.2-rhel6-x86_64
if(DEFINED ENV{PYTHON_LIBRARY})
# find_path is called twice using these variables as input and output.
# If the variables are not set as CACHE variables the second call to
# file_path fails for no apparent reason except that the variables have
# been marked as advanced by FindPythonLibs.cmake
set(PYTHON_LIBRARY $ENV{PYTHON_LIBRARY} CACHE FILEPATH "")
set(PYTHON_INCLUDE_DIR $ENV{PYTHON_INCLUDE_DIR} CACHE PATH "")
endif()
else()
set(PythonLibs_FIND_VERSION 2)
endif()
if(NOT PYTHON_EXECUTABLE)
# Regardless of auto-detection, now we need to find the interpreter to
# query the library suffix.
find_package(PythonInterp ${USE_PYTHON_VERSION} QUIET)
endif()
execute_process(COMMAND
${PYTHON_EXECUTABLE} -c "from distutils.sysconfig import get_python_lib; print(get_python_lib(1,0,prefix=''))"
OUTPUT_VARIABLE PYTHON_LIBRARY_SUFFIX OUTPUT_STRIP_TRAILING_WHITESPACE)