face-sdk-python-api/CMakeLists.txt at main · 3DiVi/face-sdk-python-api · 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
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
cmake_minimum_required(VERSION 3.27.0)

set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(PYBIND11_TAG v3.0.0)

if (UNIX)
    set(PYTHON_EXECUTABLE /usr/bin/python3 CACHE STRING "Path to Python executable")
else()
    set(PYBIND11_FINDPYTHON OFF)

    find_package(Python REQUIRED)

    set(PYTHON_EXECUTABLE ${Python_EXECUTABLE} CACHE STRING "Path to Python executable")
endif()
set(FACE_SDK_INCLUDE_PATH "FaceSDK/include" CACHE STRING "Path to include directory")

if (NOT DEFINED ${PYTHON_MINOR_VERSION})
    execute_process(
        COMMAND ${PYTHON_EXECUTABLE} --version
        OUTPUT_VARIABLE PYTHON_FULL_VERSION
    )

    message(STATUS "Full Python version: ${PYTHON_FULL_VERSION}")

    string(REGEX REPLACE "Python [0-9]+\\.([0-9]+).*" "\\1" PYTHON_MINOR_VERSION "${PYTHON_FULL_VERSION}")
endif()

message(STATUS "Python minor version: ${PYTHON_MINOR_VERSION}")

project(face_sdk_3divi)

include(FetchContent)

FetchContent_Declare(
    pybind11
    GIT_REPOSITORY https://github.com/pybind/pybind11
    GIT_TAG ${PYBIND11_TAG}
    OVERRIDE_FIND_PACKAGE
)

FetchContent_MakeAvailable(pybind11)

find_package(pybind11 REQUIRED)

pybind11_add_module(
    ${PROJECT_NAME}
    src/module.cpp
    src/FacerecService.cpp
    src/Context.cpp
    src/ProcessingBlock.cpp
    src/ContextTemplate.cpp
    src/DynamicTemplateIndex.cpp
)

target_include_directories(
    ${PROJECT_NAME} PRIVATE
    include
    ${FACE_SDK_INCLUDE_PATH}
)

if (DEFINED CUSTOM_INSTALL_PATH)
    install(TARGETS ${PROJECT_NAME} DESTINATION ${CUSTOM_INSTALL_PATH})
else()
    if (UNIX)
        add_custom_target(
            generate_pyi ALL
            DEPENDS ${PROJECT_NAME}
            WORKING_DIRECTORY ${CMAKE_BINARY_DIR}

            COMMAND ${PYTHON_EXECUTABLE} -m venv .venv
            COMMAND .venv/bin/python -m pip install pybind11-stubgen numpy
            COMMAND ${CMAKE_COMMAND} -E copy ${CMAKE_BINARY_DIR}/*.so ${CMAKE_BINARY_DIR}/.venv/lib/python3.${PYTHON_MINOR_VERSION}/site-packages
            COMMAND ${CMAKE_BINARY_DIR}/.venv/bin/pybind11-stubgen -o ${CMAKE_BINARY_DIR} ${PROJECT_NAME}
            COMMAND ${CMAKE_COMMAND} -E rm -rf .venv
        )
    else()
        add_custom_target(
            generate_pyi ALL
            DEPENDS ${PROJECT_NAME}
            WORKING_DIRECTORY ${CMAKE_BINARY_DIR}

            COMMAND ${PYTHON_EXECUTABLE} -m venv .venv
            COMMAND .venv/Scripts/python -m pip install pybind11-stubgen numpy
            COMMAND ${CMAKE_COMMAND} -E copy $<TARGET_FILE:${PROJECT_NAME}> ${CMAKE_BINARY_DIR}/.venv/Lib/site-packages
            COMMAND ${CMAKE_BINARY_DIR}/.venv/Scripts/pybind11-stubgen -o ${CMAKE_BINARY_DIR} ${PROJECT_NAME}
            COMMAND ${CMAKE_COMMAND} -E rm -rf .venv
        )
    endif()

    install(TARGETS ${PROJECT_NAME} DESTINATION .)
    install(FILES ${CMAKE_BINARY_DIR}/${PROJECT_NAME}.pyi DESTINATION .)
endif()