cmake_minimum_required(VERSION 3.10)
project(${SKBUILD_PROJECT_NAME} VERSION ${SKBUILD_PROJECT_VERSION})
option(GENERATE_STUBS "Whether to generate stubs" ON)
option(CCACHE_ENABLED "Whether to enable compiler caching, if available" ON)

set(CMAKE_POSITION_INDEPENDENT_CODE ON)
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_CUDA_STANDARD 14)
set(CMAKE_CUDA_STANDARD_REQUIRED ON)

if(CMAKE_CXX_COMPILER_ID STREQUAL "MSVC")
    # Some fixes for the Glog library.
    add_compile_definitions(GLOG_NO_ABBREVIATED_SEVERITIES)
    add_compile_definitions(GL_GLEXT_PROTOTYPES)
    add_compile_definitions(NOMINMAX)
    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /EHsc")
    # Enable object level parallel builds in Visual Studio.
    set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} /MP")
    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /MP")
endif()

find_package(colmap REQUIRED)

if(CCACHE_ENABLED)
    find_program(CCACHE ccache)
    if(CCACHE)
        message(STATUS "Enabling ccache support")
        set(CMAKE_C_COMPILER_LAUNCHER ${CCACHE})
        set(CMAKE_CXX_COMPILER_LAUNCHER ${CCACHE})
        if(CUDA_ENABLED)
            set(CMAKE_CUDA_COMPILER_LAUNCHER ${CCACHE})
        endif()
    else()
        message(STATUS "Disabling ccache support")
    endif()
else()
    message(STATUS "Disabling ccache support")
endif()

# If colmap was built with ONNX support, we need to set the rpath so the
# Python module can find the ONNX Runtime library at runtime.
set(ONNX_RPATH "")
if(DEFINED onnxruntime_LIBRARY_DIR_HINTS AND EXISTS "${onnxruntime_LIBRARY_DIR_HINTS}")
    message(STATUS "ONNX Runtime library directory: ${onnxruntime_LIBRARY_DIR_HINTS}")
    set(ONNX_RPATH "${onnxruntime_LIBRARY_DIR_HINTS}")
elseif(TARGET onnxruntime::onnxruntime)
    # Derive the library directory from the imported target.
    get_target_property(_ort_type onnxruntime::onnxruntime TYPE)
    if(_ort_type STREQUAL "INTERFACE_LIBRARY")
        # Find module creates an INTERFACE target with link libraries.
        get_target_property(_ort_libs onnxruntime::onnxruntime INTERFACE_LINK_LIBRARIES)
        list(GET _ort_libs 0 _ort_location)
    else()
        get_target_property(_ort_location onnxruntime::onnxruntime IMPORTED_LOCATION)
        if(NOT _ort_location)
            get_target_property(_ort_location onnxruntime::onnxruntime IMPORTED_LOCATION_RELEASE)
        endif()
    endif()
    if(_ort_location)
        get_filename_component(ONNX_RPATH "${_ort_location}" DIRECTORY)
        message(STATUS "ONNX Runtime library directory (from target): ${ONNX_RPATH}")
    endif()
endif()

if (CMAKE_VERSION VERSION_LESS 3.18)
    set(DEV_MODULE Development)
else()
    set(DEV_MODULE Development.Module)
endif()
find_package(Python REQUIRED COMPONENTS Interpreter ${DEV_MODULE} REQUIRED)

find_package(pybind11 2.13.0 REQUIRED)

file(GLOB_RECURSE SOURCE_FILES "${PROJECT_SOURCE_DIR}/../src/pycolmap/*.cc")
pybind11_add_module(_core ${SOURCE_FILES})
target_include_directories(_core PRIVATE ${PROJECT_SOURCE_DIR}/../src/)
target_link_libraries(_core PRIVATE colmap::colmap glog::glog Ceres::ceres)
target_compile_definitions(_core PRIVATE VERSION_INFO="${PROJECT_VERSION}")

# Set rpath to find ONNX Runtime library at runtime.
if(ONNX_RPATH)
    if(APPLE)
        set_target_properties(_core PROPERTIES
            BUILD_RPATH "${ONNX_RPATH}"
            INSTALL_RPATH "${ONNX_RPATH}"
        )
    elseif(UNIX)
        set_target_properties(_core PROPERTIES
            BUILD_RPATH "${ONNX_RPATH}"
            INSTALL_RPATH "$ORIGIN/../lib:${ONNX_RPATH}"
        )
    endif()
endif()

install(TARGETS _core LIBRARY DESTINATION pycolmap)

if(GENERATE_STUBS AND UNIX)
    message(STATUS "Enabling stubs generation")
    set(STUBGEN_OUTPUT_DIR "${CMAKE_CURRENT_BINARY_DIR}/_core")
    # Set library path for stub generation to find ONNX Runtime.
    if(APPLE AND ONNX_RPATH)
        set(STUBGEN_LIB_PATH_VAR "DYLD_LIBRARY_PATH=${ONNX_RPATH}:$ENV{DYLD_LIBRARY_PATH}")
    elseif(ONNX_RPATH)
        set(STUBGEN_LIB_PATH_VAR "LD_LIBRARY_PATH=${ONNX_RPATH}:$ENV{LD_LIBRARY_PATH}")
    else()
        set(STUBGEN_LIB_PATH_VAR "")
    endif()
    add_custom_command(
        TARGET _core POST_BUILD
        COMMAND
          "${CMAKE_COMMAND}" -E env
          "PYTHONPATH=$<TARGET_FILE_DIR:_core>:$ENV{PYTHONPATH}"
          ${STUBGEN_LIB_PATH_VAR}
          bash ${PROJECT_SOURCE_DIR}/generate_stubs.sh "${Python_EXECUTABLE}" "${CMAKE_CURRENT_BINARY_DIR}"
        WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}
        COMMENT "Generating pybind11 stubs"
        VERBATIM
    )
    install(DIRECTORY ${STUBGEN_OUTPUT_DIR} DESTINATION pycolmap)
endif()
