shithub: opus

Download patch

ref: bb8fa1fca2819ac4b7eb901a765a8854479c0b52
parent: 6c37fb150c0b316377cd8749449d5ea2c50af16d
author: Jean-Marc Valin <[email protected]>
date: Wed Jun 2 20:33:42 EDT 2010 

Allocation table now in bits/sample


--- a/libcelt/modes.c
+++ b/libcelt/modes.c
@@ -206,8 +206,9 @@
       {
          int edge, low, high;
          celt_int32 alloc;
-
-         alloc = mode->shortMdctSize*band_allocation[i*BARK_BANDS+j];
+         /* This compensates for the sampling rate. The 46000 here reflects the fact that we
+            originally tuned using both 44.1 and 48 kHz. */
+         alloc = ((celt_int32)46000*mode->shortMdctSize/mode->Fs)*band_allocation[i*BARK_BANDS+j];
          low = bark_freq[j];
          high = bark_freq[j+1];
 
@@ -216,6 +217,7 @@
          {
             celt_int32 num;
             int den, bits;
+            int N = (mode->eBands[eband+1]-mode->eBands[eband]);
             num = alloc * (edge-low);
             den = high-low;
             /* Divide with rounding */
@@ -225,6 +227,7 @@
             allocVectors[i*mode->nbEBands+eband] -= 1<<BITRES;
             if (allocVectors[i*mode->nbEBands+eband]<0)
                allocVectors[i*mode->nbEBands+eband]=0;
+            allocVectors[i*mode->nbEBands+eband] = (2*allocVectors[i*mode->nbEBands+eband]+N)/(2*N);
             /* Remove the part of the band we just allocated */
             low = edge;
             alloc -= bits;
@@ -238,13 +241,23 @@
       }
       if (eband < mode->nbEBands)
       {
+         int N = (mode->eBands[eband+1]-mode->eBands[eband]);
          allocVectors[i*mode->nbEBands+eband] = (current+128)>>(8-BITRES);
          /* Same hack as above FIXME: again */
          allocVectors[i*mode->nbEBands+eband] -= 1<<BITRES;
          if (allocVectors[i*mode->nbEBands+eband]<0)
             allocVectors[i*mode->nbEBands+eband]=0;
+         allocVectors[i*mode->nbEBands+eband] = (2*allocVectors[i*mode->nbEBands+eband]+N)/(2*N);
       }
    }
+   /*for (i=0;i<BITALLOC_SIZE;i++)
+   {
+      for (j=0;j<mode->nbEBands;j++)
+         printf ("%d ", allocVectors[i*mode->nbEBands+j]);
+      printf ("\n");
+   }
+   exit(0);*/
+
    mode->allocVectors = allocVectors;
 }
 
--- a/libcelt/rate.c
+++ b/libcelt/rate.c
@@ -207,7 +207,8 @@
       int mid = (lo+hi) >> 1;
       for (j=start;j<len;j++)
       {
-         bits1[j] = (C*M*m->allocVectors[mid*len+j] + offsets[j]);
+         int N = m->eBands[j+1]-m->eBands[j];
+         bits1[j] = (C*M*N*m->allocVectors[mid*len+j] + offsets[j]);
          if (bits1[j] < 0)
             bits1[j] = 0;
          psum += bits1[j];
@@ -223,8 +224,9 @@
    /*printf ("interp between %d and %d\n", lo, hi);*/
    for (j=start;j<len;j++)
    {
-      bits1[j] = C*M*m->allocVectors[lo*len+j] + offsets[j];
-      bits2[j] = C*M*m->allocVectors[hi*len+j] + offsets[j];
+      int N = m->eBands[j+1]-m->eBands[j];
+      bits1[j] = C*M*N*m->allocVectors[lo*len+j] + offsets[j];
+      bits2[j] = C*M*N*m->allocVectors[hi*len+j] + offsets[j];
       if (bits1[j] < 0)
          bits1[j] = 0;
       if (bits2[j] < 0)