blob: b4609d7400fe927194d8a2b89cd09ea06f51f1e1 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
#include "dcp_core.h"
void CalcRecover(IplImage *result, IplImage *input, IplImage *transmission, double A[])
{
int height = input->height;
int width = input->width;
int widthstep = input->widthStep;
int gwidthstep = transmission->widthStep;
int nch = input->nChannels;
double t0 = 0.1;
double t, tmp_res;
int i, j, k;
int e = 0, number = 0;
double a = 0;
A[0] /= 255.0;
A[1] /= 255.0;
A[2] /= 255.0;
for (i = 0; i < height; i++)
{
for (j = 0; j < width; j++)
{
t = *(uchar *)(transmission->imageData + i * gwidthstep + j);
t /= 255.0;
for (k = 0; k < 3; k++)
{
a = *(uchar *)(input->imageData + i * widthstep + j * nch + k);
a /= 255.0;
tmp_res = ((a - A[k]) / MAX(t, t0)) + A[k];
tmp_res *= 255.0;
tmp_res = tmp_res > 255 ? 255 : (tmp_res < 0 ? 0 : tmp_res);
*(uchar *)(result->imageData + i * widthstep + j * nch + k) = cvRound(tmp_res);
}
}
}
}
|
