DPCM 压缩系统的实现和分析

DPCM压缩系统的实现和分析

DPCM是差分预测编码调制的缩写，是比较典型的预测编码系统。在DPCM系统中，需要注意的是预测器的输入是已经解码以后的样本。之所以不用原始样本来做预测，是因为在解码端无法得到原始样本，只能得到存在误差的样本。因此，在DPCM编码器中实际内嵌了一个解码器，如编码器中虚线框中所示。

分别进行线性预测器和量化器的优化设计。

【实验要求】
首先读取一个
256级的灰度图像，采用自己设定的预测方法计算预测误差，并对预测误差进行8比特均匀量化（基本要求） 。还可对预测误差进行1比特、2比特和4比特的量化设计。

【代码】

【读入文件】

FILE *renFile = NULL;
	if ((renFile = fopen("D:\\数据压缩\\DPCM\\ren.yuv", "rb")) == NULL)
	{
		printf("open Linnafile fail!");
	}
	FILE *PredictedFile = NULL;
	if ((PredictedFile = fopen("D:\\数据压缩\\DPCM\\PredictedFile.yuv", "wb")) == NULL)
	{
		printf("open ForecastFile fail!");
	}
	FILE *RebuildFile = NULL;
	if ((RebuildFile = fopen("D:\\数据压缩\\DPCM\\RebuildFile.yuv", "wb")) == NULL)
	{
		printf("open RebuildFile fail!");
	}
	int w = 256;
	int h = 256;
	int N = 8;
	unsigned char *ybuffer, *ubuffer, *vbuffer, *yPre, *yRe;
	ybuffer = (unsigned char *)malloc(sizeof(char)*(w*h));
	yPre = (unsigned char *)malloc(sizeof(char)*(w*h));
	yRe = (unsigned char *)malloc(sizeof(char)*(w*h));
	ubuffer = (unsigned char *)malloc(sizeof(char)*(w*h) / 4);
	vbuffer = (unsigned char *)malloc(sizeof(char)*(w*h) / 4);

	fread(ybuffer, 1, w*h, renFile);
	fread(ubuffer, 1, w*h / 4, renFile);
	fread(vbuffer, 1, w*h / 4, renFile);

下面这一部分是因为发现在写代码的过程中，图片颜色不对。

	for (int k = 0; k < (w*h / 4); k++)
	{
		ubuffer[k] = 128;
		vbuffer[k] = 128;
	}

是这个样子的图

（图片的正反有问题，后面会解释）

【DPCM】

void DPCM(unsigned char *ybuffer, unsigned char *yPre, unsigned char *yRe, int N, int w, int h)
{
	char *error;
	error = (char *)malloc(sizeof(char)*(w*h));//量化前误差
	int *Qerror;
	Qerror = (int *)malloc(sizeof(int)*(w*h));//量化后误差
	double Nbit = pow(2.0, N);//量化bite数
	double step = 512 / Nbit;//量化间隔

	for (int i = 0; i < h; i++)
	{
		//第一列像素
		error[i*w + 0] = (char)(ybuffer[i*w + 0] - 128);
		Qerror[i*w + 0] = int((error[i*w + 0] + 255) / step + 0.5);//第一列像素预测误差值等于0
		yPre[i*w + 0] = (unsigned char)Qerror[i*w + 0];
		if (yPre[i*w + 0] > Nbit - 1)
			yPre[i*w + 0] = (unsigned char)(Nbit - 1);
		if (yPre[i*w + 0] < 0)
			yPre[i*w + 0] = 0;
		yRe[i*w + 0] = (yPre[i*w + 0] * step - 255) + 128;

	}
	for (int i = 0; i < h; i++)
	{
		for (int j = 1; j < w; j++)
		{
			error[i*w + j] = (char)(ybuffer[i*w + j] - yRe[i*w + j - 1]);
			Qerror[i*w + j] = int((error[i*w + j] + 255) / step + 0.5);//量化
			yPre[i*w + j] = (unsigned char)Qerror[i*w + j];
			if (yPre[i*w + j] > Nbit - 1)
				yPre[i*w + j] = (unsigned char)(Nbit - 1);
			if (yPre[i*w + j] < 0)
				yPre[i*w + j] = 0;
			yRe[i*w + j] = (yPre[i*w + j] * step - 255) + yRe[i*w + j - 1];//反量化
			if (yRe[i*w + j] > 255)
				yRe[i*w + j] = 255;
			if (yRe[i*w + j] < 0)
				yRe[i*w + j] = 0;
		}
	}
}

//DPCM
	DPCM(ybuffer, yPre, yRe, N, w, h);

	//计算频率
	//输出原始图像频率分布
	//FILE *renFile = NULL;
	if ((renFile = fopen("D:\\数据压缩\\DPCM\\ren.txt", "wb")) == NULL)
	{
		printf("open ren fail!");
	}
	double fre1[256] = { 0 };
	for (int i = 0; i < 256; i++)
	{
		for (int j = 0; j < w*h; j++)
		{
			if (ybuffer[j] == i)
				fre1[i]++;
		}
		fre1[i] = fre1[i] / (w*h);
		fprintf(renFile, "%lf\n", fre1[i]);
	}
	//输出重建图像频率分布
	FILE *RebuildFre = NULL;
	if ((RebuildFre = fopen("D:\\数据压缩\\DPCM\\RebuildFre.txt", "wb")) == NULL)
	{
		printf("open RebuildFre fail!");
	}
	double fre2[256] = { 0 };
	for (int i = 0; i < pow(2.0, N); i++)
	{
		for (int j = 0; j < w*h; j++)
		{
			if (yRe[j] == i)
				fre2[i]++;
		}
		fre2[i] = fre2[i] / (w*h);
		fprintf(RebuildFre, "%lf\n", fre2[i]);
	}

【PSNR】

//计算MSE和PSNR
	double MSE = 0;
	double PSNR = 0;
	for (int i = 0; i < w*h; i++)
	{
		MSE = MSE + pow((ybuffer[i] - yRe[i]), 2.0);
	}

	MSE = MSE / (w*h);
	PSNR = 10 * log10((255 * 255) / MSE);
	printf("N: %d\n", N);
	//printf("MSE is %lf\n", MSE);
	printf("PSNR is %lf\n", PSNR);

【写入】

//写入预测误差图像
	fwrite(yPre, 1, w*h, PredictedFile);
	fwrite(ubuffer, 1, w*h / 4, PredictedFile);
	fwrite(vbuffer, 1, w*h / 4, PredictedFile);


	//写入预测图像
	fwrite(yRe, 1, w*h, RebuildFile);
	fwrite(ubuffer, 1, w*h / 4, RebuildFile);
	fwrite(vbuffer, 1, w*h / 4, RebuildFile);

	fclose(renFile);
	fclose(PredictedFile);
	fclose(RebuildFile);
	fclose(renFile);
	fclose(RebuildFre);
	free(ybuffer);
	free(ubuffer);
	free(vbuffer);
	free(yPre);
	free(yRe);
	return 0;

【得到的文件】

【频率分布图】

遇到的问题：
（1）图片的颜色不对，是uv出了问题，于是加了一个判断

	for (int k = 0; k < (w*h / 4); k++)
	{
		ubuffer[k] = 128;
		vbuffer[k] = 128;
	}

（2）图片的正反不对，猜测是因为直接通过改名更改文件文件格式使图片出现了问题。

后面用现成的yuv文件就成功了。PS出问题，没能转换成功，用别的图片完成本实验