float_dsp: add x86-optimized functions for vector_fmac_scalar()
[ffmpeg.git] / libavutil / x86 / float_dsp_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
21 #include "libavutil/cpu.h"
22 #include "libavutil/float_dsp.h"
23
24 extern void ff_vector_fmul_sse(float *dst, const float *src0, const float *src1,
25                                int len);
26 extern void ff_vector_fmul_avx(float *dst, const float *src0, const float *src1,
27                                int len);
28
29 extern void ff_vector_fmac_scalar_sse(float *dst, const float *src, float mul,
30                                       int len);
31 extern void ff_vector_fmac_scalar_avx(float *dst, const float *src, float mul,
32                                       int len);
33
34 void ff_float_dsp_init_x86(AVFloatDSPContext *fdsp)
35 {
36 #if HAVE_YASM
37     int mm_flags = av_get_cpu_flags();
38
39     if (mm_flags & AV_CPU_FLAG_SSE && HAVE_SSE) {
40         fdsp->vector_fmul = ff_vector_fmul_sse;
41         fdsp->vector_fmac_scalar = ff_vector_fmac_scalar_sse;
42     }
43     if (mm_flags & AV_CPU_FLAG_AVX && HAVE_AVX) {
44         fdsp->vector_fmul = ff_vector_fmul_avx;
45         fdsp->vector_fmac_scalar = ff_vector_fmac_scalar_avx;
46     }
47 #endif
48 }