adpcm.c
上传用户:riyaled888
上传日期:2009-03-27
资源大小:7338k
文件大小:20k
- /*****************************************************************************
- * adpcm.c : adpcm variant audio decoder
- *****************************************************************************
- * Copyright (C) 2001, 2002 VideoLAN
- * $Id: adpcm.c 6961 2004-03-05 17:34:23Z sam $
- *
- * Authors: Laurent Aimar <fenrir@via.ecp.fr>
- *
- * 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., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
- *****************************************************************************/
- /*****************************************************************************
- * Preamble
- *
- * Documentation: http://www.pcisys.net/~melanson/codecs/adpcm.txt
- *****************************************************************************/
- #include <vlc/vlc.h>
- #include <vlc/decoder.h>
- /*****************************************************************************
- * Module descriptor
- *****************************************************************************/
- static int OpenDecoder( vlc_object_t * );
- static void CloseDecoder( vlc_object_t * );
- static aout_buffer_t *DecodeBlock( decoder_t *, block_t ** );
- vlc_module_begin();
- set_description( _("ADPCM audio decoder") );
- set_capability( "decoder", 50 );
- set_callbacks( OpenDecoder, CloseDecoder );
- vlc_module_end();
- /*****************************************************************************
- * Local prototypes
- *****************************************************************************/
- enum adpcm_codec_e
- {
- ADPCM_IMA_QT,
- ADPCM_IMA_WAV,
- ADPCM_MS,
- ADPCM_DK3,
- ADPCM_DK4
- };
- struct decoder_sys_t
- {
- enum adpcm_codec_e codec;
- int i_block;
- int i_samplesperblock;
- audio_date_t end_date;
- };
- static void DecodeAdpcmMs ( decoder_t *, int16_t *, uint8_t * );
- static void DecodeAdpcmImaWav( decoder_t *, int16_t *, uint8_t * );
- static void DecodeAdpcmImaQT ( decoder_t *, int16_t *, uint8_t * );
- static void DecodeAdpcmDk4 ( decoder_t *, int16_t *, uint8_t * );
- static void DecodeAdpcmDk3 ( decoder_t *, int16_t *, uint8_t * );
- static int pi_channels_maps[6] =
- {
- 0,
- AOUT_CHAN_CENTER,
- AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT,
- AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER,
- AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARLEFT,
- AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER
- | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARLEFT
- };
- /* Various table from http://www.pcisys.net/~melanson/codecs/adpcm.txt */
- static int i_index_table[16] =
- {
- -1, -1, -1, -1, 2, 4, 6, 8,
- -1, -1, -1, -1, 2, 4, 6, 8
- };
- static int i_step_table[89] =
- {
- 7, 8, 9, 10, 11, 12, 13, 14, 16, 17,
- 19, 21, 23, 25, 28, 31, 34, 37, 41, 45,
- 50, 55, 60, 66, 73, 80, 88, 97, 107, 118,
- 130, 143, 157, 173, 190, 209, 230, 253, 279, 307,
- 337, 371, 408, 449, 494, 544, 598, 658, 724, 796,
- 876, 963, 1060, 1166, 1282, 1411, 1552, 1707, 1878, 2066,
- 2272, 2499, 2749, 3024, 3327, 3660, 4026, 4428, 4871, 5358,
- 5894, 6484, 7132, 7845, 8630, 9493, 10442, 11487, 12635, 13899,
- 15289, 16818, 18500, 20350, 22385, 24623, 27086, 29794, 32767
- };
- static int i_adaptation_table[16] =
- {
- 230, 230, 230, 230, 307, 409, 512, 614,
- 768, 614, 512, 409, 307, 230, 230, 230
- };
- static int i_adaptation_coeff1[7] =
- {
- 256, 512, 0, 192, 240, 460, 392
- };
- static int i_adaptation_coeff2[7] =
- {
- 0, -256, 0, 64, 0, -208, -232
- };
- /*****************************************************************************
- * OpenDecoder: probe the decoder and return score
- *****************************************************************************/
- static int OpenDecoder( vlc_object_t *p_this )
- {
- decoder_t *p_dec = (decoder_t*)p_this;
- decoder_sys_t *p_sys;
- switch( p_dec->fmt_in.i_codec )
- {
- case VLC_FOURCC('i','m','a', '4'): /* IMA ADPCM */
- case VLC_FOURCC('m','s',0x00,0x02): /* MS ADPCM */
- case VLC_FOURCC('m','s',0x00,0x11): /* IMA ADPCM */
- case VLC_FOURCC('m','s',0x00,0x61): /* Duck DK4 ADPCM */
- case VLC_FOURCC('m','s',0x00,0x62): /* Duck DK3 ADPCM */
- break;
- default:
- return VLC_EGENERIC;
- }
- /* Allocate the memory needed to store the decoder's structure */
- if( ( p_dec->p_sys = p_sys =
- (decoder_sys_t *)malloc(sizeof(decoder_sys_t)) ) == NULL )
- {
- msg_Err( p_dec, "out of memory" );
- return VLC_EGENERIC;
- }
- if( p_dec->fmt_in.audio.i_channels <= 0 ||
- p_dec->fmt_in.audio.i_channels > 5 )
- {
- msg_Err( p_dec, "bad channels count(1-5)" );
- return VLC_EGENERIC;
- }
- if( p_dec->fmt_in.audio.i_rate <= 0 )
- {
- msg_Err( p_dec, "bad samplerate" );
- return VLC_EGENERIC;
- }
- switch( p_dec->fmt_in.i_codec )
- {
- case VLC_FOURCC('i','m','a', '4'): /* IMA ADPCM */
- p_sys->codec = ADPCM_IMA_QT;
- break;
- case VLC_FOURCC('m','s',0x00,0x11): /* IMA ADPCM */
- p_sys->codec = ADPCM_IMA_WAV;
- break;
- case VLC_FOURCC('m','s',0x00,0x02): /* MS ADPCM */
- p_sys->codec = ADPCM_MS;
- break;
- case VLC_FOURCC('m','s',0x00,0x61): /* Duck DK4 ADPCM */
- p_sys->codec = ADPCM_DK4;
- break;
- case VLC_FOURCC('m','s',0x00,0x62): /* Duck DK3 ADPCM */
- p_sys->codec = ADPCM_DK3;
- break;
- }
- if( p_dec->fmt_in.audio.i_blockalign <= 0 )
- {
- p_sys->i_block = (p_sys->codec == ADPCM_IMA_QT) ?
- 34 * p_dec->fmt_in.audio.i_channels : 1024;
- msg_Warn( p_dec, "block size undefined, using %d", p_sys->i_block );
- }
- else
- {
- p_sys->i_block = p_dec->fmt_in.audio.i_blockalign;
- }
- /* calculate samples per block */
- switch( p_sys->codec )
- {
- case ADPCM_IMA_QT:
- p_sys->i_samplesperblock = 64;
- break;
- case ADPCM_IMA_WAV:
- p_sys->i_samplesperblock =
- 2 * ( p_sys->i_block - 4 * p_dec->fmt_in.audio.i_channels ) /
- p_dec->fmt_in.audio.i_channels;
- break;
- case ADPCM_MS:
- p_sys->i_samplesperblock =
- 2 * (p_sys->i_block - 7 * p_dec->fmt_in.audio.i_channels) /
- p_dec->fmt_in.audio.i_channels + 2;
- break;
- case ADPCM_DK4:
- p_sys->i_samplesperblock =
- 2 * (p_sys->i_block - 4 * p_dec->fmt_in.audio.i_channels) /
- p_dec->fmt_in.audio.i_channels + 1;
- break;
- case ADPCM_DK3:
- p_dec->fmt_in.audio.i_channels = 2;
- p_sys->i_samplesperblock = ( 4 * ( p_sys->i_block - 16 ) + 2 )/ 3;
- break;
- }
- msg_Dbg( p_dec, "format: samplerate:%dHz channels:%d bits/sample:%d "
- "blockalign:%d samplesperblock:%d",
- p_dec->fmt_in.audio.i_rate, p_dec->fmt_in.audio.i_channels,
- p_dec->fmt_in.audio.i_bitspersample, p_sys->i_block,
- p_sys->i_samplesperblock );
- p_dec->fmt_out.i_codec = AOUT_FMT_S16_NE;
- p_dec->fmt_out.audio.i_rate = p_dec->fmt_in.audio.i_rate;
- p_dec->fmt_out.audio.i_channels = p_dec->fmt_in.audio.i_channels;
- p_dec->fmt_out.audio.i_physical_channels =
- p_dec->fmt_out.audio.i_original_channels =
- pi_channels_maps[p_dec->fmt_in.audio.i_channels];
- aout_DateInit( &p_sys->end_date, p_dec->fmt_out.audio.i_rate );
- aout_DateSet( &p_sys->end_date, 0 );
- p_dec->pf_decode_audio = DecodeBlock;
- return VLC_SUCCESS;
- }
- /*****************************************************************************
- * DecodeBlock:
- *****************************************************************************/
- static aout_buffer_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
- {
- decoder_sys_t *p_sys = p_dec->p_sys;
- block_t *p_block;
- if( !pp_block || !*pp_block ) return NULL;
- p_block = *pp_block;
- if( p_block->i_pts != 0 &&
- p_block->i_pts != aout_DateGet( &p_sys->end_date ) )
- {
- aout_DateSet( &p_sys->end_date, p_block->i_pts );
- }
- else if( !aout_DateGet( &p_sys->end_date ) )
- {
- /* We've just started the stream, wait for the first PTS. */
- block_Release( p_block );
- return NULL;
- }
- /* Don't re-use the same pts twice */
- p_block->i_pts = 0;
- if( p_block->i_buffer >= p_sys->i_block )
- {
- aout_buffer_t *p_out;
- p_out = p_dec->pf_aout_buffer_new( p_dec, p_sys->i_samplesperblock );
- if( p_out == NULL )
- {
- block_Release( p_block );
- return NULL;
- }
- p_out->start_date = aout_DateGet( &p_sys->end_date );
- p_out->end_date =
- aout_DateIncrement( &p_sys->end_date, p_sys->i_samplesperblock );
- switch( p_sys->codec )
- {
- case ADPCM_IMA_QT:
- DecodeAdpcmImaQT( p_dec, (int16_t*)p_out->p_buffer,
- p_block->p_buffer );
- break;
- case ADPCM_IMA_WAV:
- DecodeAdpcmImaWav( p_dec, (int16_t*)p_out->p_buffer,
- p_block->p_buffer );
- break;
- case ADPCM_MS:
- DecodeAdpcmMs( p_dec, (int16_t*)p_out->p_buffer,
- p_block->p_buffer );
- break;
- case ADPCM_DK4:
- DecodeAdpcmDk4( p_dec, (int16_t*)p_out->p_buffer,
- p_block->p_buffer );
- break;
- case ADPCM_DK3:
- DecodeAdpcmDk3( p_dec, (int16_t*)p_out->p_buffer,
- p_block->p_buffer );
- break;
- default:
- break;
- }
- p_block->p_buffer += p_sys->i_block;
- p_block->i_buffer -= p_sys->i_block;
- return p_out;
- }
- block_Release( p_block );
- return NULL;
- }
- /*****************************************************************************
- * CloseDecoder:
- *****************************************************************************/
- static void CloseDecoder( vlc_object_t *p_this )
- {
- decoder_t *p_dec = (decoder_t *)p_this;
- decoder_sys_t *p_sys = p_dec->p_sys;
- free( p_sys );
- }
- /*****************************************************************************
- * Local functions
- *****************************************************************************/
- #define CLAMP( v, min, max )
- if( (v) < (min) ) (v) = (min);
- if( (v) > (max) ) (v) = (max)
- #define GetByte( v )
- (v) = *p_buffer; p_buffer++;
- #define GetWord( v )
- (v) = *p_buffer; p_buffer++;
- (v) |= ( *p_buffer ) << 8; p_buffer++;
- if( (v)&0x8000 ) (v) -= 0x010000;
- /*
- * MS
- */
- typedef struct adpcm_ms_channel_s
- {
- int i_idelta;
- int i_sample1, i_sample2;
- int i_coeff1, i_coeff2;
- } adpcm_ms_channel_t;
- static int AdpcmMsExpandNibble(adpcm_ms_channel_t *p_channel,
- int i_nibble )
- {
- int i_predictor;
- int i_snibble;
- /* expand sign */
- i_snibble = i_nibble - ( i_nibble&0x08 ? 0x10 : 0 );
- i_predictor = ( p_channel->i_sample1 * p_channel->i_coeff1 +
- p_channel->i_sample2 * p_channel->i_coeff2 ) / 256 +
- i_snibble * p_channel->i_idelta;
- CLAMP( i_predictor, -32768, 32767 );
- p_channel->i_sample2 = p_channel->i_sample1;
- p_channel->i_sample1 = i_predictor;
- p_channel->i_idelta = ( i_adaptation_table[i_nibble] *
- p_channel->i_idelta ) / 256;
- if( p_channel->i_idelta < 16 )
- {
- p_channel->i_idelta = 16;
- }
- return( i_predictor );
- }
- static void DecodeAdpcmMs( decoder_t *p_dec, int16_t *p_sample,
- uint8_t *p_buffer )
- {
- decoder_sys_t *p_sys = p_dec->p_sys;
- adpcm_ms_channel_t channel[2];
- int i_nibbles;
- int b_stereo;
- int i_block_predictor;
- b_stereo = p_dec->fmt_in.audio.i_channels == 2 ? 1 : 0;
- GetByte( i_block_predictor );
- CLAMP( i_block_predictor, 0, 6 );
- channel[0].i_coeff1 = i_adaptation_coeff1[i_block_predictor];
- channel[0].i_coeff2 = i_adaptation_coeff2[i_block_predictor];
- if( b_stereo )
- {
- GetByte( i_block_predictor );
- CLAMP( i_block_predictor, 0, 6 );
- channel[1].i_coeff1 = i_adaptation_coeff1[i_block_predictor];
- channel[1].i_coeff2 = i_adaptation_coeff2[i_block_predictor];
- }
- GetWord( channel[0].i_idelta );
- if( b_stereo )
- {
- GetWord( channel[1].i_idelta );
- }
- GetWord( channel[0].i_sample1 );
- if( b_stereo )
- {
- GetWord( channel[1].i_sample1 );
- }
- GetWord( channel[0].i_sample2 );
- if( b_stereo )
- {
- GetWord( channel[1].i_sample2 );
- }
- if( b_stereo )
- {
- *p_sample++ = channel[0].i_sample2;
- *p_sample++ = channel[1].i_sample2;
- *p_sample++ = channel[0].i_sample1;
- *p_sample++ = channel[1].i_sample1;
- }
- else
- {
- *p_sample++ = channel[0].i_sample2;
- *p_sample++ = channel[0].i_sample1;
- }
- for( i_nibbles = 2 * (p_sys->i_block - 7 * p_dec->fmt_in.audio.i_channels);
- i_nibbles > 0; i_nibbles -= 2, p_buffer++ )
- {
- *p_sample++ = AdpcmMsExpandNibble( &channel[0], (*p_buffer) >> 4);
- *p_sample++ = AdpcmMsExpandNibble( &channel[b_stereo ? 1 : 0],
- (*p_buffer)&0x0f);
- }
- }
- /*
- * IMA-WAV
- */
- typedef struct adpcm_ima_wav_channel_s
- {
- int i_predictor;
- int i_step_index;
- } adpcm_ima_wav_channel_t;
- static int AdpcmImaWavExpandNibble(adpcm_ima_wav_channel_t *p_channel,
- int i_nibble )
- {
- int i_diff;
- i_diff = i_step_table[p_channel->i_step_index] >> 3;
- if( i_nibble&0x04 ) i_diff += i_step_table[p_channel->i_step_index];
- if( i_nibble&0x02 ) i_diff += i_step_table[p_channel->i_step_index]>>1;
- if( i_nibble&0x01 ) i_diff += i_step_table[p_channel->i_step_index]>>2;
- if( i_nibble&0x08 )
- p_channel->i_predictor -= i_diff;
- else
- p_channel->i_predictor += i_diff;
- CLAMP( p_channel->i_predictor, -32768, 32767 );
- p_channel->i_step_index += i_index_table[i_nibble];
- CLAMP( p_channel->i_step_index, 0, 88 );
- return( p_channel->i_predictor );
- }
- static void DecodeAdpcmImaWav( decoder_t *p_dec, int16_t *p_sample,
- uint8_t *p_buffer )
- {
- decoder_sys_t *p_sys = p_dec->p_sys;
- adpcm_ima_wav_channel_t channel[2];
- int i_nibbles;
- int b_stereo;
- b_stereo = p_dec->fmt_in.audio.i_channels == 2 ? 1 : 0;
- GetWord( channel[0].i_predictor );
- GetByte( channel[0].i_step_index );
- CLAMP( channel[0].i_step_index, 0, 88 );
- p_buffer++;
- if( b_stereo )
- {
- GetWord( channel[1].i_predictor );
- GetByte( channel[1].i_step_index );
- CLAMP( channel[1].i_step_index, 0, 88 );
- p_buffer++;
- }
- if( b_stereo )
- {
- for( i_nibbles = 2 * (p_sys->i_block - 8);
- i_nibbles > 0;
- i_nibbles -= 16 )
- {
- int i;
- for( i = 0; i < 4; i++ )
- {
- p_sample[i * 4] =
- AdpcmImaWavExpandNibble(&channel[0],p_buffer[i]&0x0f);
- p_sample[i * 4 + 2] =
- AdpcmImaWavExpandNibble(&channel[0],p_buffer[i] >> 4);
- }
- p_buffer += 4;
- for( i = 0; i < 4; i++ )
- {
- p_sample[i * 4 + 1] =
- AdpcmImaWavExpandNibble(&channel[1],p_buffer[i]&0x0f);
- p_sample[i * 4 + 3] =
- AdpcmImaWavExpandNibble(&channel[1],p_buffer[i] >> 4);
- }
- p_buffer += 4;
- p_sample += 16;
- }
- }
- else
- {
- for( i_nibbles = 2 * (p_sys->i_block - 4);
- i_nibbles > 0;
- i_nibbles -= 2, p_buffer++ )
- {
- *p_sample++ =AdpcmImaWavExpandNibble( &channel[0], (*p_buffer)&0x0f );
- *p_sample++ =AdpcmImaWavExpandNibble( &channel[0], (*p_buffer) >> 4 );
- }
- }
- }
- /*
- * Ima4 in QT file
- */
- static void DecodeAdpcmImaQT( decoder_t *p_dec, int16_t *p_sample,
- uint8_t *p_buffer )
- {
- adpcm_ima_wav_channel_t channel[2];
- int i_nibbles;
- int i_ch;
- int i_step;
- i_step = p_dec->fmt_in.audio.i_channels;
- for( i_ch = 0; i_ch < p_dec->fmt_in.audio.i_channels; i_ch++ )
- {
- /* load preambule */
- channel[i_ch].i_predictor = (int16_t)((( ( p_buffer[0] << 1 )|( p_buffer[1] >> 7 ) ))<<7);
- channel[i_ch].i_step_index = p_buffer[1]&0x7f;
- CLAMP( channel[i_ch].i_step_index, 0, 88 );
- p_buffer += 2;
- for( i_nibbles = 0; i_nibbles < 64; i_nibbles +=2 )
- {
- *p_sample = AdpcmImaWavExpandNibble( &channel[i_ch], (*p_buffer)&0x0f);
- p_sample += i_step;
- *p_sample = AdpcmImaWavExpandNibble( &channel[i_ch], (*p_buffer >> 4)&0x0f);
- p_sample += i_step;
- p_buffer++;
- }
- /* Next channel */
- p_sample += 1 - 64 * i_step;
- }
- }
- /*
- * Dk4
- */
- static void DecodeAdpcmDk4( decoder_t *p_dec, int16_t *p_sample,
- uint8_t *p_buffer )
- {
- decoder_sys_t *p_sys = p_dec->p_sys;
- adpcm_ima_wav_channel_t channel[2];
- int i_nibbles;
- int b_stereo;
- b_stereo = p_dec->fmt_in.audio.i_channels == 2 ? 1 : 0;
- GetWord( channel[0].i_predictor );
- GetByte( channel[0].i_step_index );
- CLAMP( channel[0].i_step_index, 0, 88 );
- p_buffer++;
- if( b_stereo )
- {
- GetWord( channel[1].i_predictor );
- GetByte( channel[1].i_step_index );
- CLAMP( channel[1].i_step_index, 0, 88 );
- p_buffer++;
- }
- /* first output predictor */
- *p_sample++ = channel[0].i_predictor;
- if( b_stereo )
- {
- *p_sample++ = channel[1].i_predictor;
- }
- for( i_nibbles = 0;
- i_nibbles < p_sys->i_block - 4 * (b_stereo ? 2:1 );
- i_nibbles++ )
- {
- *p_sample++ = AdpcmImaWavExpandNibble( &channel[0],
- (*p_buffer) >> 4);
- *p_sample++ = AdpcmImaWavExpandNibble( &channel[b_stereo ? 1 : 0],
- (*p_buffer)&0x0f);
- p_buffer++;
- }
- }
- /*
- * Dk3
- */
- static void DecodeAdpcmDk3( decoder_t *p_dec, int16_t *p_sample,
- uint8_t *p_buffer )
- {
- decoder_sys_t *p_sys = p_dec->p_sys;
- uint8_t *p_end = &p_buffer[p_sys->i_block];
- adpcm_ima_wav_channel_t sum;
- adpcm_ima_wav_channel_t diff;
- int i_diff_value;
- p_buffer += 10;
- GetWord( sum.i_predictor );
- GetWord( diff.i_predictor );
- GetByte( sum.i_step_index );
- GetByte( diff.i_step_index );
- i_diff_value = diff.i_predictor;
- /* we process 6 nibbles at once */
- while( p_buffer + 1 <= p_end )
- {
- /* first 3 nibbles */
- AdpcmImaWavExpandNibble( &sum,
- (*p_buffer)&0x0f);
- AdpcmImaWavExpandNibble( &diff,
- (*p_buffer) >> 4 );
- i_diff_value = ( i_diff_value + diff.i_predictor ) / 2;
- *p_sample++ = sum.i_predictor + i_diff_value;
- *p_sample++ = sum.i_predictor - i_diff_value;
- p_buffer++;
- AdpcmImaWavExpandNibble( &sum,
- (*p_buffer)&0x0f);
- *p_sample++ = sum.i_predictor + i_diff_value;
- *p_sample++ = sum.i_predictor - i_diff_value;
- /* now last 3 nibbles */
- AdpcmImaWavExpandNibble( &sum,
- (*p_buffer)>>4);
- p_buffer++;
- if( p_buffer < p_end )
- {
- AdpcmImaWavExpandNibble( &diff,
- (*p_buffer)&0x0f );
- i_diff_value = ( i_diff_value + diff.i_predictor ) / 2;
- *p_sample++ = sum.i_predictor + i_diff_value;
- *p_sample++ = sum.i_predictor - i_diff_value;
- AdpcmImaWavExpandNibble( &sum,
- (*p_buffer)>>4);
- p_buffer++;
- *p_sample++ = sum.i_predictor + i_diff_value;
- *p_sample++ = sum.i_predictor - i_diff_value;
- }
- }
- }