cmake_minimum_required(VERSION 3.5)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DGPU -DCUDNN -DOPENCV -g -std=c++11 -W -O3 -pthread")
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -DGPU -DCUDNN -DOPENCV -g -std=c++11 -W -O3")
project(demo)
set(EXECUTABLE_OUTPUT_PATH ${PROJECT_SOURCE_DIR}/build)

set(CMAKE_CXX_STANDARD 11)
add_compile_options(-fPIC -g -Wall -Wshadow -Wno-sign-compare -w)

file(GLOB_RECURSE CURRENT_HEADERS
  ./*.h
  ./lib/*.h
  ./lib/core/*.h
  ./lib/tracker_tools/*.h
  ./lib/encoder_tools/*.h
  ./lib/detecter_tools/*.h
  ./lib/detecter_tools/extra/*.h
  ./lib/Munkres_assign/*.h
  ./lib/utils/*.h
  ./lib/detecter_tools/*.hpp
  ./lib/detecter_tools/*.cuh
  ./lib/detecter_tools/darknet/*.hpp
  ./lib/detecter_tools/darknet/*.cuh)

file(GLOB sources
./*.cpp
./lib/*.cpp
./lib/core/*.cpp
./lib/tracker_tools/*.cpp
./lib/encoder_tools/*.cpp
./lib/detecter_tools/*.cpp
./lib/Munkres_assign/*.cpp
./lib/utils/*.cpp
./lib/detecter_tools/*.cpp
./lib/detecter_tools/darknet/*.c
./lib/detecter_tools/darknet/*.cpp
./lib/detecter_tools/darknet/*.cu)


source_group("Include" FILES ${CURRENT_HEADERS}) 
source_group("Source" FILES ${sources}) 


#cudaS
find_package(CUDA REQUIRED)

message("-- CUDA version:$(CUDA_VERSION)")

if(CUDA_VERSION_MAJOR GREATER 9)
        message("-- CUDA ${CUDA_VERSION_MAJOR} detected")
        set(
                CUDA_NVCC_FLAGS
                ${CUDA_NVCC_FLAGS}; 
                -gencode arch=compute_61,code=sm_61
        )
        set(CUDA_NVCC_FLAGS_RELWITHDEBINFO "--device-debug;-lineinfo")
endif()


set(TENSORFLOW_DIR /home/disk1/s_opt/tensorflow/tensorflow-1.14.0)

link_directories(/usr/local/cuda-10.0/lib64)
#link_directories(/usr/local/lib)

SET(LIBS
        opencv_core
        opencv_highgui
        opencv_imgcodecs
        opencv_imgproc
        opencv_videoio
        opencv_video
        tensorflow_cc
        tensorflow_framework   
        cudart
        cublas     
	    stdc++fs
        curand
        cuda
        cudnn
        )

include_directories(
        ${CMAKE_CURRENT_SOURCE_DIR}
        /home/disk1/s_opt/tensorflow/tensorflow-1.14.0
        /home/disk1/s_opt/tensorflow/tensorflow-1.14.0/bazel-genfiles
        /home/disk1/s_opt/tensorflow/tensorflow-1.14.0/bazel-bin/tensorflow
        /home/disk1/s_opt/tensorflow/tensorflow-1.14.0/tensorflow/contrib/makefile/downloads/nsync/public
        /home/disk1/s_opt/tensorflow/tensorflow-1.14.0/tensorflow/contrib/makefile/downloads/eigen
        /home/disk1/s_opt/tensorflow/tensorflow-1.14.0/tensorflow/contrib/makefile/downloads/absl
        /home/disk1/s_opt/tensorflow/tensorflow-1.14.0/tensorflow/contrib/makefile/downloads/protobuf/src
)
include_directories(/home/disk1/s_opt/opencv/opencv3.4.8/include) # 使用了系统的opencv，所以就不用这个了
include_directories(/usr/local/cuda/include)
include_directories(".")
include_directories("./lib")
include_directories("./lib/core")
include_directories("./lib/detecter_tools")
include_directories("./lib/encoder_tools")
include_directories("./lib/Munkres_assign")
include_directories("./lib/tracker_tools")
include_directories("./lib/utils")
include_directories("./lib/Munkres_assign/munkres")
link_directories(${TENSORFLOW_DIR}/tensorflow/contrib/makefile/downloads/nsync/builds/default.linux.c++11)
link_directories(${TENSORFLOW_DIR}/bazel-bin/tensorflow)
#link_directories(/usr/local/lib)
link_directories(/usr/local/cuda/lib64)  #
link_directories(/home/disk1/s_opt/opencv/opencv3.4.8/lib)
#link_directories("/usr/lib/x86_64-linux-gnu")


cuda_add_executable(${PROJECT_NAME}
	SDK_main.cpp
        ${sources}
        ${CURRENT_HEADERS}
        )
target_link_libraries(${PROJECT_NAME}
	${LIBS}
        )