音视频入门基础:MPEG2-TS专题(6)——FFmpeg源码中,获取MPEG2-TS传输流每个transport packet长度的实现

一、引言

从《音视频入门基础:MPEG2-TS专题(3)------TS Header简介》可以知道,TS格式有三种:分别为transport packet长度固定为188、192和204字节。而FFmpeg源码中是通过get_packet_size函数获取一段MPEG2-TS传输流中每个transport packet的长度,从而判断其是哪种TS格式的。

二、get_packet_size函数的定义

get_packet_size函数定义在FFmpeg源码(本文演示用的FFmpeg源码版本为7.0.1)的源文件libavformat/mpegts.c中:

cpp 复制代码
/* autodetect fec presence */
static int get_packet_size(AVFormatContext* s)
{
    int score, fec_score, dvhs_score;
    int margin;
    int ret;

    /*init buffer to store stream for probing */
    uint8_t buf[PROBE_PACKET_MAX_BUF] = {0};
    int buf_size = 0;
    int max_iterations = 16;

    while (buf_size < PROBE_PACKET_MAX_BUF && max_iterations--) {
        ret = avio_read_partial(s->pb, buf + buf_size, PROBE_PACKET_MAX_BUF - buf_size);
        if (ret < 0)
            return AVERROR_INVALIDDATA;
        buf_size += ret;

        score      = analyze(buf, buf_size, TS_PACKET_SIZE,      0);
        dvhs_score = analyze(buf, buf_size, TS_DVHS_PACKET_SIZE, 0);
        fec_score  = analyze(buf, buf_size, TS_FEC_PACKET_SIZE,  0);
        av_log(s, AV_LOG_TRACE, "Probe: %d, score: %d, dvhs_score: %d, fec_score: %d \n",
            buf_size, score, dvhs_score, fec_score);

        margin = mid_pred(score, fec_score, dvhs_score);

        if (buf_size < PROBE_PACKET_MAX_BUF)
            margin += PROBE_PACKET_MARGIN; /*if buffer not filled */

        if (score > margin)
            return TS_PACKET_SIZE;
        else if (dvhs_score > margin)
            return TS_DVHS_PACKET_SIZE;
        else if (fec_score > margin)
            return TS_FEC_PACKET_SIZE;
    }
    return AVERROR_INVALIDDATA;
}

该函数的作用是:读取s->pb表示的AVIOContext输入缓冲区中存贮的MPEG2-TS传输流数据,判断其每个transport packet长度。

形参s:既是输入型参数也是输出型参数。指向一个AVFormatContext类型变量。

返回值:返回TS_PACKET_SIZE(188)表示这段MPEG2-TS传输流中每个transport packet长度为188字节;返回TS_DVHS_PACKET_SIZE(192)表示transport packet长度为192字节;返回TS_FEC_PACKET_SIZE(204)表示transport packet长度为204字节;返回AVERROR_INVALIDDATA表示出错。

三、get_packet_size函数的内部实现分析

get_packet_size函数中,首先定义一个数组buf用来存贮需要被检测的MPEG2-TS传输流:

cpp 复制代码
    /*init buffer to store stream for probing */
    uint8_t buf[PROBE_PACKET_MAX_BUF] = {0};
    int buf_size = 0;
    int max_iterations = 16;

通过avio_read_partial函数读取需要被检测的MPEG2-TS传输流,存贮到"buf + buf_size"指向的缓冲区中。关于avio_read_partial函数的用法可以参考:《FFmpeg源码:avio_read_partial函数分析》:

cpp 复制代码
        ret = avio_read_partial(s->pb, buf + buf_size, PROBE_PACKET_MAX_BUF - buf_size);
        if (ret < 0)
            return AVERROR_INVALIDDATA;
        buf_size += ret;

检测这段码流符合transport packet长度固定为188字节的TS格式的程度,将对应的分数赋值给变量score。关于analyze函数的用法可以参考:《音视频入门基础:MPEG2-TS专题(5)------FFmpeg源码中,判断某文件是否为TS文件的实现》:

cpp 复制代码
     score      = analyze(buf, buf_size, TS_PACKET_SIZE,      0);

检测这段码流符合transport packet长度固定为192字节的TS格式的程度,将对应的分数赋值给变量dvhs_score:

cpp 复制代码
        dvhs_score = analyze(buf, buf_size, TS_DVHS_PACKET_SIZE, 0);

检测这段码流符合transport packet长度固定为204字节的TS格式的程度,将对应的分数赋值给变量fec_score:

cpp 复制代码
        fec_score  = analyze(buf, buf_size, TS_FEC_PACKET_SIZE,  0);

通过mid_pred函数取score、fec_score、dvhs_score的中位数,赋值给变量margin。关于mid_pred函数的用法可以参考:《FFmpeg源码:mid_pred函数分析》:

cpp 复制代码
        margin = mid_pred(score, fec_score, dvhs_score);

将score、fec_score、dvhs_score分别和它们的中位数进行比较,得到它们的最高分。根据这个最高分,判断这段码流属于哪种TS格式。返回这段MPEG2-TS传输流中每个transport packet的长度:

cpp 复制代码
        if (score > margin)
            return TS_PACKET_SIZE;
        else if (dvhs_score > margin)
            return TS_DVHS_PACKET_SIZE;
        else if (fec_score > margin)
            return TS_FEC_PACKET_SIZE;
相关推荐
美狐美颜sdk5 小时前
跨平台直播美颜SDK集成实录:Android/iOS如何适配贴纸功能
android·人工智能·ios·架构·音视频·美颜sdk·第三方美颜sdk
melonbo9 小时前
使用FFmpeg将H.264码流封装为MP4
ffmpeg·音视频·h.264
aqi0011 小时前
FFmpeg开发笔记(七十七)Android的开源音视频剪辑框架RxFFmpeg
android·ffmpeg·音视频·流媒体
慢行的骑兵13 小时前
Android音视频探索之旅 | CMake基础语法 && 创建支持Ffmpeg的Android项目
ffmpeg·音视频
Just_Paranoid13 小时前
华为云Flexus+DeepSeek征文|基于Dify构建音视频内容转录工作流
华为云·音视频·dify·maas·deepseek·flexusx
go546315846514 小时前
修改Spatial-MLLM项目,使其专注于无人机航拍视频的空间理解
人工智能·算法·机器学习·架构·音视频·无人机
叹一曲当时只道是寻常17 小时前
Softhub软件下载站实战开发(十):实现图片视频上传下载接口
golang·go·音视频
音视频牛哥19 小时前
计算机视觉的新浪潮:扩散模型(Diffusion Models)技术剖析与应用前景
人工智能·计算机视觉·ai·音视频·实时音视频·扩散模型
电子科技圈20 小时前
SmartDV推出先进的H.264和H.265视频编码器和解码器IP
音视频·h.265·h.264
feiyangqingyun21 小时前
Qt音视频开发技巧/推流带旋转角度/rtsprtmp推流/保存文件到MP4/拉流解析旋转角度
qt·音视频·qt旋转角度推流