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)
29PROTO4(_)
30PROTO4(_pack_2ch_)
31PROTO4(_pack_6ch_)
32PROTO4(_unpack_2ch_)
33
34av_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
52MULTI_CAPS_FUNC(MMX, mmx)
53MULTI_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