plugin.cmake 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. function(ocv_create_builtin_dnn_plugin name target)
  2. ocv_debug_message("ocv_create_builtin_dnn_plugin(${ARGV})")
  3. if(NOT TARGET ${target})
  4. message(FATAL_ERROR "${target} does not exist!")
  5. endif()
  6. if(NOT OpenCV_SOURCE_DIR)
  7. message(FATAL_ERROR "OpenCV_SOURCE_DIR must be set to build the plugin!")
  8. endif()
  9. message(STATUS "DNN: add builtin plugin '${name}'")
  10. set(ENABLE_PRECOMPILED_HEADERS OFF) # no support for PCH in plugins, conflicts with module's source files
  11. # TODO: update CPU optimizations scripts to support plugins
  12. add_definitions(-D__OPENCV_BUILD=1)
  13. add_definitions(-DBUILD_PLUGIN=1)
  14. include_directories("${OPENCV_MODULE_opencv_dnn_BINARY_DIR}") # Cannot open include file: 'layers/layers_common.simd_declarations.hpp'
  15. foreach(src ${ARGN})
  16. if(EXISTS "${CMAKE_CURRENT_LIST_DIR}/src/${src}")
  17. list(APPEND sources "${CMAKE_CURRENT_LIST_DIR}/src/${src}")
  18. elseif(IS_ABSOLUTE "${src}")
  19. list(APPEND sources "${src}")
  20. else()
  21. message(FATAL_ERROR "Unknown source: ${src}")
  22. endif()
  23. endforeach()
  24. if(OPENCV_MODULE_${the_module}_SOURCES_DISPATCHED)
  25. list(APPEND sources ${OPENCV_MODULE_${the_module}_SOURCES_DISPATCHED})
  26. endif()
  27. set(__${name}_DEPS_EXT "")
  28. ocv_compiler_optimization_process_sources(sources __${name}_DEPS_EXT ${name})
  29. add_library(${name} MODULE ${sources})
  30. target_include_directories(${name} PRIVATE "${CMAKE_CURRENT_BINARY_DIR}")
  31. target_link_libraries(${name} PRIVATE ${target} ${__${name}_DEPS_EXT})
  32. target_link_libraries(${name} PRIVATE ${__plugin_libs})
  33. foreach(mod opencv_dnn
  34. opencv_core
  35. opencv_imgproc
  36. opencv_dnn
  37. )
  38. ocv_target_link_libraries(${name} LINK_PRIVATE ${mod})
  39. ocv_target_include_directories(${name} "${OPENCV_MODULE_${mod}_LOCATION}/include")
  40. endforeach()
  41. if(WIN32)
  42. add_definitions(-D_USE_MATH_DEFINES)
  43. set(OPENCV_PLUGIN_VERSION "${OPENCV_DLLVERSION}" CACHE STRING "")
  44. if(CMAKE_CXX_SIZEOF_DATA_PTR EQUAL 8)
  45. set(OPENCV_PLUGIN_ARCH "_64" CACHE STRING "")
  46. else()
  47. set(OPENCV_PLUGIN_ARCH "" CACHE STRING "")
  48. endif()
  49. else()
  50. set(OPENCV_PLUGIN_VERSION "" CACHE STRING "")
  51. set(OPENCV_PLUGIN_ARCH "" CACHE STRING "")
  52. endif()
  53. set_target_properties(${name} PROPERTIES
  54. CXX_STANDARD 11
  55. CXX_VISIBILITY_PRESET hidden
  56. DEBUG_POSTFIX "${OPENCV_DEBUG_POSTFIX}"
  57. OUTPUT_NAME "${name}${OPENCV_PLUGIN_VERSION}${OPENCV_PLUGIN_ARCH}"
  58. )
  59. if(WIN32)
  60. set_target_properties(${name} PROPERTIES LIBRARY_OUTPUT_DIRECTORY ${EXECUTABLE_OUTPUT_PATH})
  61. install(TARGETS ${name} OPTIONAL LIBRARY DESTINATION ${OPENCV_BIN_INSTALL_PATH} COMPONENT plugins)
  62. else()
  63. install(TARGETS ${name} OPTIONAL LIBRARY DESTINATION ${OPENCV_LIB_INSTALL_PATH} COMPONENT plugins)
  64. endif()
  65. add_dependencies(opencv_dnn_plugins ${name})
  66. endfunction()