Using high-band noise less aggressively
[opus.git] / libcelt / quant_bands.c
index a54a78d..ba80fa9 100644 (file)
@@ -1,5 +1,6 @@
-/* (C) 2007 Jean-Marc Valin, CSIRO
-*/
+/* Copyright (c) 2007-2008 CSIRO
+   Copyright (c) 2007-2009 Xiph.Org Foundation
+   Written by Jean-Marc Valin */
 /*
    Redistribution and use in source and binary forms, with or without
    modification, are permitted provided that the following conditions
 #include <math.h>
 #include "os_support.h"
 #include "arch.h"
+#include "mathops.h"
+#include "stack_alloc.h"
 
 #ifdef FIXED_POINT
-const celt_word16_t eMeans[24] = {11520, -2048, -3072, -640, 256, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
+/* Mean energy in each band quantized in Q6 */
+const signed char eMeans[25] = {
+     124,122,115,106,100,
+      95, 91, 90, 99, 96,
+      94, 93, 98, 91, 86,
+      90, 88, 88, 90, 85,
+      64, 64, 64, 64, 64};
 #else
-const float eMeans[24] = {45.f, -8.f, -12.f, -2.5f, 1.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f};
+/* Mean energy in each band quantized in Q6 and converted back to float */
+const celt_word16 eMeans[25] = {
+      7.750000f, 7.625000f, 7.187500f, 6.625000f, 6.250000f,
+      5.937500f, 5.687500f, 5.625000f, 6.187500f, 6.000000f,
+      5.875000f, 5.812500f, 6.125000f, 5.687500f, 5.375000f,
+      5.625000f, 5.500000f, 5.500000f, 5.625000f, 5.312500f,
+      4.000000f, 4.000000f, 4.000000f, 4.000000f, 4.000000f};
 #endif
-
-/*const int frac[24] = {4, 3, 3, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2};*/
-const int frac[24] = {8, 6, 5, 4, 3, 3, 3, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2};
-
+/* prediction coefficients: 0.9, 0.8, 0.65, 0.5 */
 #ifdef FIXED_POINT
-#define Q8 256.f
-#define Q8_1 (1.f/256.f)
+static const celt_word16 pred_coef[4] = {29440, 26112, 21248, 16384};
 #else
-#define Q8 1.f
-#define Q8_1 1.f
+static const celt_word16 pred_coef[4] = {29440/32768., 26112/32768., 21248/32768., 16384/32768.};
 #endif
-static void quant_energy_mono(const CELTMode *m, celt_ener_t *eBands, celt_word16_t *oldEBands, int budget, ec_enc *enc)
+
+static int intra_decision(const celt_word16 *eBands, celt_word16 *oldEBands, int start, int end, int len, int C)
 {
-   int i;
-   int bits;
-   celt_word16_t prev = 0;
-   float coef = m->ePredCoef;
-   VARDECL(celt_word16_t *error);
-   /* The .7 is a heuristic */
-   float beta = .7*coef;
-   
-   ALLOC(error, m->nbEBands, celt_word16_t);
-   bits = ec_enc_tell(enc, 0);
-   for (i=0;i<m->nbEBands;i++)
+   int c, i;
+   celt_word32 dist = 0;
+   for (c=0;c<C;c++)
    {
-      int qi;
-      celt_word16_t q;   /* dB */
-      celt_word16_t res; /* dB */
-      celt_word16_t x;   /* dB */
-      celt_word16_t f;   /* Q8 */
-      celt_word16_t mean = (1-coef)*eMeans[i];
-      x = DB_SCALING*20*log10(.3+ENER_SCALING_1*eBands[i]);
-      res = DB_SCALING*6;
-      f = QCONST16(1.f,8)*(x-mean-coef*oldEBands[i]-prev*1.f)/res;
-#ifdef FIXED_POINT
-      /* Rounding to nearest integer here is really important! */
-      qi = (f+128)>>8;
-#else
-      qi = (int)floor(.5+f);
-#endif
-      /*ec_laplace_encode(enc, qi, i==0?11192:6192);*/
-      /*ec_laplace_encode(enc, qi, 8500-i*200);*/
-      /* If we don't have enough bits to encode all the energy, just assume something safe. */
-      if (ec_enc_tell(enc, 0) - bits > budget)
-         qi = -1;
-      else
-         ec_laplace_encode(enc, qi, 6000-i*200);
-      q = qi*res;
-      error[i] = f - SHL16(qi,8);
-      
-      /*printf("%d ", qi);*/
-      /*printf("%f %f ", pred+prev+q, x);*/
-      /*printf("%f ", x-pred);*/
-      
-      oldEBands[i] = mean+coef*oldEBands[i]+prev+q;
-      
-      prev = mean+prev+(1-beta)*q;
+      for (i=start;i<end;i++)
+      {
+         celt_word16 d = SHR16(SUB16(eBands[i+c*len], oldEBands[i+c*len]),2);
+         dist = MAC16_16(dist, d,d);
+      }
    }
-   /*bits = ec_enc_tell(enc, 0) - bits;*/
-   /*printf ("%d\n", bits);*/
+   return SHR32(dist,2*DB_SHIFT-4) > 2*C*(end-start);
+}
+
+#ifndef STATIC_MODES
+
+celt_int16 *quant_prob_alloc(const CELTMode *m)
+{
+   int i;
+   celt_int16 *prob;
+   prob = celt_alloc(4*m->nbEBands*sizeof(celt_int16));
+   if (prob==NULL)
+     return NULL;
    for (i=0;i<m->nbEBands;i++)
    {
-      int q2;
-      celt_word16_t offset = (error[i]+QCONST16(.5f,8))*frac[i];
-      /* FIXME: Instead of giving up without warning, we should degrade everything gracefully */
-      if (ec_enc_tell(enc, 0) - bits +EC_ILOG(frac[i])> budget)
-         break;
-#ifdef FIXED_POINT
-      /* Has to be without rounding */
-      q2 = offset>>8;
-#else
-      q2 = (int)floor(Q8_1*offset);
-#endif
-      if (q2 > frac[i]-1)
-         q2 = frac[i]-1;
-      ec_enc_uint(enc, q2, frac[i]);
-      offset = (Q8*(q2+.5)/frac[i])-QCONST16(.5f,8);
-      oldEBands[i] += PSHR32(MULT16_16(DB_SCALING*6,offset),8);
-      /*printf ("%f ", error[i] - offset);*/
+      prob[2*i] = 7000-i*200;
+      prob[2*i+1] = ec_laplace_get_start_freq(prob[2*i]);
    }
    for (i=0;i<m->nbEBands;i++)
    {
-      eBands[i] = ENER_SCALING*(pow(10, .05*DB_SCALING_1*oldEBands[i])-.3);
-      if (eBands[i] < 0)
-         eBands[i] = 0;
+      prob[2*m->nbEBands+2*i] = 9000-i*220;
+      prob[2*m->nbEBands+2*i+1] = ec_laplace_get_start_freq(prob[2*m->nbEBands+2*i]);
    }
-   /*printf ("%d\n", ec_enc_tell(enc, 0)-9);*/
+   return prob;
+}
 
-   /*printf ("\n");*/
+void quant_prob_free(const celt_int16 *freq)
+{
+   celt_free((celt_int16*)freq);
 }
+#endif
 
-static void unquant_energy_mono(const CELTMode *m, celt_ener_t *eBands, celt_word16_t *oldEBands, int budget, ec_dec *dec)
+static void quant_coarse_energy_impl(const CELTMode *m, int start, int end,
+      const celt_word16 *eBands, celt_word16 *oldEBands, int budget,
+      const celt_int16 *prob, celt_word16 *error, ec_enc *enc, int _C, int LM,
+      int intra, celt_word16 max_decay)
 {
-   int i;
-   int bits;
-   celt_word16_t prev = 0;
-   float coef = m->ePredCoef;
-   /* The .7 is a heuristic */
-   float beta = .7*coef;
-   bits = ec_dec_tell(dec, 0);
-   for (i=0;i<m->nbEBands;i++)
-   {
-      int qi;
-      celt_word16_t q;
-      celt_word16_t res;
-      celt_word16_t mean = (1-coef)*eMeans[i];
-      res = DB_SCALING*6.;
-      /* If we didn't have enough bits to encode all the energy, just assume something safe. */
-      if (ec_dec_tell(dec, 0) - bits > budget)
-         qi = -1;
-      else
-         qi = ec_laplace_decode(dec, 6000-i*200);
-      q = qi*res;
-      
-      /*printf("%d ", qi);*/
-      /*printf("%f %f ", pred+prev+q, x);*/
-      /*printf("%f ", x-pred);*/
-      
-      oldEBands[i] = mean+coef*oldEBands[i]+prev+q;
-      
-      prev = mean+prev+(1-beta)*q;
-   }
-   for (i=0;i<m->nbEBands;i++)
+   const int C = CHANNELS(_C);
+   int i, c;
+   celt_word32 prev[2] = {0,0};
+   celt_word16 coef;
+   celt_word16 beta;
+
+   coef = pred_coef[LM];
+
+   ec_enc_bit_prob(enc, intra, 8192);
+   if (intra)
    {
-      int q2;
-      celt_word16_t offset;
-      if (ec_dec_tell(dec, 0) - bits +EC_ILOG(frac[i])> budget)
-         break;
-      q2 = ec_dec_uint(dec, frac[i]);
-      offset = (Q8*(q2+.5)/frac[i])-QCONST16(.5f,8);
-      oldEBands[i] += PSHR32(MULT16_16(DB_SCALING*6,offset),8);
+      coef = 0;
+      prob += 2*m->nbEBands;
    }
-   for (i=0;i<m->nbEBands;i++)
+   /* No theoretical justification for this, it just works */
+   beta = MULT16_16_P15(coef,coef);
+   /* Encode at a fixed coarse resolution */
+   for (i=start;i<end;i++)
    {
-      /*printf ("%f ", error[i] - offset);*/
-      eBands[i] = ENER_SCALING*(pow(10, .05*DB_SCALING_1*oldEBands[i])-.3);
-      if (eBands[i] < 0)
-         eBands[i] = 0;
+      c=0;
+      do {
+         int bits_left;
+         int qi;
+         celt_word16 q;
+         celt_word16 x;
+         celt_word32 f;
+         x = eBands[i+c*m->nbEBands];
+#ifdef FIXED_POINT
+         f = SHL32(EXTEND32(x),15) -MULT16_16(coef,oldEBands[i+c*m->nbEBands])-prev[c];
+         /* Rounding to nearest integer here is really important! */
+         qi = (f+QCONST32(.5,DB_SHIFT+15))>>(DB_SHIFT+15);
+#else
+         f = x-coef*oldEBands[i+c*m->nbEBands]-prev[c];
+         /* Rounding to nearest integer here is really important! */
+         qi = (int)floor(.5f+f);
+#endif
+         /* Prevent the energy from going down too quickly (e.g. for bands
+            that have just one bin) */
+         if (qi < 0 && x < oldEBands[i+c*m->nbEBands]-max_decay)
+         {
+            qi += (int)SHR16(oldEBands[i+c*m->nbEBands]-max_decay-x, DB_SHIFT);
+            if (qi > 0)
+               qi = 0;
+         }
+         /* If we don't have enough bits to encode all the energy, just assume something safe.
+            We allow slightly busting the budget here */
+         bits_left = budget-(int)ec_enc_tell(enc, 0)-2*C*(end-i);
+         if (bits_left < 24)
+         {
+            if (qi > 1)
+               qi = 1;
+            if (qi < -1)
+               qi = -1;
+            if (bits_left<8)
+               qi = 0;
+         }
+         ec_laplace_encode_start(enc, &qi, prob[2*i], prob[2*i+1]);
+         error[i+c*m->nbEBands] = PSHR32(f,15) - SHL16(qi,DB_SHIFT);
+         q = SHL16(qi,DB_SHIFT);
+         
+         oldEBands[i+c*m->nbEBands] = PSHR32(MULT16_16(coef,oldEBands[i+c*m->nbEBands]) + prev[c] + SHL32(EXTEND32(q),15), 15);
+         prev[c] = prev[c] + SHL32(EXTEND32(q),15) - MULT16_16(beta,q);
+      } while (++c < C);
    }
-   /*printf ("\n");*/
 }
 
+void quant_coarse_energy(const CELTMode *m, int start, int end, int effEnd,
+      const celt_word16 *eBands, celt_word16 *oldEBands, int budget,
+      const celt_int16 *prob, celt_word16 *error, ec_enc *enc, int _C, int LM,
+      int nbAvailableBytes, int force_intra, int *delayedIntra, int two_pass)
+{
+   const int C = CHANNELS(_C);
+   int intra;
+   celt_word16 max_decay;
+   VARDECL(celt_word16, oldEBands_intra);
+   VARDECL(celt_word16, error_intra);
+   ec_enc enc_start_state;
+   ec_byte_buffer buf_start_state;
+   SAVE_STACK;
 
+   intra = force_intra || (*delayedIntra && nbAvailableBytes > end);
+   if (/*shortBlocks || */intra_decision(eBands, oldEBands, start, effEnd, m->nbEBands, C))
+      *delayedIntra = 1;
+   else
+      *delayedIntra = 0;
 
-void quant_energy(const CELTMode *m, celt_ener_t *eBands, celt_word16_t *oldEBands, int budget, ec_enc *enc)
-{
-   int C;
-   
-   C = m->nbChannels;
+   /* Encode the global flags using a simple probability model
+      (first symbols in the stream) */
+
+#ifdef FIXED_POINT
+      max_decay = MIN32(QCONST16(16,DB_SHIFT), SHL32(EXTEND32(nbAvailableBytes),DB_SHIFT-3));
+#else
+   max_decay = MIN32(16.f, .125f*nbAvailableBytes);
+#endif
+
+   enc_start_state = *enc;
+   buf_start_state = *(enc->buf);
+
+   ALLOC(oldEBands_intra, C*m->nbEBands, celt_word16);
+   ALLOC(error_intra, C*m->nbEBands, celt_word16);
+   CELT_COPY(oldEBands_intra, oldEBands, C*end);
+
+   if (two_pass || intra)
+   {
+      quant_coarse_energy_impl(m, start, end, eBands, oldEBands_intra, budget,
+            prob, error_intra, enc, C, LM, 1, max_decay);
+   }
 
-   if (C==1)
-      quant_energy_mono(m, eBands, oldEBands, budget, enc);
-   else 
-#if 1
+   if (!intra)
    {
-      int c;
-      VARDECL(celt_ener_t *E);
-      ALLOC(E, m->nbEBands, celt_ener_t);
-      for (c=0;c<C;c++)
+      ec_enc enc_intra_state;
+      ec_byte_buffer buf_intra_state;
+      int tell_intra;
+      VARDECL(unsigned char, intra_bits);
+
+      tell_intra = ec_enc_tell(enc, 3);
+
+      enc_intra_state = *enc;
+      buf_intra_state = *(enc->buf);
+
+      ALLOC(intra_bits, buf_intra_state.ptr-buf_start_state.ptr, unsigned char);
+      /* Copy bits from intra bit-stream */
+      CELT_COPY(intra_bits, buf_start_state.ptr, buf_intra_state.ptr-buf_start_state.ptr);
+
+      *enc = enc_start_state;
+      *(enc->buf) = buf_start_state;
+
+      quant_coarse_energy_impl(m, start, end, eBands, oldEBands, budget,
+            prob, error, enc, C, LM, 0, max_decay);
+
+      if (two_pass && ec_enc_tell(enc, 3) > tell_intra)
       {
-         int i;
-         for (i=0;i<m->nbEBands;i++)
-            E[i] = eBands[C*i+c];
-         quant_energy_mono(m, E, oldEBands+c*m->nbEBands, budget/C, enc);
-         for (i=0;i<m->nbEBands;i++)
-            eBands[C*i+c] = E[i];
+         *enc = enc_intra_state;
+         *(enc->buf) = buf_intra_state;
+         /* Copy bits from to bit-stream */
+         CELT_COPY(buf_start_state.ptr, intra_bits, buf_intra_state.ptr-buf_start_state.ptr);
+         CELT_COPY(oldEBands, oldEBands_intra, C*end);
+         CELT_COPY(error, error_intra, C*end);
       }
+   } else {
+      CELT_COPY(oldEBands, oldEBands_intra, C*end);
+      CELT_COPY(error, error_intra, C*end);
    }
+   RESTORE_STACK;
+}
+
+void quant_fine_energy(const CELTMode *m, int start, int end, celt_ener *eBands, celt_word16 *oldEBands, celt_word16 *error, int *fine_quant, ec_enc *enc, int _C)
+{
+   int i, c;
+   const int C = CHANNELS(_C);
+
+   /* Encode finer resolution */
+   for (i=start;i<end;i++)
+   {
+      celt_int16 frac = 1<<fine_quant[i];
+      if (fine_quant[i] <= 0)
+         continue;
+      c=0;
+      do {
+         int q2;
+         celt_word16 offset;
+#ifdef FIXED_POINT
+         /* Has to be without rounding */
+         q2 = (error[i+c*m->nbEBands]+QCONST16(.5f,DB_SHIFT))>>(DB_SHIFT-fine_quant[i]);
 #else
-      if (C==2)
+         q2 = (int)floor((error[i+c*m->nbEBands]+.5f)*frac);
+#endif
+         if (q2 > frac-1)
+            q2 = frac-1;
+         if (q2<0)
+            q2 = 0;
+         ec_enc_bits(enc, q2, fine_quant[i]);
+#ifdef FIXED_POINT
+         offset = SUB16(SHR32(SHL32(EXTEND32(q2),DB_SHIFT)+QCONST16(.5,DB_SHIFT),fine_quant[i]),QCONST16(.5f,DB_SHIFT));
+#else
+         offset = (q2+.5f)*(1<<(14-fine_quant[i]))*(1.f/16384) - .5f;
+#endif
+         oldEBands[i+c*m->nbEBands] += offset;
+         error[i+c*m->nbEBands] -= offset;
+         /*printf ("%f ", error[i] - offset);*/
+      } while (++c < C);
+   }
+}
+
+void quant_energy_finalise(const CELTMode *m, int start, int end, celt_ener *eBands, celt_word16 *oldEBands, celt_word16 *error, int *fine_quant, int *fine_priority, int bits_left, ec_enc *enc, int _C)
+{
+   int i, prio, c;
+   const int C = CHANNELS(_C);
+
+   /* Use up the remaining bits */
+   for (prio=0;prio<2;prio++)
    {
-      int i;
-      int NB = m->nbEBands;
-      float mid[NB];
-      float side[NB];
-      float left;
-      float right;
-      for (i=0;i<NB;i++)
+      for (i=start;i<end && bits_left>=C ;i++)
       {
-         //left = eBands[C*i];
-         //right = eBands[C*i+1];
-         mid[i] = ENER_SCALING_1*sqrt(eBands[C*i]*eBands[C*i] + eBands[C*i+1]*eBands[C*i+1]);
-         side[i] = 20*log10((ENER_SCALING_1*eBands[2*i]+.3)/(ENER_SCALING_1*eBands[2*i+1]+.3));
-         //printf ("%f %f ", mid[i], side[i]);
-      }
-      //printf ("\n");
-      quant_energy_mono(m, mid, oldEBands, enc);
-      for (i=0;i<NB;i++)
-         side[i] = pow(10.f,floor(.5f+side[i])/10.f);
-         
-      //quant_energy_side(m, side, oldEBands+NB, enc);
-      for (i=0;i<NB;i++)
-      {
-         eBands[C*i] = ENER_SCALING*mid[i]*sqrt(side[i]/(1.f+side[i]));
-         eBands[C*i+1] = ENER_SCALING*mid[i]*sqrt(1.f/(1.f+side[i]));
-         //printf ("%f %f ", mid[i], side[i]);
+         if (fine_quant[i] >= 7 || fine_priority[i]!=prio)
+            continue;
+         c=0;
+         do {
+            int q2;
+            celt_word16 offset;
+            q2 = error[i+c*m->nbEBands]<0 ? 0 : 1;
+            ec_enc_bits(enc, q2, 1);
+#ifdef FIXED_POINT
+            offset = SHR16(SHL16(q2,DB_SHIFT)-QCONST16(.5,DB_SHIFT),fine_quant[i]+1);
+#else
+            offset = (q2-.5f)*(1<<(14-fine_quant[i]-1))*(1.f/16384);
+#endif
+            oldEBands[i+c*m->nbEBands] += offset;
+            bits_left--;
+         } while (++c < C);
       }
+   }
+}
 
-   } else {
-      celt_fatal("more than 2 channels not supported");
+void unquant_coarse_energy(const CELTMode *m, int start, int end, celt_ener *eBands, celt_word16 *oldEBands, int intra, const celt_int16 *prob, ec_dec *dec, int _C, int LM)
+{
+   int i, c;
+   celt_word32 prev[2] = {0, 0};
+   celt_word16 coef;
+   celt_word16 beta;
+   const int C = CHANNELS(_C);
+
+   coef = pred_coef[LM];
+
+   if (intra)
+   {
+      coef = 0;
+      prob += 2*m->nbEBands;
    }
+   /* No theoretical justification for this, it just works */
+   beta = MULT16_16_P15(coef,coef);
+
+   /* Decode at a fixed coarse resolution */
+   for (i=start;i<end;i++)
+   {
+      c=0;
+      do {
+         int qi;
+         celt_word16 q;
+         qi = ec_laplace_decode_start(dec, prob[2*i], prob[2*i+1]);
+         q = SHL16(qi,DB_SHIFT);
+
+         oldEBands[i+c*m->nbEBands] = PSHR32(MULT16_16(coef,oldEBands[i+c*m->nbEBands]) + prev[c] + SHL32(EXTEND32(q),15), 15);
+         prev[c] = prev[c] + SHL32(EXTEND32(q),15) - MULT16_16(beta,q);
+      } while (++c < C);
+   }
+}
+
+void unquant_fine_energy(const CELTMode *m, int start, int end, celt_ener *eBands, celt_word16 *oldEBands, int *fine_quant, ec_dec *dec, int _C)
+{
+   int i, c;
+   const int C = CHANNELS(_C);
+   /* Decode finer resolution */
+   for (i=start;i<end;i++)
+   {
+      if (fine_quant[i] <= 0)
+         continue;
+      c=0; 
+      do {
+         int q2;
+         celt_word16 offset;
+         q2 = ec_dec_bits(dec, fine_quant[i]);
+#ifdef FIXED_POINT
+         offset = SUB16(SHR32(SHL32(EXTEND32(q2),DB_SHIFT)+QCONST16(.5,DB_SHIFT),fine_quant[i]),QCONST16(.5f,DB_SHIFT));
+#else
+         offset = (q2+.5f)*(1<<(14-fine_quant[i]))*(1.f/16384) - .5f;
 #endif
+         oldEBands[i+c*m->nbEBands] += offset;
+      } while (++c < C);
+   }
 }
 
+void unquant_energy_finalise(const CELTMode *m, int start, int end, celt_ener *eBands, celt_word16 *oldEBands, int *fine_quant,  int *fine_priority, int bits_left, ec_dec *dec, int _C)
+{
+   int i, prio, c;
+   const int C = CHANNELS(_C);
 
+   /* Use up the remaining bits */
+   for (prio=0;prio<2;prio++)
+   {
+      for (i=start;i<end && bits_left>=C ;i++)
+      {
+         if (fine_quant[i] >= 7 || fine_priority[i]!=prio)
+            continue;
+         c=0;
+         do {
+            int q2;
+            celt_word16 offset;
+            q2 = ec_dec_bits(dec, 1);
+#ifdef FIXED_POINT
+            offset = SHR16(SHL16(q2,DB_SHIFT)-QCONST16(.5,DB_SHIFT),fine_quant[i]+1);
+#else
+            offset = (q2-.5f)*(1<<(14-fine_quant[i]-1))*(1.f/16384);
+#endif
+            oldEBands[i+c*m->nbEBands] += offset;
+            bits_left--;
+         } while (++c < C);
+      }
+   }
+}
 
-void unquant_energy(const CELTMode *m, celt_ener_t *eBands, celt_word16_t *oldEBands, int budget, ec_dec *dec)
+void log2Amp(const CELTMode *m, int start, int end,
+      celt_ener *eBands, celt_word16 *oldEBands, int _C)
 {
-   int C;   
-   C = m->nbChannels;
-
-   if (C==1)
-      unquant_energy_mono(m, eBands, oldEBands, budget, dec);
-   else {
-      int c;
-      VARDECL(celt_ener_t *E);
-      ALLOC(E, m->nbEBands, celt_ener_t);
-      for (c=0;c<C;c++)
+   int c, i;
+   const int C = CHANNELS(_C);
+   c=0;
+   do {
+      for (i=start;i<m->nbEBands;i++)
       {
-         int i;
-         unquant_energy_mono(m, E, oldEBands+c*m->nbEBands, budget/C, dec);
-         for (i=0;i<m->nbEBands;i++)
-            eBands[C*i+c] = E[i];
+         celt_word16 lg = oldEBands[i+c*m->nbEBands]
+                        + SHL16((celt_word16)eMeans[i],6);
+         eBands[i+c*m->nbEBands] = PSHR32(celt_exp2(SHL16(lg,11-DB_SHIFT)),4);
+         if (oldEBands[i+c*m->nbEBands] < -QCONST16(14.f,DB_SHIFT))
+            oldEBands[i+c*m->nbEBands] = -QCONST16(14.f,DB_SHIFT);
       }
-   }
+   } while (++c < C);
+}
+
+void amp2Log2(const CELTMode *m, int effEnd, int end,
+      celt_ener *bandE, celt_word16 *bandLogE, int _C)
+{
+   int c, i;
+   const int C = CHANNELS(_C);
+   c=0;
+   do {
+      for (i=0;i<effEnd;i++)
+         bandLogE[i+c*m->nbEBands] =
+               celt_log2(MAX32(QCONST32(.001f,14),SHL32(bandE[i+c*m->nbEBands],2)))
+               - SHL16((celt_word16)eMeans[i],6);
+      for (i=effEnd;i<end;i++)
+         bandLogE[c*m->nbEBands+i] = -QCONST16(14.f,DB_SHIFT);
+   } while (++c < C);
 }