FFMPEG Tips (3) 如何读取每一帧的信息

2023-05-15,,

本文是我的《FFMPEG Tips》系列的第三篇文章,上篇文章 介绍了如何提取整个音视频码流的媒体信息,包括:封装格式、编码格式、视频的分辨率、帧率、码率、音频的采样率、位宽、通道数等等,而本文则关注得更细一点,看看如何利用 ffmpeg 读取码流中每一帧的信息。

1.  码流中每一帧的哪些信息值得关注 ?


[ ]  音频帧还是视频帧

[ ]  关键帧还是非关键帧

[ ]  帧的数据和大小

[ ]  时间戳信息

2.  为什么要关注这些信息 ?

[ ]  音频帧还是视频帧 -> 分别送入音频/视频×××

[ ]  关键帧还是非关键帧 -> 追帧优化

[ ]  帧的数据和大小 -> 取出帧的内容

[ ]  时间戳信息 -> 音视频同步

3. 如何从 ffmpeg 取出这些信息 ?


ffmpeg 提供了一个函数 av_read_frame 来完成解封装的过程,它会从码流里面提取每一个音频、视频帧,它使用了结构体 AVPacket 来记录每一帧的信息。

读取一帧数据的代码示例如下(ic 即为 AVFormatContext 对象,码流的上下文句柄):

AVPacket avpkt;
av_init_packet(&avpkt);

while (!interrupt) {
    int ret = av_read_frame(ic, &avpkt);
    if (ret < 0) {
        break;
    }
    // processing
}

av_free_packet(&avpkt);

每循环一次,就从码流中解封装并且提取了一帧数据,并存放在了 AVPacket 结构体中。

3.1 如何判断是音频帧还是视频帧

上一篇文章我们提到过,使用下面的方法,获取码流中的 video_stream_idx 和 audio_stream_idx

int video_stream_idx = av_find_best_stream(ic, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, 0);

int audio_stream_idx = av_find_best_stream(ic, AVMEDIA_TYPE_AUDIO, -1, -1, NULL, 0);

那么,此时就派上用场了,每一个 AVPacket 都有一个成员变量:stream_index,由该成员变量即可判断这个 Packet 到底是音频还是视频了:

if (avpkt.stream_index == video_stream_idx) {
    LOGD("read a video frame");
} else if (avpkt.stream_index == audio_stream_idx) {
    LOGD("read audio frame);
}

3.2 如何判断是否为关键帧

判断是否为关键帧的方法也比较简单,示例如下:

if (avpkt.flags & AV_PKT_FLAG_KEY) {
    LOGD("read a key frame");
}

3.3 如何获取帧的数据和大小

帧的数据和大小直接定义在 AVPacket 结构体中,对应的成员变量如下:

// 压缩编码的数据,一帧音频/视频
uint8_t *data;

// 数据的大小
int size;

3.4 如何获取帧的时间戳信息


每一个帧都可能携带有 2 个时间戳信息,一个是解码时间戳 dts,一个是显示时间戳 pts,解码时间戳告诉我们什么时候需要解码,显示时间戳告诉我们什么时候需要显示,只有在码流中存在 B 帧的情况下,这两个时间戳才会不一致。

这些时间戳信息不一定存在于码流中(取决于生产端),如果不存在,则其值为:AV_NOPTS_VALUE

一定要选择正确地方式打印时间戳,时间戳是使用 long long 来表示的,即 int64_t,因此打印的时候,需要使用 “%lld” 来打印,例如:

while (!interrupt) {
    int ret = av_read_frame(player->ic, &avpkt);
    if (ret < 0) {
        break;
    }
    if (avpkt.stream_index == video_stream_idx) {
        LOGD("read video frame, timestamp = %lld \n”, avpkt.pts);
    } else if (avpkt.stream_index == audio_stream_idx) {
        LOGD("read audio frame, timestamp = %lld \n”, avpkt.pts);
    }
}

由此,我们就可以通过这些 log 信息调试一下某一段音视频流的时间戳是否正确,比如是否出现了时间戳的回滚和错乱,则必然会导致播放端出现音视频不同步或者显示异常等情况。

4. 小结

关于如何使用 FFMPEG 如何读取每一帧的信息就介绍到这儿了,文章中有不清楚的地方欢迎留言或者来信 lujun.hust@gmail.com 交流,关注我的新浪微博 @卢_俊 或者 微信公众号 @Jhuster 获取最新的文章和资讯。