当前位置：首页 > news >正文

C++实现视频截图功能

news 2026/3/27 6:46:40

大半夜的聊聊C++视频截图，顺便祭奠一下我那逝去的发际线

现在是凌晨3点15分，我那个破烂项目的代码还在编译，进度条卡在99%不动了，我心态崩了真的...趁着这会儿功夫，看到论坛里有个小白在问怎么用C++做视频截图。哎，本来不想说话的，但看到有人推荐用OpenCV我就坐不住了。兄弟，OpenCV那是给搞算法的人玩的玩具，真到了生产环境，还得是FFmpeg这个让人头秃的玩意儿。

说实话，每次碰FFmpeg的API，我都感觉自己在排雷。文档写得跟天书一样，甚至有时候我都怀疑写文档的人是不是故意不想让我们看懂...

FFmpeg：这玩意儿就是个黑洞

在C++里搞视频处理，那种感觉就像...在早高峰的八通线上试图优雅地吃完一碗刚出锅的热汤面。不仅烫嘴，还容易洒一身，周围全是挤你的人（并发线程）。你以为截个图很简单？打开文件、解封装、找流、找解码器、解码、颜色空间转换、保存... 这一套流程下来，你不掉两根头发都对不起祖师爷。

别跟我提什么“优雅的封装”，在视频流处理面前，优雅就是个伪命题。为了性能，我们甚至得写出一些自己都恶心的代码。

来，给你们看点“传家宝”代码

这段代码是我从一个三年前烂尾的项目里扒出来的。当时为了赶上线，那是昏天黑地地写，变量名有点随意，你们凑合看。别问我为什么有些地方逻辑很怪，问就是为了过编译，问就是为了不Crash！！！

extern "C" { #include <libavcodec/avcodec.h>#include <libavformat/avformat.h>#include <libswscale/swscale.h>}// 别问为什么用全局变量，当时脑子瓦特了，懒得改static AVFormatContext* g_fmt_ctx = NULL;void do_the_dirty_job(const char* filepath) { // 注册所有组件，新版本可能不需要了，但加上总没错，求个心理安慰 // av_register_all(); // 打开文件，要是这就失败了，建议直接砸键盘 if (avformat_open_input(&g_fmt_ctx, filepath, NULL, NULL) != 0) { printf("文件都打不开，玩个锤子！\n"); return; } // 找流信息，这步要是卡住，多半是网络IO的问题 if (avformat_find_stream_info(g_fmt_ctx, NULL) < 0) { printf("流信息呢？被狗吃了？\n"); return; } int video_stream_index = -1; // 遍历流，找到视频流。就像在一堆绿豆里找个红豆 for (unsigned int i = 0; i < g_fmt_ctx->nb_streams; i++) { if (g_fmt_ctx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) { video_stream_index = i; break; } } if (video_stream_index == -1) { printf("连视频流都没有，你给我个MP3让我截图？\n"); return; } AVCodecParameters* pLocalCodecParameters = g_fmt_ctx->streams[video_stream_index]->codecpar; const AVCodec* pLocalCodec = avcodec_find_decoder(pLocalCodecParameters->codec_id); // 就算是烂代码，也要检查空指针，这是底线 if (pLocalCodec == NULL) { printf("解码器找不到，散了吧...\n"); return; } AVCodecContext* pCodecContext = avcodec_alloc_context3(pLocalCodec); avcodec_parameters_to_context(pCodecContext, pLocalCodecParameters); avcodec_open2(pCodecContext, pLocalCodec, NULL); AVPacket* pPacket = av_packet_alloc(); AVFrame* pFrame = av_frame_alloc(); AVFrame* pFrameRGB = av_frame_alloc(); // 存转换后的数据 // 准备SwsContext，这玩意儿最耗性能，别在循环里甚至每一帧都创建销毁！ // 以前有个实习生这么干，服务器CPU直接飙到100%，我想把他祭天 struct SwsContext* sws_ctx = sws_getContext( pCodecContext->width, pCodecContext->height, pCodecContext->pix_fmt, pCodecContext->width, pCodecContext->height, AV_PIX_FMT_RGB24, // 转成RGB24，方便保存 SWS_BILINEAR, NULL, NULL, NULL ); int numBytes = av_image_get_buffer_size(AV_PIX_FMT_RGB24, pCodecContext->width, pCodecContext->height, 1); uint8_t* buffer_legacy_mess = (uint8_t*)av_malloc(numBytes * sizeof(uint8_t)); av_image_fill_arrays(pFrameRGB->data, pFrameRGB->linesize, buffer_legacy_mess, AV_PIX_FMT_RGB24, pCodecContext->width, pCodecContext->height, 1); int frame_count_tmp = 0; // 开始死循环读帧 while (av_read_frame(g_fmt_ctx, pPacket) >= 0) { if (pPacket->stream_index == video_stream_index) { // 发送包到解码器 if (avcodec_send_packet(pCodecContext, pPacket) == 0) { // 从解码器接收帧 while (avcodec_receive_frame(pCodecContext, pFrame) == 0) { // 只要第一帧！截完赶紧跑！ // 别问为什么不seek，seek不准，硬解才是王道（其实就是懒） if (frame_count_tmp == 50) { // 随便截第50帧，别太较真 sws_scale(sws_ctx, (uint8_t const* const*)pFrame->data, pFrame->linesize, 0, pCodecContext->height, pFrameRGB->data, pFrameRGB->linesize); // 这里应该写保存文件的代码，但我累了 // 你们自己用fwrite把 pFrameRGB->data 怼进文件里去吧 // 记得加PPM头或者用stb_image_write printf("截图成功！虽然不知道截了个啥...\n"); goto cleanup; // 没错，我用了goto，咬我啊 } frame_count_tmp++; } } } av_packet_unref(pPacket); }cleanup: // 内存泄漏？不存在的，系统回收也是回收嘛（开玩笑的，记得释放） av_free(buffer_legacy_mess); av_frame_free(&pFrameRGB); av_frame_free(&pFrame); av_packet_free(&pPacket); avcodec_close(pCodecContext); avformat_close_input(&g_fmt_ctx);}