cpp_math/CMakeLists.txt at main · vincent-pollet/cpp_math · GitHub

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
cmake_minimum_required(VERSION 3.15)
cmake_policy(SET CMP0190 NEW)

project(cpp_math)

option(STUBS_ONLY "Only generate stubs (module already installed)" OFF)

# ── Cross-compilation ─────────────────────────────────────────────────────────
if(CMAKE_CROSSCOMPILING AND DEFINED ENV{PREFIX})
	message(STATUS "Cross-compiling, detecting Python from sysroot…")

	set(nanobind_ROOT        "$ENV{PREFIX}/lib/python$ENV{PY_VER}/site-packages/nanobind/cmake")
	set(PYTHON_SITE_PACKAGES "$ENV{PREFIX}/lib/python$ENV{PY_VER}/site-packages")

	find_package(Python $ENV{PY_VER} EXACT COMPONENTS Development.Module REQUIRED)

elseif(CMAKE_CROSSCOMPILING)
	message(FATAL_ERROR "Cross-compiling is not available when building without pixi.")

else()
	find_package(Python $ENV{PY_VER} EXACT COMPONENTS Interpreter Development.Module REQUIRED)
	execute_process(
				COMMAND "${Python_EXECUTABLE}" -m nanobind --cmake_dir
				OUTPUT_STRIP_TRAILING_WHITESPACE OUTPUT_VARIABLE nanobind_ROOT
		)
	execute_process(
				COMMAND "${Python_EXECUTABLE}" -c
						"import sysconfig; print(sysconfig.get_path('purelib'))"
				OUTPUT_VARIABLE PYTHON_SITE_PACKAGES
				OUTPUT_STRIP_TRAILING_WHITESPACE
		)
endif()
# ─────────────────────────────────────────────────────────────────────────────

find_package(nanobind CONFIG REQUIRED)

# ── Compiled extension ────────────────────────────────────────────────────────
if(NOT STUBS_ONLY)
	nanobind_add_module(cpp_math src/math.cpp)

	install(
				TARGETS cpp_math
				LIBRARY DESTINATION ${PYTHON_SITE_PACKAGES}/cpp_math
				ARCHIVE DESTINATION ${PYTHON_SITE_PACKAGES}/cpp_math
		)
endif()

# ── Stubs ─────────────────────────────────────────────────────────────────────
if(STUBS_ONLY)
	# The .so is assumed already installed.
	nanobind_add_stub(
				cpp_math_stub
				MODULE    cpp_math
				RECURSIVE
				OUTPUT    cpp_math.pyi
				MARKER_FILE py.typed
				OUTPUT_PATH ${PYTHON_SITE_PACKAGES}/cpp_math
				PYTHON_PATH ${PYTHON_SITE_PACKAGES}/cpp_math
		)
endif()