在涉及音视频编解码算法中,经常会涉及一些数据压缩、声音解码、图象的缩放等问题。
这里通过一个例子来推荐一种union绝妙用法(这种方法由Equator公司提供,在我们公司的图象处理算法中用得很多)。在该例子中,利用union结构n64u实现占8个字节n64类型与单字节的c0~c7的相互转换,从而达到数据压缩和分解的目的。
#include <stdio.h>
#define IN
#define OUT
#define INOUT
typedef unsigned long long n64;
typedef unsigned int n32;
typedef unsigned short n16;
typedef unsigned char n8;
typedef struct _s8t{
unsigned char c0, c1, c2, c3, c4, c5, c6, c7;
}s8t;
/*利用n64和c0~c7占用同一内存区,实现它们之间的相互转换*/
typedef union {
n64 n64;
struct {
n32 l0, l1;
} u32;
struct {
long l0, l1;
} s32;
struct {
unsigned short s0, s1, s2, s3;
} u16;
struct {
short s0, s1, s2, s3;
} s16;
struct {
unsigned char c0, c1, c2, c3, c4, c5, c6, c7;
} u8;
struct {
char c0, c1, c2, c3, c4, c5, c6, c7;
} s8;
} n64u;
#define MAX_DATA_COMPRESSED_NUM 8
int compress64_8(IN const n64* src,IN n16 n,OUT n8* dst);
int uncompress8_64(IN const n8* src,IN n16 n,OUT n64* dst);
int compress64_8(IN const n64* src,IN n16 n,OUT n8* dst)
{
n64u n64u_data;
register n64* n64ptr=(n64*)src;
register n8* n8ptr=dst;
n16 i=0,num=n;
if(NULL==n64ptr || NULL==n8ptr || n<1)
{
printf("invalid param,src 0x%x,dst 0x%x,n %d\n",n64ptr,n8ptr,n);
return 0;
}
for(i=0;i<num;i++)
{
n64u_data.n64 = *n64ptr++;
/*用简单的均值法将8个点压缩成1个点*/
*n8ptr++ = (n64u_data.u8.c0+n64u_data.u8.c1+n64u_data.u8.c2+\
n64u_data.u8.c3+n64u_data.u8.c4+n64u_data.u8.c5+\
n64u_data.u8.c6+n64u_data.u8.c7)/(sizeof(n64)/sizeof(n8));
}
return 1;
}
int uncompress8_64(IN const n8* src,IN n16 n,OUT n64* dst)
{
n64u n64u_data;
register n64* n64ptr=dst;
register n8* n8ptr=(n8*)src;
register n8 n8data;
n16 i=0,num=n;
if(NULL==n64ptr || NULL==n8ptr || n<1)
{
printf("invalid param,src 0x%x,dst 0x%x,n %d\n",n64ptr,n8ptr,n);
return 0;
}
for(i=0;i<num;i++)
{
n8data=*n8ptr++;
/*将1个点解压成8个点,采用简单的还原,所以有失真*/
n64u_data.u8.c0 = n8data;
n64u_data.u8.c1 = n8data;
n64u_data.u8.c2 = n8data;
n64u_data.u8.c3 = n8data;
n64u_data.u8.c4 = n8data;
n64u_data.u8.c5 = n8data;
n64u_data.u8.c6 = n8data;
n64u_data.u8.c7 = n8data;
*n64ptr++ = n64u_data.n64;
}
}
int main(int argc, char *argv[])
{
n64 n64data[MAX_DATA_COMPRESSED_NUM];
n8 n8data[MAX_DATA_COMPRESSED_NUM];
s8t s8t_data[MAX_DATA_COMPRESSED_NUM]
=
{/*即将被压缩的数据,一个数据占一个字节*/
{1,2,3,4,5,6,7,8},/*8个字节被压缩成单个字节*/
{2,2,3,4,5,6,7,7},
{3,2,3,4,5,6,7,6},
{4,2,3,4,5,6,7,5},
{5,2,3,4,5,6,7,4},
{6,2,3,4,5,6,7,3},
{7,2,3,4,5,6,7,2},
{8,7,6,5,4,3,2,1}
};
n16 i,n=MAX_DATA_COMPRESSED_NUM;
printf("data:\n");
for(i=0;i<n;i++)
{
n64data[i] = *(n64*)&s8t_data[i];
printf("%3u %3u %3u %3u %3u %3u %3u %3u\n",
s8t_data[i].c0,s8t_data[i].c1,s8t_data[i].c2,
s8t_data[i].c3,s8t_data[i].c4,s8t_data[i].c5,
s8t_data[i].c6,s8t_data[i].c7);
}
printf("\n");
compress64_8(n64data,n,n8data);
printf("compressed to:\n");
for(i=0;i<n;i++)
{
printf("%3u ",n8data[i]);
}
printf("\n\n");
uncompress8_64(n8data,n,n64data);
printf("uncompressed to:\n");
for(i=0;i<n;i++)
{
*(n64*)&s8t_data[i] = n64data[i];
printf("%3u %3u %3u %3u %3u %3u %3u %3u\n",
s8t_data[i].c0,s8t_data[i].c1,s8t_data[i].c2,
s8t_data[i].c3,s8t_data[i].c4,s8t_data[i].c5,
s8t_data[i].c6,s8t_data[i].c7);
}
printf("\n");
}
/*
运行结果:
data:
1 2 3 4 5 6 7 8
2 2 3 4 5 6 7 7
3 2 3 4 5 6 7 6
4 2 3 4 5 6 7 5
5 2 3 4 5 6 7 4
6 2 3 4 5 6 7 3
7 2 3 4 5 6 7 2
8 7 6 5 4 3 2 1
compressed to:
4 4 4 4 4 4 4 4
uncompressed to:
4 4 4 4 4 4 4 4
4 4 4 4 4 4 4 4
4 4 4 4 4 4 4 4
4 4 4 4 4 4 4 4
4 4 4 4 4 4 4 4
4 4 4 4 4 4 4 4
4 4 4 4 4 4 4 4
4 4 4 4 4 4 4 4
*/
如果对Equator感兴趣可以访问www.equator.com。
3. 使不同数据包兼容
union的用法如下:
struct _my_struct
{
unsigned int struct_id
typedef union _my_union
{
struct my_struct_1;
struct my_struct_2;
struct my_struct_3;
}my_union;
}my_struct;
我们公司在处理音频视频数据流方面,为了区分音频和视频数据以及来自网络和编解码的数据并减少内存占用使用了下面的数据结构。这种union使用方法在网络应用中特别常见。在数据结构TFrameBufferInfo中,用bufferType标识数据来源。
/** \struct TABufferInfoFromCodec
* \brief 来自编码器的音频数据的缓冲区信息
*/
typedef struct
{
int codecId; /**< codec ID */
int packetNum; /**< 该帧数据一共有多少个包 */
int actualNum; /**< 当前已存入的包数 */
int packetLen; /**< 正常数据包最大长度,包括包头 */
int leftPacketLen; /**< 最后一个数据包大小,包括包头 */
int frameSample; /**< 0x00表示8kHz,0x01表示11kHz*/
int timeStamp; /**< 表示该帧的编码时间 */
int dataLen; /**< 数据长度,包括包头 */
int ready; /**< 该帧数据的数据包是否都已经写入*/
unsigned char* buffer; /**< 数据指针 */
int reserve1; /**< 保留位,用0填充 */
} TABufferInfoFromCodec;
/** \struct TVBufferInfoFromCodec
* \brief 来自编码器的视频数据的缓冲区信息
*/
typedef struct
{
int codecId; /**< codec ID */
int bKeyFrame; /**< TRUE:关键帧 */
int packetNum; /**< 该帧数据一共有多少个包 */
int actualNum; /**< 当前已存入的包数 */
int packetLen; /**< 正常数据包最大长度,包括包头 */
int leftPacketLen; /**< 最后一个数据包大小,包括包头 */
int timeStamp; /**< 表示该帧的编码时间 */
int dataLen; /**< 数据长度,包括包头 */
int ready; /**< 该帧数据的数据包是否都已经写入*/
unsigned char* buffer; /**< 数据指针 */
int reserve1; /**< 保留位,用0填充 */
} TVBufferInfoFromCodec;
/** \struct TVBufferInfoToCodec
* \brief 来自网络的视频数据的缓冲区信息
*/
typedef struct
{
int codecId;
int bKeyFrame;
int packetNum;
int actualNum;
int packetLen;
int leftPacketLen;
int rtpLen;
int timeStamp;
int firstSquence;
int dataLen;
int ready;
unsigned char* buffer;
} TVBufferInfoToCodec;
/** \struct TABufferInfoToCodec
* \brief 来自网络的音频数据的缓冲区信息
*/
typedef struct
{
int codecId;
int packetNum;
int actualNum;
int packetLen;
int leftPacketLen;
int rtpLen;
int timeStamp;
int firstSquence;
int dataLen;
int ready;
unsigned char* buffer;
int reserve1;
} TABufferInfoToCodec;
/** \struct TABufferInfoToCodec
* \brief 数据缓冲区信息结构
*/
typedef struct
{
int bufferType;
union
{
TVBufferInfoFromCodec bufferInfoFromVCoder;
TABufferInfoFromCodec bufferInfoFromACoder;
TVBufferInfoToCodec bufferInfoFromVNetWork;
TABufferInfoToCodec bufferInfoFromANetWork;
} buffer_info;
} TFrameBufferInfo;
int send_to(void* stream);
int send_to(void* stream)
{
/*省略*/
}
int main(int argc, char *argv[])
{
TFrameBufferInfo tFrameBufferInfo;
TVBufferInfoFromCodec* pVBufferInfoFromCodec;
unsigned char buffer[1200*5];/*最大存放5个数据包*/
/*假设你要向网络发送一段视频数据,音频接收为1,
视频接收为2,*音频发送为3,视频发送为4*/
tFrameBufferInfo.bufferType=4;
pVBufferInfoFromCodec
=&tFrameBufferInfo.buffer_info.bufferInfoFromVCoder;
pVBufferInfoFromCodec->bKeyFrame = 1;
pVBufferInfoFromCodec->packetNum = 2;
pVBufferInfoFromCodec->actualNum = 2;
pVBufferInfoFromCodec->leftPacketLen = 123;
pVBufferInfoFromCodec->dataLen = 1323;
pVBufferInfoFromCodec->ready = 1;
pVBufferInfoFromCodec->buffer = buffer;
send_to((void*)&tFrameBufferInfo);
}