b8f3a90eefe28c13770c90a880127a7eab088d88
[ffmpeg.git] / libavresample / x86 / audio_mix_init.c
1 /*
2  * Copyright (c) 2012 Justin Ruggles <justin.ruggles@gmail.com>
3  *
4  * This file is part of Libav.
5  *
6  * Libav is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * Libav is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with Libav; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 #include "config.h"
22 #include "libavutil/cpu.h"
23 #include "libavresample/audio_mix.h"
24
25 extern void ff_mix_2_to_1_fltp_flt_sse(float **src, float **matrix, int len,
26                                        int out_ch, int in_ch);
27 extern void ff_mix_2_to_1_fltp_flt_avx(float **src, float **matrix, int len,
28                                        int out_ch, int in_ch);
29
30 extern void ff_mix_2_to_1_s16p_flt_sse2(int16_t **src, float **matrix, int len,
31                                         int out_ch, int in_ch);
32 extern void ff_mix_2_to_1_s16p_flt_sse4(int16_t **src, float **matrix, int len,
33                                         int out_ch, int in_ch);
34
35 extern void ff_mix_2_to_1_s16p_q8_sse2(int16_t **src, int16_t **matrix,
36                                        int len, int out_ch, int in_ch);
37
38 extern void ff_mix_1_to_2_fltp_flt_sse(float **src, float **matrix, int len,
39                                        int out_ch, int in_ch);
40 extern void ff_mix_1_to_2_fltp_flt_avx(float **src, float **matrix, int len,
41                                        int out_ch, int in_ch);
42
43 extern void ff_mix_1_to_2_s16p_flt_sse2(int16_t **src, float **matrix, int len,
44                                         int out_ch, int in_ch);
45 extern void ff_mix_1_to_2_s16p_flt_sse4(int16_t **src, float **matrix, int len,
46                                         int out_ch, int in_ch);
47 extern void ff_mix_1_to_2_s16p_flt_avx (int16_t **src, float **matrix, int len,
48                                         int out_ch, int in_ch);
49
50 av_cold void ff_audio_mix_init_x86(AudioMix *am)
51 {
52 #if HAVE_YASM
53     int mm_flags = av_get_cpu_flags();
54
55     if (mm_flags & AV_CPU_FLAG_SSE && HAVE_SSE) {
56         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_FLTP, AV_MIX_COEFF_TYPE_FLT,
57                               2, 1, 16, 8, "SSE", ff_mix_2_to_1_fltp_flt_sse);
58         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_FLTP, AV_MIX_COEFF_TYPE_FLT,
59                               1, 2, 16, 4, "SSE", ff_mix_1_to_2_fltp_flt_sse);
60     }
61     if (mm_flags & AV_CPU_FLAG_SSE2 && HAVE_SSE) {
62         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_S16P, AV_MIX_COEFF_TYPE_FLT,
63                               2, 1, 16, 8, "SSE2", ff_mix_2_to_1_s16p_flt_sse2);
64         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_S16P, AV_MIX_COEFF_TYPE_Q8,
65                               2, 1, 16, 8, "SSE2", ff_mix_2_to_1_s16p_q8_sse2);
66         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_S16P, AV_MIX_COEFF_TYPE_FLT,
67                               1, 2, 16, 8, "SSE2", ff_mix_1_to_2_s16p_flt_sse2);
68     }
69     if (mm_flags & AV_CPU_FLAG_SSE4 && HAVE_SSE) {
70         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_S16P, AV_MIX_COEFF_TYPE_FLT,
71                               2, 1, 16, 8, "SSE4", ff_mix_2_to_1_s16p_flt_sse4);
72         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_S16P, AV_MIX_COEFF_TYPE_FLT,
73                               1, 2, 16, 8, "SSE4", ff_mix_1_to_2_s16p_flt_sse4);
74     }
75     if (mm_flags & AV_CPU_FLAG_AVX && HAVE_AVX) {
76         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_FLTP, AV_MIX_COEFF_TYPE_FLT,
77                               2, 1, 32, 16, "AVX", ff_mix_2_to_1_fltp_flt_avx);
78         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_FLTP, AV_MIX_COEFF_TYPE_FLT,
79                               1, 2, 32, 8, "AVX", ff_mix_1_to_2_fltp_flt_avx);
80         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_S16P, AV_MIX_COEFF_TYPE_FLT,
81                               1, 2, 16, 8, "AVX", ff_mix_1_to_2_s16p_flt_avx);
82     }
83 #endif
84 }