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