【H264/AVC 句法和语义详解】(六):C语言实现Exp-Golomb指数哥伦布编码(编码篇)

上篇中我们介绍了Exp-Golomb的理论部分,这一篇我们就使用C语言来实现它。

我们已经知道,在H.264中,指数哥伦布编码有四个描述子,分别为ue(v)、se(v)、me(v)、te(v)。其中me(v)是最简单的,它直接靠查表来实现。而剩余的se(v)和te(v),是在ue(v)的基础上来实现的。所以它们的利害关系不明而喻,ue(v)就代表了指数哥伦布编码。

下面我们就先重点介绍,无符号指数哥伦布编码:ue(v)。

1. ue(v)的编码实现

由上一篇总结出,指数哥伦布编码的实现,可以分为以下几步:

在这里插入图片描述
ue(v)编码步骤

而考虑到是使用C语言,我们需要把步骤明确的更利于代码实现:

在这里插入图片描述
ue(v)编码实现步骤

图中的Buffer,通常称为缓冲器,其实在这里就是一个数组。在图中整个过程最重要的,就是要有这么一个工具,它能向Buffer中,循环写入n个比特值。并且编码下一个数时,它还能接着在上一个比特末尾,继续写入。

而在编程语言中呢,我们通常是在字节的级别上进行操作,C语言也不例外,比如数组的取值和赋值。所以我们需要定义这样一个结构体,来保存当前指针在Buffer中的位置,也即指针正在指向哪个字节,以及在这个字节中,剩余可用的比特数。

1.1 比特流操作结构体

这个结构体如下:

typedef struct
{
	uint8_t*start; // 指向buf头部指针
	uint8_t*p;     // 当前指针位置
	uint8_t* end;   // 指向buf尾部指针
	int bits_left;  // 当前读取字节的剩余可用比特个数
} bs_t;

1.2 初始化对象

以面向对象的思想,结构体就是一个类。所以我们写一个构造器,使用Buffer来初始化这个比特流操作对象。

bs_t* bs_init(bs_t* b, uint8_t* buf, size_t size)
{
	b->start = buf;  // 指向buf起始位置
	b->p = buf;      // 初始位置与start保持一致
	b->end = buf + size;    // 指向buf末尾
	b->bits_left = 8;   // 默认剩余8比特可用
	return b;
}

bs_t* bs_new(uint8_t* buf, size_t size)
{
	bs_t* b = (bs_t*)malloc(sizeof(bs_t));
	bs_init(b, buf, size);
	return b;
}

同时别忘了释放:

void bs_free(bs_t* b)
{
	free(b);
}

1.3 写入1个比特

然后我们可以拿这个对象(或句柄),写入1个比特。

void bs_write_u1(bs_t* b, uint32_t v)
{
	// 1.剩余比特先减1
	b->bits_left--;
	if (! bs_eof(b))
	{
		// 2.见文章
		(*(b->p)) |= ((v & 0x01) << b->bits_left);
	}
	// 3.判断是否写到字节末尾,如果是指针位置移向下一字节,比特位初始为8
	if (b->bits_left == 0) { b->p ++; b->bits_left = 8; }
}

/** 是否已读到末尾(end_of_file) */
int bs_eof(bs_t* b) { if (b->p >= b->end) { return 1; } else { return 0; } }

每次调用这个函数,会向Buffer中写入1个比特值。其中第二步包含四步:

1(v & 0x01):按位与计算,输入参数v的末尾比特的值,就这个函数来讲,v为01,所以(v & 0x01)的值也即v的值
(2((v & 0x01) << b->bits_left):将第(1)步计算出的比特值,移位至剩余可用比特位的最高位。
(3(*(b->p)) | ((v & 0x01) << b->bits_left):按位或运算,保留b->p指向字节之前写入比特值的同时,把第(2)步新写入的比特值加上
(4)把第(3)步结果赋值给*(b->p)

1.4 写入n个比特

/**
写入n个比特
@param b 比特流操作句柄
@param n 参数v所需的比特位个数
@param v 待写入的值
*/
void bs_write_u(bs_t* b, int n, uint32_t v)
{
	int i;
	for (i = 0; i < n; i++)
	{
		// 循环调用bs_write_u1(),写入n个比特
		bs_write_u1(b, (v >> ( n - i - 1 ))&0x01 );
	}
}

其中(v >> ( n - i - 1 ))&0x01分为两步:

1)v >> ( n - i - 1 ):把参数v向右移( n - i - 1 )位
(2)计算经过第(1)步移位后,位于最低比特位的值

1.5 写入指数哥伦布编码后的值

这一步最重要的,就是需要计算指数哥伦布编码后,所需要的比特位个数,然后直接调用 bs_write_u() 写入即可。

/**ue(v) 无符号指数哥伦布编码*/
void bs_write_ue( bs_t *b, unsigned int val)
{
	// val + 1所需的比特个数
	int i_size = 0;
	// 1.值为0~255时,所需的比特位个数表
	static const int i_size0_255[256] =
	{
		1,      // 0的二进制所需的比特个数
		1,      // 1的二进制所需的比特个数
		2,2,    // 2~3的二进制所需的比特个数
		3,3,3,3,    // 4~7的二进制所需的比特个数
		4,4,4,4,4,4,4,4,  // 8~15的二进制所需的比特个数
		5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,  // 16~31的二进制所需的比特个数
		6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,  // 32~63的二进制所需的比特个数
		6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,
		7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,  // 64~127的二进制所需的比特个数
		7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
		7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
		7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,  // 128~255的二进制所需的比特个数
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
	};
	
	if( val == 0 ) // 输入为0,直接编码为1
	{
		bs_write_u1( b, 1 );
	}
	else
	{
		// 1.指数哥伦布编码第一步,先把输入值+1
		unsigned int tmp = ++val;
		
		// 2.判断所需比特个数是否大于16位
		if( tmp >= 0x00010000 )
		{
			i_size += 16;
			tmp >>= 16;
		}
		
		// 3.判断此时所需比特个数是否大于8位
		if( tmp >= 0x100 )
		{
			i_size += 8;
			tmp >>= 8;
		}
		
		// 4.最终tmp移位至8位以内,去查表
		i_size += i_size0_255[tmp];
		
		// 5.最终得出编码val所需的总比特数:2 * i_size - 1
		// 写入Buffer
		bs_write_u( b, 2 * i_size - 1, val );
	}
}

其中的表i_size0_255[]是个关键,在H.264的各个编解码器中,查表是个很常见的优化方法。而在这里,i_size0_255[]这个表,就包含了值为0~255时,所对应的二进制值的比特位个数。

这样我们肯定会有疑问,既然表中数据只能查0~255以内的,那超过255的该怎么办?

这就是if判断中,else的处理步骤。我们都知道,2的8次方就是256,而256的二进制刚好占用(8+1)=9个比特,而小于256的值例如255,占用最多8个比特。

所以如果我们最终是想查表获得,则需要把大于8比特的值,都移位至8比特以内,这就是else处理的核心。

首先,我们需要判断val+1的值,是否大于16位,也即大于0x10000。如果是,则把所需比特个数i_size加16,然后向右移16位。

然后,再判断处理后的值,是否大于8位,也即大于0x100。如果是,则把所需比特个数i_size加8,然后向右移8位。

这样一来,只要是32位以内的数,肯定能缩短至8位以内,然后嗨皮的去查表就可以了。当然会有人说,那大于32位的数该怎么办?这个请放宽心,这个是绝对不可能出现的,要知道2的32次方是多少?是四十多亿,没错,是亿级别的!所以这个是几乎不可能的!

而且从上一篇我们就知道,指数哥伦布编码,只适合小数值比大数值概率高的编码,数值越大占得位数越大。所以除非不懂指数哥伦布编码,否则我们不可能用这么大的数。

所以最终呢,我们就得到ue(v)所需的总比特数2 * i_size - 1,然后调用bs_write_u()写入buffer就可以了。

2. se(v)的编码实现

实现了ue(v),再实现se(v)就很简单了。因为se(v)也称有符号指数哥伦布编码,也即把ue(v)无符号指数哥伦布编码拓展至了负数。它的编码方式是:

当待编码整数x≤0时,映射为偶数-2x,然后对-2x使用无符号指数哥伦布编码
当待编码整数x>0时,映射为奇整数2x-1,然后对2x-1使用无符号指数哥伦布编码

所以se(v)的实现如下:

/**se(v) 有符号指数哥伦布编码*/
void bs_write_se(bs_t* b, int32_t v)
{
	if (v <= 0)
	{
		bs_write_ue(b, -v*2);
	}
	else
	{
		bs_write_ue(b, v*2 - 1);
	}
}

3. te(v)的编码实现

/** te(v) 截断指数哥伦布编码 */
void bs_write_te( bs_t *b, int x, int val)
{
	if( x == 1 )
	{
		bs_write_u1( b, 1&~val );
	}
	else if( x > 1 )
	{
		bs_write_ue( b, val );
	}
}

其中x为语法元素范围的上限,如果上限为1,则为输入值val最低比特位的取反,否则使用ue(v)进行编码。

最后注意,给Buffer开辟存储空间时,要使用calloc(),不要用malloc(),否则会出现初始值不为0的情况。

4. 完整实例代码

#include <stdio.h>
#include <iostream>
using namespace std;
 
typedef unsigned char uint8_t;
typedef unsigned int  uint32_t;
typedef struct
{
	uint8_t *start; // 指向buf头部指针
	uint8_t *p;     // 当前指针位置
	uint8_t *end;   // 指向buf尾部指针
	int bits_left;  // 当前读取字节的剩余可用比特个数
} bs_t;
//函数申明
int bs_eof(bs_t* b);
//
bs_t* bs_init(bs_t* b, uint8_t* buf, size_t size)
{
	b->p = buf;      // 初始位置与start保持一致
	b->bits_left = 8;   // 默认剩余8比特可用
	return b;
}
 
bs_t* bs_new(uint8_t* buf, size_t size)
{
	bs_t* b = (bs_t*)malloc(sizeof(bs_t));
	bs_init(b, buf, size);
	return b;
}
 
void bs_free(bs_t* b)
{
	free(b);
}
 
void bs_write_u1(bs_t* b, uint32_t v)
{
	// 1.剩余比特先减1
	b->bits_left--;
	if (! bs_eof(b))
	{
		cout<<v;
		// 2.见文章
		*(b->p) <<= 1;
		*(b->p) |= (v & 0x01);
	}
	// 3.判断是否写到字节末尾,如果是指针位置移向下一字节,比特位初始为8
	if (b->bits_left == 0) {
		b->p++;
		b->bits_left = 8;//这样就保证了每次写入不会受比特的影响
	}
}
 
/** 是否已读到末尾(end_of_file) */
int bs_eof(bs_t* b)
{
	if (b->p >= b->end) {
		return 1;
	} else {
		return 0;
	}
}
 
/**
写入n个比特
@param b 比特流操作句柄
@param n 参数v所需的比特位个数
@param v 待写入的值
*/
void bs_write_u(bs_t* b, int n, uint32_t v)
{
	int i;
	//cout<< "n=" << n << "v=" << v <<endl;
	cout<<"转换后的数:"
	for (i = 0; i < n; i++)
	{
		// 循环调用bs_write_u1(),写入 v 的后 n 个比特
		bs_write_u1(b, (v >> ( n - i - 1 ))&0x01 );
	}
	cout<<endl;

}
 
/**ue(v) 无符号指数哥伦布编码*/
unsigned int bs_write_ue( bs_t *b, unsigned int val)
{
	// val + 1所需的比特个数
	int i_size = 0;
	// 1.值为0~255时,所需的比特位个数表
	static const int i_size0_255[256] =
	{
		1,      // 0的二进制所需的比特个数
		1,      // 1的二进制所需的比特个数
		2,2,    // 2~3的二进制所需的比特个数
		3,3,3,3,    // 4~7的二进制所需的比特个数
		4,4,4,4,4,4,4,4,  // 8~15的二进制所需的比特个数
		5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,  // 16~31的二进制所需的比特个数
		6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,  // 32~63的二进制所需的比特个数
		6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,
		7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,  // 64~127的二进制所需的比特个数
		7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
		7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
		7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,  // 128~255的二进制所需的比特个数
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
		8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
	};
	
	if( val == 0 ) // 输入为0,直接编码为1
	{
		bs_write_u1( b, 1 );
	}
	else
	{
		// 1.指数哥伦布编码第一步,先把输入值+1
		unsigned int tmp = ++val;
		
		// 2.判断所需比特个数是否大于16位
		if( tmp >= 0x00010000 )
		{
			i_size += 16;
			tmp >>= 16;
		}
		
		// 3.判断此时所需比特个数是否大于8位
		if( tmp >= 0x100 )
		{
			i_size += 8;
			tmp >>= 8;
		}
		
		// 4.最终tmp移位至8位以内,去查表
		i_size += i_size0_255[tmp];
		
		// 5.最终得出编码val所需的总比特数:2 * i_size - 1
		// 写入Buffer
		bs_write_u( b, 2 * i_size - 1, val );
		return (2 * i_size - 1);
	}
}
 
void my_print(char *s, unsigned int n)
{
	int j;
	int i;
	unsigned int num;
	i = n % 8; //16进制中每3位为一个字节,为8
	//printf("n = %u\n", n);
	num = (n >> 3) + 1;//占用几个数组元素
	j = 0;
	while (j < num) {
		
		if (j == num - 1) i = n % 8;//最高的一个字节不需要移8位
		else              i = 8;
		//printf("j = %d i = %d num = %d\n", j, i, num);
		for (; i > 0; i--) {
			cout<< ((s[j] >> (i - 1)) & 0x01) ;
		}		
		++j;
	}
	cout<<endl;
}
int main()
{	
	uint8_t buf[64] = {0};
	unsigned int i;
	memset(buf, 0, sizeof(buf));
	bs_t *pb = bs_new(buf, sizeof(buf));
	unsigned int j = 0;
	while (1) {
		cout<<"输入:"
		cin>> i;
		j += bs_write_ue(pb, i);
		cout<<"输出整个数据流:"
		my_print((char*)buf, j);
		//memset(buf, 0, sizeof(buf));
	}
	bs_free(pb);
	return 0;
}

运行结果:

输入:1
转换后的数:010
输出整个数据流:010
输入:2
转换后的数:011
输出整个数据流:010011
输入:3
转换后的数:00100
输出整个数据流:01001100100
输入:

本文源码地址如下:

1、GitHub:https://github.com/Gosivn/Exp-Golomb-C-implementation

2、CSDN:https://download.csdn.net/download/u011399342/10321706

[注]:本文的实现严重参考x264源码!

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值