git.sesse.net Git - ffmpeg/blob - libavutil/x86/pixelutils_init.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 "config.h"
  20
  21 #include "pixelutils.h"
  22 #include "cpu.h"
  23
  24 int ff_pixelutils_sad_8x8_mmx(const uint8_t *src1, ptrdiff_t stride1,
  25                               const uint8_t *src2, ptrdiff_t stride2);
  26 int ff_pixelutils_sad_8x8_mmxext(const uint8_t *src1, ptrdiff_t stride1,
  27                                  const uint8_t *src2, ptrdiff_t stride2);
  28
  29 int ff_pixelutils_sad_16x16_mmxext(const uint8_t *src1, ptrdiff_t stride1,
  30                                    const uint8_t *src2, ptrdiff_t stride2);
  31 int ff_pixelutils_sad_16x16_sse2(const uint8_t *src1, ptrdiff_t stride1,
  32                                  const uint8_t *src2, ptrdiff_t stride2);
  33 int ff_pixelutils_sad_a_16x16_sse2(const uint8_t *src1, ptrdiff_t stride1,
  34                                    const uint8_t *src2, ptrdiff_t stride2);
  35 int ff_pixelutils_sad_u_16x16_sse2(const uint8_t *src1, ptrdiff_t stride1,
  36                                    const uint8_t *src2, ptrdiff_t stride2);
  37
  38 void ff_pixelutils_sad_init_x86(av_pixelutils_sad_fn *sad, int aligned)
  39 {
  40     int cpu_flags = av_get_cpu_flags();
  41
  42     if (EXTERNAL_MMX(cpu_flags)) {
  43         sad[2] = ff_pixelutils_sad_8x8_mmx;
  44     }
  45
  46     // The best way to use SSE2 would be to do 2 SADs in parallel,
  47     // but we'd have to modify the pixelutils API to return SIMD functions.
  48
  49     // It's probably not faster to shuffle data around
  50     // to get two lines of 8 pixels into a single 16byte register,
  51     // so just use the MMX 8x8 version even when SSE2 is available.
  52     if (EXTERNAL_MMXEXT(cpu_flags)) {
  53         sad[2] = ff_pixelutils_sad_8x8_mmxext;
  54         sad[3] = ff_pixelutils_sad_16x16_mmxext;
  55     }
  56
  57     if (EXTERNAL_SSE2(cpu_flags)) {
  58         switch (aligned) {
  59         case 0: sad[3] = ff_pixelutils_sad_16x16_sse2;   break; // src1 unaligned, src2 unaligned
  60         case 1: sad[3] = ff_pixelutils_sad_u_16x16_sse2; break; // src1   aligned, src2 unaligned
  61         case 2: sad[3] = ff_pixelutils_sad_a_16x16_sse2; break; // src1   aligned, src2   aligned
  62         }
  63     }
  64 }