mirror of
https://github.com/ReactionQuake3/reaction.git
synced 2024-11-25 05:31:21 +00:00
1488 lines
46 KiB
C
1488 lines
46 KiB
C
/* Copyright (C) 2002-2006 Jean-Marc Valin
|
|
File: sb_celp.c
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions
|
|
are met:
|
|
|
|
- Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
|
|
- Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in the
|
|
documentation and/or other materials provided with the distribution.
|
|
|
|
- Neither the name of the Xiph.org Foundation nor the names of its
|
|
contributors may be used to endorse or promote products derived from
|
|
this software without specific prior written permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
|
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
|
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
|
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
|
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
|
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <math.h>
|
|
#include "sb_celp.h"
|
|
#include "filters.h"
|
|
#include "lpc.h"
|
|
#include "lsp.h"
|
|
#include "stack_alloc.h"
|
|
#include "cb_search.h"
|
|
#include "quant_lsp.h"
|
|
#include "vq.h"
|
|
#include "ltp.h"
|
|
#include "arch.h"
|
|
#include "math_approx.h"
|
|
#include "os_support.h"
|
|
|
|
#ifndef NULL
|
|
#define NULL 0
|
|
#endif
|
|
|
|
/* Default size for the encoder and decoder stack (can be changed at compile time).
|
|
This does not apply when using variable-size arrays or alloca. */
|
|
#ifndef SB_ENC_STACK
|
|
#define SB_ENC_STACK (10000*sizeof(spx_sig_t))
|
|
#endif
|
|
|
|
#ifndef SB_DEC_STACK
|
|
#define SB_DEC_STACK (6000*sizeof(spx_sig_t))
|
|
#endif
|
|
|
|
|
|
#ifdef DISABLE_WIDEBAND
|
|
void *sb_encoder_init(const SpeexMode *m)
|
|
{
|
|
speex_fatal("Wideband and Ultra-wideband are disabled");
|
|
return NULL;
|
|
}
|
|
void sb_encoder_destroy(void *state)
|
|
{
|
|
speex_fatal("Wideband and Ultra-wideband are disabled");
|
|
}
|
|
int sb_encode(void *state, void *vin, SpeexBits *bits)
|
|
{
|
|
speex_fatal("Wideband and Ultra-wideband are disabled");
|
|
return -2;
|
|
}
|
|
void *sb_decoder_init(const SpeexMode *m)
|
|
{
|
|
speex_fatal("Wideband and Ultra-wideband are disabled");
|
|
return NULL;
|
|
}
|
|
void sb_decoder_destroy(void *state)
|
|
{
|
|
speex_fatal("Wideband and Ultra-wideband are disabled");
|
|
}
|
|
int sb_decode(void *state, SpeexBits *bits, void *vout)
|
|
{
|
|
speex_fatal("Wideband and Ultra-wideband are disabled");
|
|
return -2;
|
|
}
|
|
int sb_encoder_ctl(void *state, int request, void *ptr)
|
|
{
|
|
speex_fatal("Wideband and Ultra-wideband are disabled");
|
|
return -2;
|
|
}
|
|
int sb_decoder_ctl(void *state, int request, void *ptr)
|
|
{
|
|
speex_fatal("Wideband and Ultra-wideband are disabled");
|
|
return -2;
|
|
}
|
|
#else
|
|
|
|
|
|
#ifndef M_PI
|
|
#define M_PI 3.14159265358979323846 /* pi */
|
|
#endif
|
|
|
|
#define sqr(x) ((x)*(x))
|
|
|
|
#define SUBMODE(x) st->submodes[st->submodeID]->x
|
|
|
|
#ifdef FIXED_POINT
|
|
static const spx_word16_t gc_quant_bound[16] = {125, 164, 215, 282, 370, 484, 635, 832, 1090, 1428, 1871, 2452, 3213, 4210, 5516, 7228};
|
|
static const spx_word16_t fold_quant_bound[32] = {
|
|
39, 44, 50, 57, 64, 73, 83, 94,
|
|
106, 120, 136, 154, 175, 198, 225, 255,
|
|
288, 327, 370, 420, 476, 539, 611, 692,
|
|
784, 889, 1007, 1141, 1293, 1465, 1660, 1881};
|
|
#define LSP_MARGIN 410
|
|
#define LSP_DELTA1 6553
|
|
#define LSP_DELTA2 1638
|
|
|
|
#else
|
|
|
|
static const spx_word16_t gc_quant_bound[16] = {
|
|
0.97979, 1.28384, 1.68223, 2.20426, 2.88829, 3.78458, 4.95900, 6.49787,
|
|
8.51428, 11.15642, 14.61846, 19.15484, 25.09895, 32.88761, 43.09325, 56.46588};
|
|
static const spx_word16_t fold_quant_bound[32] = {
|
|
0.30498, 0.34559, 0.39161, 0.44375, 0.50283, 0.56979, 0.64565, 0.73162,
|
|
0.82903, 0.93942, 1.06450, 1.20624, 1.36685, 1.54884, 1.75506, 1.98875,
|
|
2.25355, 2.55360, 2.89361, 3.27889, 3.71547, 4.21018, 4.77076, 5.40598,
|
|
6.12577, 6.94141, 7.86565, 8.91295, 10.09969, 11.44445, 12.96826, 14.69497};
|
|
|
|
#define LSP_MARGIN .05
|
|
#define LSP_DELTA1 .2
|
|
#define LSP_DELTA2 .05
|
|
|
|
#endif
|
|
|
|
#define QMF_ORDER 64
|
|
|
|
#ifdef FIXED_POINT
|
|
static const spx_word16_t h0[64] = {2, -7, -7, 18, 15, -39, -25, 75, 35, -130, -41, 212, 38, -327, -17, 483, -32, -689, 124, 956, -283, -1307, 543, 1780, -973, -2467, 1733, 3633, -3339, -6409, 9059, 30153, 30153, 9059, -6409, -3339, 3633, 1733, -2467, -973, 1780, 543, -1307, -283, 956, 124, -689, -32, 483, -17, -327, 38, 212, -41, -130, 35, 75, -25, -39, 15, 18, -7, -7, 2};
|
|
|
|
#else
|
|
static const float h0[64] = {
|
|
3.596189e-05f, -0.0001123515f,
|
|
-0.0001104587f, 0.0002790277f,
|
|
0.0002298438f, -0.0005953563f,
|
|
-0.0003823631f, 0.00113826f,
|
|
0.0005308539f, -0.001986177f,
|
|
-0.0006243724f, 0.003235877f,
|
|
0.0005743159f, -0.004989147f,
|
|
-0.0002584767f, 0.007367171f,
|
|
-0.0004857935f, -0.01050689f,
|
|
0.001894714f, 0.01459396f,
|
|
-0.004313674f, -0.01994365f,
|
|
0.00828756f, 0.02716055f,
|
|
-0.01485397f, -0.03764973f,
|
|
0.026447f, 0.05543245f,
|
|
-0.05095487f, -0.09779096f,
|
|
0.1382363f, 0.4600981f,
|
|
0.4600981f, 0.1382363f,
|
|
-0.09779096f, -0.05095487f,
|
|
0.05543245f, 0.026447f,
|
|
-0.03764973f, -0.01485397f,
|
|
0.02716055f, 0.00828756f,
|
|
-0.01994365f, -0.004313674f,
|
|
0.01459396f, 0.001894714f,
|
|
-0.01050689f, -0.0004857935f,
|
|
0.007367171f, -0.0002584767f,
|
|
-0.004989147f, 0.0005743159f,
|
|
0.003235877f, -0.0006243724f,
|
|
-0.001986177f, 0.0005308539f,
|
|
0.00113826f, -0.0003823631f,
|
|
-0.0005953563f, 0.0002298438f,
|
|
0.0002790277f, -0.0001104587f,
|
|
-0.0001123515f, 3.596189e-05f
|
|
};
|
|
|
|
#endif
|
|
|
|
extern const spx_word16_t lag_window[];
|
|
extern const spx_word16_t lpc_window[];
|
|
|
|
|
|
void *sb_encoder_init(const SpeexMode *m)
|
|
{
|
|
int i;
|
|
spx_int32_t tmp;
|
|
SBEncState *st;
|
|
const SpeexSBMode *mode;
|
|
|
|
st = (SBEncState*)speex_alloc(sizeof(SBEncState));
|
|
if (!st)
|
|
return NULL;
|
|
st->mode = m;
|
|
mode = (const SpeexSBMode*)m->mode;
|
|
|
|
|
|
st->st_low = speex_encoder_init(mode->nb_mode);
|
|
#if defined(VAR_ARRAYS) || defined (USE_ALLOCA)
|
|
st->stack = NULL;
|
|
#else
|
|
/*st->stack = (char*)speex_alloc_scratch(SB_ENC_STACK);*/
|
|
speex_encoder_ctl(st->st_low, SPEEX_GET_STACK, &st->stack);
|
|
#endif
|
|
|
|
st->full_frame_size = 2*mode->frameSize;
|
|
st->frame_size = mode->frameSize;
|
|
st->subframeSize = mode->subframeSize;
|
|
st->nbSubframes = mode->frameSize/mode->subframeSize;
|
|
st->windowSize = st->frame_size+st->subframeSize;
|
|
st->lpcSize=mode->lpcSize;
|
|
|
|
st->encode_submode = 1;
|
|
st->submodes=mode->submodes;
|
|
st->submodeSelect = st->submodeID=mode->defaultSubmode;
|
|
|
|
tmp=9;
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_QUALITY, &tmp);
|
|
tmp=1;
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_WIDEBAND, &tmp);
|
|
|
|
st->lpc_floor = mode->lpc_floor;
|
|
st->gamma1=mode->gamma1;
|
|
st->gamma2=mode->gamma2;
|
|
st->first=1;
|
|
|
|
st->high=(spx_word16_t*)speex_alloc((st->windowSize-st->frame_size)*sizeof(spx_word16_t));
|
|
|
|
st->h0_mem=(spx_word16_t*)speex_alloc((QMF_ORDER)*sizeof(spx_word16_t));
|
|
st->h1_mem=(spx_word16_t*)speex_alloc((QMF_ORDER)*sizeof(spx_word16_t));
|
|
|
|
st->window= lpc_window;
|
|
|
|
st->lagWindow = lag_window;
|
|
|
|
st->old_lsp = (spx_lsp_t*)speex_alloc(st->lpcSize*sizeof(spx_lsp_t));
|
|
st->old_qlsp = (spx_lsp_t*)speex_alloc(st->lpcSize*sizeof(spx_lsp_t));
|
|
st->interp_qlpc = (spx_coef_t*)speex_alloc(st->lpcSize*sizeof(spx_coef_t));
|
|
st->pi_gain = (spx_word32_t*)speex_alloc((st->nbSubframes)*sizeof(spx_word32_t));
|
|
st->exc_rms = (spx_word16_t*)speex_alloc((st->nbSubframes)*sizeof(spx_word16_t));
|
|
st->innov_rms_save = NULL;
|
|
|
|
st->mem_sp = (spx_mem_t*)speex_alloc((st->lpcSize)*sizeof(spx_mem_t));
|
|
st->mem_sp2 = (spx_mem_t*)speex_alloc((st->lpcSize)*sizeof(spx_mem_t));
|
|
st->mem_sw = (spx_mem_t*)speex_alloc((st->lpcSize)*sizeof(spx_mem_t));
|
|
|
|
for (i=0;i<st->lpcSize;i++)
|
|
st->old_lsp[i]= DIV32(MULT16_16(QCONST16(3.1415927f, LSP_SHIFT), i+1), st->lpcSize+1);
|
|
|
|
#ifndef DISABLE_VBR
|
|
st->vbr_quality = 8;
|
|
st->vbr_enabled = 0;
|
|
st->vbr_max = 0;
|
|
st->vbr_max_high = 20000; /* We just need a big value here */
|
|
st->vad_enabled = 0;
|
|
st->abr_enabled = 0;
|
|
st->relative_quality=0;
|
|
#endif /* #ifndef DISABLE_VBR */
|
|
|
|
st->complexity=2;
|
|
speex_encoder_ctl(st->st_low, SPEEX_GET_SAMPLING_RATE, &st->sampling_rate);
|
|
st->sampling_rate*=2;
|
|
#ifdef ENABLE_VALGRIND
|
|
VALGRIND_MAKE_READABLE(st, (st->stack-(char*)st));
|
|
#endif
|
|
return st;
|
|
}
|
|
|
|
void sb_encoder_destroy(void *state)
|
|
{
|
|
SBEncState *st=(SBEncState*)state;
|
|
|
|
speex_encoder_destroy(st->st_low);
|
|
#if !(defined(VAR_ARRAYS) || defined (USE_ALLOCA))
|
|
/*speex_free_scratch(st->stack);*/
|
|
#endif
|
|
|
|
speex_free(st->high);
|
|
|
|
speex_free(st->h0_mem);
|
|
speex_free(st->h1_mem);
|
|
|
|
speex_free(st->old_lsp);
|
|
speex_free(st->old_qlsp);
|
|
speex_free(st->interp_qlpc);
|
|
speex_free(st->pi_gain);
|
|
speex_free(st->exc_rms);
|
|
|
|
speex_free(st->mem_sp);
|
|
speex_free(st->mem_sp2);
|
|
speex_free(st->mem_sw);
|
|
|
|
|
|
speex_free(st);
|
|
}
|
|
|
|
|
|
int sb_encode(void *state, void *vin, SpeexBits *bits)
|
|
{
|
|
SBEncState *st;
|
|
int i, roots, sub;
|
|
char *stack;
|
|
VARDECL(spx_mem_t *mem);
|
|
VARDECL(spx_sig_t *innov);
|
|
VARDECL(spx_word16_t *target);
|
|
VARDECL(spx_word16_t *syn_resp);
|
|
VARDECL(spx_word32_t *low_pi_gain);
|
|
spx_word16_t *low;
|
|
spx_word16_t *high;
|
|
VARDECL(spx_word16_t *low_exc_rms);
|
|
VARDECL(spx_word16_t *low_innov_rms);
|
|
const SpeexSBMode *mode;
|
|
spx_int32_t dtx;
|
|
spx_word16_t *in = (spx_word16_t*)vin;
|
|
spx_word16_t e_low=0, e_high=0;
|
|
VARDECL(spx_coef_t *lpc);
|
|
VARDECL(spx_coef_t *interp_lpc);
|
|
VARDECL(spx_coef_t *bw_lpc1);
|
|
VARDECL(spx_coef_t *bw_lpc2);
|
|
VARDECL(spx_lsp_t *lsp);
|
|
VARDECL(spx_lsp_t *qlsp);
|
|
VARDECL(spx_lsp_t *interp_lsp);
|
|
VARDECL(spx_lsp_t *interp_qlsp);
|
|
|
|
st = (SBEncState*)state;
|
|
stack=st->stack;
|
|
mode = (const SpeexSBMode*)(st->mode->mode);
|
|
low = in;
|
|
high = in+st->frame_size;
|
|
|
|
/* High-band buffering / sync with low band */
|
|
/* Compute the two sub-bands by filtering with QMF h0*/
|
|
qmf_decomp(in, h0, low, high, st->full_frame_size, QMF_ORDER, st->h0_mem, stack);
|
|
|
|
#ifndef DISABLE_VBR
|
|
if (st->vbr_enabled || st->vad_enabled)
|
|
{
|
|
/* Need to compute things here before the signal is trashed by the encoder */
|
|
/*FIXME: Are the two signals (low, high) in sync? */
|
|
e_low = compute_rms16(low, st->frame_size);
|
|
e_high = compute_rms16(high, st->frame_size);
|
|
}
|
|
#endif /* #ifndef DISABLE_VBR */
|
|
|
|
ALLOC(low_innov_rms, st->nbSubframes, spx_word16_t);
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_INNOVATION_SAVE, low_innov_rms);
|
|
/* Encode the narrowband part*/
|
|
speex_encode_native(st->st_low, low, bits);
|
|
|
|
high = high - (st->windowSize-st->frame_size);
|
|
SPEEX_COPY(high, st->high, st->windowSize-st->frame_size);
|
|
SPEEX_COPY(st->high, &high[st->frame_size], st->windowSize-st->frame_size);
|
|
|
|
|
|
ALLOC(low_pi_gain, st->nbSubframes, spx_word32_t);
|
|
ALLOC(low_exc_rms, st->nbSubframes, spx_word16_t);
|
|
speex_encoder_ctl(st->st_low, SPEEX_GET_PI_GAIN, low_pi_gain);
|
|
speex_encoder_ctl(st->st_low, SPEEX_GET_EXC, low_exc_rms);
|
|
|
|
speex_encoder_ctl(st->st_low, SPEEX_GET_LOW_MODE, &dtx);
|
|
|
|
if (dtx==0)
|
|
dtx=1;
|
|
else
|
|
dtx=0;
|
|
|
|
ALLOC(lpc, st->lpcSize, spx_coef_t);
|
|
ALLOC(interp_lpc, st->lpcSize, spx_coef_t);
|
|
ALLOC(bw_lpc1, st->lpcSize, spx_coef_t);
|
|
ALLOC(bw_lpc2, st->lpcSize, spx_coef_t);
|
|
|
|
ALLOC(lsp, st->lpcSize, spx_lsp_t);
|
|
ALLOC(qlsp, st->lpcSize, spx_lsp_t);
|
|
ALLOC(interp_lsp, st->lpcSize, spx_lsp_t);
|
|
ALLOC(interp_qlsp, st->lpcSize, spx_lsp_t);
|
|
|
|
{
|
|
VARDECL(spx_word16_t *autocorr);
|
|
VARDECL(spx_word16_t *w_sig);
|
|
ALLOC(autocorr, st->lpcSize+1, spx_word16_t);
|
|
ALLOC(w_sig, st->windowSize, spx_word16_t);
|
|
/* Window for analysis */
|
|
/* FIXME: This is a kludge */
|
|
if (st->subframeSize==80)
|
|
{
|
|
for (i=0;i<st->windowSize;i++)
|
|
w_sig[i] = EXTRACT16(SHR32(MULT16_16(high[i],st->window[i>>1]),SIG_SHIFT));
|
|
} else {
|
|
for (i=0;i<st->windowSize;i++)
|
|
w_sig[i] = EXTRACT16(SHR32(MULT16_16(high[i],st->window[i]),SIG_SHIFT));
|
|
}
|
|
/* Compute auto-correlation */
|
|
_spx_autocorr(w_sig, autocorr, st->lpcSize+1, st->windowSize);
|
|
autocorr[0] = ADD16(autocorr[0],MULT16_16_Q15(autocorr[0],st->lpc_floor)); /* Noise floor in auto-correlation domain */
|
|
|
|
/* Lag windowing: equivalent to filtering in the power-spectrum domain */
|
|
for (i=0;i<st->lpcSize+1;i++)
|
|
autocorr[i] = MULT16_16_Q14(autocorr[i],st->lagWindow[i]);
|
|
|
|
/* Levinson-Durbin */
|
|
_spx_lpc(lpc, autocorr, st->lpcSize);
|
|
}
|
|
|
|
/* LPC to LSPs (x-domain) transform */
|
|
roots=lpc_to_lsp (lpc, st->lpcSize, lsp, 10, LSP_DELTA1, stack);
|
|
if (roots!=st->lpcSize)
|
|
{
|
|
roots = lpc_to_lsp (lpc, st->lpcSize, lsp, 10, LSP_DELTA2, stack);
|
|
if (roots!=st->lpcSize) {
|
|
/*If we can't find all LSP's, do some damage control and use a flat filter*/
|
|
for (i=0;i<st->lpcSize;i++)
|
|
{
|
|
lsp[i]=st->old_lsp[i];
|
|
}
|
|
}
|
|
}
|
|
|
|
#ifndef DISABLE_VBR
|
|
/* VBR code */
|
|
if ((st->vbr_enabled || st->vad_enabled) && !dtx)
|
|
{
|
|
float ratio;
|
|
if (st->abr_enabled)
|
|
{
|
|
float qual_change=0;
|
|
if (st->abr_drift2 * st->abr_drift > 0)
|
|
{
|
|
/* Only adapt if long-term and short-term drift are the same sign */
|
|
qual_change = -.00001*st->abr_drift/(1+st->abr_count);
|
|
if (qual_change>.1)
|
|
qual_change=.1;
|
|
if (qual_change<-.1)
|
|
qual_change=-.1;
|
|
}
|
|
st->vbr_quality += qual_change;
|
|
if (st->vbr_quality>10)
|
|
st->vbr_quality=10;
|
|
if (st->vbr_quality<0)
|
|
st->vbr_quality=0;
|
|
}
|
|
|
|
|
|
ratio = 2*log((1.f+e_high)/(1.f+e_low));
|
|
|
|
speex_encoder_ctl(st->st_low, SPEEX_GET_RELATIVE_QUALITY, &st->relative_quality);
|
|
if (ratio<-4)
|
|
ratio=-4;
|
|
if (ratio>2)
|
|
ratio=2;
|
|
/*if (ratio>-2)*/
|
|
if (st->vbr_enabled)
|
|
{
|
|
spx_int32_t modeid;
|
|
modeid = mode->nb_modes-1;
|
|
st->relative_quality+=1.0*(ratio+2);
|
|
if (st->relative_quality<-1)
|
|
st->relative_quality=-1;
|
|
while (modeid)
|
|
{
|
|
int v1;
|
|
float thresh;
|
|
v1=(int)floor(st->vbr_quality);
|
|
if (v1==10)
|
|
thresh = mode->vbr_thresh[modeid][v1];
|
|
else
|
|
thresh = (st->vbr_quality-v1) * mode->vbr_thresh[modeid][v1+1] +
|
|
(1+v1-st->vbr_quality) * mode->vbr_thresh[modeid][v1];
|
|
if (st->relative_quality >= thresh && st->sampling_rate*st->submodes[modeid]->bits_per_frame/st->full_frame_size <= st->vbr_max_high)
|
|
break;
|
|
modeid--;
|
|
}
|
|
speex_encoder_ctl(state, SPEEX_SET_HIGH_MODE, &modeid);
|
|
if (st->abr_enabled)
|
|
{
|
|
spx_int32_t bitrate;
|
|
speex_encoder_ctl(state, SPEEX_GET_BITRATE, &bitrate);
|
|
st->abr_drift+=(bitrate-st->abr_enabled);
|
|
st->abr_drift2 = .95*st->abr_drift2 + .05*(bitrate-st->abr_enabled);
|
|
st->abr_count += 1.0;
|
|
}
|
|
|
|
} else {
|
|
/* VAD only */
|
|
int modeid;
|
|
if (st->relative_quality<2.0)
|
|
modeid=1;
|
|
else
|
|
modeid=st->submodeSelect;
|
|
/*speex_encoder_ctl(state, SPEEX_SET_MODE, &mode);*/
|
|
st->submodeID=modeid;
|
|
|
|
}
|
|
/*fprintf (stderr, "%f %f\n", ratio, low_qual);*/
|
|
}
|
|
#endif /* #ifndef DISABLE_VBR */
|
|
|
|
if (st->encode_submode)
|
|
{
|
|
speex_bits_pack(bits, 1, 1);
|
|
if (dtx)
|
|
speex_bits_pack(bits, 0, SB_SUBMODE_BITS);
|
|
else
|
|
speex_bits_pack(bits, st->submodeID, SB_SUBMODE_BITS);
|
|
}
|
|
|
|
/* If null mode (no transmission), just set a couple things to zero*/
|
|
if (dtx || st->submodes[st->submodeID] == NULL)
|
|
{
|
|
for (i=0;i<st->frame_size;i++)
|
|
high[i]=VERY_SMALL;
|
|
|
|
for (i=0;i<st->lpcSize;i++)
|
|
st->mem_sw[i]=0;
|
|
st->first=1;
|
|
|
|
/* Final signal synthesis from excitation */
|
|
iir_mem16(high, st->interp_qlpc, high, st->frame_size, st->lpcSize, st->mem_sp, stack);
|
|
|
|
if (dtx)
|
|
return 0;
|
|
else
|
|
return 1;
|
|
}
|
|
|
|
|
|
/* LSP quantization */
|
|
SUBMODE(lsp_quant)(lsp, qlsp, st->lpcSize, bits);
|
|
|
|
if (st->first)
|
|
{
|
|
for (i=0;i<st->lpcSize;i++)
|
|
st->old_lsp[i] = lsp[i];
|
|
for (i=0;i<st->lpcSize;i++)
|
|
st->old_qlsp[i] = qlsp[i];
|
|
}
|
|
|
|
ALLOC(mem, st->lpcSize, spx_mem_t);
|
|
ALLOC(syn_resp, st->subframeSize, spx_word16_t);
|
|
ALLOC(innov, st->subframeSize, spx_sig_t);
|
|
ALLOC(target, st->subframeSize, spx_word16_t);
|
|
|
|
for (sub=0;sub<st->nbSubframes;sub++)
|
|
{
|
|
VARDECL(spx_word16_t *exc);
|
|
VARDECL(spx_word16_t *res);
|
|
VARDECL(spx_word16_t *sw);
|
|
spx_word16_t *sp;
|
|
spx_word16_t filter_ratio; /*Q7*/
|
|
int offset;
|
|
spx_word32_t rl, rh; /*Q13*/
|
|
spx_word16_t eh=0;
|
|
|
|
offset = st->subframeSize*sub;
|
|
sp=high+offset;
|
|
ALLOC(exc, st->subframeSize, spx_word16_t);
|
|
ALLOC(res, st->subframeSize, spx_word16_t);
|
|
ALLOC(sw, st->subframeSize, spx_word16_t);
|
|
|
|
/* LSP interpolation (quantized and unquantized) */
|
|
lsp_interpolate(st->old_lsp, lsp, interp_lsp, st->lpcSize, sub, st->nbSubframes);
|
|
lsp_interpolate(st->old_qlsp, qlsp, interp_qlsp, st->lpcSize, sub, st->nbSubframes);
|
|
|
|
lsp_enforce_margin(interp_lsp, st->lpcSize, LSP_MARGIN);
|
|
lsp_enforce_margin(interp_qlsp, st->lpcSize, LSP_MARGIN);
|
|
|
|
lsp_to_lpc(interp_lsp, interp_lpc, st->lpcSize,stack);
|
|
lsp_to_lpc(interp_qlsp, st->interp_qlpc, st->lpcSize, stack);
|
|
|
|
bw_lpc(st->gamma1, interp_lpc, bw_lpc1, st->lpcSize);
|
|
bw_lpc(st->gamma2, interp_lpc, bw_lpc2, st->lpcSize);
|
|
|
|
/* Compute mid-band (4000 Hz for wideband) response of low-band and high-band
|
|
filters */
|
|
st->pi_gain[sub]=LPC_SCALING;
|
|
rh = LPC_SCALING;
|
|
for (i=0;i<st->lpcSize;i+=2)
|
|
{
|
|
rh += st->interp_qlpc[i+1] - st->interp_qlpc[i];
|
|
st->pi_gain[sub] += st->interp_qlpc[i] + st->interp_qlpc[i+1];
|
|
}
|
|
|
|
rl = low_pi_gain[sub];
|
|
#ifdef FIXED_POINT
|
|
filter_ratio=EXTRACT16(SATURATE(PDIV32(SHL32(ADD32(rl,82),7),ADD32(82,rh)),32767));
|
|
#else
|
|
filter_ratio=(rl+.01)/(rh+.01);
|
|
#endif
|
|
|
|
/* Compute "real excitation" */
|
|
fir_mem16(sp, st->interp_qlpc, exc, st->subframeSize, st->lpcSize, st->mem_sp2, stack);
|
|
/* Compute energy of low-band and high-band excitation */
|
|
|
|
eh = compute_rms16(exc, st->subframeSize);
|
|
|
|
if (!SUBMODE(innovation_quant)) {/* 1 for spectral folding excitation, 0 for stochastic */
|
|
spx_word32_t g; /*Q7*/
|
|
spx_word16_t el; /*Q0*/
|
|
el = low_innov_rms[sub];
|
|
|
|
/* Gain to use if we want to use the low-band excitation for high-band */
|
|
g=PDIV32(MULT16_16(filter_ratio,eh),EXTEND32(ADD16(1,el)));
|
|
|
|
#if 0
|
|
{
|
|
char *tmp_stack=stack;
|
|
float *tmp_sig;
|
|
float g2;
|
|
ALLOC(tmp_sig, st->subframeSize, spx_sig_t);
|
|
for (i=0;i<st->lpcSize;i++)
|
|
mem[i]=st->mem_sp[i];
|
|
iir_mem2(st->low_innov+offset, st->interp_qlpc, tmp_sig, st->subframeSize, st->lpcSize, mem);
|
|
g2 = compute_rms(sp, st->subframeSize)/(.01+compute_rms(tmp_sig, st->subframeSize));
|
|
/*fprintf (stderr, "gains: %f %f\n", g, g2);*/
|
|
g = g2;
|
|
stack = tmp_stack;
|
|
}
|
|
#endif
|
|
|
|
/*print_vec(&g, 1, "gain factor");*/
|
|
/* Gain quantization */
|
|
{
|
|
int quant = scal_quant(g, fold_quant_bound, 32);
|
|
/*speex_warning_int("tata", quant);*/
|
|
if (quant<0)
|
|
quant=0;
|
|
if (quant>31)
|
|
quant=31;
|
|
speex_bits_pack(bits, quant, 5);
|
|
}
|
|
if (st->innov_rms_save)
|
|
{
|
|
st->innov_rms_save[sub] = eh;
|
|
}
|
|
st->exc_rms[sub] = eh;
|
|
} else {
|
|
spx_word16_t gc; /*Q7*/
|
|
spx_word32_t scale; /*Q14*/
|
|
spx_word16_t el; /*Q0*/
|
|
el = low_exc_rms[sub]; /*Q0*/
|
|
|
|
gc = PDIV32_16(MULT16_16(filter_ratio,1+eh),1+el);
|
|
|
|
/* This is a kludge that cleans up a historical bug */
|
|
if (st->subframeSize==80)
|
|
gc = MULT16_16_P15(QCONST16(0.70711f,15),gc);
|
|
/*printf ("%f %f %f %f\n", el, eh, filter_ratio, gc);*/
|
|
{
|
|
int qgc = scal_quant(gc, gc_quant_bound, 16);
|
|
speex_bits_pack(bits, qgc, 4);
|
|
gc = MULT16_16_Q15(QCONST16(0.87360,15),gc_quant_bound[qgc]);
|
|
}
|
|
if (st->subframeSize==80)
|
|
gc = MULT16_16_P14(QCONST16(1.4142f,14), gc);
|
|
|
|
scale = SHL32(MULT16_16(PDIV32_16(SHL32(EXTEND32(gc),SIG_SHIFT-6),filter_ratio),(1+el)),6);
|
|
|
|
compute_impulse_response(st->interp_qlpc, bw_lpc1, bw_lpc2, syn_resp, st->subframeSize, st->lpcSize, stack);
|
|
|
|
|
|
/* Reset excitation */
|
|
for (i=0;i<st->subframeSize;i++)
|
|
res[i]=VERY_SMALL;
|
|
|
|
/* Compute zero response (ringing) of A(z/g1) / ( A(z/g2) * Aq(z) ) */
|
|
for (i=0;i<st->lpcSize;i++)
|
|
mem[i]=st->mem_sp[i];
|
|
iir_mem16(res, st->interp_qlpc, res, st->subframeSize, st->lpcSize, mem, stack);
|
|
|
|
for (i=0;i<st->lpcSize;i++)
|
|
mem[i]=st->mem_sw[i];
|
|
filter_mem16(res, bw_lpc1, bw_lpc2, res, st->subframeSize, st->lpcSize, mem, stack);
|
|
|
|
/* Compute weighted signal */
|
|
for (i=0;i<st->lpcSize;i++)
|
|
mem[i]=st->mem_sw[i];
|
|
filter_mem16(sp, bw_lpc1, bw_lpc2, sw, st->subframeSize, st->lpcSize, mem, stack);
|
|
|
|
/* Compute target signal */
|
|
for (i=0;i<st->subframeSize;i++)
|
|
target[i]=SUB16(sw[i],res[i]);
|
|
|
|
signal_div(target, target, scale, st->subframeSize);
|
|
|
|
/* Reset excitation */
|
|
SPEEX_MEMSET(innov, 0, st->subframeSize);
|
|
|
|
/*print_vec(target, st->subframeSize, "\ntarget");*/
|
|
SUBMODE(innovation_quant)(target, st->interp_qlpc, bw_lpc1, bw_lpc2,
|
|
SUBMODE(innovation_params), st->lpcSize, st->subframeSize,
|
|
innov, syn_resp, bits, stack, st->complexity, SUBMODE(double_codebook));
|
|
/*print_vec(target, st->subframeSize, "after");*/
|
|
|
|
signal_mul(innov, innov, scale, st->subframeSize);
|
|
|
|
if (SUBMODE(double_codebook)) {
|
|
char *tmp_stack=stack;
|
|
VARDECL(spx_sig_t *innov2);
|
|
ALLOC(innov2, st->subframeSize, spx_sig_t);
|
|
SPEEX_MEMSET(innov2, 0, st->subframeSize);
|
|
for (i=0;i<st->subframeSize;i++)
|
|
target[i]=MULT16_16_P13(QCONST16(2.5f,13), target[i]);
|
|
|
|
SUBMODE(innovation_quant)(target, st->interp_qlpc, bw_lpc1, bw_lpc2,
|
|
SUBMODE(innovation_params), st->lpcSize, st->subframeSize,
|
|
innov2, syn_resp, bits, stack, st->complexity, 0);
|
|
signal_mul(innov2, innov2, MULT16_32_P15(QCONST16(0.4f,15),scale), st->subframeSize);
|
|
|
|
for (i=0;i<st->subframeSize;i++)
|
|
innov[i] = ADD32(innov[i],innov2[i]);
|
|
stack = tmp_stack;
|
|
}
|
|
for (i=0;i<st->subframeSize;i++)
|
|
exc[i] = PSHR32(innov[i],SIG_SHIFT);
|
|
|
|
if (st->innov_rms_save)
|
|
{
|
|
st->innov_rms_save[sub] = MULT16_16_Q15(QCONST16(.70711f, 15), compute_rms(innov, st->subframeSize));
|
|
}
|
|
st->exc_rms[sub] = compute_rms16(exc, st->subframeSize);
|
|
|
|
|
|
}
|
|
|
|
|
|
/*Keep the previous memory*/
|
|
for (i=0;i<st->lpcSize;i++)
|
|
mem[i]=st->mem_sp[i];
|
|
/* Final signal synthesis from excitation */
|
|
iir_mem16(exc, st->interp_qlpc, sp, st->subframeSize, st->lpcSize, st->mem_sp, stack);
|
|
|
|
/* Compute weighted signal again, from synthesized speech (not sure it's the right thing) */
|
|
filter_mem16(sp, bw_lpc1, bw_lpc2, sw, st->subframeSize, st->lpcSize, st->mem_sw, stack);
|
|
}
|
|
|
|
for (i=0;i<st->lpcSize;i++)
|
|
st->old_lsp[i] = lsp[i];
|
|
for (i=0;i<st->lpcSize;i++)
|
|
st->old_qlsp[i] = qlsp[i];
|
|
|
|
st->first=0;
|
|
|
|
return 1;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void *sb_decoder_init(const SpeexMode *m)
|
|
{
|
|
spx_int32_t tmp;
|
|
SBDecState *st;
|
|
const SpeexSBMode *mode;
|
|
st = (SBDecState*)speex_alloc(sizeof(SBDecState));
|
|
if (!st)
|
|
return NULL;
|
|
st->mode = m;
|
|
mode=(const SpeexSBMode*)m->mode;
|
|
st->encode_submode = 1;
|
|
|
|
st->st_low = speex_decoder_init(mode->nb_mode);
|
|
#if defined(VAR_ARRAYS) || defined (USE_ALLOCA)
|
|
st->stack = NULL;
|
|
#else
|
|
/*st->stack = (char*)speex_alloc_scratch(SB_DEC_STACK);*/
|
|
speex_decoder_ctl(st->st_low, SPEEX_GET_STACK, &st->stack);
|
|
#endif
|
|
|
|
st->full_frame_size = 2*mode->frameSize;
|
|
st->frame_size = mode->frameSize;
|
|
st->subframeSize = mode->subframeSize;
|
|
st->nbSubframes = mode->frameSize/mode->subframeSize;
|
|
st->lpcSize=mode->lpcSize;
|
|
speex_decoder_ctl(st->st_low, SPEEX_GET_SAMPLING_RATE, &st->sampling_rate);
|
|
st->sampling_rate*=2;
|
|
tmp=1;
|
|
speex_decoder_ctl(st->st_low, SPEEX_SET_WIDEBAND, &tmp);
|
|
|
|
st->submodes=mode->submodes;
|
|
st->submodeID=mode->defaultSubmode;
|
|
|
|
st->first=1;
|
|
|
|
st->g0_mem = (spx_word16_t*)speex_alloc((QMF_ORDER)*sizeof(spx_word16_t));
|
|
st->g1_mem = (spx_word16_t*)speex_alloc((QMF_ORDER)*sizeof(spx_word16_t));
|
|
|
|
st->excBuf = (spx_word16_t*)speex_alloc((st->subframeSize)*sizeof(spx_word16_t));
|
|
|
|
st->old_qlsp = (spx_lsp_t*)speex_alloc((st->lpcSize)*sizeof(spx_lsp_t));
|
|
st->interp_qlpc = (spx_coef_t*)speex_alloc(st->lpcSize*sizeof(spx_coef_t));
|
|
|
|
st->pi_gain = (spx_word32_t*)speex_alloc((st->nbSubframes)*sizeof(spx_word32_t));
|
|
st->exc_rms = (spx_word16_t*)speex_alloc((st->nbSubframes)*sizeof(spx_word16_t));
|
|
st->mem_sp = (spx_mem_t*)speex_alloc((2*st->lpcSize)*sizeof(spx_mem_t));
|
|
|
|
st->innov_save = NULL;
|
|
|
|
|
|
st->lpc_enh_enabled=0;
|
|
st->seed = 1000;
|
|
|
|
#ifdef ENABLE_VALGRIND
|
|
VALGRIND_MAKE_READABLE(st, (st->stack-(char*)st));
|
|
#endif
|
|
return st;
|
|
}
|
|
|
|
void sb_decoder_destroy(void *state)
|
|
{
|
|
SBDecState *st;
|
|
st = (SBDecState*)state;
|
|
speex_decoder_destroy(st->st_low);
|
|
#if !(defined(VAR_ARRAYS) || defined (USE_ALLOCA))
|
|
/*speex_free_scratch(st->stack);*/
|
|
#endif
|
|
|
|
speex_free(st->g0_mem);
|
|
speex_free(st->g1_mem);
|
|
speex_free(st->excBuf);
|
|
speex_free(st->old_qlsp);
|
|
speex_free(st->interp_qlpc);
|
|
speex_free(st->pi_gain);
|
|
speex_free(st->exc_rms);
|
|
speex_free(st->mem_sp);
|
|
|
|
speex_free(state);
|
|
}
|
|
|
|
static void sb_decode_lost(SBDecState *st, spx_word16_t *out, int dtx, char *stack)
|
|
{
|
|
int i;
|
|
int saved_modeid=0;
|
|
|
|
if (dtx)
|
|
{
|
|
saved_modeid=st->submodeID;
|
|
st->submodeID=1;
|
|
} else {
|
|
bw_lpc(QCONST16(0.99f,15), st->interp_qlpc, st->interp_qlpc, st->lpcSize);
|
|
}
|
|
|
|
st->first=1;
|
|
|
|
|
|
/* Final signal synthesis from excitation */
|
|
if (!dtx)
|
|
{
|
|
st->last_ener = MULT16_16_Q15(QCONST16(.9f,15),st->last_ener);
|
|
}
|
|
for (i=0;i<st->frame_size;i++)
|
|
out[i+st->frame_size] = speex_rand(st->last_ener, &st->seed);
|
|
|
|
iir_mem16(out+st->frame_size, st->interp_qlpc, out+st->frame_size, st->frame_size, st->lpcSize,
|
|
st->mem_sp, stack);
|
|
|
|
|
|
/* Reconstruct the original */
|
|
qmf_synth(out, out+st->frame_size, h0, out, st->full_frame_size, QMF_ORDER, st->g0_mem, st->g1_mem, stack);
|
|
if (dtx)
|
|
{
|
|
st->submodeID=saved_modeid;
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
int sb_decode(void *state, SpeexBits *bits, void *vout)
|
|
{
|
|
int i, sub;
|
|
SBDecState *st;
|
|
int wideband;
|
|
int ret;
|
|
char *stack;
|
|
VARDECL(spx_word32_t *low_pi_gain);
|
|
VARDECL(spx_word16_t *low_exc_rms);
|
|
VARDECL(spx_coef_t *ak);
|
|
VARDECL(spx_lsp_t *qlsp);
|
|
VARDECL(spx_lsp_t *interp_qlsp);
|
|
spx_int32_t dtx;
|
|
const SpeexSBMode *mode;
|
|
spx_word16_t *out = (spx_word16_t*)vout;
|
|
spx_word16_t *low_innov_alias;
|
|
spx_word32_t exc_ener_sum = 0;
|
|
|
|
st = (SBDecState*)state;
|
|
stack=st->stack;
|
|
mode = (const SpeexSBMode*)(st->mode->mode);
|
|
|
|
low_innov_alias = out+st->frame_size;
|
|
speex_decoder_ctl(st->st_low, SPEEX_SET_INNOVATION_SAVE, low_innov_alias);
|
|
/* Decode the low-band */
|
|
ret = speex_decode_native(st->st_low, bits, out);
|
|
|
|
speex_decoder_ctl(st->st_low, SPEEX_GET_DTX_STATUS, &dtx);
|
|
|
|
/* If error decoding the narrowband part, propagate error */
|
|
if (ret!=0)
|
|
{
|
|
return ret;
|
|
}
|
|
|
|
if (!bits)
|
|
{
|
|
sb_decode_lost(st, out, dtx, stack);
|
|
return 0;
|
|
}
|
|
|
|
if (st->encode_submode)
|
|
{
|
|
|
|
/*Check "wideband bit"*/
|
|
if (speex_bits_remaining(bits)>0)
|
|
wideband = speex_bits_peek(bits);
|
|
else
|
|
wideband = 0;
|
|
if (wideband)
|
|
{
|
|
/*Regular wideband frame, read the submode*/
|
|
wideband = speex_bits_unpack_unsigned(bits, 1);
|
|
st->submodeID = speex_bits_unpack_unsigned(bits, SB_SUBMODE_BITS);
|
|
} else
|
|
{
|
|
/*Was a narrowband frame, set "null submode"*/
|
|
st->submodeID = 0;
|
|
}
|
|
if (st->submodeID != 0 && st->submodes[st->submodeID] == NULL)
|
|
{
|
|
speex_notify("Invalid mode encountered. The stream is corrupted.");
|
|
return -2;
|
|
}
|
|
}
|
|
|
|
/* If null mode (no transmission), just set a couple things to zero*/
|
|
if (st->submodes[st->submodeID] == NULL)
|
|
{
|
|
if (dtx)
|
|
{
|
|
sb_decode_lost(st, out, 1, stack);
|
|
return 0;
|
|
}
|
|
|
|
for (i=0;i<st->frame_size;i++)
|
|
out[st->frame_size+i]=VERY_SMALL;
|
|
|
|
st->first=1;
|
|
|
|
/* Final signal synthesis from excitation */
|
|
iir_mem16(out+st->frame_size, st->interp_qlpc, out+st->frame_size, st->frame_size, st->lpcSize, st->mem_sp, stack);
|
|
|
|
qmf_synth(out, out+st->frame_size, h0, out, st->full_frame_size, QMF_ORDER, st->g0_mem, st->g1_mem, stack);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
ALLOC(low_pi_gain, st->nbSubframes, spx_word32_t);
|
|
ALLOC(low_exc_rms, st->nbSubframes, spx_word16_t);
|
|
speex_decoder_ctl(st->st_low, SPEEX_GET_PI_GAIN, low_pi_gain);
|
|
speex_decoder_ctl(st->st_low, SPEEX_GET_EXC, low_exc_rms);
|
|
|
|
ALLOC(qlsp, st->lpcSize, spx_lsp_t);
|
|
ALLOC(interp_qlsp, st->lpcSize, spx_lsp_t);
|
|
SUBMODE(lsp_unquant)(qlsp, st->lpcSize, bits);
|
|
|
|
if (st->first)
|
|
{
|
|
for (i=0;i<st->lpcSize;i++)
|
|
st->old_qlsp[i] = qlsp[i];
|
|
}
|
|
|
|
ALLOC(ak, st->lpcSize, spx_coef_t);
|
|
|
|
for (sub=0;sub<st->nbSubframes;sub++)
|
|
{
|
|
VARDECL(spx_word32_t *exc);
|
|
spx_word16_t *innov_save=NULL;
|
|
spx_word16_t *sp;
|
|
spx_word16_t filter_ratio;
|
|
spx_word16_t el=0;
|
|
int offset;
|
|
spx_word32_t rl=0,rh=0;
|
|
|
|
offset = st->subframeSize*sub;
|
|
sp=out+st->frame_size+offset;
|
|
ALLOC(exc, st->subframeSize, spx_word32_t);
|
|
/* Pointer for saving innovation */
|
|
if (st->innov_save)
|
|
{
|
|
innov_save = st->innov_save+2*offset;
|
|
SPEEX_MEMSET(innov_save, 0, 2*st->subframeSize);
|
|
}
|
|
|
|
/* LSP interpolation */
|
|
lsp_interpolate(st->old_qlsp, qlsp, interp_qlsp, st->lpcSize, sub, st->nbSubframes);
|
|
|
|
lsp_enforce_margin(interp_qlsp, st->lpcSize, LSP_MARGIN);
|
|
|
|
/* LSP to LPC */
|
|
lsp_to_lpc(interp_qlsp, ak, st->lpcSize, stack);
|
|
|
|
/* Calculate reponse ratio between the low and high filter in the middle
|
|
of the band (4000 Hz) */
|
|
|
|
st->pi_gain[sub]=LPC_SCALING;
|
|
rh = LPC_SCALING;
|
|
for (i=0;i<st->lpcSize;i+=2)
|
|
{
|
|
rh += ak[i+1] - ak[i];
|
|
st->pi_gain[sub] += ak[i] + ak[i+1];
|
|
}
|
|
|
|
rl = low_pi_gain[sub];
|
|
#ifdef FIXED_POINT
|
|
filter_ratio=EXTRACT16(SATURATE(PDIV32(SHL32(ADD32(rl,82),7),ADD32(82,rh)),32767));
|
|
#else
|
|
filter_ratio=(rl+.01)/(rh+.01);
|
|
#endif
|
|
|
|
SPEEX_MEMSET(exc, 0, st->subframeSize);
|
|
if (!SUBMODE(innovation_unquant))
|
|
{
|
|
spx_word32_t g;
|
|
int quant;
|
|
|
|
quant = speex_bits_unpack_unsigned(bits, 5);
|
|
g= spx_exp(MULT16_16(QCONST16(.125f,11),(quant-10)));
|
|
|
|
g = PDIV32(g, filter_ratio);
|
|
|
|
for (i=0;i<st->subframeSize;i+=2)
|
|
{
|
|
exc[i]=SHL32(MULT16_32_P15(MULT16_16_Q15(mode->folding_gain,low_innov_alias[offset+i]),SHL32(g,6)),SIG_SHIFT);
|
|
exc[i+1]=NEG32(SHL32(MULT16_32_P15(MULT16_16_Q15(mode->folding_gain,low_innov_alias[offset+i+1]),SHL32(g,6)),SIG_SHIFT));
|
|
}
|
|
|
|
} else {
|
|
spx_word16_t gc;
|
|
spx_word32_t scale;
|
|
int qgc = speex_bits_unpack_unsigned(bits, 4);
|
|
|
|
el = low_exc_rms[sub];
|
|
gc = MULT16_16_Q15(QCONST16(0.87360,15),gc_quant_bound[qgc]);
|
|
|
|
if (st->subframeSize==80)
|
|
gc = MULT16_16_P14(QCONST16(1.4142f,14),gc);
|
|
|
|
scale = SHL32(PDIV32(SHL32(MULT16_16(gc, el),3), filter_ratio),SIG_SHIFT-3);
|
|
SUBMODE(innovation_unquant)(exc, SUBMODE(innovation_params), st->subframeSize,
|
|
bits, stack, &st->seed);
|
|
|
|
signal_mul(exc,exc,scale,st->subframeSize);
|
|
|
|
if (SUBMODE(double_codebook)) {
|
|
char *tmp_stack=stack;
|
|
VARDECL(spx_sig_t *innov2);
|
|
ALLOC(innov2, st->subframeSize, spx_sig_t);
|
|
SPEEX_MEMSET(innov2, 0, st->subframeSize);
|
|
SUBMODE(innovation_unquant)(innov2, SUBMODE(innovation_params), st->subframeSize,
|
|
bits, stack, &st->seed);
|
|
signal_mul(innov2, innov2, MULT16_32_P15(QCONST16(0.4f,15),scale), st->subframeSize);
|
|
for (i=0;i<st->subframeSize;i++)
|
|
exc[i] = ADD32(exc[i],innov2[i]);
|
|
stack = tmp_stack;
|
|
}
|
|
|
|
}
|
|
|
|
if (st->innov_save)
|
|
{
|
|
for (i=0;i<st->subframeSize;i++)
|
|
innov_save[2*i]=EXTRACT16(PSHR32(exc[i],SIG_SHIFT));
|
|
}
|
|
|
|
iir_mem16(st->excBuf, st->interp_qlpc, sp, st->subframeSize, st->lpcSize,
|
|
st->mem_sp, stack);
|
|
for (i=0;i<st->subframeSize;i++)
|
|
st->excBuf[i]=EXTRACT16(PSHR32(exc[i],SIG_SHIFT));
|
|
for (i=0;i<st->lpcSize;i++)
|
|
st->interp_qlpc[i] = ak[i];
|
|
st->exc_rms[sub] = compute_rms16(st->excBuf, st->subframeSize);
|
|
exc_ener_sum = ADD32(exc_ener_sum, DIV32(MULT16_16(st->exc_rms[sub],st->exc_rms[sub]), st->nbSubframes));
|
|
}
|
|
st->last_ener = spx_sqrt(exc_ener_sum);
|
|
|
|
qmf_synth(out, out+st->frame_size, h0, out, st->full_frame_size, QMF_ORDER, st->g0_mem, st->g1_mem, stack);
|
|
for (i=0;i<st->lpcSize;i++)
|
|
st->old_qlsp[i] = qlsp[i];
|
|
|
|
st->first=0;
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
int sb_encoder_ctl(void *state, int request, void *ptr)
|
|
{
|
|
SBEncState *st;
|
|
st=(SBEncState*)state;
|
|
switch(request)
|
|
{
|
|
case SPEEX_GET_FRAME_SIZE:
|
|
(*(spx_int32_t*)ptr) = st->full_frame_size;
|
|
break;
|
|
case SPEEX_SET_HIGH_MODE:
|
|
st->submodeSelect = st->submodeID = (*(spx_int32_t*)ptr);
|
|
break;
|
|
case SPEEX_SET_LOW_MODE:
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_LOW_MODE, ptr);
|
|
break;
|
|
case SPEEX_SET_DTX:
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_DTX, ptr);
|
|
break;
|
|
case SPEEX_GET_DTX:
|
|
speex_encoder_ctl(st->st_low, SPEEX_GET_DTX, ptr);
|
|
break;
|
|
case SPEEX_GET_LOW_MODE:
|
|
speex_encoder_ctl(st->st_low, SPEEX_GET_LOW_MODE, ptr);
|
|
break;
|
|
case SPEEX_SET_MODE:
|
|
speex_encoder_ctl(st, SPEEX_SET_QUALITY, ptr);
|
|
break;
|
|
#ifndef DISABLE_VBR
|
|
case SPEEX_SET_VBR:
|
|
st->vbr_enabled = (*(spx_int32_t*)ptr);
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_VBR, ptr);
|
|
break;
|
|
case SPEEX_GET_VBR:
|
|
(*(spx_int32_t*)ptr) = st->vbr_enabled;
|
|
break;
|
|
case SPEEX_SET_VAD:
|
|
st->vad_enabled = (*(spx_int32_t*)ptr);
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_VAD, ptr);
|
|
break;
|
|
case SPEEX_GET_VAD:
|
|
(*(spx_int32_t*)ptr) = st->vad_enabled;
|
|
break;
|
|
#endif /* #ifndef DISABLE_VBR */
|
|
#if !defined(DISABLE_VBR) && !defined(DISABLE_FLOAT_API)
|
|
case SPEEX_SET_VBR_QUALITY:
|
|
{
|
|
spx_int32_t q;
|
|
float qual = (*(float*)ptr)+.6;
|
|
st->vbr_quality = (*(float*)ptr);
|
|
if (qual>10)
|
|
qual=10;
|
|
q=(int)floor(.5+*(float*)ptr);
|
|
if (q>10)
|
|
q=10;
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_VBR_QUALITY, &qual);
|
|
speex_encoder_ctl(state, SPEEX_SET_QUALITY, &q);
|
|
break;
|
|
}
|
|
case SPEEX_GET_VBR_QUALITY:
|
|
(*(float*)ptr) = st->vbr_quality;
|
|
break;
|
|
#endif /* #if !defined(DISABLE_VBR) && !defined(DISABLE_FLOAT_API) */
|
|
#ifndef DISABLE_VBR
|
|
case SPEEX_SET_ABR:
|
|
st->abr_enabled = (*(spx_int32_t*)ptr);
|
|
st->vbr_enabled = st->abr_enabled!=0;
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_VBR, &st->vbr_enabled);
|
|
if (st->vbr_enabled)
|
|
{
|
|
spx_int32_t i=10, rate, target;
|
|
float vbr_qual;
|
|
target = (*(spx_int32_t*)ptr);
|
|
while (i>=0)
|
|
{
|
|
speex_encoder_ctl(st, SPEEX_SET_QUALITY, &i);
|
|
speex_encoder_ctl(st, SPEEX_GET_BITRATE, &rate);
|
|
if (rate <= target)
|
|
break;
|
|
i--;
|
|
}
|
|
vbr_qual=i;
|
|
if (vbr_qual<0)
|
|
vbr_qual=0;
|
|
speex_encoder_ctl(st, SPEEX_SET_VBR_QUALITY, &vbr_qual);
|
|
st->abr_count=0;
|
|
st->abr_drift=0;
|
|
st->abr_drift2=0;
|
|
}
|
|
|
|
break;
|
|
case SPEEX_GET_ABR:
|
|
(*(spx_int32_t*)ptr) = st->abr_enabled;
|
|
break;
|
|
#endif /* #ifndef DISABLE_VBR */
|
|
|
|
case SPEEX_SET_QUALITY:
|
|
{
|
|
spx_int32_t nb_qual;
|
|
int quality = (*(spx_int32_t*)ptr);
|
|
if (quality < 0)
|
|
quality = 0;
|
|
if (quality > 10)
|
|
quality = 10;
|
|
st->submodeSelect = st->submodeID = ((const SpeexSBMode*)(st->mode->mode))->quality_map[quality];
|
|
nb_qual = ((const SpeexSBMode*)(st->mode->mode))->low_quality_map[quality];
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_MODE, &nb_qual);
|
|
}
|
|
break;
|
|
case SPEEX_SET_COMPLEXITY:
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_COMPLEXITY, ptr);
|
|
st->complexity = (*(spx_int32_t*)ptr);
|
|
if (st->complexity<1)
|
|
st->complexity=1;
|
|
break;
|
|
case SPEEX_GET_COMPLEXITY:
|
|
(*(spx_int32_t*)ptr) = st->complexity;
|
|
break;
|
|
case SPEEX_SET_BITRATE:
|
|
{
|
|
spx_int32_t i=10;
|
|
spx_int32_t rate, target;
|
|
target = (*(spx_int32_t*)ptr);
|
|
while (i>=0)
|
|
{
|
|
speex_encoder_ctl(st, SPEEX_SET_QUALITY, &i);
|
|
speex_encoder_ctl(st, SPEEX_GET_BITRATE, &rate);
|
|
if (rate <= target)
|
|
break;
|
|
i--;
|
|
}
|
|
}
|
|
break;
|
|
case SPEEX_GET_BITRATE:
|
|
speex_encoder_ctl(st->st_low, request, ptr);
|
|
/*fprintf (stderr, "before: %d\n", (*(int*)ptr));*/
|
|
if (st->submodes[st->submodeID])
|
|
(*(spx_int32_t*)ptr) += st->sampling_rate*SUBMODE(bits_per_frame)/st->full_frame_size;
|
|
else
|
|
(*(spx_int32_t*)ptr) += st->sampling_rate*(SB_SUBMODE_BITS+1)/st->full_frame_size;
|
|
/*fprintf (stderr, "after: %d\n", (*(int*)ptr));*/
|
|
break;
|
|
case SPEEX_SET_SAMPLING_RATE:
|
|
{
|
|
spx_int32_t tmp=(*(spx_int32_t*)ptr);
|
|
st->sampling_rate = tmp;
|
|
tmp>>=1;
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_SAMPLING_RATE, &tmp);
|
|
}
|
|
break;
|
|
case SPEEX_GET_SAMPLING_RATE:
|
|
(*(spx_int32_t*)ptr)=st->sampling_rate;
|
|
break;
|
|
case SPEEX_RESET_STATE:
|
|
{
|
|
int i;
|
|
st->first = 1;
|
|
for (i=0;i<st->lpcSize;i++)
|
|
st->old_lsp[i]= DIV32(MULT16_16(QCONST16(3.1415927f, LSP_SHIFT), i+1), st->lpcSize+1);
|
|
for (i=0;i<st->lpcSize;i++)
|
|
st->mem_sw[i]=st->mem_sp[i]=st->mem_sp2[i]=0;
|
|
for (i=0;i<QMF_ORDER;i++)
|
|
st->h0_mem[i]=st->h1_mem[i]=0;
|
|
}
|
|
break;
|
|
case SPEEX_SET_SUBMODE_ENCODING:
|
|
st->encode_submode = (*(spx_int32_t*)ptr);
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_SUBMODE_ENCODING, ptr);
|
|
break;
|
|
case SPEEX_GET_SUBMODE_ENCODING:
|
|
(*(spx_int32_t*)ptr) = st->encode_submode;
|
|
break;
|
|
case SPEEX_GET_LOOKAHEAD:
|
|
speex_encoder_ctl(st->st_low, SPEEX_GET_LOOKAHEAD, ptr);
|
|
(*(spx_int32_t*)ptr) = 2*(*(spx_int32_t*)ptr) + QMF_ORDER - 1;
|
|
break;
|
|
case SPEEX_SET_PLC_TUNING:
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_PLC_TUNING, ptr);
|
|
break;
|
|
case SPEEX_GET_PLC_TUNING:
|
|
speex_encoder_ctl(st->st_low, SPEEX_GET_PLC_TUNING, ptr);
|
|
break;
|
|
#ifndef DISABLE_VBR
|
|
case SPEEX_SET_VBR_MAX_BITRATE:
|
|
{
|
|
st->vbr_max = (*(spx_int32_t*)ptr);
|
|
if (SPEEX_SET_VBR_MAX_BITRATE<1)
|
|
{
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_VBR_MAX_BITRATE, &st->vbr_max);
|
|
st->vbr_max_high = 17600;
|
|
} else {
|
|
spx_int32_t low_rate;
|
|
if (st->vbr_max >= 42200)
|
|
{
|
|
st->vbr_max_high = 17600;
|
|
} else if (st->vbr_max >= 27800)
|
|
{
|
|
st->vbr_max_high = 9600;
|
|
} else if (st->vbr_max > 20600)
|
|
{
|
|
st->vbr_max_high = 5600;
|
|
} else {
|
|
st->vbr_max_high = 1800;
|
|
}
|
|
if (st->subframeSize==80)
|
|
st->vbr_max_high = 1800;
|
|
low_rate = st->vbr_max - st->vbr_max_high;
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_VBR_MAX_BITRATE, &low_rate);
|
|
}
|
|
}
|
|
break;
|
|
case SPEEX_GET_VBR_MAX_BITRATE:
|
|
(*(spx_int32_t*)ptr) = st->vbr_max;
|
|
break;
|
|
#endif /* #ifndef DISABLE_VBR */
|
|
case SPEEX_SET_HIGHPASS:
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_HIGHPASS, ptr);
|
|
break;
|
|
case SPEEX_GET_HIGHPASS:
|
|
speex_encoder_ctl(st->st_low, SPEEX_GET_HIGHPASS, ptr);
|
|
break;
|
|
|
|
|
|
/* This is all internal stuff past this point */
|
|
case SPEEX_GET_PI_GAIN:
|
|
{
|
|
int i;
|
|
spx_word32_t *g = (spx_word32_t*)ptr;
|
|
for (i=0;i<st->nbSubframes;i++)
|
|
g[i]=st->pi_gain[i];
|
|
}
|
|
break;
|
|
case SPEEX_GET_EXC:
|
|
{
|
|
int i;
|
|
for (i=0;i<st->nbSubframes;i++)
|
|
((spx_word16_t*)ptr)[i] = st->exc_rms[i];
|
|
}
|
|
break;
|
|
#ifndef DISABLE_VBR
|
|
case SPEEX_GET_RELATIVE_QUALITY:
|
|
(*(float*)ptr)=st->relative_quality;
|
|
break;
|
|
#endif /* #ifndef DISABLE_VBR */
|
|
case SPEEX_SET_INNOVATION_SAVE:
|
|
st->innov_rms_save = (spx_word16_t*)ptr;
|
|
break;
|
|
case SPEEX_SET_WIDEBAND:
|
|
speex_encoder_ctl(st->st_low, SPEEX_SET_WIDEBAND, ptr);
|
|
break;
|
|
case SPEEX_GET_STACK:
|
|
*((char**)ptr) = st->stack;
|
|
break;
|
|
default:
|
|
speex_warning_int("Unknown nb_ctl request: ", request);
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int sb_decoder_ctl(void *state, int request, void *ptr)
|
|
{
|
|
SBDecState *st;
|
|
st=(SBDecState*)state;
|
|
switch(request)
|
|
{
|
|
case SPEEX_SET_HIGH_MODE:
|
|
st->submodeID = (*(spx_int32_t*)ptr);
|
|
break;
|
|
case SPEEX_SET_LOW_MODE:
|
|
speex_decoder_ctl(st->st_low, SPEEX_SET_LOW_MODE, ptr);
|
|
break;
|
|
case SPEEX_GET_LOW_MODE:
|
|
speex_decoder_ctl(st->st_low, SPEEX_GET_LOW_MODE, ptr);
|
|
break;
|
|
case SPEEX_GET_FRAME_SIZE:
|
|
(*(spx_int32_t*)ptr) = st->full_frame_size;
|
|
break;
|
|
case SPEEX_SET_ENH:
|
|
speex_decoder_ctl(st->st_low, request, ptr);
|
|
st->lpc_enh_enabled = *((spx_int32_t*)ptr);
|
|
break;
|
|
case SPEEX_GET_ENH:
|
|
*((spx_int32_t*)ptr) = st->lpc_enh_enabled;
|
|
break;
|
|
case SPEEX_SET_MODE:
|
|
case SPEEX_SET_QUALITY:
|
|
{
|
|
spx_int32_t nb_qual;
|
|
int quality = (*(spx_int32_t*)ptr);
|
|
if (quality < 0)
|
|
quality = 0;
|
|
if (quality > 10)
|
|
quality = 10;
|
|
st->submodeID = ((const SpeexSBMode*)(st->mode->mode))->quality_map[quality];
|
|
nb_qual = ((const SpeexSBMode*)(st->mode->mode))->low_quality_map[quality];
|
|
speex_decoder_ctl(st->st_low, SPEEX_SET_MODE, &nb_qual);
|
|
}
|
|
break;
|
|
case SPEEX_GET_BITRATE:
|
|
speex_decoder_ctl(st->st_low, request, ptr);
|
|
if (st->submodes[st->submodeID])
|
|
(*(spx_int32_t*)ptr) += st->sampling_rate*SUBMODE(bits_per_frame)/st->full_frame_size;
|
|
else
|
|
(*(spx_int32_t*)ptr) += st->sampling_rate*(SB_SUBMODE_BITS+1)/st->full_frame_size;
|
|
break;
|
|
case SPEEX_SET_SAMPLING_RATE:
|
|
{
|
|
spx_int32_t tmp=(*(spx_int32_t*)ptr);
|
|
st->sampling_rate = tmp;
|
|
tmp>>=1;
|
|
speex_decoder_ctl(st->st_low, SPEEX_SET_SAMPLING_RATE, &tmp);
|
|
}
|
|
break;
|
|
case SPEEX_GET_SAMPLING_RATE:
|
|
(*(spx_int32_t*)ptr)=st->sampling_rate;
|
|
break;
|
|
case SPEEX_SET_HANDLER:
|
|
speex_decoder_ctl(st->st_low, SPEEX_SET_HANDLER, ptr);
|
|
break;
|
|
case SPEEX_SET_USER_HANDLER:
|
|
speex_decoder_ctl(st->st_low, SPEEX_SET_USER_HANDLER, ptr);
|
|
break;
|
|
case SPEEX_RESET_STATE:
|
|
{
|
|
int i;
|
|
for (i=0;i<2*st->lpcSize;i++)
|
|
st->mem_sp[i]=0;
|
|
for (i=0;i<QMF_ORDER;i++)
|
|
st->g0_mem[i]=st->g1_mem[i]=0;
|
|
st->last_ener=0;
|
|
}
|
|
break;
|
|
case SPEEX_SET_SUBMODE_ENCODING:
|
|
st->encode_submode = (*(spx_int32_t*)ptr);
|
|
speex_decoder_ctl(st->st_low, SPEEX_SET_SUBMODE_ENCODING, ptr);
|
|
break;
|
|
case SPEEX_GET_SUBMODE_ENCODING:
|
|
(*(spx_int32_t*)ptr) = st->encode_submode;
|
|
break;
|
|
case SPEEX_GET_LOOKAHEAD:
|
|
speex_decoder_ctl(st->st_low, SPEEX_GET_LOOKAHEAD, ptr);
|
|
(*(spx_int32_t*)ptr) = 2*(*(spx_int32_t*)ptr);
|
|
break;
|
|
case SPEEX_SET_HIGHPASS:
|
|
speex_decoder_ctl(st->st_low, SPEEX_SET_HIGHPASS, ptr);
|
|
break;
|
|
case SPEEX_GET_HIGHPASS:
|
|
speex_decoder_ctl(st->st_low, SPEEX_GET_HIGHPASS, ptr);
|
|
break;
|
|
case SPEEX_GET_ACTIVITY:
|
|
speex_decoder_ctl(st->st_low, SPEEX_GET_ACTIVITY, ptr);
|
|
break;
|
|
case SPEEX_GET_PI_GAIN:
|
|
{
|
|
int i;
|
|
spx_word32_t *g = (spx_word32_t*)ptr;
|
|
for (i=0;i<st->nbSubframes;i++)
|
|
g[i]=st->pi_gain[i];
|
|
}
|
|
break;
|
|
case SPEEX_GET_EXC:
|
|
{
|
|
int i;
|
|
for (i=0;i<st->nbSubframes;i++)
|
|
((spx_word16_t*)ptr)[i] = st->exc_rms[i];
|
|
}
|
|
break;
|
|
case SPEEX_GET_DTX_STATUS:
|
|
speex_decoder_ctl(st->st_low, SPEEX_GET_DTX_STATUS, ptr);
|
|
break;
|
|
case SPEEX_SET_INNOVATION_SAVE:
|
|
st->innov_save = (spx_word16_t*)ptr;
|
|
break;
|
|
case SPEEX_SET_WIDEBAND:
|
|
speex_decoder_ctl(st->st_low, SPEEX_SET_WIDEBAND, ptr);
|
|
break;
|
|
case SPEEX_GET_STACK:
|
|
*((char**)ptr) = st->stack;
|
|
break;
|
|
default:
|
|
speex_warning_int("Unknown nb_ctl request: ", request);
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
#endif
|
|
|