OpenCV：图像编解码

#include <iostream>
#include <opencv2/opencv.hpp>
#include <highgui.h>
using namespace std;
using namespace cv;

double getPSNR(Mat& src1, Mat& src2, int bb=0)
{
	int i, j;
	double sse, mse, psnr;
	sse = 0.0;
	Mat s1, s2;
	cvtColor(src1, s1, CV_BGR2GRAY);
	cvtColor(src2, s2, CV_BGR2GRAY);
	int count = 0;
	for (j = bb; j < s1.rows - bb; j++)
	{
		uchar* d = s1.ptr(j);
		uchar* s = s2.ptr(j);
		for (i = bb; i < s1.cols - bb; i++)
		{
			sse += ((d[i] - s[i]) * (d[i] - s[i]));
			count++;
		}
	}
	if (sse == 0.0 || count == 0)
	{
		return 0;
	}
	else
	{
		mse = sse / (double)(count);
		psnr = 10.0 * log10((255 * 255) / mse);
		return psnr;
	}
}

int main(int argc, char* argv) {
	
	Mat src = imread("lena256.tif");//读入一张图片
	vector<uchar> buff;//编码后的缓存
	vector<int> param = vector<int>(2);
	param[0] = CV_IMWRITE_JPEG_QUALITY;
	param[1] = 10;//default(95) 0-100
	char info[64];
	namedWindow("jpeg", CV_WINDOW_AUTOSIZE);
	//createTrackbar("quality", "jpeg", &param[1], 100);
	//createTrackbar放外面也可以
	int key = 0;
	while (key != 'q') {
		createTrackbar("quality", "jpeg", &param[1], 100);
		imencode(".jpg", src, buff, param);
		Mat jpegimage = imdecode(Mat(buff), CV_LOAD_IMAGE_COLOR);
		double psnr = getPSNR(src, jpegimage);//get PSNR
		sprintf(info, "PSNR=%.1f(dB)", psnr);
		putText(jpegimage, info, Point(0, 50), FONT_HERSHEY_SIMPLEX, 1, CV_RGB(255, 255, 255), 2);
		imshow("jpeg", jpegimage);
		key = waitKey(33);

		/***************<png>****************/
		//param[0] = CV_IMWRITE_PNG_COMPRESSION;
		//param[1] = 3;//default(3) 0-9.
		//createTrackbar("quality", "jpeg", &param[1], 10);
		//imencode(".png", src, buff, param);
		//Mat pngimage = imdecode(Mat(buff), CV_LOAD_IMAGE_COLOR);
		//double psnr = getPSNR(src, pngimage);//get PSNR
		//sprintf(info, "PSNR=%.1f(dB)", psnr);
		//putText(pngimage, info, Point(0, 50), FONT_HERSHEY_SIMPLEX, 1, CV_RGB(255, 255, 255), 2);
		//imshow("jpeg", pngimage);
		//key = waitKey(33);
		/***************</png>****************/
	}
	return 0;
}