Removing original freq-domain pitch code
[opus.git] / libcelt / pitch.c
index fc68d38..f04934e 100644 (file)
 #include "config.h"
 #endif
 
-/*#include "_kiss_fft_guts.h"
-#include "kiss_fftr.h"*/
-#include "kfft_single.h"
-
 #include "pitch.h"
-#include "psy.h"
 #include "os_support.h"
-#include "mathops.h"
 #include "modes.h"
 #include "stack_alloc.h"
-
-kiss_fftr_cfg pitch_state_alloc(int max_lag)
-{
-   return real16_fft_alloc(max_lag);
-}
-
-void pitch_state_free(kiss_fftr_cfg st)
-{
-   real16_fft_free(st);
-}
-
-#ifdef FIXED_POINT
-static void normalise16(celt_word16 *x, int len, celt_word16 val)
-{
-   int i;
-   celt_word16 maxabs;
-   maxabs = celt_maxabs16(x,len);
-   if (maxabs > val)
-   {
-      int shift = 0;
-      while (maxabs > val)
-      {
-         maxabs >>= 1;
-         shift++;
-      }
-      if (shift==0)
-         return;
-      i=0;
-      do{
-         x[i] = SHR16(x[i], shift);
-      } while (++i<len);
-   } else {
-      int shift=0;
-      if (maxabs == 0)
-         return;
-      val >>= 1;
-      while (maxabs < val)
-      {
-         val >>= 1;
-         shift++;
-      }
-      if (shift==0)
-         return;
-      i=0;
-      do{
-         x[i] = SHL16(x[i], shift);
-      } while (++i<len);
-   }
-}
-#else
-#define normalise16(x,len,val)
-#endif
-
-#define INPUT_SHIFT 15
-
-void find_spectral_pitch(const CELTMode *m, kiss_fftr_cfg fft, const struct PsyDecay *decay, const celt_sig * restrict x, const celt_sig * restrict y, const celt_word16 * restrict window, celt_word16 * restrict spectrum, int len, int max_pitch, int *pitch, int _C)
-{
-   int c, i;
-   VARDECL(celt_word16, _X);
-   VARDECL(celt_word16, _Y);
-   const celt_word16 * restrict wptr;
-#ifndef SHORTCUTS
-   VARDECL(celt_mask, curve);
-#endif
-   celt_word16 * restrict X, * restrict Y;
-   celt_word16 * restrict Xptr, * restrict Yptr;
-   const celt_sig * restrict yptr;
-   int n2;
-   int L2;
-   const int C = CHANNELS(_C);
-   const int overlap = OVERLAP(m);
-   const int lag = MAX_PERIOD;
-   SAVE_STACK;
-   n2 = lag>>1;
-   L2 = len>>1;
-   ALLOC(_X, lag, celt_word16);
-   X = _X;
-#ifndef SHORTCUTS
-   ALLOC(curve, n2, celt_mask);
-#endif
-   CELT_MEMSET(X,0,lag);
-   /* Sum all channels of the current frame and copy into X in bit-reverse order */
-   for (c=0;c<C;c++)
-   {
-      const celt_sig * restrict xptr = &x[c];
-      for (i=0;i<L2;i++)
-      {
-         X[2*BITREV(fft,i)] += SHR32(*xptr,INPUT_SHIFT);
-         xptr += C;
-         X[2*BITREV(fft,i)+1] += SHR32(*xptr,INPUT_SHIFT);
-         xptr += C;
-      }
-   }
-   /* Applying the window in the bit-reverse domain. It's a bit weird, but it
-      can help save memory */
-   wptr = window;
-   for (i=0;i<overlap>>1;i++)
-   {
-      X[2*BITREV(fft,i)]        = MULT16_16_Q15(wptr[0], X[2*BITREV(fft,i)]);
-      X[2*BITREV(fft,i)+1]      = MULT16_16_Q15(wptr[1], X[2*BITREV(fft,i)+1]);
-      X[2*BITREV(fft,L2-i-1)]   = MULT16_16_Q15(wptr[1], X[2*BITREV(fft,L2-i-1)]);
-      X[2*BITREV(fft,L2-i-1)+1] = MULT16_16_Q15(wptr[0], X[2*BITREV(fft,L2-i-1)+1]);
-      wptr += 2;
-   }
-   normalise16(X, lag, 8192);
-   /*for (i=0;i<lag;i++) printf ("%d ", X[i]);printf ("\n");*/
-   /* Forward real FFT (in-place) */
-   real16_fft_inplace(fft, X, lag);
-
-   if (spectrum)
-   {
-      for (i=0;i<lag/4;i++)
-      {
-         spectrum[2*i] = X[4*i];
-         spectrum[2*i+1] = X[4*i+1];
-      }
-   }
-#ifndef SHORTCUTS
-   compute_masking(decay, X, curve, lag);
-#endif
-   
-   /* Deferred allocation to reduce peak stack usage */
-   ALLOC(_Y, lag, celt_word16);
-   Y = _Y;
-   yptr = &y[0];
-   /* Copy first channel of the past audio into Y in bit-reverse order */
-   for (i=0;i<n2;i++)
-   {
-      Y[2*BITREV(fft,i)] = SHR32(*yptr,INPUT_SHIFT);
-      yptr += C;
-      Y[2*BITREV(fft,i)+1] = SHR32(*yptr,INPUT_SHIFT);
-      yptr += C;
-   }
-   /* Add remaining channels into Y in bit-reverse order */
-   for (c=1;c<C;c++)
-   {
-      yptr = &y[c];
-      for (i=0;i<n2;i++)
-      {
-         Y[2*BITREV(fft,i)] += SHR32(*yptr,INPUT_SHIFT);
-         yptr += C;
-         Y[2*BITREV(fft,i)+1] += SHR32(*yptr,INPUT_SHIFT);
-         yptr += C;
-      }
-   }
-   normalise16(Y, lag, 8192);
-   /* Forward real FFT (in-place) */
-   real16_fft_inplace(fft, Y, lag);
-
-   /* Compute cross-spectrum using the inverse masking curve as weighting */
-   Xptr = &X[2];
-   Yptr = &Y[2];
-   for (i=1;i<n2;i++)
-   {
-      celt_word16 Xr, Xi, n;
-      /* weight = 1/sqrt(curve) */
-      Xr = Xptr[0];
-      Xi = Xptr[1];
-#ifdef SHORTCUTS
-      /*n = SHR32(32767,(celt_ilog2(EPSILON+curve[i])>>1));*/
-      n = 1+(8192>>(celt_ilog2(1+MULT16_16(Xr,Xr)+MULT16_16(Xi,Xi))>>1));
-      /* Pre-multiply X by n, so we can keep everything in 16 bits */
-      Xr = MULT16_16_16(n, Xr);
-      Xi = MULT16_16_16(n, Xi);
-#else
-      {
-         celt_word32 t;
-#ifdef FIXED_POINT
-         int k;
-#endif
-         t = EPSILON+curve[i];
-#ifdef FIXED_POINT
-         k = celt_ilog2(t)>>1;
-#endif
-         t = VSHR32(t, (k-7)<<1);
-         n = celt_rsqrt_norm(t);
-         /* Pre-multiply X by n, so we can keep everything in 16 bits */
-         Xr = EXTRACT16(PSHR32(MULT16_16(n, Xr),3+k));
-         Xi = EXTRACT16(PSHR32(MULT16_16(n, Xi),3+k));
-      }
-#endif
-      /* Cross-spectrum between X and conj(Y) */
-      *Xptr++ = ADD16(MULT16_16_Q15(Xr, Yptr[0]), MULT16_16_Q15(Xi,Yptr[1]));
-      *Xptr++ = SUB16(MULT16_16_Q15(Xr, Yptr[1]), MULT16_16_Q15(Xi,Yptr[0]));
-      Yptr += 2;
-   }
-   /*printf ("\n");*/
-   X[0] = X[1] = 0;
-   /*for (i=0;i<lag;i++) printf ("%d ", X[i]);printf ("\n");*/
-   normalise16(X, lag, 50);
-   /* Inverse half-complex to real FFT gives us the correlation */
-   real16_ifft(fft, X, Y, lag);
-   
-   /* The peak in the correlation gives us the pitch */
-   *pitch = find_max16(Y, max_pitch);
-   /*printf ("%d ", *pitch);*/
-   RESTORE_STACK;
-}
+#include "mathops.h"
 
 void find_best_pitch(celt_word32 *xcorr, celt_word32 maxcorr, celt_word16 *y, int yshift, int len, int max_pitch, int best_pitch[2])
 {