#声明最小版本
cmake_minimum_required(VERSION 3.2)
#工程名称
project(AIRecognize LANGUAGES CXX CUDA)
# c++ 11
set(CMAKE_CXX_FLAGS "-std=c++14 -I/usr/include/mysql -Wall") 

set(CMAKE_BUILD_TYPE Debug)

# CUDA
find_package(CUDA REQUIRED)
message(STATUS "Found CUDA headers: ${CUDA_INCLUDE_DIRS}")
message(STATUS "Found CUDA libraries: ${CUDA_LIBRARIES}")

# TensorRT
set(TENSORRT_ROOT /usr/local/TensorRT-8.6.1.6)
find_path(TENSORRT_INCLUDE_DIR NvInfer.h
        HINTS ${TENSORRT_ROOT} PATH_SUFFIXES include/)
message(STATUS "Found TensorRT headers: ${TENSORRT_INCLUDE_DIR}")

# TensorRT infer
find_library(TENSORRT_LIBRARY_INFER nvinfer
        HINTS ${TENSORRT_ROOT} ${TENSORRT_BUILD} ${CUDA_TOOLKIT_ROOT_DIR}
        PATH_SUFFIXES lib lib64 lib/x64)
message(STATUS "Found TensorRT-nvinfer libs: ${TENSORRT_LIBRARY_INFER}")

# TensorRT onnxparser
find_library(TENSORRT_LIBRARY_ONNXPARSER nvonnxparser
        HINTS  ${TENSORRT_ROOT} ${TENSORRT_BUILD} ${CUDA_TOOLKIT_ROOT_DIR}
        PATH_SUFFIXES lib lib64 lib/x64)
message(STATUS "Found TensorRT-onnxparser libs: ${TENSORRT_LIBRARY_ONNXPARSER}")
set(TENSORRT_LIBRARY ${TENSORRT_LIBRARY_INFER} ${TENSORRT_LIBRARY_ONNXPARSER})

# OpenCV
find_package(OpenCV REQUIRED)

# Boost
find_package(Boost REQUIRED COMPONENTS filesystem system)

# 查找FreeType库
find_package(Freetype REQUIRED)
message(STATUS "Found Freetype libraries: ${FREETYPE_LIBRARIES}")

# 查找 librabbitmq 库
find_package(PkgConfig REQUIRED)
pkg_check_modules(RABBITMQ REQUIRED librabbitmq)

# 手动查找 hiredis 头文件和库文件
find_path(HIREDIS_INCLUDE_DIR hiredis.h PATH_SUFFIXES hiredis)
find_library(HIREDIS_LIBRARY NAMES hiredis)

# 检查是否找到 hiredis
if (HIREDIS_INCLUDE_DIR AND HIREDIS_LIBRARY)
message(STATUS "Found HIREDIS headers: ${HIREDIS_INCLUDE_DIR}")
endif()

# 查找 jsoncpp 库
find_package(jsoncpp REQUIRED)

find_package(CURL REQUIRED)

# 添加opencv头文件
include_directories(${OpenCV_INCLUDE_DIRS})
# 添加freetype头文件
include_directories(${FREETYPE_INCLUDE_DIRS})

link_directories(/usr/local/lib)

# 递归收集rtsp目录及其子目录下的所有.cpp文件
file(GLOB_RECURSE base_srcs "${CMAKE_CURRENT_SOURCE_DIR}/base/*.cpp" "${CMAKE_CURRENT_SOURCE_DIR}/base/*.h")
file(GLOB_RECURSE cls_srcs "${CMAKE_CURRENT_SOURCE_DIR}/cls/*.cpp" "${CMAKE_CURRENT_SOURCE_DIR}/cls/*.h")
file(GLOB_RECURSE deploy_srcs "${CMAKE_CURRENT_SOURCE_DIR}/deploy/*.cpp" "${CMAKE_CURRENT_SOURCE_DIR}/deploy/*.h")
#file(GLOB_RECURSE rtsp_srcs "${CMAKE_CURRENT_SOURCE_DIR}/rtsp/*.cpp" "${CMAKE_CURRENT_SOURCE_DIR}/rtsp/*.h")
#file(GLOB_RECURSE rtsp_srcs "${CMAKE_CURRENT_SOURCE_DIR}/third_party/live555/*.cpp" "${CMAKE_CURRENT_SOURCE_DIR}/third_party/live555/*.h")
#file(GLOB_RECURSE json_srcs "${CMAKE_CURRENT_SOURCE_DIR}/third_party/jsoncpp/**/*.cpp")


# 查找指定目录下的所有源文件，然后将结果存进指定变量名。
#aux_source_directory(. SRC_LIST)
aux_source_directory(${CMAKE_SOURCE_DIR} SRC_LIST)

# 生成可执行文件
#cuda_add_executable(AIRecognize ${SRC_LIST} ${base_srcs} ${cls_srcs} ${deploy_srcs} ${rtsp_srcs})
add_executable(AIRecognize ${SRC_LIST} ${base_srcs} ${cls_srcs} ${deploy_srcs} ${rtsp_srcs})

# 使用jsoncpp
#find_library(jsoncpp_lib libjsoncpp.a)

# 生成可执行文件后，为生成文件target添加库
target_include_directories(
        AIRecognize PRIVATE
    ${CMAKE_CURRENT_SOURCE_DIR}
    ${CUDA_INCLUDE_DIRS} 
    ${TENSORRT_INCLUDE_DIR} 
    ${OpenCV_INCLUDE_DIRS}     
    ${CMAKE_CURRENT_SOURCE_DIR}/include
    ${RABBITMQ_INCLUDE_DIRS}
    ${jsoncpp_INCLUDE_DIRS}
    ${CURL_INCLUDE_DIRS}
    ${HIREDIS_INCLUDE_DIR}
)

target_link_libraries(
        AIRecognize
    ${CUDA_LIBRARIES} 
    ${TENSORRT_LIBRARY} 
    ${OpenCV_LIBS}
    jsoncpp
    ${Boost_LIBRARIES}
    ${FREETYPE_LIBRARIES}
    -lmysqlclient
    ${RABBITMQ_LIBRARIES}
    ${jsoncpp_LIBRARIES}
    ${CURL_LIBRARIES}
    ${HIREDIS_LIBRARY}
    )