From f40bfc9267b13b54e6379dfe7539447662879d24 Mon Sep 17 00:00:00 2001 From: Sean Bartell Date: Sat, 25 Jun 2011 21:32:25 -0400 Subject: Add codecs to librbcodec. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Change-Id: Id7f4717d51ed02d67cb9f9cb3c0ada4a81843f97 Reviewed-on: http://gerrit.rockbox.org/137 Reviewed-by: Nils Wallménius Tested-by: Nils Wallménius --- lib/rbcodec/codecs/wmavoice.c | 195 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 195 insertions(+) create mode 100644 lib/rbcodec/codecs/wmavoice.c (limited to 'lib/rbcodec/codecs/wmavoice.c') diff --git a/lib/rbcodec/codecs/wmavoice.c b/lib/rbcodec/codecs/wmavoice.c new file mode 100644 index 0000000000..9cf5a49f1a --- /dev/null +++ b/lib/rbcodec/codecs/wmavoice.c @@ -0,0 +1,195 @@ +/*************************************************************************** + * __________ __ ___. + * Open \______ \ ____ ____ | | _\_ |__ _______ ___ + * Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ / + * Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < < + * Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \ + * \/ \/ \/ \/ \/ + * $Id$ + * + * Copyright (C) 2010 Mohamed Tarek + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY + * KIND, either express or implied. + * + ****************************************************************************/ + +#include "codeclib.h" +#include "libasf/asf.h" +#include "libwmavoice/wmavoice.h" + +CODEC_HEADER + +static AVCodecContext avctx; +static AVPacket avpkt; + +#define MAX_FRAMES 3 /*maximum number of frames per superframe*/ +#define MAX_FRAMESIZE 160 /* maximum number of samples per frame */ +#define BUFSIZE MAX_FRAMES*MAX_FRAMESIZE +static int32_t decoded[BUFSIZE] IBSS_ATTR; + + +/* This function initialises AVCodecContext with the data needed for the wmapro + * decoder to work. The required data is taken from asf_waveformatex_t because that's + * what the rockbox asf metadata parser fill/work with. In the future, when the + * codec is being optimised for on-target playback this function should not be needed. */ +static void init_codec_ctx(AVCodecContext *avctx, asf_waveformatex_t *wfx) +{ + /* Copy the extra-data */ + avctx->extradata_size = wfx->datalen; + avctx->extradata = (uint8_t *)malloc(wfx->datalen*sizeof(uint8_t)); + memcpy(avctx->extradata, wfx->data, wfx->datalen*sizeof(uint8_t)); + + avctx->block_align = wfx->blockalign; + avctx->sample_rate = wfx->rate; + avctx->channels = wfx->channels; + +} + +/* this is the codec entry point */ +enum codec_status codec_main(enum codec_entry_call_reason reason) +{ + if (reason == CODEC_LOAD) { + /* Generic codec initialisation */ + ci->configure(DSP_SET_SAMPLE_DEPTH, 31); + } + + return CODEC_OK; +} + +/* this is called for each file to process */ +enum codec_status codec_run(void) +{ + uint32_t elapsedtime; + asf_waveformatex_t wfx; /* Holds the stream properties */ + size_t resume_offset; + int res; /* Return values from asf_read_packet() and decode_packet() */ + uint8_t* audiobuf; /* Pointer to the payload of one wma pro packet */ + int audiobufsize; /* Payload size */ + int packetlength = 0; /* Logical packet size (minus the header size) */ + int outlen = 0; /* Number of bytes written to the output buffer */ + int pktcnt = 0; /* Count of the packets played */ + intptr_t param; + + /* Remember the resume position */ + resume_offset = ci->id3->offset; +restart_track: + if (codec_init()) { + LOGF("(WMA Voice) Error: Error initialising codec\n"); + return CODEC_ERROR; + } + + /* Copy the format metadata we've stored in the id3 TOC field. This + saves us from parsing it again here. */ + memcpy(&wfx, ci->id3->toc, sizeof(wfx)); + memset(&avctx, 0, sizeof(AVCodecContext)); + memset(&avpkt, 0, sizeof(AVPacket)); + + ci->configure(DSP_SWITCH_FREQUENCY, wfx.rate); + ci->configure(DSP_SET_STEREO_MODE, wfx.channels == 1 ? + STEREO_MONO : STEREO_INTERLEAVED); + codec_set_replaygain(ci->id3); + + ci->seek_buffer(0); + + /* Initialise the AVCodecContext */ + init_codec_ctx(&avctx, &wfx); + + if (wmavoice_decode_init(&avctx) < 0) { + LOGF("(WMA Voice) Error: Unsupported or corrupt file\n"); + return CODEC_ERROR; + } + + /* Now advance the file position to the first frame */ + ci->seek_buffer(ci->id3->first_frame_offset); + + elapsedtime = 0; + ci->set_elapsed(0); + + resume_offset = 0; + + /* The main decoding loop */ + + while (pktcnt < wfx.numpackets) + { + enum codec_command_action action = ci->get_command(¶m); + + if (action == CODEC_ACTION_HALT) + break; + + /* Deal with any pending seek requests */ + if (action == CODEC_ACTION_SEEK_TIME) { + ci->set_elapsed(param); + + if (param == 0) { + ci->set_elapsed(0); + ci->seek_complete(); + goto restart_track; /* Pretend you never saw this... */ + } + + elapsedtime = asf_seek(param, &wfx); + if (elapsedtime < 1){ + ci->set_elapsed(0); + ci->seek_complete(); + goto next_track; + } + + ci->set_elapsed(elapsedtime); + ci->seek_complete(); + } + +new_packet: + res = asf_read_packet(&audiobuf, &audiobufsize, &packetlength, &wfx); + + if (res < 0) { + LOGF("(WMA Voice) read_packet error %d\n",res); + return CODEC_ERROR; + } else { + avpkt.data = audiobuf; + avpkt.size = audiobufsize; + pktcnt++; + + while(avpkt.size > 0) + { + /* wmavoice_decode_packet checks for the output buffer size to + avoid overflows */ + outlen = BUFSIZE*sizeof(int32_t); + + res = wmavoice_decode_packet(&avctx, decoded, &outlen, &avpkt); + if(res < 0) { + LOGF("(WMA Voice) Error: decode_packet returned %d", res); + if(res == ERROR_WMAPRO_IN_WMAVOICE){ + /* Just skip this packet */ + ci->advance_buffer(packetlength); + goto new_packet; + } + else { + return CODEC_ERROR; + } + } + avpkt.data += res; + avpkt.size -= res; + if(outlen) { + ci->yield (); + outlen /= sizeof(int32_t); + ci->pcmbuf_insert(decoded, NULL, outlen); + elapsedtime += outlen*10/(wfx.rate/100); + ci->set_elapsed(elapsedtime); + ci->yield (); + } + } + + } + + /* Advance to the next logical packet */ + ci->advance_buffer(packetlength); + } + + return CODEC_OK; +} + -- cgit v1.2.3