90e62715db5d8ad3e63386587521b43419cdd60c
[ffmpeg.git] / libavcodec / x86 / dsputil_init.c
1 /*
2  * This file is part of Libav.
3  *
4  * Libav 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  * Libav 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 Libav; 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 #include "libavutil/attributes.h"
21 #include "libavutil/cpu.h"
22 #include "libavutil/x86/cpu.h"
23 #include "libavcodec/avcodec.h"
24 #include "libavcodec/dsputil.h"
25 #include "libavcodec/simple_idct.h"
26 #include "dsputil_x86.h"
27 #include "idct_xvid.h"
28
29 static av_cold void dsputil_init_mmx(DSPContext *c, AVCodecContext *avctx,
30                                      int cpu_flags, unsigned high_bit_depth)
31 {
32 #if HAVE_MMX_INLINE
33     c->put_pixels_clamped        = ff_put_pixels_clamped_mmx;
34     c->put_signed_pixels_clamped = ff_put_signed_pixels_clamped_mmx;
35     c->add_pixels_clamped        = ff_add_pixels_clamped_mmx;
36
37     if (!high_bit_depth) {
38         c->draw_edges   = ff_draw_edges_mmx;
39
40         switch (avctx->idct_algo) {
41         case FF_IDCT_AUTO:
42         case FF_IDCT_SIMPLEMMX:
43             c->idct_put              = ff_simple_idct_put_mmx;
44             c->idct_add              = ff_simple_idct_add_mmx;
45             c->idct                  = ff_simple_idct_mmx;
46             c->idct_permutation_type = FF_SIMPLE_IDCT_PERM;
47             break;
48         case FF_IDCT_XVIDMMX:
49             c->idct_put              = ff_idct_xvid_mmx_put;
50             c->idct_add              = ff_idct_xvid_mmx_add;
51             c->idct                  = ff_idct_xvid_mmx;
52             break;
53         }
54     }
55 #endif /* HAVE_MMX_INLINE */
56 }
57
58 static av_cold void dsputil_init_mmxext(DSPContext *c, AVCodecContext *avctx,
59                                         int cpu_flags, unsigned high_bit_depth)
60 {
61 #if HAVE_MMXEXT_INLINE
62     if (!high_bit_depth && avctx->idct_algo == FF_IDCT_XVIDMMX) {
63         c->idct_put = ff_idct_xvid_mmxext_put;
64         c->idct_add = ff_idct_xvid_mmxext_add;
65         c->idct     = ff_idct_xvid_mmxext;
66     }
67 #endif /* HAVE_MMXEXT_INLINE */
68 }
69
70 static av_cold void dsputil_init_sse2(DSPContext *c, AVCodecContext *avctx,
71                                       int cpu_flags, unsigned high_bit_depth)
72 {
73 #if HAVE_SSE2_INLINE
74     if (!high_bit_depth && avctx->idct_algo == FF_IDCT_XVIDMMX) {
75         c->idct_put              = ff_idct_xvid_sse2_put;
76         c->idct_add              = ff_idct_xvid_sse2_add;
77         c->idct                  = ff_idct_xvid_sse2;
78         c->idct_permutation_type = FF_SSE2_IDCT_PERM;
79     }
80 #endif /* HAVE_SSE2_INLINE */
81 }
82
83 av_cold void ff_dsputil_init_x86(DSPContext *c, AVCodecContext *avctx,
84                                  unsigned high_bit_depth)
85 {
86     int cpu_flags = av_get_cpu_flags();
87
88     if (X86_MMX(cpu_flags))
89         dsputil_init_mmx(c, avctx, cpu_flags, high_bit_depth);
90
91     if (X86_MMXEXT(cpu_flags))
92         dsputil_init_mmxext(c, avctx, cpu_flags, high_bit_depth);
93
94     if (X86_SSE2(cpu_flags))
95         dsputil_init_sse2(c, avctx, cpu_flags, high_bit_depth);
96
97     if (CONFIG_ENCODERS)
98         ff_dsputilenc_init_mmx(c, avctx, high_bit_depth);
99 }