FFmpeg
pixelutils_init.c
Go to the documentation of this file.
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_mmxext(const uint8_t *src1, ptrdiff_t stride1,
25  const uint8_t *src2, ptrdiff_t stride2);
26 
27 int ff_pixelutils_sad_16x16_sse2(const uint8_t *src1, ptrdiff_t stride1,
28  const uint8_t *src2, ptrdiff_t stride2);
29 int ff_pixelutils_sad_a_16x16_sse2(const uint8_t *src1, ptrdiff_t stride1,
30  const uint8_t *src2, ptrdiff_t stride2);
31 int ff_pixelutils_sad_u_16x16_sse2(const uint8_t *src1, ptrdiff_t stride1,
32  const uint8_t *src2, ptrdiff_t stride2);
33 
34 int ff_pixelutils_sad_32x32_sse2(const uint8_t *src1, ptrdiff_t stride1,
35  const uint8_t *src2, ptrdiff_t stride2);
36 int ff_pixelutils_sad_a_32x32_sse2(const uint8_t *src1, ptrdiff_t stride1,
37  const uint8_t *src2, ptrdiff_t stride2);
38 int ff_pixelutils_sad_u_32x32_sse2(const uint8_t *src1, ptrdiff_t stride1,
39  const uint8_t *src2, ptrdiff_t stride2);
40 
41 int ff_pixelutils_sad_32x32_avx2(const uint8_t *src1, ptrdiff_t stride1,
42  const uint8_t *src2, ptrdiff_t stride2);
43 int ff_pixelutils_sad_a_32x32_avx2(const uint8_t *src1, ptrdiff_t stride1,
44  const uint8_t *src2, ptrdiff_t stride2);
45 int ff_pixelutils_sad_u_32x32_avx2(const uint8_t *src1, ptrdiff_t stride1,
46  const uint8_t *src2, ptrdiff_t stride2);
47 
49 {
51 
52  // The best way to use SSE2 would be to do 2 SADs in parallel,
53  // but we'd have to modify the pixelutils API to return SIMD functions.
54 
55  // It's probably not faster to shuffle data around
56  // to get two lines of 8 pixels into a single 16byte register,
57  // so just use the MMX 8x8 version even when SSE2 is available.
60  }
61 
62  if (EXTERNAL_SSE2(cpu_flags)) {
63  switch (aligned) {
64  case 0: sad[3] = ff_pixelutils_sad_16x16_sse2; break; // src1 unaligned, src2 unaligned
65  case 1: sad[3] = ff_pixelutils_sad_u_16x16_sse2; break; // src1 aligned, src2 unaligned
66  case 2: sad[3] = ff_pixelutils_sad_a_16x16_sse2; break; // src1 aligned, src2 aligned
67  }
68  }
69 
70  if (EXTERNAL_SSE2(cpu_flags)) {
71  switch (aligned) {
72  case 0: sad[4] = ff_pixelutils_sad_32x32_sse2; break; // src1 unaligned, src2 unaligned
73  case 1: sad[4] = ff_pixelutils_sad_u_32x32_sse2; break; // src1 aligned, src2 unaligned
74  case 2: sad[4] = ff_pixelutils_sad_a_32x32_sse2; break; // src1 aligned, src2 aligned
75  }
76  }
77 
79  switch (aligned) {
80  case 0: sad[4] = ff_pixelutils_sad_32x32_avx2; break; // src1 unaligned, src2 unaligned
81  case 1: sad[4] = ff_pixelutils_sad_u_32x32_avx2; break; // src1 aligned, src2 unaligned
82  case 2: sad[4] = ff_pixelutils_sad_a_32x32_avx2; break; // src1 aligned, src2 aligned
83  }
84  }
85 }
cpu.h
src1
const pixel * src1
Definition: h264pred_template.c:421
ff_pixelutils_sad_8x8_mmxext
int ff_pixelutils_sad_8x8_mmxext(const uint8_t *src1, ptrdiff_t stride1, const uint8_t *src2, ptrdiff_t stride2)
EXTERNAL_AVX2_FAST
#define EXTERNAL_AVX2_FAST(flags)
Definition: cpu.h:79
av_pixelutils_sad_fn
int(* av_pixelutils_sad_fn)(const uint8_t *src1, ptrdiff_t stride1, const uint8_t *src2, ptrdiff_t stride2)
Sum of abs(src1[x] - src2[x])
Definition: pixelutils.h:28
av_get_cpu_flags
int av_get_cpu_flags(void)
Return the flags which specify extensions supported by the CPU.
Definition: cpu.c:109
pixelutils.h
cpu_flags
static atomic_int cpu_flags
Definition: cpu.c:56
ff_pixelutils_sad_init_x86
void ff_pixelutils_sad_init_x86(av_pixelutils_sad_fn *sad, int aligned)
Definition: pixelutils_init.c:48
ff_pixelutils_sad_32x32_sse2
int ff_pixelutils_sad_32x32_sse2(const uint8_t *src1, ptrdiff_t stride1, const uint8_t *src2, ptrdiff_t stride2)
aligned
static int aligned(int val)
Definition: dashdec.c:171
ff_pixelutils_sad_a_32x32_avx2
int ff_pixelutils_sad_a_32x32_avx2(const uint8_t *src1, ptrdiff_t stride1, const uint8_t *src2, ptrdiff_t stride2)
ff_pixelutils_sad_u_16x16_sse2
int ff_pixelutils_sad_u_16x16_sse2(const uint8_t *src1, ptrdiff_t stride1, const uint8_t *src2, ptrdiff_t stride2)
ff_pixelutils_sad_16x16_sse2
int ff_pixelutils_sad_16x16_sse2(const uint8_t *src1, ptrdiff_t stride1, const uint8_t *src2, ptrdiff_t stride2)
ff_pixelutils_sad_u_32x32_avx2
int ff_pixelutils_sad_u_32x32_avx2(const uint8_t *src1, ptrdiff_t stride1, const uint8_t *src2, ptrdiff_t stride2)
EXTERNAL_SSE2
#define EXTERNAL_SSE2(flags)
Definition: cpu.h:59
ff_pixelutils_sad_32x32_avx2
int ff_pixelutils_sad_32x32_avx2(const uint8_t *src1, ptrdiff_t stride1, const uint8_t *src2, ptrdiff_t stride2)
src2
const pixel * src2
Definition: h264pred_template.c:422
ff_pixelutils_sad_a_16x16_sse2
int ff_pixelutils_sad_a_16x16_sse2(const uint8_t *src1, ptrdiff_t stride1, const uint8_t *src2, ptrdiff_t stride2)
ff_pixelutils_sad_a_32x32_sse2
int ff_pixelutils_sad_a_32x32_sse2(const uint8_t *src1, ptrdiff_t stride1, const uint8_t *src2, ptrdiff_t stride2)
EXTERNAL_MMXEXT
#define EXTERNAL_MMXEXT(flags)
Definition: cpu.h:57
ff_pixelutils_sad_u_32x32_sse2
int ff_pixelutils_sad_u_32x32_sse2(const uint8_t *src1, ptrdiff_t stride1, const uint8_t *src2, ptrdiff_t stride2)