src/CMakeLists.txt

   1 cmake_minimum_required(VERSION 2.8)
   2
   3 set(KCF_LIB_SRC kcf.cpp kcf.h fft.cpp threadctx.hpp pragmas.h dynmem.hpp)
   4
   5 find_package(PkgConfig)
   6
   7 SET(FFT "OpenCV" CACHE STRING "Select FFT implementation")
   8 SET_PROPERTY(CACHE FFT PROPERTY STRINGS OpenCV fftw cuFFTW cuFFT)
   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(CUDA_DEBUG "Enables error cheking for cuda and cufft. " OFF)
  14 option(BIG_BATCH "Enable transforming all features from all scales together." OFF)
  15 option(TEMPLATE_COMPLEXMAT "Use ComplexMat as abstract class." ON)
  16
  17 IF(PROFILING)
  18   add_definitions(-DPROFILING )
  19   MESSAGE(STATUS "Profiling mode")
  20 ENDIF()
  21
  22 IF(BIG_BATCH)
  23   add_definitions(-DBIG_BATCH )
  24   MESSAGE(STATUS "Big_batch mode")
  25 ENDIF()
  26
  27 IF(TEMPLATE_COMPLEXMAT)
  28   add_definitions(-DTEMPLATE_COMPLEXMAT )
  29   MESSAGE(STATUS "ComplexMat abstract class version.")
  30 ENDIF()
  31
  32 SET(use_cuda OFF)
  33
  34 IF(FFT STREQUAL "OpenCV")
  35   list(APPEND KCF_LIB_SRC fft_opencv.cpp complexmat.hpp)
  36 ELSEIF(FFT STREQUAL "fftw")
  37   list(APPEND KCF_LIB_SRC fft_fftw.cpp complexmat.hpp)
  38   add_definitions(-DFFTW)
  39   pkg_check_modules(FFTW REQUIRED fftw3f)
  40 ELSEIF(FFT STREQUAL "cuFFTW")
  41   list(APPEND KCF_LIB_SRC fft_fftw.cpp complexmat.hpp)
  42   add_definitions(-DFFTW -DCUFFTW)
  43   set(use_cuda ON)
  44 ELSEIF(FFT STREQUAL "cuFFT")
  45     list(APPEND KCF_LIB_SRC fft_cufft.cpp complexmat.cuh cuda_functions.cuh)
  46     add_definitions(-DCUFFT)
  47     set(use_cuda ON)
  48     iF(CUDA_DEBUG)
  49       add_definitions(-DCUDA_DEBUG)
  50       MESSAGE(STATUS "CUDA debug mode")
  51     ENDIF()
  52 ELSE()
  53   MESSAGE(FATAL_ERROR "Invalid FFT implementation selected")
  54 ENDIF()
  55
  56 IF((FFT STREQUAL "OpenCV") AND BIG_BATCH)
  57   message(SEND_ERROR "OpenCV version does not support big batch mode.")
  58 ENDIF()
  59
  60 IF((FFT STREQUAL "cuFFT") AND (ASYNC))
  61   message(SEND_ERROR "cuFFT version does not support ASYNC and OpenMP only if used with big batch mode.")
  62 ENDIF()
  63
  64 IF(ASYNC)
  65   add_definitions(-DASYNC)
  66   MESSAGE(STATUS "ASYNC")
  67 ELSEIF(OPENMP)
  68     add_definitions(-DOPENMP)
  69     MESSAGE(STATUS "OPENMP")
  70 ENDIF() #ASYNC
  71
  72 FIND_PACKAGE( OpenCV REQUIRED )
  73
  74 IF(use_cuda)
  75   find_package(CUDA REQUIRED)
  76
  77   if (NOT ${OpenCV_USE_CUFFT})
  78     message(SEND_ERROR "OpenCV was not built with cuFFT")
  79   endif()
  80
  81   include_directories ( ${CUDA_INCLUDE_DIRS} )
  82   set(CUDA_SEPARABLE_COMPILATION ON)
  83   set(CUDA_PROPAGATE_HOST_FLAGS OFF)
  84   set(CUDA_HOST_COMPILER /usr/bin/g++)
  85   list( APPEND CUDA_NVCC_FLAGS -O3 --gpu-architecture sm_62 -std=c++11)
  86   find_cuda_helper_libs(cufftw)
  87   IF(FFT STREQUAL "cuFFT")
  88   add_subdirectory(cuda)
  89   cuda_add_library(complexmat complexmat.cu)
  90   cuda_add_library(cuda_func cuda_functions.cu)
  91   ENDIF()
  92
  93 ENDIF()
  94
  95 add_subdirectory(piotr_fhog)
  96 add_subdirectory(cn)
  97
  98 add_library(kcf STATIC ${KCF_LIB_SRC})
  99 IF(NOT TEMPLATE_COMPLEXMAT)
 100     add_library(complexmat complexmat.cpp)
 101     target_link_libraries(kcf fhog cndata complexmat ${OpenCV_LIBS})
 102 ELSE()
 103     target_link_libraries(kcf fhog cndata ${OpenCV_LIBS})
 104 ENDIF()
 105 set_target_properties(kcf PROPERTIES VERSION 1.0.0 SOVERSION 1)
 106
 107 IF(FFT STREQUAL "fftw")
 108   target_link_libraries(kcf ${FFTW_LDFLAGS})
 109   IF(OPENMP)
 110     target_link_libraries(kcf fftw3_omp)
 111   ELSEIF(NOT ASYNC)
 112     target_link_libraries(kcf fftw3_threads)
 113   ENDIF()
 114 ENDIF() #FFTW
 115
 116 IF(FFT STREQUAL "cuFFTW")
 117   target_link_libraries(kcf ${CUDA_cufft_LIBRARY} ${CUDA_cufftw_LIBRARY})
 118 ENDIF() #cuFFTW
 119
 120 IF(FFT STREQUAL "cuFFT")
 121     target_link_libraries(kcf ${CUDA_cufft_LIBRARY} ${CUDA_LIBRARIES} complexmat cuda_func)
 122 ENDIF()
 123
 124 IF(PROFILING)
 125   target_link_libraries(kcf pfm)
 126 ENDIF()