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/transmission.cpp | 51 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 DCP/transmission.cpp

(limited to 'DCP/transmission.cpp')

diff --git a/DCP/transmission.cpp b/DCP/transmission.cpp
new file mode 100644
index 0000000..5bb2ece
--- /dev/null
+++ b/DCP/transmission.cpp
@@ -0,0 +1,51 @@
+#include "dcp_core.h"
+
+void Estimate_transmission(IplImage *Transmission_maps, IplImage *InputImage, double A[], double heap[], double real_trans[])
+{
+	double w = 0.95;
+	double tmp;
+	double tran;
+	int i, j, k;
+	int e = 0;
+
+	IplImage *tmp_Trans = cvCreateImage(cvSize(width, height), IPL_DEPTH_8U, 3);
+
+	for (k = 0; k < 3; k++)
+	{
+		for (i = 1; i <= height; i++)
+		{
+			for (j = 1; j <= width; j++)
+			{
+				tmp = *(uchar *)(InputImage->imageData + (i - 1) * widthstep + (j - 1) * nch + k);
+				tmp /= A[k];
+
+				if (tmp > 255)
+					tmp = 255;
+				*(uchar *)(tmp_Trans->imageData + (i - 1) * widthstep + (j - 1) * nch + k) = tmp;
+			}
+		}
+	}
+	Cal_DarkChannel(Transmission_maps, tmp_Trans, nInteger, heap);
+
+	for (i = 1; i <= height; i++)
+	{
+		for (j = 1; j <= width; j++)
+		{
+			tran = *(uchar *)(Transmission_maps->imageData + (i - 1) * gwidthstep + (j - 1));
+
+			tran /= 255.0;
+			tran = 1 - w * tran;
+
+			real_trans[e++] = tran * 255.0;
+
+			if (tran > 1)
+				tran = 1;
+			else if (tran < 0)
+				tran = 0;
+
+			*(uchar *)(Transmission_maps->imageData + (i - 1) * gwidthstep + (j - 1)) = tran * 255;
+		}
+	}
+	//cvSaveImage("Estimated-Trans.bmp", Transmission_maps);
+	cvReleaseImage(&tmp_Trans);
+}
-- 
cgit v1.2.3