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_FIX.h" cannam@154: #include "stack_alloc.h" cannam@154: #include "tuning_parameters.h" cannam@154: cannam@154: /* Find pitch lags */ cannam@154: void silk_find_pitch_lags_FIX( cannam@154: silk_encoder_state_FIX *psEnc, /* I/O encoder state */ cannam@154: silk_encoder_control_FIX *psEncCtrl, /* I/O encoder control */ cannam@154: opus_int16 res[], /* O residual */ cannam@154: const opus_int16 x[], /* I Speech signal */ cannam@154: int arch /* I Run-time architecture */ cannam@154: ) cannam@154: { cannam@154: opus_int buf_len, i, scale; cannam@154: opus_int32 thrhld_Q13, res_nrg; cannam@154: const opus_int16 *x_ptr; cannam@154: VARDECL( opus_int16, Wsig ); cannam@154: opus_int16 *Wsig_ptr; cannam@154: opus_int32 auto_corr[ MAX_FIND_PITCH_LPC_ORDER + 1 ]; cannam@154: opus_int16 rc_Q15[ MAX_FIND_PITCH_LPC_ORDER ]; cannam@154: opus_int32 A_Q24[ MAX_FIND_PITCH_LPC_ORDER ]; cannam@154: opus_int16 A_Q12[ MAX_FIND_PITCH_LPC_ORDER ]; cannam@154: SAVE_STACK; cannam@154: cannam@154: /******************************************/ cannam@154: /* Set up buffer lengths etc based on Fs */ cannam@154: /******************************************/ cannam@154: buf_len = psEnc->sCmn.la_pitch + psEnc->sCmn.frame_length + psEnc->sCmn.ltp_mem_length; cannam@154: cannam@154: /* Safety check */ cannam@154: celt_assert( buf_len >= psEnc->sCmn.pitch_LPC_win_length ); cannam@154: cannam@154: /*************************************/ cannam@154: /* Estimate LPC AR coefficients */ cannam@154: /*************************************/ cannam@154: cannam@154: /* Calculate windowed signal */ cannam@154: cannam@154: ALLOC( Wsig, psEnc->sCmn.pitch_LPC_win_length, opus_int16 ); cannam@154: cannam@154: /* First LA_LTP samples */ cannam@154: x_ptr = x + buf_len - psEnc->sCmn.pitch_LPC_win_length; cannam@154: Wsig_ptr = Wsig; cannam@154: silk_apply_sine_window( Wsig_ptr, x_ptr, 1, psEnc->sCmn.la_pitch ); cannam@154: cannam@154: /* Middle un - windowed samples */ cannam@154: Wsig_ptr += psEnc->sCmn.la_pitch; cannam@154: x_ptr += psEnc->sCmn.la_pitch; cannam@154: silk_memcpy( Wsig_ptr, x_ptr, ( psEnc->sCmn.pitch_LPC_win_length - silk_LSHIFT( psEnc->sCmn.la_pitch, 1 ) ) * sizeof( opus_int16 ) ); cannam@154: cannam@154: /* Last LA_LTP samples */ cannam@154: Wsig_ptr += psEnc->sCmn.pitch_LPC_win_length - silk_LSHIFT( psEnc->sCmn.la_pitch, 1 ); cannam@154: x_ptr += psEnc->sCmn.pitch_LPC_win_length - silk_LSHIFT( psEnc->sCmn.la_pitch, 1 ); cannam@154: silk_apply_sine_window( Wsig_ptr, x_ptr, 2, psEnc->sCmn.la_pitch ); cannam@154: cannam@154: /* Calculate autocorrelation sequence */ cannam@154: silk_autocorr( auto_corr, &scale, Wsig, psEnc->sCmn.pitch_LPC_win_length, psEnc->sCmn.pitchEstimationLPCOrder + 1, arch ); cannam@154: cannam@154: /* Add white noise, as fraction of energy */ cannam@154: auto_corr[ 0 ] = silk_SMLAWB( auto_corr[ 0 ], auto_corr[ 0 ], SILK_FIX_CONST( FIND_PITCH_WHITE_NOISE_FRACTION, 16 ) ) + 1; cannam@154: cannam@154: /* Calculate the reflection coefficients using schur */ cannam@154: res_nrg = silk_schur( rc_Q15, auto_corr, psEnc->sCmn.pitchEstimationLPCOrder ); cannam@154: cannam@154: /* Prediction gain */ cannam@154: psEncCtrl->predGain_Q16 = silk_DIV32_varQ( auto_corr[ 0 ], silk_max_int( res_nrg, 1 ), 16 ); cannam@154: cannam@154: /* Convert reflection coefficients to prediction coefficients */ cannam@154: silk_k2a( A_Q24, rc_Q15, psEnc->sCmn.pitchEstimationLPCOrder ); cannam@154: cannam@154: /* Convert From 32 bit Q24 to 16 bit Q12 coefs */ cannam@154: for( i = 0; i < psEnc->sCmn.pitchEstimationLPCOrder; i++ ) { cannam@154: A_Q12[ i ] = (opus_int16)silk_SAT16( silk_RSHIFT( A_Q24[ i ], 12 ) ); cannam@154: } cannam@154: cannam@154: /* Do BWE */ cannam@154: silk_bwexpander( A_Q12, psEnc->sCmn.pitchEstimationLPCOrder, SILK_FIX_CONST( FIND_PITCH_BANDWIDTH_EXPANSION, 16 ) ); cannam@154: cannam@154: /*****************************************/ cannam@154: /* LPC analysis filtering */ cannam@154: /*****************************************/ cannam@154: silk_LPC_analysis_filter( res, x, A_Q12, buf_len, psEnc->sCmn.pitchEstimationLPCOrder, psEnc->sCmn.arch ); cannam@154: cannam@154: if( psEnc->sCmn.indices.signalType != TYPE_NO_VOICE_ACTIVITY && psEnc->sCmn.first_frame_after_reset == 0 ) { cannam@154: /* Threshold for pitch estimator */ cannam@154: thrhld_Q13 = SILK_FIX_CONST( 0.6, 13 ); cannam@154: thrhld_Q13 = silk_SMLABB( thrhld_Q13, SILK_FIX_CONST( -0.004, 13 ), psEnc->sCmn.pitchEstimationLPCOrder ); cannam@154: thrhld_Q13 = silk_SMLAWB( thrhld_Q13, SILK_FIX_CONST( -0.1, 21 ), psEnc->sCmn.speech_activity_Q8 ); cannam@154: thrhld_Q13 = silk_SMLABB( thrhld_Q13, SILK_FIX_CONST( -0.15, 13 ), silk_RSHIFT( psEnc->sCmn.prevSignalType, 1 ) ); cannam@154: thrhld_Q13 = silk_SMLAWB( thrhld_Q13, SILK_FIX_CONST( -0.1, 14 ), psEnc->sCmn.input_tilt_Q15 ); cannam@154: thrhld_Q13 = silk_SAT16( thrhld_Q13 ); cannam@154: cannam@154: /*****************************************/ cannam@154: /* Call pitch estimator */ cannam@154: /*****************************************/ cannam@154: if( silk_pitch_analysis_core( res, psEncCtrl->pitchL, &psEnc->sCmn.indices.lagIndex, &psEnc->sCmn.indices.contourIndex, cannam@154: &psEnc->LTPCorr_Q15, psEnc->sCmn.prevLag, psEnc->sCmn.pitchEstimationThreshold_Q16, cannam@154: (opus_int)thrhld_Q13, psEnc->sCmn.fs_kHz, psEnc->sCmn.pitchEstimationComplexity, psEnc->sCmn.nb_subfr, cannam@154: psEnc->sCmn.arch) == 0 ) cannam@154: { cannam@154: psEnc->sCmn.indices.signalType = TYPE_VOICED; cannam@154: } else { cannam@154: psEnc->sCmn.indices.signalType = TYPE_UNVOICED; cannam@154: } cannam@154: } else { cannam@154: silk_memset( psEncCtrl->pitchL, 0, sizeof( psEncCtrl->pitchL ) ); cannam@154: psEnc->sCmn.indices.lagIndex = 0; cannam@154: psEnc->sCmn.indices.contourIndex = 0; cannam@154: psEnc->LTPCorr_Q15 = 0; cannam@154: } cannam@154: RESTORE_STACK; cannam@154: }