pixelutils.c (2827B)
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 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 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 }