x86: Replace checks for CPU extensions and flags by convenience macros
[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 #include "cpu.h"
24
25 extern void ff_vector_fmul_sse(float *dst, const float *src0, const float *src1,
26                                int len);
27 extern void ff_vector_fmul_avx(float *dst, const float *src0, const float *src1,
28                                int len);
29
30 extern void ff_vector_fmac_scalar_sse(float *dst, const float *src, float mul,
31                                       int len);
32 extern void ff_vector_fmac_scalar_avx(float *dst, const float *src, float mul,
33                                       int len);
34
35 void ff_float_dsp_init_x86(AVFloatDSPContext *fdsp)
36 {
37     int mm_flags = av_get_cpu_flags();
38
39     if (EXTERNAL_SSE(mm_flags)) {
40         fdsp->vector_fmul = ff_vector_fmul_sse;
41         fdsp->vector_fmac_scalar = ff_vector_fmac_scalar_sse;
42     }
43     if (EXTERNAL_AVX(mm_flags)) {
44         fdsp->vector_fmul = ff_vector_fmul_avx;
45         fdsp->vector_fmac_scalar = ff_vector_fmac_scalar_avx;
46     }
47 }