# ==============================================================================
# Install the library
# ==============================================================================

# for CMAKE_INSTALL_INCLUDEDIR and others definition
include(GNUInstallDirs)

# include the public headers of the anira library for the install target
# TODO: File PR so that the concurrentqueue library does not have to be included that way
target_include_directories(${PROJECT_NAME}
    PUBLIC
    $<INSTALL_INTERFACE:${CMAKE_INSTALL_INCLUDEDIR}>
    $<INSTALL_INTERFACE:${CMAKE_INSTALL_INCLUDEDIR}/concurrentqueue/moodycamel>
)

# This will be problematic anyway since symbols are not found when linking the torch libraries privately
# Therefore, the install target will not work properly
if (TORCH_LIBRARIES_ALL_PRIVATE)
    target_include_directories(${PROJECT_NAME}
        PUBLIC
        $<INSTALL_INTERFACE:${CMAKE_INSTALL_INCLUDEDIR}/torch/csrc/api/include>
    )
endif()

target_link_directories(${PROJECT_NAME} PUBLIC
    $<INSTALL_INTERFACE:${CMAKE_INSTALL_LIBDIR}>
)

# define the dircetory where the library will be installed CMAKE_INSTALL_PREFIX
if(DEFINED CMAKE_INSTALL_PREFIX_INITIALIZED_TO_DEFAULT)
    message( STATUS "CMAKE_INSTALL_PREFIX will be set to ${CMAKE_CURRENT_BINARY_DIR}/${PROJECT_NAME}-${PROJECT_VERSION}" )
    set(CMAKE_INSTALL_PREFIX "${CMAKE_CURRENT_BINARY_DIR}/${PROJECT_NAME}-${PROJECT_VERSION}" CACHE PATH "Where the library will be installed to" FORCE)
else()
    message(STATUS "CMAKE_INSTALL_PREFIX was already set to ${CMAKE_INSTALL_PREFIX}")
endif()

# at install the rpath is cleared by default so we have to set it again for the installed shared library to find the other libraries
# in this case we set the rpath to the directories where the other libraries are installed
# $ORIGIN in Linux is a special token that gets replaced by the directory of the library at runtime from that point we could navigate to the other libraries
# The same token for macOS is @loader_path
if(UNIX AND NOT APPLE)
    set_target_properties(${PROJECT_NAME}
        PROPERTIES
            INSTALL_RPATH "$ORIGIN"
    )
    # This is necessary for the gtest_main library to find the gtest library at runtime
    if (ANIRA_WITH_BENCHMARK)
    set_target_properties(gtest_main
        PROPERTIES
            INSTALL_RPATH "$ORIGIN"
    )
    endif()
elseif(APPLE)
    set(OSX_RPATHS "@loader_path")
    if (CMAKE_SYSTEM_PROCESSOR STREQUAL "x86_64")
        list(APPEND OSX_RPATHS "/opt/intel/oneapi/mkl/latest/lib")
    elseif (CMAKE_SYSTEM_PROCESSOR STREQUAL "arm64")
    endif()
    set_target_properties(${PROJECT_NAME}
        PROPERTIES
            INSTALL_RPATH "${OSX_RPATHS}"
    )
    if (ANIRA_WITH_BENCHMARK)
    set_target_properties(gtest_main
        PROPERTIES
            INSTALL_RPATH "@loader_path"
    )
    endif()
endif()


# the variant with PUBLIC_HEADER property unfortunately does not preserve the folder structure therefore we use the simple install directory command
install(DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/include/anira
    DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}
    COMPONENT dev
)

# install the target and create export-set
install(TARGETS ${PROJECT_NAME} concurrentqueue nlohmann_json
    EXPORT "aniraTargets"
    # these get default values from GNUInstallDirs
    RUNTIME DESTINATION ${CMAKE_INSTALL_LIBDIR} # .dll files 
    COMPONENT runtime
    LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR} # .so or .dylib files
    COMPONENT runtime NAMELINK_COMPONENT dev
    ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR} # .lib files
    COMPONENT dev
)

# libtorch has cmake config files that we can use to install the library later with find_package and then just link to it
if(ANIRA_WITH_LIBTORCH)
    install(DIRECTORY "${LIBTORCH_ROOTDIR}/include/"
        DESTINATION "${CMAKE_INSTALL_INCLUDEDIR}"
        COMPONENT deps-backends
    )
    install(DIRECTORY "${LIBTORCH_ROOTDIR}/lib/"
        DESTINATION "${CMAKE_INSTALL_LIBDIR}"
        COMPONENT deps-backends
    )
    install(DIRECTORY "${LIBTORCH_ROOTDIR}/share/cmake/"
        DESTINATION "${CMAKE_INSTALL_LIBDIR}/cmake"
        COMPONENT deps-backends
    )
endif()

# the other ones don't have cmake config files so we have to install them manually
if(ANIRA_WITH_ONNXRUNTIME)
    if(UNIX AND NOT APPLE AND CMAKE_SYSTEM_PROCESSOR STREQUAL "armv7l")
        install(DIRECTORY "${ONNXRUNTIME_ROOTDIR}/include/onnxruntime/"
            DESTINATION "${CMAKE_INSTALL_INCLUDEDIR}"
            COMPONENT deps-backends
        )
    else()
        install(DIRECTORY "${ONNXRUNTIME_ROOTDIR}/include/"
            DESTINATION "${CMAKE_INSTALL_INCLUDEDIR}"
            COMPONENT deps-backends
        )
    endif()
    install(DIRECTORY "${ONNXRUNTIME_ROOTDIR}/lib/"
        DESTINATION "${CMAKE_INSTALL_LIBDIR}"
        COMPONENT deps-backends
    )
endif()

if(ANIRA_WITH_TFLITE)
    install(DIRECTORY "${TENSORFLOWLITE_ROOTDIR}/include/"
        DESTINATION "${CMAKE_INSTALL_INCLUDEDIR}"
        COMPONENT deps-backends
    )
    install(DIRECTORY "${TENSORFLOWLITE_ROOTDIR}/lib/"
        DESTINATION "${CMAKE_INSTALL_LIBDIR}"
        COMPONENT deps-backends
    )
endif()

# ==============================================================================
# Generate cmake config files
# ==============================================================================

# generate and install export file in the folder cmake with the name of the project and namespace
# this generates files called aniraTargets.cmake, aniraTargets-debug.cmake, aniraTargets-release.cmake
install(EXPORT "aniraTargets"
    NAMESPACE ${PROJECT_NAME}::
    DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/${PROJECT_NAME}
    COMPONENT dev
)

include(CMakePackageConfigHelpers)

# create config file from the template file Config.cmake.in and specify the install destination
configure_package_config_file(${CMAKE_CURRENT_SOURCE_DIR}/Config.cmake.in
    "${CMAKE_CURRENT_BINARY_DIR}/aniraConfig.cmake"
    INSTALL_DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/${PROJECT_NAME}
)

# generate the version file for the config file
write_basic_package_version_file(
    "${CMAKE_CURRENT_BINARY_DIR}/aniraConfigVersion.cmake"
    VERSION "${PROJECT_VERSION}"
    COMPATIBILITY AnyNewerVersion
)

# install config files
install(FILES
    "${CMAKE_CURRENT_BINARY_DIR}/aniraConfig.cmake"
    "${CMAKE_CURRENT_BINARY_DIR}/aniraConfigVersion.cmake"
    DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/${PROJECT_NAME}
    COMPONENT dev
)