Blame libcelt/rate.c

Packit 664db3
/* (C) 2007-2008 Jean-Marc Valin, CSIRO
Packit 664db3
*/
Packit 664db3
/*
Packit 664db3
   Redistribution and use in source and binary forms, with or without
Packit 664db3
   modification, are permitted provided that the following conditions
Packit 664db3
   are met:
Packit 664db3
   
Packit 664db3
   - Redistributions of source code must retain the above copyright
Packit 664db3
   notice, this list of conditions and the following disclaimer.
Packit 664db3
   
Packit 664db3
   - Redistributions in binary form must reproduce the above copyright
Packit 664db3
   notice, this list of conditions and the following disclaimer in the
Packit 664db3
   documentation and/or other materials provided with the distribution.
Packit 664db3
   
Packit 664db3
   - Neither the name of the Xiph.org Foundation nor the names of its
Packit 664db3
   contributors may be used to endorse or promote products derived from
Packit 664db3
   this software without specific prior written permission.
Packit 664db3
   
Packit 664db3
   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
Packit 664db3
   ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
Packit 664db3
   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
Packit 664db3
   A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
Packit 664db3
   CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
Packit 664db3
   EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
Packit 664db3
   PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
Packit 664db3
   PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
Packit 664db3
   LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
Packit 664db3
   NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
Packit 664db3
   SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
Packit 664db3
*/
Packit 664db3
Packit 664db3
#ifdef HAVE_CONFIG_H
Packit 664db3
#include "config.h"
Packit 664db3
#endif
Packit 664db3
Packit 664db3
#include <math.h>
Packit 664db3
#include "modes.h"
Packit 664db3
#include "cwrs.h"
Packit 664db3
#include "arch.h"
Packit 664db3
#include "os_support.h"
Packit 664db3
Packit 664db3
#include "entcode.h"
Packit 664db3
#include "rate.h"
Packit 664db3
Packit 664db3
Packit 664db3
#ifndef STATIC_MODES
Packit 664db3
Packit 664db3
celt_int16_t **compute_alloc_cache(CELTMode *m, int C)
Packit 664db3
{
Packit 664db3
   int i, prevN;
Packit 664db3
   celt_int16_t **bits;
Packit 664db3
   const celt_int16_t *eBands = m->eBands;
Packit 664db3
Packit 664db3
   bits = celt_alloc(m->nbEBands*sizeof(celt_int16_t*));
Packit 664db3
   
Packit 664db3
   prevN = -1;
Packit 664db3
   for (i=0;i<m->nbEBands;i++)
Packit 664db3
   {
Packit 664db3
      int N = C*(eBands[i+1]-eBands[i]);
Packit 664db3
      if (N == prevN && eBands[i] < m->pitchEnd)
Packit 664db3
      {
Packit 664db3
         bits[i] = bits[i-1];
Packit 664db3
      } else {
Packit 664db3
         bits[i] = celt_alloc(MAX_PULSES*sizeof(celt_int16_t));
Packit 664db3
         get_required_bits(bits[i], N, MAX_PULSES, BITRES);
Packit 664db3
         prevN = N;
Packit 664db3
      }
Packit 664db3
   }
Packit 664db3
   return bits;
Packit 664db3
}
Packit 664db3
Packit 664db3
#endif /* !STATIC_MODES */
Packit 664db3
Packit 664db3
Packit 664db3
Packit 664db3
static int interp_bits2pulses(const CELTMode *m, int *bits1, int *bits2, int *ebits1, int *ebits2, int total, int *bits, int *ebits, int len)
Packit 664db3
{
Packit 664db3
   int esum, psum;
Packit 664db3
   int lo, hi;
Packit 664db3
   int j;
Packit 664db3
   const int C = CHANNELS(m);
Packit 664db3
   SAVE_STACK;
Packit 664db3
   lo = 0;
Packit 664db3
   hi = 1<
Packit 664db3
   while (hi-lo != 1)
Packit 664db3
   {
Packit 664db3
      int mid = (lo+hi)>>1;
Packit 664db3
      psum = 0;
Packit 664db3
      esum = 0;
Packit 664db3
      for (j=0;j
Packit 664db3
      {
Packit 664db3
         esum += (((1<<BITRES)-mid)*ebits1[j] + mid*ebits2[j] + (1<<(BITRES-1)))>>BITRES;
Packit 664db3
         psum += ((1<
Packit 664db3
      }
Packit 664db3
      if (psum > (total-C*esum)<
Packit 664db3
         hi = mid;
Packit 664db3
      else
Packit 664db3
         lo = mid;
Packit 664db3
   }
Packit 664db3
   esum = 0;
Packit 664db3
   psum = 0;
Packit 664db3
   /*printf ("interp bisection gave %d\n", lo);*/
Packit 664db3
   for (j=0;j
Packit 664db3
   {
Packit 664db3
      ebits[j] = (((1<<BITRES)-lo)*ebits1[j] + lo*ebits2[j] + (1<<(BITRES-1)))>>BITRES;
Packit 664db3
      esum += ebits[j];
Packit 664db3
   }
Packit 664db3
   for (j=0;j
Packit 664db3
   {
Packit 664db3
      bits[j] = ((1<
Packit 664db3
      psum += bits[j];
Packit 664db3
   }
Packit 664db3
   /* Allocate the remaining bits */
Packit 664db3
   {
Packit 664db3
      int left, perband;
Packit 664db3
      left = ((total-C*esum)<
Packit 664db3
      perband = left/len;
Packit 664db3
      for (j=0;j
Packit 664db3
         bits[j] += perband;
Packit 664db3
      left = left-len*perband;
Packit 664db3
      for (j=0;j
Packit 664db3
         bits[j]++;
Packit 664db3
   }
Packit 664db3
   RESTORE_STACK;
Packit 664db3
   return (total-C*esum)<
Packit 664db3
}
Packit 664db3
Packit 664db3
void compute_allocation(const CELTMode *m, int *offsets, const int *stereo_mode, int total, int *pulses, int *ebits)
Packit 664db3
{
Packit 664db3
   int lo, hi, len, j;
Packit 664db3
   int remaining_bits;
Packit 664db3
   VARDECL(int, bits1);
Packit 664db3
   VARDECL(int, bits2);
Packit 664db3
   VARDECL(int, ebits1);
Packit 664db3
   VARDECL(int, ebits2);
Packit 664db3
   const int C = CHANNELS(m);
Packit 664db3
   SAVE_STACK;
Packit 664db3
   
Packit 664db3
   len = m->nbEBands;
Packit 664db3
   ALLOC(bits1, len, int);
Packit 664db3
   ALLOC(bits2, len, int);
Packit 664db3
   ALLOC(ebits1, len, int);
Packit 664db3
   ALLOC(ebits2, len, int);
Packit 664db3
Packit 664db3
   lo = 0;
Packit 664db3
   hi = m->nbAllocVectors - 1;
Packit 664db3
   while (hi-lo != 1)
Packit 664db3
   {
Packit 664db3
      int psum = 0;
Packit 664db3
      int mid = (lo+hi) >> 1;
Packit 664db3
      for (j=0;j
Packit 664db3
      {
Packit 664db3
         bits1[j] = (m->allocVectors[mid*len+j] + offsets[j])<
Packit 664db3
         if (bits1[j] < 0)
Packit 664db3
            bits1[j] = 0;
Packit 664db3
         psum += bits1[j];
Packit 664db3
         /*printf ("%d ", bits[j]);*/
Packit 664db3
      }
Packit 664db3
      /*printf ("\n");*/
Packit 664db3
      if (psum > (total-C*m->energy_alloc[mid*(len+1)+len])<
Packit 664db3
         hi = mid;
Packit 664db3
      else
Packit 664db3
         lo = mid;
Packit 664db3
      /*printf ("lo = %d, hi = %d\n", lo, hi);*/
Packit 664db3
   }
Packit 664db3
   /*printf ("interp between %d and %d\n", lo, hi);*/
Packit 664db3
   for (j=0;j
Packit 664db3
   {
Packit 664db3
      ebits1[j] = m->energy_alloc[lo*(len+1)+j];
Packit 664db3
      ebits2[j] = m->energy_alloc[hi*(len+1)+j];
Packit 664db3
      bits1[j] = m->allocVectors[lo*len+j] + offsets[j];
Packit 664db3
      bits2[j] = m->allocVectors[hi*len+j] + offsets[j];
Packit 664db3
      if (bits1[j] < 0)
Packit 664db3
         bits1[j] = 0;
Packit 664db3
      if (bits2[j] < 0)
Packit 664db3
         bits2[j] = 0;
Packit 664db3
   }
Packit 664db3
   remaining_bits = interp_bits2pulses(m, bits1, bits2, ebits1, ebits2, total, pulses, ebits, len);
Packit 664db3
   RESTORE_STACK;
Packit 664db3
}
Packit 664db3