cmake_minimum_required(VERSION 3.10)

# Set the project name in a variable
set(project_name yolov10_cpp)
project(${project_name})
set(CMAKE_CXX_STANDARD 17)

find_package(OpenCV REQUIRED)

# Find ONNX Runtime package
find_path(ONNXRUNTIME_INCLUDE_DIR onnxruntime_c_api.h
    HINTS /opt/onnxruntime-linux-x64-rocm-1.18.0/include
)
find_library(ONNXRUNTIME_LIBRARY onnxruntime
    HINTS /opt/onnxruntime-linux-x64-rocm-1.18.0/lib
)

if(NOT ONNXRUNTIME_INCLUDE_DIR)
    message(FATAL_ERROR "ONNX Runtime include directory not found")
endif()
if(NOT ONNXRUNTIME_LIBRARY)
    message(FATAL_ERROR "ONNX Runtime library not found")
endif()

add_library(${project_name}-lib
    src/placeholder.cpp
    src/ia/inference.cpp
    src/ia/inference.h
)

target_include_directories(${project_name}-lib PUBLIC src)
target_include_directories(${project_name}-lib PUBLIC ${ONNXRUNTIME_INCLUDE_DIR})

target_link_libraries(${project_name}-lib
    PUBLIC ${OpenCV_LIBS}
    PUBLIC ${ONNXRUNTIME_LIBRARY}
)

# Add the main executable
add_executable(${project_name} 
    ./src/main.cpp
)
target_include_directories(${project_name} PUBLIC ${ONNXRUNTIME_INCLUDE_DIR})
target_link_libraries(${project_name} ${project_name}-lib)

# Add the video executable
add_executable(${project_name}_video 
    ./src/video.cpp
)

# Add the video executable
add_executable(${project_name}_video_rtsp 
    ./src/video_rtsp.cpp
)

target_include_directories(${project_name}_video PUBLIC ${ONNXRUNTIME_INCLUDE_DIR})
target_link_libraries(${project_name}_video ${project_name}-lib)

target_include_directories(${project_name}_video_rtsp PUBLIC ${ONNXRUNTIME_INCLUDE_DIR})
target_link_libraries(${project_name}_video_rtsp ${project_name}-lib)