libavutil/pixelutils.c

   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 }