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);
213 padd->handle = aac_open();
216 static void aacdec_close(struct filter_node *fn)
218 struct private_aacdec_data *padd = fn->private_data;
220 NeAACDecClose(padd->handle);
224 fn->private_data = NULL;
227 static void aacdec_post_select(__a_unused struct sched *s, struct task *t)
229 struct filter_node *fn = container_of(t, struct filter_node, task);
230 struct btr_node *btrn = fn->btrn;
231 struct private_aacdec_data *padd = fn->private_data;
233 unsigned char *p, *inbuf, *outbuffer;
235 size_t len, skip, consumed = 0, loaded, iqs;
239 ret = prepare_filter_node(fn);
244 len = btr_next_buffer(btrn, (char **)&inbuf);
245 iqs = btr_get_input_queue_size(btrn);
246 if (!padd->initialized) {
247 unsigned long rate = 0;
248 unsigned char channels = 0;
249 ret = aac_find_esds(inbuf, len, &skip, &padd->decoder_length);
251 PARA_INFO_LOG("%s\n", para_strerror(-ret));
252 ret = NeAACDecInit(padd->handle, inbuf,
253 len, &rate, &channels);
254 PARA_INFO_LOG("decoder init: %d\n", ret);
256 ret = -E_AACDEC_INIT;
261 PARA_INFO_LOG("decoder len: %lu\n",
262 padd->decoder_length);
264 p = inbuf + consumed;
265 ret = -E_AACDEC_INIT;
266 if (NeAACDecInit2(padd->handle, p,
267 padd->decoder_length, &rate,
271 padd->samplerate = rate;
272 padd->channels = channels;
273 PARA_INFO_LOG("rate: %u, channels: %d\n",
274 padd->samplerate, padd->channels);
275 padd->initialized = 1;
277 if (padd->decoder_length > 0) {
280 ret = aac_find_entry_point(inbuf + consumed,
281 len - consumed, &skip);
288 PARA_INFO_LOG("entry: %zu\n", padd->entry);
291 if (padd->consumed_total + len < padd->entry)
293 if (padd->consumed_total < padd->entry)
294 consumed = padd->entry - padd->consumed_total;
296 for (; consumed < len; consumed++)
297 if ((inbuf[consumed] & 0xfe) == 0x20)
301 p = inbuf + consumed;
302 //PARA_CRIT_LOG("consumed: %zu (%zu + %zu), have: %zu\n", padd->consumed_total + consumed,
303 // padd->consumed_total, consumed, len - consumed);
304 outbuffer = NeAACDecDecode(padd->handle, &padd->frame_info, p,
306 if (padd->frame_info.error) {
307 int err = padd->frame_info.error;
309 if (padd->error_count++ > MAX_ERRORS)
311 PARA_ERROR_LOG("frame_error: %d (%s), consumed: %zu + %zd + %lu\n",
312 err, NeAACDecGetErrorMessage(padd->frame_info.error),
313 padd->consumed_total, consumed,
314 padd->frame_info.bytesconsumed);
315 PARA_ERROR_LOG("%s\n", NeAACDecGetErrorMessage(
316 padd->frame_info.error));
317 consumed++; /* catch 21 */
320 padd->error_count = 0;
321 //PARA_CRIT_LOG("decoder ate %lu\n", padd->frame_info.bytesconsumed);
322 consumed += padd->frame_info.bytesconsumed;
324 if (!padd->frame_info.samples)
326 btr_buf = para_malloc(2 * padd->frame_info.samples);
328 for (i = 0; i < padd->frame_info.samples; i++) {
329 short sh = ((short *)outbuffer)[i];
330 write_int16_host_endian(btr_buf + loaded, sh);
333 btr_add_output(btr_buf, loaded, btrn);
338 padd->consumed_total += ret;
339 btr_consume(btrn, ret);
350 * the init function of the aacdec filter
352 * \param f pointer to the filter struct to initialize
356 void aacdec_filter_init(struct filter *f)
358 f->open = aacdec_open;
360 f->close = aacdec_close;
361 f->pre_select = aacdec_pre_select;
362 f->post_select = aacdec_post_select;
363 f->post_select = aacdec_post_select;
364 f->execute = aacdec_execute;