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