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

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