198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project/* Copyright (C) 2002 Jean-Marc Valin */
298913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project/**
398913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   @file ltp.h
498913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   @brief Long-Term Prediction functions
598913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project*/
698913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project/*
798913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   Redistribution and use in source and binary forms, with or without
898913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   modification, are permitted provided that the following conditions
998913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   are met:
1098913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project
1198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   - Redistributions of source code must retain the above copyright
1298913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   notice, this list of conditions and the following disclaimer.
1398913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project
1498913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   - Redistributions in binary form must reproduce the above copyright
1598913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   notice, this list of conditions and the following disclaimer in the
1698913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   documentation and/or other materials provided with the distribution.
1798913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project
1898913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   - Neither the name of the Xiph.org Foundation nor the names of its
1998913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   contributors may be used to endorse or promote products derived from
2098913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   this software without specific prior written permission.
2198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project
2298913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
2398913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
2498913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
2598913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
2698913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
2798913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
2898913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
2998913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
3098913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
3198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
3298913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
3398913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project*/
3498913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project
3598913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project#include <speex/speex_bits.h>
3698913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project#include "arch.h"
3798913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project
3898913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project/** LTP parameters. */
3998913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projecttypedef struct {
4098913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   const signed char *gain_cdbk;
4198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   int     gain_bits;
4298913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project   int     pitch_bits;
4398913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project} ltp_params;
4498913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project
4598913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project#ifdef FIXED_POINT
4698913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project#define gain_3tap_to_1tap(g) (ABS(g[1]) + (g[0]>0 ? g[0] : -SHR16(g[0],1)) + (g[2]>0 ? g[2] : -SHR16(g[2],1)))
4798913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project#else
4898913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project#define gain_3tap_to_1tap(g) (ABS(g[1]) + (g[0]>0 ? g[0] : -.5*g[0]) + (g[2]>0 ? g[2] : -.5*g[2]))
4998913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project#endif
5098913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project
5198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word32_t inner_prod(const spx_word16_t *x, const spx_word16_t *y, int len);
5298913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectvoid pitch_xcorr(const spx_word16_t *_x, const spx_word16_t *_y, spx_word32_t *corr, int len, int nb_pitch, char *stack);
5398913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project
5498913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectvoid open_loop_nbest_pitch(spx_word16_t *sw, int start, int end, int len, int *pitch, spx_word16_t *gain, int N, char *stack);
5598913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project
5698913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project
5798913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project/** Finds the best quantized 3-tap pitch predictor by analysis by synthesis */
5898913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint pitch_search_3tap(
5998913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t target[],                 /* Target vector */
6098913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t *sw,
6198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_coef_t ak[],                     /* LPCs for this subframe */
6298913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_coef_t awk1[],                   /* Weighted LPCs #1 for this subframe */
6398913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_coef_t awk2[],                   /* Weighted LPCs #2 for this subframe */
6498913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_sig_t exc[],                    /* Overlapping codebook */
6598913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectconst void *par,
6698913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   start,                    /* Smallest pitch value allowed */
6798913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   end,                      /* Largest pitch value allowed */
6898913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t pitch_coef,               /* Voicing (pitch) coefficient */
6998913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   p,                        /* Number of LPC coeffs */
7098913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   nsf,                      /* Number of samples in subframe */
7198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source ProjectSpeexBits *bits,
7298913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectchar *stack,
7398913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t *exc2,
7498913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t *r,
7598913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   complexity,
7698913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   cdbk_offset,
7798913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint plc_tuning,
7898913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word32_t *cumul_gain
7998913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project);
8098913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project
8198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project/*Unquantize adaptive codebook and update pitch contribution*/
8298913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectvoid pitch_unquant_3tap(
8398913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t exc[],             /* Input excitation */
8498913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word32_t exc_out[],         /* Output excitation */
8598913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   start,                    /* Smallest pitch value allowed */
8698913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   end,                      /* Largest pitch value allowed */
8798913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t pitch_coef,        /* Voicing (pitch) coefficient */
8898913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectconst void *par,
8998913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   nsf,                      /* Number of samples in subframe */
9098913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint *pitch_val,
9198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t *gain_val,
9298913fed6520d8849fb2e246be943e04474aefaThe Android Open Source ProjectSpeexBits *bits,
9398913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectchar *stack,
9498913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint lost,
9598913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint subframe_offset,
9698913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t last_pitch_gain,
9798913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint cdbk_offset
9898913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project);
9998913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project
10098913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project/** Forced pitch delay and gain */
10198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint forced_pitch_quant(
10298913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t target[],                 /* Target vector */
10398913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t *sw,
10498913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_coef_t ak[],                     /* LPCs for this subframe */
10598913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_coef_t awk1[],                   /* Weighted LPCs #1 for this subframe */
10698913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_coef_t awk2[],                   /* Weighted LPCs #2 for this subframe */
10798913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_sig_t exc[],                    /* Excitation */
10898913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectconst void *par,
10998913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   start,                    /* Smallest pitch value allowed */
11098913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   end,                      /* Largest pitch value allowed */
11198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t pitch_coef,               /* Voicing (pitch) coefficient */
11298913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   p,                        /* Number of LPC coeffs */
11398913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   nsf,                      /* Number of samples in subframe */
11498913fed6520d8849fb2e246be943e04474aefaThe Android Open Source ProjectSpeexBits *bits,
11598913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectchar *stack,
11698913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t *exc2,
11798913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t *r,
11898913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint complexity,
11998913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint cdbk_offset,
12098913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint plc_tuning,
12198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word32_t *cumul_gain
12298913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project);
12398913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project
12498913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project/** Unquantize forced pitch delay and gain */
12598913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectvoid forced_pitch_unquant(
12698913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t exc[],             /* Input excitation */
12798913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word32_t exc_out[],         /* Output excitation */
12898913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   start,                    /* Smallest pitch value allowed */
12998913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   end,                      /* Largest pitch value allowed */
13098913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t pitch_coef,        /* Voicing (pitch) coefficient */
13198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectconst void *par,
13298913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint   nsf,                      /* Number of samples in subframe */
13398913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint *pitch_val,
13498913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t *gain_val,
13598913fed6520d8849fb2e246be943e04474aefaThe Android Open Source ProjectSpeexBits *bits,
13698913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectchar *stack,
13798913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint lost,
13898913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint subframe_offset,
13998913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectspx_word16_t last_pitch_gain,
14098913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Projectint cdbk_offset
14198913fed6520d8849fb2e246be943e04474aefaThe Android Open Source Project);
142