|
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 |
|