Main Page
Related Pages
Modules
Data Structures
Files
Examples
File List
Globals
libavcodec
x86
fft.c
Go to the documentation of this file.
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 "
libavutil/cpu.h
"
20
#include "
libavcodec/dsputil.h
"
21
#include "
libavcodec/dct.h
"
22
#include "
fft.h
"
23
24
av_cold
void
ff_fft_init_mmx
(
FFTContext
*s)
25
{
26
#if HAVE_YASM
27
int
has_vectors =
av_get_cpu_flags
();
28
if
(has_vectors &
AV_CPU_FLAG_AVX
&&
HAVE_AVX
&& s->
nbits
>= 5) {
29
/* AVX for SB */
30
s->
imdct_calc
=
ff_imdct_calc_sse
;
31
s->
imdct_half
=
ff_imdct_half_avx
;
32
s->
fft_permute
=
ff_fft_permute_sse
;
33
s->
fft_calc
=
ff_fft_calc_avx
;
34
s->
fft_permutation
=
FF_FFT_PERM_AVX
;
35
}
else
if
(has_vectors &
AV_CPU_FLAG_SSE
&&
HAVE_SSE
) {
36
/* SSE for P3/P4/K8 */
37
s->
imdct_calc
=
ff_imdct_calc_sse
;
38
s->
imdct_half
=
ff_imdct_half_sse
;
39
s->
fft_permute
=
ff_fft_permute_sse
;
40
s->
fft_calc
=
ff_fft_calc_sse
;
41
s->
fft_permutation
=
FF_FFT_PERM_SWAP_LSBS
;
42
}
43
#endif
44
}
45
46
#if CONFIG_DCT
47
av_cold
void
ff_dct_init_mmx
(
DCTContext
*s)
48
{
49
#if HAVE_YASM
50
int
has_vectors =
av_get_cpu_flags
();
51
if
(has_vectors &
AV_CPU_FLAG_AVX
&&
HAVE_AVX
)
52
s->
dct32
=
ff_dct32_float_avx
;
53
else
if
(has_vectors &
AV_CPU_FLAG_SSE2
&&
HAVE_SSE
)
54
s->
dct32
=
ff_dct32_float_sse2
;
55
else
if
(has_vectors &
AV_CPU_FLAG_SSE
&&
HAVE_SSE
)
56
s->
dct32
=
ff_dct32_float_sse
;
57
#endif
58
}
59
#endif
60