FFmpeg
fft_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 "libavutil/attributes.h"
22 #include "libavutil/cpu.h"
23 #include "libavutil/x86/cpu.h"
24 
25 #include "fft.h"
26 
28 {
30 
31  if (s->nbits > 16)
32  return;
33 
34 #if ARCH_X86_32
36  s->imdct_calc = ff_imdct_calc_3dnow;
37  s->imdct_half = ff_imdct_half_3dnow;
38  s->fft_calc = ff_fft_calc_3dnow;
39  }
40 
42  s->imdct_calc = ff_imdct_calc_3dnowext;
43  s->imdct_half = ff_imdct_half_3dnowext;
44  s->fft_calc = ff_fft_calc_3dnowext;
45  }
46 #endif /* ARCH_X86_32 */
47 
48  if (EXTERNAL_SSE(cpu_flags)) {
49  s->imdct_calc = ff_imdct_calc_sse;
50  s->imdct_half = ff_imdct_half_sse;
51  s->fft_permute = ff_fft_permute_sse;
52  s->fft_calc = ff_fft_calc_sse;
53  s->fft_permutation = FF_FFT_PERM_SWAP_LSBS;
54  }
55 
56  if (EXTERNAL_AVX_FAST(cpu_flags) && s->nbits >= 5) {
57  s->imdct_half = ff_imdct_half_avx;
58  s->fft_calc = ff_fft_calc_avx;
59  s->fft_permutation = FF_FFT_PERM_AVX;
60  }
61 }
EXTERNAL_AMD3DNOWEXT
#define EXTERNAL_AMD3DNOWEXT(flags)
Definition: cpu.h:55
cpu.h
EXTERNAL_AMD3DNOW
#define EXTERNAL_AMD3DNOW(flags)
Definition: cpu.h:54
ff_imdct_calc_3dnowext
void ff_imdct_calc_3dnowext(FFTContext *s, FFTSample *output, const FFTSample *input)
av_get_cpu_flags
int av_get_cpu_flags(void)
Return the flags which specify extensions supported by the CPU.
Definition: cpu.c:96
ff_imdct_calc_sse
void ff_imdct_calc_sse(FFTContext *s, FFTSample *output, const FFTSample *input)
FF_FFT_PERM_SWAP_LSBS
@ FF_FFT_PERM_SWAP_LSBS
Definition: fft.h:66
cpu_flags
static atomic_int cpu_flags
Definition: cpu.c:50
ff_imdct_calc_3dnow
void ff_imdct_calc_3dnow(FFTContext *s, FFTSample *output, const FFTSample *input)
fft.h
ff_imdct_half_avx
void ff_imdct_half_avx(FFTContext *s, FFTSample *output, const FFTSample *input)
ff_fft_init_x86
av_cold void ff_fft_init_x86(FFTContext *s)
Definition: fft_init.c:27
av_cold
#define av_cold
Definition: attributes.h:90
s
#define s(width, name)
Definition: cbs_vp9.c:257
EXTERNAL_SSE
#define EXTERNAL_SSE(flags)
Definition: cpu.h:58
EXTERNAL_AVX_FAST
#define EXTERNAL_AVX_FAST(flags)
Definition: cpu.h:71
ff_fft_calc_sse
void ff_fft_calc_sse(FFTContext *s, FFTComplex *z)
ff_fft_permute_sse
void ff_fft_permute_sse(FFTContext *s, FFTComplex *z)
cpu.h
ff_fft_calc_3dnow
void ff_fft_calc_3dnow(FFTContext *s, FFTComplex *z)
ff_imdct_half_3dnowext
void ff_imdct_half_3dnowext(FFTContext *s, FFTSample *output, const FFTSample *input)
attributes.h
FFTContext
Definition: fft.h:75
ff_imdct_half_sse
void ff_imdct_half_sse(FFTContext *s, FFTSample *output, const FFTSample *input)
ff_fft_calc_avx
void ff_fft_calc_avx(FFTContext *s, FFTComplex *z)
FF_FFT_PERM_AVX
@ FF_FFT_PERM_AVX
Definition: fft.h:67
ff_fft_calc_3dnowext
void ff_fft_calc_3dnowext(FFTContext *s, FFTComplex *z)
ff_imdct_half_3dnow
void ff_imdct_half_3dnow(FFTContext *s, FFTSample *output, const FFTSample *input)