简介
webrtc的modules中有一个模块desktop_capture,该模块负责截屏,目前只支持windows和mac平台,android,ios没有实现。
desktop_capture中有两种截屏方式,第一种是截单个窗口,叫做WindowCapturer,
第二种是截整个屏幕,叫做ScreenCapturer。
window_capture/screen_capture都继承于基类DesktopCapturer:
// Abstract interface for screen and window capturers.
class DesktopCapturer {
public:
//初始化截屏,设置数据回调
virtual void Start(Callback* callback) = 0;
//截一张图,数据直接进入callback
virtual void Capture(const DesktopRegion& region) = 0;
};
一、WindowCapture
WindowCapture主要增加了获取窗口列表,和设置截屏窗口id的接口:
virtual bool GetWindowList(WindowList* windows);
virtual bool SelectWindow(WindowId id);
二、ScreenCapture
ScreenCapture主要增加了获取屏幕列表,和设置截屏屏幕id的接口:
virtual bool GetScreenList(ScreenList* screens);
virtual bool SelectScreen(ScreenId id);
三、使用流程
接口都比较简单,很容易使用,大概的流程如下:
- 创建对象
- 初始化截屏,设置回调函数
- 开启线程循环截图
screen_capture_ = webrtc::ScreenCapturer::Create(webrtc::DesktopCaptureOptions::CreateDefault());
screen_capture_->SelectScreen(0);
bool ImageCaptureThreadFunc(void* param)
{
webrtc::DesktopCapturer* capture = static_cast<webrtc::DesktopCapturer*>(param);
capture->Capture(webrtc::DesktopRegion(webrtc::DesktopRect()));
Sleep(100);
return true;
}
四、截屏数据处理
截屏后得到的数据格式是rgb,需要使用libyuv将数据从rgb转换为yuv420,然后传入编码器和进行本地渲染。
转换时注意填写正确的原始数据类型,windows下格式为webrtc::kARGB
void OnCaptureCompleted(webrtc::DesktopFrame* frame) {
if (frame == NULL) {
//error,stop capture
StopImageCapture();
return;
}
int width = frame->size().width();
int height = frame->size().height();
int half_width = (width + 1) / 2;
webrtc::VideoFrame i420frame;
i420frame.CreateEmptyFrame(width, height, width, half_width, half_width);
int ret = webrtc::ConvertToI420(webrtc::kARGB, frame->data(), 0, 0, width, height, 4, webrtc::kVideoRotation_0, &i420frame);
if (ret != 0) {
return;
}
//后续处理
OnIncomingCapturedFrame(0, i420frame);
//需要释放内存
delete frame;
}
五、传递给videoSendStream
通过VideoSendStream的input接口可以把采集到的图像投递进去,编码发送。
video_stream_s_->Input()->IncomingCapturedFrame(videoFrame);