cmake_minimum_required(VERSION 2.8)
project(VisionCOre)

add_compile_options(-std=c++14)
add_compile_options(-g)

# setup OpenCV
find_package(OpenCV REQUIRED)

# setup ZED
option(LINK_SHARED_ZED "Link with the ZED SDK shared executable" ON)
if (NOT LINK_SHARED_ZED AND MSVC)
    message(FATAL_ERROR "LINK_SHARED_ZED OFF : ZED SDK static libraries not available on Windows")
endif()

find_package(ZED 2 REQUIRED)

# setup tensorRT flags
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
set(BUILD_DEPS "YES" CACHE BOOL "If YES, will install dependencies into sandbox.  Automatically reset to NO after dependencies are installed.")

# setup CUDA
find_package(CUDA)
message("-- CUDA version: ${CUDA_VERSION}")

set(
    CUDA_NVCC_FLAGS
    ${CUDA_NVCC_FLAGS};
    -O3
    -gencode arch=compute_53,code=sm_53
    -gencode arch=compute_62,code=sm_62
)

if(CUDA_VERSION_MAJOR GREATER 9)
    message("-- CUDA ${CUDA_VERSION_MAJOR} detected, enabling SM_72")

    set(
        CUDA_NVCC_FLAGS
        ${CUDA_NVCC_FLAGS};
        -gencode arch=compute_72,code=sm_72
    )
endif()

include_directories(
    src
    /usr/local/cuda/include
    ${ZED_INCLUDE_DIRS}
)

link_directories(${ZED_LIBRARY_DIR})
link_directories(${CUDA_LIBRARY_DIRS})

set(SOURCES
    src/Main.cpp
)

if (LINK_SHARED_ZED)
    SET(ZED_LIBS ${ZED_LIBRARIES} ${CUDA_CUDA_LIBRARY} ${CUDA_CUDART_LIBRARY} ${CUDA_NPP_LIBRARIES_ZED})
else()
    SET(ZED_LIBS ${ZED_STATIC_LIBRARIES} ${CUDA_CUDA_LIBRARY} ${CUDA_LIBRARY})
endif()

add_executable(Main ${SOURCES})
target_link_libraries(Main nvinfer nvinfer_plugin nvonnxparser
    pthread ${ZED_LIBS} ${OpenCV_LIBS})