yading@10: /* yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: #include "libavutil/cpu.h" yading@10: #include "libavutil/x86/cpu.h" yading@10: #include "libavcodec/dct.h" yading@10: #include "fft.h" yading@10: yading@10: av_cold void ff_fft_init_x86(FFTContext *s) yading@10: { yading@10: int has_vectors = av_get_cpu_flags(); yading@10: #if ARCH_X86_32 yading@10: if (EXTERNAL_AMD3DNOW(has_vectors)) { yading@10: /* 3DNow! for K6-2/3 */ yading@10: s->imdct_calc = ff_imdct_calc_3dnow; yading@10: s->imdct_half = ff_imdct_half_3dnow; yading@10: s->fft_calc = ff_fft_calc_3dnow; yading@10: } yading@10: if (EXTERNAL_AMD3DNOWEXT(has_vectors)) { yading@10: /* 3DNowEx for K7 */ yading@10: s->imdct_calc = ff_imdct_calc_3dnowext; yading@10: s->imdct_half = ff_imdct_half_3dnowext; yading@10: s->fft_calc = ff_fft_calc_3dnowext; yading@10: } yading@10: #endif yading@10: if (EXTERNAL_SSE(has_vectors)) { yading@10: /* SSE for P3/P4/K8 */ yading@10: s->imdct_calc = ff_imdct_calc_sse; yading@10: s->imdct_half = ff_imdct_half_sse; yading@10: s->fft_permute = ff_fft_permute_sse; yading@10: s->fft_calc = ff_fft_calc_sse; yading@10: s->fft_permutation = FF_FFT_PERM_SWAP_LSBS; yading@10: } yading@10: if (EXTERNAL_AVX(has_vectors) && s->nbits >= 5) { yading@10: /* AVX for SB */ yading@10: s->imdct_half = ff_imdct_half_avx; yading@10: s->fft_calc = ff_fft_calc_avx; yading@10: s->fft_permutation = FF_FFT_PERM_AVX; yading@10: } yading@10: } yading@10: yading@10: #if CONFIG_DCT yading@10: av_cold void ff_dct_init_x86(DCTContext *s) yading@10: { yading@10: int has_vectors = av_get_cpu_flags(); yading@10: if (EXTERNAL_SSE(has_vectors)) yading@10: s->dct32 = ff_dct32_float_sse; yading@10: if (EXTERNAL_SSE2(has_vectors)) yading@10: s->dct32 = ff_dct32_float_sse2; yading@10: if (EXTERNAL_AVX(has_vectors)) yading@10: s->dct32 = ff_dct32_float_avx; yading@10: } yading@10: #endif