2 * Copyright (C) 2006-2009 Andre Noll <maan@systemlinux.org>
4 * Licensed under the GPL v2. For licencing details see COPYING.
7 * based in parts on libfaad, Copyright (C) 2003-2005 M. Bakker,
11 /** \file aacdec_filter.c paraslash's aac (m4a) decoder. */
20 #include "buffer_tree.h"
26 /** the output buffer size */
27 #define AAC_OUTBUF_SIZE (32 * 1024)
29 /** give up decoding after that many errors */
33 * data specific to the aacdec filter
35 * \sa filter, filter_node
37 struct private_aacdec_data {
38 /** the return value of aac_open */
39 NeAACDecHandle handle;
40 /** info about the currently decoded frame */
41 NeAACDecFrameInfo frame_info;
42 /** whether this instance of the aac decoder is already initialized */
45 * return value of aac_find_esds(). Used to call the right aacdec
48 unsigned long decoder_length;
49 /** number of times the decoder returned an error */
51 /** number of bytes already consumed from the imput stream */
52 size_t consumed_total;
53 /** return value of aac_find_entry_point */
55 /** The number of channels of the current stream. */
56 unsigned int channels;
57 /** Current sample rate in Hz. */
58 unsigned int samplerate;
61 #define AACDEC_MAX_PENDING (640 * 1024)
63 static void aacdec_pre_select(struct sched *s, struct task *t)
65 struct filter_node *fn = container_of(t, struct filter_node, task);
66 size_t iqs = btr_get_input_queue_size(fn->btrn);
71 if (btr_bytes_pending(fn->btrn) > AACDEC_MAX_PENDING)
72 return; /* FIXME, should use reasonable bound on timeout */
73 s->timeout.tv_sec = 0;
74 s->timeout.tv_usec = 1;
77 static ssize_t aacdec(char *input_buffer, size_t len, struct filter_node *fn)
79 struct private_aacdec_data *padd = fn->private_data;
80 struct filter_chain *fc = fn->fc;
82 unsigned char *p, *outbuffer;
83 unsigned char *inbuf = (unsigned char*)input_buffer;
84 size_t skip, consumed = 0;
86 if (fn->loaded > fn->bufsize * 3 / 5)
88 ret = *fc->input_error;
94 if (!padd->initialized) {
95 unsigned long rate = 0;
96 unsigned char channels = 0;
97 ret = aac_find_esds(inbuf, len, &skip, &padd->decoder_length);
99 PARA_INFO_LOG("%s\n", para_strerror(-ret));
100 ret = NeAACDecInit(padd->handle, inbuf,
101 len, &rate, &channels);
102 PARA_INFO_LOG("decoder init: %d\n", ret);
104 ret = -E_AACDEC_INIT;
109 PARA_INFO_LOG("decoder len: %lu\n",
110 padd->decoder_length);
112 p = inbuf + consumed;
113 ret = -E_AACDEC_INIT;
114 if (NeAACDecInit2(padd->handle, p,
115 padd->decoder_length, &rate,
119 fc->samplerate = rate;
120 fc->channels = channels;
121 PARA_INFO_LOG("rate: %u, channels: %d\n",
122 fc->samplerate, fc->channels);
123 padd->initialized = 1;
125 if (padd->decoder_length > 0) {
128 ret = aac_find_entry_point(inbuf + consumed,
129 len - consumed, &skip);
136 PARA_INFO_LOG("entry: %zu\n", padd->entry);
139 if (padd->consumed_total + len < padd->entry)
141 if (padd->consumed_total < padd->entry)
142 consumed = padd->entry - padd->consumed_total;
144 for (; consumed < len; consumed++)
145 if ((inbuf[consumed] & 0xfe) == 0x20)
149 p = inbuf + consumed;
150 outbuffer = NeAACDecDecode(padd->handle, &padd->frame_info, p,
152 if (padd->frame_info.error) {
154 if (padd->error_count++ > MAX_ERRORS)
156 PARA_ERROR_LOG("frame_error: %d, consumed: %zu + %zd + %lu\n",
157 padd->frame_info.error, padd->consumed_total,
158 consumed, padd->frame_info.bytesconsumed);
159 PARA_ERROR_LOG("%s\n", NeAACDecGetErrorMessage(
160 padd->frame_info.error));
161 consumed++; /* catch 21 */
164 padd->error_count = 0;
165 consumed += padd->frame_info.bytesconsumed;
167 if (!padd->frame_info.samples)
169 ret = -E_AAC_OVERRUN;
170 if (padd->frame_info.samples * 2 + fn->loaded > fn->bufsize)
172 for (i = 0; i < padd->frame_info.samples; i++) {
173 short *s = (short *)outbuffer;
174 write_int16_host_endian(fn->buf + fn->loaded, s[i]);
181 padd->consumed_total += ret;
185 static int aacdec_execute(struct btr_node *btrn, const char *cmd, char **result)
187 struct filter_node *fn = btr_context(btrn);
188 struct private_aacdec_data *padd = fn->private_data;
190 if (!strcmp(cmd, "samplerate")) {
191 if (padd->samplerate == 0)
192 return -ERRNO_TO_PARA_ERROR(ENAVAIL);
193 *result = make_message("%u", padd->samplerate);
196 if (!strcmp(cmd, "channels")) {
197 if (padd->channels == 0)
198 return -ERRNO_TO_PARA_ERROR(ENAVAIL);
199 *result = make_message("%u", padd->channels);
202 return -ERRNO_TO_PARA_ERROR(ENOTSUP);
205 static void aacdec_open(struct filter_node *fn)
207 struct private_aacdec_data *padd = para_calloc(sizeof(*padd));
209 fn->private_data = padd;
210 fn->bufsize = AAC_OUTBUF_SIZE;
211 fn->buf = para_calloc(fn->bufsize);
212 padd->handle = aac_open();
215 static void aacdec_close(struct filter_node *fn)
217 struct private_aacdec_data *padd = fn->private_data;
219 NeAACDecClose(padd->handle);
223 fn->private_data = NULL;
226 static void aacdec_post_select(__a_unused struct sched *s, struct task *t)
228 struct filter_node *fn = container_of(t, struct filter_node, task);
229 struct btr_node *btrn = fn->btrn;
230 struct private_aacdec_data *padd = fn->private_data;
232 unsigned char *p, *inbuf, *outbuffer;
234 size_t len, skip, consumed = 0, loaded, iqs;
238 iqs = btr_get_input_queue_size(btrn);
240 if (!btr_no_parent(btrn)) /* wait for more data */
243 if (btr_bytes_pending(btrn) > AACDEC_MAX_PENDING)
245 /* avoid "buffer too small" errors from the decoder */
247 len = btr_next_buffer(btrn, (char **)&inbuf);
250 if (btr_merge(btrn) < 2) /* only one buffer left */
253 //PARA_CRIT_LOG("next buffer: %zu\n", len);
255 if (btr_no_parent(btrn)) {
261 if (!padd->initialized) {
262 unsigned long rate = 0;
263 unsigned char channels = 0;
264 ret = aac_find_esds(inbuf, len, &skip, &padd->decoder_length);
266 PARA_INFO_LOG("%s\n", para_strerror(-ret));
267 ret = NeAACDecInit(padd->handle, inbuf,
268 len, &rate, &channels);
269 PARA_INFO_LOG("decoder init: %d\n", ret);
271 ret = -E_AACDEC_INIT;
276 PARA_INFO_LOG("decoder len: %lu\n",
277 padd->decoder_length);
279 p = inbuf + consumed;
280 ret = -E_AACDEC_INIT;
281 if (NeAACDecInit2(padd->handle, p,
282 padd->decoder_length, &rate,
286 padd->samplerate = rate;
287 padd->channels = channels;
288 PARA_INFO_LOG("rate: %u, channels: %d\n",
289 padd->samplerate, padd->channels);
290 padd->initialized = 1;
292 if (padd->decoder_length > 0) {
295 ret = aac_find_entry_point(inbuf + consumed,
296 len - consumed, &skip);
303 PARA_INFO_LOG("entry: %zu\n", padd->entry);
306 if (padd->consumed_total + len < padd->entry)
308 if (padd->consumed_total < padd->entry)
309 consumed = padd->entry - padd->consumed_total;
311 for (; consumed < len; consumed++)
312 if ((inbuf[consumed] & 0xfe) == 0x20)
316 p = inbuf + consumed;
317 //PARA_CRIT_LOG("consumed: %zu (%zu + %zu), have: %zu\n", padd->consumed_total + consumed,
318 // padd->consumed_total, consumed, len - consumed);
319 outbuffer = NeAACDecDecode(padd->handle, &padd->frame_info, p,
321 if (padd->frame_info.error) {
322 int err = padd->frame_info.error;
324 if (padd->error_count++ > MAX_ERRORS)
326 PARA_ERROR_LOG("frame_error: %d (%s), consumed: %zu + %zd + %lu\n",
327 err, NeAACDecGetErrorMessage(padd->frame_info.error),
328 padd->consumed_total, consumed,
329 padd->frame_info.bytesconsumed);
330 PARA_ERROR_LOG("%s\n", NeAACDecGetErrorMessage(
331 padd->frame_info.error));
332 consumed++; /* catch 21 */
335 padd->error_count = 0;
336 //PARA_CRIT_LOG("decoder ate %lu\n", padd->frame_info.bytesconsumed);
337 consumed += padd->frame_info.bytesconsumed;
339 if (!padd->frame_info.samples)
341 btr_buf = para_malloc(2 * padd->frame_info.samples);
343 for (i = 0; i < padd->frame_info.samples; i++) {
344 short sh = ((short *)outbuffer)[i];
345 write_int16_host_endian(btr_buf + loaded, sh);
348 btr_add_output(btr_buf, loaded, btrn);
353 padd->consumed_total += ret;
354 btr_consume(btrn, ret);
365 * the init function of the aacdec filter
367 * \param f pointer to the filter struct to initialize
371 void aacdec_filter_init(struct filter *f)
373 f->open = aacdec_open;
375 f->close = aacdec_close;
376 f->pre_select = aacdec_pre_select;
377 f->post_select = aacdec_post_select;
378 f->post_select = aacdec_post_select;
379 f->execute = aacdec_execute;