fft.h 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175
  1. /*
  2. * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
  3. * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
  4. *
  5. * This file is part of FFmpeg.
  6. *
  7. * FFmpeg is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU Lesser General Public
  9. * License as published by the Free Software Foundation; either
  10. * version 2.1 of the License, or (at your option) any later version.
  11. *
  12. * FFmpeg is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with FFmpeg; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. #ifndef AVCODEC_FFT_H
  22. #define AVCODEC_FFT_H
  23. #ifndef FFT_FLOAT
  24. #define FFT_FLOAT 1
  25. #endif
  26. #ifndef FFT_FIXED_32
  27. #define FFT_FIXED_32 0
  28. #endif
  29. #include <stdint.h>
  30. #include "config.h"
  31. #include "libavutil/mem.h"
  32. #if FFT_FLOAT
  33. #include "avfft.h"
  34. #define FFT_NAME(x) x
  35. typedef float FFTDouble;
  36. #else
  37. #if FFT_FIXED_32
  38. #define Q31(x) (int)((x)*2147483648.0 + 0.5)
  39. #define FFT_NAME(x) x ## _fixed_32
  40. typedef int32_t FFTSample;
  41. #else /* FFT_FIXED_32 */
  42. #define FFT_NAME(x) x ## _fixed
  43. typedef int16_t FFTSample;
  44. #endif /* FFT_FIXED_32 */
  45. typedef struct FFTComplex {
  46. FFTSample re, im;
  47. } FFTComplex;
  48. typedef int FFTDouble;
  49. typedef struct FFTContext FFTContext;
  50. #endif /* FFT_FLOAT */
  51. typedef struct FFTDComplex {
  52. FFTDouble re, im;
  53. } FFTDComplex;
  54. /* FFT computation */
  55. enum fft_permutation_type {
  56. FF_FFT_PERM_DEFAULT,
  57. FF_FFT_PERM_SWAP_LSBS,
  58. FF_FFT_PERM_AVX,
  59. };
  60. enum mdct_permutation_type {
  61. FF_MDCT_PERM_NONE,
  62. FF_MDCT_PERM_INTERLEAVE,
  63. };
  64. struct FFTContext {
  65. int nbits;
  66. int inverse;
  67. uint16_t *revtab;
  68. FFTComplex *tmp_buf;
  69. int mdct_size; /* size of MDCT (i.e. number of input data * 2) */
  70. int mdct_bits; /* n = 2^nbits */
  71. /* pre/post rotation tables */
  72. FFTSample *tcos;
  73. FFTSample *tsin;
  74. /**
  75. * Do the permutation needed BEFORE calling fft_calc().
  76. */
  77. void (*fft_permute)(struct FFTContext *s, FFTComplex *z);
  78. /**
  79. * Do a complex FFT with the parameters defined in ff_fft_init(). The
  80. * input data must be permuted before. No 1.0/sqrt(n) normalization is done.
  81. */
  82. void (*fft_calc)(struct FFTContext *s, FFTComplex *z);
  83. void (*imdct_calc)(struct FFTContext *s, FFTSample *output, const FFTSample *input);
  84. void (*imdct_half)(struct FFTContext *s, FFTSample *output, const FFTSample *input);
  85. void (*mdct_calc)(struct FFTContext *s, FFTSample *output, const FFTSample *input);
  86. void (*mdct_calcw)(struct FFTContext *s, FFTDouble *output, const FFTSample *input);
  87. enum fft_permutation_type fft_permutation;
  88. enum mdct_permutation_type mdct_permutation;
  89. uint32_t *revtab32;
  90. };
  91. #if CONFIG_HARDCODED_TABLES
  92. #define COSTABLE_CONST const
  93. #else
  94. #define COSTABLE_CONST
  95. #endif
  96. #define COSTABLE(size) \
  97. COSTABLE_CONST DECLARE_ALIGNED(32, FFTSample, FFT_NAME(ff_cos_##size))[size/2]
  98. extern COSTABLE(16);
  99. extern COSTABLE(32);
  100. extern COSTABLE(64);
  101. extern COSTABLE(128);
  102. extern COSTABLE(256);
  103. extern COSTABLE(512);
  104. extern COSTABLE(1024);
  105. extern COSTABLE(2048);
  106. extern COSTABLE(4096);
  107. extern COSTABLE(8192);
  108. extern COSTABLE(16384);
  109. extern COSTABLE(32768);
  110. extern COSTABLE(65536);
  111. extern COSTABLE(131072);
  112. extern COSTABLE_CONST FFTSample* const FFT_NAME(ff_cos_tabs)[18];
  113. #define ff_init_ff_cos_tabs FFT_NAME(ff_init_ff_cos_tabs)
  114. /**
  115. * Initialize the cosine table in ff_cos_tabs[index]
  116. * @param index index in ff_cos_tabs array of the table to initialize
  117. */
  118. void ff_init_ff_cos_tabs(int index);
  119. #define ff_fft_init FFT_NAME(ff_fft_init)
  120. #define ff_fft_end FFT_NAME(ff_fft_end)
  121. /**
  122. * Set up a complex FFT.
  123. * @param nbits log2 of the length of the input array
  124. * @param inverse if 0 perform the forward transform, if 1 perform the inverse
  125. */
  126. int ff_fft_init(FFTContext *s, int nbits, int inverse);
  127. void ff_fft_init_aarch64(FFTContext *s);
  128. void ff_fft_init_x86(FFTContext *s);
  129. void ff_fft_init_arm(FFTContext *s);
  130. void ff_fft_init_mips(FFTContext *s);
  131. void ff_fft_init_ppc(FFTContext *s);
  132. void ff_fft_fixed_init_arm(FFTContext *s);
  133. void ff_fft_end(FFTContext *s);
  134. #define ff_mdct_init FFT_NAME(ff_mdct_init)
  135. #define ff_mdct_end FFT_NAME(ff_mdct_end)
  136. int ff_mdct_init(FFTContext *s, int nbits, int inverse, double scale);
  137. void ff_mdct_end(FFTContext *s);
  138. #endif /* AVCODEC_FFT_H */