Coverage Report

Created: 2026-02-26 07:37

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/src/opus/silk/NSQ.c
Line
Count
Source
1
/***********************************************************************
2
Copyright (c) 2006-2011, Skype Limited. All rights reserved.
3
Redistribution and use in source and binary forms, with or without
4
modification, are permitted provided that the following conditions
5
are met:
6
- Redistributions of source code must retain the above copyright notice,
7
this list of conditions and the following disclaimer.
8
- Redistributions in binary form must reproduce the above copyright
9
notice, this list of conditions and the following disclaimer in the
10
documentation and/or other materials provided with the distribution.
11
- Neither the name of Internet Society, IETF or IETF Trust, nor the
12
names of specific contributors, may be used to endorse or promote
13
products derived from this software without specific prior written
14
permission.
15
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
16
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
19
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
20
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
21
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
22
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
23
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
24
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
25
POSSIBILITY OF SUCH DAMAGE.
26
***********************************************************************/
27
28
#ifdef HAVE_CONFIG_H
29
#include "config.h"
30
#endif
31
32
#include "main.h"
33
#include "stack_alloc.h"
34
#include "NSQ.h"
35
36
37
static OPUS_INLINE void silk_nsq_scale_states(
38
    const silk_encoder_state *psEncC,           /* I    Encoder State                   */
39
    silk_nsq_state      *NSQ,                   /* I/O  NSQ state                       */
40
    const opus_int16    x16[],                  /* I    input                           */
41
    opus_int32          x_sc_Q10[],             /* O    input scaled with 1/Gain        */
42
    const opus_int16    sLTP[],                 /* I    re-whitened LTP state in Q0     */
43
    opus_int32          sLTP_Q15[],             /* O    LTP state matching scaled input */
44
    opus_int            subfr,                  /* I    subframe number                 */
45
    const opus_int      LTP_scale_Q14,          /* I                                    */
46
    const opus_int32    Gains_Q16[ MAX_NB_SUBFR ], /* I                                 */
47
    const opus_int      pitchL[ MAX_NB_SUBFR ], /* I    Pitch lag                       */
48
    const opus_int      signal_type             /* I    Signal type                     */
49
);
50
51
#if !defined(OPUS_X86_MAY_HAVE_SSE4_1)
52
static OPUS_INLINE void silk_noise_shape_quantizer(
53
    silk_nsq_state      *NSQ,                   /* I/O  NSQ state                       */
54
    opus_int            signalType,             /* I    Signal type                     */
55
    const opus_int32    x_sc_Q10[],             /* I                                    */
56
    opus_int8           pulses[],               /* O                                    */
57
    opus_int16          xq[],                   /* O                                    */
58
    opus_int32          sLTP_Q15[],             /* I/O  LTP state                       */
59
    const opus_int16    a_Q12[],                /* I    Short term prediction coefs     */
60
    const opus_int16    b_Q14[],                /* I    Long term prediction coefs      */
61
    const opus_int16    AR_shp_Q13[],           /* I    Noise shaping AR coefs          */
62
    opus_int            lag,                    /* I    Pitch lag                       */
63
    opus_int32          HarmShapeFIRPacked_Q14, /* I                                    */
64
    opus_int            Tilt_Q14,               /* I    Spectral tilt                   */
65
    opus_int32          LF_shp_Q14,             /* I                                    */
66
    opus_int32          Gain_Q16,               /* I                                    */
67
    opus_int            Lambda_Q10,             /* I                                    */
68
    opus_int            offset_Q10,             /* I                                    */
69
    opus_int            length,                 /* I    Input length                    */
70
    opus_int            shapingLPCOrder,        /* I    Noise shaping AR filter order   */
71
    opus_int            predictLPCOrder,        /* I    Prediction filter order         */
72
    int                 arch                    /* I    Architecture                    */
73
);
74
#endif
75
76
void silk_NSQ_c
77
(
78
    const silk_encoder_state    *psEncC,                                      /* I    Encoder State                   */
79
    silk_nsq_state              *NSQ,                                         /* I/O  NSQ state                       */
80
    SideInfoIndices             *psIndices,                                   /* I/O  Quantization Indices            */
81
    const opus_int16            x16[],                                        /* I    Input                           */
82
    opus_int8                   pulses[],                                     /* O    Quantized pulse signal          */
83
    const opus_int16            *PredCoef_Q12,                                /* I    Short term prediction coefs     */
84
    const opus_int16            LTPCoef_Q14[ LTP_ORDER * MAX_NB_SUBFR ],      /* I    Long term prediction coefs      */
85
    const opus_int16            AR_Q13[ MAX_NB_SUBFR * MAX_SHAPE_LPC_ORDER ], /* I    Noise shaping coefs             */
86
    const opus_int              HarmShapeGain_Q14[ MAX_NB_SUBFR ],            /* I    Long term shaping coefs         */
87
    const opus_int              Tilt_Q14[ MAX_NB_SUBFR ],                     /* I    Spectral tilt                   */
88
    const opus_int32            LF_shp_Q14[ MAX_NB_SUBFR ],                   /* I    Low frequency shaping coefs     */
89
    const opus_int32            Gains_Q16[ MAX_NB_SUBFR ],                    /* I    Quantization step sizes         */
90
    const opus_int              pitchL[ MAX_NB_SUBFR ],                       /* I    Pitch lags                      */
91
    const opus_int              Lambda_Q10,                                   /* I    Rate/distortion tradeoff        */
92
    const opus_int              LTP_scale_Q14                                 /* I    LTP state scaling               */
93
)
94
22.3M
{
95
22.3M
    opus_int            k, lag, start_idx, LSF_interpolation_flag;
96
22.3M
    const opus_int16    *A_Q12, *B_Q14, *AR_shp_Q13;
97
22.3M
    opus_int16          *pxq;
98
22.3M
    VARDECL( opus_int32, sLTP_Q15 );
99
22.3M
    VARDECL( opus_int16, sLTP );
100
22.3M
    opus_int32          HarmShapeFIRPacked_Q14;
101
22.3M
    opus_int            offset_Q10;
102
22.3M
    VARDECL( opus_int32, x_sc_Q10 );
103
22.3M
    SAVE_STACK;
104
105
22.3M
    NSQ->rand_seed = psIndices->Seed;
106
107
    /* Set unvoiced lag to the previous one, overwrite later for voiced */
108
22.3M
    lag = NSQ->lagPrev;
109
110
22.3M
    silk_assert( NSQ->prev_gain_Q16 != 0 );
111
112
22.3M
    offset_Q10 = silk_Quantization_Offsets_Q10[ psIndices->signalType >> 1 ][ psIndices->quantOffsetType ];
113
114
22.3M
    if( psIndices->NLSFInterpCoef_Q2 == 4 ) {
115
22.3M
        LSF_interpolation_flag = 0;
116
22.3M
    } else {
117
0
        LSF_interpolation_flag = 1;
118
0
    }
119
120
22.3M
    ALLOC( sLTP_Q15, psEncC->ltp_mem_length + psEncC->frame_length, opus_int32 );
121
22.3M
    ALLOC( sLTP, psEncC->ltp_mem_length + psEncC->frame_length, opus_int16 );
122
22.3M
    ALLOC( x_sc_Q10, psEncC->subfr_length, opus_int32 );
123
    /* Set up pointers to start of sub frame */
124
22.3M
    NSQ->sLTP_shp_buf_idx = psEncC->ltp_mem_length;
125
22.3M
    NSQ->sLTP_buf_idx     = psEncC->ltp_mem_length;
126
22.3M
    pxq                   = &NSQ->xq[ psEncC->ltp_mem_length ];
127
95.2M
    for( k = 0; k < psEncC->nb_subfr; k++ ) {
128
72.8M
        A_Q12      = &PredCoef_Q12[ (( k >> 1 ) | ( 1 - LSF_interpolation_flag )) * MAX_LPC_ORDER ];
129
72.8M
        B_Q14      = &LTPCoef_Q14[ k * LTP_ORDER ];
130
72.8M
        AR_shp_Q13 = &AR_Q13[ k * MAX_SHAPE_LPC_ORDER ];
131
132
        /* Noise shape parameters */
133
72.8M
        silk_assert( HarmShapeGain_Q14[ k ] >= 0 );
134
72.8M
        HarmShapeFIRPacked_Q14  =                          silk_RSHIFT( HarmShapeGain_Q14[ k ], 2 );
135
72.8M
        HarmShapeFIRPacked_Q14 |= silk_LSHIFT( (opus_int32)silk_RSHIFT( HarmShapeGain_Q14[ k ], 1 ), 16 );
136
137
72.8M
        NSQ->rewhite_flag = 0;
138
72.8M
        if( psIndices->signalType == TYPE_VOICED ) {
139
            /* Voiced */
140
1.96M
            lag = pitchL[ k ];
141
142
            /* Re-whitening */
143
1.96M
            if( ( k & ( 3 - silk_LSHIFT( LSF_interpolation_flag, 1 ) ) ) == 0 ) {
144
                /* Rewhiten with new A coefs */
145
547k
                start_idx = psEncC->ltp_mem_length - lag - psEncC->predictLPCOrder - LTP_ORDER / 2;
146
547k
                celt_assert( start_idx > 0 );
147
148
547k
                silk_LPC_analysis_filter( &sLTP[ start_idx ], &NSQ->xq[ start_idx + k * psEncC->subfr_length ],
149
547k
                    A_Q12, psEncC->ltp_mem_length - start_idx, psEncC->predictLPCOrder, psEncC->arch );
150
151
547k
                NSQ->rewhite_flag = 1;
152
547k
                NSQ->sLTP_buf_idx = psEncC->ltp_mem_length;
153
547k
            }
154
1.96M
        }
155
156
72.8M
        silk_nsq_scale_states( psEncC, NSQ, x16, x_sc_Q10, sLTP, sLTP_Q15, k, LTP_scale_Q14, Gains_Q16, pitchL, psIndices->signalType );
157
158
72.8M
        silk_noise_shape_quantizer( NSQ, psIndices->signalType, x_sc_Q10, pulses, pxq, sLTP_Q15, A_Q12, B_Q14,
159
72.8M
            AR_shp_Q13, lag, HarmShapeFIRPacked_Q14, Tilt_Q14[ k ], LF_shp_Q14[ k ], Gains_Q16[ k ], Lambda_Q10,
160
72.8M
            offset_Q10, psEncC->subfr_length, psEncC->shapingLPCOrder, psEncC->predictLPCOrder, psEncC->arch );
161
162
72.8M
        x16    += psEncC->subfr_length;
163
72.8M
        pulses += psEncC->subfr_length;
164
72.8M
        pxq    += psEncC->subfr_length;
165
72.8M
    }
166
167
    /* Update lagPrev for next frame */
168
22.3M
    NSQ->lagPrev = pitchL[ psEncC->nb_subfr - 1 ];
169
170
    /* Save quantized speech and noise shaping signals */
171
22.3M
    silk_memmove( NSQ->xq,           &NSQ->xq[           psEncC->frame_length ], psEncC->ltp_mem_length * sizeof( opus_int16 ) );
172
22.3M
    silk_memmove( NSQ->sLTP_shp_Q14, &NSQ->sLTP_shp_Q14[ psEncC->frame_length ], psEncC->ltp_mem_length * sizeof( opus_int32 ) );
173
22.3M
    RESTORE_STACK;
174
22.3M
}
175
176
/******************************/
177
/* silk_noise_shape_quantizer */
178
/******************************/
179
180
#if !defined(OPUS_X86_MAY_HAVE_SSE4_1)
181
static OPUS_INLINE
182
#endif
183
void silk_noise_shape_quantizer(
184
    silk_nsq_state      *NSQ,                   /* I/O  NSQ state                       */
185
    opus_int            signalType,             /* I    Signal type                     */
186
    const opus_int32    x_sc_Q10[],             /* I                                    */
187
    opus_int8           pulses[],               /* O                                    */
188
    opus_int16          xq[],                   /* O                                    */
189
    opus_int32          sLTP_Q15[],             /* I/O  LTP state                       */
190
    const opus_int16    a_Q12[],                /* I    Short term prediction coefs     */
191
    const opus_int16    b_Q14[],                /* I    Long term prediction coefs      */
192
    const opus_int16    AR_shp_Q13[],           /* I    Noise shaping AR coefs          */
193
    opus_int            lag,                    /* I    Pitch lag                       */
194
    opus_int32          HarmShapeFIRPacked_Q14, /* I                                    */
195
    opus_int            Tilt_Q14,               /* I    Spectral tilt                   */
196
    opus_int32          LF_shp_Q14,             /* I                                    */
197
    opus_int32          Gain_Q16,               /* I                                    */
198
    opus_int            Lambda_Q10,             /* I                                    */
199
    opus_int            offset_Q10,             /* I                                    */
200
    opus_int            length,                 /* I    Input length                    */
201
    opus_int            shapingLPCOrder,        /* I    Noise shaping AR filter order   */
202
    opus_int            predictLPCOrder,        /* I    Prediction filter order         */
203
    int                 arch                    /* I    Architecture                    */
204
)
205
226M
{
206
226M
    opus_int     i;
207
226M
    opus_int32   LTP_pred_Q13, LPC_pred_Q10, n_AR_Q12, n_LTP_Q13;
208
226M
    opus_int32   n_LF_Q12, r_Q10, rr_Q10, q1_Q0, q1_Q10, q2_Q10, rd1_Q20, rd2_Q20;
209
226M
    opus_int32   exc_Q14, LPC_exc_Q14, xq_Q14, Gain_Q10;
210
226M
    opus_int32   tmp1, tmp2, sLF_AR_shp_Q14;
211
226M
    opus_int32   *psLPC_Q14, *shp_lag_ptr, *pred_lag_ptr;
212
#ifdef silk_short_prediction_create_arch_coef
213
    opus_int32   a_Q12_arch[MAX_LPC_ORDER];
214
#endif
215
216
226M
    shp_lag_ptr  = &NSQ->sLTP_shp_Q14[ NSQ->sLTP_shp_buf_idx - lag + HARM_SHAPE_FIR_TAPS / 2 ];
217
226M
    pred_lag_ptr = &sLTP_Q15[ NSQ->sLTP_buf_idx - lag + LTP_ORDER / 2 ];
218
226M
    Gain_Q10     = silk_RSHIFT( Gain_Q16, 6 );
219
220
    /* Set up short term AR state */
221
226M
    psLPC_Q14 = &NSQ->sLPC_Q14[ NSQ_LPC_BUF_LENGTH - 1 ];
222
223
#ifdef silk_short_prediction_create_arch_coef
224
    silk_short_prediction_create_arch_coef(a_Q12_arch, a_Q12, predictLPCOrder);
225
#endif
226
227
11.2G
    for( i = 0; i < length; i++ ) {
228
        /* Generate dither */
229
11.0G
        NSQ->rand_seed = silk_RAND( NSQ->rand_seed );
230
231
        /* Short-term prediction */
232
11.0G
        LPC_pred_Q10 = silk_noise_shape_quantizer_short_prediction(psLPC_Q14, a_Q12, a_Q12_arch, predictLPCOrder, arch);
233
234
        /* Long-term prediction */
235
11.0G
        if( signalType == TYPE_VOICED ) {
236
            /* Unrolled loop */
237
            /* Avoids introducing a bias because silk_SMLAWB() always rounds to -inf */
238
288M
            LTP_pred_Q13 = 2;
239
288M
            LTP_pred_Q13 = silk_SMLAWB( LTP_pred_Q13, pred_lag_ptr[  0 ], b_Q14[ 0 ] );
240
288M
            LTP_pred_Q13 = silk_SMLAWB( LTP_pred_Q13, pred_lag_ptr[ -1 ], b_Q14[ 1 ] );
241
288M
            LTP_pred_Q13 = silk_SMLAWB( LTP_pred_Q13, pred_lag_ptr[ -2 ], b_Q14[ 2 ] );
242
288M
            LTP_pred_Q13 = silk_SMLAWB( LTP_pred_Q13, pred_lag_ptr[ -3 ], b_Q14[ 3 ] );
243
288M
            LTP_pred_Q13 = silk_SMLAWB( LTP_pred_Q13, pred_lag_ptr[ -4 ], b_Q14[ 4 ] );
244
288M
            pred_lag_ptr++;
245
10.7G
        } else {
246
10.7G
            LTP_pred_Q13 = 0;
247
10.7G
        }
248
249
        /* Noise shape feedback */
250
11.0G
        celt_assert( ( shapingLPCOrder & 1 ) == 0 );   /* check that order is even */
251
11.0G
        n_AR_Q12 = silk_NSQ_noise_shape_feedback_loop(&NSQ->sDiff_shp_Q14, NSQ->sAR2_Q14, AR_shp_Q13, shapingLPCOrder, arch);
252
253
11.0G
        n_AR_Q12 = silk_SMLAWB( n_AR_Q12, NSQ->sLF_AR_shp_Q14, Tilt_Q14 );
254
255
11.0G
        n_LF_Q12 = silk_SMULWB( NSQ->sLTP_shp_Q14[ NSQ->sLTP_shp_buf_idx - 1 ], LF_shp_Q14 );
256
11.0G
        n_LF_Q12 = silk_SMLAWT( n_LF_Q12, NSQ->sLF_AR_shp_Q14, LF_shp_Q14 );
257
258
11.0G
        celt_assert( lag > 0 || signalType != TYPE_VOICED );
259
260
        /* Combine prediction and noise shaping signals */
261
11.0G
        tmp1 = silk_SUB32_ovflw( silk_LSHIFT32( LPC_pred_Q10, 2 ), n_AR_Q12 );  /* Q12 */
262
11.0G
        tmp1 = silk_SUB32_ovflw( tmp1, n_LF_Q12 );                              /* Q12 */
263
11.0G
        if( lag > 0 ) {
264
            /* Symmetric, packed FIR coefficients */
265
361M
            n_LTP_Q13 = silk_SMULWB( silk_ADD_SAT32( shp_lag_ptr[ 0 ], shp_lag_ptr[ -2 ] ), HarmShapeFIRPacked_Q14 );
266
361M
            n_LTP_Q13 = silk_SMLAWT( n_LTP_Q13, shp_lag_ptr[ -1 ],                      HarmShapeFIRPacked_Q14 );
267
361M
            n_LTP_Q13 = silk_LSHIFT( n_LTP_Q13, 1 );
268
361M
            shp_lag_ptr++;
269
270
361M
            tmp2 = silk_SUB32( LTP_pred_Q13, n_LTP_Q13 );                       /* Q13 */
271
361M
            tmp1 = silk_ADD32_ovflw( tmp2, silk_LSHIFT32( tmp1, 1 ) );          /* Q13 */
272
361M
            tmp1 = silk_RSHIFT_ROUND( tmp1, 3 );                                /* Q10 */
273
10.6G
        } else {
274
10.6G
            tmp1 = silk_RSHIFT_ROUND( tmp1, 2 );                                /* Q10 */
275
10.6G
        }
276
277
11.0G
        r_Q10 = silk_SUB32( x_sc_Q10[ i ], tmp1 );                              /* residual error Q10 */
278
279
        /* Flip sign depending on dither */
280
11.0G
        if( NSQ->rand_seed < 0 ) {
281
5.31G
            r_Q10 = -r_Q10;
282
5.31G
        }
283
11.0G
        r_Q10 = silk_LIMIT_32( r_Q10, -(31 << 10), 30 << 10 );
284
285
        /* Find two quantization level candidates and measure their rate-distortion */
286
11.0G
        q1_Q10 = silk_SUB32( r_Q10, offset_Q10 );
287
11.0G
        q1_Q0 = silk_RSHIFT( q1_Q10, 10 );
288
11.0G
        if (Lambda_Q10 > 2048) {
289
            /* For aggressive RDO, the bias becomes more than one pulse. */
290
48.1M
            int rdo_offset = Lambda_Q10/2 - 512;
291
48.1M
            if (q1_Q10 > rdo_offset) {
292
5.36M
                q1_Q0 = silk_RSHIFT( q1_Q10 - rdo_offset, 10 );
293
42.8M
            } else if (q1_Q10 < -rdo_offset) {
294
7.11M
                q1_Q0 = silk_RSHIFT( q1_Q10 + rdo_offset, 10 );
295
35.7M
            } else if (q1_Q10 < 0) {
296
20.5M
                q1_Q0 = -1;
297
20.5M
            } else {
298
15.1M
                q1_Q0 = 0;
299
15.1M
            }
300
48.1M
        }
301
11.0G
        if( q1_Q0 > 0 ) {
302
57.9M
            q1_Q10  = silk_SUB32( silk_LSHIFT( q1_Q0, 10 ), QUANT_LEVEL_ADJUST_Q10 );
303
57.9M
            q1_Q10  = silk_ADD32( q1_Q10, offset_Q10 );
304
57.9M
            q2_Q10  = silk_ADD32( q1_Q10, 1024 );
305
57.9M
            rd1_Q20 = silk_SMULBB( q1_Q10, Lambda_Q10 );
306
57.9M
            rd2_Q20 = silk_SMULBB( q2_Q10, Lambda_Q10 );
307
10.9G
        } else if( q1_Q0 == 0 ) {
308
451M
            q1_Q10  = offset_Q10;
309
451M
            q2_Q10  = silk_ADD32( q1_Q10, 1024 - QUANT_LEVEL_ADJUST_Q10 );
310
451M
            rd1_Q20 = silk_SMULBB( q1_Q10, Lambda_Q10 );
311
451M
            rd2_Q20 = silk_SMULBB( q2_Q10, Lambda_Q10 );
312
10.4G
        } else if( q1_Q0 == -1 ) {
313
10.4G
            q2_Q10  = offset_Q10;
314
10.4G
            q1_Q10  = silk_SUB32( q2_Q10, 1024 - QUANT_LEVEL_ADJUST_Q10 );
315
10.4G
            rd1_Q20 = silk_SMULBB( -q1_Q10, Lambda_Q10 );
316
10.4G
            rd2_Q20 = silk_SMULBB(  q2_Q10, Lambda_Q10 );
317
10.4G
        } else {            /* Q1_Q0 < -1 */
318
83.3M
            q1_Q10  = silk_ADD32( silk_LSHIFT( q1_Q0, 10 ), QUANT_LEVEL_ADJUST_Q10 );
319
83.3M
            q1_Q10  = silk_ADD32( q1_Q10, offset_Q10 );
320
83.3M
            q2_Q10  = silk_ADD32( q1_Q10, 1024 );
321
83.3M
            rd1_Q20 = silk_SMULBB( -q1_Q10, Lambda_Q10 );
322
83.3M
            rd2_Q20 = silk_SMULBB( -q2_Q10, Lambda_Q10 );
323
83.3M
        }
324
11.0G
        rr_Q10  = silk_SUB32( r_Q10, q1_Q10 );
325
11.0G
        rd1_Q20 = silk_SMLABB( rd1_Q20, rr_Q10, rr_Q10 );
326
11.0G
        rr_Q10  = silk_SUB32( r_Q10, q2_Q10 );
327
11.0G
        rd2_Q20 = silk_SMLABB( rd2_Q20, rr_Q10, rr_Q10 );
328
329
11.0G
        if( rd2_Q20 < rd1_Q20 ) {
330
10.4G
            q1_Q10 = q2_Q10;
331
10.4G
        }
332
333
11.0G
        pulses[ i ] = (opus_int8)silk_RSHIFT_ROUND( q1_Q10, 10 );
334
335
        /* Excitation */
336
11.0G
        exc_Q14 = silk_LSHIFT( q1_Q10, 4 );
337
11.0G
        if ( NSQ->rand_seed < 0 ) {
338
5.31G
           exc_Q14 = -exc_Q14;
339
5.31G
        }
340
341
        /* Add predictions */
342
11.0G
        LPC_exc_Q14 = silk_ADD_LSHIFT32( exc_Q14, LTP_pred_Q13, 1 );
343
11.0G
        xq_Q14      = silk_ADD32_ovflw( LPC_exc_Q14, silk_LSHIFT32( LPC_pred_Q10, 4 ) );
344
345
        /* Scale XQ back to normal level before saving */
346
11.0G
        xq[ i ] = (opus_int16)silk_SAT16( silk_RSHIFT_ROUND( silk_SMULWW( xq_Q14, Gain_Q10 ), 8 ) );
347
348
        /* Update states */
349
11.0G
        psLPC_Q14++;
350
11.0G
        *psLPC_Q14 = xq_Q14;
351
11.0G
        NSQ->sDiff_shp_Q14 = silk_SUB32_ovflw( xq_Q14, silk_LSHIFT32( x_sc_Q10[ i ], 4 ) );
352
11.0G
        sLF_AR_shp_Q14 = silk_SUB32_ovflw( NSQ->sDiff_shp_Q14, silk_LSHIFT32( n_AR_Q12, 2 ) );
353
11.0G
        NSQ->sLF_AR_shp_Q14 = sLF_AR_shp_Q14;
354
355
11.0G
        NSQ->sLTP_shp_Q14[ NSQ->sLTP_shp_buf_idx ] = silk_SUB32_ovflw(sLF_AR_shp_Q14, silk_LSHIFT32(n_LF_Q12, 2));
356
11.0G
        sLTP_Q15[ NSQ->sLTP_buf_idx ] = silk_LSHIFT( LPC_exc_Q14, 1 );
357
11.0G
        NSQ->sLTP_shp_buf_idx++;
358
11.0G
        NSQ->sLTP_buf_idx++;
359
360
        /* Make dither dependent on quantized signal */
361
11.0G
        NSQ->rand_seed = silk_ADD32_ovflw( NSQ->rand_seed, pulses[ i ] );
362
11.0G
    }
363
364
    /* Update LPC synth buffer */
365
226M
    silk_memcpy( NSQ->sLPC_Q14, &NSQ->sLPC_Q14[ length ], NSQ_LPC_BUF_LENGTH * sizeof( opus_int32 ) );
366
226M
}
367
368
static OPUS_INLINE void silk_nsq_scale_states(
369
    const silk_encoder_state *psEncC,           /* I    Encoder State                   */
370
    silk_nsq_state      *NSQ,                   /* I/O  NSQ state                       */
371
    const opus_int16    x16[],                  /* I    input                           */
372
    opus_int32          x_sc_Q10[],             /* O    input scaled with 1/Gain        */
373
    const opus_int16    sLTP[],                 /* I    re-whitened LTP state in Q0     */
374
    opus_int32          sLTP_Q15[],             /* O    LTP state matching scaled input */
375
    opus_int            subfr,                  /* I    subframe number                 */
376
    const opus_int      LTP_scale_Q14,          /* I                                    */
377
    const opus_int32    Gains_Q16[ MAX_NB_SUBFR ], /* I                                 */
378
    const opus_int      pitchL[ MAX_NB_SUBFR ], /* I    Pitch lag                       */
379
    const opus_int      signal_type             /* I    Signal type                     */
380
)
381
72.8M
{
382
72.8M
    opus_int   i, lag;
383
72.8M
    opus_int32 gain_adj_Q16, inv_gain_Q31, inv_gain_Q26;
384
385
72.8M
    lag          = pitchL[ subfr ];
386
72.8M
    inv_gain_Q31 = silk_INVERSE32_varQ( silk_max( Gains_Q16[ subfr ], 1 ), 47 );
387
72.8M
    silk_assert( inv_gain_Q31 != 0 );
388
389
    /* Scale input */
390
72.8M
    inv_gain_Q26 = silk_RSHIFT_ROUND( inv_gain_Q31, 5 );
391
3.48G
    for( i = 0; i < psEncC->subfr_length; i++ ) {
392
3.41G
        x_sc_Q10[ i ] = silk_SMULWW( x16[ i ], inv_gain_Q26 );
393
3.41G
    }
394
395
    /* After rewhitening the LTP state is un-scaled, so scale with inv_gain_Q16 */
396
72.8M
    if( NSQ->rewhite_flag ) {
397
547k
        if( subfr == 0 ) {
398
            /* Do LTP downscaling */
399
547k
            inv_gain_Q31 = silk_LSHIFT( silk_SMULWB( inv_gain_Q31, LTP_scale_Q14 ), 2 );
400
547k
        }
401
37.8M
        for( i = NSQ->sLTP_buf_idx - lag - LTP_ORDER / 2; i < NSQ->sLTP_buf_idx; i++ ) {
402
37.3M
            silk_assert( i < MAX_FRAME_LENGTH );
403
37.3M
            sLTP_Q15[ i ] = silk_SMULWB( inv_gain_Q31, sLTP[ i ] );
404
37.3M
        }
405
547k
    }
406
407
    /* Adjust for changing gain */
408
72.8M
    if( Gains_Q16[ subfr ] != NSQ->prev_gain_Q16 ) {
409
3.70M
        gain_adj_Q16 =  silk_DIV32_varQ( NSQ->prev_gain_Q16, Gains_Q16[ subfr ], 16 );
410
411
        /* Scale long-term shaping state */
412
681M
        for( i = NSQ->sLTP_shp_buf_idx - psEncC->ltp_mem_length; i < NSQ->sLTP_shp_buf_idx; i++ ) {
413
677M
            NSQ->sLTP_shp_Q14[ i ] = silk_SMULWW( gain_adj_Q16, NSQ->sLTP_shp_Q14[ i ] );
414
677M
        }
415
416
        /* Scale long-term prediction state */
417
3.70M
        if( signal_type == TYPE_VOICED && NSQ->rewhite_flag == 0 ) {
418
73.4M
            for( i = NSQ->sLTP_buf_idx - lag - LTP_ORDER / 2; i < NSQ->sLTP_buf_idx; i++ ) {
419
72.3M
                sLTP_Q15[ i ] = silk_SMULWW( gain_adj_Q16, sLTP_Q15[ i ] );
420
72.3M
            }
421
1.08M
        }
422
423
3.70M
        NSQ->sLF_AR_shp_Q14 = silk_SMULWW( gain_adj_Q16, NSQ->sLF_AR_shp_Q14 );
424
3.70M
        NSQ->sDiff_shp_Q14 = silk_SMULWW( gain_adj_Q16, NSQ->sDiff_shp_Q14 );
425
426
        /* Scale short-term prediction and shaping states */
427
63.0M
        for( i = 0; i < NSQ_LPC_BUF_LENGTH; i++ ) {
428
59.3M
            NSQ->sLPC_Q14[ i ] = silk_SMULWW( gain_adj_Q16, NSQ->sLPC_Q14[ i ] );
429
59.3M
        }
430
92.7M
        for( i = 0; i < MAX_SHAPE_LPC_ORDER; i++ ) {
431
89.0M
            NSQ->sAR2_Q14[ i ] = silk_SMULWW( gain_adj_Q16, NSQ->sAR2_Q14[ i ] );
432
89.0M
        }
433
434
        /* Save inverse gain */
435
3.70M
        NSQ->prev_gain_Q16 = Gains_Q16[ subfr ];
436
3.70M
    }
437
72.8M
}