DCTLs/Median.dctl at master · rodrigomas/DCTLs · GitHub

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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
__DEVICE__ float median(float p_Table[], int m) {
float temp;
int i, j;
for(i = 0; i < m - 1; i++) {
        for(j = i + 1; j < m; j++) {
            if(p_Table[j] < p_Table[i]) {
                temp = p_Table[i];
                p_Table[i] = p_Table[j];
                p_Table[j] = temp;
                return p_Table[m / 2];
            }
        }
    }

  }

__DEVICE__ float3 transform(int p_Width, int p_Height, int p_X, int p_Y, __TEXTURE__ p_TexR, __TEXTURE__ p_TexG, __TEXTURE__ p_TexB)
{

 float TableR[9] =
{
    _tex2D(p_TexR, p_X - 1, p_Y + 1),
    _tex2D(p_TexR, p_X, p_Y + 1),
    _tex2D(p_TexR, p_X + 1, p_Y + 1),
    _tex2D(p_TexR, p_X - 1, p_Y),
    _tex2D(p_TexR, p_X, p_Y),
    _tex2D(p_TexR, p_X + 1, p_Y),
    _tex2D(p_TexR, p_X - 1, p_Y - 1),
    _tex2D(p_TexR, p_X, p_Y - 1),
    _tex2D(p_TexR, p_X + 1, p_Y - 1)
};

 float TableG[9] =
{
    _tex2D(p_TexG, p_X - 1, p_Y + 1),
    _tex2D(p_TexG, p_X, p_Y + 1),
    _tex2D(p_TexG, p_X + 1, p_Y + 1),
    _tex2D(p_TexG, p_X - 1, p_Y),
    _tex2D(p_TexG, p_X, p_Y),
    _tex2D(p_TexG, p_X + 1, p_Y),
    _tex2D(p_TexG, p_X - 1, p_Y - 1),
    _tex2D(p_TexG, p_X, p_Y - 1),
    _tex2D(p_TexG, p_X + 1, p_Y - 1)
};

 float TableB[9] =
{
    _tex2D(p_TexB, p_X - 1, p_Y + 1),
    _tex2D(p_TexB, p_X, p_Y + 1),
    _tex2D(p_TexB, p_X + 1, p_Y + 1),
    _tex2D(p_TexB, p_X - 1, p_Y),
    _tex2D(p_TexB, p_X, p_Y),
    _tex2D(p_TexB, p_X + 1, p_Y),
    _tex2D(p_TexB, p_X - 1, p_Y - 1),
    _tex2D(p_TexB, p_X, p_Y - 1),
    _tex2D(p_TexB, p_X + 1, p_Y - 1)
};

const float r = median(TableR, 9);
const float g = median(TableG, 9);
const float b = median(TableB, 9);

return make_float3(r, g, b);

}