ColorAnalysis.cpp

#include "D:\Solve-Puzzle-Not-Puzzled\main.h"
#include "ColorInfo.h"


vector<ColorInfo> ImgColorDescriptor(Mat img, int row, int col, int colorNumber);
vector <vector<float>> ColorSort(vector<ColorInfo> origin, ColorInfo piece);
ColorInfo AnalysisColor(Mat img,int);
ColorInfo AnalysisColorPNG(Mat img,int);
vector <vector<float>> ColorAnalysis(Mat, Mat, int, int,int);
vector<ColorInfo> OriginColorInfo;

void ShowImageColorInfo(Mat img,vector<ColorInfo> color,int row,int col){

	int i=0;
	Mat part;
	vector<ColorInfo>::iterator it;
	
	for (it = color.begin(); it != color.end(); ++it)
	{
		
		//printf("%d-%d\n", i / col + 1, i%col+1);
		cout << i << "  " << endl;
		it->ShowColorInfo(); 
		//part = img(Range((int)((i / col)*(float)img.rows / row), (int)((float)img.rows / row*((i / col) + 1))), Range((int)((i%col)*(float)img.cols / col), (int)((float)img.cols / col*((i%col) + 1))));	//·|¶V¨Ó¶V¤p
		//namedWindow("part", CV_WINDOW_AUTOSIZE);
		//imshow("part", part);
		//printf("\n\n");
		i++;
		//waitKey(1);
	}
	

}

ColorInfo AnalysisColorPNG(Mat img, int colorNumber){
	int i, j, m;
	int N = colorNumber;    /*class the color into N colors*/
	int h,size=0;

	vector<float> colorArray(N, 0);
	map<int, float> colorPrortion;
	for (i = 0; i < img.rows; ++i){

		Vec3b *p = img.ptr<Vec3b>(i);
		for (j = 0; j < img.cols; ++j){
			if (p[j][1] < COLOR_RANGE / 6){
				if (p[j][2] < COLOR_RANGE / 3){				//black
					h = N-1;
				}
				else if (p[j][2] > 2 * COLOR_RANGE / 3){	//white
					h = N-3;
				}
				else{										//gray
					h = N-2;
				}

			}
			else{
				h = (float)p[j][0] / COLOR_RANGE * N;

			}
			if (p[j][1] == 0 && p[j][2] == 0){
				//cout << "none";
				
			}
			else{
				colorArray[h]++;
				size++;
			}
		}
	}
	for (i = 0; i < colorArray.size(); i++){
		colorArray[i] /= size;
	}
	ColorInfo colorInfo(colorArray);
	colorInfo.ShowColorInfo();
	return colorInfo;

}

ColorInfo AnalysisColor(Mat img, int colorNumber){
	int i, j,m;
	int N = colorNumber;    /*class the color into N colors*/
	int h;
	
	vector<float> colorArray(N,0);
	map<int, float> colorPrortion;
	for (i = 0; i < img.rows; ++i){

		Vec3b *p = img.ptr<Vec3b>(i);
		for (j = 0; j < img.cols; ++j){
			if (p[j][1] < COLOR_RANGE / 6){
				if (p[j][2] < COLOR_RANGE / 3){				//black
					h = N-1;
				}
				else if (p[j][2] > 2 * COLOR_RANGE / 3){	//white
					h = N-3;
				}
				else{										//gray
					h = N-2;
				}
			}
			else{
				h = (float)p[j][0] / COLOR_RANGE * N;
			}
			colorArray[h]++;
		}
	}
	for (i = 0; i < colorArray.size(); i++){
		colorArray[i] /= (img.rows*img.cols);
	}
	ColorInfo colorInfo(colorArray);
	return colorInfo;
}


void PrintColorImage(Mat img, int channel){

	int i, j;
	int h;

	for (i = 0; i < img.rows; ++i){

		Vec3b *p = img.ptr<Vec3b>(i);
		for (j = 0; j < img.cols; ++j){
			h = (float)p[j][channel]/255*6;
			printf("%d ", h);
		}
		printf("\n");
	}
}

void PrintNonColorImage(Mat img){

	int i, j;
	for (i = 0; i < img.rows; ++i){

		uchar *p = img.ptr<uchar>(i);
		for (j = 0; j < img.cols; ++j){
			printf("%3d ", p[j]);
		}
		printf("\n");
	}
}

float ColorCompare(ColorInfo c1, ColorInfo c2,int colorNumber){
	
	int i;
	int size = colorNumber;
 	float compare = 0.0,a;
	
	for (i = 0; i < size; i++){
	
			a = abs(c1.get_colorProportion(i) - c2.get_colorProportion(i));

		compare = compare + a;
	}
	return compare;
}


ColorInfo PieceColorDescriptor(Mat img,int colorNumber){

	Mat hsv;
	cvtColor(img, hsv, CV_BGR2HSV_FULL);

	ColorInfo pieceColor = AnalysisColor(hsv, colorNumber);
	pieceColor.ShowColorInfo();

	return pieceColor;
}




vector<ColorInfo> ImgColorDescriptor(Mat img, int row, int col,int colorNumber){

	Mat hsv;
	vector<Mat> hsv_split, part_split;
	Mat part,part2;
	int i, j;
	int width, height;
	width = hsv.cols / col;
	height = hsv.rows / row;
	cvtColor(img, hsv, CV_BGR2HSV_FULL);
	OriginColorInfo.clear();
	//cout << hsv.type() << endl;
	for (i = 0; i < row; i++){
		for (j = 0; j < col; j++){
			
			part = hsv(Range(hsv.rows / row*i, hsv.rows / row*(i + 1)), Range(hsv.cols / col*j, hsv.cols / col*(j + 1)));
			part2 = img(Range(img.rows / row*i, img.rows / row*(i + 1)), Range(img.cols / col*j, img.cols / col*(j + 1)));
			OriginColorInfo.push_back(AnalysisColor(part, colorNumber));
			/*namedWindow("part", CV_WINDOW_AUTOSIZE);
			imshow("part", part2);
			waitKey(1);*/
		}
	}

//	ShowImageColorInfo(img, OriginColorInfo, row, col);



	/*split(hsv, hsv_split);
	split(part, part_split);
	namedWindow("­ì¹Ï", CV_WINDOW_AUTOSIZE);
	namedWindow("HSV", CV_WINDOW_AUTOSIZE);
	namedWindow("part", CV_WINDOW_AUTOSIZE);
	imshow("­ì¹Ï",img);
	imshow("HSV",hsv);
	imshow("part", part);
	namedWindow("hue", CV_WINDOW_AUTOSIZE);
	namedWindow("saturation", CV_WINDOW_AUTOSIZE);
	namedWindow("value", CV_WINDOW_AUTOSIZE);
	imshow("hue", hsv_split.at(0));
	imshow("saturation", hsv_split.at(1));
	imshow("value", hsv_split.at(2));
	*/
	//PrintColorImage(part, 0);

	return OriginColorInfo;
}


vector <vector<float>> ColorCompareArray(vector<ColorInfo> origin, ColorInfo piece,int colorNumber){

	vector <vector<float>> color_array;
	vector<float> colorInfo;
	int i;
	float compare, max = 0.0, min = 100.0;
	for (i = 0; i < origin.size(); i++)
	{
		compare = ColorCompare(origin[i], piece, colorNumber);
		colorInfo.push_back((float)i);
		colorInfo.push_back(compare);
		color_array.push_back(colorInfo);
		colorInfo.clear();

		if (compare > max)
			max = compare;
		if (compare < min)
			min = compare;
		//cout <<"color  " <<color_array[i][0] << "   " << color_array[i][1] << endl;
	}

	for (i = 0; i < color_array.size(); i++)
	{
		color_array[i][1] = (color_array[i][1] - min)*0.99 / (max - min) + 0.01;
	}

	return color_array;
}

vector <vector<float>> ColorSort(vector <vector<float>> color_sort_array){

	int i;
	sort(color_sort_array.begin(), color_sort_array.end(), sortCompare);

	for (i = 0; i < color_sort_array.size(); i++)
	{
		cout <<i+1<<"  color sort  " <<color_sort_array[i][0] << " : " << color_sort_array[i][1] << endl;
	}

	return color_sort_array;
}


vector <vector<float>> ColorAnalysis(Mat original, Mat piece, int row, int col,int colorNumber){

	vector <vector<float>> colorInfo;
	Mat piece_hsv;
	ColorInfo pieceColor;
	cvtColor(piece, piece_hsv, CV_BGR2HSV_FULL);
	pieceColor = AnalysisColorPNG(piece_hsv, colorNumber);
	colorInfo = ColorCompareArray(OriginColorInfo, pieceColor, colorNumber);
	ColorSort(colorInfo);
	return colorInfo;
}