FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
libfdk-aacdec.c
Go to the documentation of this file.
1 /*
2  * AAC decoder wrapper
3  * Copyright (c) 2012 Martin Storsjo
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 #include <fdk-aac/aacdecoder_lib.h>
23 
24 #include "libavutil/channel_layout.h"
25 #include "libavutil/common.h"
26 #include "libavutil/opt.h"
27 #include "avcodec.h"
28 #include "internal.h"
29 
30 enum ConcealMethod {
31  CONCEAL_METHOD_DEFAULT = -1,
32  CONCEAL_METHOD_SPECTRAL_MUTING = 0,
33  CONCEAL_METHOD_NOISE_SUBSTITUTION = 1,
34  CONCEAL_METHOD_ENERGY_INTERPOLATION = 2,
35  CONCEAL_METHOD_NB,
36 };
37 
38 typedef struct FDKAACDecContext {
39  const AVClass *class;
40  HANDLE_AACDECODER handle;
41  int initialized;
42  enum ConcealMethod conceal_method;
43 } FDKAACDecContext;
44 
45 #define OFFSET(x) offsetof(FDKAACDecContext, x)
46 #define AD AV_OPT_FLAG_AUDIO_PARAM | AV_OPT_FLAG_DECODING_PARAM
47 static const AVOption fdk_aac_dec_options[] = {
48  { "conceal", "Error concealment method", OFFSET(conceal_method), AV_OPT_TYPE_INT, { .i64 = CONCEAL_METHOD_DEFAULT }, CONCEAL_METHOD_DEFAULT, CONCEAL_METHOD_NB - 1, AD, "conceal" },
49  { "default", "Default", 0, AV_OPT_TYPE_CONST, { .i64 = CONCEAL_METHOD_DEFAULT }, INT_MIN, INT_MAX, AD, "conceal" },
50  { "spectral", "Spectral muting", 0, AV_OPT_TYPE_CONST, { .i64 = CONCEAL_METHOD_SPECTRAL_MUTING }, INT_MIN, INT_MAX, AD, "conceal" },
51  { "noise", "Noise Substitution", 0, AV_OPT_TYPE_CONST, { .i64 = CONCEAL_METHOD_NOISE_SUBSTITUTION }, INT_MIN, INT_MAX, AD, "conceal" },
52  { "energy", "Energy Interpolation", 0, AV_OPT_TYPE_CONST, { .i64 = CONCEAL_METHOD_ENERGY_INTERPOLATION }, INT_MIN, INT_MAX, AD, "conceal" },
53  { NULL }
54 };
55 
56 static const AVClass fdk_aac_dec_class = {
57  "libfdk-aac decoder", av_default_item_name, fdk_aac_dec_options, LIBAVUTIL_VERSION_INT
58 };
59 
60 static int get_stream_info(AVCodecContext *avctx)
61 {
62  FDKAACDecContext *s = avctx->priv_data;
63  CStreamInfo *info = aacDecoder_GetStreamInfo(s->handle);
64  int channel_counts[9] = { 0 };
65  int i, ch_error = 0;
66  uint64_t ch_layout = 0;
67 
68  if (!info) {
69  av_log(avctx, AV_LOG_ERROR, "Unable to get stream info\n");
70  return AVERROR_UNKNOWN;
71  }
72 
73  if (info->sampleRate <= 0) {
74  av_log(avctx, AV_LOG_ERROR, "Stream info not initialized\n");
75  return AVERROR_UNKNOWN;
76  }
77  avctx->sample_rate = info->sampleRate;
78  avctx->frame_size = info->frameSize;
79 
80  for (i = 0; i < info->numChannels; i++) {
81  AUDIO_CHANNEL_TYPE ctype = info->pChannelType[i];
82  if (ctype <= ACT_NONE || ctype > ACT_TOP) {
83  av_log(avctx, AV_LOG_WARNING, "unknown channel type\n");
84  break;
85  }
86  channel_counts[ctype]++;
87  }
88  av_log(avctx, AV_LOG_DEBUG,
89  "%d channels - front:%d side:%d back:%d lfe:%d top:%d\n",
90  info->numChannels,
91  channel_counts[ACT_FRONT], channel_counts[ACT_SIDE],
92  channel_counts[ACT_BACK], channel_counts[ACT_LFE],
93  channel_counts[ACT_FRONT_TOP] + channel_counts[ACT_SIDE_TOP] +
94  channel_counts[ACT_BACK_TOP] + channel_counts[ACT_TOP]);
95 
96  switch (channel_counts[ACT_FRONT]) {
97  case 4:
98  ch_layout |= AV_CH_LAYOUT_STEREO | AV_CH_FRONT_LEFT_OF_CENTER |
99  AV_CH_FRONT_RIGHT_OF_CENTER;
100  break;
101  case 3:
102  ch_layout |= AV_CH_LAYOUT_STEREO | AV_CH_FRONT_CENTER;
103  break;
104  case 2:
105  ch_layout |= AV_CH_LAYOUT_STEREO;
106  break;
107  case 1:
108  ch_layout |= AV_CH_FRONT_CENTER;
109  break;
110  default:
111  av_log(avctx, AV_LOG_WARNING,
112  "unsupported number of front channels: %d\n",
113  channel_counts[ACT_FRONT]);
114  ch_error = 1;
115  break;
116  }
117  if (channel_counts[ACT_SIDE] > 0) {
118  if (channel_counts[ACT_SIDE] == 2) {
119  ch_layout |= AV_CH_SIDE_LEFT | AV_CH_SIDE_RIGHT;
120  } else {
121  av_log(avctx, AV_LOG_WARNING,
122  "unsupported number of side channels: %d\n",
123  channel_counts[ACT_SIDE]);
124  ch_error = 1;
125  }
126  }
127  if (channel_counts[ACT_BACK] > 0) {
128  switch (channel_counts[ACT_BACK]) {
129  case 3:
130  ch_layout |= AV_CH_BACK_LEFT | AV_CH_BACK_RIGHT | AV_CH_BACK_CENTER;
131  break;
132  case 2:
133  ch_layout |= AV_CH_BACK_LEFT | AV_CH_BACK_RIGHT;
134  break;
135  case 1:
136  ch_layout |= AV_CH_BACK_CENTER;
137  break;
138  default:
139  av_log(avctx, AV_LOG_WARNING,
140  "unsupported number of back channels: %d\n",
141  channel_counts[ACT_BACK]);
142  ch_error = 1;
143  break;
144  }
145  }
146  if (channel_counts[ACT_LFE] > 0) {
147  if (channel_counts[ACT_LFE] == 1) {
148  ch_layout |= AV_CH_LOW_FREQUENCY;
149  } else {
150  av_log(avctx, AV_LOG_WARNING,
151  "unsupported number of LFE channels: %d\n",
152  channel_counts[ACT_LFE]);
153  ch_error = 1;
154  }
155  }
156  if (!ch_error &&
157  av_get_channel_layout_nb_channels(ch_layout) != info->numChannels) {
158  av_log(avctx, AV_LOG_WARNING, "unsupported channel configuration\n");
159  ch_error = 1;
160  }
161  if (ch_error)
162  avctx->channel_layout = 0;
163  else
164  avctx->channel_layout = ch_layout;
165 
166  avctx->channels = info->numChannels;
167 
168  return 0;
169 }
170 
171 static av_cold int fdk_aac_decode_close(AVCodecContext *avctx)
172 {
173  FDKAACDecContext *s = avctx->priv_data;
174 
175  if (s->handle)
176  aacDecoder_Close(s->handle);
177 
178  return 0;
179 }
180 
181 static av_cold int fdk_aac_decode_init(AVCodecContext *avctx)
182 {
183  FDKAACDecContext *s = avctx->priv_data;
184  AAC_DECODER_ERROR err;
185 
186  s->handle = aacDecoder_Open(avctx->extradata_size ? TT_MP4_RAW : TT_MP4_ADTS, 1);
187  if (!s->handle) {
188  av_log(avctx, AV_LOG_ERROR, "Error opening decoder\n");
189  return AVERROR_UNKNOWN;
190  }
191 
192  if (avctx->extradata_size) {
193  if ((err = aacDecoder_ConfigRaw(s->handle, &avctx->extradata,
194  &avctx->extradata_size)) != AAC_DEC_OK) {
195  av_log(avctx, AV_LOG_ERROR, "Unable to set extradata\n");
196  return AVERROR_INVALIDDATA;
197  }
198  }
199 
200  if (s->conceal_method != CONCEAL_METHOD_DEFAULT) {
201  if ((err = aacDecoder_SetParam(s->handle, AAC_CONCEAL_METHOD,
202  s->conceal_method)) != AAC_DEC_OK) {
203  av_log(avctx, AV_LOG_ERROR, "Unable to set error concealment method\n");
204  return AVERROR_UNKNOWN;
205  }
206  }
207 
208  avctx->sample_fmt = AV_SAMPLE_FMT_S16;
209 
210  return 0;
211 }
212 
213 static int fdk_aac_decode_frame(AVCodecContext *avctx, void *data,
214  int *got_frame_ptr, AVPacket *avpkt)
215 {
216  FDKAACDecContext *s = avctx->priv_data;
217  AVFrame *frame = data;
218  int ret;
219  AAC_DECODER_ERROR err;
220  UINT valid = avpkt->size;
221  uint8_t *buf, *tmpptr = NULL;
222  int buf_size;
223 
224  err = aacDecoder_Fill(s->handle, &avpkt->data, &avpkt->size, &valid);
225  if (err != AAC_DEC_OK) {
226  av_log(avctx, AV_LOG_ERROR, "aacDecoder_Fill() failed: %x\n", err);
227  return AVERROR_INVALIDDATA;
228  }
229 
230  if (s->initialized) {
231  frame->nb_samples = avctx->frame_size;
232  if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
233  return ret;
234  buf = frame->extended_data[0];
235  buf_size = avctx->channels * frame->nb_samples *
236  av_get_bytes_per_sample(avctx->sample_fmt);
237  } else {
238  buf_size = 50 * 1024;
239  buf = tmpptr = av_malloc(buf_size);
240  if (!buf)
241  return AVERROR(ENOMEM);
242  }
243 
244  err = aacDecoder_DecodeFrame(s->handle, (INT_PCM *) buf, buf_size, 0);
245  if (err == AAC_DEC_NOT_ENOUGH_BITS) {
246  ret = avpkt->size - valid;
247  goto end;
248  }
249  if (err != AAC_DEC_OK) {
250  av_log(avctx, AV_LOG_ERROR,
251  "aacDecoder_DecodeFrame() failed: %x\n", err);
252  ret = AVERROR_UNKNOWN;
253  goto end;
254  }
255 
256  if (!s->initialized) {
257  if ((ret = get_stream_info(avctx)) < 0)
258  goto end;
259  s->initialized = 1;
260  frame->nb_samples = avctx->frame_size;
261  }
262 
263  if (tmpptr) {
264  frame->nb_samples = avctx->frame_size;
265  if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
266  goto end;
267  memcpy(frame->extended_data[0], tmpptr,
268  avctx->channels * avctx->frame_size *
269  av_get_bytes_per_sample(avctx->sample_fmt));
270  }
271 
272  *got_frame_ptr = 1;
273  ret = avpkt->size - valid;
274 
275 end:
276  av_free(tmpptr);
277  return ret;
278 }
279 
280 static av_cold void fdk_aac_decode_flush(AVCodecContext *avctx)
281 {
282  FDKAACDecContext *s = avctx->priv_data;
283  AAC_DECODER_ERROR err;
284 
285  if (!s->handle)
286  return;
287 
288  if ((err = aacDecoder_SetParam(s->handle,
289  AAC_TPDEC_CLEAR_BUFFER, 1)) != AAC_DEC_OK)
290  av_log(avctx, AV_LOG_WARNING, "failed to clear buffer when flushing\n");
291 }
292 
293 AVCodec ff_libfdk_aac_decoder = {
294  .name = "libfdk_aac",
295  .long_name = NULL_IF_CONFIG_SMALL("Fraunhofer FDK AAC"),
296  .type = AVMEDIA_TYPE_AUDIO,
297  .id = AV_CODEC_ID_AAC,
298  .priv_data_size = sizeof(FDKAACDecContext),
299  .init = fdk_aac_decode_init,
300  .decode = fdk_aac_decode_frame,
301  .close = fdk_aac_decode_close,
302  .flush = fdk_aac_decode_flush,
303  .capabilities = CODEC_CAP_DR1,
304  .priv_class = &fdk_aac_dec_class,
305 };
Generated on Sun Mar 23 2014 23:49:58 for FFmpeg by   doxygen 1.8.2