From 17fbf4355747b1ceb9caca120aec9830a44621cf Mon Sep 17 00:00:00 2001
From: JinweiClarkChao <clarkzjw@gmail.com>
Date: Mon, 24 Aug 2015 10:15:00 +0800
Subject: init

---
 DCP/airlight.cpp | 97 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 DCP/airlight.cpp

(limited to 'DCP/airlight.cpp')

diff --git a/DCP/airlight.cpp b/DCP/airlight.cpp
new file mode 100644
index 0000000..c6df18f
--- /dev/null
+++ b/DCP/airlight.cpp
@@ -0,0 +1,97 @@
+#include "dcp_core.h"
+
+void Estimate_A(IplImage *DarkChannel, IplImage *InputImage, struct Max_Pixel_Selector *MaxPixel, double A[])
+{
+	double num = height * width * 0.001;
+	double num1per = num * 0.01;   //num1per is the number of 1 percent of top 0.1 pixels
+	//k is the index of the color channel
+	int k = 0;
+	int patchsize;
+	int dark_i, dark_j;
+	uchar Max = 0;
+
+	IplImage *mask = cvCreateImage(cvSize(width, height), IPL_DEPTH_8U, 1);
+	cvZero(mask);
+
+	int e = 0, i, j;
+	for (i = 1; i <= height; i++)
+	{
+		for (j = 1; j <= width; j++)
+		{
+			MaxPixel[e].Pixel = *(uchar *)(DarkChannel->imageData + (i - 1) * gwidthstep + (j - 1));
+			MaxPixel[e].i = i;
+			MaxPixel[e].j = j;
+			e++;
+		}
+	}
+	sort(MaxPixel, MaxPixel + e, comp1);
+
+	for (i = 0; i < num; i++)
+	{
+		dark_i = MaxPixel[i].i;
+		dark_j = MaxPixel[i].j;
+		*(uchar *)(mask->imageData + (dark_i - 1) * gwidthstep + (dark_j - 1)) = 255;
+		//只有当mask标记为255之后，才说明该块区域亮度值是在暗通道的前0.1%之内的
+	}
+
+	if (num1per < 9)
+		patchsize = 3;
+	else
+		patchsize = sqrt(num1per);
+
+	for (k = 0; k < 3; k++)
+	{
+		int e, x, y;
+		int tmp = (patchsize - 1) / 2;
+		int st_row, ed_row;
+		int st_col, ed_col;
+
+		double A_tmp = 0;
+		double aver = 0;  // aver 为指定大小patch内InputImage值的平均值
+		int flag;  //flag用来获取mask的标记
+
+		for (i = 1; i <= num; i++)
+		{
+			aver = 0;
+			dark_i = MaxPixel[i].i;
+			dark_j = MaxPixel[i].j;
+
+			st_row = dark_i - tmp, ed_row = dark_i + tmp;
+			st_col = dark_j - tmp, ed_col = dark_j + tmp;
+
+			if (st_row <= 0)
+				st_row = 1;
+			if (ed_row>height)
+				ed_row = height;
+			if (st_col <= 0)
+				st_col = 1;
+			if (ed_col>width)
+				ed_col = width;
+
+			e = 0;
+			for (x = st_row; x <= ed_row; x++)
+			{
+				for (y = st_col; y <= ed_col; y++)
+				{
+					flag = *(uchar *)(mask->imageData + (x - 1) * gwidthstep + (y - 1));
+					if (flag == 0)   //未被标记说明该点不在前0.1%对应的区域内，跳过
+					{
+						continue;
+					}
+					else
+					{
+						aver += *(uchar *)(InputImage->imageData + (x - 1) * widthstep + (y - 1) *nch + k);
+						e++;
+					}
+				}
+			}
+			aver /= e;
+			if (A_tmp < aver)
+			{
+				A_tmp = aver;
+			}
+		}
+		A[k] = A_tmp / 255.0;
+	}
+	cvReleaseImage(&mask);
+}
\ No newline at end of file
-- 
cgit v1.2.3