CMakeLists.txt

cmake_minimum_required(VERSION 3.16)
project("FaceTransform" LANGUAGES CXX CUDA)


set(CMAKE_CXX_STANDARD 17)
find_package(CUDA REQUIRED)

IF (CUDA_VERSION VERSION_LESS "11.0")
    set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode arch=compute_30,code=sm_30")
ENDIF()

set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode arch=compute_35,code=sm_35")

set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode arch=compute_50,code=sm_50")
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode arch=compute_52,code=sm_52")

IF (CUDA_VERSION VERSION_GREATER "7.6")
    set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode arch=compute_60,code=sm_60")
    set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode arch=compute_61,code=sm_61")
    set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode arch=compute_62,code=sm_62")
ENDIF()

IF ((CUDA_VERSION VERSION_GREATER "9.0") OR (CUDA_VERSION VERSION_EQUAL "9.0"))
    set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode arch=compute_70,code=sm_70")
ENDIF()

IF ((CUDA_VERSION VERSION_GREATER "10.0") OR (CUDA_VERSION VERSION_EQUAL "10.0"))
    set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode arch=compute_75,code=sm_75")
ENDIF()

IF ((CUDA_VERSION VERSION_GREATER "11.0") OR (CUDA_VERSION VERSION_EQUAL "11.0"))
    set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode arch=compute_80,code=sm_80")
ENDIF()


# THIS LINE SHOULD BE CHANGE DEPENDING ON THE ARCHITECTURE OF THE GPU YOU INTEND TO USE
#set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode arch=compute_50,code=sm_50")

set(CMAKE_BUILD_TYPE "Release")
set(MODULE_SOURCE "${PROJECT_SOURCE_DIR}/src/")
set(EXTERNALS "${PROJECT_SOURCE_DIR}/ext")

set(LIB_DIR "../lib")
SET(LIBRARY_OUTPUT_PATH ${LIB_DIR}/)
# pybind11
add_subdirectory(${EXTERNALS}/pybind11)

#OpenCv
set(OpenCV_DIR ${EXTERNALS}/opencv2/)

# Homography library
add_library(
    utils_lib STATIC
    ${MODULE_SOURCE}/utils/utils.cpp
    ${MODULE_SOURCE}/utils/cuda_func.cu
    ${MODULE_SOURCE}/utils/homographie.cpp
)

target_include_directories(
    utils_lib
    PRIVATE "${OpenCV_DIR}/include"
)
target_link_libraries(utils_lib 
    PRIVATE ${OpenCV_DIR}/opencv_world455.lib)

# Copy needed dll in output folder
add_custom_command(TARGET utils_lib POST_BUILD
COMMAND ${CMAKE_COMMAND} -E copy_if_different
    ${OpenCV_DIR}/opencv_world455.dll
    $<TARGET_FILE_DIR:utils_lib>)


#pybind module
pybind11_add_module( 
    FaceTransform MODULE
    # FaceSwap Sources
    ${MODULE_SOURCE}/faceswap/faceswap_func.cpp
    # Bigger Face
    ${MODULE_SOURCE}/biggerface/biggerface.cpp
    # Add beard 
    ${MODULE_SOURCE}/beard/Image.cpp
    ${MODULE_SOURCE}/beard/addBeard_func.cpp
    #Wrapper
    ${MODULE_SOURCE}/wrapper.cpp
)

target_link_libraries(FaceTransform 
    PRIVATE utils_lib
)

target_include_directories (
    FaceTransform
    PRIVATE ${MODULE_SOURCE}/faceswap
    PRIVATE ${MODULE_SOURCE}/biggerface
    PRIVATE ${MODULE_SOURCE}/beard
    PRIVATE ${MODULE_SOURCE}/utils
)