libcelt/kiss_fft.c

   1 /*
   2 Copyright (c) 2003-2004, Mark Borgerding
   3 Lots of modifications by Jean-Marc Valin
   4 Copyright (c) 2005-2007, Xiph.Org Foundation
   5 Copyright (c) 2008,      Xiph.Org Foundation, CSIRO
   6
   7 All rights reserved.
   8
   9 Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
  10
  11     * Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
  12     * 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.
  13     * Neither the author nor the names of any contributors may be used to endorse or promote products derived from this software without specific prior written permission.
  14
  15 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 COPYRIGHT OWNER 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.
  16 */
  17
  18 #ifndef SKIP_CONFIG_H
  19 #  ifdef HAVE_CONFIG_H
  20 #    include "config.h"
  21 #  endif
  22 #endif
  23
  24 #include "_kiss_fft_guts.h"
  25 #include "arch.h"
  26 #include "os_support.h"
  27 #include "mathops.h"
  28 #include "stack_alloc.h"
  29
  30 /* The guts header contains all the multiplication and addition macros that are defined for
  31    complex numbers.  It also delares the kf_ internal functions.
  32 */
  33
  34 static void kf_bfly2(
  35                      kiss_fft_cpx * Fout,
  36                      const size_t fstride,
  37                      const kiss_fft_state *st,
  38                      int m,
  39                      int N,
  40                      int mm
  41                     )
  42 {
  43    kiss_fft_cpx * Fout2;
  44    const kiss_twiddle_cpx * tw1;
  45    int i,j;
  46    kiss_fft_cpx * Fout_beg = Fout;
  47    for (i=0;i<N;i++)
  48    {
  49       Fout = Fout_beg + i*mm;
  50       Fout2 = Fout + m;
  51       tw1 = st->twiddles;
  52       for(j=0;j<m;j++)
  53       {
  54          kiss_fft_cpx t;
  55          Fout->r = SHR(Fout->r, 1);Fout->i = SHR(Fout->i, 1);
  56          Fout2->r = SHR(Fout2->r, 1);Fout2->i = SHR(Fout2->i, 1);
  57          C_MUL (t,  *Fout2 , *tw1);
  58          tw1 += fstride;
  59          C_SUB( *Fout2 ,  *Fout , t );
  60          C_ADDTO( *Fout ,  t );
  61          ++Fout2;
  62          ++Fout;
  63       }
  64    }
  65 }
  66
  67 static void ki_bfly2(
  68                      kiss_fft_cpx * Fout,
  69                      const size_t fstride,
  70                      const kiss_fft_state *st,
  71                      int m,
  72                      int N,
  73                      int mm
  74                     )
  75 {
  76    kiss_fft_cpx * Fout2;
  77    const kiss_twiddle_cpx * tw1;
  78    kiss_fft_cpx t;
  79    int i,j;
  80    kiss_fft_cpx * Fout_beg = Fout;
  81    for (i=0;i<N;i++)
  82    {
  83       Fout = Fout_beg + i*mm;
  84       Fout2 = Fout + m;
  85       tw1 = st->twiddles;
  86       for(j=0;j<m;j++)
  87       {
  88          C_MULC (t,  *Fout2 , *tw1);
  89          tw1 += fstride;
  90          C_SUB( *Fout2 ,  *Fout , t );
  91          C_ADDTO( *Fout ,  t );
  92          ++Fout2;
  93          ++Fout;
  94       }
  95    }
  96 }
  97
  98 static void kf_bfly4(
  99                      kiss_fft_cpx * Fout,
 100                      const size_t fstride,
 101                      const kiss_fft_state *st,
 102                      int m,
 103                      int N,
 104                      int mm
 105                     )
 106 {
 107    const kiss_twiddle_cpx *tw1,*tw2,*tw3;
 108    kiss_fft_cpx scratch[6];
 109    const size_t m2=2*m;
 110    const size_t m3=3*m;
 111    int i, j;
 112
 113    kiss_fft_cpx * Fout_beg = Fout;
 114    for (i=0;i<N;i++)
 115    {
 116       Fout = Fout_beg + i*mm;
 117       tw3 = tw2 = tw1 = st->twiddles;
 118       for (j=0;j<m;j++)
 119       {
 120          C_MUL4(scratch[0],Fout[m] , *tw1 );
 121          C_MUL4(scratch[1],Fout[m2] , *tw2 );
 122          C_MUL4(scratch[2],Fout[m3] , *tw3 );
 123
 124          Fout->r = PSHR(Fout->r, 2);
 125          Fout->i = PSHR(Fout->i, 2);
 126          C_SUB( scratch[5] , *Fout, scratch[1] );
 127          C_ADDTO(*Fout, scratch[1]);
 128          C_ADD( scratch[3] , scratch[0] , scratch[2] );
 129          C_SUB( scratch[4] , scratch[0] , scratch[2] );
 130          Fout[m2].r = PSHR(Fout[m2].r, 2);
 131          Fout[m2].i = PSHR(Fout[m2].i, 2);
 132          C_SUB( Fout[m2], *Fout, scratch[3] );
 133          tw1 += fstride;
 134          tw2 += fstride*2;
 135          tw3 += fstride*3;
 136          C_ADDTO( *Fout , scratch[3] );
 137
 138          Fout[m].r = scratch[5].r + scratch[4].i;
 139          Fout[m].i = scratch[5].i - scratch[4].r;
 140          Fout[m3].r = scratch[5].r - scratch[4].i;
 141          Fout[m3].i = scratch[5].i + scratch[4].r;
 142          ++Fout;
 143       }
 144    }
 145 }
 146
 147 static void ki_bfly4(
 148                      kiss_fft_cpx * Fout,
 149                      const size_t fstride,
 150                      const kiss_fft_state *st,
 151                      int m,
 152                      int N,
 153                      int mm
 154                     )
 155 {
 156    const kiss_twiddle_cpx *tw1,*tw2,*tw3;
 157    kiss_fft_cpx scratch[6];
 158    const size_t m2=2*m;
 159    const size_t m3=3*m;
 160    int i, j;
 161
 162    kiss_fft_cpx * Fout_beg = Fout;
 163    for (i=0;i<N;i++)
 164    {
 165       Fout = Fout_beg + i*mm;
 166       tw3 = tw2 = tw1 = st->twiddles;
 167       for (j=0;j<m;j++)
 168       {
 169          C_MULC(scratch[0],Fout[m] , *tw1 );
 170          C_MULC(scratch[1],Fout[m2] , *tw2 );
 171          C_MULC(scratch[2],Fout[m3] , *tw3 );
 172
 173          C_SUB( scratch[5] , *Fout, scratch[1] );
 174          C_ADDTO(*Fout, scratch[1]);
 175          C_ADD( scratch[3] , scratch[0] , scratch[2] );
 176          C_SUB( scratch[4] , scratch[0] , scratch[2] );
 177          C_SUB( Fout[m2], *Fout, scratch[3] );
 178          tw1 += fstride;
 179          tw2 += fstride*2;
 180          tw3 += fstride*3;
 181          C_ADDTO( *Fout , scratch[3] );
 182
 183          Fout[m].r = scratch[5].r - scratch[4].i;
 184          Fout[m].i = scratch[5].i + scratch[4].r;
 185          Fout[m3].r = scratch[5].r + scratch[4].i;
 186          Fout[m3].i = scratch[5].i - scratch[4].r;
 187          ++Fout;
 188       }
 189    }
 190 }
 191
 192 #ifndef RADIX_TWO_ONLY
 193
 194 static void kf_bfly3(
 195                      kiss_fft_cpx * Fout,
 196                      const size_t fstride,
 197                      const kiss_fft_state *st,
 198                      int m,
 199                      int N,
 200                      int mm
 201                     )
 202 {
 203    int i;
 204    size_t k;
 205    const size_t m2 = 2*m;
 206    const kiss_twiddle_cpx *tw1,*tw2;
 207    kiss_fft_cpx scratch[5];
 208    kiss_twiddle_cpx epi3;
 209
 210    kiss_fft_cpx * Fout_beg = Fout;
 211    epi3 = st->twiddles[fstride*m];
 212    for (i=0;i<N;i++)
 213    {
 214       Fout = Fout_beg + i*mm;
 215       tw1=tw2=st->twiddles;
 216       k=m;
 217       do {
 218          C_FIXDIV(*Fout,3); C_FIXDIV(Fout[m],3); C_FIXDIV(Fout[m2],3);
 219
 220          C_MUL(scratch[1],Fout[m] , *tw1);
 221          C_MUL(scratch[2],Fout[m2] , *tw2);
 222
 223          C_ADD(scratch[3],scratch[1],scratch[2]);
 224          C_SUB(scratch[0],scratch[1],scratch[2]);
 225          tw1 += fstride;
 226          tw2 += fstride*2;
 227
 228          Fout[m].r = Fout->r - HALF_OF(scratch[3].r);
 229          Fout[m].i = Fout->i - HALF_OF(scratch[3].i);
 230
 231          C_MULBYSCALAR( scratch[0] , epi3.i );
 232
 233          C_ADDTO(*Fout,scratch[3]);
 234
 235          Fout[m2].r = Fout[m].r + scratch[0].i;
 236          Fout[m2].i = Fout[m].i - scratch[0].r;
 237
 238          Fout[m].r -= scratch[0].i;
 239          Fout[m].i += scratch[0].r;
 240
 241          ++Fout;
 242       } while(--k);
 243    }
 244 }
 245
 246 static void ki_bfly3(
 247                      kiss_fft_cpx * Fout,
 248                      const size_t fstride,
 249                      const kiss_fft_state *st,
 250                      size_t m,
 251                      int N,
 252                      int mm
 253                     )
 254 {
 255    size_t i, k;
 256    const size_t m2 = 2*m;
 257    const kiss_twiddle_cpx *tw1,*tw2;
 258    kiss_fft_cpx scratch[5];
 259    kiss_twiddle_cpx epi3;
 260
 261    kiss_fft_cpx * Fout_beg = Fout;
 262    epi3 = st->twiddles[fstride*m];
 263    for (i=0;i<N;i++)
 264    {
 265       Fout = Fout_beg + i*mm;
 266       tw1=tw2=st->twiddles;
 267       k=m;
 268       do{
 269
 270          C_MULC(scratch[1],Fout[m] , *tw1);
 271          C_MULC(scratch[2],Fout[m2] , *tw2);
 272
 273          C_ADD(scratch[3],scratch[1],scratch[2]);
 274          C_SUB(scratch[0],scratch[1],scratch[2]);
 275          tw1 += fstride;
 276          tw2 += fstride*2;
 277
 278          Fout[m].r = Fout->r - HALF_OF(scratch[3].r);
 279          Fout[m].i = Fout->i - HALF_OF(scratch[3].i);
 280
 281          C_MULBYSCALAR( scratch[0] , -epi3.i );
 282
 283          C_ADDTO(*Fout,scratch[3]);
 284
 285          Fout[m2].r = Fout[m].r + scratch[0].i;
 286          Fout[m2].i = Fout[m].i - scratch[0].r;
 287
 288          Fout[m].r -= scratch[0].i;
 289          Fout[m].i += scratch[0].r;
 290
 291          ++Fout;
 292       }while(--k);
 293    }
 294 }
 295
 296
 297 static void kf_bfly5(
 298                      kiss_fft_cpx * Fout,
 299                      const size_t fstride,
 300                      const kiss_fft_state *st,
 301                      int m,
 302                      int N,
 303                      int mm
 304                     )
 305 {
 306    kiss_fft_cpx *Fout0,*Fout1,*Fout2,*Fout3,*Fout4;
 307    int i, u;
 308    kiss_fft_cpx scratch[13];
 309    const kiss_twiddle_cpx * twiddles = st->twiddles;
 310    const kiss_twiddle_cpx *tw;
 311    kiss_twiddle_cpx ya,yb;
 312    kiss_fft_cpx * Fout_beg = Fout;
 313
 314    ya = twiddles[fstride*m];
 315    yb = twiddles[fstride*2*m];
 316    tw=st->twiddles;
 317
 318    for (i=0;i<N;i++)
 319    {
 320       Fout = Fout_beg + i*mm;
 321       Fout0=Fout;
 322       Fout1=Fout0+m;
 323       Fout2=Fout0+2*m;
 324       Fout3=Fout0+3*m;
 325       Fout4=Fout0+4*m;
 326
 327       for ( u=0; u<m; ++u ) {
 328          C_FIXDIV( *Fout0,5); C_FIXDIV( *Fout1,5); C_FIXDIV( *Fout2,5); C_FIXDIV( *Fout3,5); C_FIXDIV( *Fout4,5);
 329          scratch[0] = *Fout0;
 330
 331          C_MUL(scratch[1] ,*Fout1, tw[u*fstride]);
 332          C_MUL(scratch[2] ,*Fout2, tw[2*u*fstride]);
 333          C_MUL(scratch[3] ,*Fout3, tw[3*u*fstride]);
 334          C_MUL(scratch[4] ,*Fout4, tw[4*u*fstride]);
 335
 336          C_ADD( scratch[7],scratch[1],scratch[4]);
 337          C_SUB( scratch[10],scratch[1],scratch[4]);
 338          C_ADD( scratch[8],scratch[2],scratch[3]);
 339          C_SUB( scratch[9],scratch[2],scratch[3]);
 340
 341          Fout0->r += scratch[7].r + scratch[8].r;
 342          Fout0->i += scratch[7].i + scratch[8].i;
 343
 344          scratch[5].r = scratch[0].r + S_MUL(scratch[7].r,ya.r) + S_MUL(scratch[8].r,yb.r);
 345          scratch[5].i = scratch[0].i + S_MUL(scratch[7].i,ya.r) + S_MUL(scratch[8].i,yb.r);
 346
 347          scratch[6].r =  S_MUL(scratch[10].i,ya.i) + S_MUL(scratch[9].i,yb.i);
 348          scratch[6].i = -S_MUL(scratch[10].r,ya.i) - S_MUL(scratch[9].r,yb.i);
 349
 350          C_SUB(*Fout1,scratch[5],scratch[6]);
 351          C_ADD(*Fout4,scratch[5],scratch[6]);
 352
 353          scratch[11].r = scratch[0].r + S_MUL(scratch[7].r,yb.r) + S_MUL(scratch[8].r,ya.r);
 354          scratch[11].i = scratch[0].i + S_MUL(scratch[7].i,yb.r) + S_MUL(scratch[8].i,ya.r);
 355          scratch[12].r = - S_MUL(scratch[10].i,yb.i) + S_MUL(scratch[9].i,ya.i);
 356          scratch[12].i = S_MUL(scratch[10].r,yb.i) - S_MUL(scratch[9].r,ya.i);
 357
 358          C_ADD(*Fout2,scratch[11],scratch[12]);
 359          C_SUB(*Fout3,scratch[11],scratch[12]);
 360
 361          ++Fout0;++Fout1;++Fout2;++Fout3;++Fout4;
 362       }
 363    }
 364 }
 365
 366 static void ki_bfly5(
 367                      kiss_fft_cpx * Fout,
 368                      const size_t fstride,
 369                      const kiss_fft_state *st,
 370                      int m,
 371                      int N,
 372                      int mm
 373                     )
 374 {
 375    kiss_fft_cpx *Fout0,*Fout1,*Fout2,*Fout3,*Fout4;
 376    int i, u;
 377    kiss_fft_cpx scratch[13];
 378    const kiss_twiddle_cpx * twiddles = st->twiddles;
 379    const kiss_twiddle_cpx *tw;
 380    kiss_twiddle_cpx ya,yb;
 381    kiss_fft_cpx * Fout_beg = Fout;
 382
 383    ya = twiddles[fstride*m];
 384    yb = twiddles[fstride*2*m];
 385    tw=st->twiddles;
 386
 387    for (i=0;i<N;i++)
 388    {
 389       Fout = Fout_beg + i*mm;
 390       Fout0=Fout;
 391       Fout1=Fout0+m;
 392       Fout2=Fout0+2*m;
 393       Fout3=Fout0+3*m;
 394       Fout4=Fout0+4*m;
 395
 396       for ( u=0; u<m; ++u ) {
 397          scratch[0] = *Fout0;
 398
 399          C_MULC(scratch[1] ,*Fout1, tw[u*fstride]);
 400          C_MULC(scratch[2] ,*Fout2, tw[2*u*fstride]);
 401          C_MULC(scratch[3] ,*Fout3, tw[3*u*fstride]);
 402          C_MULC(scratch[4] ,*Fout4, tw[4*u*fstride]);
 403
 404          C_ADD( scratch[7],scratch[1],scratch[4]);
 405          C_SUB( scratch[10],scratch[1],scratch[4]);
 406          C_ADD( scratch[8],scratch[2],scratch[3]);
 407          C_SUB( scratch[9],scratch[2],scratch[3]);
 408
 409          Fout0->r += scratch[7].r + scratch[8].r;
 410          Fout0->i += scratch[7].i + scratch[8].i;
 411
 412          scratch[5].r = scratch[0].r + S_MUL(scratch[7].r,ya.r) + S_MUL(scratch[8].r,yb.r);
 413          scratch[5].i = scratch[0].i + S_MUL(scratch[7].i,ya.r) + S_MUL(scratch[8].i,yb.r);
 414
 415          scratch[6].r = -S_MUL(scratch[10].i,ya.i) - S_MUL(scratch[9].i,yb.i);
 416          scratch[6].i =  S_MUL(scratch[10].r,ya.i) + S_MUL(scratch[9].r,yb.i);
 417
 418          C_SUB(*Fout1,scratch[5],scratch[6]);
 419          C_ADD(*Fout4,scratch[5],scratch[6]);
 420
 421          scratch[11].r = scratch[0].r + S_MUL(scratch[7].r,yb.r) + S_MUL(scratch[8].r,ya.r);
 422          scratch[11].i = scratch[0].i + S_MUL(scratch[7].i,yb.r) + S_MUL(scratch[8].i,ya.r);
 423          scratch[12].r =  S_MUL(scratch[10].i,yb.i) - S_MUL(scratch[9].i,ya.i);
 424          scratch[12].i = -S_MUL(scratch[10].r,yb.i) + S_MUL(scratch[9].r,ya.i);
 425
 426          C_ADD(*Fout2,scratch[11],scratch[12]);
 427          C_SUB(*Fout3,scratch[11],scratch[12]);
 428
 429          ++Fout0;++Fout1;++Fout2;++Fout3;++Fout4;
 430       }
 431    }
 432 }
 433
 434 #endif
 435
 436 static
 437 void compute_bitrev_table(
 438          int Fout,
 439          celt_int16 *f,
 440          const size_t fstride,
 441          int in_stride,
 442          celt_int16 * factors,
 443          const kiss_fft_state *st
 444             )
 445 {
 446    const int p=*factors++; /* the radix  */
 447    const int m=*factors++; /* stage's fft length/p */
 448
 449     /*printf ("fft %d %d %d %d %d %d\n", p*m, m, p, s2, fstride*in_stride, N);*/
 450    if (m==1)
 451    {
 452       int j;
 453       for (j=0;j<p;j++)
 454       {
 455          *f = Fout+j;
 456          f += fstride*in_stride;
 457       }
 458    } else {
 459       int j;
 460       for (j=0;j<p;j++)
 461       {
 462          compute_bitrev_table( Fout , f, fstride*p, in_stride, factors,st);
 463          f += fstride*in_stride;
 464          Fout += m;
 465       }
 466    }
 467 }
 468
 469
 470 static void kf_work(
 471         kiss_fft_cpx * Fout,
 472         const kiss_fft_cpx * f,
 473         size_t fstride,
 474         int in_stride,
 475         const celt_int16 * factors,
 476         const kiss_fft_state *st,
 477         int N,
 478         int s2,
 479         int m2
 480         )
 481 {
 482     const int p=*factors++; /* the radix  */
 483     const int m=*factors++; /* stage's fft length/p */
 484     /*printf ("fft %d %d %d %d %d %d %d\n", p*m, m, p, s2, fstride*in_stride, N, m2);*/
 485     if (m!=1)
 486         kf_work( Fout , f, fstride*p, in_stride, factors,st, N*p, fstride*in_stride, m);
 487
 488     /* Compensate for longer twiddles table (when sharing) */
 489     if (st->shift>0)
 490        fstride <<= st->shift;
 491     switch (p) {
 492         case 2: kf_bfly2(Fout,fstride,st,m, N, m2); break;
 493         case 4: kf_bfly4(Fout,fstride,st,m, N, m2); break;
 494 #ifndef RADIX_TWO_ONLY
 495         case 3: kf_bfly3(Fout,fstride,st,m, N, m2); break;
 496         case 5: kf_bfly5(Fout,fstride,st,m, N, m2); break;
 497 #else
 498        default: celt_fatal("kiss_fft: only powers of two enabled");
 499 #endif
 500     }
 501 }
 502
 503
 504 static void ki_work(
 505              kiss_fft_cpx * Fout,
 506              const kiss_fft_cpx * f,
 507              size_t fstride,
 508              int in_stride,
 509              const celt_int16 * factors,
 510              const kiss_fft_state *st,
 511              int N,
 512              int s2,
 513              int m2
 514             )
 515 {
 516    const int p=*factors++; /* the radix  */
 517    const int m=*factors++; /* stage's fft length/p */
 518    /*printf ("fft %d %d %d %d %d %d %d\n", p*m, m, p, s2, fstride*in_stride, N, m2);*/
 519    if (m!=1)
 520       ki_work( Fout , f, fstride*p, in_stride, factors,st, N*p, fstride*in_stride, m);
 521
 522    /* Compensate for longer twiddles table (when sharing) */
 523    if (st->shift>0)
 524       fstride <<= st->shift;
 525    switch (p) {
 526       case 2: ki_bfly2(Fout,fstride,st,m, N, m2); break;
 527       case 4: ki_bfly4(Fout,fstride,st,m, N, m2); break;
 528 #ifndef RADIX_TWO_ONLY
 529       case 3: ki_bfly3(Fout,fstride,st,m, N, m2); break;
 530       case 5: ki_bfly5(Fout,fstride,st,m, N, m2); break;
 531 #else
 532       default: celt_fatal("kiss_fft: only powers of two enabled");
 533 #endif
 534    }
 535 }
 536
 537 /*  facbuf is populated by p1,m1,p2,m2, ...
 538     where
 539     p[i] * m[i] = m[i-1]
 540     m0 = n                  */
 541 static
 542 int kf_factor(int n,celt_int16 * facbuf)
 543 {
 544     int p=4;
 545
 546     /*factor out powers of 4, powers of 2, then any remaining primes */
 547     do {
 548         while (n % p) {
 549             switch (p) {
 550                 case 4: p = 2; break;
 551                 case 2: p = 3; break;
 552                 default: p += 2; break;
 553             }
 554             if (p>32000 || (celt_int32)p*(celt_int32)p > n)
 555                 p = n;          /* no more factors, skip to end */
 556         }
 557         n /= p;
 558         if (p>5)
 559         {
 560            celt_warning("Only powers of 2, 3 and 5 are supported");
 561            return 0;
 562         }
 563         *facbuf++ = p;
 564         *facbuf++ = n;
 565     } while (n > 1);
 566     return 1;
 567 }
 568
 569 static void compute_twiddles(kiss_twiddle_cpx *twiddles, int nfft)
 570 {
 571    int i;
 572 #ifdef FIXED_POINT
 573    for (i=0;i<nfft;++i) {
 574       celt_word32 phase = -i;
 575       kf_cexp2(twiddles+i, DIV32(SHL32(phase,17),nfft));
 576    }
 577 #else
 578    for (i=0;i<nfft;++i) {
 579       const double pi=3.14159265358979323846264338327;
 580       double phase = ( -2*pi /nfft ) * i;
 581       kf_cexp(twiddles+i, phase );
 582    }
 583 #endif
 584 }
 585
 586
 587 /*
 588  *
 589  * User-callable function to allocate all necessary storage space for the fft.
 590  *
 591  * The return value is a contiguous block of memory, allocated with malloc.  As such,
 592  * It can be freed with free(), rather than a kiss_fft-specific function.
 593  * */
 594 kiss_fft_state *kiss_fft_alloc_twiddles(int nfft,void * mem,size_t * lenmem,  const kiss_fft_state *base)
 595 {
 596     kiss_fft_state *st=NULL;
 597     size_t memneeded = sizeof(struct kiss_fft_state); /* twiddle factors*/
 598
 599     if ( lenmem==NULL ) {
 600         st = ( kiss_fft_state*)KISS_FFT_MALLOC( memneeded );
 601     }else{
 602         if (mem != NULL && *lenmem >= memneeded)
 603             st = (kiss_fft_state*)mem;
 604         *lenmem = memneeded;
 605     }
 606     if (st) {
 607         celt_int16 *bitrev;
 608         kiss_twiddle_cpx *twiddles;
 609
 610         st->nfft=nfft;
 611 #ifndef FIXED_POINT
 612         st->scale = 1./nfft;
 613 #endif
 614         if (base != NULL)
 615         {
 616            st->twiddles = base->twiddles;
 617            st->shift = 0;
 618            while (nfft<<st->shift != base->nfft && st->shift < 32)
 619               st->shift++;
 620            /* FIXME: Report error and do proper cleanup */
 621            if (st->shift>=32)
 622               return NULL;
 623         } else {
 624            st->twiddles = twiddles = (kiss_twiddle_cpx*)KISS_FFT_MALLOC(sizeof(kiss_twiddle_cpx)*nfft);
 625            compute_twiddles(twiddles, nfft);
 626            st->shift = -1;
 627         }
 628         if (!kf_factor(nfft,st->factors))
 629         {
 630            kiss_fft_free(st);
 631            return NULL;
 632         }
 633
 634         /* bitrev */
 635         st->bitrev = bitrev = (celt_int16*)KISS_FFT_MALLOC(sizeof(celt_int16)*nfft);
 636         compute_bitrev_table(0, bitrev, 1,1, st->factors,st);
 637     }
 638     return st;
 639 }
 640
 641 kiss_fft_state *kiss_fft_alloc(int nfft,void * mem,size_t * lenmem )
 642 {
 643    return kiss_fft_alloc_twiddles(nfft, mem, lenmem, NULL);
 644 }
 645
 646
 647 static void kiss_fft_stride(const kiss_fft_state *st,const kiss_fft_cpx *fin,kiss_fft_cpx *fout,int in_stride)
 648 {
 649     if (fin == fout)
 650     {
 651        celt_fatal("In-place FFT not supported");
 652     } else {
 653        /* Bit-reverse the input */
 654        int i;
 655        for (i=0;i<st->nfft;i++)
 656        {
 657           fout[st->bitrev[i]] = fin[i];
 658 #ifndef FIXED_POINT
 659           fout[st->bitrev[i]].r *= st->scale;
 660           fout[st->bitrev[i]].i *= st->scale;
 661 #endif
 662        }
 663        kf_work( fout, fin, 1,in_stride, st->factors,st, 1, in_stride, 1);
 664     }
 665 }
 666
 667 void kiss_fft(const kiss_fft_state *cfg,const kiss_fft_cpx *fin,kiss_fft_cpx *fout)
 668 {
 669     kiss_fft_stride(cfg,fin,fout,1);
 670 }
 671
 672 static void kiss_ifft_stride(const kiss_fft_state *st,const kiss_fft_cpx *fin,kiss_fft_cpx *fout,int in_stride)
 673 {
 674    if (fin == fout)
 675    {
 676       celt_fatal("In-place FFT not supported");
 677    } else {
 678       /* Bit-reverse the input */
 679       int i;
 680       for (i=0;i<st->nfft;i++)
 681          fout[st->bitrev[i]] = fin[i];
 682       ki_work( fout, fin, 1,in_stride, st->factors,st, 1, in_stride, 1);
 683    }
 684 }
 685
 686 void kiss_ifft(const kiss_fft_state *cfg,const kiss_fft_cpx *fin,kiss_fft_cpx *fout)
 687 {
 688    kiss_ifft_stride(cfg,fin,fout,1);
 689 }
 690
 691 void kiss_fft_free(const kiss_fft_state *cfg)
 692 {
 693    celt_free((celt_int16*)cfg->bitrev);
 694    if (cfg->shift < 0)
 695       celt_free((kiss_twiddle_cpx*)cfg->twiddles);
 696    celt_free((kiss_fft_state*)cfg);
 697 }