CMakeLists.txt

   1 cmake_minimum_required(VERSION 3.0.2)
   2
   3 project(kcf_tracker)
   4
   5 SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -Wall -Wextra -pedantic -O3 -Wno-long-long -fno-omit-frame-pointer")
   6
   7 SET(FFT "OpenCV" CACHE STRING "Select FFT implementation")
   8 SET_PROPERTY(CACHE FFT PROPERTY STRINGS OpenCV OpenCV_cuFFT fftw cuFFTW)
   9 MESSAGE(STATUS "FFT implementation: ${FFT}")
  10
  11 option(OPENMP "Use OpenMP library. Works with FFTW and OpenCV implementation." OFF)
  12 option(ASYNC "Works only if OPENCV_CUFFT is not ON. Will enable C++ async directive." OFF)
  13 option(DEBUG_MODE "Additional terminal outputs and screens. " OFF)
  14 option(PROFILING "Enable profiling using perf_event_open together with libpfm4. " OFF)
  15
  16 IF(DEBUG_MODE )
  17   add_definitions(-DDEBUG_MODE )
  18   MESSAGE(STATUS "Debug mode")
  19 ENDIF() #DEBUG_MODE
  20
  21 IF(PROFILING)
  22   add_definitions(-DPROFILING )
  23   MESSAGE(STATUS "Profiling mode")
  24 ENDIF() #DEBUG_MODE
  25
  26 SET(use_cuda OFF)
  27
  28 IF(FFT STREQUAL "OpenCV")
  29 ELSEIF(FFT STREQUAL "OpenCV_cuFFT")
  30   add_definitions(-DOPENCV_CUFFT)
  31   set(use_cuda ON)
  32 ELSEIF(FFT STREQUAL "fftw")
  33   add_definitions(-DFFTW)
  34 ELSEIF(FFT STREQUAL "cuFFTW")
  35   add_definitions(-DFFTW -DCUFFTW)
  36   set(use_cuda ON)
  37 ELSE()
  38   MESSAGE(FATAL_ERROR "Invalid FFT implementation selected")
  39 ENDIF()
  40
  41 IF(ASYNC AND NOT OPENMP)
  42   add_definitions(-DASYNC)
  43   MESSAGE(STATUS "ASYNC")
  44 ENDIF() #ASYNC
  45
  46 FIND_PACKAGE( OpenCV REQUIRED )
  47 link_directories ( ${OpenCV_LIB_DIR} )
  48 MESSAGE(STATUS "OpenCV_LIB_DIR: ${OpenCV_LIB_DIR} ")
  49
  50 IF ("${CMAKE_CXX_COMPILER_ID}" STREQUAL "GNU")
  51   IF(NOT OPENMP)
  52     SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -pthread -Wno-unknown-pragmas")
  53   ELSE()
  54     MESSAGE(STATUS "OpenMP")
  55     SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fopenmp")
  56   ENDIF() #ASYNC
  57 ENDIF ()
  58
  59 include_directories ( ${OpenCV_INCLUDE_DIRS} )
  60 MESSAGE(STATUS "OpenCV_INCLUDE_DIRS: ${OpenCV_INCLUDE_DIRS}")
  61
  62 INCLUDE_DIRECTORIES( ${CMAKE_BINARY_DIR}/)
  63 include_directories(${CMAKE_SOURCE_DIR}/src/ ${CMAKE_SOURCE_DIR}/src/piotr_fhog ${CMAKE_SOURCE_DIR}/src/cn)
  64
  65 include_directories(${CMAKE_SOURCE_DIR}/sse2neon)
  66
  67 IF(use_cuda)
  68   find_package(CUDA REQUIRED)
  69
  70   include_directories ( ${CUDA_INCLUDE_DIRS} )
  71   MESSAGE(STATUS "CUDA_INCLUDE_DIRS: ${CUDA_INCLUDE_DIRS}")
  72   set(CUDA_SEPARABLE_COMPILATION ON)
  73   set(CUDA_PROPAGATE_HOST_FLAGS OFF)
  74   set(CUDA_HOST_COMPILER /usr/bin/g++)
  75   list( APPEND CUDA_NVCC_FLAGS "-O3 --gpu-architecture compute_62 -std=c++11")
  76   find_cuda_helper_libs(cufftw)
  77 ENDIF()
  78
  79 add_subdirectory(src)
  80
  81 IF(NOT use_cuda)
  82   add_executable(kcf_vot main_vot.cpp vot.hpp)
  83 ELSE()
  84   cuda_add_executable( kcf_vot main_vot.cpp vot.hpp )
  85   target_link_libraries(kcf_vot ${CUDA_LIBRARIES})
  86 ENDIF() #OPENCV_CUFFT
  87
  88 target_link_libraries(kcf_vot ${OpenCV_LIBS} kcf)
  89
  90 IF(FFT STREQUAL "fftw")
  91   target_link_libraries(kcf_vot fftw3f)
  92   IF(OPENMP)
  93     target_link_libraries(kcf_vot fftw3_omp)
  94   ENDIF()
  95 ENDIF() #FFTW
  96
  97 IF(FFT STREQUAL "cuFFTW")
  98   target_link_libraries(kcf_vot ${CUDA_cufft_LIBRARY} ${CUDA_cufftw_LIBRARY})
  99 ENDIF() #cuFFTW
 100
 101 IF(PROFILING)
 102   target_link_libraries(kcf_vot pfm)
 103 ENDIF()