apps/codecs/cook.c

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  * Copyright (C) 2009 Mohamed Tarek
   9  *
  10  * This program is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU General Public License
  12  * as published by the Free Software Foundation; either version 2
  13  * of the License, or (at your option) any later version.
  14  *
  15  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  16  * KIND, either express or implied.
  17  *
  18  ****************************************************************************/
  19
  20 #include <string.h>
  21
  22 #include "logf.h"
  23 #include "codeclib.h"
  24 #include "inttypes.h"
  25 #include "libcook/cook.h"
  26
  27 #define DATA_HEADER_SIZE 18 /* size of DATA chunk header in a rm file */
  28
  29 CODEC_HEADER
  30
  31 RMContext rmctx;
  32 RMPacket pkt;
  33 COOKContext q IBSS_ATTR;
  34
  35 static void init_rm(RMContext *rmctx)
  36 {
  37     memcpy(rmctx, (void*)(( (intptr_t)ci->id3->id3v2buf + 3 ) &~ 3), sizeof(RMContext));
  38 }
  39
  40 /* this is the codec entry point */
  41 enum codec_status codec_main(void)
  42 {
  43     static size_t buff_size;
  44     int datasize, res, consumed, i, j, time_offset;
  45     uint8_t *bit_buffer;
  46     int16_t outbuf[2048] __attribute__((aligned(32)));
  47     uint16_t fs,sps,h;
  48     uint32_t packet_count;
  49     int scrambling_unit_size, flag_seek_block = 0;
  50
  51 next_track:
  52     if (codec_init()) {
  53         DEBUGF("codec init failed\n");
  54         return CODEC_ERROR;
  55     }
  56     while (!*ci->taginfo_ready && !ci->stop_codec)
  57         ci->sleep(1);
  58
  59     codec_set_replaygain(ci->id3);
  60     ci->memset(&rmctx,0,sizeof(RMContext));
  61     ci->memset(&pkt,0,sizeof(RMPacket));
  62     ci->memset(&q,0,sizeof(COOKContext));
  63
  64     init_rm(&rmctx);
  65
  66     ci->configure(DSP_SET_FREQUENCY, ci->id3->frequency);
  67     ci->configure(DSP_SET_SAMPLE_DEPTH, 16);
  68     ci->configure(DSP_SET_STEREO_MODE, rmctx.nb_channels == 1 ?
  69                   STEREO_MONO : STEREO_INTERLEAVED);
  70
  71     packet_count = rmctx.nb_packets;
  72     rmctx.audio_framesize = rmctx.block_align;
  73     rmctx.block_align = rmctx.sub_packet_size;
  74     fs = rmctx.audio_framesize;
  75     sps= rmctx.block_align;
  76     h = rmctx.sub_packet_h;
  77     scrambling_unit_size = h*fs;
  78
  79     res =cook_decode_init(&rmctx, &q);
  80     if(res < 0) {
  81         DEBUGF("failed to initialize cook decoder\n");
  82         return CODEC_ERROR;
  83     }
  84
  85     ci->set_elapsed(0);
  86     ci->advance_buffer(rmctx.data_offset + DATA_HEADER_SIZE);
  87
  88     /* The main decoder loop */
  89 seek_start :
  90     while(packet_count)
  91     {
  92         bit_buffer = (uint8_t *) ci->request_buffer(&buff_size, scrambling_unit_size);
  93         consumed = rm_get_packet(&bit_buffer, &rmctx, &pkt);
  94         if(consumed < 0) {
  95             DEBUGF("rm_get_packet failed\n");
  96             return CODEC_ERROR;
  97         }
  98
  99         for(i = 0; i < rmctx.audio_pkt_cnt*(fs/sps) ; i++)
 100         {
 101             ci->yield();
 102             if (ci->stop_codec || ci->new_track)
 103                 goto done;
 104
 105              if (ci->seek_time) {
 106                 ci->seek_buffer(rmctx.data_offset + DATA_HEADER_SIZE);
 107                 packet_count = rmctx.nb_packets;
 108                 rmctx.audio_pkt_cnt = 0;
 109                 rmctx.frame_number = 0;
 110                 if (ci->seek_time == 1) {
 111                     ci->set_elapsed(0);
 112                     ci->seek_complete();
 113                     goto seek_start;
 114                 }
 115                 j = 0;
 116                 while(1) {
 117                     rmctx.audio_pkt_cnt = 0;
 118                     bit_buffer = (uint8_t *) ci->request_buffer(&buff_size, scrambling_unit_size);
 119                     consumed = rm_get_packet(&bit_buffer, &rmctx, &pkt);
 120                     if(consumed < 0) {
 121                         DEBUGF("rm_get_packet failed\n");
 122                         return CODEC_ERROR;
 123                     }
 124                     if(rmctx.audiotimestamp < (unsigned) ci->seek_time) {
 125                         ci->advance_buffer(consumed);
 126                         packet_count -= rmctx.audio_pkt_cnt;
 127                         rmctx.frame_number += h*(fs/sps);
 128                         j+= consumed;
 129                     }
 130                     if(rmctx.audiotimestamp > (unsigned) ci->seek_time) {
 131                         flag_seek_block = 1;
 132                         rmctx.audio_pkt_cnt = 0;
 133                         break;
 134                     }
 135                 }
 136
 137                 if(flag_seek_block) {
 138                     ci->seek_buffer(rmctx.data_offset + DATA_HEADER_SIZE + j - consumed);
 139                     bit_buffer = (uint8_t *) ci->request_buffer(&buff_size, scrambling_unit_size);
 140                     consumed = rm_get_packet(&bit_buffer, &rmctx, &pkt);
 141                     packet_count += rmctx.audio_pkt_cnt;
 142                     rmctx.frame_number -= h*(fs/sps);
 143                     time_offset = ci->seek_time - rmctx.audiotimestamp;
 144                     i = (time_offset/((sps * 8 * 1000)/rmctx.bit_rate));
 145                     flag_seek_block = 0;
 146                     ci->set_elapsed(rmctx.audiotimestamp+(1000*8*sps/rmctx.bit_rate)*i);
 147                     ci->seek_complete();
 148                 }
 149              }
 150             res = cook_decode_frame(&rmctx,&q, outbuf, &datasize, pkt.frames[i], rmctx.block_align);
 151             rmctx.frame_number++;
 152
 153             /* skip the first two frames; no valid audio */
 154             if(rmctx.frame_number < 3) continue;
 155
 156             if(res != rmctx.block_align) {
 157                 DEBUGF("codec error\n");
 158                 return CODEC_ERROR;
 159             }
 160
 161             ci->pcmbuf_insert(outbuf, NULL, rmctx.samples_pf_pc / rmctx.nb_channels);
 162             ci->set_elapsed(rmctx.audiotimestamp+(1000*8*sps/rmctx.bit_rate)*i);
 163         }
 164         packet_count -= rmctx.audio_pkt_cnt;
 165         rmctx.audio_pkt_cnt = 0;
 166         ci->advance_buffer(consumed);
 167     }
 168
 169     done :
 170     if (ci->request_next_track())
 171         goto next_track;
 172
 173     return CODEC_OK;
 174 }