mp3lib/sr1.c

   1 // #define NEWBUFFERING
   2 //#define DEBUG_RESYNC
   3
   4 /* 1 frame = 4608 byte PCM */
   5
   6 #define LOCAL static inline
   7
   8 //#undef LOCAL
   9 //#define LOCAL
  10
  11 #include        <stdlib.h>
  12 #include        <stdio.h>
  13 #include        <string.h>
  14 #include        <math.h>
  15
  16 #include "mpg123.h"
  17 #include "huffman.h"
  18 #include "mp3.h"
  19 #include "mpbswap.h"
  20 #include "cpudetect.h"
  21 #include "mp_msg.h"
  22 #include "libmpcodecs/ad_mp3lib.h"
  23 #include "libvo/fastmemcpy.h"
  24
  25 #include "libavutil/common.h"
  26
  27 #if ARCH_X86_64
  28 // 3DNow! and 3DNow!Ext routines don't compile under AMD64
  29 #undef HAVE_AMD3DNOW
  30 #undef HAVE_AMD3DNOWEXT
  31 #define HAVE_AMD3DNOW 0
  32 #define HAVE_AMD3DNOWEXT 0
  33 #endif
  34
  35 //static FILE* mp3_file=NULL;
  36
  37 int MP3_frames=0;
  38 int MP3_eof=0;
  39 int MP3_pause=0;
  40 int MP3_filesize=0;
  41 int MP3_fpos=0;      // current file position
  42 int MP3_framesize=0; // current framesize
  43 int MP3_bitrate=0;   // current bitrate
  44 int MP3_samplerate=0;  // current samplerate
  45 int MP3_resync=0;
  46 int MP3_channels=0;
  47 int MP3_bps=2;
  48
  49 static long outscale = 32768;
  50 #include "tabinit.c"
  51
  52 #if 1
  53 LOCAL int mp3_read(char *buf,int size){
  54 //  int len=fread(buf,1,size,mp3_file);
  55   int len=mplayer_audio_read(buf,size);
  56   if(len>0) MP3_fpos+=len;
  57 //  if(len!=size) MP3_eof=1;
  58   return len;
  59 }
  60 #else
  61 int mp3_read(char *buf,int size);
  62 #endif
  63 /*
  64  * Modified for use with MPlayer, for details see the changelog at
  65  * http://svn.mplayerhq.hu/mplayer/trunk/
  66  * $Id$
  67  */
  68
  69
  70 //void mp3_seek(int pos){
  71 //  fseek(mp3_file,pos,SEEK_SET);
  72 //  return MP3_fpos = ftell(mp3_file);
  73 //}
  74
  75 /*       Frame reader           */
  76
  77 #define MAXFRAMESIZE 1280
  78 #define MAXFRAMESIZE2 (512+MAXFRAMESIZE)
  79
  80 static int fsizeold=0,ssize=0;
  81 static unsigned char bsspace[2][MAXFRAMESIZE2]; /* !!!!! */
  82 static unsigned char *bsbufold=bsspace[0]+512;
  83 static unsigned char *bsbuf=bsspace[1]+512;
  84 static int bsnum=0;
  85
  86 static int bitindex;
  87 static unsigned char *wordpointer;
  88 static int bitsleft;
  89
  90 static unsigned char *pcm_sample;   /* outbuffer address */
  91 static int pcm_point = 0;           /* outbuffer offset */
  92
  93 static struct frame fr;
  94
  95 static int tabsel_123[2][3][16] = {
  96    { {0,32,64,96,128,160,192,224,256,288,320,352,384,416,448,},
  97      {0,32,48,56, 64, 80, 96,112,128,160,192,224,256,320,384,},
  98      {0,32,40,48, 56, 64, 80, 96,112,128,160,192,224,256,320,} },
  99
 100    { {0,32,48,56,64,80,96,112,128,144,160,176,192,224,256,},
 101      {0,8,16,24,32,40,48,56,64,80,96,112,128,144,160,},
 102      {0,8,16,24,32,40,48,56,64,80,96,112,128,144,160,} }
 103 };
 104
 105 static int freqs[9] = { 44100, 48000, 32000, 22050, 24000, 16000 , 11025 , 12000 , 8000 };
 106
 107 LOCAL unsigned int getbits(short number_of_bits)
 108 {
 109   unsigned rval;
 110 //  if(MP3_frames>=7741) printf("getbits: bits=%d  bitsleft=%d  wordptr=%x\n",number_of_bits,bitsleft,wordpointer);
 111   if((bitsleft-=number_of_bits)<0) return 0;
 112   if(!number_of_bits) return 0;
 113          rval = wordpointer[0];
 114          rval <<= 8;
 115          rval |= wordpointer[1];
 116          rval <<= 8;
 117          rval |= wordpointer[2];
 118          rval <<= bitindex;
 119          rval &= 0xffffff;
 120          bitindex += number_of_bits;
 121          rval >>= (24-number_of_bits);
 122          wordpointer += (bitindex>>3);
 123          bitindex &= 7;
 124   return rval;
 125 }
 126
 127
 128 LOCAL unsigned int getbits_fast(short number_of_bits)
 129 {
 130   unsigned rval;
 131 //  if(MP3_frames>=7741) printf("getbits_fast: bits=%d  bitsleft=%d  wordptr=%x\n",number_of_bits,bitsleft,wordpointer);
 132   if((bitsleft-=number_of_bits)<0) return 0;
 133   if(!number_of_bits) return 0;
 134 #if ARCH_X86
 135   rval = bswap_16(*((uint16_t *)wordpointer));
 136 #else
 137   /*
 138    * we may not be able to address unaligned 16-bit data on non-x86 cpus.
 139    * Fall back to some portable code.
 140    */
 141   rval = wordpointer[0] << 8 | wordpointer[1];
 142 #endif
 143          rval <<= bitindex;
 144          rval &= 0xffff;
 145          bitindex += number_of_bits;
 146          rval >>= (16-number_of_bits);
 147          wordpointer += (bitindex>>3);
 148          bitindex &= 7;
 149   return rval;
 150 }
 151
 152 LOCAL unsigned int get1bit(void)
 153 {
 154   unsigned char rval;
 155 //  if(MP3_frames>=7741) printf("get1bit: bitsleft=%d  wordptr=%x\n",bitsleft,wordpointer);
 156   if((--bitsleft)<0) return 0;
 157   rval = *wordpointer << bitindex;
 158   bitindex++;
 159   wordpointer += (bitindex>>3);
 160   bitindex &= 7;
 161   return (rval >> 7) & 1;
 162 }
 163
 164 LOCAL void set_pointer(int backstep)
 165 {
 166 //  if(backstep!=512 && backstep>fsizeold)
 167 //    printf("\rWarning! backstep (%d>%d)                                         \n",backstep,fsizeold);
 168   wordpointer = bsbuf + ssize - backstep;
 169   if (backstep) fast_memcpy(wordpointer,bsbufold+fsizeold-backstep,backstep);
 170   bitindex = 0;
 171   bitsleft+=8*backstep;
 172 //  printf("Backstep %d  (bitsleft=%d)\n",backstep,bitsleft);
 173 }
 174
 175 LOCAL int stream_head_read(unsigned char *hbuf,uint32_t *newhead){
 176   if(mp3_read(hbuf,4) != 4) return FALSE;
 177 #if ARCH_X86
 178   *newhead = bswap_32(*((uint32_t*)hbuf));
 179 #else
 180   /*
 181    * we may not be able to address unaligned 32-bit data on non-x86 cpus.
 182    * Fall back to some portable code.
 183    */
 184   *newhead =
 185       hbuf[0] << 24 |
 186       hbuf[1] << 16 |
 187       hbuf[2] <<  8 |
 188       hbuf[3];
 189 #endif
 190   return TRUE;
 191 }
 192
 193 LOCAL int stream_head_shift(unsigned char *hbuf,uint32_t *head){
 194   *((uint32_t*)hbuf) >>= 8;
 195   if(mp3_read(hbuf+3,1) != 1) return 0;
 196   *head <<= 8;
 197   *head |= hbuf[3];
 198   return 1;
 199 }
 200
 201 /*
 202  * decode a header and write the information
 203  * into the frame structure
 204  */
 205 LOCAL int decode_header(struct frame *fr,uint32_t newhead){
 206
 207     // head_check:
 208     if( (newhead & 0xffe00000) != 0xffe00000 ||
 209         (newhead & 0x0000fc00) == 0x0000fc00) return FALSE;
 210
 211     fr->lay = 4-((newhead>>17)&3);
 212 //    if(fr->lay!=3) return FALSE;
 213
 214     if( newhead & (1<<20) ) {
 215       fr->lsf = (newhead & (1<<19)) ? 0x0 : 0x1;
 216       fr->mpeg25 = 0;
 217     } else {
 218       fr->lsf = 1;
 219       fr->mpeg25 = 1;
 220     }
 221
 222     if(fr->mpeg25)
 223       fr->sampling_frequency = 6 + ((newhead>>10)&0x3);
 224     else
 225       fr->sampling_frequency = ((newhead>>10)&0x3) + (fr->lsf*3);
 226
 227     if(fr->sampling_frequency>8) return FALSE;  // valid: 0..8
 228
 229     fr->error_protection = ((newhead>>16)&0x1)^0x1;
 230     fr->bitrate_index = ((newhead>>12)&0xf);
 231     fr->padding   = ((newhead>>9)&0x1);
 232     fr->extension = ((newhead>>8)&0x1);
 233     fr->mode      = ((newhead>>6)&0x3);
 234     fr->mode_ext  = ((newhead>>4)&0x3);
 235     fr->copyright = ((newhead>>3)&0x1);
 236     fr->original  = ((newhead>>2)&0x1);
 237     fr->emphasis  = newhead & 0x3;
 238
 239     MP3_channels = fr->stereo    = (fr->mode == MPG_MD_MONO) ? 1 : 2;
 240
 241     if(!fr->bitrate_index){
 242 //      fprintf(stderr,"Free format not supported.\n");
 243       return FALSE;
 244     }
 245
 246 switch(fr->lay){
 247   case 2:
 248     MP3_bitrate=tabsel_123[fr->lsf][1][fr->bitrate_index];
 249     MP3_samplerate=freqs[fr->sampling_frequency];
 250     fr->framesize = MP3_bitrate * 144000;
 251     fr->framesize /= MP3_samplerate;
 252     MP3_framesize=fr->framesize;
 253     fr->framesize += fr->padding - 4;
 254     break;
 255   case 3:
 256     if(fr->lsf)
 257       ssize = (fr->stereo == 1) ? 9 : 17;
 258     else
 259       ssize = (fr->stereo == 1) ? 17 : 32;
 260     if(fr->error_protection) ssize += 2;
 261
 262     MP3_bitrate=tabsel_123[fr->lsf][2][fr->bitrate_index];
 263     MP3_samplerate=freqs[fr->sampling_frequency];
 264     fr->framesize  = MP3_bitrate * 144000;
 265     fr->framesize /= MP3_samplerate<<(fr->lsf);
 266     MP3_framesize=fr->framesize;
 267     fr->framesize += fr->padding - 4;
 268     break;
 269   case 1:
 270 //    fr->jsbound = (fr->mode == MPG_MD_JOINT_STEREO) ? (fr->mode_ext<<2)+4 : 32;
 271     MP3_bitrate=tabsel_123[fr->lsf][0][fr->bitrate_index];
 272     MP3_samplerate=freqs[fr->sampling_frequency];
 273     fr->framesize  = MP3_bitrate * 12000;
 274     fr->framesize /= MP3_samplerate;
 275     MP3_framesize  = ((fr->framesize+fr->padding)<<2);
 276     fr->framesize  = MP3_framesize-4;
 277 //    printf("framesize=%d\n",fr->framesize);
 278     break;
 279   default:
 280     MP3_framesize=fr->framesize=0;
 281 //    fprintf(stderr,"Sorry, unsupported layer type.\n");
 282     return 0;
 283 }
 284     if(fr->framesize<=0 || fr->framesize>MAXFRAMESIZE) return FALSE;
 285
 286     return 1;
 287 }
 288
 289
 290 LOCAL int stream_read_frame_body(int size){
 291
 292   /* flip/init buffer for Layer 3 */
 293   bsbufold = bsbuf;
 294   bsbuf = bsspace[bsnum]+512;
 295   bsnum = (bsnum + 1) & 1;
 296
 297   if( mp3_read(bsbuf,size) != size) return 0; // broken frame
 298
 299   bitindex = 0;
 300   wordpointer = (unsigned char *) bsbuf;
 301   bitsleft=8*size;
 302
 303   return 1;
 304 }
 305
 306
 307 /*****************************************************************
 308  * read next frame     return number of frames read.
 309  */
 310 LOCAL int read_frame(struct frame *fr){
 311   uint32_t newhead;
 312   union {
 313     unsigned char buf[8];
 314     unsigned long dummy; // for alignment
 315   } hbuf;
 316   int skipped,resyncpos;
 317   int frames=0;
 318
 319 resync:
 320   skipped=MP3_fpos;
 321   resyncpos=MP3_fpos;
 322
 323   set_pointer(512);
 324   fsizeold=fr->framesize;       /* for Layer3 */
 325   if(!stream_head_read(hbuf.buf,&newhead)) return 0;
 326   if(!decode_header(fr,newhead)){
 327     // invalid header! try to resync stream!
 328 #ifdef DEBUG_RESYNC
 329     printf("ReSync: searching for a valid header...  (pos=%X)\n",MP3_fpos);
 330 #endif
 331 retry1:
 332     while(!decode_header(fr,newhead)){
 333       if(!stream_head_shift(hbuf.buf,&newhead)) return 0;
 334     }
 335     resyncpos=MP3_fpos-4;
 336     // found valid header
 337 #ifdef DEBUG_RESYNC
 338     printf("ReSync: found valid hdr at %X  fsize=%ld  ",resyncpos,fr->framesize);
 339 #endif
 340     if(!stream_read_frame_body(fr->framesize)) return 0; // read body
 341     set_pointer(512);
 342     fsizeold=fr->framesize;       /* for Layer3 */
 343     if(!stream_head_read(hbuf.buf,&newhead)) return 0;
 344     if(!decode_header(fr,newhead)){
 345       // invalid hdr! go back...
 346 #ifdef DEBUG_RESYNC
 347       printf("INVALID\n");
 348 #endif
 349 //      mp3_seek(resyncpos+1);
 350       if(!stream_head_read(hbuf.buf,&newhead)) return 0;
 351       goto retry1;
 352     }
 353 #ifdef DEBUG_RESYNC
 354     printf("OK!\n");
 355     ++frames;
 356 #endif
 357   }
 358
 359   skipped=resyncpos-skipped;
 360 //  if(skipped && !MP3_resync) printf("\r%d bad bytes skipped  (resync at 0x%X)                          \n",skipped,resyncpos);
 361
 362 //  printf("%8X [%08X] %d %d (%d)%s%s\n",MP3_fpos-4,newhead,fr->framesize,fr->mode,fr->mode_ext,fr->error_protection?" CRC":"",fr->padding?" PAD":"");
 363
 364   /* read main data into memory */
 365   if(!stream_read_frame_body(fr->framesize)){
 366     printf("\nBroken frame at 0x%X                                                  \n",resyncpos);
 367     return 0;
 368   }
 369   ++frames;
 370
 371   if(MP3_resync){
 372     MP3_resync=0;
 373     if(frames==1) goto resync;
 374   }
 375
 376   return frames;
 377 }
 378
 379 static int _has_mmx = 0;  // used by layer2.c, layer3.c to pre-scale coeffs
 380
 381 /******************************************************************************/
 382 /*           PUBLIC FUNCTIONS                  */
 383 /******************************************************************************/
 384
 385 void (*dct64_MMX_func)(short *, short *, real *);
 386
 387 #include "layer2.c"
 388 #include "layer3.c"
 389 #include "layer1.c"
 390
 391 #include "cpudetect.h"
 392
 393 // Init decoder tables.  Call first, once!
 394 #ifdef CONFIG_FAKE_MONO
 395 void MP3_Init(int fakemono){
 396 #else
 397 void MP3_Init(void){
 398 #endif
 399
 400 //gCpuCaps.hasMMX=gCpuCaps.hasMMX2=gCpuCaps.hasSSE=0; // for testing!
 401
 402     _has_mmx = 0;
 403     dct36_func = dct36;
 404
 405     make_decode_tables(outscale);
 406
 407 #if HAVE_MMX
 408     if (gCpuCaps.hasMMX)
 409     {
 410         _has_mmx = 1;
 411         synth_func = synth_1to1_MMX;
 412     }
 413 #endif
 414
 415 #if HAVE_AMD3DNOWEXT
 416     if (gCpuCaps.has3DNowExt)
 417     {
 418         dct36_func=dct36_3dnowex;
 419         dct64_MMX_func= dct64_MMX_3dnowex;
 420         mp_msg(MSGT_DECAUDIO,MSGL_V,"mp3lib: using 3DNow!Ex optimized decore!\n");
 421     }
 422     else
 423 #endif
 424 #if HAVE_AMD3DNOW
 425     if (gCpuCaps.has3DNow)
 426     {
 427         dct36_func = dct36_3dnow;
 428         dct64_MMX_func = dct64_MMX_3dnow;
 429         mp_msg(MSGT_DECAUDIO,MSGL_V,"mp3lib: using 3DNow! optimized decore!\n");
 430     }
 431     else
 432 #endif
 433 #if HAVE_SSE
 434     if (gCpuCaps.hasSSE)
 435     {
 436         dct64_MMX_func = dct64_sse;
 437         mp_msg(MSGT_DECAUDIO,MSGL_V,"mp3lib: using SSE optimized decore!\n");
 438     }
 439     else
 440 #endif
 441 #if ARCH_X86_32
 442 #if HAVE_MMX
 443     if (gCpuCaps.hasMMX)
 444     {
 445         dct64_MMX_func = dct64_MMX;
 446         mp_msg(MSGT_DECAUDIO,MSGL_V,"mp3lib: using MMX optimized decore!\n");
 447     }
 448     else
 449 #endif
 450     if (gCpuCaps.cpuType >= CPUTYPE_I586)
 451     {
 452         synth_func = synth_1to1_pent;
 453         mp_msg(MSGT_DECAUDIO,MSGL_V,"mp3lib: using Pentium optimized decore!\n");
 454     }
 455     else
 456 #endif /* ARCH_X86_32 */
 457 #if HAVE_ALTIVEC
 458     if (gCpuCaps.hasAltiVec)
 459     {
 460         mp_msg(MSGT_DECAUDIO,MSGL_V,"mp3lib: using AltiVec optimized decore!\n");
 461     }
 462     else
 463 #endif
 464     {
 465         synth_func = NULL; /* use default c version */
 466         mp_msg(MSGT_DECAUDIO,MSGL_V,"mp3lib: using generic C decore!\n");
 467     }
 468
 469 #ifdef CONFIG_FAKE_MONO
 470     if (fakemono == 1)
 471         fr.synth=synth_1to1_l;
 472     else if (fakemono == 2)
 473         fr.synth=synth_1to1_r;
 474     else
 475         fr.synth=synth_1to1;
 476 #else
 477     fr.synth=synth_1to1;
 478 #endif
 479     fr.synth_mono=synth_1to1_mono2stereo;
 480     fr.down_sample=0;
 481     fr.down_sample_sblimit = SBLIMIT>>(fr.down_sample);
 482
 483     init_layer2();
 484     init_layer3(fr.down_sample_sblimit);
 485     mp_msg(MSGT_DECAUDIO,MSGL_V,"MP3lib: init layer2&3 finished, tables done\n");
 486 }
 487
 488 #if 0
 489
 490 void MP3_Close(void){
 491   MP3_eof=1;
 492   if(mp3_file) fclose(mp3_file);
 493   mp3_file=NULL;
 494 }
 495
 496 // Open a file, init buffers. Call once per file!
 497 int MP3_Open(char *filename,int buffsize){
 498   MP3_eof=1;   // lock decoding
 499   MP3_pause=1; // lock playing
 500   if(mp3_file) MP3_Close(); // close prev. file
 501   MP3_frames=0;
 502
 503   mp3_file=fopen(filename,"rb");
 504 //  printf("MP3_Open: file='%s'",filename);
 505 //  if(!mp3_file){ printf(" not found!\n"); return 0;} else printf("Ok!\n");
 506   if(!mp3_file) return 0;
 507
 508   MP3_filesize=MP3_PrintTAG();
 509   fseek(mp3_file,0,SEEK_SET);
 510
 511   MP3_InitBuffers(buffsize);
 512   if(!tables_done_flag) MP3_Init();
 513   MP3_eof=0;  // allow decoding
 514   MP3_pause=0; // allow playing
 515   return MP3_filesize;
 516 }
 517
 518 #endif
 519
 520 // Read & decode a single frame. Called by sound driver.
 521 int MP3_DecodeFrame(unsigned char *hova,short single){
 522    pcm_sample = hova;
 523    pcm_point = 0;
 524    if(!read_frame(&fr)) return 0;
 525    if(single==-2){ set_pointer(512); return 1; }
 526    if(fr.error_protection) getbits(16); /* skip crc */
 527    fr.single=single;
 528    switch(fr.lay){
 529      case 2: do_layer2(&fr,single);break;
 530      case 3: do_layer3(&fr,single);break;
 531      case 1: do_layer1(&fr,single);break;
 532      default:
 533          return 0;      // unsupported
 534    }
 535 //   ++MP3_frames;
 536    return pcm_point ? pcm_point : 2;
 537 }
 538
 539 // Prints last frame header in ascii.
 540 void MP3_PrintHeader(void){
 541         static char *modes[4] = { "Stereo", "Joint-Stereo", "Dual-Channel", "Single-Channel" };
 542         static char *layers[4] = { "???" , "I", "II", "III" };
 543
 544         mp_msg(MSGT_DECAUDIO,MSGL_V,"\rMPEG %s, Layer %s, %d Hz %d kbit %s, BPF: %d\n",
 545                 fr.mpeg25 ? "2.5" : (fr.lsf ? "2.0" : "1.0"),
 546                 layers[fr.lay],freqs[fr.sampling_frequency],
 547     tabsel_123[fr.lsf][fr.lay-1][fr.bitrate_index],
 548                 modes[fr.mode],fr.framesize+4);
 549         mp_msg(MSGT_DECAUDIO,MSGL_V,"Channels: %d, copyright: %s, original: %s, CRC: %s, emphasis: %d\n",
 550                 fr.stereo,fr.copyright?"Yes":"No",
 551                 fr.original?"Yes":"No",fr.error_protection?"Yes":"No",
 552                 fr.emphasis);
 553 }
 554
 555 #if 0
 556 #include "genre.h"
 557
 558 // Read & print ID3 TAG. Do not call when playing!!!  returns filesize.
 559 int MP3_PrintTAG(void){
 560         struct id3tag {
 561                 char tag[3];
 562                 char title[30];
 563                 char artist[30];
 564                 char album[30];
 565                 char year[4];
 566                 char comment[30];
 567                 unsigned char genre;
 568         };
 569         struct id3tag tag;
 570         char title[31]={0,};
 571         char artist[31]={0,};
 572         char album[31]={0,};
 573         char year[5]={0,};
 574         char comment[31]={0,};
 575         char genre[31]={0,};
 576   int fsize;
 577   int ret;
 578
 579   fseek(mp3_file,0,SEEK_END);
 580   fsize=ftell(mp3_file);
 581   if(fseek(mp3_file,-128,SEEK_END)) return fsize;
 582   ret=fread(&tag,128,1,mp3_file);
 583   if(ret!=1 || tag.tag[0]!='T' || tag.tag[1]!='A' || tag.tag[2]!='G') return fsize;
 584
 585         strncpy(title,tag.title,30);
 586         strncpy(artist,tag.artist,30);
 587         strncpy(album,tag.album,30);
 588         strncpy(year,tag.year,4);
 589         strncpy(comment,tag.comment,30);
 590
 591         if ( tag.genre <= sizeof(genre_table)/sizeof(*genre_table) ) {
 592                 strncpy(genre, genre_table[tag.genre], 30);
 593         } else {
 594                 strncpy(genre,"Unknown",30);
 595         }
 596
 597 //      printf("\n");
 598         printf("Title  : %30s  Artist: %s\n",title,artist);
 599         printf("Album  : %30s  Year  : %4s\n",album,year);
 600         printf("Comment: %30s  Genre : %s\n",comment,genre);
 601         printf("\n");
 602   return fsize-128;
 603 }
 604
 605 #endif