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

一、实验目的

1.1 理解图像文件的基本组成。

1.2 掌握结构体作为复杂数据对象的用法。进一步熟悉由问题到程序的解决方案，并掌握编程细节：如内存分配、倒序读写、字节序、文件读写过程等。

二、实验内容

2.1 BMP文件的组成结构

BMP（全称Bitmap）是Windows操作系统中的标准图像文件格式，可以分成两类：设备相关位图（DDB）和设备无关位图（DIB）。

采用位映射存储格式，除了图像深度可选以外，在绝大多数应用中不采用其他任何压缩，因此，BMP文件所占用的空间很大。BMP文件的图像深度可选lbit、4bit、8bit、16bit及24bit。

BMP文件存储数据时，图像的扫描方式是按从左到右、从下到上的顺序。由于BMP文件格式是Windows环境中交换与图有关的数据的一种标准，因此在Windows环境中运行的图形图像软件都支持BMP图像格式。

典型的BMP图像文件由四部分组成：

（1）位图头文件数据结构，它包含BMP图像文件的类型、显示内容等信息；

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

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

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

相应的数据结构可表示如下:位图文件头主要包括：

typedef struct tagBITMAPFILEHEADER {
    WORD bfType;       /* 说明文件的类型*/
    DWORD bfSize;      /* 说明文件的大小，用字节为单位*/
    WORD bfReserved1;  /* 保留，设置为0 */
    WORD bfReserved2;  /* 保留，设置为0 */
    DWORD bfOffBits;   /* 说明从BITMAPFILEHEADER结构开始到实际的图像数据之间的字节偏移量*/
} BITMAPFILEHEADER;

(2) 位图信息头主要包括：

typedef struct tagBITMAPINFOHEADER {
    DWORD biSize;   /* 说明结构体所需字节数*/
    LONG biWidth;   /* 以像素为单位说明图像的宽度*/
    LONG biHeight;  /* 以像素为单位说明图像的高速*/
    WORD biPlanes;  /* 说明位面数，必须为 1 */
    WORD biBitCount;  /* 说明位数/像素，1、2、4、8、24 */
    DWORD biCompression; /* 说明图像是否压缩及压缩类型 BI_RGB，BI_RLE8，BI_RLE4，BI_BITFIELDS */
    DWORD biSizeImage;  /* 以字节为单位说明图像大小，必须是4的整数倍*/
    LONG biXPelsPerMeter;  /*目标设备的水平分辨率，像素/米*/
    LONG biYPelsPerMeter;  /*目标设备的垂直分辨率，像素/米*/
    DWORD biClrUsed;   /* 说明图像实际用到的颜色数，如果为0，则颜色数为2的biBitCount次方*/
    DWORD  biClrImportant;  /*说明对图像显示有重要影响的颜色索引的数目，如果是0，表示都重要。*/
} BITMAPINFOHEADER;

(3) 调色板实际上是一个数组，它所包含的元素与位图所具有的颜色数相同，决定于biClrUsed和biBitCount字段。数组中每个元素的类型是一个RGBQUAD结构。真彩色无调色板部分。

typedef struct tagRGBQUAD {
    BYTE rgbBlue;     /*指定蓝色分量*/
    BYTE rgbGreen;    /*指定绿色分量*/
    BYTE rgbRed;      /*指定红色分量*/
    BYTE rgbReserved; /*保留，指定为0*/
} RGBQUAD;

(4) 紧跟在调色板之后的是图像数据字节阵列。对于用到调色板的位图，图像数据就是该像素颜色在调色板中的索引值（逻辑色）。对于真彩色图，图像数据就是实际的R、G、B值。图像的每一扫描行由表示图像像素的连续的字节组成，每一行的字节数取决于图像的颜色数目和用像素表示的图像宽度。规定每一扫描行的字节数必须是4的整倍数，也就是DWORD对齐的。扫描行是由底向上存储的，这就是说，阵列中的第一个字节表示位图左下角的像素，而最后一个字节表示位图右上角的像素。

2.2 字节序

不同的计算机系统采用不同的字节序存储数据，同样一个4字节的32位整数，在内存中存储的方式不同。字节序分为小尾字节序(Little Endian)和大尾字节序(Big Endian)。Intel处理器大多数使用小尾字节序，Motorola处理器大多数使用大尾(Big Endian)字节序。

小尾就是低位字节排放在内存的低端，高位字节排放在内存的高端，即所谓的“低位在前，高位在后”。大尾就是高位字节排放在内存的低端，低位字节排放在内存的高端，即所谓的“高位在前，低位在后”。TCP/IP各层协议将字节序定义为大尾，因此TCP/IP协议中使用的字节序通常称之为网络字节序。

在实现BMP文件头信息的写入和读出时，需要注意整数保存时的字节序。例如：文件大小是以Intel序保存的。在编程前先用二进制打开方式观察BMP文件各个部分的数据存储格式。

三、实验步骤

3.1 BMP序列转YUV文件

（1）在图像处理软件中自行生成多个BMP文件，至少含5个不同的场景画面，要求带含有班级、学号后四位和本人姓名（缩写或昵称均可）的logo。（基本要求为24bit的BMP，进阶要求为支持小于24bit的BMP。）

（2）编写将第一步所生成的多个BMP文件转化为YUV文件，要求可在命令行中设置每个画面出现的帧数。最后形成的YUV文件应至少包含200帧。重点掌握函数定义、缓冲区分配、倒序读写、结构体的操作。

（3）对整个程序进行调试，并将生成的YUV文件用播放软件观看，验证是否正确。

3.2 自行选择的图像文件序列转为YUV文件

（1）在图像处理软件中自行生成所选择格式的图象文件序列，要求带含有班级、学号后四位和本人姓名（缩写或昵称均可）的logo。

（2）编写程序实现该图像文件序列转换为YUV文件，要求在命令行中设置每个画面出现的帧数。最后形成的YUV文件应至少包含200帧。重点掌握函数定义、缓冲区分配、结构体或类的操作。

（3）调试程序，并验证程序的正确性。

四、实验代码及结果

4.1 头文件

#pragma once
#include<windows.h>

int bmp2rgb(unsigned char* bbuf, int w, int h, unsigned char* rgbbuf);
void InitLookupTable();
int rgb2yuv(unsigned long w, unsigned long h, unsigned char* rgbdata, unsigned char* y, unsigned char* u, unsigned char* v);

4.2 主程序

#include<stdio.h>
#include"bmp2yuv.h"
#include<stdlib.h>
#include<windows.h>


int main(int argc, char* argv[])  
{
	BITMAPFILEHEADER file_header;  //定义两个结构体变量
	BITMAPINFOHEADER info_header;

	FILE* bmpFile = NULL;   
	FILE* yuvFile = NULL;

	int frameWidth;
	int frameHeight;
	int framenumber;
	int count;  //图像的重复帧数
	int sum = 0;  //合成的yuv总帧数
	unsigned short bitcount;  //位深
	unsigned char* bmpBuf, * rgbBuf, * yBuf, * uBuf, * vBuf;
	const int pic_num = 5;
	const char* bmpfile[pic_num] = { "1.bmp","2.bmp","3.bmp","4.bmp","5.bmp" };
	const char* yuvfile = "out.yuv";

	yuvFile = fopen(argv[11], "wb");  //打开输出文件，二进制写入yuv文件
	for (int i = 0; i < 2 * pic_num; i = i + 2)
	{
		bmpFile = fopen(argv[i + 1], "rb");
		printf("\n读取bmp文件:%s\n", argv[i + 1]);
		framenumber = atoi(argv[i + 2]);  //写入每个图像的重复帧数
		count = 0;

		//通过文件头判断是否为bmp文件
		if (fread(&file_header, sizeof(BITMAPFILEHEADER), 1, bmpFile) != 1)
		{
			printf("read file header error!\n");
			exit(0);
		}
		if (file_header.bfType != 0x4D42)
		{
			printf("not bmp file!\n");
			exit(0);
		}
		else
		{
			printf("this is a bmp file!\n");
		}

		//位图信息头
		if (fread(&info_header, sizeof(BITMAPINFOHEADER), 1, bmpFile) != 1)
		{
			printf("read info header error!\n");
			exit(0);
		}

		//读取图像宽、高、位数
		frameWidth = info_header.biWidth;
		frameHeight = info_header.biHeight;
		bitcount = info_header.biBitCount;

		//开辟缓冲区
		rgbBuf = (unsigned char*)malloc(frameWidth * frameHeight * 3);
		yBuf = (unsigned char*)malloc(frameWidth * frameHeight);
		uBuf = (unsigned char*)malloc((frameWidth * frameHeight) / 4);
		vBuf = (unsigned char*)malloc((frameWidth * frameHeight) / 4);

		//信息头中宽和高以像素为单位，转换为字节为单位
		int width = frameWidth * info_header.biBitCount / 8;
		int height = frameHeight;
		bmpBuf = ((unsigned char*)malloc(sizeof(unsigned char) * width * height));

		//文件指针读取文件有效数据
		fread(bmpBuf, width * height, 1, bmpFile);

		//读bmp文件得rgb数据
		bmp2rgb(bmpBuf, width, height, rgbBuf);

		//rgb转yuv
		rgb2yuv(frameWidth, frameHeight, rgbBuf, yBuf, uBuf, vBuf);

		//将重复次数的图像写入yuv文件
		for (int a = 0; a < framenumber; a++)
		{
			fwrite(yBuf, 1, frameWidth * frameHeight, yuvFile);
			fwrite(uBuf, 1, (frameWidth * frameHeight) / 4, yuvFile);
			fwrite(vBuf, 1, (frameWidth * frameHeight) / 4, yuvFile);

			++count;
		}
		sum += count;
	}
	printf("\n生成YUV序列，共%d帧。\n", sum);

	//释放空间
	free(rgbBuf);
	free(yBuf);
	free(uBuf);
	free(vBuf);
	fclose(bmpFile);
	fclose(yuvFile);
	return 0;
}

4.3 rgb2yuv

#include<stdlib.h>
#include"bmp2yuv.h"

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

//查找表
void InitLookupTable()
{
	for (int i = 0; i < 256; i++)
	{
		RGBYUV02990[i] = (float)0.2990 * i;
		RGBYUV05870[i] = (float)0.5870 * i;
		RGBYUV01140[i] = (float)0.1140 * i;
		RGBYUV01684[i] = (float)0.1684 * i;
		RGBYUV03316[i] = (float)0.3316 * i;
		RGBYUV04187[i] = (float)0.4187 * i;
		RGBYUV00813[i] = (float)0.0813 * i;
	}
}

int rgb2yuv(unsigned long w, unsigned long h, unsigned char* rgbdata, unsigned char* y, unsigned char* u, unsigned char* v)
{
	InitLookupTable();//初始化查找表

	//定义缓冲区，4：2：0格式uv要下采样
	unsigned char* utemp = NULL;
	unsigned char* vtemp = NULL;
	utemp = (unsigned char*)malloc(w * h);
	vtemp = (unsigned char*)malloc(w * h);
	unsigned long i, r, g, b, nSize;

	//rgb转yuv
	for (i = 0, nSize = 0; nSize < w * h * 3; nSize += 3)
	{
		b = rgbdata[nSize];
		g = rgbdata[nSize + 1];
		r = rgbdata[nSize + 2];
		y[i] = (unsigned char)(RGBYUV02990[r] + RGBYUV05870[g] + RGBYUV01140[b]);
		utemp[i] = (unsigned char)(-RGBYUV01684[r] - RGBYUV03316[g] + b / 2 + 128);
		vtemp[i] = (unsigned char)(r / 2 - RGBYUV04187[g] - RGBYUV00813[b] + 128);
		i++;
	}

	// 4：4：4到4：2：0格式，u、v均为y的1/4,需进行下采样
	int k = 0;
	for (i = 0; i < h; i += 2)
	{
		for (unsigned long j = 0; j < w; j += 2)
		{
			u[k] = (utemp[i * w + j] + utemp[(i + 1) * w + j] + utemp[i * w + j + 1] + utemp[(i + 1) * w + j + 1]) / 4;
			v[k] = (vtemp[i * w + j] + vtemp[(i + 1) * w + j] + vtemp[i * w + j + 1] + vtemp[(i + 1) * w + j + 1]) / 4;
			k++;
		}
	}

	//限电平处理
	for (i = 0; i < w * h; i++)
	{
		if (y[i] < 16)
			y[i] = 16;
		if (y[i] > 235)
			y[i] = 235;
	}

	for (i = 0; i < w * h / 4; i++)
	{
		if (u[i] < 16)
			u[i] = 16;
		if (u[i] > 240)
			u[i] = 240;
		if (v[i] < 16)
			v[i] = 16;
		if (v[i] > 240)
			v[i] = 240;
	}
	if (utemp)
		free(utemp);
	if (vtemp)
		free(vtemp);
	return 0;
}

4.4 bmp2rgb

#include<stdlib.h>
#include<string.h>
#include"bmp2yuv.h"
int bmp2rgb(unsigned char* bbuf, int w, int h, unsigned char* rgbbuf)
{
	unsigned char* rgb = ((unsigned char*)malloc(w * h));

	for (int i = 0; i < h; i++)
	{
		for (int j = 0; j < w; j++)
		{
			rgb[i * w + j] = bbuf[(h - i - 1) * w + j];  
		}
	}
	memcpy(rgbbuf, rgb, w * h);
	free(rgb);
	return 0;
}

4.5 输入的bmp位图

大小为1152x1024，位图深度为24bit

输入命令，生成总计250帧图像

 

查看生成的yuv文件（此处转为gif方便查看）