CMakeLists.txt 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319
  1. cmake_minimum_required(VERSION 3.27)
  2. project(DepthGuide)
  3. set(CMAKE_CXX_STANDARD 20)
  4. add_executable(${PROJECT_NAME} src/main.cpp
  5. src/ai/impl/fast_sam.cpp
  6. src/ai/impl/zmq_client.cpp
  7. src/image_process/impl/camera_calibrator.cpp
  8. src/image_process/impl/image_process_ui.cpp
  9. src/image_process/impl/process_funcs.cpp
  10. src/image_process/impl/versatile_convertor.cpp
  11. src/image_process_v5/video_stabilization.cpp
  12. src/impl/main_impl.cpp
  13. src/impl/apps/app_selector/app_selector.cpp
  14. src/impl/apps/debug/app_debug.cpp
  15. src/impl/apps/depth_guide/depth_guide.cpp
  16. src/impl/apps/depth_guide_v2/depth_guide_v2.cpp
  17. src/impl/apps/endo_guide/endo_guide.cpp
  18. src/impl/apps/remote_ar/remote_ar.cpp
  19. src/impl/apps/scene_player/scene_player.cpp
  20. src/impl/apps/tiny_player/tiny_player.cpp
  21. src/codec/image_codec.cpp
  22. src/codec/image_decoder.cpp
  23. src/codec/image_encoder.cpp
  24. src/codec/mesh_decoder.cpp
  25. src/codec/mesh_encoder.cpp
  26. src/codec/pc_decoder.cpp
  27. src/codec/pc_encoder.cpp
  28. src/codec/scene_decoder.cpp
  29. src/codec/scene_encoder.cpp
  30. src/core/impl/event_timer.cpp
  31. src/core/impl/image_utility_v2.cpp
  32. src/core/impl/math_helper.cpp
  33. src/core/impl/memory_pool.cpp
  34. src/core/impl/object_manager.cpp
  35. src/core/impl/pc_utility.cpp
  36. src/module_v3/registration.cpp
  37. src/module/experiment/impl/calib_eval.cpp
  38. src/module/guidance/impl/acl_guide.cpp
  39. src/module/guidance/impl/cursor_guide.cpp
  40. src/module/guidance/impl/guide_helper.cpp
  41. src/module/impl/augment_manager.cpp
  42. src/module/impl/augment_manager_v2.cpp
  43. src/module/impl/camera_augment_helper.cpp
  44. src/module/impl/camera_augment_helper_v2.cpp
  45. src/module/impl/depth_guide_controller.cpp
  46. src/module/impl/image_augment_helper.cpp
  47. src/module/impl/image_augment_helper_v2.cpp
  48. src/module/impl/image_player.cpp
  49. src/module/impl/image_saver.cpp
  50. src/module/impl/image_viewer.cpp
  51. src/module/impl/image_streamer.cpp
  52. src/module/impl/pc_viewer.cpp
  53. src/module/impl/sophiar_monitor.cpp
  54. src/network_v3/sender_tcp.cpp
  55. src/network_v3/sender_udp_fec.cpp
  56. # src/network_v3/receiver_tcp.cpp
  57. src/network_v3/receiver_udp_fec.cpp
  58. # src/network/impl/crc_checker.cpp
  59. src/network/impl/fragment/third_party/rs.c
  60. # src/network/impl/fragment/frag_basic.cpp
  61. # src/network/impl/fragment/frag_rs.cpp
  62. # src/network/impl/multiplexer.cpp
  63. # src/network/impl/network.cpp
  64. src/render_v3/scene_render.cpp
  65. src/render/impl/render_line.cpp
  66. src/render/impl/render_mesh.cpp
  67. src/render/impl/render_pc.cpp
  68. src/render/impl/render_scene.cpp
  69. src/render/impl/render_texture.cpp
  70. src/render/impl/render_tools.cpp
  71. src/render/impl/render_utility.cpp)
  72. target_include_directories(${PROJECT_NAME} PRIVATE src)
  73. #target_compile_options(${PROJECT_NAME} PRIVATE -g -pg)
  74. #target_link_options(${PROJECT_NAME} PRIVATE -g -pg)
  75. #target_compile_options(${PROJECT_NAME} PRIVATE -fsanitize=leak)
  76. #target_link_options(${PROJECT_NAME} PRIVATE -fsanitize=leak)
  77. # image process sub-module
  78. add_subdirectory(src/image_process/cuda_impl)
  79. target_link_libraries(${PROJECT_NAME} ImageProcessCuda)
  80. #target_sources(${PROJECT_NAME} PRIVATE
  81. # src/image_process_v3/image_process.cpp)
  82. # old image process sub-module
  83. add_subdirectory(src/image_process_v3/cuda_impl)
  84. target_link_libraries(${PROJECT_NAME} ImageProcessCudaV3)
  85. target_sources(${PROJECT_NAME} PRIVATE
  86. src/image_process_v3/image_process.cpp)
  87. find_package(PkgConfig REQUIRED)
  88. # CUDA config
  89. find_package(CUDAToolkit REQUIRED)
  90. target_link_libraries(${PROJECT_NAME} CUDA::cudart CUDA::cuda_driver)
  91. # CCCL config
  92. find_package(CCCL REQUIRED)
  93. target_link_libraries(${PROJECT_NAME} CCCL::CCCL)
  94. # spdlog config
  95. find_package(spdlog REQUIRED)
  96. target_link_libraries(${PROJECT_NAME} spdlog::spdlog)
  97. target_compile_definitions(${PROJECT_NAME} PRIVATE SPDLOG_ACTIVE_LEVEL=SPDLOG_LEVEL_TRACE)
  98. # OpenCV config
  99. cmake_policy(SET CMP0146 OLD)
  100. find_package(OpenCV REQUIRED COMPONENTS cudaimgproc calib3d imgcodecs opencv_cudastereo)
  101. target_include_directories(${PROJECT_NAME} PRIVATE ${OpenCV_INCLUDE_DIRS})
  102. target_link_libraries(${PROJECT_NAME} ${OpenCV_LIBS})
  103. # glfw config
  104. if (WIN32)
  105. set(GLFW_INCLUDE_DIR C:/BuildEssentials/VS2019Libs/include)
  106. set(GLFW_LIB_DIR C:/BuildEssentials/VS2019Libs/lib)
  107. find_library(GLFW_LIB glfw3 HINTS ${GLFW_LIB_DIR})
  108. target_include_directories(${PROJECT_NAME} PRIVATE ${GLFW_INCLUDE_DIR})
  109. target_link_libraries(${PROJECT_NAME} ${GLFW_LIB})
  110. else ()
  111. find_package(glfw3 REQUIRED)
  112. target_link_libraries(${PROJECT_NAME} glfw)
  113. endif ()
  114. # glad config
  115. if (WIN32)
  116. set(GLAD_DIR C:/BuildEssentials/Library/glad)
  117. else ()
  118. set(GLAD_DIR /home/tpx/ext/src/glad)
  119. endif ()
  120. target_include_directories(${PROJECT_NAME} PRIVATE ${GLAD_DIR}/include)
  121. target_sources(${PROJECT_NAME} PRIVATE ${GLAD_DIR}/src/gl.c)
  122. # glm config
  123. find_package(glm REQUIRED)
  124. target_link_libraries(${PROJECT_NAME} glm::glm)
  125. target_compile_definitions(${PROJECT_NAME} PRIVATE GLM_ENABLE_EXPERIMENTAL)
  126. # imgui config
  127. if (WIN32)
  128. set(IMGUI_DIR C:/BuildEssentials/Library/imgui-1.89.5)
  129. else ()
  130. set(IMGUI_DIR /home/tpx/ext/src/imgui-1.91.0)
  131. endif ()
  132. set(IMGUI_BACKENDS_DIR ${IMGUI_DIR}/backends)
  133. target_include_directories(${PROJECT_NAME} PRIVATE ${IMGUI_DIR} ${IMGUI_BACKENDS_DIR})
  134. set(IMGUI_IMPL_FILES ${IMGUI_DIR}/imgui.cpp
  135. ${IMGUI_DIR}/imgui_draw.cpp
  136. ${IMGUI_DIR}/imgui_tables.cpp
  137. ${IMGUI_DIR}/imgui_widgets.cpp
  138. ${IMGUI_DIR}/imgui_demo.cpp
  139. ${IMGUI_BACKENDS_DIR}/imgui_impl_glfw.cpp
  140. ${IMGUI_BACKENDS_DIR}/imgui_impl_opengl3.cpp)
  141. target_sources(${PROJECT_NAME} PRIVATE ${IMGUI_IMPL_FILES})
  142. foreach(source_file ${IMGUI_IMPL_FILES})
  143. set_source_files_properties(${source_file} PROPERTIES COMPILE_OPTIONS "-O3")
  144. set_source_files_properties(${source_file} PROPERTIES COMPILE_OPTIONS "-DNDEBUG")
  145. endforeach()
  146. # ImGuiFileDialog config
  147. set(ImGuiFileDialog_DIR /home/tpx/ext/src/ImGuiFileDialog-0.6.7)
  148. add_subdirectory(${ImGuiFileDialog_DIR} third_party/imgui_file_dialog)
  149. target_include_directories(ImGuiFileDialog PRIVATE ${IMGUI_DIR})
  150. target_link_libraries(${PROJECT_NAME} ImGuiFileDialog)
  151. # imGuIZMO config
  152. set(IMGUIZMO_DIR /home/tpx/ext/src/imGuIZMO.quat-3.0/imGuIZMO.quat)
  153. target_include_directories(${PROJECT_NAME} PRIVATE ${IMGUIZMO_DIR})
  154. target_sources(${PROJECT_NAME} PRIVATE
  155. ${IMGUIZMO_DIR}/imGuIZMOquat.cpp)
  156. target_compile_definitions(${PROJECT_NAME} PRIVATE
  157. IMGUIZMO_IMGUI_FOLDER=${IMGUI_DIR}/)
  158. # NanoVG config
  159. set(NANOVG_DIR /home/tpx/ext/src/nanovg-f93799c)
  160. target_include_directories(${PROJECT_NAME} PRIVATE ${NANOVG_DIR}/src)
  161. target_sources(${PROJECT_NAME} PRIVATE ${NANOVG_DIR}/src/nanovg.c)
  162. # yaml-cpp
  163. find_package(yaml-cpp REQUIRED)
  164. target_include_directories(${PROJECT_NAME} PRIVATE ${YAML_CPP_INCLUDE_DIR})
  165. target_link_libraries(${PROJECT_NAME} ${YAML_CPP_LIBRARIES})
  166. if (WIN32)
  167. target_link_directories(${PROJECT_NAME} PRIVATE C:/BuildEssentials/VS2019Libs/lib)
  168. endif ()
  169. # JSON config
  170. find_package(nlohmann_json REQUIRED)
  171. target_link_libraries(${PROJECT_NAME} nlohmann_json::nlohmann_json)
  172. # Boost config
  173. find_package(Boost REQUIRED COMPONENTS iostreams)
  174. target_include_directories(${PROJECT_NAME} PRIVATE ${Boost_INCLUDE_DIRS})
  175. target_link_libraries(${PROJECT_NAME} ${Boost_LIBRARIES})
  176. # VTK config
  177. find_package(VTK REQUIRED)
  178. target_link_libraries(${PROJECT_NAME} ${VTK_LIBRARIES})
  179. vtk_module_autoinit(TARGETS ${PROJECT_NAME} MODULES ${VTK_LIBRARIES})
  180. target_sources(${PROJECT_NAME} PRIVATE src/render_v3/vtk_viewer.cpp)
  181. # PCL config
  182. find_package(PCL REQUIRED)
  183. target_include_directories(${PROJECT_NAME} PRIVATE ${PCL_INCLUDE_DIRS})
  184. target_link_directories(${PROJECT_NAME} PRIVATE ${PCL_LIBRARY_DIRS})
  185. target_compile_definitions(${PROJECT_NAME} PRIVATE ${PCL_DEFINITIONS})
  186. target_link_libraries(${PROJECT_NAME} ${PCL_LIBRARIES})
  187. # Eigen3 config
  188. find_package(Eigen3 REQUIRED)
  189. target_link_libraries(${PROJECT_NAME} Eigen3::Eigen)
  190. # Orbbec config
  191. set(OrbbecSDK_ROOT_DIR /home/tpx/ext/src/OrbbecSDK_v1.10.12/SDK)
  192. set(OrbbecSDK_LIBRARY_DIRS ${OrbbecSDK_ROOT_DIR}/lib)
  193. set(OrbbecSDK_INCLUDE_DIR ${OrbbecSDK_ROOT_DIR}/include)
  194. target_include_directories(${PROJECT_NAME} PRIVATE ${OrbbecSDK_INCLUDE_DIR})
  195. target_link_directories(${PROJECT_NAME} PRIVATE ${OrbbecSDK_LIBRARY_DIRS})
  196. target_link_libraries(${PROJECT_NAME} OrbbecSDK)
  197. target_sources(${PROJECT_NAME} PRIVATE
  198. src/device/impl/orb_camera.cpp
  199. src/device/impl/orb_camera_ui.cpp)
  200. # MVS config
  201. if (WIN32)
  202. set(MVS_DIR "C:/BuildEssentials/Library/MVS/Development")
  203. set(MVS_INCLUDE_DIR ${MVS_DIR}/Includes)
  204. set(MVS_LIB_DIR ${MVS_DIR}/Libraries/win64)
  205. else ()
  206. set(MVS_DIR /opt/MVS)
  207. set(MVS_INCLUDE_DIR ${MVS_DIR}/include)
  208. set(MVS_LIB_DIR ${MVS_DIR}/lib/64)
  209. endif ()
  210. find_library(MVS_LIB MvCameraControl HINTS ${MVS_LIB_DIR})
  211. target_include_directories(${PROJECT_NAME} PRIVATE ${MVS_INCLUDE_DIR})
  212. target_link_libraries(${PROJECT_NAME} ${MVS_LIB})
  213. target_sources(${PROJECT_NAME} PRIVATE
  214. src/device/impl/mvs_camera.cpp
  215. src/device/impl/mvs_camera_ui.cpp)
  216. # libuvc config
  217. find_package(libuvc REQUIRED)
  218. target_link_libraries(${PROJECT_NAME} LibUVC::UVCShared LibUSB::LibUSB)
  219. target_sources(${PROJECT_NAME} PRIVATE
  220. src/device/impl/uvc_camera.cpp
  221. src/device/impl/uvc_camera_ui.cpp)
  222. # Crypto++ config
  223. set(CRYPTOPP_DIR /home/tpx/usr)
  224. set(CRYPTOPP_LIB_DIR ${CRYPTOPP_DIR}/lib)
  225. find_library(CRYPTOPP_LIB cryptopp HINTS ${CRYPTOPP_LIB_DIR})
  226. set(CRYPTOPP_INCLUDE_DIR ${CRYPTOPP_DIR}/include)
  227. target_include_directories(${PROJECT_NAME} PRIVATE ${CRYPTOPP_INCLUDE_DIR})
  228. target_link_libraries(${PROJECT_NAME} ${CRYPTOPP_LIB})
  229. # xxHash config
  230. pkg_search_module(XXHASH REQUIRED libxxhash)
  231. target_include_directories(${PROJECT_NAME} PRIVATE ${XXHASH_INCLUDE_DIRS})
  232. target_link_directories(${PROJECT_NAME} PRIVATE /home/tpx/usr/lib)
  233. target_link_libraries(${PROJECT_NAME} ${XXHASH_LIBRARIES})
  234. # NvEnc config
  235. if (WIN32)
  236. set(NVCODEC_DIR C:/BuildEssentials/CUDA/Video_Codec_SDK_12.0.16)
  237. set(NVENC_LIB_DIR ${NVCODEC_DIR}/Lib/x64)
  238. find_library(NVENC_LIB nvencodeapi HINTS ${NVENC_LIB_DIR})
  239. else ()
  240. set(NVCODEC_DIR /home/tpx/src/Video_Codec_SDK_12.2.72)
  241. find_library(NVENC_LIB nvidia-encode)
  242. endif ()
  243. set(NVCODEC_INCLUDE_DIR ${NVCODEC_DIR}/Interface)
  244. target_include_directories(${PROJECT_NAME} PRIVATE ${NVCODEC_INCLUDE_DIR})
  245. target_link_libraries(${PROJECT_NAME} ${NVENC_LIB})
  246. target_sources(${PROJECT_NAME} PRIVATE src/codec/encoder_nvenc.cpp)
  247. # NvDec config
  248. if (WIN32)
  249. set(NVCODEC_DIR C:/BuildEssentials/CUDA/Video_Codec_SDK_12.0.16)
  250. find_library(NVDEC_LIB nvcuvid HINTS ${NVCODEC_DIR}/Lib/x64)
  251. else ()
  252. set(NVCODEC_DIR /home/tpx/src/Video_Codec_SDK_12.0.16)
  253. find_library(NVDEC_LIB nvcuvid)
  254. endif ()
  255. set(NVCODEC_INCLUDE_DIR ${NVCODEC_DIR}/Interface)
  256. target_include_directories(${PROJECT_NAME} PRIVATE ${NVCODEC_INCLUDE_DIR})
  257. target_link_libraries(${PROJECT_NAME} ${NVDEC_LIB})
  258. target_sources(${PROJECT_NAME} PRIVATE src/codec/decoder_nvdec.cpp)
  259. # nvJPEG config
  260. target_link_libraries(${PROJECT_NAME} CUDA::nvjpeg)
  261. #target_sources(${PROJECT_NAME} PRIVATE src/codec/encoder_jpeg.cpp)
  262. #target_sources(${PROJECT_NAME} PRIVATE src/codec/decoder_nvjpeg.cpp)
  263. # assimp config
  264. find_package(assimp REQUIRED)
  265. target_link_libraries(${PROJECT_NAME} assimp::assimp)
  266. # ZeroMQ config
  267. find_package(azmq REQUIRED)
  268. target_link_libraries(${PROJECT_NAME} Azmq::azmq)
  269. # BS::thread_pool config
  270. set(BSTP_DIR /home/tpx/ext/src/thread-pool-4.1.0)
  271. target_include_directories(${PROJECT_NAME} PRIVATE ${BSTP_DIR}/include)
  272. # Sophiar2 config
  273. if (WIN32)
  274. set(Sophiar2DIR D:/Program/Robot/Sophiar2)
  275. else ()
  276. set(Sophiar2DIR /home/tpx/ext/project/Sophiar2)
  277. endif ()
  278. add_subdirectory(${Sophiar2DIR}/src Sophiar2Lib)
  279. target_include_directories(${PROJECT_NAME} PRIVATE ${Sophiar2DIR}/src)
  280. target_link_libraries(${PROJECT_NAME} Sophiar2Lib)
  281. add_compile_definitions(EIGEN_DONT_VECTORIZE)
  282. add_compile_definitions(EIGEN_DISABLE_UNALIGNED_ARRAY_ASSERT)