FFMPEG录屏（1）----录制桌面

柔情只为你懂 2023-07-13 10:39 23阅读 0赞

[https://blog.csdn.net/peilinok/article/details/102981560][https_blog.csdn.net_peilinok_article_details_102981560]

首先录制桌面有很多种方法，原生windows api进行GDI抓屏，Mirror，Direct等，本文针对ffmpeg gdi抓屏进行介绍，也开始录屏软件开发之旅。

准备ffmpeg，无所谓是动态库、静态库，请自行前往下载并引入工程。

在本系列文章中，一些FFMPEG的结构、函数并不会做过多说明，请自行了解，重点放在了桌面录制的整体流程。

参考资料：

ffmpeg 源代码简单分析

初始化FFMPEG

av\_register\_all();  
avdevice\_register\_all();  
准备屏幕参数，包含了帧率、录制区域的起始坐标、大小、是否绘制鼠标等参数。

char buff\_video\_size\[50\] = \{ 0 \};  
sprintf\_s(buff\_video\_size, 50, "%dx%d", rect.right - rect.left, rect.bottom - rect.top);  
  
AVDictionary \*options = NULL;  
av\_dict\_set\_int(&options, "framerate", fps, AV\_DICT\_MATCH\_CASE);  
av\_dict\_set\_int(&options, "offset\_x", rect.left, AV\_DICT\_MATCH\_CASE);  
av\_dict\_set\_int(&options, "offset\_y", rect.top, AV\_DICT\_MATCH\_CASE);  
av\_dict\_set(&options, "video\_size", buff\_video\_size, AV\_DICT\_MATCH\_CASE);  
av\_dict\_set\_int(&options, "draw\_mouse", 1, AV\_DICT\_MATCH\_CASE);  
创建一个AVformatContext句柄，这个结构了不得，将会贯穿整个录屏软件的所有模块，描述了众多音视频信息。

AVFormatContext \*\_fmt\_ctx = avformat\_alloc\_context();  
获取gdigrab

AVInputFormat \*\_input\_fmt = av\_find\_input\_format("gdigrab");  
打开gidgrab，这时候就要把准备好的捕获区域等配置信息一同传入api

avformat\_open\_input(&\_fmt\_ctx, "desktop", \_input\_fmt, &options);  
获取桌面流信息和索引

avformat\_find\_stream\_info(\_fmt\_ctx, NULL);  
int stream\_index = -1;  
for (int i = 0; i < \_fmt\_ctx->nb\_streams; i++) \{  
if (\_fmt\_ctx->streams\[i\]->codec->codec\_type == AVMEDIA\_TYPE\_VIDEO) \{  
stream\_index = i;  
break;  
\}  
\}  
请务必保存上面获取的索引，他将用于后面的捕获线程。

为gidgrab创建对应的解码器，这一点在后续的声音录制中也很重要，可以通过解码器获取数据的基本信息，如数据格式、采样率等等。

AVCodecContext \*\_codec\_ctx = \_fmt\_ctx->streams\[stream\_index\]->codec;  
AVCodec \*\_codec = avcodec\_find\_decoder(\_codec\_ctx->codec\_id);  
查找到对应的解码器后，打开或初始化解码器。

avcodec\_open2(\_codec\_ctx, \_codec, NULL);  
至此我们的初始化工作已经完成，这其中的API调用请务必判断所有的返回值，并进行相应的错误处理。最后，在初始化函数中进行最后一步，释放存放配置信息的字典。

av\_dict\_free(&options);  
接下来就是我们的第一个录制线程了，请以喜欢或者习惯的方式启动一个线程。

为捕获出来的数据初始化两个结构体，AVFrame和AVPacket,这两个结构体也将在各种捕获、压缩、转码过程中出现，最好能读一下结构体介绍。

AVPacket \*packet = (AVPacket\*)av\_malloc(sizeof(AVPacket));  
AVFrame \*frame\_rgb = av\_frame\_alloc();  
随后就是一个while循环进行屏幕数据的捕获以及解码。注意av\_read\_frame函数将会在某些情况下阻塞，例如读取网络流时，在本例中他将尽可能的按照我们所设定的帧率返回桌面图像数据。

int got\_pic = 0;  
while (\_running == true) \{  
ret = av\_read\_frame(\_fmt\_ctx, packet);  
  
if (ret < 0) \{  
al\_fatal("read frame failed:%d", ret);  
break;  
\}  
  
if (packet->stream\_index == \_stream\_index) \{  
ret = avcodec\_decode\_video2(\_codec\_ctx, frame\_rgb, &got\_pic, packet);  
if (ret < 0) \{  
if (\_on\_error) \_on\_error(AE\_FFMPEG\_DECODE\_FRAME\_FAILED);  
al\_fatal("decode desktop frame failed");  
break;  
\}  
  
if (got\_pic) \{  
//do something here with frame\_rgba  
\}  
\}  
  
av\_free\_packet(packet);  
\}  
  
av\_free(frame\_rgb);  
以上就是通过ffmpeg进行gdi抓屏的全部过程了。下一篇将对桌面数据压缩进行介绍。

CSDN源码下载

GitHub持续更新地址

————————————————  
版权声明：本文为CSDN博主「ssslar」的原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接及本声明。  
原文链接：https://blog.csdn.net/peilinok/article/details/102981560

[https_blog.csdn.net_peilinok_article_details_102981560]: https://blog.csdn.net/peilinok/article/details/102981560