obs-webrtc: Add new WHEP Source

Co-authored-by: Kevin Wang <[email protected]
Sean-Der · Jul 8, 2024 · 04c2a67 · 04c2a67
1 parent 16325de
commit 04c2a67
Show file tree

Hide file tree

Showing 6 changed files with 662 additions and 6 deletions.
diff --git a/plugins/obs-webrtc/CMakeLists.txt b/plugins/obs-webrtc/CMakeLists.txt
@@ -10,15 +10,48 @@ endif()
 
 find_package(LibDataChannel 0.20 REQUIRED)
 find_package(CURL REQUIRED)
+find_package(
+  FFmpeg REQUIRED
+  COMPONENTS avcodec
+             avfilter
+             avdevice
+             avutil
+             swscale
+             avformat
+             swresample)
+
+if(NOT TARGET OBS::media-playback)
+  add_subdirectory("${CMAKE_SOURCE_DIR}/deps/media-playback" "${CMAKE_BINARY_DIR}/deps/media-playback")
+endif()
 
 add_library(obs-webrtc MODULE)
 add_library(OBS::webrtc ALIAS obs-webrtc)
 
 target_sources(
-  obs-webrtc PRIVATE # cmake-format: sortable
-                     obs-webrtc.cpp webrtc-utils.h whep-source.cpp whep-source.h whip-output.cpp whip-output.h whip-service.cpp whip-service.h)
+  obs-webrtc
+  PRIVATE # cmake-format: sortable
+          obs-webrtc.cpp
+          webrtc-utils.h
+          whep-source.cpp
+          whep-source.h
+          whip-output.cpp
+          whip-output.h
+          whip-service.cpp
+          whip-service.h)
 
-target_link_libraries(obs-webrtc PRIVATE OBS::libobs LibDataChannel::LibDataChannel CURL::libcurl)
+target_link_libraries(
+  obs-webrtc
+  PRIVATE OBS::libobs
+          OBS::media-playback
+          LibDataChannel::LibDataChannel
+          CURL::libcurl
+          FFmpeg::avcodec
+          FFmpeg::avfilter
+          FFmpeg::avformat
+          FFmpeg::avdevice
+          FFmpeg::avutil
+          FFmpeg::swscale
+          FFmpeg::swresample)
 
 # cmake-format: off
 set_target_properties_obs(obs-webrtc PROPERTIES FOLDER plugins PREFIX "")

diff --git a/plugins/obs-webrtc/cmake/legacy.cmake b/plugins/obs-webrtc/cmake/legacy.cmake
@@ -8,14 +8,46 @@ endif()
 
 find_package(LibDataChannel 0.20 REQUIRED)
 find_package(CURL REQUIRED)
+find_package(
+  FFmpeg REQUIRED
+  COMPONENTS avcodec
+             avfilter
+             avdevice
+             avutil
+             swscale
+             avformat
+             swresample)
+if(NOT TARGET OBS::media-playback)
+  add_subdirectory("${CMAKE_SOURCE_DIR}/deps/media-playback" "${CMAKE_BINARY_DIR}/deps/media-playback")
+endif()
 
 add_library(obs-webrtc MODULE)
 add_library(OBS::webrtc ALIAS obs-webrtc)
 
-target_sources(obs-webrtc PRIVATE obs-webrtc.cpp whip-output.cpp whip-output.h whip-service.cpp whip-service.h
-                                  webrtc-utils.h)
+target_sources(
+  obs-webrtc
+  PRIVATE obs-webrtc.cpp
+          whip-output.cpp
+          whip-output.h
+          whip-service.cpp
+          whip-service.h
+          whep-source.cpp
+          whep-source.h
+          webrtc-utils.h)
 
-target_link_libraries(obs-webrtc PRIVATE OBS::libobs LibDataChannel::LibDataChannel CURL::libcurl)
+target_link_libraries(
+  obs-webrtc
+  PRIVATE OBS::libobs
+          OBS::media-playback
+          LibDataChannel::LibDataChannel
+          CURL::libcurl
+          FFmpeg::avcodec
+          FFmpeg::avfilter
+          FFmpeg::avformat
+          FFmpeg::avdevice
+          FFmpeg::avutil
+          FFmpeg::swscale
+          FFmpeg::swresample)
 
 set_target_properties(obs-webrtc PROPERTIES FOLDER "plugins")
 

diff --git a/plugins/obs-webrtc/data/locale/en-US.ini b/plugins/obs-webrtc/data/locale/en-US.ini
@@ -1,4 +1,5 @@
 Output.Name="WHIP Output"
+Source.Name="WHEP Source"
 Service.Name="WHIP Service"
 Service.BearerToken="Bearer Token"
 

diff --git a/plugins/obs-webrtc/obs-webrtc.cpp b/plugins/obs-webrtc/obs-webrtc.cpp
@@ -1,6 +1,7 @@
 #include <obs-module.h>
 
 #include "whip-output.h"
+#include "whep-source.h"
 #include "whip-service.h"
 
 OBS_DECLARE_MODULE()
@@ -14,6 +15,7 @@ bool obs_module_load()
 {
 	register_whip_output();
 	register_whip_service();
+	register_whep_source();
 
 	return true;
 }