cmake_minimum_required(VERSION 3.10)

# Set the project name and version
project(TestJPEG VERSION 1.0)

# Specify C++ Standard
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED True)
set(CMAKE_BUILD_TYPE Debug)

# Jetson Multimedia API Path
set(JETSON_MULTIMEDIA_API_PATH "/usr/src/jetson_multimedia_api")

# DeepStream SDK Path
set(DEEPSTREAM_PATH "/opt/nvidia/deepstream/deepstream-6.3")

# Add include directories
include_directories(
    # /home/rekor/denczek/alprmodules/openalprgpu/src
    # /home/rekor/denczek/alprmodules/alprsupport
    /usr/include/opencv4
    # /home/rekor/denczek/alprmodules/openalprgpu/src/cuda/
    # /home/rekor/denczek/alprmodules/alprsupport/alprlog
    /usr/local/cuda-11.4/include
    ${JETSON_MULTIMEDIA_API_PATH}/include
    ${JETSON_MULTIMEDIA_API_PATH}/include/libjpeg-8b
)

# Add the executable
add_executable(test_jpeg
    NvSurfEncoder.cpp
    test_jpeg.cpp
    ${JETSON_MULTIMEDIA_API_PATH}/samples/common/classes/NvJpegEncoder.cpp
    ${JETSON_MULTIMEDIA_API_PATH}/samples/common/classes/NvUtils.cpp
    ${JETSON_MULTIMEDIA_API_PATH}/samples/common/classes/NvElementProfiler.cpp
    ${JETSON_MULTIMEDIA_API_PATH}/samples/common/classes/NvElement.cpp
    ${JETSON_MULTIMEDIA_API_PATH}/samples/common/classes/NvBuffer.cpp
    ${JETSON_MULTIMEDIA_API_PATH}/samples/common/classes/NvJpegEncoder.cpp
    ${JETSON_MULTIMEDIA_API_PATH}/samples/common/classes/NvLogging.cpp
    ${JETSON_MULTIMEDIA_API_PATH}/samples/common/classes/NvJpegDecoder.cpp
    ${JETSON_MULTIMEDIA_API_PATH}/samples/common/classes/NvBufSurface.cpp
)

find_package(CUDA)

if(CUDA_FOUND)
    include_directories(${CUDA_INCLUDE_DIRS})
endif()

# Link libraries
target_link_libraries(test_jpeg
    /usr/lib/aarch64-linux-gnu/libopencv_world.so
    /usr/lib/aarch64-linux-gnu/libboost_filesystem.so.1.71.0
    /usr/lib/aarch64-linux-gnu/libcuda.so
    /usr/lib/aarch64-linux-gnu/tegra/libnvjpeg.so
    ${DEEPSTREAM_PATH}/lib/libnvdsgst_meta.so
    ${DEEPSTREAM_PATH}/lib/libnvds_meta.so
    ${DEEPSTREAM_PATH}/lib/libnvbufsurface.so
    ${DEEPSTREAM_PATH}/lib/libnvbufsurftransform.so
    ${CUDA_LIBRARIES}
    /usr/local/cuda-11.4/lib64/libcudart_static.a
    -lpthread
    -ldl
    -lrt
    -lEGL
    -lGLESv2
    -lcuda
)