blob: c41a290bd50f2818fbb0bde483ac4bb206297f8e (
plain)
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
66
67
68
69
70
71
|
#define SOBEL_RADIUS 1
kernel void kernel_edgesobel(global float4 *in,
global float4 *out,
const int horizontal,
const int vertical,
const int keep_sign,
const int has_alpha)
{
int gidx = get_global_id(0);
int gidy = get_global_id(1);
float4 hor_grad = 0.0f;
float4 ver_grad = 0.0f;
float4 gradient = 0.0f;
int dst_width = get_global_size(0);
int src_width = dst_width + SOBEL_RADIUS * 2;
int i = gidx + SOBEL_RADIUS, j = gidy + SOBEL_RADIUS;
int gid1d = i + j * src_width;
float4 pix_fl = in[gid1d - 1 - src_width];
float4 pix_fm = in[gid1d - src_width];
float4 pix_fr = in[gid1d + 1 - src_width];
float4 pix_ml = in[gid1d - 1 ];
float4 pix_mm = in[gid1d ];
float4 pix_mr = in[gid1d + 1 ];
float4 pix_bl = in[gid1d - 1 + src_width];
float4 pix_bm = in[gid1d + src_width];
float4 pix_br = in[gid1d + 1 + src_width];
if (horizontal)
{
hor_grad +=
- 1.0f * pix_fl + 1.0f * pix_fr
- 2.0f * pix_ml + 2.0f * pix_mr
- 1.0f * pix_bl + 1.0f * pix_br;
}
if (vertical)
{
ver_grad +=
- 1.0f * pix_fl - 2.0f * pix_fm
- 1.0f * pix_fr + 1.0f * pix_bl
+ 2.0f * pix_bm + 1.0f * pix_br;
}
if (horizontal && vertical)
{
gradient = sqrt(
hor_grad * hor_grad +
ver_grad * ver_grad) / 1.41f;
}
else
{
if (keep_sign)
gradient = hor_grad + ver_grad;
else
gradient = fabs(hor_grad + ver_grad);
}
if (has_alpha)
{
gradient.w = pix_mm.w;
}
else
{
gradient.w = 1.0f;
}
out[gidx + gidy * dst_width] = gradient;
}
|
