c022bb8d51
Remove ENABLE_IAX option Add an improved FGCom standalone as part of FG in utils directory
253 lines
6.4 KiB
C
253 lines
6.4 KiB
C
/* Copyright (C) 2002 Jean-Marc Valin
|
|
File: vq.c
|
|
Vector quantization
|
|
|
|
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 "vq.h"
|
|
#include "stack_alloc.h"
|
|
|
|
int scal_quant(spx_word16_t in, const spx_word16_t *boundary, int entries)
|
|
{
|
|
int i=0;
|
|
while (i<entries-1 && in>boundary[0])
|
|
{
|
|
boundary++;
|
|
i++;
|
|
}
|
|
return i;
|
|
}
|
|
|
|
int scal_quant32(spx_word32_t in, const spx_word32_t *boundary, int entries)
|
|
{
|
|
int i=0;
|
|
while (i<entries-1 && in>boundary[0])
|
|
{
|
|
boundary++;
|
|
i++;
|
|
}
|
|
return i;
|
|
}
|
|
|
|
/*Finds the index of the entry in a codebook that best matches the input*/
|
|
int vq_index(float *in, const float *codebook, int len, int entries)
|
|
{
|
|
int i,j;
|
|
float min_dist=0;
|
|
int best_index=0;
|
|
for (i=0;i<entries;i++)
|
|
{
|
|
float dist=0;
|
|
for (j=0;j<len;j++)
|
|
{
|
|
float tmp = in[j]-*codebook++;
|
|
dist += tmp*tmp;
|
|
}
|
|
if (i==0 || dist<min_dist)
|
|
{
|
|
min_dist=dist;
|
|
best_index=i;
|
|
}
|
|
}
|
|
return best_index;
|
|
}
|
|
|
|
#ifdef _USE_SSE
|
|
#include <xmmintrin.h>
|
|
#include "misc.h"
|
|
void vq_nbest(spx_word16_t *_in, const __m128 *codebook, int len, int entries, __m128 *E, int N, int *nbest, spx_word32_t *best_dist, char *stack)
|
|
{
|
|
int i,j,k,used;
|
|
VARDECL(float *dist);
|
|
VARDECL(__m128 *in);
|
|
__m128 half;
|
|
used = 0;
|
|
ALLOC(dist, entries, float);
|
|
half = _mm_set_ps1(.5f);
|
|
ALLOC(in, len, __m128);
|
|
for (i=0;i<len;i++)
|
|
in[i] = _mm_set_ps1(_in[i]);
|
|
for (i=0;i<entries>>2;i++)
|
|
{
|
|
__m128 d = _mm_mul_ps(E[i], half);
|
|
for (j=0;j<len;j++)
|
|
d = _mm_sub_ps(d, _mm_mul_ps(in[j], *codebook++));
|
|
_mm_storeu_ps(dist+4*i, d);
|
|
}
|
|
for (i=0;i<entries;i++)
|
|
{
|
|
if (i<N || dist[i]<best_dist[N-1])
|
|
{
|
|
for (k=N-1; (k >= 1) && (k > used || dist[i] < best_dist[k-1]); k--)
|
|
{
|
|
best_dist[k]=best_dist[k-1];
|
|
nbest[k] = nbest[k-1];
|
|
}
|
|
best_dist[k]=dist[i];
|
|
nbest[k]=i;
|
|
used++;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
#else
|
|
|
|
#if defined(SHORTCUTS) && (defined(ARM4_ASM) || defined(ARM5E_ASM))
|
|
#include "vq_arm4.h"
|
|
#else
|
|
|
|
/*Finds the indices of the n-best entries in a codebook*/
|
|
void vq_nbest(spx_word16_t *in, const spx_word16_t *codebook, int len, int entries, spx_word32_t *E, int N, int *nbest, spx_word32_t *best_dist, char *stack)
|
|
{
|
|
int i,j,k,used;
|
|
used = 0;
|
|
for (i=0;i<entries;i++)
|
|
{
|
|
spx_word32_t dist=0;
|
|
for (j=0;j<len;j++)
|
|
dist = MAC16_16(dist,in[j],*codebook++);
|
|
#ifdef FIXED_POINT
|
|
dist=SUB32(SHR32(E[i],1),dist);
|
|
#else
|
|
dist=.5f*E[i]-dist;
|
|
#endif
|
|
if (i<N || dist<best_dist[N-1])
|
|
{
|
|
for (k=N-1; (k >= 1) && (k > used || dist < best_dist[k-1]); k--)
|
|
{
|
|
best_dist[k]=best_dist[k-1];
|
|
nbest[k] = nbest[k-1];
|
|
}
|
|
best_dist[k]=dist;
|
|
nbest[k]=i;
|
|
used++;
|
|
}
|
|
}
|
|
}
|
|
#endif
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifdef _USE_SSE
|
|
|
|
void vq_nbest_sign(spx_word16_t *_in, const __m128 *codebook, int len, int entries, __m128 *E, int N, int *nbest, spx_word32_t *best_dist, char *stack)
|
|
{
|
|
int i,j,k,used;
|
|
VARDECL(float *dist);
|
|
VARDECL(__m128 *in);
|
|
__m128 half;
|
|
used = 0;
|
|
ALLOC(dist, entries, float);
|
|
half = _mm_set_ps1(.5f);
|
|
ALLOC(in, len, __m128);
|
|
for (i=0;i<len;i++)
|
|
in[i] = _mm_set_ps1(_in[i]);
|
|
for (i=0;i<entries>>2;i++)
|
|
{
|
|
__m128 d = _mm_setzero_ps();
|
|
for (j=0;j<len;j++)
|
|
d = _mm_add_ps(d, _mm_mul_ps(in[j], *codebook++));
|
|
_mm_storeu_ps(dist+4*i, d);
|
|
}
|
|
for (i=0;i<entries;i++)
|
|
{
|
|
int sign;
|
|
if (dist[i]>0)
|
|
{
|
|
sign=0;
|
|
dist[i]=-dist[i];
|
|
} else
|
|
{
|
|
sign=1;
|
|
}
|
|
dist[i] += .5f*((float*)E)[i];
|
|
if (i<N || dist[i]<best_dist[N-1])
|
|
{
|
|
for (k=N-1; (k >= 1) && (k > used || dist[i] < best_dist[k-1]); k--)
|
|
{
|
|
best_dist[k]=best_dist[k-1];
|
|
nbest[k] = nbest[k-1];
|
|
}
|
|
best_dist[k]=dist[i];
|
|
nbest[k]=i;
|
|
used++;
|
|
if (sign)
|
|
nbest[k]+=entries;
|
|
}
|
|
}
|
|
}
|
|
|
|
#else
|
|
|
|
/*Finds the indices of the n-best entries in a codebook with sign*/
|
|
void vq_nbest_sign(spx_word16_t *in, const spx_word16_t *codebook, int len, int entries, spx_word32_t *E, int N, int *nbest, spx_word32_t *best_dist, char *stack)
|
|
{
|
|
int i,j,k, sign, used;
|
|
used=0;
|
|
for (i=0;i<entries;i++)
|
|
{
|
|
spx_word32_t dist=0;
|
|
for (j=0;j<len;j++)
|
|
dist = MAC16_16(dist,in[j],*codebook++);
|
|
if (dist>0)
|
|
{
|
|
sign=0;
|
|
dist=-dist;
|
|
} else
|
|
{
|
|
sign=1;
|
|
}
|
|
#ifdef FIXED_POINT
|
|
dist = ADD32(dist,SHR32(E[i],1));
|
|
#else
|
|
dist = ADD32(dist,.5f*E[i]);
|
|
#endif
|
|
if (i<N || dist<best_dist[N-1])
|
|
{
|
|
for (k=N-1; (k >= 1) && (k > used || dist < best_dist[k-1]); k--)
|
|
{
|
|
best_dist[k]=best_dist[k-1];
|
|
nbest[k] = nbest[k-1];
|
|
}
|
|
best_dist[k]=dist;
|
|
nbest[k]=i;
|
|
used++;
|
|
if (sign)
|
|
nbest[k]+=entries;
|
|
}
|
|
}
|
|
}
|
|
#endif
|