apps/codecs/wma.c

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  * $Id$
   9  *
  10  * Copyright (C) 2007 Dave Chapman
  11  *
  12  * This program is free software; you can redistribute it and/or
  13  * modify it under the terms of the GNU General Public License
  14  * as published by the Free Software Foundation; either version 2
  15  * of the License, or (at your option) any later version.
  16  *
  17  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  18  * KIND, either express or implied.
  19  *
  20  ****************************************************************************/
  21
  22 #include "codeclib.h"
  23 #include "libasf/asf.h"
  24 #include "libwma/wmadec.h"
  25
  26 CODEC_HEADER
  27
  28 /* NOTE: WMADecodeContext is 120152 bytes (on x86) */
  29 static WMADecodeContext wmadec;
  30
  31 /* this is the codec entry point */
  32 enum codec_status codec_main(enum codec_entry_call_reason reason)
  33 {
  34     if (reason == CODEC_LOAD) {
  35         /* Generic codec initialisation */
  36         ci->configure(DSP_SET_SAMPLE_DEPTH, 29);
  37     }
  38
  39     return CODEC_OK;
  40 }
  41
  42 /* this is called for each file to process */
  43 enum codec_status codec_run(void)
  44 {
  45     uint32_t elapsedtime;
  46     asf_waveformatex_t wfx;
  47     size_t resume_offset;
  48     int i;
  49     int wmares;
  50     int res = 0;
  51     uint8_t* audiobuf;
  52     int audiobufsize;
  53     int packetlength = 0;
  54     int errcount = 0;
  55     intptr_t param;
  56
  57     /* Proper reset of the decoder context. */
  58     memset(&wmadec, 0, sizeof(wmadec));
  59
  60     /* Remember the resume position - when the codec is opened, the
  61        playback engine will reset it. */
  62     resume_offset = ci->id3->offset;
  63
  64 restart_track:
  65     if (codec_init()) {
  66         LOGF("WMA: Error initialising codec\n");
  67         return CODEC_ERROR;
  68     }
  69
  70     /* Copy the format metadata we've stored in the id3 TOC field.  This
  71        saves us from parsing it again here. */
  72     memcpy(&wfx, ci->id3->toc, sizeof(wfx));
  73
  74     ci->seek_buffer(ci->id3->first_frame_offset);
  75     if (wma_decode_init(&wmadec,&wfx) < 0) {
  76         LOGF("WMA: Unsupported or corrupt file\n");
  77         return CODEC_ERROR;
  78     }
  79
  80     if (resume_offset > ci->id3->first_frame_offset)
  81     {
  82         /* Get start of current packet */
  83         int packet_offset = (resume_offset - ci->id3->first_frame_offset)
  84             % wfx.packet_size;
  85         ci->seek_buffer(resume_offset - packet_offset);
  86         elapsedtime = asf_get_timestamp(&i);
  87         ci->set_elapsed(elapsedtime);
  88     }
  89     else
  90     {
  91         /* Now advance the file position to the first frame */
  92         ci->seek_buffer(ci->id3->first_frame_offset);
  93         elapsedtime = 0;
  94     }
  95
  96     resume_offset = 0;
  97     ci->configure(DSP_SWITCH_FREQUENCY, wfx.rate);
  98     ci->configure(DSP_SET_STEREO_MODE, wfx.channels == 1 ?
  99                   STEREO_MONO : STEREO_NONINTERLEAVED);
 100     codec_set_replaygain(ci->id3);
 101
 102     /* The main decoding loop */
 103     while (res >= 0)
 104     {
 105         enum codec_command_action action = ci->get_command(&param);
 106
 107         if (action == CODEC_ACTION_HALT)
 108             break;
 109
 110         /* Deal with any pending seek requests */
 111         if (action == CODEC_ACTION_SEEK_TIME) {
 112
 113             if (param == 0) {
 114                 ci->set_elapsed(0);
 115                 ci->seek_complete();
 116                 goto restart_track; /* Pretend you never saw this... */
 117             }
 118
 119             elapsedtime = asf_seek(param, &wfx);
 120             if (elapsedtime < 1){
 121                 ci->set_elapsed(0);
 122                 ci->seek_complete();
 123                 break;
 124             }
 125             /*DEBUGF("Seek returned %d\n", (int)elapsedtime);*/
 126
 127             /*flush the wma decoder state*/
 128             wmadec.last_superframe_len = 0;
 129             wmadec.last_bitoffset = 0;
 130
 131             ci->set_elapsed(elapsedtime);
 132             ci->seek_complete();
 133         }
 134         errcount = 0;
 135 new_packet:
 136         res = asf_read_packet(&audiobuf, &audiobufsize, &packetlength, &wfx);
 137
 138         if (res < 0) {
 139             /* We'll try to recover from a parse error a certain number of
 140              * times. If we succeed, the error counter will be reset.
 141              */
 142
 143             if (res == ASF_ERROR_EOF) {
 144                 /* File ended - not an error */
 145                 break;
 146             }
 147
 148             errcount++;
 149             DEBUGF("read_packet error %d, errcount %d\n",wmares, errcount);
 150             if (errcount > 5) {
 151                 return CODEC_ERROR;
 152             } else {
 153                 ci->advance_buffer(packetlength);
 154                 goto new_packet;
 155             }
 156         } else if (res > 0) {
 157             wma_decode_superframe_init(&wmadec, audiobuf, audiobufsize);
 158
 159             for (i=0; i < wmadec.nb_frames; i++)
 160             {
 161                 wmares = wma_decode_superframe_frame(&wmadec,
 162                                                      audiobuf, audiobufsize);
 163
 164                 ci->yield ();
 165
 166                 if (wmares < 0) {
 167                     /* Do the above, but for errors in decode. */
 168                     errcount++;
 169                     DEBUGF("WMA decode error %d, errcount %d\n",wmares, errcount);
 170                     if (errcount > 5) {
 171                         return CODEC_ERROR;
 172                     } else {
 173                         ci->advance_buffer(packetlength);
 174                         goto new_packet;
 175                     }
 176                 } else if (wmares > 0) {
 177                     ci->pcmbuf_insert((*wmadec.frame_out)[0], (*wmadec.frame_out)[1], wmares);
 178                     elapsedtime += (wmares*10)/(wfx.rate/100);
 179                     ci->set_elapsed(elapsedtime);
 180                 }
 181             }
 182         }
 183
 184         ci->advance_buffer(packetlength);
 185     }
 186
 187     /*LOGF("WMA: Decoded %ld samples\n",elapsedtime*wfx.rate/1000);*/
 188     return CODEC_OK;
 189 }