mirror of
https://github.com/opencv/opencv.git
synced 2024-12-15 09:49:13 +08:00
158 lines
5.5 KiB
CMake
158 lines
5.5 KiB
CMake
if(ANDROID OR IOS)
|
|
ocv_module_disable(gpu)
|
|
endif()
|
|
|
|
set(the_description "GPU-accelerated Computer Vision")
|
|
ocv_add_module(gpu opencv_imgproc opencv_calib3d opencv_objdetect opencv_video opencv_nonfree opencv_legacy)
|
|
|
|
ocv_module_include_directories("${CMAKE_CURRENT_SOURCE_DIR}/src/cuda")
|
|
|
|
ocv_module_include_directories("${CMAKE_CURRENT_SOURCE_DIR}/../highgui/src")
|
|
|
|
file(GLOB lib_hdrs "include/opencv2/${name}/*.hpp" "include/opencv2/${name}/*.h")
|
|
file(GLOB lib_int_hdrs "src/*.hpp" "src/*.h")
|
|
file(GLOB lib_cuda_hdrs "src/cuda/*.hpp" "src/cuda/*.h")
|
|
file(GLOB lib_device_hdrs "src/opencv2/gpu/device/*.hpp" "src/opencv2/gpu/device/*.h")
|
|
file(GLOB lib_device_hdrs_detail "src/opencv2/gpu/device/detail/*.hpp" "src/opencv2/gpu/device/detail/*.h")
|
|
file(GLOB lib_srcs "src/*.cpp")
|
|
file(GLOB lib_cuda "src/cuda/*.cu*")
|
|
|
|
source_group("Include" FILES ${lib_hdrs})
|
|
source_group("Src\\Host" FILES ${lib_srcs} ${lib_int_hdrs})
|
|
source_group("Src\\Cuda" FILES ${lib_cuda} ${lib_cuda_hdrs})
|
|
source_group("Device" FILES ${lib_device_hdrs})
|
|
source_group("Device\\Detail" FILES ${lib_device_hdrs_detail})
|
|
|
|
if (HAVE_CUDA)
|
|
file(GLOB_RECURSE ncv_srcs "src/nvidia/*.cpp")
|
|
file(GLOB_RECURSE ncv_cuda "src/nvidia/*.cu")
|
|
file(GLOB_RECURSE ncv_hdrs "src/nvidia/*.hpp" "src/nvidia/*.h")
|
|
set(ncv_files ${ncv_srcs} ${ncv_hdrs} ${ncv_cuda})
|
|
|
|
source_group("Src\\NVidia" FILES ${ncv_files})
|
|
ocv_include_directories("src/nvidia" "src/nvidia/core" "src/nvidia/NPP_staging" ${CUDA_INCLUDE_DIRS})
|
|
|
|
#set (CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} "-keep")
|
|
#set (CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} "-Xcompiler;/EHsc-;")
|
|
|
|
if(MSVC)
|
|
if(NOT ENABLE_NOISY_WARNINGS)
|
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /wd4211 /wd4201 /wd4100 /wd4505 /wd4408")
|
|
|
|
foreach(var CMAKE_CXX_FLAGS CMAKE_CXX_FLAGS_RELEASE CMAKE_CXX_FLAGS_DEBUG)
|
|
string(REPLACE "/W4" "/W3" ${var} "${${var}}")
|
|
endforeach()
|
|
|
|
set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} -Xcompiler /wd4251)
|
|
endif()
|
|
endif()
|
|
|
|
ocv_cuda_compile(cuda_objs ${lib_cuda} ${ncv_cuda})
|
|
#CUDA_BUILD_CLEAN_TARGET()
|
|
|
|
|
|
set(cuda_link_libs ${CUDA_LIBRARIES} ${CUDA_npp_LIBRARY})
|
|
|
|
|
|
if(NOT APPLE)
|
|
unset(CUDA_nvcuvid_LIBRARY CACHE)
|
|
find_cuda_helper_libs(nvcuvid)
|
|
set(cuda_link_libs ${cuda_link_libs} ${CUDA_nvcuvid_LIBRARY})
|
|
endif()
|
|
|
|
if(WIN32)
|
|
unset(CUDA_nvcuvenc_LIBRARY CACHE)
|
|
find_cuda_helper_libs(nvcuvenc)
|
|
set(cuda_link_libs ${cuda_link_libs} ${CUDA_nvcuvenc_LIBRARY})
|
|
endif()
|
|
|
|
if(NOT APPLE AND WITH_FFMPEG)
|
|
set(cuda_link_libs ${cuda_link_libs} ${HIGHGUI_LIBRARIES})
|
|
endif()
|
|
else()
|
|
set(lib_cuda "")
|
|
set(cuda_objs "")
|
|
set(cuda_link_libs "")
|
|
set(ncv_files "")
|
|
endif()
|
|
|
|
ocv_set_module_sources(
|
|
HEADERS ${lib_hdrs}
|
|
SOURCES ${lib_int_hdrs} ${lib_cuda_hdrs} ${lib_device_hdrs} ${lib_device_hdrs_detail} ${lib_srcs} ${lib_cuda} ${ncv_files} ${cuda_objs}
|
|
)
|
|
|
|
ocv_create_module(${cuda_link_libs})
|
|
|
|
if(HAVE_CUDA)
|
|
if(HAVE_CUFFT)
|
|
CUDA_ADD_CUFFT_TO_TARGET(${the_module})
|
|
endif()
|
|
|
|
if(HAVE_CUBLAS)
|
|
CUDA_ADD_CUBLAS_TO_TARGET(${the_module})
|
|
endif()
|
|
|
|
install(FILES src/nvidia/NPP_staging/NPP_staging.hpp src/nvidia/core/NCV.hpp
|
|
DESTINATION ${OPENCV_INCLUDE_PREFIX}/opencv2/${name}
|
|
COMPONENT main)
|
|
endif()
|
|
|
|
ocv_add_precompiled_headers(${the_module})
|
|
|
|
################################################################################################################
|
|
################################ GPU Module Tests #####################################################
|
|
################################################################################################################
|
|
file(GLOB test_srcs "test/*.cpp")
|
|
file(GLOB test_hdrs "test/*.hpp" "test/*.h")
|
|
if(HAVE_CUDA)
|
|
file(GLOB nvidia "test/nvidia/*.cpp" "test/nvidia/*.hpp" "test/nvidia/*.h")
|
|
set(nvidia FILES "Src\\\\\\\\NVidia" ${nvidia}) # 8 ugly backslashes :'(
|
|
else()
|
|
set(nvidia "")
|
|
endif()
|
|
|
|
ocv_add_accuracy_tests(FILES "Include" ${test_hdrs}
|
|
FILES "Src" ${test_srcs}
|
|
${nvidia})
|
|
ocv_add_perf_tests()
|
|
|
|
|
|
|
|
set(perf_cpu_path "${CMAKE_CURRENT_SOURCE_DIR}/perf_cpu")
|
|
if(BUILD_PERF_TESTS AND EXISTS "${perf_cpu_path}")
|
|
# opencv_highgui is required for imread/imwrite
|
|
set(perf_deps ${the_module} opencv_ts opencv_highgui opencv_imgproc opencv_calib3d opencv_objdetect opencv_video opencv_nonfree)
|
|
ocv_check_dependencies(${perf_deps})
|
|
|
|
if(OCV_DEPENDENCIES_FOUND)
|
|
set(the_target "opencv_perf_gpu_cpu")
|
|
|
|
ocv_module_include_directories(${perf_deps} "${perf_cpu_path}")
|
|
|
|
if(NOT OPENCV_PERF_${the_module}_CPU_SOURCES)
|
|
file(GLOB perf_srcs "${perf_cpu_path}/*.cpp")
|
|
file(GLOB perf_hdrs "${perf_cpu_path}/*.hpp" "${perf_cpu_path}/*.h")
|
|
source_group("Src" FILES ${perf_srcs})
|
|
source_group("Include" FILES ${perf_hdrs})
|
|
set(OPENCV_PERF_${the_module}_CPU_SOURCES ${perf_srcs} ${perf_hdrs})
|
|
endif()
|
|
|
|
add_executable(${the_target} ${OPENCV_PERF_${the_module}_CPU_SOURCES})
|
|
target_link_libraries(${the_target} ${OPENCV_MODULE_${the_module}_DEPS} ${perf_deps} ${OPENCV_LINKER_LIBS})
|
|
|
|
# Additional target properties
|
|
set_target_properties(${the_target} PROPERTIES
|
|
DEBUG_POSTFIX "${OPENCV_DEBUG_POSTFIX}"
|
|
RUNTIME_OUTPUT_DIRECTORY "${EXECUTABLE_OUTPUT_PATH}"
|
|
)
|
|
|
|
if(ENABLE_SOLUTION_FOLDERS)
|
|
set_target_properties(${the_target} PROPERTIES FOLDER "tests performance")
|
|
endif()
|
|
|
|
ocv_add_precompiled_headers(${the_target})
|
|
else(OCV_DEPENDENCIES_FOUND)
|
|
#TODO: warn about unsatisfied dependencies
|
|
endif(OCV_DEPENDENCIES_FOUND)
|
|
endif()
|