1 /*
2 * Copyright (C) 2012 Michael Niedermayer (michaelni@gmx.at)
3 *
4 * This file is part of libswresample
5 *
6 * libswresample 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 * libswresample 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 libswresample; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #include "libavutil/x86/cpu.h"
22 #include "libswresample/swresample_internal.h"
23 #include "libswresample/audioconvert.h"
24
25 #define PROTO(pre, in, out, cap) void ff ## pre ## in## _to_ ##out## _a_ ##cap(uint8_t **dst, const uint8_t **src, int len);
26 #define PROTO2(pre, out, cap) PROTO(pre, int16, out, cap) PROTO(pre, int32, out, cap) PROTO(pre, float, out, cap)
27 #define PROTO3(pre, cap) PROTO2(pre, int16, cap) PROTO2(pre, int32, cap) PROTO2(pre, float, cap)
28 #define PROTO4(pre) PROTO3(pre, mmx) PROTO3(pre, sse) PROTO3(pre, sse2) PROTO3(pre, ssse3) PROTO3(pre, sse4) PROTO3(pre, avx)
29 PROTO4(_)
PROTO4(_pack_2ch_)30 PROTO4(_pack_2ch_)
31 PROTO4(_pack_6ch_)
32 PROTO4(_unpack_2ch_)
33
34 av_cold void swri_audio_convert_init_x86(struct AudioConvert *ac,
35 enum AVSampleFormat out_fmt,
36 enum AVSampleFormat in_fmt,
37 int channels){
38 int mm_flags = av_get_cpu_flags();
39
40 ac->simd_f= NULL;
41
42 //FIXME add memcpy case
43
44 #define MULTI_CAPS_FUNC(flag, cap) \
45 if (EXTERNAL_##flag(mm_flags)) {\
46 if( out_fmt == AV_SAMPLE_FMT_S32 && in_fmt == AV_SAMPLE_FMT_S16 || out_fmt == AV_SAMPLE_FMT_S32P && in_fmt == AV_SAMPLE_FMT_S16P)\
47 ac->simd_f = ff_int16_to_int32_a_ ## cap;\
48 if( out_fmt == AV_SAMPLE_FMT_S16 && in_fmt == AV_SAMPLE_FMT_S32 || out_fmt == AV_SAMPLE_FMT_S16P && in_fmt == AV_SAMPLE_FMT_S32P)\
49 ac->simd_f = ff_int32_to_int16_a_ ## cap;\
50 }
51
52 MULTI_CAPS_FUNC(MMX, mmx)
53 MULTI_CAPS_FUNC(SSE2, sse2)
54
55 if(EXTERNAL_MMX(mm_flags)) {
56 if(channels == 6) {
57 if( out_fmt == AV_SAMPLE_FMT_FLT && in_fmt == AV_SAMPLE_FMT_FLTP || out_fmt == AV_SAMPLE_FMT_S32 && in_fmt == AV_SAMPLE_FMT_S32P)
58 ac->simd_f = ff_pack_6ch_float_to_float_a_mmx;
59 }
60 }
61
62 if(EXTERNAL_SSE2(mm_flags)) {
63 if( out_fmt == AV_SAMPLE_FMT_FLT && in_fmt == AV_SAMPLE_FMT_S32 || out_fmt == AV_SAMPLE_FMT_FLTP && in_fmt == AV_SAMPLE_FMT_S32P)
64 ac->simd_f = ff_int32_to_float_a_sse2;
65 if( out_fmt == AV_SAMPLE_FMT_FLT && in_fmt == AV_SAMPLE_FMT_S16 || out_fmt == AV_SAMPLE_FMT_FLTP && in_fmt == AV_SAMPLE_FMT_S16P)
66 ac->simd_f = ff_int16_to_float_a_sse2;
67 if( out_fmt == AV_SAMPLE_FMT_S32 && in_fmt == AV_SAMPLE_FMT_FLT || out_fmt == AV_SAMPLE_FMT_S32P && in_fmt == AV_SAMPLE_FMT_FLTP)
68 ac->simd_f = ff_float_to_int32_a_sse2;
69 if( out_fmt == AV_SAMPLE_FMT_S16 && in_fmt == AV_SAMPLE_FMT_FLT || out_fmt == AV_SAMPLE_FMT_S16P && in_fmt == AV_SAMPLE_FMT_FLTP)
70 ac->simd_f = ff_float_to_int16_a_sse2;
71
72 if(channels == 2) {
73 if( out_fmt == AV_SAMPLE_FMT_FLT && in_fmt == AV_SAMPLE_FMT_FLTP || out_fmt == AV_SAMPLE_FMT_S32 && in_fmt == AV_SAMPLE_FMT_S32P)
74 ac->simd_f = ff_pack_2ch_int32_to_int32_a_sse2;
75 if( out_fmt == AV_SAMPLE_FMT_S16 && in_fmt == AV_SAMPLE_FMT_S16P)
76 ac->simd_f = ff_pack_2ch_int16_to_int16_a_sse2;
77 if( out_fmt == AV_SAMPLE_FMT_S32 && in_fmt == AV_SAMPLE_FMT_S16P)
78 ac->simd_f = ff_pack_2ch_int16_to_int32_a_sse2;
79 if( out_fmt == AV_SAMPLE_FMT_S16 && in_fmt == AV_SAMPLE_FMT_S32P)
80 ac->simd_f = ff_pack_2ch_int32_to_int16_a_sse2;
81
82 if( out_fmt == AV_SAMPLE_FMT_FLTP && in_fmt == AV_SAMPLE_FMT_FLT || out_fmt == AV_SAMPLE_FMT_S32P && in_fmt == AV_SAMPLE_FMT_S32)
83 ac->simd_f = ff_unpack_2ch_int32_to_int32_a_sse2;
84 if( out_fmt == AV_SAMPLE_FMT_S16P && in_fmt == AV_SAMPLE_FMT_S16)
85 ac->simd_f = ff_unpack_2ch_int16_to_int16_a_sse2;
86 if( out_fmt == AV_SAMPLE_FMT_S32P && in_fmt == AV_SAMPLE_FMT_S16)
87 ac->simd_f = ff_unpack_2ch_int16_to_int32_a_sse2;
88 if( out_fmt == AV_SAMPLE_FMT_S16P && in_fmt == AV_SAMPLE_FMT_S32)
89 ac->simd_f = ff_unpack_2ch_int32_to_int16_a_sse2;
90
91 if( out_fmt == AV_SAMPLE_FMT_FLT && in_fmt == AV_SAMPLE_FMT_S32P)
92 ac->simd_f = ff_pack_2ch_int32_to_float_a_sse2;
93 if( out_fmt == AV_SAMPLE_FMT_S32 && in_fmt == AV_SAMPLE_FMT_FLTP)
94 ac->simd_f = ff_pack_2ch_float_to_int32_a_sse2;
95 if( out_fmt == AV_SAMPLE_FMT_FLT && in_fmt == AV_SAMPLE_FMT_S16P)
96 ac->simd_f = ff_pack_2ch_int16_to_float_a_sse2;
97 if( out_fmt == AV_SAMPLE_FMT_S16 && in_fmt == AV_SAMPLE_FMT_FLTP)
98 ac->simd_f = ff_pack_2ch_float_to_int16_a_sse2;
99 if( out_fmt == AV_SAMPLE_FMT_FLTP && in_fmt == AV_SAMPLE_FMT_S32)
100 ac->simd_f = ff_unpack_2ch_int32_to_float_a_sse2;
101 if( out_fmt == AV_SAMPLE_FMT_S32P && in_fmt == AV_SAMPLE_FMT_FLT)
102 ac->simd_f = ff_unpack_2ch_float_to_int32_a_sse2;
103 if( out_fmt == AV_SAMPLE_FMT_FLTP && in_fmt == AV_SAMPLE_FMT_S16)
104 ac->simd_f = ff_unpack_2ch_int16_to_float_a_sse2;
105 if( out_fmt == AV_SAMPLE_FMT_S16P && in_fmt == AV_SAMPLE_FMT_FLT)
106 ac->simd_f = ff_unpack_2ch_float_to_int16_a_sse2;
107 }
108 }
109 if(EXTERNAL_SSSE3(mm_flags)) {
110 if(channels == 2) {
111 if( out_fmt == AV_SAMPLE_FMT_S16P && in_fmt == AV_SAMPLE_FMT_S16)
112 ac->simd_f = ff_unpack_2ch_int16_to_int16_a_ssse3;
113 if( out_fmt == AV_SAMPLE_FMT_S32P && in_fmt == AV_SAMPLE_FMT_S16)
114 ac->simd_f = ff_unpack_2ch_int16_to_int32_a_ssse3;
115 if( out_fmt == AV_SAMPLE_FMT_FLTP && in_fmt == AV_SAMPLE_FMT_S16)
116 ac->simd_f = ff_unpack_2ch_int16_to_float_a_ssse3;
117 }
118 }
119 if(EXTERNAL_SSE4(mm_flags)) {
120 if(channels == 6) {
121 if( out_fmt == AV_SAMPLE_FMT_FLT && in_fmt == AV_SAMPLE_FMT_FLTP || out_fmt == AV_SAMPLE_FMT_S32 && in_fmt == AV_SAMPLE_FMT_S32P)
122 ac->simd_f = ff_pack_6ch_float_to_float_a_sse4;
123 if( out_fmt == AV_SAMPLE_FMT_FLT && in_fmt == AV_SAMPLE_FMT_S32P)
124 ac->simd_f = ff_pack_6ch_int32_to_float_a_sse4;
125 if( out_fmt == AV_SAMPLE_FMT_S32 && in_fmt == AV_SAMPLE_FMT_FLTP)
126 ac->simd_f = ff_pack_6ch_float_to_int32_a_sse4;
127 }
128 }
129 if(EXTERNAL_AVX(mm_flags)) {
130 if( out_fmt == AV_SAMPLE_FMT_FLT && in_fmt == AV_SAMPLE_FMT_S32 || out_fmt == AV_SAMPLE_FMT_FLTP && in_fmt == AV_SAMPLE_FMT_S32P)
131 ac->simd_f = ff_int32_to_float_a_avx;
132 if(channels == 6) {
133 if( out_fmt == AV_SAMPLE_FMT_FLT && in_fmt == AV_SAMPLE_FMT_FLTP || out_fmt == AV_SAMPLE_FMT_S32 && in_fmt == AV_SAMPLE_FMT_S32P)
134 ac->simd_f = ff_pack_6ch_float_to_float_a_avx;
135 if( out_fmt == AV_SAMPLE_FMT_FLT && in_fmt == AV_SAMPLE_FMT_S32P)
136 ac->simd_f = ff_pack_6ch_int32_to_float_a_avx;
137 if( out_fmt == AV_SAMPLE_FMT_S32 && in_fmt == AV_SAMPLE_FMT_FLTP)
138 ac->simd_f = ff_pack_6ch_float_to_int32_a_avx;
139 }
140 }
141 }
142