手把手教你用FFmpeg根据时间戳定位当前帧获取Bitmap

2024-01-13 13:37:30

使用 FFmpeg 从视频中提取关键帧图像

定位视频中的关键帧

随着互联网视频的爆发式增长，视频处理技术变得越来越重要。FFmpeg 是一款功能强大的开源多媒体框架，广泛应用于视频处理领域。本文将指导您使用 FFmpeg API，根据时间戳定位视频中的关键帧，并获取对应的 Bitmap。这对于从视频中提取图像、创建视频缩略图等应用场景非常有用。

首先，打开视频文件并获取视频流信息。FFmpeg 的 avformat_open_input() 和 avformat_find_stream_info() 函数可用于此目的。接下来，使用 av_seek_frame() 函数定位关键帧，该函数基于指定的时间戳。

获取关键帧的 Bitmap

定位关键帧后，可以通过 av_read_frame() 函数获取对应的 Bitmap。它将返回一个 AVPacket 结构，包含有关帧的信息。要解码帧并将其转换为 Bitmap，可以使用 avcodec_decode_video2() 和 sws_getContext() 函数。

以下是示例代码，展示如何使用 FFmpeg API 定位关键帧并获取 Bitmap：

// 打开视频文件
AVFormatContext* fmt_ctx = avformat_open_input("test.mp4", NULL, NULL);

// 获取视频流信息
avformat_find_stream_info(fmt_ctx, NULL);

// 定位第一个关键帧
av_seek_frame(fmt_ctx, 0, 0, AVSEEK_FLAG_BACKWARD);

// 获取关键帧的 Bitmap
AVPacket pkt;
av_read_frame(fmt_ctx, &pkt);
AVCodecContext* codec_ctx = fmt_ctx->streams[0]->codec;
AVFrame* frame = av_frame_alloc();
avcodec_decode_video2(codec_ctx, frame, NULL, &pkt);
SwsContext* sws_ctx = sws_getContext(codec_ctx->width, codec_ctx->height, codec_ctx->pix_fmt, codec_ctx->width, codec_ctx->height, AV_PIX_FMT_RGB24, SWS_BICUBIC, NULL, NULL, NULL);
uint8_t* rgb_frame_buffer = (uint8_t*)av_malloc(codec_ctx->width * codec_ctx->height * 3);
avpicture_fill((AVPicture*)frame, rgb_frame_buffer, AV_PIX_FMT_RGB24, codec_ctx->width, codec_ctx->height);
sws_scale(sws_ctx, (const uint8_t* const*)frame->data, frame->linesize, 0, codec_ctx->height, frame->data, frame->linesize);
SDL_Surface* surface = SDL_CreateRGBSurfaceFrom(rgb_frame_buffer, codec_ctx->width, codec_ctx->height, 24, codec_ctx->width * 3, 0x0000FF, 0x00FF00, 0xFF0000, 0);
SDL_SaveBMP(surface, "frame.bmp");