avfilter/x86/vf_blend : reorganize DIFFERENCE macro to reduce line duplication betwee...
[ffmpeg.git] / libavfilter / x86 / vf_threshold_init.c
1 /*
2  * Copyright (c) 2015 Paul B Mahol
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 #include "libavutil/attributes.h"
22 #include "libavutil/cpu.h"
23 #include "libavutil/x86/cpu.h"
24 #include "libavfilter/threshold.h"
25
26 #define THRESHOLD_FUNC(depth, opt) \
27 void ff_threshold##depth##_##opt(const uint8_t *in, const uint8_t *threshold,\
28                                 const uint8_t *min, const uint8_t *max,     \
29                                 uint8_t *out,                               \
30                                 ptrdiff_t ilinesize, ptrdiff_t tlinesize,   \
31                                 ptrdiff_t flinesize, ptrdiff_t slinesize,   \
32                                 ptrdiff_t olinesize,                        \
33                                 int w, int h);
34
35 THRESHOLD_FUNC(8, sse4)
36 THRESHOLD_FUNC(8, avx2)
37 THRESHOLD_FUNC(16, sse4)
38 THRESHOLD_FUNC(16, avx2)
39
40 av_cold void ff_threshold_init_x86(ThresholdContext *s)
41 {
42     int cpu_flags = av_get_cpu_flags();
43
44     if (s->depth == 8) {
45         if (EXTERNAL_SSE4(cpu_flags)) {
46             s->threshold = ff_threshold8_sse4;
47         }
48         if (EXTERNAL_AVX2_FAST(cpu_flags)) {
49             s->threshold = ff_threshold8_avx2;
50         }
51     } else if (s->depth == 16) {
52         if (EXTERNAL_SSE4(cpu_flags)) {
53             s->threshold = ff_threshold16_sse4;
54         }
55         if (EXTERNAL_AVX2_FAST(cpu_flags)) {
56             s->threshold = ff_threshold16_avx2;
57         }
58     }
59 }