实验二 图像文件的转换和读写 bmp转yuv

实验目的：
1.理解bmp文件的基本组成；
2.实现bmp序列向yuv文件的转换（基本要求为24bit的bmp文件，进阶要求为支持小于24bit的bmp文件），并用YUV Viewer播放验证。

BMP文件格式：
典型的 BMP 图像文件由四部分组成：
（1）位图头文件数据结构，它包含 BMP 图像文件的类型、显示内容等信息；
位图文件头：
typedef struct tagBITMAPFILEHEADER {WORD bfType; /* 说明文件的类型 /
DWORD bfSize; / 说明文件的大小，用字节为单位 /
WORD bfReserved1; / 保留，设置为 0 /
WORD bfReserved2; / 保留，设置为 0 /
DWORD bfOffBits; / 说明从 BITMAPFILEHEADER 结构开始到实际的图像数据之间的字节偏移量 /
} BITMAPFILEHEADER;
（2）位图信息数据结构，它包含有 BMP 图像的宽、高、压缩方法，以及定义颜色等信息；
位图信息头：
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）调色板，这个部分是可选的，有些位图需要调色板，有些位图，比如真彩色图（24位的 BMP）就不需要调色板；
本实验的BMP文件为24位的真彩色图，不需要调色板。
（4）位图数据，这部分的内容根据 BMP 位图使用的位数不同而不同，在 24 位图中直接使用 RGB，而其他的小于 24 位的使用调色板中颜色索引值。
R、G、B所在位置，直接把此段数据当做普通RGB文件进行处理即可。
使用windows.h中的BITMAPFILEHEADER 和BITMAPINFOHEADER 即可直接读出位图文件头和位图信息头，并使用其.biWidth和.biHeight直接读出图像的宽和高，并对剩余数据进行普通的RGBtoYUV转换，分别对多个BMP文件进行以上操作并将得到的数据放入同一YUV文件中即可获得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;

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);
}

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);
	}
    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;
		}
        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);
		}
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 ++;
		}
	}
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;

}

查看结果