Chris@69: /* Copyright (c) 2007-2008 CSIRO
Chris@69:    Copyright (c) 2007-2010 Xiph.Org Foundation
Chris@69:    Copyright (c) 2008 Gregory Maxwell
Chris@69:    Written by Jean-Marc Valin and Gregory Maxwell */
Chris@69: /*
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: 
Chris@69:    - Redistributions of source code must retain the above copyright
Chris@69:    notice, this list of conditions and the following disclaimer.
Chris@69: 
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: 
Chris@69:    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
Chris@69:    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
Chris@69:    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
Chris@69:    A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
Chris@69:    OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
Chris@69:    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
Chris@69:    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
Chris@69:    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
Chris@69:    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
Chris@69:    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
Chris@69:    SOFTWARE, EVEN IF ADVISED OF THE 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: #define CELT_ENCODER_C
Chris@69: 
Chris@69: #include "cpu_support.h"
Chris@69: #include "os_support.h"
Chris@69: #include "mdct.h"
Chris@69: #include <math.h>
Chris@69: #include "celt.h"
Chris@69: #include "pitch.h"
Chris@69: #include "bands.h"
Chris@69: #include "modes.h"
Chris@69: #include "entcode.h"
Chris@69: #include "quant_bands.h"
Chris@69: #include "rate.h"
Chris@69: #include "stack_alloc.h"
Chris@69: #include "mathops.h"
Chris@69: #include "float_cast.h"
Chris@69: #include <stdarg.h>
Chris@69: #include "celt_lpc.h"
Chris@69: #include "vq.h"
Chris@69: 
Chris@69: 
Chris@69: /** Encoder state
Chris@69:  @brief Encoder state
Chris@69:  */
Chris@69: struct OpusCustomEncoder {
Chris@69:    const OpusCustomMode *mode;     /**< Mode used by the encoder */
Chris@69:    int channels;
Chris@69:    int stream_channels;
Chris@69: 
Chris@69:    int force_intra;
Chris@69:    int clip;
Chris@69:    int disable_pf;
Chris@69:    int complexity;
Chris@69:    int upsample;
Chris@69:    int start, end;
Chris@69: 
Chris@69:    opus_int32 bitrate;
Chris@69:    int vbr;
Chris@69:    int signalling;
Chris@69:    int constrained_vbr;      /* If zero, VBR can do whatever it likes with the rate */
Chris@69:    int loss_rate;
Chris@69:    int lsb_depth;
Chris@69:    int lfe;
Chris@69:    int disable_inv;
Chris@69:    int arch;
Chris@69: 
Chris@69:    /* Everything beyond this point gets cleared on a reset */
Chris@69: #define ENCODER_RESET_START rng
Chris@69: 
Chris@69:    opus_uint32 rng;
Chris@69:    int spread_decision;
Chris@69:    opus_val32 delayedIntra;
Chris@69:    int tonal_average;
Chris@69:    int lastCodedBands;
Chris@69:    int hf_average;
Chris@69:    int tapset_decision;
Chris@69: 
Chris@69:    int prefilter_period;
Chris@69:    opus_val16 prefilter_gain;
Chris@69:    int prefilter_tapset;
Chris@69: #ifdef RESYNTH
Chris@69:    int prefilter_period_old;
Chris@69:    opus_val16 prefilter_gain_old;
Chris@69:    int prefilter_tapset_old;
Chris@69: #endif
Chris@69:    int consec_transient;
Chris@69:    AnalysisInfo analysis;
Chris@69:    SILKInfo silk_info;
Chris@69: 
Chris@69:    opus_val32 preemph_memE[2];
Chris@69:    opus_val32 preemph_memD[2];
Chris@69: 
Chris@69:    /* VBR-related parameters */
Chris@69:    opus_int32 vbr_reservoir;
Chris@69:    opus_int32 vbr_drift;
Chris@69:    opus_int32 vbr_offset;
Chris@69:    opus_int32 vbr_count;
Chris@69:    opus_val32 overlap_max;
Chris@69:    opus_val16 stereo_saving;
Chris@69:    int intensity;
Chris@69:    opus_val16 *energy_mask;
Chris@69:    opus_val16 spec_avg;
Chris@69: 
Chris@69: #ifdef RESYNTH
Chris@69:    /* +MAX_PERIOD/2 to make space for overlap */
Chris@69:    celt_sig syn_mem[2][2*MAX_PERIOD+MAX_PERIOD/2];
Chris@69: #endif
Chris@69: 
Chris@69:    celt_sig in_mem[1]; /* Size = channels*mode->overlap */
Chris@69:    /* celt_sig prefilter_mem[],  Size = channels*COMBFILTER_MAXPERIOD */
Chris@69:    /* opus_val16 oldBandE[],     Size = channels*mode->nbEBands */
Chris@69:    /* opus_val16 oldLogE[],      Size = channels*mode->nbEBands */
Chris@69:    /* opus_val16 oldLogE2[],     Size = channels*mode->nbEBands */
Chris@69:    /* opus_val16 energyError[],  Size = channels*mode->nbEBands */
Chris@69: };
Chris@69: 
Chris@69: int celt_encoder_get_size(int channels)
Chris@69: {
Chris@69:    CELTMode *mode = opus_custom_mode_create(48000, 960, NULL);
Chris@69:    return opus_custom_encoder_get_size(mode, channels);
Chris@69: }
Chris@69: 
Chris@69: OPUS_CUSTOM_NOSTATIC int opus_custom_encoder_get_size(const CELTMode *mode, int channels)
Chris@69: {
Chris@69:    int size = sizeof(struct CELTEncoder)
Chris@69:          + (channels*mode->overlap-1)*sizeof(celt_sig)    /* celt_sig in_mem[channels*mode->overlap]; */
Chris@69:          + channels*COMBFILTER_MAXPERIOD*sizeof(celt_sig) /* celt_sig prefilter_mem[channels*COMBFILTER_MAXPERIOD]; */
Chris@69:          + 4*channels*mode->nbEBands*sizeof(opus_val16);  /* opus_val16 oldBandE[channels*mode->nbEBands]; */
Chris@69:                                                           /* opus_val16 oldLogE[channels*mode->nbEBands]; */
Chris@69:                                                           /* opus_val16 oldLogE2[channels*mode->nbEBands]; */
Chris@69:                                                           /* opus_val16 energyError[channels*mode->nbEBands]; */
Chris@69:    return size;
Chris@69: }
Chris@69: 
Chris@69: #ifdef CUSTOM_MODES
Chris@69: CELTEncoder *opus_custom_encoder_create(const CELTMode *mode, int channels, int *error)
Chris@69: {
Chris@69:    int ret;
Chris@69:    CELTEncoder *st = (CELTEncoder *)opus_alloc(opus_custom_encoder_get_size(mode, channels));
Chris@69:    /* init will handle the NULL case */
Chris@69:    ret = opus_custom_encoder_init(st, mode, channels);
Chris@69:    if (ret != OPUS_OK)
Chris@69:    {
Chris@69:       opus_custom_encoder_destroy(st);
Chris@69:       st = NULL;
Chris@69:    }
Chris@69:    if (error)
Chris@69:       *error = ret;
Chris@69:    return st;
Chris@69: }
Chris@69: #endif /* CUSTOM_MODES */
Chris@69: 
Chris@69: static int opus_custom_encoder_init_arch(CELTEncoder *st, const CELTMode *mode,
Chris@69:                                          int channels, int arch)
Chris@69: {
Chris@69:    if (channels < 0 || channels > 2)
Chris@69:       return OPUS_BAD_ARG;
Chris@69: 
Chris@69:    if (st==NULL || mode==NULL)
Chris@69:       return OPUS_ALLOC_FAIL;
Chris@69: 
Chris@69:    OPUS_CLEAR((char*)st, opus_custom_encoder_get_size(mode, channels));
Chris@69: 
Chris@69:    st->mode = mode;
Chris@69:    st->stream_channels = st->channels = channels;
Chris@69: 
Chris@69:    st->upsample = 1;
Chris@69:    st->start = 0;
Chris@69:    st->end = st->mode->effEBands;
Chris@69:    st->signalling = 1;
Chris@69:    st->arch = arch;
Chris@69: 
Chris@69:    st->constrained_vbr = 1;
Chris@69:    st->clip = 1;
Chris@69: 
Chris@69:    st->bitrate = OPUS_BITRATE_MAX;
Chris@69:    st->vbr = 0;
Chris@69:    st->force_intra  = 0;
Chris@69:    st->complexity = 5;
Chris@69:    st->lsb_depth=24;
Chris@69: 
Chris@69:    opus_custom_encoder_ctl(st, OPUS_RESET_STATE);
Chris@69: 
Chris@69:    return OPUS_OK;
Chris@69: }
Chris@69: 
Chris@69: #ifdef CUSTOM_MODES
Chris@69: int opus_custom_encoder_init(CELTEncoder *st, const CELTMode *mode, int channels)
Chris@69: {
Chris@69:    return opus_custom_encoder_init_arch(st, mode, channels, opus_select_arch());
Chris@69: }
Chris@69: #endif
Chris@69: 
Chris@69: int celt_encoder_init(CELTEncoder *st, opus_int32 sampling_rate, int channels,
Chris@69:                       int arch)
Chris@69: {
Chris@69:    int ret;
Chris@69:    ret = opus_custom_encoder_init_arch(st,
Chris@69:            opus_custom_mode_create(48000, 960, NULL), channels, arch);
Chris@69:    if (ret != OPUS_OK)
Chris@69:       return ret;
Chris@69:    st->upsample = resampling_factor(sampling_rate);
Chris@69:    return OPUS_OK;
Chris@69: }
Chris@69: 
Chris@69: #ifdef CUSTOM_MODES
Chris@69: void opus_custom_encoder_destroy(CELTEncoder *st)
Chris@69: {
Chris@69:    opus_free(st);
Chris@69: }
Chris@69: #endif /* CUSTOM_MODES */
Chris@69: 
Chris@69: 
Chris@69: static int transient_analysis(const opus_val32 * OPUS_RESTRICT in, int len, int C,
Chris@69:                               opus_val16 *tf_estimate, int *tf_chan, int allow_weak_transients,
Chris@69:                               int *weak_transient)
Chris@69: {
Chris@69:    int i;
Chris@69:    VARDECL(opus_val16, tmp);
Chris@69:    opus_val32 mem0,mem1;
Chris@69:    int is_transient = 0;
Chris@69:    opus_int32 mask_metric = 0;
Chris@69:    int c;
Chris@69:    opus_val16 tf_max;
Chris@69:    int len2;
Chris@69:    /* Forward masking: 6.7 dB/ms. */
Chris@69: #ifdef FIXED_POINT
Chris@69:    int forward_shift = 4;
Chris@69: #else
Chris@69:    opus_val16 forward_decay = QCONST16(.0625f,15);
Chris@69: #endif
Chris@69:    /* Table of 6*64/x, trained on real data to minimize the average error */
Chris@69:    static const unsigned char inv_table[128] = {
Chris@69:          255,255,156,110, 86, 70, 59, 51, 45, 40, 37, 33, 31, 28, 26, 25,
Chris@69:           23, 22, 21, 20, 19, 18, 17, 16, 16, 15, 15, 14, 13, 13, 12, 12,
Chris@69:           12, 12, 11, 11, 11, 10, 10, 10,  9,  9,  9,  9,  9,  9,  8,  8,
Chris@69:            8,  8,  8,  7,  7,  7,  7,  7,  7,  6,  6,  6,  6,  6,  6,  6,
Chris@69:            6,  6,  6,  6,  6,  6,  6,  6,  6,  5,  5,  5,  5,  5,  5,  5,
Chris@69:            5,  5,  5,  5,  5,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,
Chris@69:            4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  4,  3,  3,
Chris@69:            3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  2,
Chris@69:    };
Chris@69:    SAVE_STACK;
Chris@69:    ALLOC(tmp, len, opus_val16);
Chris@69: 
Chris@69:    *weak_transient = 0;
Chris@69:    /* For lower bitrates, let's be more conservative and have a forward masking
Chris@69:       decay of 3.3 dB/ms. This avoids having to code transients at very low
Chris@69:       bitrate (mostly for hybrid), which can result in unstable energy and/or
Chris@69:       partial collapse. */
Chris@69:    if (allow_weak_transients)
Chris@69:    {
Chris@69: #ifdef FIXED_POINT
Chris@69:       forward_shift = 5;
Chris@69: #else
Chris@69:       forward_decay = QCONST16(.03125f,15);
Chris@69: #endif
Chris@69:    }
Chris@69:    len2=len/2;
Chris@69:    for (c=0;c<C;c++)
Chris@69:    {
Chris@69:       opus_val32 mean;
Chris@69:       opus_int32 unmask=0;
Chris@69:       opus_val32 norm;
Chris@69:       opus_val16 maxE;
Chris@69:       mem0=0;
Chris@69:       mem1=0;
Chris@69:       /* High-pass filter: (1 - 2*z^-1 + z^-2) / (1 - z^-1 + .5*z^-2) */
Chris@69:       for (i=0;i<len;i++)
Chris@69:       {
Chris@69:          opus_val32 x,y;
Chris@69:          x = SHR32(in[i+c*len],SIG_SHIFT);
Chris@69:          y = ADD32(mem0, x);
Chris@69: #ifdef FIXED_POINT
Chris@69:          mem0 = mem1 + y - SHL32(x,1);
Chris@69:          mem1 = x - SHR32(y,1);
Chris@69: #else
Chris@69:          mem0 = mem1 + y - 2*x;
Chris@69:          mem1 = x - .5f*y;
Chris@69: #endif
Chris@69:          tmp[i] = SROUND16(y, 2);
Chris@69:          /*printf("%f ", tmp[i]);*/
Chris@69:       }
Chris@69:       /*printf("\n");*/
Chris@69:       /* First few samples are bad because we don't propagate the memory */
Chris@69:       OPUS_CLEAR(tmp, 12);
Chris@69: 
Chris@69: #ifdef FIXED_POINT
Chris@69:       /* Normalize tmp to max range */
Chris@69:       {
Chris@69:          int shift=0;
Chris@69:          shift = 14-celt_ilog2(MAX16(1, celt_maxabs16(tmp, len)));
Chris@69:          if (shift!=0)
Chris@69:          {
Chris@69:             for (i=0;i<len;i++)
Chris@69:                tmp[i] = SHL16(tmp[i], shift);
Chris@69:          }
Chris@69:       }
Chris@69: #endif
Chris@69: 
Chris@69:       mean=0;
Chris@69:       mem0=0;
Chris@69:       /* Grouping by two to reduce complexity */
Chris@69:       /* Forward pass to compute the post-echo threshold*/
Chris@69:       for (i=0;i<len2;i++)
Chris@69:       {
Chris@69:          opus_val16 x2 = PSHR32(MULT16_16(tmp[2*i],tmp[2*i]) + MULT16_16(tmp[2*i+1],tmp[2*i+1]),16);
Chris@69:          mean += x2;
Chris@69: #ifdef FIXED_POINT
Chris@69:          /* FIXME: Use PSHR16() instead */
Chris@69:          tmp[i] = mem0 + PSHR32(x2-mem0,forward_shift);
Chris@69: #else
Chris@69:          tmp[i] = mem0 + MULT16_16_P15(forward_decay,x2-mem0);
Chris@69: #endif
Chris@69:          mem0 = tmp[i];
Chris@69:       }
Chris@69: 
Chris@69:       mem0=0;
Chris@69:       maxE=0;
Chris@69:       /* Backward pass to compute the pre-echo threshold */
Chris@69:       for (i=len2-1;i>=0;i--)
Chris@69:       {
Chris@69:          /* Backward masking: 13.9 dB/ms. */
Chris@69: #ifdef FIXED_POINT
Chris@69:          /* FIXME: Use PSHR16() instead */
Chris@69:          tmp[i] = mem0 + PSHR32(tmp[i]-mem0,3);
Chris@69: #else
Chris@69:          tmp[i] = mem0 + MULT16_16_P15(QCONST16(0.125f,15),tmp[i]-mem0);
Chris@69: #endif
Chris@69:          mem0 = tmp[i];
Chris@69:          maxE = MAX16(maxE, mem0);
Chris@69:       }
Chris@69:       /*for (i=0;i<len2;i++)printf("%f ", tmp[i]/mean);printf("\n");*/
Chris@69: 
Chris@69:       /* Compute the ratio of the "frame energy" over the harmonic mean of the energy.
Chris@69:          This essentially corresponds to a bitrate-normalized temporal noise-to-mask
Chris@69:          ratio */
Chris@69: 
Chris@69:       /* As a compromise with the old transient detector, frame energy is the
Chris@69:          geometric mean of the energy and half the max */
Chris@69: #ifdef FIXED_POINT
Chris@69:       /* Costs two sqrt() to avoid overflows */
Chris@69:       mean = MULT16_16(celt_sqrt(mean), celt_sqrt(MULT16_16(maxE,len2>>1)));
Chris@69: #else
Chris@69:       mean = celt_sqrt(mean * maxE*.5*len2);
Chris@69: #endif
Chris@69:       /* Inverse of the mean energy in Q15+6 */
Chris@69:       norm = SHL32(EXTEND32(len2),6+14)/ADD32(EPSILON,SHR32(mean,1));
Chris@69:       /* Compute harmonic mean discarding the unreliable boundaries
Chris@69:          The data is smooth, so we only take 1/4th of the samples */
Chris@69:       unmask=0;
Chris@69:       /* We should never see NaNs here. If we find any, then something really bad happened and we better abort
Chris@69:          before it does any damage later on. If these asserts are disabled (no hardening), then the table
Chris@69:          lookup a few lines below (id = ...) is likely to crash dur to an out-of-bounds read. DO NOT FIX
Chris@69:          that crash on NaN since it could result in a worse issue later on. */
Chris@69:       celt_assert(!celt_isnan(tmp[0]));
Chris@69:       celt_assert(!celt_isnan(norm));
Chris@69:       for (i=12;i<len2-5;i+=4)
Chris@69:       {
Chris@69:          int id;
Chris@69: #ifdef FIXED_POINT
Chris@69:          id = MAX32(0,MIN32(127,MULT16_32_Q15(tmp[i]+EPSILON,norm))); /* Do not round to nearest */
Chris@69: #else
Chris@69:          id = (int)MAX32(0,MIN32(127,floor(64*norm*(tmp[i]+EPSILON)))); /* Do not round to nearest */
Chris@69: #endif
Chris@69:          unmask += inv_table[id];
Chris@69:       }
Chris@69:       /*printf("%d\n", unmask);*/
Chris@69:       /* Normalize, compensate for the 1/4th of the sample and the factor of 6 in the inverse table */
Chris@69:       unmask = 64*unmask*4/(6*(len2-17));
Chris@69:       if (unmask>mask_metric)
Chris@69:       {
Chris@69:          *tf_chan = c;
Chris@69:          mask_metric = unmask;
Chris@69:       }
Chris@69:    }
Chris@69:    is_transient = mask_metric>200;
Chris@69:    /* For low bitrates, define "weak transients" that need to be
Chris@69:       handled differently to avoid partial collapse. */
Chris@69:    if (allow_weak_transients && is_transient && mask_metric<600) {
Chris@69:       is_transient = 0;
Chris@69:       *weak_transient = 1;
Chris@69:    }
Chris@69:    /* Arbitrary metric for VBR boost */
Chris@69:    tf_max = MAX16(0,celt_sqrt(27*mask_metric)-42);
Chris@69:    /* *tf_estimate = 1 + MIN16(1, sqrt(MAX16(0, tf_max-30))/20); */
Chris@69:    *tf_estimate = celt_sqrt(MAX32(0, SHL32(MULT16_16(QCONST16(0.0069,14),MIN16(163,tf_max)),14)-QCONST32(0.139,28)));
Chris@69:    /*printf("%d %f\n", tf_max, mask_metric);*/
Chris@69:    RESTORE_STACK;
Chris@69: #ifdef FUZZING
Chris@69:    is_transient = rand()&0x1;
Chris@69: #endif
Chris@69:    /*printf("%d %f %d\n", is_transient, (float)*tf_estimate, tf_max);*/
Chris@69:    return is_transient;
Chris@69: }
Chris@69: 
Chris@69: /* Looks for sudden increases of energy to decide whether we need to patch
Chris@69:    the transient decision */
Chris@69: static int patch_transient_decision(opus_val16 *newE, opus_val16 *oldE, int nbEBands,
Chris@69:       int start, int end, int C)
Chris@69: {
Chris@69:    int i, c;
Chris@69:    opus_val32 mean_diff=0;
Chris@69:    opus_val16 spread_old[26];
Chris@69:    /* Apply an aggressive (-6 dB/Bark) spreading function to the old frame to
Chris@69:       avoid false detection caused by irrelevant bands */
Chris@69:    if (C==1)
Chris@69:    {
Chris@69:       spread_old[start] = oldE[start];
Chris@69:       for (i=start+1;i<end;i++)
Chris@69:          spread_old[i] = MAX16(spread_old[i-1]-QCONST16(1.0f, DB_SHIFT), oldE[i]);
Chris@69:    } else {
Chris@69:       spread_old[start] = MAX16(oldE[start],oldE[start+nbEBands]);
Chris@69:       for (i=start+1;i<end;i++)
Chris@69:          spread_old[i] = MAX16(spread_old[i-1]-QCONST16(1.0f, DB_SHIFT),
Chris@69:                                MAX16(oldE[i],oldE[i+nbEBands]));
Chris@69:    }
Chris@69:    for (i=end-2;i>=start;i--)
Chris@69:       spread_old[i] = MAX16(spread_old[i], spread_old[i+1]-QCONST16(1.0f, DB_SHIFT));
Chris@69:    /* Compute mean increase */
Chris@69:    c=0; do {
Chris@69:       for (i=IMAX(2,start);i<end-1;i++)
Chris@69:       {
Chris@69:          opus_val16 x1, x2;
Chris@69:          x1 = MAX16(0, newE[i + c*nbEBands]);
Chris@69:          x2 = MAX16(0, spread_old[i]);
Chris@69:          mean_diff = ADD32(mean_diff, EXTEND32(MAX16(0, SUB16(x1, x2))));
Chris@69:       }
Chris@69:    } while (++c<C);
Chris@69:    mean_diff = DIV32(mean_diff, C*(end-1-IMAX(2,start)));
Chris@69:    /*printf("%f %f %d\n", mean_diff, max_diff, count);*/
Chris@69:    return mean_diff > QCONST16(1.f, DB_SHIFT);
Chris@69: }
Chris@69: 
Chris@69: /** Apply window and compute the MDCT for all sub-frames and
Chris@69:     all channels in a frame */
Chris@69: static void compute_mdcts(const CELTMode *mode, int shortBlocks, celt_sig * OPUS_RESTRICT in,
Chris@69:                           celt_sig * OPUS_RESTRICT out, int C, int CC, int LM, int upsample,
Chris@69:                           int arch)
Chris@69: {
Chris@69:    const int overlap = mode->overlap;
Chris@69:    int N;
Chris@69:    int B;
Chris@69:    int shift;
Chris@69:    int i, b, c;
Chris@69:    if (shortBlocks)
Chris@69:    {
Chris@69:       B = shortBlocks;
Chris@69:       N = mode->shortMdctSize;
Chris@69:       shift = mode->maxLM;
Chris@69:    } else {
Chris@69:       B = 1;
Chris@69:       N = mode->shortMdctSize<<LM;
Chris@69:       shift = mode->maxLM-LM;
Chris@69:    }
Chris@69:    c=0; do {
Chris@69:       for (b=0;b<B;b++)
Chris@69:       {
Chris@69:          /* Interleaving the sub-frames while doing the MDCTs */
Chris@69:          clt_mdct_forward(&mode->mdct, in+c*(B*N+overlap)+b*N,
Chris@69:                           &out[b+c*N*B], mode->window, overlap, shift, B,
Chris@69:                           arch);
Chris@69:       }
Chris@69:    } while (++c<CC);
Chris@69:    if (CC==2&&C==1)
Chris@69:    {
Chris@69:       for (i=0;i<B*N;i++)
Chris@69:          out[i] = ADD32(HALF32(out[i]), HALF32(out[B*N+i]));
Chris@69:    }
Chris@69:    if (upsample != 1)
Chris@69:    {
Chris@69:       c=0; do
Chris@69:       {
Chris@69:          int bound = B*N/upsample;
Chris@69:          for (i=0;i<bound;i++)
Chris@69:             out[c*B*N+i] *= upsample;
Chris@69:          OPUS_CLEAR(&out[c*B*N+bound], B*N-bound);
Chris@69:       } while (++c<C);
Chris@69:    }
Chris@69: }
Chris@69: 
Chris@69: 
Chris@69: void celt_preemphasis(const opus_val16 * OPUS_RESTRICT pcmp, celt_sig * OPUS_RESTRICT inp,
Chris@69:                         int N, int CC, int upsample, const opus_val16 *coef, celt_sig *mem, int clip)
Chris@69: {
Chris@69:    int i;
Chris@69:    opus_val16 coef0;
Chris@69:    celt_sig m;
Chris@69:    int Nu;
Chris@69: 
Chris@69:    coef0 = coef[0];
Chris@69:    m = *mem;
Chris@69: 
Chris@69:    /* Fast path for the normal 48kHz case and no clipping */
Chris@69:    if (coef[1] == 0 && upsample == 1 && !clip)
Chris@69:    {
Chris@69:       for (i=0;i<N;i++)
Chris@69:       {
Chris@69:          opus_val16 x;
Chris@69:          x = SCALEIN(pcmp[CC*i]);
Chris@69:          /* Apply pre-emphasis */
Chris@69:          inp[i] = SHL32(x, SIG_SHIFT) - m;
Chris@69:          m = SHR32(MULT16_16(coef0, x), 15-SIG_SHIFT);
Chris@69:       }
Chris@69:       *mem = m;
Chris@69:       return;
Chris@69:    }
Chris@69: 
Chris@69:    Nu = N/upsample;
Chris@69:    if (upsample!=1)
Chris@69:    {
Chris@69:       OPUS_CLEAR(inp, N);
Chris@69:    }
Chris@69:    for (i=0;i<Nu;i++)
Chris@69:       inp[i*upsample] = SCALEIN(pcmp[CC*i]);
Chris@69: 
Chris@69: #ifndef FIXED_POINT
Chris@69:    if (clip)
Chris@69:    {
Chris@69:       /* Clip input to avoid encoding non-portable files */
Chris@69:       for (i=0;i<Nu;i++)
Chris@69:          inp[i*upsample] = MAX32(-65536.f, MIN32(65536.f,inp[i*upsample]));
Chris@69:    }
Chris@69: #else
Chris@69:    (void)clip; /* Avoids a warning about clip being unused. */
Chris@69: #endif
Chris@69: #ifdef CUSTOM_MODES
Chris@69:    if (coef[1] != 0)
Chris@69:    {
Chris@69:       opus_val16 coef1 = coef[1];
Chris@69:       opus_val16 coef2 = coef[2];
Chris@69:       for (i=0;i<N;i++)
Chris@69:       {
Chris@69:          celt_sig x, tmp;
Chris@69:          x = inp[i];
Chris@69:          /* Apply pre-emphasis */
Chris@69:          tmp = MULT16_16(coef2, x);
Chris@69:          inp[i] = tmp + m;
Chris@69:          m = MULT16_32_Q15(coef1, inp[i]) - MULT16_32_Q15(coef0, tmp);
Chris@69:       }
Chris@69:    } else
Chris@69: #endif
Chris@69:    {
Chris@69:       for (i=0;i<N;i++)
Chris@69:       {
Chris@69:          opus_val16 x;
Chris@69:          x = inp[i];
Chris@69:          /* Apply pre-emphasis */
Chris@69:          inp[i] = SHL32(x, SIG_SHIFT) - m;
Chris@69:          m = SHR32(MULT16_16(coef0, x), 15-SIG_SHIFT);
Chris@69:       }
Chris@69:    }
Chris@69:    *mem = m;
Chris@69: }
Chris@69: 
Chris@69: 
Chris@69: 
Chris@69: static opus_val32 l1_metric(const celt_norm *tmp, int N, int LM, opus_val16 bias)
Chris@69: {
Chris@69:    int i;
Chris@69:    opus_val32 L1;
Chris@69:    L1 = 0;
Chris@69:    for (i=0;i<N;i++)
Chris@69:       L1 += EXTEND32(ABS16(tmp[i]));
Chris@69:    /* When in doubt, prefer good freq resolution */
Chris@69:    L1 = MAC16_32_Q15(L1, LM*bias, L1);
Chris@69:    return L1;
Chris@69: 
Chris@69: }
Chris@69: 
Chris@69: static int tf_analysis(const CELTMode *m, int len, int isTransient,
Chris@69:       int *tf_res, int lambda, celt_norm *X, int N0, int LM,
Chris@69:       opus_val16 tf_estimate, int tf_chan, int *importance)
Chris@69: {
Chris@69:    int i;
Chris@69:    VARDECL(int, metric);
Chris@69:    int cost0;
Chris@69:    int cost1;
Chris@69:    VARDECL(int, path0);
Chris@69:    VARDECL(int, path1);
Chris@69:    VARDECL(celt_norm, tmp);
Chris@69:    VARDECL(celt_norm, tmp_1);
Chris@69:    int sel;
Chris@69:    int selcost[2];
Chris@69:    int tf_select=0;
Chris@69:    opus_val16 bias;
Chris@69: 
Chris@69:    SAVE_STACK;
Chris@69:    bias = MULT16_16_Q14(QCONST16(.04f,15), MAX16(-QCONST16(.25f,14), QCONST16(.5f,14)-tf_estimate));
Chris@69:    /*printf("%f ", bias);*/
Chris@69: 
Chris@69:    ALLOC(metric, len, int);
Chris@69:    ALLOC(tmp, (m->eBands[len]-m->eBands[len-1])<<LM, celt_norm);
Chris@69:    ALLOC(tmp_1, (m->eBands[len]-m->eBands[len-1])<<LM, celt_norm);
Chris@69:    ALLOC(path0, len, int);
Chris@69:    ALLOC(path1, len, int);
Chris@69: 
Chris@69:    for (i=0;i<len;i++)
Chris@69:    {
Chris@69:       int k, N;
Chris@69:       int narrow;
Chris@69:       opus_val32 L1, best_L1;
Chris@69:       int best_level=0;
Chris@69:       N = (m->eBands[i+1]-m->eBands[i])<<LM;
Chris@69:       /* band is too narrow to be split down to LM=-1 */
Chris@69:       narrow = (m->eBands[i+1]-m->eBands[i])==1;
Chris@69:       OPUS_COPY(tmp, &X[tf_chan*N0 + (m->eBands[i]<<LM)], N);
Chris@69:       /* Just add the right channel if we're in stereo */
Chris@69:       /*if (C==2)
Chris@69:          for (j=0;j<N;j++)
Chris@69:             tmp[j] = ADD16(SHR16(tmp[j], 1),SHR16(X[N0+j+(m->eBands[i]<<LM)], 1));*/
Chris@69:       L1 = l1_metric(tmp, N, isTransient ? LM : 0, bias);
Chris@69:       best_L1 = L1;
Chris@69:       /* Check the -1 case for transients */
Chris@69:       if (isTransient && !narrow)
Chris@69:       {
Chris@69:          OPUS_COPY(tmp_1, tmp, N);
Chris@69:          haar1(tmp_1, N>>LM, 1<<LM);
Chris@69:          L1 = l1_metric(tmp_1, N, LM+1, bias);
Chris@69:          if (L1<best_L1)
Chris@69:          {
Chris@69:             best_L1 = L1;
Chris@69:             best_level = -1;
Chris@69:          }
Chris@69:       }
Chris@69:       /*printf ("%f ", L1);*/
Chris@69:       for (k=0;k<LM+!(isTransient||narrow);k++)
Chris@69:       {
Chris@69:          int B;
Chris@69: 
Chris@69:          if (isTransient)
Chris@69:             B = (LM-k-1);
Chris@69:          else
Chris@69:             B = k+1;
Chris@69: 
Chris@69:          haar1(tmp, N>>k, 1<<k);
Chris@69: 
Chris@69:          L1 = l1_metric(tmp, N, B, bias);
Chris@69: 
Chris@69:          if (L1 < best_L1)
Chris@69:          {
Chris@69:             best_L1 = L1;
Chris@69:             best_level = k+1;
Chris@69:          }
Chris@69:       }
Chris@69:       /*printf ("%d ", isTransient ? LM-best_level : best_level);*/
Chris@69:       /* metric is in Q1 to be able to select the mid-point (-0.5) for narrower bands */
Chris@69:       if (isTransient)
Chris@69:          metric[i] = 2*best_level;
Chris@69:       else
Chris@69:          metric[i] = -2*best_level;
Chris@69:       /* For bands that can't be split to -1, set the metric to the half-way point to avoid
Chris@69:          biasing the decision */
Chris@69:       if (narrow && (metric[i]==0 || metric[i]==-2*LM))
Chris@69:          metric[i]-=1;
Chris@69:       /*printf("%d ", metric[i]/2 + (!isTransient)*LM);*/
Chris@69:    }
Chris@69:    /*printf("\n");*/
Chris@69:    /* Search for the optimal tf resolution, including tf_select */
Chris@69:    tf_select = 0;
Chris@69:    for (sel=0;sel<2;sel++)
Chris@69:    {
Chris@69:       cost0 = importance[0]*abs(metric[0]-2*tf_select_table[LM][4*isTransient+2*sel+0]);
Chris@69:       cost1 = importance[0]*abs(metric[0]-2*tf_select_table[LM][4*isTransient+2*sel+1]) + (isTransient ? 0 : lambda);
Chris@69:       for (i=1;i<len;i++)
Chris@69:       {
Chris@69:          int curr0, curr1;
Chris@69:          curr0 = IMIN(cost0, cost1 + lambda);
Chris@69:          curr1 = IMIN(cost0 + lambda, cost1);
Chris@69:          cost0 = curr0 + importance[i]*abs(metric[i]-2*tf_select_table[LM][4*isTransient+2*sel+0]);
Chris@69:          cost1 = curr1 + importance[i]*abs(metric[i]-2*tf_select_table[LM][4*isTransient+2*sel+1]);
Chris@69:       }
Chris@69:       cost0 = IMIN(cost0, cost1);
Chris@69:       selcost[sel]=cost0;
Chris@69:    }
Chris@69:    /* For now, we're conservative and only allow tf_select=1 for transients.
Chris@69:     * If tests confirm it's useful for non-transients, we could allow it. */
Chris@69:    if (selcost[1]<selcost[0] && isTransient)
Chris@69:       tf_select=1;
Chris@69:    cost0 = importance[0]*abs(metric[0]-2*tf_select_table[LM][4*isTransient+2*tf_select+0]);
Chris@69:    cost1 = importance[0]*abs(metric[0]-2*tf_select_table[LM][4*isTransient+2*tf_select+1]) + (isTransient ? 0 : lambda);
Chris@69:    /* Viterbi forward pass */
Chris@69:    for (i=1;i<len;i++)
Chris@69:    {
Chris@69:       int curr0, curr1;
Chris@69:       int from0, from1;
Chris@69: 
Chris@69:       from0 = cost0;
Chris@69:       from1 = cost1 + lambda;
Chris@69:       if (from0 < from1)
Chris@69:       {
Chris@69:          curr0 = from0;
Chris@69:          path0[i]= 0;
Chris@69:       } else {
Chris@69:          curr0 = from1;
Chris@69:          path0[i]= 1;
Chris@69:       }
Chris@69: 
Chris@69:       from0 = cost0 + lambda;
Chris@69:       from1 = cost1;
Chris@69:       if (from0 < from1)
Chris@69:       {
Chris@69:          curr1 = from0;
Chris@69:          path1[i]= 0;
Chris@69:       } else {
Chris@69:          curr1 = from1;
Chris@69:          path1[i]= 1;
Chris@69:       }
Chris@69:       cost0 = curr0 + importance[i]*abs(metric[i]-2*tf_select_table[LM][4*isTransient+2*tf_select+0]);
Chris@69:       cost1 = curr1 + importance[i]*abs(metric[i]-2*tf_select_table[LM][4*isTransient+2*tf_select+1]);
Chris@69:    }
Chris@69:    tf_res[len-1] = cost0 < cost1 ? 0 : 1;
Chris@69:    /* Viterbi backward pass to check the decisions */
Chris@69:    for (i=len-2;i>=0;i--)
Chris@69:    {
Chris@69:       if (tf_res[i+1] == 1)
Chris@69:          tf_res[i] = path1[i+1];
Chris@69:       else
Chris@69:          tf_res[i] = path0[i+1];
Chris@69:    }
Chris@69:    /*printf("%d %f\n", *tf_sum, tf_estimate);*/
Chris@69:    RESTORE_STACK;
Chris@69: #ifdef FUZZING
Chris@69:    tf_select = rand()&0x1;
Chris@69:    tf_res[0] = rand()&0x1;
Chris@69:    for (i=1;i<len;i++)
Chris@69:       tf_res[i] = tf_res[i-1] ^ ((rand()&0xF) == 0);
Chris@69: #endif
Chris@69:    return tf_select;
Chris@69: }
Chris@69: 
Chris@69: static void tf_encode(int start, int end, int isTransient, int *tf_res, int LM, int tf_select, ec_enc *enc)
Chris@69: {
Chris@69:    int curr, i;
Chris@69:    int tf_select_rsv;
Chris@69:    int tf_changed;
Chris@69:    int logp;
Chris@69:    opus_uint32 budget;
Chris@69:    opus_uint32 tell;
Chris@69:    budget = enc->storage*8;
Chris@69:    tell = ec_tell(enc);
Chris@69:    logp = isTransient ? 2 : 4;
Chris@69:    /* Reserve space to code the tf_select decision. */
Chris@69:    tf_select_rsv = LM>0 && tell+logp+1 <= budget;
Chris@69:    budget -= tf_select_rsv;
Chris@69:    curr = tf_changed = 0;
Chris@69:    for (i=start;i<end;i++)
Chris@69:    {
Chris@69:       if (tell+logp<=budget)
Chris@69:       {
Chris@69:          ec_enc_bit_logp(enc, tf_res[i] ^ curr, logp);
Chris@69:          tell = ec_tell(enc);
Chris@69:          curr = tf_res[i];
Chris@69:          tf_changed |= curr;
Chris@69:       }
Chris@69:       else
Chris@69:          tf_res[i] = curr;
Chris@69:       logp = isTransient ? 4 : 5;
Chris@69:    }
Chris@69:    /* Only code tf_select if it would actually make a difference. */
Chris@69:    if (tf_select_rsv &&
Chris@69:          tf_select_table[LM][4*isTransient+0+tf_changed]!=
Chris@69:          tf_select_table[LM][4*isTransient+2+tf_changed])
Chris@69:       ec_enc_bit_logp(enc, tf_select, 1);
Chris@69:    else
Chris@69:       tf_select = 0;
Chris@69:    for (i=start;i<end;i++)
Chris@69:       tf_res[i] = tf_select_table[LM][4*isTransient+2*tf_select+tf_res[i]];
Chris@69:    /*for(i=0;i<end;i++)printf("%d ", isTransient ? tf_res[i] : LM+tf_res[i]);printf("\n");*/
Chris@69: }
Chris@69: 
Chris@69: 
Chris@69: static int alloc_trim_analysis(const CELTMode *m, const celt_norm *X,
Chris@69:       const opus_val16 *bandLogE, int end, int LM, int C, int N0,
Chris@69:       AnalysisInfo *analysis, opus_val16 *stereo_saving, opus_val16 tf_estimate,
Chris@69:       int intensity, opus_val16 surround_trim, opus_int32 equiv_rate, int arch)
Chris@69: {
Chris@69:    int i;
Chris@69:    opus_val32 diff=0;
Chris@69:    int c;
Chris@69:    int trim_index;
Chris@69:    opus_val16 trim = QCONST16(5.f, 8);
Chris@69:    opus_val16 logXC, logXC2;
Chris@69:    /* At low bitrate, reducing the trim seems to help. At higher bitrates, it's less
Chris@69:       clear what's best, so we're keeping it as it was before, at least for now. */
Chris@69:    if (equiv_rate < 64000) {
Chris@69:       trim = QCONST16(4.f, 8);
Chris@69:    } else if (equiv_rate < 80000) {
Chris@69:       opus_int32 frac = (equiv_rate-64000) >> 10;
Chris@69:       trim = QCONST16(4.f, 8) + QCONST16(1.f/16.f, 8)*frac;
Chris@69:    }
Chris@69:    if (C==2)
Chris@69:    {
Chris@69:       opus_val16 sum = 0; /* Q10 */
Chris@69:       opus_val16 minXC; /* Q10 */
Chris@69:       /* Compute inter-channel correlation for low frequencies */
Chris@69:       for (i=0;i<8;i++)
Chris@69:       {
Chris@69:          opus_val32 partial;
Chris@69:          partial = celt_inner_prod(&X[m->eBands[i]<<LM], &X[N0+(m->eBands[i]<<LM)],
Chris@69:                (m->eBands[i+1]-m->eBands[i])<<LM, arch);
Chris@69:          sum = ADD16(sum, EXTRACT16(SHR32(partial, 18)));
Chris@69:       }
Chris@69:       sum = MULT16_16_Q15(QCONST16(1.f/8, 15), sum);
Chris@69:       sum = MIN16(QCONST16(1.f, 10), ABS16(sum));
Chris@69:       minXC = sum;
Chris@69:       for (i=8;i<intensity;i++)
Chris@69:       {
Chris@69:          opus_val32 partial;
Chris@69:          partial = celt_inner_prod(&X[m->eBands[i]<<LM], &X[N0+(m->eBands[i]<<LM)],
Chris@69:                (m->eBands[i+1]-m->eBands[i])<<LM, arch);
Chris@69:          minXC = MIN16(minXC, ABS16(EXTRACT16(SHR32(partial, 18))));
Chris@69:       }
Chris@69:       minXC = MIN16(QCONST16(1.f, 10), ABS16(minXC));
Chris@69:       /*printf ("%f\n", sum);*/
Chris@69:       /* mid-side savings estimations based on the LF average*/
Chris@69:       logXC = celt_log2(QCONST32(1.001f, 20)-MULT16_16(sum, sum));
Chris@69:       /* mid-side savings estimations based on min correlation */
Chris@69:       logXC2 = MAX16(HALF16(logXC), celt_log2(QCONST32(1.001f, 20)-MULT16_16(minXC, minXC)));
Chris@69: #ifdef FIXED_POINT
Chris@69:       /* Compensate for Q20 vs Q14 input and convert output to Q8 */
Chris@69:       logXC = PSHR32(logXC-QCONST16(6.f, DB_SHIFT),DB_SHIFT-8);
Chris@69:       logXC2 = PSHR32(logXC2-QCONST16(6.f, DB_SHIFT),DB_SHIFT-8);
Chris@69: #endif
Chris@69: 
Chris@69:       trim += MAX16(-QCONST16(4.f, 8), MULT16_16_Q15(QCONST16(.75f,15),logXC));
Chris@69:       *stereo_saving = MIN16(*stereo_saving + QCONST16(0.25f, 8), -HALF16(logXC2));
Chris@69:    }
Chris@69: 
Chris@69:    /* Estimate spectral tilt */
Chris@69:    c=0; do {
Chris@69:       for (i=0;i<end-1;i++)
Chris@69:       {
Chris@69:          diff += bandLogE[i+c*m->nbEBands]*(opus_int32)(2+2*i-end);
Chris@69:       }
Chris@69:    } while (++c<C);
Chris@69:    diff /= C*(end-1);
Chris@69:    /*printf("%f\n", diff);*/
Chris@69:    trim -= MAX32(-QCONST16(2.f, 8), MIN32(QCONST16(2.f, 8), SHR32(diff+QCONST16(1.f, DB_SHIFT),DB_SHIFT-8)/6 ));
Chris@69:    trim -= SHR16(surround_trim, DB_SHIFT-8);
Chris@69:    trim -= 2*SHR16(tf_estimate, 14-8);
Chris@69: #ifndef DISABLE_FLOAT_API
Chris@69:    if (analysis->valid)
Chris@69:    {
Chris@69:       trim -= MAX16(-QCONST16(2.f, 8), MIN16(QCONST16(2.f, 8),
Chris@69:             (opus_val16)(QCONST16(2.f, 8)*(analysis->tonality_slope+.05f))));
Chris@69:    }
Chris@69: #else
Chris@69:    (void)analysis;
Chris@69: #endif
Chris@69: 
Chris@69: #ifdef FIXED_POINT
Chris@69:    trim_index = PSHR32(trim, 8);
Chris@69: #else
Chris@69:    trim_index = (int)floor(.5f+trim);
Chris@69: #endif
Chris@69:    trim_index = IMAX(0, IMIN(10, trim_index));
Chris@69:    /*printf("%d\n", trim_index);*/
Chris@69: #ifdef FUZZING
Chris@69:    trim_index = rand()%11;
Chris@69: #endif
Chris@69:    return trim_index;
Chris@69: }
Chris@69: 
Chris@69: static int stereo_analysis(const CELTMode *m, const celt_norm *X,
Chris@69:       int LM, int N0)
Chris@69: {
Chris@69:    int i;
Chris@69:    int thetas;
Chris@69:    opus_val32 sumLR = EPSILON, sumMS = EPSILON;
Chris@69: 
Chris@69:    /* Use the L1 norm to model the entropy of the L/R signal vs the M/S signal */
Chris@69:    for (i=0;i<13;i++)
Chris@69:    {
Chris@69:       int j;
Chris@69:       for (j=m->eBands[i]<<LM;j<m->eBands[i+1]<<LM;j++)
Chris@69:       {
Chris@69:          opus_val32 L, R, M, S;
Chris@69:          /* We cast to 32-bit first because of the -32768 case */
Chris@69:          L = EXTEND32(X[j]);
Chris@69:          R = EXTEND32(X[N0+j]);
Chris@69:          M = ADD32(L, R);
Chris@69:          S = SUB32(L, R);
Chris@69:          sumLR = ADD32(sumLR, ADD32(ABS32(L), ABS32(R)));
Chris@69:          sumMS = ADD32(sumMS, ADD32(ABS32(M), ABS32(S)));
Chris@69:       }
Chris@69:    }
Chris@69:    sumMS = MULT16_32_Q15(QCONST16(0.707107f, 15), sumMS);
Chris@69:    thetas = 13;
Chris@69:    /* We don't need thetas for lower bands with LM<=1 */
Chris@69:    if (LM<=1)
Chris@69:       thetas -= 8;
Chris@69:    return MULT16_32_Q15((m->eBands[13]<<(LM+1))+thetas, sumMS)
Chris@69:          > MULT16_32_Q15(m->eBands[13]<<(LM+1), sumLR);
Chris@69: }
Chris@69: 
Chris@69: #define MSWAP(a,b) do {opus_val16 tmp = a;a=b;b=tmp;} while(0)
Chris@69: static opus_val16 median_of_5(const opus_val16 *x)
Chris@69: {
Chris@69:    opus_val16 t0, t1, t2, t3, t4;
Chris@69:    t2 = x[2];
Chris@69:    if (x[0] > x[1])
Chris@69:    {
Chris@69:       t0 = x[1];
Chris@69:       t1 = x[0];
Chris@69:    } else {
Chris@69:       t0 = x[0];
Chris@69:       t1 = x[1];
Chris@69:    }
Chris@69:    if (x[3] > x[4])
Chris@69:    {
Chris@69:       t3 = x[4];
Chris@69:       t4 = x[3];
Chris@69:    } else {
Chris@69:       t3 = x[3];
Chris@69:       t4 = x[4];
Chris@69:    }
Chris@69:    if (t0 > t3)
Chris@69:    {
Chris@69:       MSWAP(t0, t3);
Chris@69:       MSWAP(t1, t4);
Chris@69:    }
Chris@69:    if (t2 > t1)
Chris@69:    {
Chris@69:       if (t1 < t3)
Chris@69:          return MIN16(t2, t3);
Chris@69:       else
Chris@69:          return MIN16(t4, t1);
Chris@69:    } else {
Chris@69:       if (t2 < t3)
Chris@69:          return MIN16(t1, t3);
Chris@69:       else
Chris@69:          return MIN16(t2, t4);
Chris@69:    }
Chris@69: }
Chris@69: 
Chris@69: static opus_val16 median_of_3(const opus_val16 *x)
Chris@69: {
Chris@69:    opus_val16 t0, t1, t2;
Chris@69:    if (x[0] > x[1])
Chris@69:    {
Chris@69:       t0 = x[1];
Chris@69:       t1 = x[0];
Chris@69:    } else {
Chris@69:       t0 = x[0];
Chris@69:       t1 = x[1];
Chris@69:    }
Chris@69:    t2 = x[2];
Chris@69:    if (t1 < t2)
Chris@69:       return t1;
Chris@69:    else if (t0 < t2)
Chris@69:       return t2;
Chris@69:    else
Chris@69:       return t0;
Chris@69: }
Chris@69: 
Chris@69: static opus_val16 dynalloc_analysis(const opus_val16 *bandLogE, const opus_val16 *bandLogE2,
Chris@69:       int nbEBands, int start, int end, int C, int *offsets, int lsb_depth, const opus_int16 *logN,
Chris@69:       int isTransient, int vbr, int constrained_vbr, const opus_int16 *eBands, int LM,
Chris@69:       int effectiveBytes, opus_int32 *tot_boost_, int lfe, opus_val16 *surround_dynalloc,
Chris@69:       AnalysisInfo *analysis, int *importance, int *spread_weight)
Chris@69: {
Chris@69:    int i, c;
Chris@69:    opus_int32 tot_boost=0;
Chris@69:    opus_val16 maxDepth;
Chris@69:    VARDECL(opus_val16, follower);
Chris@69:    VARDECL(opus_val16, noise_floor);
Chris@69:    SAVE_STACK;
Chris@69:    ALLOC(follower, C*nbEBands, opus_val16);
Chris@69:    ALLOC(noise_floor, C*nbEBands, opus_val16);
Chris@69:    OPUS_CLEAR(offsets, nbEBands);
Chris@69:    /* Dynamic allocation code */
Chris@69:    maxDepth=-QCONST16(31.9f, DB_SHIFT);
Chris@69:    for (i=0;i<end;i++)
Chris@69:    {
Chris@69:       /* Noise floor must take into account eMeans, the depth, the width of the bands
Chris@69:          and the preemphasis filter (approx. square of bark band ID) */
Chris@69:       noise_floor[i] = MULT16_16(QCONST16(0.0625f, DB_SHIFT),logN[i])
Chris@69:             +QCONST16(.5f,DB_SHIFT)+SHL16(9-lsb_depth,DB_SHIFT)-SHL16(eMeans[i],6)
Chris@69:             +MULT16_16(QCONST16(.0062,DB_SHIFT),(i+5)*(i+5));
Chris@69:    }
Chris@69:    c=0;do
Chris@69:    {
Chris@69:       for (i=0;i<end;i++)
Chris@69:          maxDepth = MAX16(maxDepth, bandLogE[c*nbEBands+i]-noise_floor[i]);
Chris@69:    } while (++c<C);
Chris@69:    {
Chris@69:       /* Compute a really simple masking model to avoid taking into account completely masked
Chris@69:          bands when computing the spreading decision. */
Chris@69:       VARDECL(opus_val16, mask);
Chris@69:       VARDECL(opus_val16, sig);
Chris@69:       ALLOC(mask, nbEBands, opus_val16);
Chris@69:       ALLOC(sig, nbEBands, opus_val16);
Chris@69:       for (i=0;i<end;i++)
Chris@69:          mask[i] = bandLogE[i]-noise_floor[i];
Chris@69:       if (C==2)
Chris@69:       {
Chris@69:          for (i=0;i<end;i++)
Chris@69:             mask[i] = MAX16(mask[i], bandLogE[nbEBands+i]-noise_floor[i]);
Chris@69:       }
Chris@69:       OPUS_COPY(sig, mask, end);
Chris@69:       for (i=1;i<end;i++)
Chris@69:          mask[i] = MAX16(mask[i], mask[i-1] - QCONST16(2.f, DB_SHIFT));
Chris@69:       for (i=end-2;i>=0;i--)
Chris@69:          mask[i] = MAX16(mask[i], mask[i+1] - QCONST16(3.f, DB_SHIFT));
Chris@69:       for (i=0;i<end;i++)
Chris@69:       {
Chris@69:          /* Compute SMR: Mask is never more than 72 dB below the peak and never below the noise floor.*/
Chris@69:          opus_val16 smr = sig[i]-MAX16(MAX16(0, maxDepth-QCONST16(12.f, DB_SHIFT)), mask[i]);
Chris@69:          /* Clamp SMR to make sure we're not shifting by something negative or too large. */
Chris@69: #ifdef FIXED_POINT
Chris@69:          /* FIXME: Use PSHR16() instead */
Chris@69:          int shift = -PSHR32(MAX16(-QCONST16(5.f, DB_SHIFT), MIN16(0, smr)), DB_SHIFT);
Chris@69: #else
Chris@69:          int shift = IMIN(5, IMAX(0, -(int)floor(.5f + smr)));
Chris@69: #endif
Chris@69:          spread_weight[i] = 32 >> shift;
Chris@69:       }
Chris@69:       /*for (i=0;i<end;i++)
Chris@69:          printf("%d ", spread_weight[i]);
Chris@69:       printf("\n");*/
Chris@69:    }
Chris@69:    /* Make sure that dynamic allocation can't make us bust the budget */
Chris@69:    if (effectiveBytes > 50 && LM>=1 && !lfe)
Chris@69:    {
Chris@69:       int last=0;
Chris@69:       c=0;do
Chris@69:       {
Chris@69:          opus_val16 offset;
Chris@69:          opus_val16 tmp;
Chris@69:          opus_val16 *f;
Chris@69:          f = &follower[c*nbEBands];
Chris@69:          f[0] = bandLogE2[c*nbEBands];
Chris@69:          for (i=1;i<end;i++)
Chris@69:          {
Chris@69:             /* The last band to be at least 3 dB higher than the previous one
Chris@69:                is the last we'll consider. Otherwise, we run into problems on
Chris@69:                bandlimited signals. */
Chris@69:             if (bandLogE2[c*nbEBands+i] > bandLogE2[c*nbEBands+i-1]+QCONST16(.5f,DB_SHIFT))
Chris@69:                last=i;
Chris@69:             f[i] = MIN16(f[i-1]+QCONST16(1.5f,DB_SHIFT), bandLogE2[c*nbEBands+i]);
Chris@69:          }
Chris@69:          for (i=last-1;i>=0;i--)
Chris@69:             f[i] = MIN16(f[i], MIN16(f[i+1]+QCONST16(2.f,DB_SHIFT), bandLogE2[c*nbEBands+i]));
Chris@69: 
Chris@69:          /* Combine with a median filter to avoid dynalloc triggering unnecessarily.
Chris@69:             The "offset" value controls how conservative we are -- a higher offset
Chris@69:             reduces the impact of the median filter and makes dynalloc use more bits. */
Chris@69:          offset = QCONST16(1.f, DB_SHIFT);
Chris@69:          for (i=2;i<end-2;i++)
Chris@69:             f[i] = MAX16(f[i], median_of_5(&bandLogE2[c*nbEBands+i-2])-offset);
Chris@69:          tmp = median_of_3(&bandLogE2[c*nbEBands])-offset;
Chris@69:          f[0] = MAX16(f[0], tmp);
Chris@69:          f[1] = MAX16(f[1], tmp);
Chris@69:          tmp = median_of_3(&bandLogE2[c*nbEBands+end-3])-offset;
Chris@69:          f[end-2] = MAX16(f[end-2], tmp);
Chris@69:          f[end-1] = MAX16(f[end-1], tmp);
Chris@69: 
Chris@69:          for (i=0;i<end;i++)
Chris@69:             f[i] = MAX16(f[i], noise_floor[i]);
Chris@69:       } while (++c<C);
Chris@69:       if (C==2)
Chris@69:       {
Chris@69:          for (i=start;i<end;i++)
Chris@69:          {
Chris@69:             /* Consider 24 dB "cross-talk" */
Chris@69:             follower[nbEBands+i] = MAX16(follower[nbEBands+i], follower[         i]-QCONST16(4.f,DB_SHIFT));
Chris@69:             follower[         i] = MAX16(follower[         i], follower[nbEBands+i]-QCONST16(4.f,DB_SHIFT));
Chris@69:             follower[i] = HALF16(MAX16(0, bandLogE[i]-follower[i]) + MAX16(0, bandLogE[nbEBands+i]-follower[nbEBands+i]));
Chris@69:          }
Chris@69:       } else {
Chris@69:          for (i=start;i<end;i++)
Chris@69:          {
Chris@69:             follower[i] = MAX16(0, bandLogE[i]-follower[i]);
Chris@69:          }
Chris@69:       }
Chris@69:       for (i=start;i<end;i++)
Chris@69:          follower[i] = MAX16(follower[i], surround_dynalloc[i]);
Chris@69:       for (i=start;i<end;i++)
Chris@69:       {
Chris@69: #ifdef FIXED_POINT
Chris@69:          importance[i] = PSHR32(13*celt_exp2(MIN16(follower[i], QCONST16(4.f, DB_SHIFT))), 16);
Chris@69: #else
Chris@69:          importance[i] = (int)floor(.5f+13*celt_exp2(MIN16(follower[i], QCONST16(4.f, DB_SHIFT))));
Chris@69: #endif
Chris@69:       }
Chris@69:       /* For non-transient CBR/CVBR frames, halve the dynalloc contribution */
Chris@69:       if ((!vbr || constrained_vbr)&&!isTransient)
Chris@69:       {
Chris@69:          for (i=start;i<end;i++)
Chris@69:             follower[i] = HALF16(follower[i]);
Chris@69:       }
Chris@69:       for (i=start;i<end;i++)
Chris@69:       {
Chris@69:          if (i<8)
Chris@69:             follower[i] *= 2;
Chris@69:          if (i>=12)
Chris@69:             follower[i] = HALF16(follower[i]);
Chris@69:       }
Chris@69: #ifdef DISABLE_FLOAT_API
Chris@69:       (void)analysis;
Chris@69: #else
Chris@69:       if (analysis->valid)
Chris@69:       {
Chris@69:          for (i=start;i<IMIN(LEAK_BANDS, end);i++)
Chris@69:             follower[i] = follower[i] +  QCONST16(1.f/64.f, DB_SHIFT)*analysis->leak_boost[i];
Chris@69:       }
Chris@69: #endif
Chris@69:       for (i=start;i<end;i++)
Chris@69:       {
Chris@69:          int width;
Chris@69:          int boost;
Chris@69:          int boost_bits;
Chris@69: 
Chris@69:          follower[i] = MIN16(follower[i], QCONST16(4, DB_SHIFT));
Chris@69: 
Chris@69:          width = C*(eBands[i+1]-eBands[i])<<LM;
Chris@69:          if (width<6)
Chris@69:          {
Chris@69:             boost = (int)SHR32(EXTEND32(follower[i]),DB_SHIFT);
Chris@69:             boost_bits = boost*width<<BITRES;
Chris@69:          } else if (width > 48) {
Chris@69:             boost = (int)SHR32(EXTEND32(follower[i])*8,DB_SHIFT);
Chris@69:             boost_bits = (boost*width<<BITRES)/8;
Chris@69:          } else {
Chris@69:             boost = (int)SHR32(EXTEND32(follower[i])*width/6,DB_SHIFT);
Chris@69:             boost_bits = boost*6<<BITRES;
Chris@69:          }
Chris@69:          /* For CBR and non-transient CVBR frames, limit dynalloc to 2/3 of the bits */
Chris@69:          if ((!vbr || (constrained_vbr&&!isTransient))
Chris@69:                && (tot_boost+boost_bits)>>BITRES>>3 > 2*effectiveBytes/3)
Chris@69:          {
Chris@69:             opus_int32 cap = ((2*effectiveBytes/3)<<BITRES<<3);
Chris@69:             offsets[i] = cap-tot_boost;
Chris@69:             tot_boost = cap;
Chris@69:             break;
Chris@69:          } else {
Chris@69:             offsets[i] = boost;
Chris@69:             tot_boost += boost_bits;
Chris@69:          }
Chris@69:       }
Chris@69:    } else {
Chris@69:       for (i=start;i<end;i++)
Chris@69:          importance[i] = 13;
Chris@69:    }
Chris@69:    *tot_boost_ = tot_boost;
Chris@69:    RESTORE_STACK;
Chris@69:    return maxDepth;
Chris@69: }
Chris@69: 
Chris@69: 
Chris@69: static int run_prefilter(CELTEncoder *st, celt_sig *in, celt_sig *prefilter_mem, int CC, int N,
Chris@69:       int prefilter_tapset, int *pitch, opus_val16 *gain, int *qgain, int enabled, int nbAvailableBytes, AnalysisInfo *analysis)
Chris@69: {
Chris@69:    int c;
Chris@69:    VARDECL(celt_sig, _pre);
Chris@69:    celt_sig *pre[2];
Chris@69:    const CELTMode *mode;
Chris@69:    int pitch_index;
Chris@69:    opus_val16 gain1;
Chris@69:    opus_val16 pf_threshold;
Chris@69:    int pf_on;
Chris@69:    int qg;
Chris@69:    int overlap;
Chris@69:    SAVE_STACK;
Chris@69: 
Chris@69:    mode = st->mode;
Chris@69:    overlap = mode->overlap;
Chris@69:    ALLOC(_pre, CC*(N+COMBFILTER_MAXPERIOD), celt_sig);
Chris@69: 
Chris@69:    pre[0] = _pre;
Chris@69:    pre[1] = _pre + (N+COMBFILTER_MAXPERIOD);
Chris@69: 
Chris@69: 
Chris@69:    c=0; do {
Chris@69:       OPUS_COPY(pre[c], prefilter_mem+c*COMBFILTER_MAXPERIOD, COMBFILTER_MAXPERIOD);
Chris@69:       OPUS_COPY(pre[c]+COMBFILTER_MAXPERIOD, in+c*(N+overlap)+overlap, N);
Chris@69:    } while (++c<CC);
Chris@69: 
Chris@69:    if (enabled)
Chris@69:    {
Chris@69:       VARDECL(opus_val16, pitch_buf);
Chris@69:       ALLOC(pitch_buf, (COMBFILTER_MAXPERIOD+N)>>1, opus_val16);
Chris@69: 
Chris@69:       pitch_downsample(pre, pitch_buf, COMBFILTER_MAXPERIOD+N, CC, st->arch);
Chris@69:       /* Don't search for the fir last 1.5 octave of the range because
Chris@69:          there's too many false-positives due to short-term correlation */
Chris@69:       pitch_search(pitch_buf+(COMBFILTER_MAXPERIOD>>1), pitch_buf, N,
Chris@69:             COMBFILTER_MAXPERIOD-3*COMBFILTER_MINPERIOD, &pitch_index,
Chris@69:             st->arch);
Chris@69:       pitch_index = COMBFILTER_MAXPERIOD-pitch_index;
Chris@69: 
Chris@69:       gain1 = remove_doubling(pitch_buf, COMBFILTER_MAXPERIOD, COMBFILTER_MINPERIOD,
Chris@69:             N, &pitch_index, st->prefilter_period, st->prefilter_gain, st->arch);
Chris@69:       if (pitch_index > COMBFILTER_MAXPERIOD-2)
Chris@69:          pitch_index = COMBFILTER_MAXPERIOD-2;
Chris@69:       gain1 = MULT16_16_Q15(QCONST16(.7f,15),gain1);
Chris@69:       /*printf("%d %d %f %f\n", pitch_change, pitch_index, gain1, st->analysis.tonality);*/
Chris@69:       if (st->loss_rate>2)
Chris@69:          gain1 = HALF32(gain1);
Chris@69:       if (st->loss_rate>4)
Chris@69:          gain1 = HALF32(gain1);
Chris@69:       if (st->loss_rate>8)
Chris@69:          gain1 = 0;
Chris@69:    } else {
Chris@69:       gain1 = 0;
Chris@69:       pitch_index = COMBFILTER_MINPERIOD;
Chris@69:    }
Chris@69: #ifndef DISABLE_FLOAT_API
Chris@69:    if (analysis->valid)
Chris@69:       gain1 = (opus_val16)(gain1 * analysis->max_pitch_ratio);
Chris@69: #else
Chris@69:    (void)analysis;
Chris@69: #endif
Chris@69:    /* Gain threshold for enabling the prefilter/postfilter */
Chris@69:    pf_threshold = QCONST16(.2f,15);
Chris@69: 
Chris@69:    /* Adjusting the threshold based on rate and continuity */
Chris@69:    if (abs(pitch_index-st->prefilter_period)*10>pitch_index)
Chris@69:       pf_threshold += QCONST16(.2f,15);
Chris@69:    if (nbAvailableBytes<25)
Chris@69:       pf_threshold += QCONST16(.1f,15);
Chris@69:    if (nbAvailableBytes<35)
Chris@69:       pf_threshold += QCONST16(.1f,15);
Chris@69:    if (st->prefilter_gain > QCONST16(.4f,15))
Chris@69:       pf_threshold -= QCONST16(.1f,15);
Chris@69:    if (st->prefilter_gain > QCONST16(.55f,15))
Chris@69:       pf_threshold -= QCONST16(.1f,15);
Chris@69: 
Chris@69:    /* Hard threshold at 0.2 */
Chris@69:    pf_threshold = MAX16(pf_threshold, QCONST16(.2f,15));
Chris@69:    if (gain1<pf_threshold)
Chris@69:    {
Chris@69:       gain1 = 0;
Chris@69:       pf_on = 0;
Chris@69:       qg = 0;
Chris@69:    } else {
Chris@69:       /*This block is not gated by a total bits check only because
Chris@69:         of the nbAvailableBytes check above.*/
Chris@69:       if (ABS16(gain1-st->prefilter_gain)<QCONST16(.1f,15))
Chris@69:          gain1=st->prefilter_gain;
Chris@69: 
Chris@69: #ifdef FIXED_POINT
Chris@69:       qg = ((gain1+1536)>>10)/3-1;
Chris@69: #else
Chris@69:       qg = (int)floor(.5f+gain1*32/3)-1;
Chris@69: #endif
Chris@69:       qg = IMAX(0, IMIN(7, qg));
Chris@69:       gain1 = QCONST16(0.09375f,15)*(qg+1);
Chris@69:       pf_on = 1;
Chris@69:    }
Chris@69:    /*printf("%d %f\n", pitch_index, gain1);*/
Chris@69: 
Chris@69:    c=0; do {
Chris@69:       int offset = mode->shortMdctSize-overlap;
Chris@69:       st->prefilter_period=IMAX(st->prefilter_period, COMBFILTER_MINPERIOD);
Chris@69:       OPUS_COPY(in+c*(N+overlap), st->in_mem+c*(overlap), overlap);
Chris@69:       if (offset)
Chris@69:          comb_filter(in+c*(N+overlap)+overlap, pre[c]+COMBFILTER_MAXPERIOD,
Chris@69:                st->prefilter_period, st->prefilter_period, offset, -st->prefilter_gain, -st->prefilter_gain,
Chris@69:                st->prefilter_tapset, st->prefilter_tapset, NULL, 0, st->arch);
Chris@69: 
Chris@69:       comb_filter(in+c*(N+overlap)+overlap+offset, pre[c]+COMBFILTER_MAXPERIOD+offset,
Chris@69:             st->prefilter_period, pitch_index, N-offset, -st->prefilter_gain, -gain1,
Chris@69:             st->prefilter_tapset, prefilter_tapset, mode->window, overlap, st->arch);
Chris@69:       OPUS_COPY(st->in_mem+c*(overlap), in+c*(N+overlap)+N, overlap);
Chris@69: 
Chris@69:       if (N>COMBFILTER_MAXPERIOD)
Chris@69:       {
Chris@69:          OPUS_COPY(prefilter_mem+c*COMBFILTER_MAXPERIOD, pre[c]+N, COMBFILTER_MAXPERIOD);
Chris@69:       } else {
Chris@69:          OPUS_MOVE(prefilter_mem+c*COMBFILTER_MAXPERIOD, prefilter_mem+c*COMBFILTER_MAXPERIOD+N, COMBFILTER_MAXPERIOD-N);
Chris@69:          OPUS_COPY(prefilter_mem+c*COMBFILTER_MAXPERIOD+COMBFILTER_MAXPERIOD-N, pre[c]+COMBFILTER_MAXPERIOD, N);
Chris@69:       }
Chris@69:    } while (++c<CC);
Chris@69: 
Chris@69:    RESTORE_STACK;
Chris@69:    *gain = gain1;
Chris@69:    *pitch = pitch_index;
Chris@69:    *qgain = qg;
Chris@69:    return pf_on;
Chris@69: }
Chris@69: 
Chris@69: static int compute_vbr(const CELTMode *mode, AnalysisInfo *analysis, opus_int32 base_target,
Chris@69:       int LM, opus_int32 bitrate, int lastCodedBands, int C, int intensity,
Chris@69:       int constrained_vbr, opus_val16 stereo_saving, int tot_boost,
Chris@69:       opus_val16 tf_estimate, int pitch_change, opus_val16 maxDepth,
Chris@69:       int lfe, int has_surround_mask, opus_val16 surround_masking,
Chris@69:       opus_val16 temporal_vbr)
Chris@69: {
Chris@69:    /* The target rate in 8th bits per frame */
Chris@69:    opus_int32 target;
Chris@69:    int coded_bins;
Chris@69:    int coded_bands;
Chris@69:    opus_val16 tf_calibration;
Chris@69:    int nbEBands;
Chris@69:    const opus_int16 *eBands;
Chris@69: 
Chris@69:    nbEBands = mode->nbEBands;
Chris@69:    eBands = mode->eBands;
Chris@69: 
Chris@69:    coded_bands = lastCodedBands ? lastCodedBands : nbEBands;
Chris@69:    coded_bins = eBands[coded_bands]<<LM;
Chris@69:    if (C==2)
Chris@69:       coded_bins += eBands[IMIN(intensity, coded_bands)]<<LM;
Chris@69: 
Chris@69:    target = base_target;
Chris@69: 
Chris@69:    /*printf("%f %f %f %f %d %d ", st->analysis.activity, st->analysis.tonality, tf_estimate, st->stereo_saving, tot_boost, coded_bands);*/
Chris@69: #ifndef DISABLE_FLOAT_API
Chris@69:    if (analysis->valid && analysis->activity<.4)
Chris@69:       target -= (opus_int32)((coded_bins<<BITRES)*(.4f-analysis->activity));
Chris@69: #endif
Chris@69:    /* Stereo savings */
Chris@69:    if (C==2)
Chris@69:    {
Chris@69:       int coded_stereo_bands;
Chris@69:       int coded_stereo_dof;
Chris@69:       opus_val16 max_frac;
Chris@69:       coded_stereo_bands = IMIN(intensity, coded_bands);
Chris@69:       coded_stereo_dof = (eBands[coded_stereo_bands]<<LM)-coded_stereo_bands;
Chris@69:       /* Maximum fraction of the bits we can save if the signal is mono. */
Chris@69:       max_frac = DIV32_16(MULT16_16(QCONST16(0.8f, 15), coded_stereo_dof), coded_bins);
Chris@69:       stereo_saving = MIN16(stereo_saving, QCONST16(1.f, 8));
Chris@69:       /*printf("%d %d %d ", coded_stereo_dof, coded_bins, tot_boost);*/
Chris@69:       target -= (opus_int32)MIN32(MULT16_32_Q15(max_frac,target),
Chris@69:                       SHR32(MULT16_16(stereo_saving-QCONST16(0.1f,8),(coded_stereo_dof<<BITRES)),8));
Chris@69:    }
Chris@69:    /* Boost the rate according to dynalloc (minus the dynalloc average for calibration). */
Chris@69:    target += tot_boost-(19<<LM);
Chris@69:    /* Apply transient boost, compensating for average boost. */
Chris@69:    tf_calibration = QCONST16(0.044f,14);
Chris@69:    target += (opus_int32)SHL32(MULT16_32_Q15(tf_estimate-tf_calibration, target),1);
Chris@69: 
Chris@69: #ifndef DISABLE_FLOAT_API
Chris@69:    /* Apply tonality boost */
Chris@69:    if (analysis->valid && !lfe)
Chris@69:    {
Chris@69:       opus_int32 tonal_target;
Chris@69:       float tonal;
Chris@69: 
Chris@69:       /* Tonality boost (compensating for the average). */
Chris@69:       tonal = MAX16(0.f,analysis->tonality-.15f)-0.12f;
Chris@69:       tonal_target = target + (opus_int32)((coded_bins<<BITRES)*1.2f*tonal);
Chris@69:       if (pitch_change)
Chris@69:          tonal_target +=  (opus_int32)((coded_bins<<BITRES)*.8f);
Chris@69:       /*printf("%f %f ", analysis->tonality, tonal);*/
Chris@69:       target = tonal_target;
Chris@69:    }
Chris@69: #else
Chris@69:    (void)analysis;
Chris@69:    (void)pitch_change;
Chris@69: #endif
Chris@69: 
Chris@69:    if (has_surround_mask&&!lfe)
Chris@69:    {
Chris@69:       opus_int32 surround_target = target + (opus_int32)SHR32(MULT16_16(surround_masking,coded_bins<<BITRES), DB_SHIFT);
Chris@69:       /*printf("%f %d %d %d %d %d %d ", surround_masking, coded_bins, st->end, st->intensity, surround_target, target, st->bitrate);*/
Chris@69:       target = IMAX(target/4, surround_target);
Chris@69:    }
Chris@69: 
Chris@69:    {
Chris@69:       opus_int32 floor_depth;
Chris@69:       int bins;
Chris@69:       bins = eBands[nbEBands-2]<<LM;
Chris@69:       /*floor_depth = SHR32(MULT16_16((C*bins<<BITRES),celt_log2(SHL32(MAX16(1,sample_max),13))), DB_SHIFT);*/
Chris@69:       floor_depth = (opus_int32)SHR32(MULT16_16((C*bins<<BITRES),maxDepth), DB_SHIFT);
Chris@69:       floor_depth = IMAX(floor_depth, target>>2);
Chris@69:       target = IMIN(target, floor_depth);
Chris@69:       /*printf("%f %d\n", maxDepth, floor_depth);*/
Chris@69:    }
Chris@69: 
Chris@69:    /* Make VBR less aggressive for constrained VBR because we can't keep a higher bitrate
Chris@69:       for long. Needs tuning. */
Chris@69:    if ((!has_surround_mask||lfe) && constrained_vbr)
Chris@69:    {
Chris@69:       target = base_target + (opus_int32)MULT16_32_Q15(QCONST16(0.67f, 15), target-base_target);
Chris@69:    }
Chris@69: 
Chris@69:    if (!has_surround_mask && tf_estimate < QCONST16(.2f, 14))
Chris@69:    {
Chris@69:       opus_val16 amount;
Chris@69:       opus_val16 tvbr_factor;
Chris@69:       amount = MULT16_16_Q15(QCONST16(.0000031f, 30), IMAX(0, IMIN(32000, 96000-bitrate)));
Chris@69:       tvbr_factor = SHR32(MULT16_16(temporal_vbr, amount), DB_SHIFT);
Chris@69:       target += (opus_int32)MULT16_32_Q15(tvbr_factor, target);
Chris@69:    }
Chris@69: 
Chris@69:    /* Don't allow more than doubling the rate */
Chris@69:    target = IMIN(2*base_target, target);
Chris@69: 
Chris@69:    return target;
Chris@69: }
Chris@69: 
Chris@69: int celt_encode_with_ec(CELTEncoder * OPUS_RESTRICT st, const opus_val16 * pcm, int frame_size, unsigned char *compressed, int nbCompressedBytes, ec_enc *enc)
Chris@69: {
Chris@69:    int i, c, N;
Chris@69:    opus_int32 bits;
Chris@69:    ec_enc _enc;
Chris@69:    VARDECL(celt_sig, in);
Chris@69:    VARDECL(celt_sig, freq);
Chris@69:    VARDECL(celt_norm, X);
Chris@69:    VARDECL(celt_ener, bandE);
Chris@69:    VARDECL(opus_val16, bandLogE);
Chris@69:    VARDECL(opus_val16, bandLogE2);
Chris@69:    VARDECL(int, fine_quant);
Chris@69:    VARDECL(opus_val16, error);
Chris@69:    VARDECL(int, pulses);
Chris@69:    VARDECL(int, cap);
Chris@69:    VARDECL(int, offsets);
Chris@69:    VARDECL(int, importance);
Chris@69:    VARDECL(int, spread_weight);
Chris@69:    VARDECL(int, fine_priority);
Chris@69:    VARDECL(int, tf_res);
Chris@69:    VARDECL(unsigned char, collapse_masks);
Chris@69:    celt_sig *prefilter_mem;
Chris@69:    opus_val16 *oldBandE, *oldLogE, *oldLogE2, *energyError;
Chris@69:    int shortBlocks=0;
Chris@69:    int isTransient=0;
Chris@69:    const int CC = st->channels;
Chris@69:    const int C = st->stream_channels;
Chris@69:    int LM, M;
Chris@69:    int tf_select;
Chris@69:    int nbFilledBytes, nbAvailableBytes;
Chris@69:    int start;
Chris@69:    int end;
Chris@69:    int effEnd;
Chris@69:    int codedBands;
Chris@69:    int alloc_trim;
Chris@69:    int pitch_index=COMBFILTER_MINPERIOD;
Chris@69:    opus_val16 gain1 = 0;
Chris@69:    int dual_stereo=0;
Chris@69:    int effectiveBytes;
Chris@69:    int dynalloc_logp;
Chris@69:    opus_int32 vbr_rate;
Chris@69:    opus_int32 total_bits;
Chris@69:    opus_int32 total_boost;
Chris@69:    opus_int32 balance;
Chris@69:    opus_int32 tell;
Chris@69:    opus_int32 tell0_frac;
Chris@69:    int prefilter_tapset=0;
Chris@69:    int pf_on;
Chris@69:    int anti_collapse_rsv;
Chris@69:    int anti_collapse_on=0;
Chris@69:    int silence=0;
Chris@69:    int tf_chan = 0;
Chris@69:    opus_val16 tf_estimate;
Chris@69:    int pitch_change=0;
Chris@69:    opus_int32 tot_boost;
Chris@69:    opus_val32 sample_max;
Chris@69:    opus_val16 maxDepth;
Chris@69:    const OpusCustomMode *mode;
Chris@69:    int nbEBands;
Chris@69:    int overlap;
Chris@69:    const opus_int16 *eBands;
Chris@69:    int secondMdct;
Chris@69:    int signalBandwidth;
Chris@69:    int transient_got_disabled=0;
Chris@69:    opus_val16 surround_masking=0;
Chris@69:    opus_val16 temporal_vbr=0;
Chris@69:    opus_val16 surround_trim = 0;
Chris@69:    opus_int32 equiv_rate;
Chris@69:    int hybrid;
Chris@69:    int weak_transient = 0;
Chris@69:    int enable_tf_analysis;
Chris@69:    VARDECL(opus_val16, surround_dynalloc);
Chris@69:    ALLOC_STACK;
Chris@69: 
Chris@69:    mode = st->mode;
Chris@69:    nbEBands = mode->nbEBands;
Chris@69:    overlap = mode->overlap;
Chris@69:    eBands = mode->eBands;
Chris@69:    start = st->start;
Chris@69:    end = st->end;
Chris@69:    hybrid = start != 0;
Chris@69:    tf_estimate = 0;
Chris@69:    if (nbCompressedBytes<2 || pcm==NULL)
Chris@69:    {
Chris@69:       RESTORE_STACK;
Chris@69:       return OPUS_BAD_ARG;
Chris@69:    }
Chris@69: 
Chris@69:    frame_size *= st->upsample;
Chris@69:    for (LM=0;LM<=mode->maxLM;LM++)
Chris@69:       if (mode->shortMdctSize<<LM==frame_size)
Chris@69:          break;
Chris@69:    if (LM>mode->maxLM)
Chris@69:    {
Chris@69:       RESTORE_STACK;
Chris@69:       return OPUS_BAD_ARG;
Chris@69:    }
Chris@69:    M=1<<LM;
Chris@69:    N = M*mode->shortMdctSize;
Chris@69: 
Chris@69:    prefilter_mem = st->in_mem+CC*(overlap);
Chris@69:    oldBandE = (opus_val16*)(st->in_mem+CC*(overlap+COMBFILTER_MAXPERIOD));
Chris@69:    oldLogE = oldBandE + CC*nbEBands;
Chris@69:    oldLogE2 = oldLogE + CC*nbEBands;
Chris@69:    energyError = oldLogE2 + CC*nbEBands;
Chris@69: 
Chris@69:    if (enc==NULL)
Chris@69:    {
Chris@69:       tell0_frac=tell=1;
Chris@69:       nbFilledBytes=0;
Chris@69:    } else {
Chris@69:       tell0_frac=ec_tell_frac(enc);
Chris@69:       tell=ec_tell(enc);
Chris@69:       nbFilledBytes=(tell+4)>>3;
Chris@69:    }
Chris@69: 
Chris@69: #ifdef CUSTOM_MODES
Chris@69:    if (st->signalling && enc==NULL)
Chris@69:    {
Chris@69:       int tmp = (mode->effEBands-end)>>1;
Chris@69:       end = st->end = IMAX(1, mode->effEBands-tmp);
Chris@69:       compressed[0] = tmp<<5;
Chris@69:       compressed[0] |= LM<<3;
Chris@69:       compressed[0] |= (C==2)<<2;
Chris@69:       /* Convert "standard mode" to Opus header */
Chris@69:       if (mode->Fs==48000 && mode->shortMdctSize==120)
Chris@69:       {
Chris@69:          int c0 = toOpus(compressed[0]);
Chris@69:          if (c0<0)
Chris@69:          {
Chris@69:             RESTORE_STACK;
Chris@69:             return OPUS_BAD_ARG;
Chris@69:          }
Chris@69:          compressed[0] = c0;
Chris@69:       }
Chris@69:       compressed++;
Chris@69:       nbCompressedBytes--;
Chris@69:    }
Chris@69: #else
Chris@69:    celt_assert(st->signalling==0);
Chris@69: #endif
Chris@69: 
Chris@69:    /* Can't produce more than 1275 output bytes */
Chris@69:    nbCompressedBytes = IMIN(nbCompressedBytes,1275);
Chris@69:    nbAvailableBytes = nbCompressedBytes - nbFilledBytes;
Chris@69: 
Chris@69:    if (st->vbr && st->bitrate!=OPUS_BITRATE_MAX)
Chris@69:    {
Chris@69:       opus_int32 den=mode->Fs>>BITRES;
Chris@69:       vbr_rate=(st->bitrate*frame_size+(den>>1))/den;
Chris@69: #ifdef CUSTOM_MODES
Chris@69:       if (st->signalling)
Chris@69:          vbr_rate -= 8<<BITRES;
Chris@69: #endif
Chris@69:       effectiveBytes = vbr_rate>>(3+BITRES);
Chris@69:    } else {
Chris@69:       opus_int32 tmp;
Chris@69:       vbr_rate = 0;
Chris@69:       tmp = st->bitrate*frame_size;
Chris@69:       if (tell>1)
Chris@69:          tmp += tell;
Chris@69:       if (st->bitrate!=OPUS_BITRATE_MAX)
Chris@69:          nbCompressedBytes = IMAX(2, IMIN(nbCompressedBytes,
Chris@69:                (tmp+4*mode->Fs)/(8*mode->Fs)-!!st->signalling));
Chris@69:       effectiveBytes = nbCompressedBytes - nbFilledBytes;
Chris@69:    }
Chris@69:    equiv_rate = ((opus_int32)nbCompressedBytes*8*50 >> (3-LM)) - (40*C+20)*((400>>LM) - 50);
Chris@69:    if (st->bitrate != OPUS_BITRATE_MAX)
Chris@69:       equiv_rate = IMIN(equiv_rate, st->bitrate - (40*C+20)*((400>>LM) - 50));
Chris@69: 
Chris@69:    if (enc==NULL)
Chris@69:    {
Chris@69:       ec_enc_init(&_enc, compressed, nbCompressedBytes);
Chris@69:       enc = &_enc;
Chris@69:    }
Chris@69: 
Chris@69:    if (vbr_rate>0)
Chris@69:    {
Chris@69:       /* Computes the max bit-rate allowed in VBR mode to avoid violating the
Chris@69:           target rate and buffering.
Chris@69:          We must do this up front so that bust-prevention logic triggers
Chris@69:           correctly if we don't have enough bits. */
Chris@69:       if (st->constrained_vbr)
Chris@69:       {
Chris@69:          opus_int32 vbr_bound;
Chris@69:          opus_int32 max_allowed;
Chris@69:          /* We could use any multiple of vbr_rate as bound (depending on the
Chris@69:              delay).
Chris@69:             This is clamped to ensure we use at least two bytes if the encoder
Chris@69:              was entirely empty, but to allow 0 in hybrid mode. */
Chris@69:          vbr_bound = vbr_rate;
Chris@69:          max_allowed = IMIN(IMAX(tell==1?2:0,
Chris@69:                (vbr_rate+vbr_bound-st->vbr_reservoir)>>(BITRES+3)),
Chris@69:                nbAvailableBytes);
Chris@69:          if(max_allowed < nbAvailableBytes)
Chris@69:          {
Chris@69:             nbCompressedBytes = nbFilledBytes+max_allowed;
Chris@69:             nbAvailableBytes = max_allowed;
Chris@69:             ec_enc_shrink(enc, nbCompressedBytes);
Chris@69:          }
Chris@69:       }
Chris@69:    }
Chris@69:    total_bits = nbCompressedBytes*8;
Chris@69: 
Chris@69:    effEnd = end;
Chris@69:    if (effEnd > mode->effEBands)
Chris@69:       effEnd = mode->effEBands;
Chris@69: 
Chris@69:    ALLOC(in, CC*(N+overlap), celt_sig);
Chris@69: 
Chris@69:    sample_max=MAX32(st->overlap_max, celt_maxabs16(pcm, C*(N-overlap)/st->upsample));
Chris@69:    st->overlap_max=celt_maxabs16(pcm+C*(N-overlap)/st->upsample, C*overlap/st->upsample);
Chris@69:    sample_max=MAX32(sample_max, st->overlap_max);
Chris@69: #ifdef FIXED_POINT
Chris@69:    silence = (sample_max==0);
Chris@69: #else
Chris@69:    silence = (sample_max <= (opus_val16)1/(1<<st->lsb_depth));
Chris@69: #endif
Chris@69: #ifdef FUZZING
Chris@69:    if ((rand()&0x3F)==0)
Chris@69:       silence = 1;
Chris@69: #endif
Chris@69:    if (tell==1)
Chris@69:       ec_enc_bit_logp(enc, silence, 15);
Chris@69:    else
Chris@69:       silence=0;
Chris@69:    if (silence)
Chris@69:    {
Chris@69:       /*In VBR mode there is no need to send more than the minimum. */
Chris@69:       if (vbr_rate>0)
Chris@69:       {
Chris@69:          effectiveBytes=nbCompressedBytes=IMIN(nbCompressedBytes, nbFilledBytes+2);
Chris@69:          total_bits=nbCompressedBytes*8;
Chris@69:          nbAvailableBytes=2;
Chris@69:          ec_enc_shrink(enc, nbCompressedBytes);
Chris@69:       }
Chris@69:       /* Pretend we've filled all the remaining bits with zeros
Chris@69:             (that's what the initialiser did anyway) */
Chris@69:       tell = nbCompressedBytes*8;
Chris@69:       enc->nbits_total+=tell-ec_tell(enc);
Chris@69:    }
Chris@69:    c=0; do {
Chris@69:       int need_clip=0;
Chris@69: #ifndef FIXED_POINT
Chris@69:       need_clip = st->clip && sample_max>65536.f;
Chris@69: #endif
Chris@69:       celt_preemphasis(pcm+c, in+c*(N+overlap)+overlap, N, CC, st->upsample,
Chris@69:                   mode->preemph, st->preemph_memE+c, need_clip);
Chris@69:    } while (++c<CC);
Chris@69: 
Chris@69: 
Chris@69: 
Chris@69:    /* Find pitch period and gain */
Chris@69:    {
Chris@69:       int enabled;
Chris@69:       int qg;
Chris@69:       enabled = ((st->lfe&&nbAvailableBytes>3) || nbAvailableBytes>12*C) && !hybrid && !silence && !st->disable_pf
Chris@69:             && st->complexity >= 5;
Chris@69: 
Chris@69:       prefilter_tapset = st->tapset_decision;
Chris@69:       pf_on = run_prefilter(st, in, prefilter_mem, CC, N, prefilter_tapset, &pitch_index, &gain1, &qg, enabled, nbAvailableBytes, &st->analysis);
Chris@69:       if ((gain1 > QCONST16(.4f,15) || st->prefilter_gain > QCONST16(.4f,15)) && (!st->analysis.valid || st->analysis.tonality > .3)
Chris@69:             && (pitch_index > 1.26*st->prefilter_period || pitch_index < .79*st->prefilter_period))
Chris@69:          pitch_change = 1;
Chris@69:       if (pf_on==0)
Chris@69:       {
Chris@69:          if(!hybrid && tell+16<=total_bits)
Chris@69:             ec_enc_bit_logp(enc, 0, 1);
Chris@69:       } else {
Chris@69:          /*This block is not gated by a total bits check only because
Chris@69:            of the nbAvailableBytes check above.*/
Chris@69:          int octave;
Chris@69:          ec_enc_bit_logp(enc, 1, 1);
Chris@69:          pitch_index += 1;
Chris@69:          octave = EC_ILOG(pitch_index)-5;
Chris@69:          ec_enc_uint(enc, octave, 6);
Chris@69:          ec_enc_bits(enc, pitch_index-(16<<octave), 4+octave);
Chris@69:          pitch_index -= 1;
Chris@69:          ec_enc_bits(enc, qg, 3);
Chris@69:          ec_enc_icdf(enc, prefilter_tapset, tapset_icdf, 2);
Chris@69:       }
Chris@69:    }
Chris@69: 
Chris@69:    isTransient = 0;
Chris@69:    shortBlocks = 0;
Chris@69:    if (st->complexity >= 1 && !st->lfe)
Chris@69:    {
Chris@69:       /* Reduces the likelihood of energy instability on fricatives at low bitrate
Chris@69:          in hybrid mode. It seems like we still want to have real transients on vowels
Chris@69:          though (small SILK quantization offset value). */
Chris@69:       int allow_weak_transients = hybrid && effectiveBytes<15 && st->silk_info.signalType != 2;
Chris@69:       isTransient = transient_analysis(in, N+overlap, CC,
Chris@69:             &tf_estimate, &tf_chan, allow_weak_transients, &weak_transient);
Chris@69:    }
Chris@69:    if (LM>0 && ec_tell(enc)+3<=total_bits)
Chris@69:    {
Chris@69:       if (isTransient)
Chris@69:          shortBlocks = M;
Chris@69:    } else {
Chris@69:       isTransient = 0;
Chris@69:       transient_got_disabled=1;
Chris@69:    }
Chris@69: 
Chris@69:    ALLOC(freq, CC*N, celt_sig); /**< Interleaved signal MDCTs */
Chris@69:    ALLOC(bandE,nbEBands*CC, celt_ener);
Chris@69:    ALLOC(bandLogE,nbEBands*CC, opus_val16);
Chris@69: 
Chris@69:    secondMdct = shortBlocks && st->complexity>=8;
Chris@69:    ALLOC(bandLogE2, C*nbEBands, opus_val16);
Chris@69:    if (secondMdct)
Chris@69:    {
Chris@69:       compute_mdcts(mode, 0, in, freq, C, CC, LM, st->upsample, st->arch);
Chris@69:       compute_band_energies(mode, freq, bandE, effEnd, C, LM, st->arch);
Chris@69:       amp2Log2(mode, effEnd, end, bandE, bandLogE2, C);
Chris@69:       for (i=0;i<C*nbEBands;i++)
Chris@69:          bandLogE2[i] += HALF16(SHL16(LM, DB_SHIFT));
Chris@69:    }
Chris@69: 
Chris@69:    compute_mdcts(mode, shortBlocks, in, freq, C, CC, LM, st->upsample, st->arch);
Chris@69:    /* This should catch any NaN in the CELT input. Since we're not supposed to see any (they're filtered
Chris@69:       at the Opus layer), just abort. */
Chris@69:    celt_assert(!celt_isnan(freq[0]) && (C==1 || !celt_isnan(freq[N])));
Chris@69:    if (CC==2&&C==1)
Chris@69:       tf_chan = 0;
Chris@69:    compute_band_energies(mode, freq, bandE, effEnd, C, LM, st->arch);
Chris@69: 
Chris@69:    if (st->lfe)
Chris@69:    {
Chris@69:       for (i=2;i<end;i++)
Chris@69:       {
Chris@69:          bandE[i] = IMIN(bandE[i], MULT16_32_Q15(QCONST16(1e-4f,15),bandE[0]));
Chris@69:          bandE[i] = MAX32(bandE[i], EPSILON);
Chris@69:       }
Chris@69:    }
Chris@69:    amp2Log2(mode, effEnd, end, bandE, bandLogE, C);
Chris@69: 
Chris@69:    ALLOC(surround_dynalloc, C*nbEBands, opus_val16);
Chris@69:    OPUS_CLEAR(surround_dynalloc, end);
Chris@69:    /* This computes how much masking takes place between surround channels */
Chris@69:    if (!hybrid&&st->energy_mask&&!st->lfe)
Chris@69:    {
Chris@69:       int mask_end;
Chris@69:       int midband;
Chris@69:       int count_dynalloc;
Chris@69:       opus_val32 mask_avg=0;
Chris@69:       opus_val32 diff=0;
Chris@69:       int count=0;
Chris@69:       mask_end = IMAX(2,st->lastCodedBands);
Chris@69:       for (c=0;c<C;c++)
Chris@69:       {
Chris@69:          for(i=0;i<mask_end;i++)
Chris@69:          {
Chris@69:             opus_val16 mask;
Chris@69:             mask = MAX16(MIN16(st->energy_mask[nbEBands*c+i],
Chris@69:                    QCONST16(.25f, DB_SHIFT)), -QCONST16(2.0f, DB_SHIFT));
Chris@69:             if (mask > 0)
Chris@69:                mask = HALF16(mask);
Chris@69:             mask_avg += MULT16_16(mask, eBands[i+1]-eBands[i]);
Chris@69:             count += eBands[i+1]-eBands[i];
Chris@69:             diff += MULT16_16(mask, 1+2*i-mask_end);
Chris@69:          }
Chris@69:       }
Chris@69:       celt_assert(count>0);
Chris@69:       mask_avg = DIV32_16(mask_avg,count);
Chris@69:       mask_avg += QCONST16(.2f, DB_SHIFT);
Chris@69:       diff = diff*6/(C*(mask_end-1)*(mask_end+1)*mask_end);
Chris@69:       /* Again, being conservative */
Chris@69:       diff = HALF32(diff);
Chris@69:       diff = MAX32(MIN32(diff, QCONST32(.031f, DB_SHIFT)), -QCONST32(.031f, DB_SHIFT));
Chris@69:       /* Find the band that's in the middle of the coded spectrum */
Chris@69:       for (midband=0;eBands[midband+1] < eBands[mask_end]/2;midband++);
Chris@69:       count_dynalloc=0;
Chris@69:       for(i=0;i<mask_end;i++)
Chris@69:       {
Chris@69:          opus_val32 lin;
Chris@69:          opus_val16 unmask;
Chris@69:          lin = mask_avg + diff*(i-midband);
Chris@69:          if (C==2)
Chris@69:             unmask = MAX16(st->energy_mask[i], st->energy_mask[nbEBands+i]);
Chris@69:          else
Chris@69:             unmask = st->energy_mask[i];
Chris@69:          unmask = MIN16(unmask, QCONST16(.0f, DB_SHIFT));
Chris@69:          unmask -= lin;
Chris@69:          if (unmask > QCONST16(.25f, DB_SHIFT))
Chris@69:          {
Chris@69:             surround_dynalloc[i] = unmask - QCONST16(.25f, DB_SHIFT);
Chris@69:             count_dynalloc++;
Chris@69:          }
Chris@69:       }
Chris@69:       if (count_dynalloc>=3)
Chris@69:       {
Chris@69:          /* If we need dynalloc in many bands, it's probably because our
Chris@69:             initial masking rate was too low. */
Chris@69:          mask_avg += QCONST16(.25f, DB_SHIFT);
Chris@69:          if (mask_avg>0)
Chris@69:          {
Chris@69:             /* Something went really wrong in the original calculations,
Chris@69:                disabling masking. */
Chris@69:             mask_avg = 0;
Chris@69:             diff = 0;
Chris@69:             OPUS_CLEAR(surround_dynalloc, mask_end);
Chris@69:          } else {
Chris@69:             for(i=0;i<mask_end;i++)
Chris@69:                surround_dynalloc[i] = MAX16(0, surround_dynalloc[i]-QCONST16(.25f, DB_SHIFT));
Chris@69:          }
Chris@69:       }
Chris@69:       mask_avg += QCONST16(.2f, DB_SHIFT);
Chris@69:       /* Convert to 1/64th units used for the trim */
Chris@69:       surround_trim = 64*diff;
Chris@69:       /*printf("%d %d ", mask_avg, surround_trim);*/
Chris@69:       surround_masking = mask_avg;
Chris@69:    }
Chris@69:    /* Temporal VBR (but not for LFE) */
Chris@69:    if (!st->lfe)
Chris@69:    {
Chris@69:       opus_val16 follow=-QCONST16(10.0f,DB_SHIFT);
Chris@69:       opus_val32 frame_avg=0;
Chris@69:       opus_val16 offset = shortBlocks?HALF16(SHL16(LM, DB_SHIFT)):0;
Chris@69:       for(i=start;i<end;i++)
Chris@69:       {
Chris@69:          follow = MAX16(follow-QCONST16(1.f, DB_SHIFT), bandLogE[i]-offset);
Chris@69:          if (C==2)
Chris@69:             follow = MAX16(follow, bandLogE[i+nbEBands]-offset);
Chris@69:          frame_avg += follow;
Chris@69:       }
Chris@69:       frame_avg /= (end-start);
Chris@69:       temporal_vbr = SUB16(frame_avg,st->spec_avg);
Chris@69:       temporal_vbr = MIN16(QCONST16(3.f, DB_SHIFT), MAX16(-QCONST16(1.5f, DB_SHIFT), temporal_vbr));
Chris@69:       st->spec_avg += MULT16_16_Q15(QCONST16(.02f, 15), temporal_vbr);
Chris@69:    }
Chris@69:    /*for (i=0;i<21;i++)
Chris@69:       printf("%f ", bandLogE[i]);
Chris@69:    printf("\n");*/
Chris@69: 
Chris@69:    if (!secondMdct)
Chris@69:    {
Chris@69:       OPUS_COPY(bandLogE2, bandLogE, C*nbEBands);
Chris@69:    }
Chris@69: 
Chris@69:    /* Last chance to catch any transient we might have missed in the
Chris@69:       time-domain analysis */
Chris@69:    if (LM>0 && ec_tell(enc)+3<=total_bits && !isTransient && st->complexity>=5 && !st->lfe && !hybrid)
Chris@69:    {
Chris@69:       if (patch_transient_decision(bandLogE, oldBandE, nbEBands, start, end, C))
Chris@69:       {
Chris@69:          isTransient = 1;
Chris@69:          shortBlocks = M;
Chris@69:          compute_mdcts(mode, shortBlocks, in, freq, C, CC, LM, st->upsample, st->arch);
Chris@69:          compute_band_energies(mode, freq, bandE, effEnd, C, LM, st->arch);
Chris@69:          amp2Log2(mode, effEnd, end, bandE, bandLogE, C);
Chris@69:          /* Compensate for the scaling of short vs long mdcts */
Chris@69:          for (i=0;i<C*nbEBands;i++)
Chris@69:             bandLogE2[i] += HALF16(SHL16(LM, DB_SHIFT));
Chris@69:          tf_estimate = QCONST16(.2f,14);
Chris@69:       }
Chris@69:    }
Chris@69: 
Chris@69:    if (LM>0 && ec_tell(enc)+3<=total_bits)
Chris@69:       ec_enc_bit_logp(enc, isTransient, 3);
Chris@69: 
Chris@69:    ALLOC(X, C*N, celt_norm);         /**< Interleaved normalised MDCTs */
Chris@69: 
Chris@69:    /* Band normalisation */
Chris@69:    normalise_bands(mode, freq, X, bandE, effEnd, C, M);
Chris@69: 
Chris@69:    enable_tf_analysis = effectiveBytes>=15*C && !hybrid && st->complexity>=2 && !st->lfe;
Chris@69: 
Chris@69:    ALLOC(offsets, nbEBands, int);
Chris@69:    ALLOC(importance, nbEBands, int);
Chris@69:    ALLOC(spread_weight, nbEBands, int);
Chris@69: 
Chris@69:    maxDepth = dynalloc_analysis(bandLogE, bandLogE2, nbEBands, start, end, C, offsets,
Chris@69:          st->lsb_depth, mode->logN, isTransient, st->vbr, st->constrained_vbr,
Chris@69:          eBands, LM, effectiveBytes, &tot_boost, st->lfe, surround_dynalloc, &st->analysis, importance, spread_weight);
Chris@69: 
Chris@69:    ALLOC(tf_res, nbEBands, int);
Chris@69:    /* Disable variable tf resolution for hybrid and at very low bitrate */
Chris@69:    if (enable_tf_analysis)
Chris@69:    {
Chris@69:       int lambda;
Chris@69:       lambda = IMAX(80, 20480/effectiveBytes + 2);
Chris@69:       tf_select = tf_analysis(mode, effEnd, isTransient, tf_res, lambda, X, N, LM, tf_estimate, tf_chan, importance);
Chris@69:       for (i=effEnd;i<end;i++)
Chris@69:          tf_res[i] = tf_res[effEnd-1];
Chris@69:    } else if (hybrid && weak_transient)
Chris@69:    {
Chris@69:       /* For weak transients, we rely on the fact that improving time resolution using
Chris@69:          TF on a long window is imperfect and will not result in an energy collapse at
Chris@69:          low bitrate. */
Chris@69:       for (i=0;i<end;i++)
Chris@69:          tf_res[i] = 1;
Chris@69:       tf_select=0;
Chris@69:    } else if (hybrid && effectiveBytes<15 && st->silk_info.signalType != 2)
Chris@69:    {
Chris@69:       /* For low bitrate hybrid, we force temporal resolution to 5 ms rather than 2.5 ms. */
Chris@69:       for (i=0;i<end;i++)
Chris@69:          tf_res[i] = 0;
Chris@69:       tf_select=isTransient;
Chris@69:    } else {
Chris@69:       for (i=0;i<end;i++)
Chris@69:          tf_res[i] = isTransient;
Chris@69:       tf_select=0;
Chris@69:    }
Chris@69: 
Chris@69:    ALLOC(error, C*nbEBands, opus_val16);
Chris@69:    c=0;
Chris@69:    do {
Chris@69:       for (i=start;i<end;i++)
Chris@69:       {
Chris@69:          /* When the energy is stable, slightly bias energy quantization towards
Chris@69:             the previous error to make the gain more stable (a constant offset is
Chris@69:             better than fluctuations). */
Chris@69:          if (ABS32(SUB32(bandLogE[i+c*nbEBands], oldBandE[i+c*nbEBands])) < QCONST16(2.f, DB_SHIFT))
Chris@69:          {
Chris@69:             bandLogE[i+c*nbEBands] -= MULT16_16_Q15(energyError[i+c*nbEBands], QCONST16(0.25f, 15));
Chris@69:          }
Chris@69:       }
Chris@69:    } while (++c < C);
Chris@69:    quant_coarse_energy(mode, start, end, effEnd, bandLogE,
Chris@69:          oldBandE, total_bits, error, enc,
Chris@69:          C, LM, nbAvailableBytes, st->force_intra,
Chris@69:          &st->delayedIntra, st->complexity >= 4, st->loss_rate, st->lfe);
Chris@69: 
Chris@69:    tf_encode(start, end, isTransient, tf_res, LM, tf_select, enc);
Chris@69: 
Chris@69:    if (ec_tell(enc)+4<=total_bits)
Chris@69:    {
Chris@69:       if (st->lfe)
Chris@69:       {
Chris@69:          st->tapset_decision = 0;
Chris@69:          st->spread_decision = SPREAD_NORMAL;
Chris@69:       } else if (hybrid)
Chris@69:       {
Chris@69:          if (st->complexity == 0)
Chris@69:             st->spread_decision = SPREAD_NONE;
Chris@69:          else if (isTransient)
Chris@69:             st->spread_decision = SPREAD_NORMAL;
Chris@69:          else
Chris@69:             st->spread_decision = SPREAD_AGGRESSIVE;
Chris@69:       } else if (shortBlocks || st->complexity < 3 || nbAvailableBytes < 10*C)
Chris@69:       {
Chris@69:          if (st->complexity == 0)
Chris@69:             st->spread_decision = SPREAD_NONE;
Chris@69:          else
Chris@69:             st->spread_decision = SPREAD_NORMAL;
Chris@69:       } else {
Chris@69:          /* Disable new spreading+tapset estimator until we can show it works
Chris@69:             better than the old one. So far it seems like spreading_decision()
Chris@69:             works best. */
Chris@69: #if 0
Chris@69:          if (st->analysis.valid)
Chris@69:          {
Chris@69:             static const opus_val16 spread_thresholds[3] = {-QCONST16(.6f, 15), -QCONST16(.2f, 15), -QCONST16(.07f, 15)};
Chris@69:             static const opus_val16 spread_histeresis[3] = {QCONST16(.15f, 15), QCONST16(.07f, 15), QCONST16(.02f, 15)};
Chris@69:             static const opus_val16 tapset_thresholds[2] = {QCONST16(.0f, 15), QCONST16(.15f, 15)};
Chris@69:             static const opus_val16 tapset_histeresis[2] = {QCONST16(.1f, 15), QCONST16(.05f, 15)};
Chris@69:             st->spread_decision = hysteresis_decision(-st->analysis.tonality, spread_thresholds, spread_histeresis, 3, st->spread_decision);
Chris@69:             st->tapset_decision = hysteresis_decision(st->analysis.tonality_slope, tapset_thresholds, tapset_histeresis, 2, st->tapset_decision);
Chris@69:          } else
Chris@69: #endif
Chris@69:          {
Chris@69:             st->spread_decision = spreading_decision(mode, X,
Chris@69:                   &st->tonal_average, st->spread_decision, &st->hf_average,
Chris@69:                   &st->tapset_decision, pf_on&&!shortBlocks, effEnd, C, M, spread_weight);
Chris@69:          }
Chris@69:          /*printf("%d %d\n", st->tapset_decision, st->spread_decision);*/
Chris@69:          /*printf("%f %d %f %d\n\n", st->analysis.tonality, st->spread_decision, st->analysis.tonality_slope, st->tapset_decision);*/
Chris@69:       }
Chris@69:       ec_enc_icdf(enc, st->spread_decision, spread_icdf, 5);
Chris@69:    }
Chris@69: 
Chris@69:    /* For LFE, everything interesting is in the first band */
Chris@69:    if (st->lfe)
Chris@69:       offsets[0] = IMIN(8, effectiveBytes/3);
Chris@69:    ALLOC(cap, nbEBands, int);
Chris@69:    init_caps(mode,cap,LM,C);
Chris@69: 
Chris@69:    dynalloc_logp = 6;
Chris@69:    total_bits<<=BITRES;
Chris@69:    total_boost = 0;
Chris@69:    tell = ec_tell_frac(enc);
Chris@69:    for (i=start;i<end;i++)
Chris@69:    {
Chris@69:       int width, quanta;
Chris@69:       int dynalloc_loop_logp;
Chris@69:       int boost;
Chris@69:       int j;
Chris@69:       width = C*(eBands[i+1]-eBands[i])<<LM;
Chris@69:       /* quanta is 6 bits, but no more than 1 bit/sample
Chris@69:          and no less than 1/8 bit/sample */
Chris@69:       quanta = IMIN(width<<BITRES, IMAX(6<<BITRES, width));
Chris@69:       dynalloc_loop_logp = dynalloc_logp;
Chris@69:       boost = 0;
Chris@69:       for (j = 0; tell+(dynalloc_loop_logp<<BITRES) < total_bits-total_boost
Chris@69:             && boost < cap[i]; j++)
Chris@69:       {
Chris@69:          int flag;
Chris@69:          flag = j<offsets[i];
Chris@69:          ec_enc_bit_logp(enc, flag, dynalloc_loop_logp);
Chris@69:          tell = ec_tell_frac(enc);
Chris@69:          if (!flag)
Chris@69:             break;
Chris@69:          boost += quanta;
Chris@69:          total_boost += quanta;
Chris@69:          dynalloc_loop_logp = 1;
Chris@69:       }
Chris@69:       /* Making dynalloc more likely */
Chris@69:       if (j)
Chris@69:          dynalloc_logp = IMAX(2, dynalloc_logp-1);
Chris@69:       offsets[i] = boost;
Chris@69:    }
Chris@69: 
Chris@69:    if (C==2)
Chris@69:    {
Chris@69:       static const opus_val16 intensity_thresholds[21]=
Chris@69:       /* 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19  20  off*/
Chris@69:         {  1, 2, 3, 4, 5, 6, 7, 8,16,24,36,44,50,56,62,67,72,79,88,106,134};
Chris@69:       static const opus_val16 intensity_histeresis[21]=
Chris@69:         {  1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 3, 3, 4, 5, 6,  8, 8};
Chris@69: 
Chris@69:       /* Always use MS for 2.5 ms frames until we can do a better analysis */
Chris@69:       if (LM!=0)
Chris@69:          dual_stereo = stereo_analysis(mode, X, LM, N);
Chris@69: 
Chris@69:       st->intensity = hysteresis_decision((opus_val16)(equiv_rate/1000),
Chris@69:             intensity_thresholds, intensity_histeresis, 21, st->intensity);
Chris@69:       st->intensity = IMIN(end,IMAX(start, st->intensity));
Chris@69:    }
Chris@69: 
Chris@69:    alloc_trim = 5;
Chris@69:    if (tell+(6<<BITRES) <= total_bits - total_boost)
Chris@69:    {
Chris@69:       if (start > 0 || st->lfe)
Chris@69:       {
Chris@69:          st->stereo_saving = 0;
Chris@69:          alloc_trim = 5;
Chris@69:       } else {
Chris@69:          alloc_trim = alloc_trim_analysis(mode, X, bandLogE,
Chris@69:             end, LM, C, N, &st->analysis, &st->stereo_saving, tf_estimate,
Chris@69:             st->intensity, surround_trim, equiv_rate, st->arch);
Chris@69:       }
Chris@69:       ec_enc_icdf(enc, alloc_trim, trim_icdf, 7);
Chris@69:       tell = ec_tell_frac(enc);
Chris@69:    }
Chris@69: 
Chris@69:    /* Variable bitrate */
Chris@69:    if (vbr_rate>0)
Chris@69:    {
Chris@69:      opus_val16 alpha;
Chris@69:      opus_int32 delta;
Chris@69:      /* The target rate in 8th bits per frame */
Chris@69:      opus_int32 target, base_target;
Chris@69:      opus_int32 min_allowed;
Chris@69:      int lm_diff = mode->maxLM - LM;
Chris@69: 
Chris@69:      /* Don't attempt to use more than 510 kb/s, even for frames smaller than 20 ms.
Chris@69:         The CELT allocator will just not be able to use more than that anyway. */
Chris@69:      nbCompressedBytes = IMIN(nbCompressedBytes,1275>>(3-LM));
Chris@69:      if (!hybrid)
Chris@69:      {
Chris@69:         base_target = vbr_rate - ((40*C+20)<<BITRES);
Chris@69:      } else {
Chris@69:         base_target = IMAX(0, vbr_rate - ((9*C+4)<<BITRES));
Chris@69:      }
Chris@69: 
Chris@69:      if (st->constrained_vbr)
Chris@69:         base_target += (st->vbr_offset>>lm_diff);
Chris@69: 
Chris@69:      if (!hybrid)
Chris@69:      {
Chris@69:         target = compute_vbr(mode, &st->analysis, base_target, LM, equiv_rate,
Chris@69:            st->lastCodedBands, C, st->intensity, st->constrained_vbr,
Chris@69:            st->stereo_saving, tot_boost, tf_estimate, pitch_change, maxDepth,
Chris@69:            st->lfe, st->energy_mask!=NULL, surround_masking,
Chris@69:            temporal_vbr);
Chris@69:      } else {
Chris@69:         target = base_target;
Chris@69:         /* Tonal frames (offset<100) need more bits than noisy (offset>100) ones. */
Chris@69:         if (st->silk_info.offset < 100) target += 12 << BITRES >> (3-LM);
Chris@69:         if (st->silk_info.offset > 100) target -= 18 << BITRES >> (3-LM);
Chris@69:         /* Boosting bitrate on transients and vowels with significant temporal
Chris@69:            spikes. */
Chris@69:         target += (opus_int32)MULT16_16_Q14(tf_estimate-QCONST16(.25f,14), (50<<BITRES));
Chris@69:         /* If we have a strong transient, let's make sure it has enough bits to code
Chris@69:            the first two bands, so that it can use folding rather than noise. */
Chris@69:         if (tf_estimate > QCONST16(.7f,14))
Chris@69:            target = IMAX(target, 50<<BITRES);
Chris@69:      }
Chris@69:      /* The current offset is removed from the target and the space used
Chris@69:         so far is added*/
Chris@69:      target=target+tell;
Chris@69:      /* In VBR mode the frame size must not be reduced so much that it would
Chris@69:          result in the encoder running out of bits.
Chris@69:         The margin of 2 bytes ensures that none of the bust-prevention logic
Chris@69:          in the decoder will have triggered so far. */
Chris@69:      min_allowed = ((tell+total_boost+(1<<(BITRES+3))-1)>>(BITRES+3)) + 2;
Chris@69:      /* Take into account the 37 bits we need to have left in the packet to
Chris@69:         signal a redundant frame in hybrid mode. Creating a shorter packet would
Chris@69:         create an entropy coder desync. */
Chris@69:      if (hybrid)
Chris@69:         min_allowed = IMAX(min_allowed, (tell0_frac+(37<<BITRES)+total_boost+(1<<(BITRES+3))-1)>>(BITRES+3));
Chris@69: 
Chris@69:      nbAvailableBytes = (target+(1<<(BITRES+2)))>>(BITRES+3);
Chris@69:      nbAvailableBytes = IMAX(min_allowed,nbAvailableBytes);
Chris@69:      nbAvailableBytes = IMIN(nbCompressedBytes,nbAvailableBytes);
Chris@69: 
Chris@69:      /* By how much did we "miss" the target on that frame */
Chris@69:      delta = target - vbr_rate;
Chris@69: 
Chris@69:      target=nbAvailableBytes<<(BITRES+3);
Chris@69: 
Chris@69:      /*If the frame is silent we don't adjust our drift, otherwise
Chris@69:        the encoder will shoot to very high rates after hitting a
Chris@69:        span of silence, but we do allow the bitres to refill.
Chris@69:        This means that we'll undershoot our target in CVBR/VBR modes
Chris@69:        on files with lots of silence. */
Chris@69:      if(silence)
Chris@69:      {
Chris@69:        nbAvailableBytes = 2;
Chris@69:        target = 2*8<<BITRES;
Chris@69:        delta = 0;
Chris@69:      }
Chris@69: 
Chris@69:      if (st->vbr_count < 970)
Chris@69:      {
Chris@69:         st->vbr_count++;
Chris@69:         alpha = celt_rcp(SHL32(EXTEND32(st->vbr_count+20),16));
Chris@69:      } else
Chris@69:         alpha = QCONST16(.001f,15);
Chris@69:      /* How many bits have we used in excess of what we're allowed */
Chris@69:      if (st->constrained_vbr)
Chris@69:         st->vbr_reservoir += target - vbr_rate;
Chris@69:      /*printf ("%d\n", st->vbr_reservoir);*/
Chris@69: 
Chris@69:      /* Compute the offset we need to apply in order to reach the target */
Chris@69:      if (st->constrained_vbr)
Chris@69:      {
Chris@69:         st->vbr_drift += (opus_int32)MULT16_32_Q15(alpha,(delta*(1<<lm_diff))-st->vbr_offset-st->vbr_drift);
Chris@69:         st->vbr_offset = -st->vbr_drift;
Chris@69:      }
Chris@69:      /*printf ("%d\n", st->vbr_drift);*/
Chris@69: 
Chris@69:      if (st->constrained_vbr && st->vbr_reservoir < 0)
Chris@69:      {
Chris@69:         /* We're under the min value -- increase rate */
Chris@69:         int adjust = (-st->vbr_reservoir)/(8<<BITRES);
Chris@69:         /* Unless we're just coding silence */
Chris@69:         nbAvailableBytes += silence?0:adjust;
Chris@69:         st->vbr_reservoir = 0;
Chris@69:         /*printf ("+%d\n", adjust);*/
Chris@69:      }
Chris@69:      nbCompressedBytes = IMIN(nbCompressedBytes,nbAvailableBytes);
Chris@69:      /*printf("%d\n", nbCompressedBytes*50*8);*/
Chris@69:      /* This moves the raw bits to take into account the new compressed size */
Chris@69:      ec_enc_shrink(enc, nbCompressedBytes);
Chris@69:    }
Chris@69: 
Chris@69:    /* Bit allocation */
Chris@69:    ALLOC(fine_quant, nbEBands, int);
Chris@69:    ALLOC(pulses, nbEBands, int);
Chris@69:    ALLOC(fine_priority, nbEBands, int);
Chris@69: 
Chris@69:    /* bits =           packet size                    - where we are - safety*/
Chris@69:    bits = (((opus_int32)nbCompressedBytes*8)<<BITRES) - ec_tell_frac(enc) - 1;
Chris@69:    anti_collapse_rsv = isTransient&&LM>=2&&bits>=((LM+2)<<BITRES) ? (1<<BITRES) : 0;
Chris@69:    bits -= anti_collapse_rsv;
Chris@69:    signalBandwidth = end-1;
Chris@69: #ifndef DISABLE_FLOAT_API
Chris@69:    if (st->analysis.valid)
Chris@69:    {
Chris@69:       int min_bandwidth;
Chris@69:       if (equiv_rate < (opus_int32)32000*C)
Chris@69:          min_bandwidth = 13;
Chris@69:       else if (equiv_rate < (opus_int32)48000*C)
Chris@69:          min_bandwidth = 16;
Chris@69:       else if (equiv_rate < (opus_int32)60000*C)
Chris@69:          min_bandwidth = 18;
Chris@69:       else  if (equiv_rate < (opus_int32)80000*C)
Chris@69:          min_bandwidth = 19;
Chris@69:       else
Chris@69:          min_bandwidth = 20;
Chris@69:       signalBandwidth = IMAX(st->analysis.bandwidth, min_bandwidth);
Chris@69:    }
Chris@69: #endif
Chris@69:    if (st->lfe)
Chris@69:       signalBandwidth = 1;
Chris@69:    codedBands = clt_compute_allocation(mode, start, end, offsets, cap,
Chris@69:          alloc_trim, &st->intensity, &dual_stereo, bits, &balance, pulses,
Chris@69:          fine_quant, fine_priority, C, LM, enc, 1, st->lastCodedBands, signalBandwidth);
Chris@69:    if (st->lastCodedBands)
Chris@69:       st->lastCodedBands = IMIN(st->lastCodedBands+1,IMAX(st->lastCodedBands-1,codedBands));
Chris@69:    else
Chris@69:       st->lastCodedBands = codedBands;
Chris@69: 
Chris@69:    quant_fine_energy(mode, start, end, oldBandE, error, fine_quant, enc, C);
Chris@69: 
Chris@69:    /* Residual quantisation */
Chris@69:    ALLOC(collapse_masks, C*nbEBands, unsigned char);
Chris@69:    quant_all_bands(1, mode, start, end, X, C==2 ? X+N : NULL, collapse_masks,
Chris@69:          bandE, pulses, shortBlocks, st->spread_decision,
Chris@69:          dual_stereo, st->intensity, tf_res, nbCompressedBytes*(8<<BITRES)-anti_collapse_rsv,
Chris@69:          balance, enc, LM, codedBands, &st->rng, st->complexity, st->arch, st->disable_inv);
Chris@69: 
Chris@69:    if (anti_collapse_rsv > 0)
Chris@69:    {
Chris@69:       anti_collapse_on = st->consec_transient<2;
Chris@69: #ifdef FUZZING
Chris@69:       anti_collapse_on = rand()&0x1;
Chris@69: #endif
Chris@69:       ec_enc_bits(enc, anti_collapse_on, 1);
Chris@69:    }
Chris@69:    quant_energy_finalise(mode, start, end, oldBandE, error, fine_quant, fine_priority, nbCompressedBytes*8-ec_tell(enc), enc, C);
Chris@69:    OPUS_CLEAR(energyError, nbEBands*CC);
Chris@69:    c=0;
Chris@69:    do {
Chris@69:       for (i=start;i<end;i++)
Chris@69:       {
Chris@69:          energyError[i+c*nbEBands] = MAX16(-QCONST16(0.5f, 15), MIN16(QCONST16(0.5f, 15), error[i+c*nbEBands]));
Chris@69:       }
Chris@69:    } while (++c < C);
Chris@69: 
Chris@69:    if (silence)
Chris@69:    {
Chris@69:       for (i=0;i<C*nbEBands;i++)
Chris@69:          oldBandE[i] = -QCONST16(28.f,DB_SHIFT);
Chris@69:    }
Chris@69: 
Chris@69: #ifdef RESYNTH
Chris@69:    /* Re-synthesis of the coded audio if required */
Chris@69:    {
Chris@69:       celt_sig *out_mem[2];
Chris@69: 
Chris@69:       if (anti_collapse_on)
Chris@69:       {
Chris@69:          anti_collapse(mode, X, collapse_masks, LM, C, N,
Chris@69:                start, end, oldBandE, oldLogE, oldLogE2, pulses, st->rng);
Chris@69:       }
Chris@69: 
Chris@69:       c=0; do {
Chris@69:          OPUS_MOVE(st->syn_mem[c], st->syn_mem[c]+N, 2*MAX_PERIOD-N+overlap/2);
Chris@69:       } while (++c<CC);
Chris@69: 
Chris@69:       c=0; do {
Chris@69:          out_mem[c] = st->syn_mem[c]+2*MAX_PERIOD-N;
Chris@69:       } while (++c<CC);
Chris@69: 
Chris@69:       celt_synthesis(mode, X, out_mem, oldBandE, start, effEnd,
Chris@69:                      C, CC, isTransient, LM, st->upsample, silence, st->arch);
Chris@69: 
Chris@69:       c=0; do {
Chris@69:          st->prefilter_period=IMAX(st->prefilter_period, COMBFILTER_MINPERIOD);
Chris@69:          st->prefilter_period_old=IMAX(st->prefilter_period_old, COMBFILTER_MINPERIOD);
Chris@69:          comb_filter(out_mem[c], out_mem[c], st->prefilter_period_old, st->prefilter_period, mode->shortMdctSize,
Chris@69:                st->prefilter_gain_old, st->prefilter_gain, st->prefilter_tapset_old, st->prefilter_tapset,
Chris@69:                mode->window, overlap);
Chris@69:          if (LM!=0)
Chris@69:             comb_filter(out_mem[c]+mode->shortMdctSize, out_mem[c]+mode->shortMdctSize, st->prefilter_period, pitch_index, N-mode->shortMdctSize,
Chris@69:                   st->prefilter_gain, gain1, st->prefilter_tapset, prefilter_tapset,
Chris@69:                   mode->window, overlap);
Chris@69:       } while (++c<CC);
Chris@69: 
Chris@69:       /* We reuse freq[] as scratch space for the de-emphasis */
Chris@69:       deemphasis(out_mem, (opus_val16*)pcm, N, CC, st->upsample, mode->preemph, st->preemph_memD);
Chris@69:       st->prefilter_period_old = st->prefilter_period;
Chris@69:       st->prefilter_gain_old = st->prefilter_gain;
Chris@69:       st->prefilter_tapset_old = st->prefilter_tapset;
Chris@69:    }
Chris@69: #endif
Chris@69: 
Chris@69:    st->prefilter_period = pitch_index;
Chris@69:    st->prefilter_gain = gain1;
Chris@69:    st->prefilter_tapset = prefilter_tapset;
Chris@69: #ifdef RESYNTH
Chris@69:    if (LM!=0)
Chris@69:    {
Chris@69:       st->prefilter_period_old = st->prefilter_period;
Chris@69:       st->prefilter_gain_old = st->prefilter_gain;
Chris@69:       st->prefilter_tapset_old = st->prefilter_tapset;
Chris@69:    }
Chris@69: #endif
Chris@69: 
Chris@69:    if (CC==2&&C==1) {
Chris@69:       OPUS_COPY(&oldBandE[nbEBands], oldBandE, nbEBands);
Chris@69:    }
Chris@69: 
Chris@69:    if (!isTransient)
Chris@69:    {
Chris@69:       OPUS_COPY(oldLogE2, oldLogE, CC*nbEBands);
Chris@69:       OPUS_COPY(oldLogE, oldBandE, CC*nbEBands);
Chris@69:    } else {
Chris@69:       for (i=0;i<CC*nbEBands;i++)
Chris@69:          oldLogE[i] = MIN16(oldLogE[i], oldBandE[i]);
Chris@69:    }
Chris@69:    /* In case start or end were to change */
Chris@69:    c=0; do
Chris@69:    {
Chris@69:       for (i=0;i<start;i++)
Chris@69:       {
Chris@69:          oldBandE[c*nbEBands+i]=0;
Chris@69:          oldLogE[c*nbEBands+i]=oldLogE2[c*nbEBands+i]=-QCONST16(28.f,DB_SHIFT);
Chris@69:       }
Chris@69:       for (i=end;i<nbEBands;i++)
Chris@69:       {
Chris@69:          oldBandE[c*nbEBands+i]=0;
Chris@69:          oldLogE[c*nbEBands+i]=oldLogE2[c*nbEBands+i]=-QCONST16(28.f,DB_SHIFT);
Chris@69:       }
Chris@69:    } while (++c<CC);
Chris@69: 
Chris@69:    if (isTransient || transient_got_disabled)
Chris@69:       st->consec_transient++;
Chris@69:    else
Chris@69:       st->consec_transient=0;
Chris@69:    st->rng = enc->rng;
Chris@69: 
Chris@69:    /* If there's any room left (can only happen for very high rates),
Chris@69:       it's already filled with zeros */
Chris@69:    ec_enc_done(enc);
Chris@69: 
Chris@69: #ifdef CUSTOM_MODES
Chris@69:    if (st->signalling)
Chris@69:       nbCompressedBytes++;
Chris@69: #endif
Chris@69: 
Chris@69:    RESTORE_STACK;
Chris@69:    if (ec_get_error(enc))
Chris@69:       return OPUS_INTERNAL_ERROR;
Chris@69:    else
Chris@69:       return nbCompressedBytes;
Chris@69: }
Chris@69: 
Chris@69: 
Chris@69: #ifdef CUSTOM_MODES
Chris@69: 
Chris@69: #ifdef FIXED_POINT
Chris@69: int opus_custom_encode(CELTEncoder * OPUS_RESTRICT st, const opus_int16 * pcm, int frame_size, unsigned char *compressed, int nbCompressedBytes)
Chris@69: {
Chris@69:    return celt_encode_with_ec(st, pcm, frame_size, compressed, nbCompressedBytes, NULL);
Chris@69: }
Chris@69: 
Chris@69: #ifndef DISABLE_FLOAT_API
Chris@69: int opus_custom_encode_float(CELTEncoder * OPUS_RESTRICT st, const float * pcm, int frame_size, unsigned char *compressed, int nbCompressedBytes)
Chris@69: {
Chris@69:    int j, ret, C, N;
Chris@69:    VARDECL(opus_int16, in);
Chris@69:    ALLOC_STACK;
Chris@69: 
Chris@69:    if (pcm==NULL)
Chris@69:       return OPUS_BAD_ARG;
Chris@69: 
Chris@69:    C = st->channels;
Chris@69:    N = frame_size;
Chris@69:    ALLOC(in, C*N, opus_int16);
Chris@69: 
Chris@69:    for (j=0;j<C*N;j++)
Chris@69:      in[j] = FLOAT2INT16(pcm[j]);
Chris@69: 
Chris@69:    ret=celt_encode_with_ec(st,in,frame_size,compressed,nbCompressedBytes, NULL);
Chris@69: #ifdef RESYNTH
Chris@69:    for (j=0;j<C*N;j++)
Chris@69:       ((float*)pcm)[j]=in[j]*(1.f/32768.f);
Chris@69: #endif
Chris@69:    RESTORE_STACK;
Chris@69:    return ret;
Chris@69: }
Chris@69: #endif /* DISABLE_FLOAT_API */
Chris@69: #else
Chris@69: 
Chris@69: int opus_custom_encode(CELTEncoder * OPUS_RESTRICT st, const opus_int16 * pcm, int frame_size, unsigned char *compressed, int nbCompressedBytes)
Chris@69: {
Chris@69:    int j, ret, C, N;
Chris@69:    VARDECL(celt_sig, in);
Chris@69:    ALLOC_STACK;
Chris@69: 
Chris@69:    if (pcm==NULL)
Chris@69:       return OPUS_BAD_ARG;
Chris@69: 
Chris@69:    C=st->channels;
Chris@69:    N=frame_size;
Chris@69:    ALLOC(in, C*N, celt_sig);
Chris@69:    for (j=0;j<C*N;j++) {
Chris@69:      in[j] = SCALEOUT(pcm[j]);
Chris@69:    }
Chris@69: 
Chris@69:    ret = celt_encode_with_ec(st,in,frame_size,compressed,nbCompressedBytes, NULL);
Chris@69: #ifdef RESYNTH
Chris@69:    for (j=0;j<C*N;j++)
Chris@69:       ((opus_int16*)pcm)[j] = FLOAT2INT16(in[j]);
Chris@69: #endif
Chris@69:    RESTORE_STACK;
Chris@69:    return ret;
Chris@69: }
Chris@69: 
Chris@69: int opus_custom_encode_float(CELTEncoder * OPUS_RESTRICT st, const float * pcm, int frame_size, unsigned char *compressed, int nbCompressedBytes)
Chris@69: {
Chris@69:    return celt_encode_with_ec(st, pcm, frame_size, compressed, nbCompressedBytes, NULL);
Chris@69: }
Chris@69: 
Chris@69: #endif
Chris@69: 
Chris@69: #endif /* CUSTOM_MODES */
Chris@69: 
Chris@69: int opus_custom_encoder_ctl(CELTEncoder * OPUS_RESTRICT st, int request, ...)
Chris@69: {
Chris@69:    va_list ap;
Chris@69: 
Chris@69:    va_start(ap, request);
Chris@69:    switch (request)
Chris@69:    {
Chris@69:       case OPUS_SET_COMPLEXITY_REQUEST:
Chris@69:       {
Chris@69:          int value = va_arg(ap, opus_int32);
Chris@69:          if (value<0 || value>10)
Chris@69:             goto bad_arg;
Chris@69:          st->complexity = value;
Chris@69:       }
Chris@69:       break;
Chris@69:       case CELT_SET_START_BAND_REQUEST:
Chris@69:       {
Chris@69:          opus_int32 value = va_arg(ap, opus_int32);
Chris@69:          if (value<0 || value>=st->mode->nbEBands)
Chris@69:             goto bad_arg;
Chris@69:          st->start = value;
Chris@69:       }
Chris@69:       break;
Chris@69:       case CELT_SET_END_BAND_REQUEST:
Chris@69:       {
Chris@69:          opus_int32 value = va_arg(ap, opus_int32);
Chris@69:          if (value<1 || value>st->mode->nbEBands)
Chris@69:             goto bad_arg;
Chris@69:          st->end = value;
Chris@69:       }
Chris@69:       break;
Chris@69:       case CELT_SET_PREDICTION_REQUEST:
Chris@69:       {
Chris@69:          int value = va_arg(ap, opus_int32);
Chris@69:          if (value<0 || value>2)
Chris@69:             goto bad_arg;
Chris@69:          st->disable_pf = value<=1;
Chris@69:          st->force_intra = value==0;
Chris@69:       }
Chris@69:       break;
Chris@69:       case OPUS_SET_PACKET_LOSS_PERC_REQUEST:
Chris@69:       {
Chris@69:          int value = va_arg(ap, opus_int32);
Chris@69:          if (value<0 || value>100)
Chris@69:             goto bad_arg;
Chris@69:          st->loss_rate = value;
Chris@69:       }
Chris@69:       break;
Chris@69:       case OPUS_SET_VBR_CONSTRAINT_REQUEST:
Chris@69:       {
Chris@69:          opus_int32 value = va_arg(ap, opus_int32);
Chris@69:          st->constrained_vbr = value;
Chris@69:       }
Chris@69:       break;
Chris@69:       case OPUS_SET_VBR_REQUEST:
Chris@69:       {
Chris@69:          opus_int32 value = va_arg(ap, opus_int32);
Chris@69:          st->vbr = value;
Chris@69:       }
Chris@69:       break;
Chris@69:       case OPUS_SET_BITRATE_REQUEST:
Chris@69:       {
Chris@69:          opus_int32 value = va_arg(ap, opus_int32);
Chris@69:          if (value<=500 && value!=OPUS_BITRATE_MAX)
Chris@69:             goto bad_arg;
Chris@69:          value = IMIN(value, 260000*st->channels);
Chris@69:          st->bitrate = value;
Chris@69:       }
Chris@69:       break;
Chris@69:       case CELT_SET_CHANNELS_REQUEST:
Chris@69:       {
Chris@69:          opus_int32 value = va_arg(ap, opus_int32);
Chris@69:          if (value<1 || value>2)
Chris@69:             goto bad_arg;
Chris@69:          st->stream_channels = value;
Chris@69:       }
Chris@69:       break;
Chris@69:       case OPUS_SET_LSB_DEPTH_REQUEST:
Chris@69:       {
Chris@69:           opus_int32 value = va_arg(ap, opus_int32);
Chris@69:           if (value<8 || value>24)
Chris@69:              goto bad_arg;
Chris@69:           st->lsb_depth=value;
Chris@69:       }
Chris@69:       break;
Chris@69:       case OPUS_GET_LSB_DEPTH_REQUEST:
Chris@69:       {
Chris@69:           opus_int32 *value = va_arg(ap, opus_int32*);
Chris@69:           *value=st->lsb_depth;
Chris@69:       }
Chris@69:       break;
Chris@69:       case OPUS_SET_PHASE_INVERSION_DISABLED_REQUEST:
Chris@69:       {
Chris@69:           opus_int32 value = va_arg(ap, opus_int32);
Chris@69:           if(value<0 || value>1)
Chris@69:           {
Chris@69:              goto bad_arg;
Chris@69:           }
Chris@69:           st->disable_inv = value;
Chris@69:       }
Chris@69:       break;
Chris@69:       case OPUS_GET_PHASE_INVERSION_DISABLED_REQUEST:
Chris@69:       {
Chris@69:           opus_int32 *value = va_arg(ap, opus_int32*);
Chris@69:           if (!value)
Chris@69:           {
Chris@69:              goto bad_arg;
Chris@69:           }
Chris@69:           *value = st->disable_inv;
Chris@69:       }
Chris@69:       break;
Chris@69:       case OPUS_RESET_STATE:
Chris@69:       {
Chris@69:          int i;
Chris@69:          opus_val16 *oldBandE, *oldLogE, *oldLogE2;
Chris@69:          oldBandE = (opus_val16*)(st->in_mem+st->channels*(st->mode->overlap+COMBFILTER_MAXPERIOD));
Chris@69:          oldLogE = oldBandE + st->channels*st->mode->nbEBands;
Chris@69:          oldLogE2 = oldLogE + st->channels*st->mode->nbEBands;
Chris@69:          OPUS_CLEAR((char*)&st->ENCODER_RESET_START,
Chris@69:                opus_custom_encoder_get_size(st->mode, st->channels)-
Chris@69:                ((char*)&st->ENCODER_RESET_START - (char*)st));
Chris@69:          for (i=0;i<st->channels*st->mode->nbEBands;i++)
Chris@69:             oldLogE[i]=oldLogE2[i]=-QCONST16(28.f,DB_SHIFT);
Chris@69:          st->vbr_offset = 0;
Chris@69:          st->delayedIntra = 1;
Chris@69:          st->spread_decision = SPREAD_NORMAL;
Chris@69:          st->tonal_average = 256;
Chris@69:          st->hf_average = 0;
Chris@69:          st->tapset_decision = 0;
Chris@69:       }
Chris@69:       break;
Chris@69: #ifdef CUSTOM_MODES
Chris@69:       case CELT_SET_INPUT_CLIPPING_REQUEST:
Chris@69:       {
Chris@69:          opus_int32 value = va_arg(ap, opus_int32);
Chris@69:          st->clip = value;
Chris@69:       }
Chris@69:       break;
Chris@69: #endif
Chris@69:       case CELT_SET_SIGNALLING_REQUEST:
Chris@69:       {
Chris@69:          opus_int32 value = va_arg(ap, opus_int32);
Chris@69:          st->signalling = value;
Chris@69:       }
Chris@69:       break;
Chris@69:       case CELT_SET_ANALYSIS_REQUEST:
Chris@69:       {
Chris@69:          AnalysisInfo *info = va_arg(ap, AnalysisInfo *);
Chris@69:          if (info)
Chris@69:             OPUS_COPY(&st->analysis, info, 1);
Chris@69:       }
Chris@69:       break;
Chris@69:       case CELT_SET_SILK_INFO_REQUEST:
Chris@69:       {
Chris@69:          SILKInfo *info = va_arg(ap, SILKInfo *);
Chris@69:          if (info)
Chris@69:             OPUS_COPY(&st->silk_info, info, 1);
Chris@69:       }
Chris@69:       break;
Chris@69:       case CELT_GET_MODE_REQUEST:
Chris@69:       {
Chris@69:          const CELTMode ** value = va_arg(ap, const CELTMode**);
Chris@69:          if (value==0)
Chris@69:             goto bad_arg;
Chris@69:          *value=st->mode;
Chris@69:       }
Chris@69:       break;
Chris@69:       case OPUS_GET_FINAL_RANGE_REQUEST:
Chris@69:       {
Chris@69:          opus_uint32 * value = va_arg(ap, opus_uint32 *);
Chris@69:          if (value==0)
Chris@69:             goto bad_arg;
Chris@69:          *value=st->rng;
Chris@69:       }
Chris@69:       break;
Chris@69:       case OPUS_SET_LFE_REQUEST:
Chris@69:       {
Chris@69:           opus_int32 value = va_arg(ap, opus_int32);
Chris@69:           st->lfe = value;
Chris@69:       }
Chris@69:       break;
Chris@69:       case OPUS_SET_ENERGY_MASK_REQUEST:
Chris@69:       {
Chris@69:           opus_val16 *value = va_arg(ap, opus_val16*);
Chris@69:           st->energy_mask = value;
Chris@69:       }
Chris@69:       break;
Chris@69:       default:
Chris@69:          goto bad_request;
Chris@69:    }
Chris@69:    va_end(ap);
Chris@69:    return OPUS_OK;
Chris@69: bad_arg:
Chris@69:    va_end(ap);
Chris@69:    return OPUS_BAD_ARG;
Chris@69: bad_request:
Chris@69:    va_end(ap);
Chris@69:    return OPUS_UNIMPLEMENTED;
Chris@69: }