ToB企服应用市场:ToB评测及商务社交产业平台
标题:
WebRTC视频 01 - 视频采集整体架构
[打印本页]
作者:
张裕
时间:
2024-11-23 18:23
标题:
WebRTC视频 01 - 视频采集整体架构
WebRTC视频 01 - 视频采集整体架构（本文）
WebRTC视频 02 - 视频采集类 VideoCaptureModule
WebRTC视频 03 - 视频采集类 VideoCaptureDS 上篇
WebRTC视频 04 - 视频采集类 VideoCaptureDS 中篇
WebRTC视频 05 - 视频采集类 VideoCaptureDS 下篇
一、前言：
我们从1对1通信说起，假如有一天，你和你情敌使用X信进行1v1通信，想象一下画面是不是一个大画面中有一个小画面？这在布局中就叫做PIP（picture in picture）；这个随手一点，看似在1s不到就完成的动作，内里却经过了很多复杂的操作，我们本日开始写一系列文章介绍下这俩帅哥的图片怎么体现的。
二、宏观流程：
起首开始呼唤的时间会对摄像头和体现屏幕进行初始化。
摄像头采集完数据之后会进行分发，一路给本地进行渲染（你本身就画出来了），另外一路送给编码器进行编码。
编码后的数据通过网络模块发送（这内里其实非常复杂，后续专题分析）。
接收端通过网络模块接收数据，并进行一些排序，去掉RTP头等操作，得到编码后的视频帧。
视频帧送给解码模块进行解码。
将解码后的数据进行渲染（你女票也画出来了）。
三、类图：
内里最重要的就是VideoCaptureModule，这是一个抽象类，差别平台有本身的实现。
VideoTrack是负责将整个链路创建起来。并不负责处理详细数据，数据由VideoCaptureModule、VcmCapture、VideoBroadcaster处理。
VcmCapture是PeerConnectionClient这个demo实现的一个类，内里包含了vcm_，又实现了TestVideoCapture（内里有broadcaster），所以这个类是左手连接数据源vcm，右手连接broadcaster；
VcmCapture包含的成员（其实是父类包含）VideoBroadCaster，负责分发给本地渲染器和Encoder；
上面就是核心类，接下来看其他类，也就是看看怎么使用上面的核心类的。
采集视频流的时间，起首会创建CaptureTrackSource。这个类内里会有capture（也就是VcmCapture）；
然后就是创建VideoTrack，内里有个成员变量video_source，就是上面创建的CaptureTrackSource的接口类指针，也就是说创建的收入接收了入参CaptureTrackSource，拥有了它；
在PeerConnectionClient这个demo中，调用StartLocalRenderer的时间，就会调用VideoTrack（还记得吗，它是专门负责创建链路的）的AddOrUpdateSink，然后就会调用CaptureTrackSource的AddOrUpdateSink，接着调用VideoBroadCaster的AddOrUpdateSink。这个时间VideoBroadCaster就会将本地渲染器添加到本身的列表中，由于之前
VideoCaptureModule****已经将采集模块启动起来
了，因此，数据就源源不断的从VideoCaptureModule进入到VcmCapture，然后再进入VideoBroadCaster当中。
VideoBroadcaster收到数据，发现目的列表中有数据的时间，就会将数据转发给这些目的。也就是本地渲染器，这个时间本地就可以看到本身视频了；
同样，在媒体协商进行到最后一步的时间，就会将编码器添加到VideoBroadCaster的目的列表当中，这样给本地渲染器分发数据的时间，同时也给编码器分发。
四、代码走读：
前面说了VideoTrack主要职责是创建数据链路，将数据源和数据消费者串起来，我们现在看看这条通道是怎么创建起来的。
我们先按照经验猜一下（其实我是看了代码的，冒充猜一下）：
起首，创建VcmCapture，因为它持有数据源VideoCaptureModule和分发器VideoBroadcaster，就可以把数据生产者和消费者连起来，形成通路；
有了VcmCapture，先给它创建数据源source，由于数据源CaptureTrackSource必要通过VideoTrack管理，因此，我们创建CaptureTrackSource之后，必要再创建一个VideoTrack，再将CaptureTrackSource交给VideoTrack持有，详细步骤：
将上面的CaptureTrackSource对象video_device作为参数传入，创建VideoTrack，这样VideoTrack就持有了CaptureTrackSource（看上面类图）；
至此，VcmCapture中就有了源，源产生的数据必要交给消费者，也就是VideoBroadcaster，然后VideoBroadcaster可以分发给本地渲染器和视频编码器；
三个重要对象创建好，就可以通过：VideoTrack -> CaptureTrackSource -> VcmCapture -> VideoBroadcaster 完成的链路搭建；
看代码：
代码入口：
// 代码路径：examples\peerconnection\client\conductor.cc
bool Conductor::InitializePeerConnection() {
// 创建PeerConnection部分省略...
// 添加track到PeerConnection中
AddTracks();
return peer_connection_ != nullptr;
}
复制代码
AddTracks内里会：
创建VcmCapture；
创建数据生产者；
创建数据消费者；
并将生产者到消费者的链路创建起来；
将VideoTrack这个管理者加入到PeerConnection当中；
已经删除非关键代码。
// 代码路径：examples\peerconnection\client\conductor.cc
void Conductor::AddTracks() {
if (!peer_connection_->GetSenders().empty()) {
return; // Already added tracks.
}
// 1、构建一个数据源 CaptureTrackSource （里面会创建 VcmCapturer ）
rtc::scoped_refptr<CapturerTrackSource> video_device = CapturerTrackSource::Create();
if (video_device) {
// 2、构建一个 VideoTrack , 返回其代理类
rtc::scoped_refptr<webrtc::VideoTrackInterface> video_track_(
peer_connection_factory_->CreateVideoTrack(kVideoLabel, video_device));
// 3、开始本地渲染
main_wnd_->StartLocalRenderer(video_track_);
// 4、将VideoTrack添加到PeerConnection当中管理
result_or_error = peer_connection_->AddTrack(video_track_, {kStreamId});
if (!result_or_error.ok()) {
RTC_LOG(LS_ERROR) << "Failed to add video track to PeerConnection: "
<< result_or_error.error().message();
}
}
main_wnd_->SwitchToStreamingUI();
}
复制代码
分开看下上面几个关键步骤：
1）创建CapturerTrackSource：
// 代码路径：examples\peerconnection\client\conductor.cc
static rtc::scoped_refptr<CapturerTrackSource> Create() {
const size_t kWidth = 640;
const size_t kHeight = 480;
const size_t kFps = 30;
std::unique_ptr<webrtc::test::VcmCapturer> capturer;
// 创建一个DeviceInfo对象，里面包含视频采集设备的属性信息
std::unique_ptr<webrtc::VideoCaptureModule::DeviceInfo> info(
webrtc::VideoCaptureFactory::CreateDeviceInfo());
if (!info) {
return nullptr;
}
// 获取采集设备数量（因为有些设备有多个摄像头），并遍历每个采集设备
int num_devices = info->NumberOfDevices();
for (int i = 0; i < num_devices; ++i) {
// 为每个采集设备创建VcmCapture，里面会实例化vcm对象 VideoCaptureImpl
capturer = absl::WrapUnique(webrtc::test::VcmCapturer::Create(kWidth, kHeight, kFps, i));
if (capturer) {
// 以VcmCapture为入参，创建CapturerTrackSource对象，并返回
return new rtc::RefCountedObject<CapturerTrackSource>(std::move(capturer));
}
}
return nullptr;
}
复制代码
上面就是创建了一个CapturerTrackSource对象，为什么我说是一个呢？因为，纵然你有多个摄像头，找到第一个可用的，并创建了CapturerTrackSource就返回了。并且，在创建CapturerTrackSource对象的时间传入了一个VcmCapture对象，并持有了。这个VcmCapture内里又会创建详细的数据源采集类对象，即VideoCaptureImpl类型的capturer，由于人脑栈有限，先不深究capturer如何创建的，继续回头看主干，也就是Conductor::AddTracks()函数。
至此，拉皮条的VcmCapture有了，CapturerTrackSource有了，数据源VideoCaptureImpl有了，记着我们的目的是创建链路，那么还必要创建管理者VideoTrack，以及数据分发器VideoBroadcaster。
2）创建VideoTrack：
下面代码就不是examples，了是webrtc内核代码了。
// 代码路径：pc\peer_connection_factory.cc
rtc::scoped_refptr<VideoTrackInterface> PeerConnectionFactory::CreateVideoTrack(
const std::string& id,
VideoTrackSourceInterface* source) {
RTC_DCHECK(signaling_thread()->IsCurrent());
// 构建一个VideoTrack对象
rtc::scoped_refptr<VideoTrackInterface> track(VideoTrack::Create(id, source, worker_thread()));
return VideoTrackProxy::Create(signaling_thread(), worker_thread(), track);
}
复制代码
留意是工作线程，一定要记着本身在哪个线程实行。还有，返回的是一个VideoTrack的代理类。
// 代码路径：pc\video_track.cc
rtc::scoped_refptr<VideoTrack> VideoTrack::Create(
const std::string& id,
VideoTrackSourceInterface* source,
rtc::Thread* worker_thread) {
// 创建了一个带有引用计数的VideoTrack对象，并返回了指针
rtc::RefCountedObject<VideoTrack>* track =
new rtc::RefCountedObject<VideoTrack>(id, source, worker_thread);
return track;
}
复制代码
不明白这个智能指针的，可以去看看我的另外一篇博客：https://blog.csdn.net/Ziwubiancheng/article/details/142985264?spm=1001.2014.3001.5501
3）创建链路：
至此，我们创建好了VcmCapture，并且创建好了详细数的数据采集类VideoCaptureImpl，详细的数据分发器VideoBroadcaster，以及其管理者VideoTrack。那么，管理者VideoTrack什么时间（when），在哪儿（where），通过何种方式（how），创建了什么样（what）的数据链路呢？我们详细分析下：
起首，有两条链路，想想之前哪个视频PIP画面，因此，必要一条本地渲染链路，以及一条远端渲染链路。
a）本地渲染链路：
入口就在：Conductor::AddTracks()的main_wnd_->StartLocalRenderer(video_track_);
// 代码路径：examples\peerconnection\client\main_wnd.cc
// 开始本地渲染
void MainWnd::StartLocalRenderer(webrtc::VideoTrackInterface* local_video) {
// VideoRenderer 构造函数里面会调用 AddOrUpdateSink，一路调用到 VideoBroadcaster 当中
// 这个 local_video 是一个 VideoTrack 对象
local_renderer_.reset(new VideoRenderer(handle(), 1, 1, local_video));
}
复制代码
看看VideoRenderer构造函数：
// 代码路径：examples\peerconnection\client\main_wnd.cc
MainWnd::VideoRenderer::VideoRenderer(
HWND wnd,
int width,
int height,
webrtc::VideoTrackInterface* track_to_render)
: wnd_(wnd), rendered_track_(track_to_render) {
::InitializeCriticalSection(&buffer_lock_);
ZeroMemory(&bmi_, sizeof(bmi_));
bmi_.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
bmi_.bmiHeader.biPlanes = 1;
bmi_.bmiHeader.biBitCount = 32;
bmi_.bmiHeader.biCompression = BI_RGB;
bmi_.bmiHeader.biWidth = width;
bmi_.bmiHeader.biHeight = -height;
bmi_.bmiHeader.biSizeImage =
width * height * (bmi_.bmiHeader.biBitCount >> 3);
// 这是一个 VideoTrack 对象，会将this(渲染器)添加到 VideoTrack 当中
rendered_track_->AddOrUpdateSink(this, rtc::VideoSinkWants());
}
复制代码
再进去看看这个AddOrUpdateSink方法（留意看上面的VideoRenderer对于VideoTrack来说就是个sink）
// 代码路径：pc\video_track.cc
// AddOrUpdateSink and RemoveSink should be called on the worker
// thread.
void VideoTrack::AddOrUpdateSink(rtc::VideoSinkInterface<VideoFrame>* sink,
const rtc::VideoSinkWants& wants) {
RTC_DCHECK(worker_thread_->IsCurrent());
VideoSourceBase::AddOrUpdateSink(sink, wants);
rtc::VideoSinkWants modified_wants = wants;
modified_wants.black_frames = !enabled();
// video_source_ 是 CapturerTrackSource
video_source_->AddOrUpdateSink(sink, modified_wants);
}
复制代码
CaptureTrackSource是VideoTrackSource的子类，因此，会去调用VideoTrackSource：
// 代码路径：pc\video_track_source.cc
void VideoTrackSource::AddOrUpdateSink(
rtc::VideoSinkInterface<VideoFrame>* sink,
const rtc::VideoSinkWants& wants) {
RTC_DCHECK(worker_thread_checker_.IsCurrent());
// 直接调用 source里面的方法，这个source是 TestVideoCapturer
source()->AddOrUpdateSink(sink, wants);
}
复制代码
进去TestVideoCapturer看看：
void TestVideoCapturer::AddOrUpdateSink(
rtc::VideoSinkInterface<VideoFrame>* sink,
const rtc::VideoSinkWants& wants) {
broadcaster_.AddOrUpdateSink(sink, wants);
UpdateVideoAdapter();
}
复制代码
TestVideoCapturer内里又调用了broadcaster的Add方法（记着我们设置渲染器的目的，就是终极设置给broadcaster）
看看VideoBroadcaster内里做了啥：
// 代码路径：media\base\video_broadcaster.cc
void VideoBroadcaster::AddOrUpdateSink(
VideoSinkInterface<webrtc::VideoFrame>* sink,
const VideoSinkWants& wants) {
RTC_DCHECK(sink != nullptr);
webrtc::MutexLock lock(&sinks_and_wants_lock_);
if (!FindSinkPair(sink)) {
// |Sink| is a new sink, which didn't receive previous frame.
previous_frame_sent_to_all_sinks_ = false;
}
// 又调用了基类的方法
VideoSourceBase::AddOrUpdateSink(sink, wants);
UpdateWants();
}
复制代码
进去看看其基类VideoSourceBase：
// 代码路径：media\base\video_source_base.cc
void VideoSourceBase::AddOrUpdateSink(
VideoSinkInterface<webrtc::VideoFrame>* sink,
const VideoSinkWants& wants) {
RTC_DCHECK(sink != nullptr);
SinkPair* sink_pair = FindSinkPair(sink);
if (!sink_pair) {
// 直接放到成员变量sinks里面了
sinks_.push_back(SinkPair(sink, wants));
} else {
sink_pair->wants = wants;
}
}
复制代码
由于之前
VideoCaptureModule
已经将采集模块启动起来了，因此，数据就远远不断的从VideoCaptureModule进入到VcmCapture，然后，再进入VideoBroadCaster当中，broadcaster就会给sinks内里全部成员发一份数据。
至此，本地渲染链路就启动起来了。至于，拿到这些数据如何渲染到屏幕上，后续再分析。
b）远端发送链路：
上面创建了本地渲染链路，那么数据分发器VideoBroadCaster内里通常还会编码发送给远端。
对于VideoStreamEncoder，在媒体协商之后，通过VideoTrack将VideoStreamEncoder添加到VideoBroadcaster当中。这样在VideoBroadcaster当中就有两个输出端。本地渲染器和Encoder；
编码后的数据通过PacedSender传给网络传输模块；
由于大量媒体协商的内容在之前介绍过，我们就看下调用栈，关注我们视频相关内容即可。
调用栈：
AdaptedVideoTrackSource::AddOrUpdateSink
VideoSourceSinkController::SetSource
VideoStreamEncoder::SetSource
VideoSendStream::SetSource
WebRtcVideoChannel::WebRtcVideoSendStream::RecreateWebRtcStream
WebRtcVideoChannel::WebRtcVideoSendStream::SetCodec
WebRtcVideoChannel::WebRtcVideoSendStream::SetSendParameters
WebRtcVideoChannel::ApplyChangedParams
WebRtcVideoChannel::SetSendParameters(应用获取到的编码参数设置)
VideoChannel::SetRemoteContent_w
// 切换到工作线程
BaseChannel::SetRemoteContent
SdpOfferAnswerHandler::PushdownMediaDescription（根据SDP媒体部分的描述，更新内部对象）
SdpOfferAnswerHandler::UpdateSessionState（更新媒体协商状态机、媒体流、编解码器）
SdpOfferAnswerHandler::ApplyRemoteDescription
SdpOfferAnswerHandler::SetRemoteDescription
PeerConnection::SetRemoteDescription
复制代码
AdaptedVideoTrackSource::AddOrUpdateSink当中：
// 代码路径：media\base\adapted_video_track_source.cc
void AdaptedVideoTrackSource::AddOrUpdateSink(
rtc::VideoSinkInterface<webrtc::VideoFrame>* sink,
const rtc::VideoSinkWants& wants) {
// 添加到broadcaster当中了
broadcaster_.AddOrUpdateSink(sink, wants);
OnSinkWantsChanged(broadcaster_.wants());
}
复制代码
至此，SetRemoteDescription的时间就将视频编码器添加进去视频分发器VideoBroadcaster了。
c）视频数据流动：
那么，视频数据毕竟是如何进入到视频分发器VideoBroadcaster的呢？思路如下：
我们通过DirectShow采集到摄像头的视频数据之后，会通过Receive函数进入；
终极走到VcmCapture::OnFrame，再调用其父类TestVideoCapturer的OnFrame；
我们知道TestVideoCapture内里包含VideoBroadcaster，就可以通过它进行分发了；
详细调用栈如下：
TestVideoCapturer::OnFrame
VcmCapturer::OnFrame
VideoCaptureImpl::DeliverCapturedFrame
VideoCaptureImpl::IncomingFrame
CaptureSinkFilter::ProcessCapturedFrame
CaptureInputPin::Receive
复制代码
看看详细函数：
// 代码路径：modules\video_capture\windows\sink_filter_ds.cc
/**
* 接收采集到的视频数据时候，首先会进入到这儿
* @param media_sample：就是采集到的数据
*/
STDMETHODIMP CaptureInputPin::Receive(IMediaSample* media_sample) {
RTC_DCHECK_RUN_ON(&capture_checker_);
// 通过Filter()获取到这个pin所属的filter，也就是sinkFilter
CaptureSinkFilter* const filter = static_cast<CaptureSinkFilter*>(Filter());
// 收到数据之后调用这个方法将数据从pin传给filter
filter->ProcessCapturedFrame(sample_props.pbBuffer, sample_props.lActual,
resulting_capability_);
return S_OK;
}
复制代码
// 代码路径：modules\video_capture\video_capture_impl.cc
/**
* 通过 SinkFilter 获取到数据之后，会调用此函数，
* 这个函数会将采集到的数据统一转换为I420格式的数据（因为用户request的格式是I420）
*/
int32_t VideoCaptureImpl::IncomingFrame(uint8_t* videoFrame,
size_t videoFrameLength,
const VideoCaptureCapability& frameInfo,
int64_t captureTime /*=0*/) {
// 由于我们最终采集的数据肯定是YUV，下面计算一些YUV相关的参数
int stride_y = width;
int stride_uv = (width + 1) / 2;
int target_width = width;
int target_height = abs(height);
// SetApplyRotation doesn't take any lock. Make a local copy here.
// 采集到数据帧是否进行了旋转
bool apply_rotation = apply_rotation_;
// 如果进行了旋转，那么，还要旋转回来
if (apply_rotation) {
// Rotating resolution when for 90/270 degree rotations.
if (_rotateFrame == kVideoRotation_90 ||
_rotateFrame == kVideoRotation_270) {
target_width = abs(height);
target_height = width;
}
}
// Setting absolute height (in case it was negative).
// In Windows, the image starts bottom left, instead of top left.
// Setting a negative source height, inverts the image (within LibYuv).
// TODO(nisse): Use a pool?
// 由于我们采集的数据不是I420，因此我们分配个I420的buffer，将数据转换为I420
rtc::scoped_refptr<I420Buffer> buffer = I420Buffer::Create(
target_width, target_height, stride_y, stride_uv, stride_uv);
libyuv::RotationMode rotation_mode = libyuv::kRotate0;
if (apply_rotation) {
switch (_rotateFrame) {
case kVideoRotation_0:
rotation_mode = libyuv::kRotate0;
break;
case kVideoRotation_90:
rotation_mode = libyuv::kRotate90;
break;
case kVideoRotation_180:
rotation_mode = libyuv::kRotate180;
break;
case kVideoRotation_270:
rotation_mode = libyuv::kRotate270;
break;
}
}
// 通过libyuv的方法将数据转换成I420
const int conversionResult = libyuv::ConvertToI420(
videoFrame, videoFrameLength, buffer.get()->MutableDataY(),
buffer.get()->StrideY(), buffer.get()->MutableDataU(),
buffer.get()->StrideU(), buffer.get()->MutableDataV(),
buffer.get()->StrideV(), 0, 0, // No Cropping
width, height, target_width, target_height, rotation_mode,
ConvertVideoType(frameInfo.videoType));
if (conversionResult < 0) {
RTC_LOG(LS_ERROR) << "Failed to convert capture frame from type "
<< static_cast<int>(frameInfo.videoType) << "to I420.";
return -1;
}
// 将转换后的数据重新封装成一个 VideoFrame 格式
VideoFrame captureFrame =
VideoFrame::Builder()
.set_video_frame_buffer(buffer)
.set_timestamp_rtp(0)
.set_timestamp_ms(rtc::TimeMillis())
.set_rotation(!apply_rotation ? _rotateFrame : kVideoRotation_0)
.build();
captureFrame.set_ntp_time_ms(captureTime);
// 里面会调用 RegisterCaptureDataCallback 的onFrame，将数据传给onFrame函数
DeliverCapturedFrame(captureFrame);
return 0;
}
复制代码
重点关注最后的DeliverCapturedFrame函数
// 代码路径：modules\video_capture\video_capture_impl.cc
/**
* 里面会调用 RegisterCaptureDataCallback 的onFrame，将数据传给onFrame函数
*/
int32_t VideoCaptureImpl::DeliverCapturedFrame(VideoFrame& captureFrame) {
UpdateFrameCount(); // frame count used for local frame rate callback.
if (_dataCallBack) {
_dataCallBack->OnFrame(captureFrame);
}
return 0;
}
复制代码
然后就到了VcmCpaturer
// 接收采集到视频数据（格式已经转换成用户请求的了）
void VcmCapturer::OnFrame(const VideoFrame& frame) {
TestVideoCapturer::OnFrame(frame);
}
复制代码
到了熟悉的TestVideoCapturer
/**
* 从 VcmCapturer::OnFrame 抛上来的
*/
void TestVideoCapturer::OnFrame(const VideoFrame& original_frame) {
int cropped_width = 0;
int cropped_height = 0;
int out_width = 0;
int out_height = 0;
// 对原始视频帧进行处理（比如你加一些特效）
VideoFrame frame = MaybePreprocess(original_frame);
if (out_height != frame.height() || out_width != frame.width()) {
// 缩放部分省略...
} else {
// 如果不需要缩放，那么直接交给 VideoBroadcaster 进行分发
// No adaptations needed, just return the frame as is.
broadcaster_.OnFrame(frame);
}
}
复制代码
这样，就通过broadcaster分发给其内部已经添加的sink了。
五、总结：
本章主要介绍了视频数据采集的关键类VcmCapture、VideoTrack、VideoBroadcaster，VideoCapture。并且交接了这几个类的主要职责，以及如何利用他们创建一条数据链路的。后续，对详细的引擎再做分析。
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。
欢迎光临 ToB企服应用市场:ToB评测及商务社交产业平台 (https://dis.qidao123.com/)
Powered by Discuz! X3.4
