2c46c6361bb09d5595f14dff7ad5f4bb35323df9
[ffmpeg.git] / libavcodec / x86 / fft.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 "libavcodec/dsputil.h"
20 #include "fft.h"
21
22 av_cold void ff_fft_init_mmx(FFTContext *s)
23 {
24 #if HAVE_YASM
25     int has_vectors = mm_support();
26     if (has_vectors & FF_MM_SSE && HAVE_SSE) {
27         /* SSE for P3/P4/K8 */
28         s->imdct_calc  = ff_imdct_calc_sse;
29         s->imdct_half  = ff_imdct_half_sse;
30         s->fft_permute = ff_fft_permute_sse;
31         s->fft_calc    = ff_fft_calc_sse;
32     } else if (has_vectors & FF_MM_3DNOWEXT && HAVE_AMD3DNOWEXT) {
33         /* 3DNowEx for K7 */
34         s->imdct_calc = ff_imdct_calc_3dn2;
35         s->imdct_half = ff_imdct_half_3dn2;
36         s->fft_calc   = ff_fft_calc_3dn2;
37     } else if (has_vectors & FF_MM_3DNOW && HAVE_AMD3DNOW) {
38         /* 3DNow! for K6-2/3 */
39         s->imdct_calc = ff_imdct_calc_3dn;
40         s->imdct_half = ff_imdct_half_3dn;
41         s->fft_calc   = ff_fft_calc_3dn;
42     }
43 #endif
44 }