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

一、实验目的

自行生成多个BMP文件，至少含5个不同的场景画面，要求带含有班级、学号后四位和本人姓名的logo。
编写代码实现将第一步所生成的多个BMP文件转化为YUV文件，要求可在命令行中设置每个画面出现的帧数，且最后形成的YUV文件应至少包含200帧。
对整个程序进行调试，并将生成的YUV文件用播放软件观看，验证是否正确。

二、BMP文件格式解析

典型的 BMP 图像文件由四部分组成：
（1）位图头文件数据结构，包含 BMP 图像文件的类型、显示内容等信息；

（2）位图信息数据结构，包含有 BMP 图像的宽、高、压缩方法，以及定义颜色等信息；

（3）调色板，这个部分是可选的，有些位图需要调色板，有些位图，比如真彩色图（24位的 BMP）就不需要调色板；

（4）位图数据，根据 BMP 位图使用的位数不同而不同，在 24 位图中直接使用 RGB，而其他的小于 24 位的使用调色板中颜色索引值。

• 注意
  

三、实验代码

(1)读取头文件

下面展示一些 内联代码片。

tagBITMAPFILEHEADER file_header;
			tagBITMAPINFOHEADER info_header;

			if (fread(&file_header, sizeof(tagBITMAPFILEHEADER), 1, bmpFile) != 1)
			{
				printf("read file header error!");
				exit(0);
			}

			if (file_header.bfType != 0x4D42)
			{
				printf("Not bmp file!");
				exit(0);
			}

			else
			{
				printf("this is a bmp file!");
			}

			if (fread(&info_header, sizeof(tagBITMAPINFOHEADER), 1, bmpFile) != 1)
			{
				printf("read info header error!");
				exit(0);
			}

			frameWidth = info_header.biWidth;
			frameHeight = info_header.biHeight;

(2)主代码

１、定义相关变量，对于输出的out.yuv文件设置相应的指针。
２、设置for循环，用以循环读取输入的ｘ的bmp文件数据。
３、对于每一个输入的bmp文件，先读取其File_header和Info_header，在读取实际的位图文件图像数据。
４、调用Lab１中RGB2YUV转换函数，进行转换。
５、将转换好的yuv数据写入输出的out.yuv文件，并循环40次。
６、释放缓冲、关闭文件。

bmpBuf = (u_int8_t*)malloc(frameWidth * frameHeight * 3);

			/* get the output buffers for a frame */
			yBuf = (u_int8_t*)malloc(frameWidth * frameHeight);
			uBuf = (u_int8_t*)malloc((frameWidth * frameHeight) / 4);
			vBuf = (u_int8_t*)malloc((frameWidth * frameHeight) / 4);

			fread(bmpBuf, 1, frameWidth * frameHeight * 3, bmpFile);

			if (BMP2YUV(frameWidth, frameHeight, bmpBuf, yBuf, uBuf, vBuf, flip))
			{
				printf("error");
				return 0;
			}

			int i;

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

			for (i = 0; i < frameWidth * frameHeight / 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;
			}

			
				fwrite(yBuf, 1, frameWidth * frameHeight, yuvFile);
				fwrite(uBuf, 1, (frameWidth * frameHeight) / 4, yuvFile);
				fwrite(vBuf, 1, (frameWidth * frameHeight) / 4, yuvFile);
			

			/* cleanup */
			fclose(bmpFile);
			

(3)bmp2yuv

int BMP2YUV (int x_dim, int y_dim, void *bmp, void *y_out, void *u_out, void *v_out, int flip)
{
	static int init_done = 0;

	long i, j, size;
	unsigned char *r, *g, *b;
	unsigned char *y, *u, *v;
	unsigned char *pu1, *pu2, *pv1, *pv2, *psu, *psv;
	unsigned char *y_buffer, *u_buffer, *v_buffer;
	unsigned char *sub_u_buf, *sub_v_buf;

	if (init_done == 0)
	{
		InitLookupTable();
		init_done = 1;
	}

	// check to see if x_dim and y_dim are divisible by 2
	if ((x_dim % 2) || (y_dim % 2)) return 1;
	size = x_dim * y_dim;

	// allocate memory
	y_buffer = (unsigned char *)y_out;
	sub_u_buf = (unsigned char *)u_out;
	sub_v_buf = (unsigned char *)v_out;
	u_buffer = (unsigned char *)malloc(size * sizeof(unsigned char));
	v_buffer = (unsigned char *)malloc(size * sizeof(unsigned char));
	if (!(u_buffer && v_buffer))
	{
		if (u_buffer) free(u_buffer);
		if (v_buffer) free(v_buffer);
		printf("aaaaaaaaaaaaaaaa");
		return 2;
	}

	b = (unsigned char *)bmp;
	y = y_buffer;
	u = u_buffer;
	v = v_buffer;

	// convert RGB to 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 ++;
			}
		}
	} else {
		for (i = 0; i < size; 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 ++;
		}
	}

	// subsample UV
	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;
		}
	}

	free(u_buffer);
	free(v_buffer);

	return 0;
}

注意

• 对于程序中需要输入的数据，可以使用命令参数写入程序中，既简单又高效。
• 在读取File_header和Info_header的数据时，要使用结构体变量，既便捷又能保证高准确率，便于后续调用结构体变量中的图像宽biWidth、高biHeight。

四、实验结果

五、实验反思

（1）使用了命令参数和结构体变量，极大减轻了代码的书写量，简单高效。
（2）由于本次实验所使用的bmp文件均为32bit深度，是真彩图像，但其数据部分为各像素的B、G、R、A数据的循环，所以在RGB2YUV的转换函数中，需要调整b指针的＂b＋＝３＂为＂b＋＝４＂。
（3）由于输出的out.yuv文件只有一个，且需要追加写入数据，所以使用out.yuv文件的yuvFile指针打开文件、关闭out.yuv文件的操作都需要在读取输入的laｘ的bmp文件数据的循环外执行，但对于每个bmp文件的打开、数据的读取和相关内存的释放则需要在该循环中实现。