Updating dump_modes to include the MDCT and FFT. More work needed.
authorJean-Marc Valin <jean-marc.valin@octasic.com>
Wed, 25 Aug 2010 22:01:50 +0000 (18:01 -0400)
committerJean-Marc Valin <jean-marc.valin@octasic.com>
Wed, 25 Aug 2010 22:01:50 +0000 (18:01 -0400)
libcelt/_kiss_fft_guts.h
libcelt/dump_modes.c
libcelt/kiss_fft.c
libcelt/kiss_fft.h
libcelt/mdct.h
tests/dft-test.c

index 74d9ecb..5d5aa3b 100644 (file)
@@ -24,22 +24,6 @@ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
    typedef struct { kiss_fft_scalar r; kiss_fft_scalar i; }kiss_fft_cpx; */
 #include "kiss_fft.h"
 
-#define MAXFACTORS 8
-/* e.g. an fft of length 128 has 4 factors 
- as far as kissfft is concerned
- 4*4*4*2
- */
-
-struct kiss_fft_state{
-    int nfft;
-#ifndef FIXED_POINT
-    kiss_fft_scalar scale;
-#endif
-    int shift;
-    celt_int16 factors[2*MAXFACTORS];
-    celt_int16 *bitrev;
-    kiss_twiddle_cpx *twiddles;
-};
 
 /*
   Explanation of macros dealing with complex math:
index 8c05845..627ab96 100644 (file)
@@ -122,14 +122,22 @@ void dump_modes(FILE *file, CELTMode **modes, int nb_modes)
       fprintf(file, "#endif\n");
       fprintf(file, "\n");
 
-      /* FFTs */
-      /*
-      for (i=0;i<mode->mdct.maxshift;i++)
+      /* FFT twiddles */
+      fprintf(file, "#ifndef FFT_TWIDDLES%d_%d\n", mode->Fs, mdctSize);
+      fprintf(file, "#define FFT_TWIDDLES%d_%d\n", mode->Fs, mdctSize);
+      fprintf (file, "static const kiss_twiddle_cpx fft_twiddles%d_%d[%d] = {\n",
+            mode->Fs, mdctSize, mode->mdct.kfft[0]->nfft);
+      for (j=0;j<mode->mdct.kfft[0]->nfft;j++)
+         fprintf (file, "{" WORD16 ", " WORD16 "}, ", mode->mdct.kfft[0]->twiddles[j].r, mode->mdct.kfft[0]->twiddles[j].i);
+      fprintf (file, "};\n");
+
+      /* FFT Bitrev tables */
+      for (i=0;i<=mode->mdct.maxshift;i++)
       {
          fprintf(file, "#ifndef FFT_BITREV%d_%d\n", mode->Fs, mdctSize>>i);
          fprintf(file, "#define FFT_BITREV%d_%d\n", mode->Fs, mdctSize>>i);
          fprintf (file, "static const celt_int16 fft_bitrev%d_%d[%d] = {\n",
-               mode->Fs, mdctSize/i, mode->mdct.kfft[i]->nfft);
+               mode->Fs, mdctSize>>i, mode->mdct.kfft[i]->nfft);
          for (j=0;j<mode->mdct.kfft[i]->nfft;j++)
             fprintf (file, "%d, ", mode->mdct.kfft[i]->bitrev[j]);
          fprintf (file, "};\n");
@@ -137,7 +145,46 @@ void dump_modes(FILE *file, CELTMode **modes, int nb_modes)
          fprintf(file, "#endif\n");
          fprintf(file, "\n");
       }
-      */
+
+      /* FFT States */
+      for (i=0;i<=mode->mdct.maxshift;i++)
+      {
+         fprintf(file, "#ifndef FFT_STATE%d_%d\n", mode->Fs, mdctSize>>i);
+         fprintf(file, "#define FFT_STATE%d_%d\n", mode->Fs, mdctSize>>i);
+         fprintf (file, "static const kiss_fft_state fft_state%d_%d = {\n",
+               mode->Fs, mdctSize>>i);
+         fprintf (file, "%d,\t/* nfft */\n", mode->mdct.kfft[i]->nfft);
+#ifndef FIXED_POINT
+         fprintf (file, "%f,\t/* scale */\n", mode->mdct.kfft[i]->scale);
+#endif
+         fprintf (file, "%d,\t/* shift */\n", mode->mdct.kfft[i]->shift);
+         fprintf (file, "{");
+         for (j=0;j<2*MAXFACTORS;j++)
+            fprintf (file, "%d, ", mode->mdct.kfft[i]->factors[j]);
+         fprintf (file, "},\t/* factors */\n");
+         fprintf (file, "fft_bitrev%d_%d,\t/* bitrev */\n", mode->Fs, mdctSize>>i);
+         fprintf (file, "fft_twiddles%d_%d,\t/* bitrev */\n", mode->Fs, mdctSize);
+         fprintf (file, "};\n");
+
+         fprintf(file, "#endif\n");
+         fprintf(file, "\n");
+      }
+
+      fprintf(file, "#endif\n");
+      fprintf(file, "\n");
+
+      /* MDCT twiddles */
+      fprintf(file, "#ifndef MDCT_TWIDDLES%d_%d\n", mode->Fs, mdctSize);
+      fprintf(file, "#define MDCT_TWIDDLES%d_%d\n", mode->Fs, mdctSize);
+      fprintf (file, "static const celt_word16 mdct_twiddles%d_%d[%d] = {\n",
+            mode->Fs, mdctSize, mode->mdct.n/4);
+      for (j=0;j<mode->mdct.n/4;j++)
+         fprintf (file, WORD16 ", ", mode->mdct.trig[j]);
+      fprintf (file, "};\n");
+
+      fprintf(file, "#endif\n");
+      fprintf(file, "\n");
+
 
       /* Print the actual mode data */
       fprintf(file, "static const CELTMode mode%d_%d_%d = {\n", mode->Fs, mdctSize, mode->overlap);
@@ -154,8 +201,10 @@ void dump_modes(FILE *file, CELTMode **modes, int nb_modes)
       fprintf(file, "%d,\t/* nbAllocVectors */\n", mode->nbAllocVectors);
       fprintf(file, "allocVectors%d_%d,\t/* allocVectors */\n", mode->Fs, mdctSize);
 
-      fprintf(file, "{%d, %d, 0, 0},\t", mode->mdct.n, mode->mdct.maxshift);
-      fprintf (file, "/* mdct */\n");
+      fprintf(file, "{%d, %d, {", mode->mdct.n, mode->mdct.maxshift);
+      for (i=0;i<=mode->mdct.maxshift;i++)
+         fprintf(file, "fft_state%d_%d, ", mode->Fs, mdctSize>>i);
+      fprintf (file, "}, mdct_twiddles%d_%d},\t/* mdct */\n", mode->Fs, mdctSize);
 
       fprintf(file, "window%d,\t/* window */\n", mode->overlap);
       fprintf(file, "%d,\t/* maxLM */\n", mode->maxLM);
index b1159be..4513498 100644 (file)
@@ -34,7 +34,7 @@ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 static void kf_bfly2(
                      kiss_fft_cpx * Fout,
                      const size_t fstride,
-                     const kiss_fft_cfg st,
+                     const kiss_fft_state *st,
                      int m,
                      int N,
                      int mm
@@ -67,7 +67,7 @@ static void kf_bfly2(
 static void ki_bfly2(
                      kiss_fft_cpx * Fout,
                      const size_t fstride,
-                     const kiss_fft_cfg st,
+                     const kiss_fft_state *st,
                      int m,
                      int N,
                      int mm
@@ -98,7 +98,7 @@ static void ki_bfly2(
 static void kf_bfly4(
                      kiss_fft_cpx * Fout,
                      const size_t fstride,
-                     const kiss_fft_cfg st,
+                     const kiss_fft_state *st,
                      int m,
                      int N,
                      int mm
@@ -147,7 +147,7 @@ static void kf_bfly4(
 static void ki_bfly4(
                      kiss_fft_cpx * Fout,
                      const size_t fstride,
-                     const kiss_fft_cfg st,
+                     const kiss_fft_state *st,
                      int m,
                      int N,
                      int mm
@@ -194,7 +194,7 @@ static void ki_bfly4(
 static void kf_bfly3(
                      kiss_fft_cpx * Fout,
                      const size_t fstride,
-                     const kiss_fft_cfg st,
+                     const kiss_fft_state *st,
                      int m,
                      int N,
                      int mm
@@ -246,7 +246,7 @@ static void kf_bfly3(
 static void ki_bfly3(
                      kiss_fft_cpx * Fout,
                      const size_t fstride,
-                     const kiss_fft_cfg st,
+                     const kiss_fft_state *st,
                      size_t m,
                      int N,
                      int mm
@@ -297,7 +297,7 @@ static void ki_bfly3(
 static void kf_bfly5(
                      kiss_fft_cpx * Fout,
                      const size_t fstride,
-                     const kiss_fft_cfg st,
+                     const kiss_fft_state *st,
                      int m,
                      int N,
                      int mm
@@ -366,7 +366,7 @@ static void kf_bfly5(
 static void ki_bfly5(
                      kiss_fft_cpx * Fout,
                      const size_t fstride,
-                     const kiss_fft_cfg st,
+                     const kiss_fft_state *st,
                      int m,
                      int N,
                      int mm
@@ -440,7 +440,7 @@ void compute_bitrev_table(
          const size_t fstride,
          int in_stride,
          celt_int16 * factors,
-         const kiss_fft_cfg st
+         const kiss_fft_state *st
             )
 {
    const int p=*factors++; /* the radix  */
@@ -473,7 +473,7 @@ static void kf_work(
         size_t fstride,
         int in_stride,
         celt_int16 * factors,
-        const kiss_fft_cfg st,
+        const kiss_fft_state *st,
         int N,
         int s2,
         int m2
@@ -507,7 +507,7 @@ static void ki_work(
              size_t fstride,
              int in_stride,
              celt_int16 * factors,
-             const kiss_fft_cfg st,
+             const kiss_fft_state *st,
              int N,
              int s2,
              int m2
@@ -591,16 +591,16 @@ static void compute_twiddles(kiss_twiddle_cpx *twiddles, int nfft)
  * The return value is a contiguous block of memory, allocated with malloc.  As such,
  * It can be freed with free(), rather than a kiss_fft-specific function.
  * */
-kiss_fft_cfg kiss_fft_alloc_twiddles(int nfft,void * mem,size_t * lenmem,  kiss_fft_cfg base)
+kiss_fft_state *kiss_fft_alloc_twiddles(int nfft,void * mem,size_t * lenmem,  kiss_fft_state *base)
 {
-    kiss_fft_cfg st=NULL;
+    kiss_fft_state *st=NULL;
     size_t memneeded = sizeof(struct kiss_fft_state); /* twiddle factors*/
 
     if ( lenmem==NULL ) {
-        st = ( kiss_fft_cfg)KISS_FFT_MALLOC( memneeded );
+        st = ( kiss_fft_state*)KISS_FFT_MALLOC( memneeded );
     }else{
         if (mem != NULL && *lenmem >= memneeded)
-            st = (kiss_fft_cfg)mem;
+            st = (kiss_fft_state*)mem;
         *lenmem = memneeded;
     }
     if (st) {
@@ -635,13 +635,13 @@ kiss_fft_cfg kiss_fft_alloc_twiddles(int nfft,void * mem,size_t * lenmem,  kiss_
     return st;
 }
 
-kiss_fft_cfg kiss_fft_alloc(int nfft,void * mem,size_t * lenmem )
+kiss_fft_state *kiss_fft_alloc(int nfft,void * mem,size_t * lenmem )
 {
    return kiss_fft_alloc_twiddles(nfft, mem, lenmem, NULL);
 }
 
     
-static void kiss_fft_stride(kiss_fft_cfg st,const kiss_fft_cpx *fin,kiss_fft_cpx *fout,int in_stride)
+static void kiss_fft_stride(kiss_fft_state *st,const kiss_fft_cpx *fin,kiss_fft_cpx *fout,int in_stride)
 {
     if (fin == fout) 
     {
@@ -661,12 +661,12 @@ static void kiss_fft_stride(kiss_fft_cfg st,const kiss_fft_cpx *fin,kiss_fft_cpx
     }
 }
 
-void kiss_fft(kiss_fft_cfg cfg,const kiss_fft_cpx *fin,kiss_fft_cpx *fout)
+void kiss_fft(kiss_fft_state *cfg,const kiss_fft_cpx *fin,kiss_fft_cpx *fout)
 {
     kiss_fft_stride(cfg,fin,fout,1);
 }
 
-static void kiss_ifft_stride(kiss_fft_cfg st,const kiss_fft_cpx *fin,kiss_fft_cpx *fout,int in_stride)
+static void kiss_ifft_stride(kiss_fft_state *st,const kiss_fft_cpx *fin,kiss_fft_cpx *fout,int in_stride)
 {
    if (fin == fout) 
    {
@@ -680,12 +680,12 @@ static void kiss_ifft_stride(kiss_fft_cfg st,const kiss_fft_cpx *fin,kiss_fft_cp
    }
 }
 
-void kiss_ifft(kiss_fft_cfg cfg,const kiss_fft_cpx *fin,kiss_fft_cpx *fout)
+void kiss_ifft(kiss_fft_state *cfg,const kiss_fft_cpx *fin,kiss_fft_cpx *fout)
 {
    kiss_ifft_stride(cfg,fin,fout,1);
 }
 
-void kiss_fft_free(kiss_fft_cfg cfg)
+void kiss_fft_free(kiss_fft_state *cfg)
 {
    celt_free(cfg->bitrev);
    if (cfg->shift < 0)
index 3d88c6d..4fdb491 100644 (file)
@@ -97,7 +97,24 @@ typedef struct {
    kiss_twiddle_scalar i;
 }kiss_twiddle_cpx;
 
-typedef struct kiss_fft_state* kiss_fft_cfg;
+#define MAXFACTORS 8
+/* e.g. an fft of length 128 has 4 factors
+ as far as kissfft is concerned
+ 4*4*4*2
+ */
+
+typedef struct kiss_fft_state{
+    int nfft;
+#ifndef FIXED_POINT
+    kiss_fft_scalar scale;
+#endif
+    int shift;
+    celt_int16 factors[2*MAXFACTORS];
+    celt_int16 *bitrev;
+    kiss_twiddle_cpx *twiddles;
+} kiss_fft_state;
+
+//typedef struct kiss_fft_state* kiss_fft_cfg;
 
 /** 
  *  kiss_fft_alloc
@@ -122,9 +139,9 @@ typedef struct kiss_fft_state* kiss_fft_cfg;
  *      buffer size in *lenmem.
  * */
 
-kiss_fft_cfg kiss_fft_alloc_twiddles(int nfft,void * mem,size_t * lenmem,  kiss_fft_cfg base);
+kiss_fft_state *kiss_fft_alloc_twiddles(int nfft,void * mem,size_t * lenmem,  kiss_fft_state *base);
 
-kiss_fft_cfg kiss_fft_alloc(int nfft,void * mem,size_t * lenmem); 
+kiss_fft_state *kiss_fft_alloc(int nfft,void * mem,size_t * lenmem);
 
 /**
  * kiss_fft(cfg,in_out_buf)
@@ -136,10 +153,10 @@ kiss_fft_cfg kiss_fft_alloc(int nfft,void * mem,size_t * lenmem);
  * Note that each element is complex and can be accessed like
     f[k].r and f[k].i
  * */
-void kiss_fft(kiss_fft_cfg cfg,const kiss_fft_cpx *fin,kiss_fft_cpx *fout);
-void kiss_ifft(kiss_fft_cfg cfg,const kiss_fft_cpx *fin,kiss_fft_cpx *fout);
+void kiss_fft(kiss_fft_state *cfg,const kiss_fft_cpx *fin,kiss_fft_cpx *fout);
+void kiss_ifft(kiss_fft_state *cfg,const kiss_fft_cpx *fin,kiss_fft_cpx *fout);
 
-void kiss_fft_free(kiss_fft_cfg cfg);
+void kiss_fft_free(kiss_fft_state *cfg);
 
 
 #ifdef __cplusplus
index 9170cb7..d2a2c36 100644 (file)
@@ -52,7 +52,7 @@
 typedef struct {
    int n;
    int maxshift;
-   kiss_fft_cfg kfft[4];
+   kiss_fft_state *kfft[4];
    kiss_twiddle_scalar * restrict trig;
 } mdct_lookup;
 
index 8f3f1d7..749205a 100644 (file)
@@ -68,7 +68,7 @@ void test1d(int nfft,int isinverse)
 
     kiss_fft_cpx  * in = (kiss_fft_cpx*)malloc(buflen);
     kiss_fft_cpx  * out= (kiss_fft_cpx*)malloc(buflen);
-    kiss_fft_cfg  cfg = kiss_fft_alloc(nfft,0,0);
+    kiss_fft_state *cfg = kiss_fft_alloc(nfft,0,0);
     int k;
 
     for (k=0;k<nfft;++k) {