数据压缩第四次作业

彩色空间转换
实验一：RGB转YUV
转换前的原图片：

代码实现：

#include <iostream>
#include<cstdlib>

using namespace std;

int main()
{
	FILE* down;
	FILE* YUV;
	const int width = 256;
	const int height = 256;
	int size=0;
	//打开文件，定义新文件
	fopen_s(&down, "D:\down.rgb", "rb");
	fopen_s(&YUV, "D:\down1.yuv", "wb");
	//计算rgb文件的长度
	fseek(down, 0L, SEEK_END);
	size = ftell(down);
	fseek(down, 0L, SEEK_SET);
	//定义缓存区
	unsigned char* buffer = new unsigned char[size];
	unsigned char* Y_BUFFER = new unsigned char[size / 3];
	unsigned char* U_BUFFER = new unsigned char[size / 12];
	unsigned char* V_BUFFER = new unsigned char[size / 12];
	//每个像素中RGB读入缓存区
	//fread的用法：fread(存入地点，每次读的字节数，共读字节数，读取文件名)
	fread(buffer, 1, size, down);
	unsigned char* r = new unsigned char[size / 3];
	unsigned char* g = new unsigned char[size / 3];
	unsigned char* b = new unsigned char[size / 3];

	//把读取的内容分别放入RGB数组中 RGB储存顺序是B、G、R，所以用模函数
	//定义RGB数组
	int m = 0, n = 0, k = 0;
	for (int i = 0; i < size; i++)
	{
		if (i % 3 == 0)
		{
			b[m] = *(buffer + i);
			m++;
		}
		else if (i % 3 == 1)
		{
			g[n] = *(buffer + i);
			n++;
		}
		else
		{
			r[k] = *(buffer + i);
			k++;
		}
	}
	//计算Y分量
	for (int i = 0; i < size / 3; i++)
	{
		Y_BUFFER[i] = r[i] * 0.2990 + g[i] * 0.5870 + b[i] * 0.1140;
		if (Y_BUFFER[i] > 235)
		{
			Y_BUFFER[i] = 235;
		}
		if (Y_BUFFER[i] < 16)
		{
			Y_BUFFER[i] = 16;
		}
	}
	//计算U分量
	int u = 0;
	for (int i = 0; i < size / 3; )
	{
		U_BUFFER[u] = (r[i] + r[i + 1] + r[i + width] + r[i + width + 1]) / 4 * (-0.1684) +
			(g[i] + g[i + 1] + g[i + width] + g[i + width + 1]) / 4 * (-0.3316) +
			(b[i] + b[i + 1] + b[i + width] + b[i + width + 1]) / 4 * (0.500) + 128;
		if (U_BUFFER[u] > 235)
		{
			U_BUFFER[u] = 235;
		}
		if (U_BUFFER[u] < 16)
		{
			U_BUFFER[u] = 16;
		}
		if (i % width == 254)
		{
			i = i + 2 + width;
		}
		else
		{
			i = i + 2;
		}
		u++;
	}
	//计算V分量
	int v = 0;
	for (int i = 0; i < size / 3; )
	{
		V_BUFFER[v] = (r[i] + r[i + 1] + r[i + width] + r[i + width + 1]) / 4 * (0.5) +
			(g[i] + g[i + 1] + g[i + width] + g[i + width + 1]) / 4 * (-0.4187) +
			(b[i] + b[i + 1] + b[i + width] + b[i + width + 1]) / 4 * (-0.0813) + 128;
		if (V_BUFFER[v] > 235)
		{
			V_BUFFER[v] = 235;
		}
		if (V_BUFFER[v] < 16)
		{
			U_BUFFER[v] = 16;
		}
		if (i % width == 254)
		{
			i = i + 2 + width;
		}
		else
		{
			i = i + 2;
		}
		v++;
	}
	fwrite(Y_BUFFER, 1, size / 3, YUV);
	fwrite(U_BUFFER, 1, size / 12, YUV);
	fwrite(V_BUFFER, 1, size / 12, YUV);
	delete[](r);
	delete[](g);
	delete[](b);
	delete[](buffer);
	delete[](Y_BUFFER);
	delete[](U_BUFFER);
	delete[](V_BUFFER);
	fclose(down);
	fclose(YUV);
	system("pause");
}

得到的YUV文件：

实验二：YUV转RGB

代码实现：

#include <iostream>
#include<cstdlib>

using namespace std;

int main(int argc, char* argv[])
{
	FILE* YUV;
	FILE* RGB;
	const int width = 256;
	const int height = 256;
	int size = 0;
	//打开文件，定义新文件
	fopen_s(&YUV, "D:\down1.yuv", "rb");
	fopen_s(&RGB, "D:\down2.rgb", "wb");
	//计算rgb文件的长度
	fseek(YUV, 0L, SEEK_END);
	size = ftell(YUV);
	fseek(YUV, 0L, SEEK_SET);
	//定义缓存区
	unsigned char* buffer = new unsigned char[size];
	unsigned char* Y_BUFFER = new unsigned char[size * 2 / 3];
	unsigned char* U_BUFFER = new unsigned char[size / 6];
	unsigned char* V_BUFFER = new unsigned char[size / 6];
	fread(buffer, 1, size, YUV);

	unsigned char* r = new unsigned char[size * 2 / 3];
	unsigned char* g = new unsigned char[size * 2 / 3];
	unsigned char* b = new unsigned char[size * 2 / 3];

	for (int i = 0; i < (size * 2 / 3); i++)
	{
		Y_BUFFER[i] = buffer[i];
	}
	for (int i = 0; i < (size * 1 / 6); i++)
	{
		U_BUFFER[i] = buffer[i + (size * 2 / 3)];
	}
	for (int i = 0; i < (size * 1 / 6); i++)
	{
		V_BUFFER[i] = buffer[i + (size * 2 / 3)];
	}

	int R = 0;
	for (int i = 0; i < size * 2 / 3;)
	{
		r[i] = Y_BUFFER[i] + 1.4075 * (V_BUFFER[R] - 128);
		r[i + 1] = Y_BUFFER[i + 1] + 1.4075 * (V_BUFFER[R] - 128);
		r[i + 256] = Y_BUFFER[i + 256] + 1.4075 * (V_BUFFER[R] - 128);
		r[i + 257] = Y_BUFFER[i + 257] + 1.4075 * (V_BUFFER[R] - 128);
		if (i % 256 == 254)
		{
			i = i + 258;
		}
		else
		{
			i = i + 2;
		}
		R++;
	}

	int G = 0;
	for (int i = 0; i < size * 2 / 3;)
	{
		g[i] = Y_BUFFER[i] - 0.3455 * (U_BUFFER[G] - 128) - 0.7169 * (V_BUFFER[G] - 128);
		g[i + 1] = Y_BUFFER[i + 1] - 0.3455 * (U_BUFFER[G] - 128) - 0.7169 * (V_BUFFER[G] - 128);
		g[i + 256] = Y_BUFFER[i + 256] - 0.3455 * (U_BUFFER[G] - 128) - 0.7169 * (V_BUFFER[G] - 128);
		g[i + 257] = Y_BUFFER[i + 257] - 0.3455 * (U_BUFFER[G] - 128) - 0.7169 * (V_BUFFER[G] - 128);
		if (i % 256 == 254)
		{
			i = i + 258;
		}
		else
		{
			i = i + 2;
		}
		G++;
	}

	int B = 0;
	for (int i = 0; i < size * 2 / 3;)
	{
		b[i] = Y_BUFFER[i] + 1.779 * (U_BUFFER[B] - 128);
		b[i + 1] = Y_BUFFER[i + 1] + 1.779 * (U_BUFFER[B] - 128);
		b[i + 256] = Y_BUFFER[i + 256] + 1.779 * (U_BUFFER[B] - 128);
		b[i + 257] = Y_BUFFER[i + 257] + 1.779 * (U_BUFFER[B] - 128);
		if (i % 256 == 254)
		{
			i = i + 258;
		}
		else
		{
			i = i + 2;
		}
		B++;
	}


	fwrite(b, 1, size * 2 / 3, RGB);
	fwrite(g, 1, size * 2 / 3, RGB);
	fwrite(r, 1, size * 2 / 3, RGB);

	delete[](r);
	delete[](g);
	delete[](b);
	delete[](buffer);
	delete[](Y_BUFFER);
	delete[](U_BUFFER);
	delete[](V_BUFFER);
	fclose(RGB);
	fclose(YUV);
	system("pause");
}

转换后的图片：

在YUVviewerPlus软件中打开的，原先的down.rgb无法在这个软件中打开，转换后的rgb文件能打开，但是发现色彩有失真，扫描也有问题。。。
再去看看是哪里出了问题