avutil: add pixelutils API
[ffmpeg.git] / 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     if (EXTERNAL_MMXEXT(cpu_flags)) {
47         sad[2] = ff_pixelutils_sad_8x8_mmxext;
48         sad[3] = ff_pixelutils_sad_16x16_mmxext;
49     }
50
51     if (EXTERNAL_SSE2(cpu_flags)) {
52         switch (aligned) {
53         case 0: sad[3] = ff_pixelutils_sad_16x16_sse2;   break; // src1 unaligned, src2 unaligned
54         case 1: sad[3] = ff_pixelutils_sad_u_16x16_sse2; break; // src1   aligned, src2 unaligned
55         case 2: sad[3] = ff_pixelutils_sad_a_16x16_sse2; break; // src1   aligned, src2   aligned
56         }
57     }
58 }