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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
#ifndef SUMTHRESHOLD_H
#define SUMTHRESHOLD_H
#include <cstddef>
#include <cstring>
#include "../../structures/image2d.h"
#include "../../structures/mask2d.h"
class SumThreshold {
public:
struct VerticalScratch {
VerticalScratch(size_t width, size_t height);
std::unique_ptr<int[], decltype(&free)> lastFlaggedPos;
std::unique_ptr<num_t[], decltype(&free)> sum;
std::unique_ptr<int[], decltype(&free)> count;
};
template <size_t Length>
static void Horizontal(const Image2D* input, Mask2D* mask, num_t threshold);
template <size_t Length>
static void Vertical(const Image2D* input, Mask2D* mask, num_t threshold);
template <size_t Length>
static void HorizontalLarge(const Image2D* input, Mask2D* mask,
Mask2D* scratch, num_t threshold);
#ifdef __SSE__
template <size_t Length>
static void VerticalLargeSSE(const Image2D* input, Mask2D* mask,
Mask2D* scratch, num_t threshold);
static void VerticalLargeSSE(const Image2D* input, Mask2D* mask,
Mask2D* scratch, size_t length, num_t threshold);
template <size_t Length>
static void HorizontalLargeSSE(const Image2D* input, Mask2D* mask,
Mask2D* scratch, num_t threshold);
static void HorizontalLargeSSE(const Image2D* input, Mask2D* mask,
Mask2D* scratch, size_t length,
num_t threshold);
#endif
#ifdef __AVX2__
template <size_t Length>
static void VerticalLargeAVX(const Image2D* input, Mask2D* mask,
Mask2D* scratch, num_t threshold);
static void VerticalLargeAVX(const Image2D* input, Mask2D* mask,
Mask2D* scratch, size_t length, num_t threshold);
static void HorizontalAVXDumas(const Image2D* input, Mask2D* mask,
size_t length, num_t threshold);
static void VerticalAVXDumas(const Image2D* input, Mask2D* mask,
VerticalScratch* scratch, size_t length,
num_t threshold);
template <size_t Length>
static void HorizontalAVXDumas(const Image2D* input, Mask2D* mask,
num_t threshold);
template <size_t Length>
static void VerticalAVXDumas(const Image2D* input, Mask2D* mask,
VerticalScratch* scratch, num_t threshold);
#endif
template <size_t Length>
static void VerticalLarge(const Image2D* input, Mask2D* mask, Mask2D* scratch,
num_t threshold);
template <size_t Length>
static void Large(const Image2D* input, Mask2D* mask, num_t hThreshold,
num_t vThreshold) {
HorizontalLarge<Length>(input, mask, hThreshold);
VerticalLarge<Length>(input, mask, vThreshold);
}
static void VerticalLarge(const Image2D* input, Mask2D* mask, Mask2D* scratch,
VerticalScratch* vScratch, size_t length,
num_t threshold) {
#if defined(__AVX2__)
VerticalAVXDumas(input, mask, vScratch, length, threshold);
#elif defined(__SSE__)
VerticalLargeSSE(input, mask, scratch, length, threshold);
#else
VerticalLargeReference(input, mask, scratch, length, threshold);
#endif
}
static void VerticalLargeReference(const Image2D* input, Mask2D* mask,
Mask2D* scratch, size_t length,
num_t threshold);
static void HorizontalLargeReference(const Image2D* input, Mask2D* mask,
Mask2D* scratch, size_t length,
num_t threshold);
static void HorizontalLarge(const Image2D* input, Mask2D* mask,
Mask2D* scratch, size_t length, num_t threshold) {
#if defined(__AVX2__)
if (length >= 64)
HorizontalAVXDumas(input, mask, length, threshold);
else
HorizontalLargeSSE(input, mask, scratch, length, threshold);
#elif defined(__SSE__)
HorizontalLargeSSE(input, mask, scratch, length, threshold);
#else
HorizontalLargeReference(input, mask, scratch, length, threshold);
#endif
}
};
#endif
|
