1 /*
2 * This file is part of FFmpeg.
3 *
4 * FFmpeg is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * FFmpeg is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with FFmpeg; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17 */
18
19 #include <stddef.h>
20
21 #include "config.h"
22 #include "pixelutils.h"
23
24 #if CONFIG_PIXELUTILS
25 #include <stdlib.h>
26 #include <string.h>
27
28 #include "attributes.h"
29 #include "macros.h"
30
31 #include "x86/pixelutils.h"
32
sad_wxh(const uint8_t *src1, ptrdiff_t stride1, const uint8_t *src2, ptrdiff_t stride2, int w, int h)33 static av_always_inline int sad_wxh(const uint8_t *src1, ptrdiff_t stride1,
34 const uint8_t *src2, ptrdiff_t stride2,
35 int w, int h)
36 {
37 int x, y, sum = 0;
38
39 for (y = 0; y < h; y++) {
40 for (x = 0; x < w; x++)
41 sum += abs(src1[x] - src2[x]);
42 src1 += stride1;
43 src2 += stride2;
44 }
45 return sum;
46 }
47
48 #define DECLARE_BLOCK_FUNCTIONS(size) \
49 static int block_sad_##size##x##size##_c(const uint8_t *src1, ptrdiff_t stride1, \
50 const uint8_t *src2, ptrdiff_t stride2) \
51 { \
52 return sad_wxh(src1, stride1, src2, stride2, size, size); \
53 }
54
55 DECLARE_BLOCK_FUNCTIONS(2)
56 DECLARE_BLOCK_FUNCTIONS(4)
57 DECLARE_BLOCK_FUNCTIONS(8)
58 DECLARE_BLOCK_FUNCTIONS(16)
59 DECLARE_BLOCK_FUNCTIONS(32)
60
61 static const av_pixelutils_sad_fn sad_c[] = {
62 block_sad_2x2_c,
63 block_sad_4x4_c,
64 block_sad_8x8_c,
65 block_sad_16x16_c,
66 block_sad_32x32_c,
67 };
68 #else
69 #include "log.h"
70 #endif /* CONFIG_PIXELUTILS */
71
av_pixelutils_get_sad_fn(int w_bits, int h_bits, int aligned, void *log_ctx)72 av_pixelutils_sad_fn av_pixelutils_get_sad_fn(int w_bits, int h_bits, int aligned, void *log_ctx)
73 {
74 #if !CONFIG_PIXELUTILS
75 av_log(log_ctx, AV_LOG_ERROR, "pixelutils support is required "
76 "but libavutil is not compiled with it\n");
77 return NULL;
78 #else
79 av_pixelutils_sad_fn sad[FF_ARRAY_ELEMS(sad_c)];
80
81 memcpy(sad, sad_c, sizeof(sad));
82
83 if (w_bits < 1 || w_bits > FF_ARRAY_ELEMS(sad) ||
84 h_bits < 1 || h_bits > FF_ARRAY_ELEMS(sad))
85 return NULL;
86 if (w_bits != h_bits) // only squared sad for now
87 return NULL;
88
89 #if ARCH_X86
90 ff_pixelutils_sad_init_x86(sad, aligned);
91 #endif
92
93 return sad[w_bits - 1];
94 #endif
95 }
96