audioconvert.c 9.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224
  1. /*
  2. * audio conversion
  3. * Copyright (c) 2006 Michael Niedermayer <michaelni@gmx.at>
  4. *
  5. * This file is part of FFmpeg.
  6. *
  7. * FFmpeg is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU Lesser General Public
  9. * License as published by the Free Software Foundation; either
  10. * version 2.1 of the License, or (at your option) any later version.
  11. *
  12. * FFmpeg is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with FFmpeg; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. /**
  22. * @file
  23. * audio conversion
  24. * @author Michael Niedermayer <michaelni@gmx.at>
  25. */
  26. #include "libavutil/avstring.h"
  27. #include "libavutil/avassert.h"
  28. #include "libavutil/libm.h"
  29. #include "libavutil/samplefmt.h"
  30. #include "audioconvert.h"
  31. #define CONV_FUNC_NAME(dst_fmt, src_fmt) conv_ ## src_fmt ## _to_ ## dst_fmt
  32. //FIXME rounding ?
  33. #define CONV_FUNC(ofmt, otype, ifmt, expr)\
  34. static void CONV_FUNC_NAME(ofmt, ifmt)(uint8_t *po, const uint8_t *pi, int is, int os, uint8_t *end)\
  35. {\
  36. uint8_t *end2 = end - 3*os;\
  37. while(po < end2){\
  38. *(otype*)po = expr; pi += is; po += os;\
  39. *(otype*)po = expr; pi += is; po += os;\
  40. *(otype*)po = expr; pi += is; po += os;\
  41. *(otype*)po = expr; pi += is; po += os;\
  42. }\
  43. while(po < end){\
  44. *(otype*)po = expr; pi += is; po += os;\
  45. }\
  46. }
  47. //FIXME put things below under ifdefs so we do not waste space for cases no codec will need
  48. CONV_FUNC(AV_SAMPLE_FMT_U8 , uint8_t, AV_SAMPLE_FMT_U8 , *(const uint8_t*)pi)
  49. CONV_FUNC(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_U8 , (*(const uint8_t*)pi - 0x80)<<8)
  50. CONV_FUNC(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_U8 , (*(const uint8_t*)pi - 0x80)<<24)
  51. CONV_FUNC(AV_SAMPLE_FMT_FLT, float , AV_SAMPLE_FMT_U8 , (*(const uint8_t*)pi - 0x80)*(1.0f/ (1<<7)))
  52. CONV_FUNC(AV_SAMPLE_FMT_DBL, double , AV_SAMPLE_FMT_U8 , (*(const uint8_t*)pi - 0x80)*(1.0 / (1<<7)))
  53. CONV_FUNC(AV_SAMPLE_FMT_U8 , uint8_t, AV_SAMPLE_FMT_S16, (*(const int16_t*)pi>>8) + 0x80)
  54. CONV_FUNC(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_S16, *(const int16_t*)pi)
  55. CONV_FUNC(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_S16, *(const int16_t*)pi<<16)
  56. CONV_FUNC(AV_SAMPLE_FMT_FLT, float , AV_SAMPLE_FMT_S16, *(const int16_t*)pi*(1.0f/ (1<<15)))
  57. CONV_FUNC(AV_SAMPLE_FMT_DBL, double , AV_SAMPLE_FMT_S16, *(const int16_t*)pi*(1.0 / (1<<15)))
  58. CONV_FUNC(AV_SAMPLE_FMT_U8 , uint8_t, AV_SAMPLE_FMT_S32, (*(const int32_t*)pi>>24) + 0x80)
  59. CONV_FUNC(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_S32, *(const int32_t*)pi>>16)
  60. CONV_FUNC(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_S32, *(const int32_t*)pi)
  61. CONV_FUNC(AV_SAMPLE_FMT_FLT, float , AV_SAMPLE_FMT_S32, *(const int32_t*)pi*(1.0f/ (1U<<31)))
  62. CONV_FUNC(AV_SAMPLE_FMT_DBL, double , AV_SAMPLE_FMT_S32, *(const int32_t*)pi*(1.0 / (1U<<31)))
  63. CONV_FUNC(AV_SAMPLE_FMT_U8 , uint8_t, AV_SAMPLE_FMT_FLT, av_clip_uint8( lrintf(*(const float*)pi * (1<<7)) + 0x80))
  64. CONV_FUNC(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, av_clip_int16( lrintf(*(const float*)pi * (1<<15))))
  65. CONV_FUNC(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, av_clipl_int32(llrintf(*(const float*)pi * (1U<<31))))
  66. CONV_FUNC(AV_SAMPLE_FMT_FLT, float , AV_SAMPLE_FMT_FLT, *(const float*)pi)
  67. CONV_FUNC(AV_SAMPLE_FMT_DBL, double , AV_SAMPLE_FMT_FLT, *(const float*)pi)
  68. CONV_FUNC(AV_SAMPLE_FMT_U8 , uint8_t, AV_SAMPLE_FMT_DBL, av_clip_uint8( lrint(*(const double*)pi * (1<<7)) + 0x80))
  69. CONV_FUNC(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, av_clip_int16( lrint(*(const double*)pi * (1<<15))))
  70. CONV_FUNC(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, av_clipl_int32(llrint(*(const double*)pi * (1U<<31))))
  71. CONV_FUNC(AV_SAMPLE_FMT_FLT, float , AV_SAMPLE_FMT_DBL, *(const double*)pi)
  72. CONV_FUNC(AV_SAMPLE_FMT_DBL, double , AV_SAMPLE_FMT_DBL, *(const double*)pi)
  73. #define FMT_PAIR_FUNC(out, in) [out + AV_SAMPLE_FMT_NB*in] = CONV_FUNC_NAME(out, in)
  74. static conv_func_type * const fmt_pair_to_conv_functions[AV_SAMPLE_FMT_NB*AV_SAMPLE_FMT_NB] = {
  75. FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8 , AV_SAMPLE_FMT_U8 ),
  76. FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_U8 ),
  77. FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_U8 ),
  78. FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_U8 ),
  79. FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_U8 ),
  80. FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8 , AV_SAMPLE_FMT_S16),
  81. FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_S16),
  82. FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_S16),
  83. FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_S16),
  84. FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_S16),
  85. FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8 , AV_SAMPLE_FMT_S32),
  86. FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_S32),
  87. FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_S32),
  88. FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_S32),
  89. FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_S32),
  90. FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8 , AV_SAMPLE_FMT_FLT),
  91. FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_FLT),
  92. FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_FLT),
  93. FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_FLT),
  94. FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_FLT),
  95. FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8 , AV_SAMPLE_FMT_DBL),
  96. FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_DBL),
  97. FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_DBL),
  98. FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_DBL),
  99. FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_DBL),
  100. };
  101. static void cpy1(uint8_t **dst, const uint8_t **src, int len){
  102. memcpy(*dst, *src, len);
  103. }
  104. static void cpy2(uint8_t **dst, const uint8_t **src, int len){
  105. memcpy(*dst, *src, 2*len);
  106. }
  107. static void cpy4(uint8_t **dst, const uint8_t **src, int len){
  108. memcpy(*dst, *src, 4*len);
  109. }
  110. static void cpy8(uint8_t **dst, const uint8_t **src, int len){
  111. memcpy(*dst, *src, 8*len);
  112. }
  113. AudioConvert *swri_audio_convert_alloc(enum AVSampleFormat out_fmt,
  114. enum AVSampleFormat in_fmt,
  115. int channels, const int *ch_map,
  116. int flags)
  117. {
  118. AudioConvert *ctx;
  119. conv_func_type *f = fmt_pair_to_conv_functions[av_get_packed_sample_fmt(out_fmt) + AV_SAMPLE_FMT_NB*av_get_packed_sample_fmt(in_fmt)];
  120. if (!f)
  121. return NULL;
  122. ctx = av_mallocz(sizeof(*ctx));
  123. if (!ctx)
  124. return NULL;
  125. if(channels == 1){
  126. in_fmt = av_get_planar_sample_fmt( in_fmt);
  127. out_fmt = av_get_planar_sample_fmt(out_fmt);
  128. }
  129. ctx->channels = channels;
  130. ctx->conv_f = f;
  131. ctx->ch_map = ch_map;
  132. if (in_fmt == AV_SAMPLE_FMT_U8 || in_fmt == AV_SAMPLE_FMT_U8P)
  133. memset(ctx->silence, 0x80, sizeof(ctx->silence));
  134. if(out_fmt == in_fmt && !ch_map) {
  135. switch(av_get_bytes_per_sample(in_fmt)){
  136. case 1:ctx->simd_f = cpy1; break;
  137. case 2:ctx->simd_f = cpy2; break;
  138. case 4:ctx->simd_f = cpy4; break;
  139. case 8:ctx->simd_f = cpy8; break;
  140. }
  141. }
  142. if(HAVE_YASM && HAVE_MMX) swri_audio_convert_init_x86(ctx, out_fmt, in_fmt, channels);
  143. if(ARCH_ARM) swri_audio_convert_init_arm(ctx, out_fmt, in_fmt, channels);
  144. return ctx;
  145. }
  146. void swri_audio_convert_free(AudioConvert **ctx)
  147. {
  148. av_freep(ctx);
  149. }
  150. int swri_audio_convert(AudioConvert *ctx, AudioData *out, AudioData *in, int len)
  151. {
  152. int ch;
  153. int off=0;
  154. const int os= (out->planar ? 1 :out->ch_count) *out->bps;
  155. unsigned misaligned = 0;
  156. av_assert0(ctx->channels == out->ch_count);
  157. if (ctx->in_simd_align_mask) {
  158. int planes = in->planar ? in->ch_count : 1;
  159. unsigned m = 0;
  160. for (ch = 0; ch < planes; ch++)
  161. m |= (intptr_t)in->ch[ch];
  162. misaligned |= m & ctx->in_simd_align_mask;
  163. }
  164. if (ctx->out_simd_align_mask) {
  165. int planes = out->planar ? out->ch_count : 1;
  166. unsigned m = 0;
  167. for (ch = 0; ch < planes; ch++)
  168. m |= (intptr_t)out->ch[ch];
  169. misaligned |= m & ctx->out_simd_align_mask;
  170. }
  171. //FIXME optimize common cases
  172. if(ctx->simd_f && !ctx->ch_map && !misaligned){
  173. off = len&~15;
  174. av_assert1(off>=0);
  175. av_assert1(off<=len);
  176. av_assert2(ctx->channels == SWR_CH_MAX || !in->ch[ctx->channels]);
  177. if(off>0){
  178. if(out->planar == in->planar){
  179. int planes = out->planar ? out->ch_count : 1;
  180. for(ch=0; ch<planes; ch++){
  181. ctx->simd_f(out->ch+ch, (const uint8_t **)in->ch+ch, off * (out->planar ? 1 :out->ch_count));
  182. }
  183. }else{
  184. ctx->simd_f(out->ch, (const uint8_t **)in->ch, off);
  185. }
  186. }
  187. if(off == len)
  188. return 0;
  189. }
  190. for(ch=0; ch<ctx->channels; ch++){
  191. const int ich= ctx->ch_map ? ctx->ch_map[ch] : ch;
  192. const int is= ich < 0 ? 0 : (in->planar ? 1 : in->ch_count) * in->bps;
  193. const uint8_t *pi= ich < 0 ? ctx->silence : in->ch[ich];
  194. uint8_t *po= out->ch[ch];
  195. uint8_t *end= po + os*len;
  196. if(!po)
  197. continue;
  198. ctx->conv_f(po+off*os, pi+off*is, is, os, end);
  199. }
  200. return 0;
  201. }