init

1 files changed, 51 insertions, 0 deletions

diff --git a/DCP/darkchannel.cpp b/DCP/darkchannel.cpp
new file mode 100644
index 0000000..a6cd530
--- /dev/null
+++ b/DCP/darkchannel.cpp
@@ -0,0 +1,51 @@
+#include "dcp_core.h"

+

+static inline int minbgr(int b, int g, int r)

+{

+        b = b > g ? g : b;

+        b = b > r ? r : b;

+        return b;

+}

+

+void CalcDarkChannel(IplImage *darkchannel, IplImage *input, int radius)

+{

+        int height = input->height;

+        int width = input->width;

+        int widthstep = input->widthStep;

+        int gwidthstep = darkchannel->widthStep;

+        int nch = input->nChannels;

+

+        int st_row, ed_row;

+        int st_col, ed_col;

+

+        for (int i = 1; i <= height; i++)

+        {

+                for (int j = 1; j <= width; j++)

+                {

+                        st_row = i - radius, ed_row = i + radius;

+                        st_col = j - radius, ed_col = j + radius;

+

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

+

+                        int min = 0;

+                        int tmp[3];

+                        for (int m = st_row; m <= ed_row; m++)

+                        {

+                                for (int n = st_col; n <= ed_col; n++)

+                                {

+                                        for (int k = 0; k < 3; k++)

+                                                tmp[k] = *(uchar *)(input->imageData + (i - 1) * widthstep + (j - 1) * nch + k);

+                                        min = minbgr(tmp[0], tmp[1], tmp[2]);

+                                }

+                        }

+                        *(uchar *)(darkchannel->imageData + (i - 1) * gwidthstep + (j - 1)) = min;

+                }

+        }

+}