FFmpeg源码:ff_h2645_extract_rbsp函数分析

2024-06-20 20:44

本文主要是介绍FFmpeg源码:ff_h2645_extract_rbsp函数分析,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

一、ff_h2645_extract_rbsp函数的声明

ff_h2645_extract_rbsp函数的声明放在FFmpeg源码(本文演示用的FFmpeg源码版本为5.0.3,该ffmpeg在CentOS 7.5上通过10.2.1版本的gcc编译)的头文件libavcodec/h2645_parse.h中。

/*** Extract the raw (unescaped) bitstream.*/
int ff_h2645_extract_rbsp(const uint8_t *src, int length, H2645RBSP *rbsp,H2645NAL *nal, int small_padding);

该函数在H.264/H.265的解码时被调用。作用是将去掉第一个startcode的H.264/H.265码流(以下全部以H.264码流为例) 中的第一个NALU 提取出来,分别去掉和保留防竞争字节,存贮到形参nal 指向的缓冲区中。关于 NALU和防竞争字节的概念可以参考:音视频入门基础:H.264专题(3)——EBSP, RBSP和SODB

形参src:输入型参数。指向缓冲区的指针,该缓冲区存放 去掉第一个startcode(起始码)后的H.264码流。

形参length:输入型参数。指针src指向的缓冲区的长度,单位为字节。

形参rbsp为H2645RBSP类型,为输出型参数。

结构体H2645RBSP 定义如下:

typedef struct H2645RBSP {uint8_t *rbsp_buffer;AVBufferRef *rbsp_buffer_ref;int rbsp_buffer_alloc_size;int rbsp_buffer_size;
} H2645RBSP;

执行ff_h2645_extract_rbsp函数后,

rbsp->rbsp_buffer 变为:去掉startcode和防竞争字节后的H.264码流,可能包含多个NALU。

rbsp->rbsp_buffer_size 变为:rbsp->rbsp_buffer的大小,单位为字节。

形参nal为H2645NAL类型,为输出型参数。

结构体H2645NAL定义如下:

typedef struct H2645NAL {const uint8_t *data;int size;/*** Size, in bits, of just the data, excluding the stop bit and any trailing* padding. I.e. what HEVC calls SODB.*/int size_bits;int raw_size;const uint8_t *raw_data;GetBitContext gb;/*** NAL unit type*/int type;/*** H.264 only, nal_ref_idc*/int ref_idc;/*** HEVC only, nuh_temporal_id_plus_1 - 1*/int temporal_id;/** HEVC only, identifier of layer to which nal unit belongs*/int nuh_layer_id;int skipped_bytes;int skipped_bytes_pos_size;int *skipped_bytes_pos;
} H2645NAL;

执行ff_h2645_extract_rbsp函数后,

nal->data变为:指向缓冲区的指针。该缓冲区存放 “指针src指向的缓冲区中的第一个NALU”,该NALU去掉了startcode和防竞争字节,但保留了NALU Header。

nal->size变为:nal->data指向的缓冲区的大小,单位为字节。

nal->raw_data变为:指向缓冲区的指针。该缓冲区存放 “指针src指向的缓冲区中的第一个NALU”,该NALU去掉了startcode,但保留了防竞争字节和NALU Header。

nal->raw_size变为:nal->raw_data指向的缓冲区的大小,单位为字节。

形参small_padding:输入型参数。值一般等于1,可以忽略。

返回值:整形。值等同于nal->raw_size,为nal->raw_data指向的缓冲区的大小,单位为字节

二、ff_h2645_extract_rbsp函数的定义

ff_h2645_extract_rbsp函数定义在libavcodec/h2645_parse.c中:

int ff_h2645_extract_rbsp(const uint8_t *src, int length,H2645RBSP *rbsp, H2645NAL *nal, int small_padding)
{int i, si, di;uint8_t *dst;nal->skipped_bytes = 0;
#define STARTCODE_TEST                                                  \if (i + 2 < length && src[i + 1] == 0 && src[i + 2] <= 3) {     \if (src[i + 2] != 3 && src[i + 2] != 0) {                   \/* startcode, so we must be past the end */             \length = i;                                             \}                                                           \break;                                                      \}
#if HAVE_FAST_UNALIGNED
#define FIND_FIRST_ZERO                                                 \if (i > 0 && !src[i])                                           \i--;                                                        \while (src[i])                                                  \i++
#if HAVE_FAST_64BITfor (i = 0; i + 1 < length; i += 9) {if (!((~AV_RN64(src + i) &(AV_RN64(src + i) - 0x0100010001000101ULL)) &0x8000800080008080ULL))continue;FIND_FIRST_ZERO;STARTCODE_TEST;i -= 7;}
#elsefor (i = 0; i + 1 < length; i += 5) {if (!((~AV_RN32(src + i) &(AV_RN32(src + i) - 0x01000101U)) &0x80008080U))continue;FIND_FIRST_ZERO;STARTCODE_TEST;i -= 3;}
#endif /* HAVE_FAST_64BIT */
#elsefor (i = 0; i + 1 < length; i += 2) {if (src[i])continue;if (i > 0 && src[i - 1] == 0)i--;STARTCODE_TEST;}
#endif /* HAVE_FAST_UNALIGNED */if (i >= length - 1 && small_padding) { // no escaped 0nal->data     =nal->raw_data = src;nal->size     =nal->raw_size = length;return length;} else if (i > length)i = length;dst = &rbsp->rbsp_buffer[rbsp->rbsp_buffer_size];memcpy(dst, src, i);si = di = i;while (si + 2 < length) {// remove escapes (very rare 1:2^22)if (src[si + 2] > 3) {dst[di++] = src[si++];dst[di++] = src[si++];} else if (src[si] == 0 && src[si + 1] == 0 && src[si + 2] != 0) {if (src[si + 2] == 3) { // escapedst[di++] = 0;dst[di++] = 0;si       += 3;if (nal->skipped_bytes_pos) {nal->skipped_bytes++;if (nal->skipped_bytes_pos_size < nal->skipped_bytes) {nal->skipped_bytes_pos_size *= 2;av_assert0(nal->skipped_bytes_pos_size >= nal->skipped_bytes);av_reallocp_array(&nal->skipped_bytes_pos,nal->skipped_bytes_pos_size,sizeof(*nal->skipped_bytes_pos));if (!nal->skipped_bytes_pos) {nal->skipped_bytes_pos_size = 0;return AVERROR(ENOMEM);}}if (nal->skipped_bytes_pos)nal->skipped_bytes_pos[nal->skipped_bytes-1] = di - 1;}continue;} else // next start codegoto nsc;}dst[di++] = src[si++];}while (si < length)dst[di++] = src[si++];nsc:memset(dst + di, 0, AV_INPUT_BUFFER_PADDING_SIZE);nal->data = dst;nal->size = di;nal->raw_data = src;nal->raw_size = si;rbsp->rbsp_buffer_size += si;return si;
}

三、ff_h2645_extract_rbsp函数的内部实现原理分析

ff_h2645_extract_rbsp函数中存在如下代码:

int ff_h2645_extract_rbsp(const uint8_t *src, int length,H2645RBSP *rbsp, H2645NAL *nal, int small_padding)
{//...#define STARTCODE_TEST                                                  \if (i + 2 < length && src[i + 1] == 0 && src[i + 2] <= 3) {     \if (src[i + 2] != 3 && src[i + 2] != 0) {                   \/* startcode, so we must be past the end */             \length = i;                                             \}                                                           \break;                                                      \}//...for (i = 0; i + 1 < length; i += 2) {if (src[i])continue;if (i > 0 && src[i - 1] == 0)i--;STARTCODE_TEST;}//...
}

其中STARTCODE_TEST是宏定义。将宏展开,上述代码相当于:

int ff_h2645_extract_rbsp(const uint8_t *src, int length,H2645RBSP *rbsp, H2645NAL *nal, int small_padding)
{//...for (i = 0; i + 1 < length; i += 2) {if (src[i])continue;if (i > 0 && src[i - 1] == 0)i--;if (i + 2 < length && src[i + 1] == 0 && src[i + 2] <= 3) {     if (src[i + 2] != 3 && src[i + 2] != 0) {                   /* startcode, so we must be past the end */             length = i;                                             }                                                           break;                                                      }}//...
}

上述代码中,首先会通过语句:

    for (i = 0; i + 1 < length; i += 2) {if (src[i])continue;if (i > 0 && src[i - 1] == 0)i--;//...}

来判断H.264码流中是否存在ASCII 码为 0 (值为'\0')的的字符,如果存在则表明接下来的数据中可能会出现startcode(起始码)或防竞争字节。然后执行下面代码

if (i + 2 < length && src[i + 1] == 0 && src[i + 2] <= 3) {     if (src[i + 2] != 3 && src[i + 2] != 0) {                   /* startcode, so we must be past the end */             length = i;                                             }                                                           break;                                                      
}

来判断是否是起始码,如果是起始码或防竞争字节就通过break;跳出循环。

继续执行语句。满足下面条件,说明是防竞争字节:

else if (src[si] == 0 && src[si + 1] == 0 && src[si + 2] != 0) {if (src[si + 2] == 3) { // escape//...
}

如果是防竞争字节,通过下面语句去掉防竞争字节:

dst[di++] = 0;dst[di++] = 0;si       += 3;
//...

如果不满足条件if (src[si + 2] == 3),说明遇到下一个起始码,表示这个NALU结束了。执行else语句,跳转到“nsc”:

if (src[si + 2] == 3) { // escape
//...
}
else // next start codegoto nsc;
//...

跳转到“nsc”后,给输出型参数赋值,并返回。

nsc:memset(dst + di, 0, AV_INPUT_BUFFER_PADDING_SIZE);nal->data = dst;nal->size = di;nal->raw_data = src;nal->raw_size = si;rbsp->rbsp_buffer_size += si;return si;

四、通过修改ff_h2645_extract_rbsp函数降低FFmpeg转码时的cpu使用率

由于ff_h2645_extract_rbsp函数在H.264/H.265的解码时被调用。所以理论上修改该函数(使用算法优化,用空间换时间等策略)可以降低FFmpeg转码时的cpu使用率。具体可以参考:Imagine Computing创新技术大赛赛道2参赛攻略 - 007gzs

这篇关于FFmpeg源码:ff_h2645_extract_rbsp函数分析的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1079209

相关文章

Python正则表达式语法及re模块中的常用函数详解

《Python正则表达式语法及re模块中的常用函数详解》这篇文章主要给大家介绍了关于Python正则表达式语法及re模块中常用函数的相关资料,正则表达式是一种强大的字符串处理工具,可以用于匹配、切分、... 目录概念、作用和步骤语法re模块中的常用函数总结 概念、作用和步骤概念: 本身也是一个字符串,其中

Java程序进程起来了但是不打印日志的原因分析

《Java程序进程起来了但是不打印日志的原因分析》:本文主要介绍Java程序进程起来了但是不打印日志的原因分析,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录Java程序进程起来了但是不打印日志的原因1、日志配置问题2、日志文件权限问题3、日志文件路径问题4、程序

Java 正则表达式URL 匹配与源码全解析

《Java正则表达式URL匹配与源码全解析》在Web应用开发中,我们经常需要对URL进行格式验证,今天我们结合Java的Pattern和Matcher类,深入理解正则表达式在实际应用中... 目录1.正则表达式分解:2. 添加域名匹配 (2)3. 添加路径和查询参数匹配 (3) 4. 最终优化版本5.设计思

Java字符串操作技巧之语法、示例与应用场景分析

《Java字符串操作技巧之语法、示例与应用场景分析》在Java算法题和日常开发中,字符串处理是必备的核心技能,本文全面梳理Java中字符串的常用操作语法,结合代码示例、应用场景和避坑指南,可快速掌握字... 目录引言1. 基础操作1.1 创建字符串1.2 获取长度1.3 访问字符2. 字符串处理2.1 子字

shell编程之函数与数组的使用详解

《shell编程之函数与数组的使用详解》:本文主要介绍shell编程之函数与数组的使用,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录shell函数函数的用法俩个数求和系统资源监控并报警函数函数变量的作用范围函数的参数递归函数shell数组获取数组的长度读取某下的

MySQL高级查询之JOIN、子查询、窗口函数实际案例

《MySQL高级查询之JOIN、子查询、窗口函数实际案例》:本文主要介绍MySQL高级查询之JOIN、子查询、窗口函数实际案例的相关资料,JOIN用于多表关联查询,子查询用于数据筛选和过滤,窗口函... 目录前言1. JOIN(连接查询)1.1 内连接(INNER JOIN)1.2 左连接(LEFT JOI

MySQL中FIND_IN_SET函数与INSTR函数用法解析

《MySQL中FIND_IN_SET函数与INSTR函数用法解析》:本文主要介绍MySQL中FIND_IN_SET函数与INSTR函数用法解析,本文通过实例代码给大家介绍的非常详细,感兴趣的朋友一... 目录一、功能定义与语法1、FIND_IN_SET函数2、INSTR函数二、本质区别对比三、实际场景案例分

Python 迭代器和生成器概念及场景分析

《Python迭代器和生成器概念及场景分析》yield是Python中实现惰性计算和协程的核心工具,结合send()、throw()、close()等方法,能够构建高效、灵活的数据流和控制流模型,这... 目录迭代器的介绍自定义迭代器省略的迭代器生产器的介绍yield的普通用法yield的高级用法yidle

C++ Sort函数使用场景分析

《C++Sort函数使用场景分析》sort函数是algorithm库下的一个函数,sort函数是不稳定的,即大小相同的元素在排序后相对顺序可能发生改变,如果某些场景需要保持相同元素间的相对顺序,可使... 目录C++ Sort函数详解一、sort函数调用的两种方式二、sort函数使用场景三、sort函数排序

C语言函数递归实际应用举例详解

《C语言函数递归实际应用举例详解》程序调用自身的编程技巧称为递归,递归做为一种算法在程序设计语言中广泛应用,:本文主要介绍C语言函数递归实际应用举例的相关资料,文中通过代码介绍的非常详细,需要的朋... 目录前言一、递归的概念与思想二、递归的限制条件 三、递归的实际应用举例(一)求 n 的阶乘(二)顺序打印