git.sesse.net Git - ffmpeg/blob - 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 av_cold void ff_audio_mix_init_x86(AudioMix *am)
  44 {
  45 #if HAVE_YASM
  46     int mm_flags = av_get_cpu_flags();
  47
  48     if (mm_flags & AV_CPU_FLAG_SSE && HAVE_SSE) {
  49         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_FLTP, AV_MIX_COEFF_TYPE_FLT,
  50                               2, 1, 16, 8, "SSE", ff_mix_2_to_1_fltp_flt_sse);
  51         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_FLTP, AV_MIX_COEFF_TYPE_FLT,
  52                               1, 2, 16, 4, "SSE", ff_mix_1_to_2_fltp_flt_sse);
  53     }
  54     if (mm_flags & AV_CPU_FLAG_SSE2 && HAVE_SSE) {
  55         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_S16P, AV_MIX_COEFF_TYPE_FLT,
  56                               2, 1, 16, 8, "SSE2", ff_mix_2_to_1_s16p_flt_sse2);
  57         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_S16P, AV_MIX_COEFF_TYPE_Q8,
  58                               2, 1, 16, 8, "SSE2", ff_mix_2_to_1_s16p_q8_sse2);
  59     }
  60     if (mm_flags & AV_CPU_FLAG_SSE4 && HAVE_SSE) {
  61         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_S16P, AV_MIX_COEFF_TYPE_FLT,
  62                               2, 1, 16, 8, "SSE4", ff_mix_2_to_1_s16p_flt_sse4);
  63     }
  64     if (mm_flags & AV_CPU_FLAG_AVX && HAVE_AVX) {
  65         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_FLTP, AV_MIX_COEFF_TYPE_FLT,
  66                               2, 1, 32, 16, "AVX", ff_mix_2_to_1_fltp_flt_avx);
  67         ff_audio_mix_set_func(am, AV_SAMPLE_FMT_FLTP, AV_MIX_COEFF_TYPE_FLT,
  68                               1, 2, 32, 8, "AVX", ff_mix_1_to_2_fltp_flt_avx);
  69     }
  70 #endif
  71 }