数据压缩 实验二 BMP序列转YUV文件

BMP序列转YUV文件

BMP文件的组成结构

典型的 BMP 图像文件由四部分组成：
（1）位图头文件数据结构，它包含 BMP 图像文件的类型、显示内容等信息；
（2）位图信息数据结构，它包含有 BMP 图像的宽、高、压缩方法，以及定义颜色等信
息；
（3）调色板，这个部分是可选的，有些位图需要调色板，真彩色图（24位的 BMP）不需要调色板；
（4）位图数据，这部分的内容根据 BMP 位图使用的位数不同而不同，在 24 位图中直接
使用 RGB，而其他的小于 24 位的使用调色板中颜色索引值。

实现算法

命令行参数

修改命令行参数依次为需要转换的BMP文件名、输出的YUV文件名和每个画面出现的帧数。

主函数

声明变量

    BITMAPFILEHEADER File_header;
	BITMAPINFOHEADER Info_header;
	char* yuvFileName = NULL;
	char* bmpFileName[7]{};
	FILE* bmpFile=NULL;
	FILE* yuvFile = NULL;
	u_int Fr = 30;
	u_int8_t* bmpBuf = NULL;
	u_int8_t* yBuf = NULL;
	u_int8_t* uBuf = NULL;
	u_int8_t* vBuf = NULL;
	u_int32_t videoFramesWritten = 0;
    u_int width, height;
	bool flip = FALSE;                 
	unsigned int i,j;

获得YUV文件名和每个图片的帧数，创建YUV文件

    yuvFileName = argv[8];
	Fr = atoi(argv[9]);
	 
	fopen_s(&yuvFile, yuvFileName, "wb");
	if (yuvFile == NULL)
	{
		printf("cannot find yuv file\n");
		exit(1);
	}
	else
	{
		printf("The output yuv file is %s\n", yuvFileName);
	}

循环处理各个BMP图像，得到该BMP图像的文件名，用结构体组织分别读出位图文件头与信息头，从信息头中得到图像的宽和高

for (j = 0; j < 7; j++)
	{
		bmpFileName[j] = argv[j+1];
		fopen_s(&bmpFile, bmpFileName[j], "rb");
		if (fread(&File_header, sizeof(BITMAPFILEHEADER), 1, bmpFile) != 1)
		{
			printf("Read file header error!");
			exit(0);
		}
		if (File_header.bfType != 0x4D42)
		{
			printf("Not bmp file!");
			exit(0);
		}
		if (fread(&Info_header, sizeof(BITMAPINFOHEADER), 1, bmpFile) != 1)
		{
			printf("Read info header error!");
			exit(0);
		}

		width = Info_header.biWidth;
		height = Info_header.biHeight;

分配缓存区

        bmpBuf = (u_int8_t*)malloc(width * height * 3);
		yBuf = (u_int8_t*)malloc(width * height);
		uBuf = (u_int8_t*)malloc((width * height) / 4);
		vBuf = (u_int8_t*)malloc((width * height) / 4);

		if (bmpBuf == NULL || yBuf == NULL || uBuf == NULL || vBuf == NULL)
		{
			printf("no enought memory\n");
			exit(1);
		}

读出位图数据并用BMP2YUV函数转换为YUV图像，为防止信号变动造成过载，在256级上端留20级，下端留16级作为信号超越动态范围的保护带

        if(fread(bmpBuf, 1, width * height * 3, bmpFile)!= 0)
		{
			if (BMP2YUV(width, height, bmpBuf, yBuf, uBuf, vBuf, flip))
			{
				printf("error");
				return 0;
			}

			for (i = 0; i < width * height; i++)
			{
				if (yBuf[i] < 16) yBuf[i] = 16;
				if (yBuf[i] > 235) yBuf[i] = 235;
			}

			for (i = 0; i < width * height / 4; i++)
			{
				if (uBuf[i] < 16) uBuf[i] = 16;
				if (uBuf[i] > 240) uBuf[i] = 240;

				if (vBuf[i] < 16) vBuf[i] = 16;
				if (vBuf[i] > 240) vBuf[i] = 240;
			}

按照每个图像的帧数Fr将得到的数据循环写入YUV文件

            for (i = 0; i < Fr; i++)
			{
				fwrite(yBuf, 1, width * height, yuvFile);
				fwrite(uBuf, 1, (width * height) / 4, yuvFile);
				fwrite(vBuf, 1, (width * height) / 4, yuvFile);
			}

BMP2YUV函数

本次实验使用24位BMP，其图像数据是RGB值。存储数据时，BMP图像的扫描方式是按从左到右、从下到上的顺序，而YUV是按从左到右、从上到下的顺序，所以转换中图像要上下翻转

YUV与RGB转换公式：
Y＝0.2990R+0.5870G+0.1140B
U＝-0.1684R-0.3316G+0.5B
V＝0.5R-0.4187G-0.0813B

RGB转换为YUV

    if (!flip) {
		for (j = 0; j < y_dim; j ++)
		{
			y = y_buffer + (y_dim - j - 1) * x_dim;
			u = u_buffer + (y_dim - j - 1) * x_dim;
			v = v_buffer + (y_dim - j - 1) * x_dim;

			for (i = 0; i < x_dim; i ++) {
				g = b + 1;
				r = b + 2;
				*y = (unsigned char)(  RGBYUV02990[*r] + RGBYUV05870[*g] + RGBYUV01140[*b]);
				*u = (unsigned char)(- RGBYUV01684[*r] - RGBYUV03316[*g] + (*b)/2          + 128);
				*v = (unsigned char)(  (*r)/2          - RGBYUV04187[*g] - RGBYUV00813[*b] + 128);
				b += 3;
				y ++;
				u ++;
				v ++;
			}
		}

将YUV的色度格式由4：4：4通过取平均化为4：2：0，使UV分量取样点数分别为Y分量的四分之一

    for (j = 0; j < y_dim/2; j ++)
	{
		psu = sub_u_buf + j * x_dim / 2;
		psv = sub_v_buf + j * x_dim / 2;
		pu1 = u_buffer + 2 * j * x_dim;
		pu2 = u_buffer + (2 * j + 1) * x_dim;
		pv1 = v_buffer + 2 * j * x_dim;
		pv2 = v_buffer + (2 * j + 1) * x_dim;
		for (i = 0; i < x_dim/2; i ++)
		{
			*psu = (*pu1 + *(pu1+1) + *pu2 + *(pu2+1)) / 4;
			*psv = (*pv1 + *(pv1+1) + *pv2 + *(pv2+1)) / 4;
			psu ++;
			psv ++;
			pu1 += 2;
			pu2 += 2;
			pv1 += 2;
			pv2 += 2;
		}
	}

部分查找表

static float RGBYUV02990[256], RGBYUV05870[256], RGBYUV01140[256];
static float RGBYUV01684[256], RGBYUV03316[256];
static float RGBYUV04187[256], RGBYUV00813[256];

void InitLookupTable()
{
	int i;

	for (i = 0; i < 256; i++) RGBYUV02990[i] = (float)0.2990 * i;
	for (i = 0; i < 256; i++) RGBYUV05870[i] = (float)0.5870 * i;
	for (i = 0; i < 256; i++) RGBYUV01140[i] = (float)0.1140 * i;
	for (i = 0; i < 256; i++) RGBYUV01684[i] = (float)0.1684 * i;
	for (i = 0; i < 256; i++) RGBYUV03316[i] = (float)0.3316 * i;
	for (i = 0; i < 256; i++) RGBYUV04187[i] = (float)0.4187 * i;
	for (i = 0; i < 256; i++) RGBYUV00813[i] = (float)0.0813 * i;
}

实验结果

用yuv播放器查看结果

画面1

画面2

画面3

画面4

画面5

画面6

画面7

每个画面30帧，帧率为30Hz，每个画面持续1秒，结果与预期一致。