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