init

1 files changed, 51 insertions, 0 deletions

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);

+}