Mercurial > libavcodec.hg
view mpegaudiodec.c @ 38:5bf15419d47e libavcodec
changed quant matrix order (should fix mmx mpeg decoding bug)
author | glantau |
---|---|
date | Tue, 07 Aug 2001 22:47:08 +0000 |
parents | 986e461dc072 |
children | 46ee548e47e4 |
line wrap: on
line source
/* * MPEG Audio decoder * Copyright (c) 2001 Gerard Lantau. * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */ #include <stdlib.h> #include <stdio.h> #include <string.h> #include "avcodec.h" #include "mpglib/mpg123.h" /* * TODO: * - add free format * - do not rely anymore on mpglib (first step: implement dct64 and decoding filter) */ #define HEADER_SIZE 4 #define BACKSTEP_SIZE 512 typedef struct MPADecodeContext { struct mpstr mpstr; UINT8 inbuf1[2][MAXFRAMESIZE + BACKSTEP_SIZE]; /* input buffer */ int inbuf_index; UINT8 *inbuf_ptr, *inbuf; int frame_size; int error_protection; int layer; int sample_rate; int bit_rate; int old_frame_size; GetBitContext gb; } MPADecodeContext; /* XXX: suppress that mess */ struct mpstr *gmp; GetBitContext *gmp_gb; static MPADecodeContext *gmp_s; /* XXX: merge constants with encoder */ static const unsigned short mp_bitrate_tab[2][3][15] = { { {0, 32, 64, 96, 128, 160, 192, 224, 256, 288, 320, 352, 384, 416, 448 }, {0, 32, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320, 384 }, {0, 32, 40, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320 } }, { {0, 32, 48, 56, 64, 80, 96, 112, 128, 144, 160, 176, 192, 224, 256}, {0, 8, 16, 24, 32, 40, 48, 56, 64, 80, 96, 112, 128, 144, 160}, {0, 8, 16, 24, 32, 40, 48, 56, 64, 80, 96, 112, 128, 144, 160} } }; static unsigned short mp_freq_tab[3] = { 44100, 48000, 32000 }; static int decode_init(AVCodecContext * avctx) { MPADecodeContext *s = avctx->priv_data; struct mpstr *mp = &s->mpstr; static int init; mp->fr.single = -1; mp->synth_bo = 1; if(!init) { init = 1; make_decode_tables(32767); init_layer2(); init_layer3(SBLIMIT); } s->inbuf_index = 0; s->inbuf = &s->inbuf1[s->inbuf_index][BACKSTEP_SIZE]; s->inbuf_ptr = s->inbuf; return 0; } /* fast header check for resync */ static int check_header(UINT32 header) { /* header */ if ((header & 0xffe00000) != 0xffe00000) return -1; /* layer check */ if (((header >> 17) & 3) == 0) return -1; /* bit rate : currently no free format supported */ if (((header >> 12) & 0xf) == 0xf || ((header >> 12) & 0xf) == 0x0) return -1; /* frequency */ if (((header >> 10) & 3) == 3) return -1; return 0; } /* header decoding. MUST check the header before because no consistency check is done there */ static void decode_header(MPADecodeContext *s, UINT32 header) { struct frame *fr = &s->mpstr.fr; int sample_rate, frame_size; if (header & (1<<20)) { fr->lsf = (header & (1<<19)) ? 0 : 1; fr->mpeg25 = 0; } else { fr->lsf = 1; fr->mpeg25 = 1; } s->layer = 4 - ((header >> 17) & 3); /* extract frequency */ fr->sampling_frequency = ((header >> 10) & 3); sample_rate = mp_freq_tab[fr->sampling_frequency] >> (fr->lsf + fr->mpeg25); fr->sampling_frequency += 3 * (fr->lsf + fr->mpeg25); s->error_protection = ((header>>16) & 1) ^ 1; fr->bitrate_index = ((header>>12)&0xf); fr->padding = ((header>>9)&0x1); fr->extension = ((header>>8)&0x1); fr->mode = ((header>>6)&0x3); fr->mode_ext = ((header>>4)&0x3); fr->copyright = ((header>>3)&0x1); fr->original = ((header>>2)&0x1); fr->emphasis = header & 0x3; fr->stereo = (fr->mode == MPG_MD_MONO) ? 1 : 2; frame_size = mp_bitrate_tab[fr->lsf][s->layer - 1][fr->bitrate_index]; s->bit_rate = frame_size * 1000; switch(s->layer) { case 1: frame_size = (frame_size * 12000) / sample_rate; frame_size = ((frame_size + fr->padding) << 2); break; case 2: frame_size = (frame_size * 144000) / sample_rate; frame_size += fr->padding; break; case 3: frame_size = (frame_size * 144000) / (sample_rate << fr->lsf); frame_size += fr->padding; break; } s->frame_size = frame_size; s->sample_rate = sample_rate; #if 0 printf("layer%d, %d Hz, %d kbits/s, %s\n", s->layer, s->sample_rate, s->bit_rate, fr->stereo ? "stereo" : "mono"); #endif } static int mp_decode_frame(MPADecodeContext *s, short *samples) { int nb_bytes; init_get_bits(&s->gb, s->inbuf + HEADER_SIZE, s->inbuf_ptr - s->inbuf - HEADER_SIZE); /* skip error protection field */ if (s->error_protection) get_bits(&s->gb, 16); /* XXX: horrible: global! */ gmp = &s->mpstr; gmp_s = s; gmp_gb = &s->gb; nb_bytes = 0; switch(s->layer) { case 1: do_layer1(&s->mpstr.fr,(unsigned char *)samples, &nb_bytes); break; case 2: do_layer2(&s->mpstr.fr,(unsigned char *)samples, &nb_bytes); break; case 3: do_layer3(&s->mpstr.fr,(unsigned char *)samples, &nb_bytes); s->inbuf_index ^= 1; s->inbuf = &s->inbuf1[s->inbuf_index][BACKSTEP_SIZE]; s->old_frame_size = s->frame_size; break; default: break; } return nb_bytes; } /* * seek back in the stream for backstep bytes (at most 511 bytes, and * at most in last frame). Note that this is slightly incorrect (data * can span more than one block!) */ int set_pointer(long backstep) { UINT8 *ptr; /* compute current position in stream */ ptr = gmp_gb->buf_ptr - (gmp_gb->bit_cnt >> 3); /* copy old data before current one */ ptr -= backstep; memcpy(ptr, gmp_s->inbuf1[gmp_s->inbuf_index ^ 1] + BACKSTEP_SIZE + gmp_s->old_frame_size - backstep, backstep); /* init get bits again */ init_get_bits(gmp_gb, ptr, gmp_s->frame_size + backstep); return 0; } static int decode_frame(AVCodecContext * avctx, void *data, int *data_size, UINT8 * buf, int buf_size) { MPADecodeContext *s = avctx->priv_data; UINT32 header; UINT8 *buf_ptr; int len, out_size; short *out_samples = data; *data_size = 0; buf_ptr = buf; while (buf_size > 0) { len = s->inbuf_ptr - s->inbuf; if (s->frame_size == 0) { /* no header seen : find one. We need at least 7 bytes to parse it */ len = HEADER_SIZE - len; if (len > buf_size) len = buf_size; memcpy(s->inbuf_ptr, buf_ptr, len); buf_ptr += len; s->inbuf_ptr += len; buf_size -= len; if ((s->inbuf_ptr - s->inbuf) == HEADER_SIZE) { header = (s->inbuf[0] << 24) | (s->inbuf[1] << 16) | (s->inbuf[2] << 8) | s->inbuf[3]; if (check_header(header) < 0) { /* no sync found : move by one byte (inefficient, but simple!) */ memcpy(s->inbuf, s->inbuf + 1, HEADER_SIZE - 1); s->inbuf_ptr--; } else { decode_header(s, header); /* update codec info */ avctx->sample_rate = s->sample_rate; avctx->channels = s->mpstr.fr.stereo ? 2 : 1; avctx->bit_rate = s->bit_rate; } } } else if (len < s->frame_size) { len = s->frame_size - len; if (len > buf_size) len = buf_size; memcpy(s->inbuf_ptr, buf_ptr, len); buf_ptr += len; s->inbuf_ptr += len; buf_size -= len; } else { out_size = mp_decode_frame(s, out_samples); s->inbuf_ptr = s->inbuf; s->frame_size = 0; *data_size = out_size; break; } } return buf_ptr - buf; } AVCodec mp3_decoder = { "mpegaudio", CODEC_TYPE_AUDIO, CODEC_ID_MP2, sizeof(MPADecodeContext), decode_init, NULL, NULL, decode_frame, };