Add PerfectReflectionAlgorithm.cpp

Author: BBuf

Correction algorithm/PerfectReflectionAlgorithm.cpp

@@ -0,0 +1,72 @@
+Mat PerfectReflectionAlgorithm(Mat src) {
+	int row = src.rows;
+	int col = src.cols;
+	Mat dst(row, col, CV_8UC3);
+	int HistRGB[767] = { 0 };
+	int MaxVal = 0;
+	for (int i = 0; i < row; i++) {
+		for (int j = 0; j < col; j++) {
+			MaxVal = max(MaxVal, (int)src.at<Vec3b>(i, j)[0]);
+			MaxVal = max(MaxVal, (int)src.at<Vec3b>(i, j)[1]);
+			MaxVal = max(MaxVal, (int)src.at<Vec3b>(i, j)[2]);
+			int sum = src.at<Vec3b>(i, j)[0] + src.at<Vec3b>(i, j)[1] + src.at<Vec3b>(i, j)[2];
+			HistRGB[sum]++;
+		}
+	}
+	int Threshold = 0;
+	int sum = 0;
+	for (int i = 766; i >= 0; i--) {
+		sum += HistRGB[i];
+		if (sum > row * col * 0.1) {
+			Threshold = i;
+			break;
+		}
+	}
+	int AvgB = 0;
+	int AvgG = 0;
+	int AvgR = 0;
+	int cnt = 0;
+	for (int i = 0; i < row; i++) {
+		for (int j = 0; j < col; j++) {
+			int sumP = src.at<Vec3b>(i, j)[0] + src.at<Vec3b>(i, j)[1] + src.at<Vec3b>(i, j)[2];
+			if (sumP > Threshold) {
+				AvgB += src.at<Vec3b>(i, j)[0];
+				AvgG += src.at<Vec3b>(i, j)[1];
+				AvgR += src.at<Vec3b>(i, j)[2];
+				cnt++;
+			}
+		}
+	}
+	AvgB /= cnt;
+	AvgG /= cnt;
+	AvgR /= cnt;
+	for (int i = 0; i < row; i++) {
+		for (int j = 0; j < col; j++) {
+			int Blue = src.at<Vec3b>(i, j)[0] * MaxVal / AvgB;
+			int Green = src.at<Vec3b>(i, j)[1] * MaxVal / AvgG;
+			int Red = src.at<Vec3b>(i, j)[2] * MaxVal / AvgR;
+			if (Red > 255) {
+				Red = 255;
+			}
+			else if (Red < 0) {
+				Red = 0;
+			}
+			if (Green > 255) {
+				Green = 255;
+			}
+			else if (Green < 0) {
+				Green = 0;
+			}
+			if (Blue > 255) {
+				Blue = 255;
+			}
+			else if (Blue < 0) {
+				Blue = 0;
+			}
+			dst.at<Vec3b>(i, j)[0] = Blue;
+			dst.at<Vec3b>(i, j)[1] = Green;
+			dst.at<Vec3b>(i, j)[2] = Red;
+		}
+	}
+	return dst;
+}

Correction algorithm/README.md

@@ -2,4 +2,5 @@
 
 - GammaCorrection.cpp Gamma 矫正C++实现。原理附在了代码中
 - HistograEqualization.cpp 直方图均衡化C++实现。原理请看：https://blog.csdn.net/just_sort/article/details/85013803
-- Gray World.cpp 对灰度世界算法的实现。原理请看：https://blog.csdn.net/just_sort/article/details/85638420
+- Gray World.cpp 对灰度世界算法的C++实现。原理请看：https://blog.csdn.net/just_sort/article/details/85638420
+- PerfectReflectionAlgorithm.cpp 对完美反射算法的C++实现。原理请看：https://blog.csdn.net/just_sort/article/details/85982871