diff --git a/libcelt/quant_bands.c b/libcelt/quant_bands.c index acd59659f87ef9232d4c33f929c29f52d21a99d5..f33bf3f8f1bea4e1953d5aecce0875ad7e12887d 100644 --- a/libcelt/quant_bands.c +++ b/libcelt/quant_bands.c @@ -271,11 +271,12 @@ void quant_coarse_energy(const CELTMode *m, int start, int end, int effEnd, celt_uint32 tell; int badness1=0; celt_int32 intra_bias; + celt_word32 new_distortion; SAVE_STACK; intra = force_intra || (!two_pass && *delayedIntra>2*C*(end-start) && nbAvailableBytes > (end-start)*C); intra_bias = ((budget**delayedIntra*loss_rate)/(C*512)); - *delayedIntra = loss_distortion(eBands, oldEBands, start, effEnd, m->nbEBands, C); + new_distortion = loss_distortion(eBands, oldEBands, start, effEnd, m->nbEBands, C); tell = ec_tell(enc); if (tell+3 > budget) @@ -336,11 +337,19 @@ void quant_coarse_energy(const CELTMode *m, int start, int end, int effEnd, intra_bits, nintra_bytes - nstart_bytes); CELT_COPY(oldEBands, oldEBands_intra, C*m->nbEBands); CELT_COPY(error, error_intra, C*m->nbEBands); + intra = 1; } } else { CELT_COPY(oldEBands, oldEBands_intra, C*m->nbEBands); CELT_COPY(error, error_intra, C*m->nbEBands); } + + if (intra) + *delayedIntra = new_distortion; + else + *delayedIntra = ADD32(MULT16_32_Q15(MULT16_16_Q15(pred_coef[LM], pred_coef[LM]),*delayedIntra), + new_distortion); + RESTORE_STACK; }