cannam@154: /*********************************************************************** cannam@154: Copyright (c) 2006-2011, Skype Limited. All rights reserved. cannam@154: Redistribution and use in source and binary forms, with or without cannam@154: modification, are permitted provided that the following conditions cannam@154: are met: cannam@154: - Redistributions of source code must retain the above copyright notice, cannam@154: this list of conditions and the following disclaimer. cannam@154: - Redistributions in binary form must reproduce the above copyright cannam@154: notice, this list of conditions and the following disclaimer in the cannam@154: documentation and/or other materials provided with the distribution. cannam@154: - Neither the name of Internet Society, IETF or IETF Trust, nor the cannam@154: names of specific contributors, may be used to endorse or promote cannam@154: products derived from this software without specific prior written cannam@154: permission. cannam@154: THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" cannam@154: AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE cannam@154: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE cannam@154: ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE cannam@154: LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR cannam@154: CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF cannam@154: SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS cannam@154: INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN cannam@154: CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) cannam@154: ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE cannam@154: POSSIBILITY OF SUCH DAMAGE. cannam@154: ***********************************************************************/ cannam@154: cannam@154: #ifdef HAVE_CONFIG_H cannam@154: #include "config.h" cannam@154: #endif cannam@154: cannam@154: #include "main.h" cannam@154: #include "stack_alloc.h" cannam@154: cannam@154: /*********************************************/ cannam@154: /* Encode quantization indices of excitation */ cannam@154: /*********************************************/ cannam@154: cannam@154: static OPUS_INLINE opus_int combine_and_check( /* return ok */ cannam@154: opus_int *pulses_comb, /* O */ cannam@154: const opus_int *pulses_in, /* I */ cannam@154: opus_int max_pulses, /* I max value for sum of pulses */ cannam@154: opus_int len /* I number of output values */ cannam@154: ) cannam@154: { cannam@154: opus_int k, sum; cannam@154: cannam@154: for( k = 0; k < len; k++ ) { cannam@154: sum = pulses_in[ 2 * k ] + pulses_in[ 2 * k + 1 ]; cannam@154: if( sum > max_pulses ) { cannam@154: return 1; cannam@154: } cannam@154: pulses_comb[ k ] = sum; cannam@154: } cannam@154: cannam@154: return 0; cannam@154: } cannam@154: cannam@154: /* Encode quantization indices of excitation */ cannam@154: void silk_encode_pulses( cannam@154: ec_enc *psRangeEnc, /* I/O compressor data structure */ cannam@154: const opus_int signalType, /* I Signal type */ cannam@154: const opus_int quantOffsetType, /* I quantOffsetType */ cannam@154: opus_int8 pulses[], /* I quantization indices */ cannam@154: const opus_int frame_length /* I Frame length */ cannam@154: ) cannam@154: { cannam@154: opus_int i, k, j, iter, bit, nLS, scale_down, RateLevelIndex = 0; cannam@154: opus_int32 abs_q, minSumBits_Q5, sumBits_Q5; cannam@154: VARDECL( opus_int, abs_pulses ); cannam@154: VARDECL( opus_int, sum_pulses ); cannam@154: VARDECL( opus_int, nRshifts ); cannam@154: opus_int pulses_comb[ 8 ]; cannam@154: opus_int *abs_pulses_ptr; cannam@154: const opus_int8 *pulses_ptr; cannam@154: const opus_uint8 *cdf_ptr; cannam@154: const opus_uint8 *nBits_ptr; cannam@154: SAVE_STACK; cannam@154: cannam@154: silk_memset( pulses_comb, 0, 8 * sizeof( opus_int ) ); /* Fixing Valgrind reported problem*/ cannam@154: cannam@154: /****************************/ cannam@154: /* Prepare for shell coding */ cannam@154: /****************************/ cannam@154: /* Calculate number of shell blocks */ cannam@154: silk_assert( 1 << LOG2_SHELL_CODEC_FRAME_LENGTH == SHELL_CODEC_FRAME_LENGTH ); cannam@154: iter = silk_RSHIFT( frame_length, LOG2_SHELL_CODEC_FRAME_LENGTH ); cannam@154: if( iter * SHELL_CODEC_FRAME_LENGTH < frame_length ) { cannam@154: celt_assert( frame_length == 12 * 10 ); /* Make sure only happens for 10 ms @ 12 kHz */ cannam@154: iter++; cannam@154: silk_memset( &pulses[ frame_length ], 0, SHELL_CODEC_FRAME_LENGTH * sizeof(opus_int8)); cannam@154: } cannam@154: cannam@154: /* Take the absolute value of the pulses */ cannam@154: ALLOC( abs_pulses, iter * SHELL_CODEC_FRAME_LENGTH, opus_int ); cannam@154: silk_assert( !( SHELL_CODEC_FRAME_LENGTH & 3 ) ); cannam@154: for( i = 0; i < iter * SHELL_CODEC_FRAME_LENGTH; i+=4 ) { cannam@154: abs_pulses[i+0] = ( opus_int )silk_abs( pulses[ i + 0 ] ); cannam@154: abs_pulses[i+1] = ( opus_int )silk_abs( pulses[ i + 1 ] ); cannam@154: abs_pulses[i+2] = ( opus_int )silk_abs( pulses[ i + 2 ] ); cannam@154: abs_pulses[i+3] = ( opus_int )silk_abs( pulses[ i + 3 ] ); cannam@154: } cannam@154: cannam@154: /* Calc sum pulses per shell code frame */ cannam@154: ALLOC( sum_pulses, iter, opus_int ); cannam@154: ALLOC( nRshifts, iter, opus_int ); cannam@154: abs_pulses_ptr = abs_pulses; cannam@154: for( i = 0; i < iter; i++ ) { cannam@154: nRshifts[ i ] = 0; cannam@154: cannam@154: while( 1 ) { cannam@154: /* 1+1 -> 2 */ cannam@154: scale_down = combine_and_check( pulses_comb, abs_pulses_ptr, silk_max_pulses_table[ 0 ], 8 ); cannam@154: /* 2+2 -> 4 */ cannam@154: scale_down += combine_and_check( pulses_comb, pulses_comb, silk_max_pulses_table[ 1 ], 4 ); cannam@154: /* 4+4 -> 8 */ cannam@154: scale_down += combine_and_check( pulses_comb, pulses_comb, silk_max_pulses_table[ 2 ], 2 ); cannam@154: /* 8+8 -> 16 */ cannam@154: scale_down += combine_and_check( &sum_pulses[ i ], pulses_comb, silk_max_pulses_table[ 3 ], 1 ); cannam@154: cannam@154: if( scale_down ) { cannam@154: /* We need to downscale the quantization signal */ cannam@154: nRshifts[ i ]++; cannam@154: for( k = 0; k < SHELL_CODEC_FRAME_LENGTH; k++ ) { cannam@154: abs_pulses_ptr[ k ] = silk_RSHIFT( abs_pulses_ptr[ k ], 1 ); cannam@154: } cannam@154: } else { cannam@154: /* Jump out of while(1) loop and go to next shell coding frame */ cannam@154: break; cannam@154: } cannam@154: } cannam@154: abs_pulses_ptr += SHELL_CODEC_FRAME_LENGTH; cannam@154: } cannam@154: cannam@154: /**************/ cannam@154: /* Rate level */ cannam@154: /**************/ cannam@154: /* find rate level that leads to fewest bits for coding of pulses per block info */ cannam@154: minSumBits_Q5 = silk_int32_MAX; cannam@154: for( k = 0; k < N_RATE_LEVELS - 1; k++ ) { cannam@154: nBits_ptr = silk_pulses_per_block_BITS_Q5[ k ]; cannam@154: sumBits_Q5 = silk_rate_levels_BITS_Q5[ signalType >> 1 ][ k ]; cannam@154: for( i = 0; i < iter; i++ ) { cannam@154: if( nRshifts[ i ] > 0 ) { cannam@154: sumBits_Q5 += nBits_ptr[ SILK_MAX_PULSES + 1 ]; cannam@154: } else { cannam@154: sumBits_Q5 += nBits_ptr[ sum_pulses[ i ] ]; cannam@154: } cannam@154: } cannam@154: if( sumBits_Q5 < minSumBits_Q5 ) { cannam@154: minSumBits_Q5 = sumBits_Q5; cannam@154: RateLevelIndex = k; cannam@154: } cannam@154: } cannam@154: ec_enc_icdf( psRangeEnc, RateLevelIndex, silk_rate_levels_iCDF[ signalType >> 1 ], 8 ); cannam@154: cannam@154: /***************************************************/ cannam@154: /* Sum-Weighted-Pulses Encoding */ cannam@154: /***************************************************/ cannam@154: cdf_ptr = silk_pulses_per_block_iCDF[ RateLevelIndex ]; cannam@154: for( i = 0; i < iter; i++ ) { cannam@154: if( nRshifts[ i ] == 0 ) { cannam@154: ec_enc_icdf( psRangeEnc, sum_pulses[ i ], cdf_ptr, 8 ); cannam@154: } else { cannam@154: ec_enc_icdf( psRangeEnc, SILK_MAX_PULSES + 1, cdf_ptr, 8 ); cannam@154: for( k = 0; k < nRshifts[ i ] - 1; k++ ) { cannam@154: ec_enc_icdf( psRangeEnc, SILK_MAX_PULSES + 1, silk_pulses_per_block_iCDF[ N_RATE_LEVELS - 1 ], 8 ); cannam@154: } cannam@154: ec_enc_icdf( psRangeEnc, sum_pulses[ i ], silk_pulses_per_block_iCDF[ N_RATE_LEVELS - 1 ], 8 ); cannam@154: } cannam@154: } cannam@154: cannam@154: /******************/ cannam@154: /* Shell Encoding */ cannam@154: /******************/ cannam@154: for( i = 0; i < iter; i++ ) { cannam@154: if( sum_pulses[ i ] > 0 ) { cannam@154: silk_shell_encoder( psRangeEnc, &abs_pulses[ i * SHELL_CODEC_FRAME_LENGTH ] ); cannam@154: } cannam@154: } cannam@154: cannam@154: /****************/ cannam@154: /* LSB Encoding */ cannam@154: /****************/ cannam@154: for( i = 0; i < iter; i++ ) { cannam@154: if( nRshifts[ i ] > 0 ) { cannam@154: pulses_ptr = &pulses[ i * SHELL_CODEC_FRAME_LENGTH ]; cannam@154: nLS = nRshifts[ i ] - 1; cannam@154: for( k = 0; k < SHELL_CODEC_FRAME_LENGTH; k++ ) { cannam@154: abs_q = (opus_int8)silk_abs( pulses_ptr[ k ] ); cannam@154: for( j = nLS; j > 0; j-- ) { cannam@154: bit = silk_RSHIFT( abs_q, j ) & 1; cannam@154: ec_enc_icdf( psRangeEnc, bit, silk_lsb_iCDF, 8 ); cannam@154: } cannam@154: bit = abs_q & 1; cannam@154: ec_enc_icdf( psRangeEnc, bit, silk_lsb_iCDF, 8 ); cannam@154: } cannam@154: } cannam@154: } cannam@154: cannam@154: /****************/ cannam@154: /* Encode signs */ cannam@154: /****************/ cannam@154: silk_encode_signs( psRangeEnc, pulses, frame_length, signalType, quantOffsetType, sum_pulses ); cannam@154: RESTORE_STACK; cannam@154: }