git.sesse.net Git - ffmpeg/blob - libavcodec/faad.c

   1 /*
   2  * Faad decoder
   3  * Copyright (c) 2003 Zdenek Kabelac.
   4  * Copyright (c) 2004 Thomas Raivio.
   5  *
   6  * This library 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 of the License, or (at your option) any later version.
  10  *
  11  * This library 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 this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19  */
  20
  21 /**
  22  * @file faad.c
  23  * AAC decoder.
  24  *
  25  * still a bit unfinished - but it plays something
  26  */
  27
  28 #include "avcodec.h"
  29 #include "faad.h"
  30
  31 #ifndef FAADAPI
  32 #define FAADAPI
  33 #endif
  34
  35 /*
  36  * when CONFIG_FAADBIN is defined the libfaad will be opened at runtime
  37  */
  38 //#undef CONFIG_FAADBIN
  39 //#define CONFIG_FAADBIN
  40
  41 #ifdef CONFIG_FAADBIN
  42 #include <dlfcn.h>
  43 static const char* libfaadname = "libfaad.so.0";
  44 #else
  45 #define dlopen(a)
  46 #define dlclose(a)
  47 #endif
  48
  49 typedef struct {
  50     void* handle;               /* dlopen handle */
  51     void* faac_handle;          /* FAAD library handle */
  52     int sample_size;
  53     int init;
  54
  55     /* faad calls */
  56     faacDecHandle FAADAPI (*faacDecOpen)(void);
  57     faacDecConfigurationPtr FAADAPI (*faacDecGetCurrentConfiguration)(faacDecHandle hDecoder);
  58 #ifndef FAAD2_VERSION
  59         int FAADAPI (*faacDecSetConfiguration)(faacDecHandle hDecoder,
  60                                            faacDecConfigurationPtr config);
  61         int FAADAPI (*faacDecInit)(faacDecHandle hDecoder,
  62                                 unsigned char *buffer,
  63                                 unsigned long *samplerate,
  64                                 unsigned long *channels);
  65         int FAADAPI (*faacDecInit2)(faacDecHandle hDecoder, unsigned char *pBuffer,
  66                                 unsigned long SizeOfDecoderSpecificInfo,
  67                                 unsigned long *samplerate, unsigned long *channels);
  68         int FAADAPI (*faacDecDecode)(faacDecHandle hDecoder,
  69                                 unsigned char *buffer,
  70                                 unsigned long *bytesconsumed,
  71                                 short *sample_buffer,
  72                                 unsigned long *samples);
  73 #else
  74         unsigned char FAADAPI (*faacDecSetConfiguration)(faacDecHandle hDecoder,
  75                                                      faacDecConfigurationPtr config);
  76         long FAADAPI (*faacDecInit)(faacDecHandle hDecoder,
  77                                    unsigned char *buffer,
  78                                  unsigned long buffer_size,
  79                                  unsigned long *samplerate,
  80                                  unsigned char *channels);
  81         char FAADAPI (*faacDecInit2)(faacDecHandle hDecoder, unsigned char *pBuffer,
  82                                  unsigned long SizeOfDecoderSpecificInfo,
  83                                  unsigned long *samplerate, unsigned char *channels);
  84         void *FAADAPI (*faacDecDecode)(faacDecHandle hDecoder,
  85                                          faacDecFrameInfo *hInfo,
  86                                          unsigned char *buffer,
  87                                                                  unsigned long buffer_size);
  88         char* FAADAPI (*faacDecGetErrorMessage)(unsigned char errcode);
  89 #endif
  90
  91     void FAADAPI (*faacDecClose)(faacDecHandle hDecoder);
  92
  93
  94 } FAACContext;
  95
  96 static const unsigned long faac_srates[] =
  97 {
  98     96000, 88200, 64000, 48000, 44100, 32000,
  99     24000, 22050, 16000, 12000, 11025, 8000
 100 };
 101
 102 static int faac_init_mp4(AVCodecContext *avctx)
 103 {
 104     FAACContext *s = (FAACContext *) avctx->priv_data;
 105     unsigned long samplerate;
 106 #ifndef FAAD2_VERSION
 107     unsigned long channels;
 108 #else
 109     unsigned char channels;
 110 #endif
 111     int r = 0;
 112
 113     if (avctx->extradata){
 114         r = s->faacDecInit2(s->faac_handle, (uint8_t*) avctx->extradata,
 115                             avctx->extradata_size,
 116                             &samplerate, &channels);
 117         if (r < 0){
 118             av_log(avctx, AV_LOG_ERROR,
 119                    "faacDecInit2 failed r:%d   sr:%ld  ch:%ld  s:%d\n",
 120                    r, samplerate, (long)channels, avctx->extradata_size);
 121         } else {
 122             avctx->sample_rate = samplerate;
 123             avctx->channels = channels;
 124             s->init = 1;
 125         }
 126     }
 127
 128     return r;
 129 }
 130
 131 static int faac_decode_frame(AVCodecContext *avctx,
 132                              void *data, int *data_size,
 133                              uint8_t *buf, int buf_size)
 134 {
 135     FAACContext *s = (FAACContext *) avctx->priv_data;
 136 #ifndef FAAD2_VERSION
 137     unsigned long bytesconsumed;
 138     short *sample_buffer = NULL;
 139     unsigned long samples;
 140     int out;
 141 #else
 142     faacDecFrameInfo frame_info;
 143     void *out;
 144 #endif
 145     if(buf_size == 0)
 146         return 0;
 147 #ifndef FAAD2_VERSION
 148     out = s->faacDecDecode(s->faac_handle,
 149                            (unsigned char*)buf,
 150                            &bytesconsumed,
 151                            data,
 152                            &samples);
 153     samples *= s->sample_size;
 154     if (data_size)
 155         *data_size = samples;
 156     return (buf_size < (int)bytesconsumed)
 157         ? buf_size : (int)bytesconsumed;
 158 #else
 159
 160     if(!s->init){
 161         unsigned long srate;
 162         unsigned char channels;
 163         int r = s->faacDecInit(s->faac_handle, buf, buf_size, &srate, &channels);
 164         if(r < 0){
 165             av_log(avctx, AV_LOG_ERROR, "faac: codec init failed: %s\n",
 166                    s->faacDecGetErrorMessage(frame_info.error));
 167             return -1;
 168         }
 169         avctx->sample_rate = srate;
 170         avctx->channels = channels;
 171         s->init = 1;
 172     }
 173
 174     out = s->faacDecDecode(s->faac_handle, &frame_info, (unsigned char*)buf, (unsigned long)buf_size);
 175
 176     if (frame_info.error > 0) {
 177         av_log(avctx, AV_LOG_ERROR, "faac: frame decoding failed: %s\n",
 178                 s->faacDecGetErrorMessage(frame_info.error));
 179         return -1;
 180     }
 181
 182     frame_info.samples *= s->sample_size;
 183     memcpy(data, out, frame_info.samples); // CHECKME - can we cheat this one
 184
 185     if (data_size)
 186         *data_size = frame_info.samples;
 187
 188     return (buf_size < (int)frame_info.bytesconsumed)
 189         ? buf_size : (int)frame_info.bytesconsumed;
 190 #endif
 191 }
 192
 193 static int faac_decode_end(AVCodecContext *avctx)
 194 {
 195     FAACContext *s = (FAACContext *) avctx->priv_data;
 196
 197     if (s->faacDecClose)
 198         s->faacDecClose(s->faac_handle);
 199
 200     dlclose(s->handle);
 201     return 0;
 202 }
 203
 204 static int faac_decode_init(AVCodecContext *avctx)
 205 {
 206     FAACContext *s = (FAACContext *) avctx->priv_data;
 207     faacDecConfigurationPtr faac_cfg;
 208
 209 #ifdef CONFIG_FAADBIN
 210     const char* err = 0;
 211
 212     s->handle = dlopen(libfaadname, RTLD_LAZY);
 213     if (!s->handle)
 214     {
 215         av_log(avctx, AV_LOG_ERROR, "FAAD library: %s could not be opened! \n%s\n",
 216                 libfaadname, dlerror());
 217         return -1;
 218     }
 219 #define dfaac(a, b) \
 220     do { static const char* n = "faacDec" #a; \
 221     if ((s->faacDec ## a = b dlsym( s->handle, n )) == NULL) { err = n; break; } } while(0)
 222     for(;;) {
 223 #else  /* !CONFIG_FAADBIN */
 224 #define dfaac(a, b)     s->faacDec ## a = faacDec ## a
 225 #endif /* CONFIG_FAADBIN */
 226
 227         // resolve all needed function calls
 228         dfaac(Open, (faacDecHandle FAADAPI (*)(void)));
 229         dfaac(GetCurrentConfiguration, (faacDecConfigurationPtr
 230                                         FAADAPI (*)(faacDecHandle)));
 231 #ifndef FAAD2_VERSION
 232         dfaac(SetConfiguration, (int FAADAPI (*)(faacDecHandle,
 233                                                            faacDecConfigurationPtr)));
 234
 235         dfaac(Init, (int FAADAPI (*)(faacDecHandle, unsigned char*,
 236                                      unsigned long*, unsigned long*)));
 237     dfaac(Init2, (int FAADAPI (*)(faacDecHandle, unsigned char*,
 238                                        unsigned long, unsigned long*,
 239                                        unsigned long*)));
 240     dfaac(Close, (void FAADAPI (*)(faacDecHandle hDecoder)));
 241         dfaac(Decode, (int FAADAPI (*)(faacDecHandle, unsigned char*,
 242                              unsigned long*, short*, unsigned long*)));
 243 #else
 244         dfaac(SetConfiguration, (unsigned char FAADAPI (*)(faacDecHandle,
 245                                                            faacDecConfigurationPtr)));
 246         dfaac(Init, (long FAADAPI (*)(faacDecHandle, unsigned char*,
 247                                      unsigned long, unsigned long*, unsigned char*)));
 248         dfaac(Init2, (char FAADAPI (*)(faacDecHandle, unsigned char*,
 249                                        unsigned long, unsigned long*,
 250                                        unsigned char*)));
 251         dfaac(Decode, (void *FAADAPI (*)(faacDecHandle, faacDecFrameInfo*,
 252                              unsigned char*, unsigned long)));
 253         dfaac(GetErrorMessage, (char* FAADAPI (*)(unsigned char)));
 254 #endif
 255 #undef dfacc
 256
 257 #ifdef CONFIG_FAADBIN
 258         break;
 259     }
 260     if (err) {
 261         dlclose(s->handle);
 262         av_log(avctx, AV_LOG_ERROR, "FAAD library: cannot resolve %s in %s!\n",
 263                 err, libfaadname);
 264         return -1;
 265     }
 266 #endif
 267
 268     s->faac_handle = s->faacDecOpen();
 269     if (!s->faac_handle) {
 270         av_log(avctx, AV_LOG_ERROR, "FAAD library: cannot create handler!\n");
 271         faac_decode_end(avctx);
 272         return -1;
 273     }
 274
 275
 276     faac_cfg = s->faacDecGetCurrentConfiguration(s->faac_handle);
 277
 278     if (faac_cfg) {
 279         switch (avctx->bits_per_sample) {
 280         case 8: av_log(avctx, AV_LOG_ERROR, "FAADlib unsupported bps %d\n", avctx->bits_per_sample); break;
 281         default:
 282         case 16:
 283 #ifdef FAAD2_VERSION
 284             faac_cfg->outputFormat = FAAD_FMT_16BIT;
 285 #endif
 286             s->sample_size = 2;
 287             break;
 288         case 24:
 289 #ifdef FAAD2_VERSION
 290             faac_cfg->outputFormat = FAAD_FMT_24BIT;
 291 #endif
 292             s->sample_size = 3;
 293             break;
 294         case 32:
 295 #ifdef FAAD2_VERSION
 296             faac_cfg->outputFormat = FAAD_FMT_32BIT;
 297 #endif
 298             s->sample_size = 4;
 299             break;
 300         }
 301
 302         faac_cfg->defSampleRate = (!avctx->sample_rate) ? 44100 : avctx->sample_rate;
 303         faac_cfg->defObjectType = LC;
 304     }
 305
 306     s->faacDecSetConfiguration(s->faac_handle, faac_cfg);
 307
 308     faac_init_mp4(avctx);
 309
 310     return 0;
 311 }
 312
 313 #define AAC_CODEC(id, name)     \
 314 AVCodec name ## _decoder = {    \
 315     #name,                      \
 316     CODEC_TYPE_AUDIO,           \
 317     id,                         \
 318     sizeof(FAACContext),        \
 319     faac_decode_init,           \
 320     NULL,                       \
 321     faac_decode_end,            \
 322     faac_decode_frame,          \
 323 }
 324
 325 // FIXME - raw AAC files - maybe just one entry will be enough
 326 AAC_CODEC(CODEC_ID_AAC, aac);
 327 // If it's mp4 file - usually embeded into Qt Mov
 328 AAC_CODEC(CODEC_ID_MPEG4AAC, mpeg4aac);
 329
 330 #undef AAC_CODEC