59e3580153a3dfebcb958823a50e33fb5c34e5cf
[ffmpeg.git] / libavcodec / x86 / mpegvideoenc.c
1 /*
2  * The simplest mpeg encoder (well, it was the simplest!)
3  * Copyright (c) 2000,2001 Fabrice Bellard
4  *
5  * This file is part of Libav.
6  *
7  * Libav is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * Libav is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with Libav; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include "libavutil/cpu.h"
23 #include "libavutil/x86/asm.h"
24 #include "libavutil/x86/cpu.h"
25 #include "libavcodec/avcodec.h"
26 #include "libavcodec/dsputil.h"
27 #include "libavcodec/mpegvideo.h"
28 #include "dsputil_mmx.h"
29
30 extern uint16_t ff_inv_zigzag_direct16[64];
31
32 #if HAVE_MMX_INLINE
33 #define COMPILE_TEMPLATE_MMXEXT 0
34 #define COMPILE_TEMPLATE_SSE2   0
35 #define COMPILE_TEMPLATE_SSSE3  0
36 #define RENAME(a) a ## _MMX
37 #define RENAMEl(a) a ## _mmx
38 #include "mpegvideoenc_template.c"
39 #endif /* HAVE_MMX_INLINE */
40
41 #if HAVE_MMXEXT_INLINE
42 #undef COMPILE_TEMPLATE_SSSE3
43 #undef COMPILE_TEMPLATE_SSE2
44 #undef COMPILE_TEMPLATE_MMXEXT
45 #define COMPILE_TEMPLATE_MMXEXT 1
46 #define COMPILE_TEMPLATE_SSE2   0
47 #define COMPILE_TEMPLATE_SSSE3  0
48 #undef RENAME
49 #undef RENAMEl
50 #define RENAME(a) a ## _MMX2
51 #define RENAMEl(a) a ## _mmx2
52 #include "mpegvideoenc_template.c"
53 #endif /* HAVE_MMXEXT_INLINE */
54
55 #if HAVE_SSE2_INLINE
56 #undef COMPILE_TEMPLATE_MMXEXT
57 #undef COMPILE_TEMPLATE_SSE2
58 #undef COMPILE_TEMPLATE_SSSE3
59 #define COMPILE_TEMPLATE_MMXEXT 0
60 #define COMPILE_TEMPLATE_SSE2   1
61 #define COMPILE_TEMPLATE_SSSE3  0
62 #undef RENAME
63 #undef RENAMEl
64 #define RENAME(a) a ## _SSE2
65 #define RENAMEl(a) a ## _sse2
66 #include "mpegvideoenc_template.c"
67 #endif /* HAVE_SSE2_INLINE */
68
69 #if HAVE_SSSE3_INLINE
70 #undef COMPILE_TEMPLATE_MMXEXT
71 #undef COMPILE_TEMPLATE_SSE2
72 #undef COMPILE_TEMPLATE_SSSE3
73 #define COMPILE_TEMPLATE_MMXEXT 0
74 #define COMPILE_TEMPLATE_SSE2   1
75 #define COMPILE_TEMPLATE_SSSE3  1
76 #undef RENAME
77 #undef RENAMEl
78 #define RENAME(a) a ## _SSSE3
79 #define RENAMEl(a) a ## _sse2
80 #include "mpegvideoenc_template.c"
81 #endif /* HAVE_SSSE3_INLINE */
82
83 void ff_MPV_encode_init_x86(MpegEncContext *s)
84 {
85     int mm_flags = av_get_cpu_flags();
86     const int dct_algo = s->avctx->dct_algo;
87
88     if (dct_algo == FF_DCT_AUTO || dct_algo == FF_DCT_MMX) {
89 #if HAVE_MMX_INLINE
90         if (INLINE_MMX(mm_flags))
91             s->dct_quantize = dct_quantize_MMX;
92 #endif
93 #if HAVE_MMXEXT_INLINE
94         if (INLINE_MMXEXT(mm_flags))
95             s->dct_quantize = dct_quantize_MMX2;
96 #endif
97 #if HAVE_SSE2_INLINE
98         if (INLINE_SSE2(mm_flags))
99             s->dct_quantize = dct_quantize_SSE2;
100 #endif
101 #if HAVE_SSSE3_INLINE
102         if (INLINE_SSSE3(mm_flags))
103             s->dct_quantize = dct_quantize_SSSE3;
104 #endif
105     }
106 }