Chris@69: /*********************************************************************** Chris@69: Copyright (c) 2006-2011, Skype Limited. All rights reserved. Chris@69: Redistribution and use in source and binary forms, with or without Chris@69: modification, are permitted provided that the following conditions Chris@69: are met: Chris@69: - Redistributions of source code must retain the above copyright notice, Chris@69: this list of conditions and the following disclaimer. Chris@69: - Redistributions in binary form must reproduce the above copyright Chris@69: notice, this list of conditions and the following disclaimer in the Chris@69: documentation and/or other materials provided with the distribution. Chris@69: - Neither the name of Internet Society, IETF or IETF Trust, nor the Chris@69: names of specific contributors, may be used to endorse or promote Chris@69: products derived from this software without specific prior written Chris@69: permission. Chris@69: THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" Chris@69: AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE Chris@69: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE Chris@69: ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE Chris@69: LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR Chris@69: CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF Chris@69: SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS Chris@69: INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN Chris@69: CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) Chris@69: ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE Chris@69: POSSIBILITY OF SUCH DAMAGE. Chris@69: ***********************************************************************/ Chris@69: Chris@69: #ifdef HAVE_CONFIG_H Chris@69: #include "config.h" Chris@69: #endif Chris@69: Chris@69: #include "main_FIX.h" Chris@69: #include "stack_alloc.h" Chris@69: #include "tuning_parameters.h" Chris@69: Chris@69: /* Find pitch lags */ Chris@69: void silk_find_pitch_lags_FIX( Chris@69: silk_encoder_state_FIX *psEnc, /* I/O encoder state */ Chris@69: silk_encoder_control_FIX *psEncCtrl, /* I/O encoder control */ Chris@69: opus_int16 res[], /* O residual */ Chris@69: const opus_int16 x[], /* I Speech signal */ Chris@69: int arch /* I Run-time architecture */ Chris@69: ) Chris@69: { Chris@69: opus_int buf_len, i, scale; Chris@69: opus_int32 thrhld_Q13, res_nrg; Chris@69: const opus_int16 *x_ptr; Chris@69: VARDECL( opus_int16, Wsig ); Chris@69: opus_int16 *Wsig_ptr; Chris@69: opus_int32 auto_corr[ MAX_FIND_PITCH_LPC_ORDER + 1 ]; Chris@69: opus_int16 rc_Q15[ MAX_FIND_PITCH_LPC_ORDER ]; Chris@69: opus_int32 A_Q24[ MAX_FIND_PITCH_LPC_ORDER ]; Chris@69: opus_int16 A_Q12[ MAX_FIND_PITCH_LPC_ORDER ]; Chris@69: SAVE_STACK; Chris@69: Chris@69: /******************************************/ Chris@69: /* Set up buffer lengths etc based on Fs */ Chris@69: /******************************************/ Chris@69: buf_len = psEnc->sCmn.la_pitch + psEnc->sCmn.frame_length + psEnc->sCmn.ltp_mem_length; Chris@69: Chris@69: /* Safety check */ Chris@69: celt_assert( buf_len >= psEnc->sCmn.pitch_LPC_win_length ); Chris@69: Chris@69: /*************************************/ Chris@69: /* Estimate LPC AR coefficients */ Chris@69: /*************************************/ Chris@69: Chris@69: /* Calculate windowed signal */ Chris@69: Chris@69: ALLOC( Wsig, psEnc->sCmn.pitch_LPC_win_length, opus_int16 ); Chris@69: Chris@69: /* First LA_LTP samples */ Chris@69: x_ptr = x + buf_len - psEnc->sCmn.pitch_LPC_win_length; Chris@69: Wsig_ptr = Wsig; Chris@69: silk_apply_sine_window( Wsig_ptr, x_ptr, 1, psEnc->sCmn.la_pitch ); Chris@69: Chris@69: /* Middle un - windowed samples */ Chris@69: Wsig_ptr += psEnc->sCmn.la_pitch; Chris@69: x_ptr += psEnc->sCmn.la_pitch; Chris@69: silk_memcpy( Wsig_ptr, x_ptr, ( psEnc->sCmn.pitch_LPC_win_length - silk_LSHIFT( psEnc->sCmn.la_pitch, 1 ) ) * sizeof( opus_int16 ) ); Chris@69: Chris@69: /* Last LA_LTP samples */ Chris@69: Wsig_ptr += psEnc->sCmn.pitch_LPC_win_length - silk_LSHIFT( psEnc->sCmn.la_pitch, 1 ); Chris@69: x_ptr += psEnc->sCmn.pitch_LPC_win_length - silk_LSHIFT( psEnc->sCmn.la_pitch, 1 ); Chris@69: silk_apply_sine_window( Wsig_ptr, x_ptr, 2, psEnc->sCmn.la_pitch ); Chris@69: Chris@69: /* Calculate autocorrelation sequence */ Chris@69: silk_autocorr( auto_corr, &scale, Wsig, psEnc->sCmn.pitch_LPC_win_length, psEnc->sCmn.pitchEstimationLPCOrder + 1, arch ); Chris@69: Chris@69: /* Add white noise, as fraction of energy */ Chris@69: auto_corr[ 0 ] = silk_SMLAWB( auto_corr[ 0 ], auto_corr[ 0 ], SILK_FIX_CONST( FIND_PITCH_WHITE_NOISE_FRACTION, 16 ) ) + 1; Chris@69: Chris@69: /* Calculate the reflection coefficients using schur */ Chris@69: res_nrg = silk_schur( rc_Q15, auto_corr, psEnc->sCmn.pitchEstimationLPCOrder ); Chris@69: Chris@69: /* Prediction gain */ Chris@69: psEncCtrl->predGain_Q16 = silk_DIV32_varQ( auto_corr[ 0 ], silk_max_int( res_nrg, 1 ), 16 ); Chris@69: Chris@69: /* Convert reflection coefficients to prediction coefficients */ Chris@69: silk_k2a( A_Q24, rc_Q15, psEnc->sCmn.pitchEstimationLPCOrder ); Chris@69: Chris@69: /* Convert From 32 bit Q24 to 16 bit Q12 coefs */ Chris@69: for( i = 0; i < psEnc->sCmn.pitchEstimationLPCOrder; i++ ) { Chris@69: A_Q12[ i ] = (opus_int16)silk_SAT16( silk_RSHIFT( A_Q24[ i ], 12 ) ); Chris@69: } Chris@69: Chris@69: /* Do BWE */ Chris@69: silk_bwexpander( A_Q12, psEnc->sCmn.pitchEstimationLPCOrder, SILK_FIX_CONST( FIND_PITCH_BANDWIDTH_EXPANSION, 16 ) ); Chris@69: Chris@69: /*****************************************/ Chris@69: /* LPC analysis filtering */ Chris@69: /*****************************************/ Chris@69: silk_LPC_analysis_filter( res, x, A_Q12, buf_len, psEnc->sCmn.pitchEstimationLPCOrder, psEnc->sCmn.arch ); Chris@69: Chris@69: if( psEnc->sCmn.indices.signalType != TYPE_NO_VOICE_ACTIVITY && psEnc->sCmn.first_frame_after_reset == 0 ) { Chris@69: /* Threshold for pitch estimator */ Chris@69: thrhld_Q13 = SILK_FIX_CONST( 0.6, 13 ); Chris@69: thrhld_Q13 = silk_SMLABB( thrhld_Q13, SILK_FIX_CONST( -0.004, 13 ), psEnc->sCmn.pitchEstimationLPCOrder ); Chris@69: thrhld_Q13 = silk_SMLAWB( thrhld_Q13, SILK_FIX_CONST( -0.1, 21 ), psEnc->sCmn.speech_activity_Q8 ); Chris@69: thrhld_Q13 = silk_SMLABB( thrhld_Q13, SILK_FIX_CONST( -0.15, 13 ), silk_RSHIFT( psEnc->sCmn.prevSignalType, 1 ) ); Chris@69: thrhld_Q13 = silk_SMLAWB( thrhld_Q13, SILK_FIX_CONST( -0.1, 14 ), psEnc->sCmn.input_tilt_Q15 ); Chris@69: thrhld_Q13 = silk_SAT16( thrhld_Q13 ); Chris@69: Chris@69: /*****************************************/ Chris@69: /* Call pitch estimator */ Chris@69: /*****************************************/ Chris@69: if( silk_pitch_analysis_core( res, psEncCtrl->pitchL, &psEnc->sCmn.indices.lagIndex, &psEnc->sCmn.indices.contourIndex, Chris@69: &psEnc->LTPCorr_Q15, psEnc->sCmn.prevLag, psEnc->sCmn.pitchEstimationThreshold_Q16, Chris@69: (opus_int)thrhld_Q13, psEnc->sCmn.fs_kHz, psEnc->sCmn.pitchEstimationComplexity, psEnc->sCmn.nb_subfr, Chris@69: psEnc->sCmn.arch) == 0 ) Chris@69: { Chris@69: psEnc->sCmn.indices.signalType = TYPE_VOICED; Chris@69: } else { Chris@69: psEnc->sCmn.indices.signalType = TYPE_UNVOICED; Chris@69: } Chris@69: } else { Chris@69: silk_memset( psEncCtrl->pitchL, 0, sizeof( psEncCtrl->pitchL ) ); Chris@69: psEnc->sCmn.indices.lagIndex = 0; Chris@69: psEnc->sCmn.indices.contourIndex = 0; Chris@69: psEnc->LTPCorr_Q15 = 0; Chris@69: } Chris@69: RESTORE_STACK; Chris@69: }