6f00af49f98a95bf8d91acb7c3b4f3afc8408ceb
[ffmpeg.git] / libavcodec / audioconvert.c
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 /**
23  * @file audioconvert.c
24  * audio conversion
25  * @author Michael Niedermayer <michaelni@gmx.at>
26  */
27
28 #include "audioconvert.h"
29
30 typedef struct SampleFmtInfo {
31     const char *name;
32     int bits;
33 } SampleFmtInfo;
34
35 /** this table gives more information about formats */
36 static const SampleFmtInfo sample_fmt_info[SAMPLE_FMT_NB] = {
37     [SAMPLE_FMT_U8]  = { .name = "u8",  .bits = 8 },
38     [SAMPLE_FMT_S16] = { .name = "s16", .bits = 16 },
39     [SAMPLE_FMT_S24] = { .name = "s24", .bits = 24 },
40     [SAMPLE_FMT_S32] = { .name = "s32", .bits = 32 },
41     [SAMPLE_FMT_FLT] = { .name = "flt", .bits = 32 }
42 };
43
44 const char *avcodec_get_sample_fmt_name(int sample_fmt)
45 {
46     if (sample_fmt < 0 || sample_fmt >= SAMPLE_FMT_NB)
47         return NULL;
48     return sample_fmt_info[sample_fmt].name;
49 }
50
51 enum SampleFormat avcodec_get_sample_fmt(const char* name)
52 {
53     int i;
54
55     for (i=0; i < SAMPLE_FMT_NB; i++)
56         if (!strcmp(sample_fmt_info[i].name, name))
57             return i;
58     return SAMPLE_FMT_NONE;
59 }
60
61 void avcodec_sample_fmt_string (char *buf, int buf_size, int sample_fmt)
62 {
63     /* print header */
64     if (sample_fmt < 0)
65         snprintf (buf, buf_size, "name  " " depth");
66     else if (sample_fmt < SAMPLE_FMT_NB) {
67         SampleFmtInfo info= sample_fmt_info[sample_fmt];
68         snprintf (buf, buf_size, "%-6s" "   %2d ", info.name, info.bits);
69     }
70 }
71
72 int av_audio_convert(void *maybe_dspcontext_or_something_av_convert_specific,
73                      void *out[6], int out_stride[6], enum SampleFormat out_fmt,
74                      void * in[6], int  in_stride[6], enum SampleFormat  in_fmt, int len){
75     int ch;
76     const int isize= FFMIN( in_fmt+1, 4);
77     const int osize= FFMIN(out_fmt+1, 4);
78     const int fmt_pair= out_fmt + 5*in_fmt;
79
80     //FIXME optimize common cases
81
82     for(ch=0; ch<6; ch++){
83         const int is=  in_stride[ch] * isize;
84         const int os= out_stride[ch] * osize;
85         uint8_t *pi=  in[ch];
86         uint8_t *po= out[ch];
87         uint8_t *end= po + os;
88         if(!out[ch])
89             continue;
90
91 #define CONV(ofmt, otype, ifmt, expr)\
92 if(fmt_pair == ofmt + 5*ifmt){\
93     do{\
94         *(otype*)po = expr; pi += is; po += os;\
95     }while(po < end);\
96 }
97
98 //FIXME put things below under ifdefs so we do not waste space for cases no codec will need
99 //FIXME rounding and clipping ?
100
101              CONV(SAMPLE_FMT_U8 , uint8_t, SAMPLE_FMT_U8 ,  *(uint8_t*)pi)
102         else CONV(SAMPLE_FMT_S16, int16_t, SAMPLE_FMT_U8 , (*(uint8_t*)pi - 0x80)<<8)
103         else CONV(SAMPLE_FMT_S32, int32_t, SAMPLE_FMT_U8 , (*(uint8_t*)pi - 0x80)<<24)
104         else CONV(SAMPLE_FMT_FLT, float  , SAMPLE_FMT_U8 , (*(uint8_t*)pi - 0x80)*(1.0 / (1<<7)))
105         else CONV(SAMPLE_FMT_U8 , uint8_t, SAMPLE_FMT_S16, (*(int16_t*)pi>>8) + 0x80)
106         else CONV(SAMPLE_FMT_S16, int16_t, SAMPLE_FMT_S16,  *(int16_t*)pi)
107         else CONV(SAMPLE_FMT_S32, int32_t, SAMPLE_FMT_S16,  *(int16_t*)pi<<16)
108         else CONV(SAMPLE_FMT_FLT, float  , SAMPLE_FMT_S16,  *(int16_t*)pi*(1.0 / (1<<15)))
109         else CONV(SAMPLE_FMT_U8 , uint8_t, SAMPLE_FMT_S32, (*(int32_t*)pi>>24) + 0x80)
110         else CONV(SAMPLE_FMT_S16, int16_t, SAMPLE_FMT_S32,  *(int32_t*)pi>>16)
111         else CONV(SAMPLE_FMT_S32, int32_t, SAMPLE_FMT_S32,  *(int32_t*)pi)
112         else CONV(SAMPLE_FMT_FLT, float  , SAMPLE_FMT_S32,  *(int32_t*)pi*(1.0 / (1<<31)))
113         else CONV(SAMPLE_FMT_U8 , uint8_t, SAMPLE_FMT_FLT, lrintf(*(float*)pi * (1<<7)) + 0x80)
114         else CONV(SAMPLE_FMT_S16, int16_t, SAMPLE_FMT_FLT, lrintf(*(float*)pi * (1<<15)))
115         else CONV(SAMPLE_FMT_S32, int32_t, SAMPLE_FMT_FLT, lrintf(*(float*)pi * (1<<31)))
116         else CONV(SAMPLE_FMT_FLT, float  , SAMPLE_FMT_FLT, *(float*)pi)
117         else return -1;
118     }
119     return 0;
120 }