ffmpeg/libavcodec/avfft.c

   1 /*
   2  * This file is part of FFmpeg.
   3  *
   4  * FFmpeg is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * FFmpeg is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with FFmpeg; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  17  */
  18
  19 #include "libavutil/attributes.h"
  20 #include "libavutil/mem.h"
  21 #include "avfft.h"
  22 #include "fft.h"
  23 #include "rdft.h"
  24 #include "dct.h"
  25
  26 /* FFT */
  27
  28 FFTContext *av_fft_init(int nbits, int inverse)
  29 {
  30     FFTContext *s = av_mallocz(sizeof(*s));
  31
  32     if (s && ff_fft_init(s, nbits, inverse))
  33         av_freep(&s);
  34
  35     return s;
  36 }
  37
  38 void av_fft_permute(FFTContext *s, FFTComplex *z)
  39 {
  40     s->fft_permute(s, z);
  41 }
  42
  43 void av_fft_calc(FFTContext *s, FFTComplex *z)
  44 {
  45     s->fft_calc(s, z);
  46 }
  47
  48 av_cold void av_fft_end(FFTContext *s)
  49 {
  50     if (s) {
  51         ff_fft_end(s);
  52         av_free(s);
  53     }
  54 }
  55
  56 #if CONFIG_MDCT
  57
  58 FFTContext *av_mdct_init(int nbits, int inverse, double scale)
  59 {
  60     FFTContext *s = av_malloc(sizeof(*s));
  61
  62     if (s && ff_mdct_init(s, nbits, inverse, scale))
  63         av_freep(&s);
  64
  65     return s;
  66 }
  67
  68 void av_imdct_calc(FFTContext *s, FFTSample *output, const FFTSample *input)
  69 {
  70     s->imdct_calc(s, output, input);
  71 }
  72
  73 void av_imdct_half(FFTContext *s, FFTSample *output, const FFTSample *input)
  74 {
  75     s->imdct_half(s, output, input);
  76 }
  77
  78 void av_mdct_calc(FFTContext *s, FFTSample *output, const FFTSample *input)
  79 {
  80     s->mdct_calc(s, output, input);
  81 }
  82
  83 av_cold void av_mdct_end(FFTContext *s)
  84 {
  85     if (s) {
  86         ff_mdct_end(s);
  87         av_free(s);
  88     }
  89 }
  90
  91 #endif /* CONFIG_MDCT */
  92
  93 #if CONFIG_RDFT
  94
  95 RDFTContext *av_rdft_init(int nbits, enum RDFTransformType trans)
  96 {
  97     RDFTContext *s = av_malloc(sizeof(*s));
  98
  99     if (s && ff_rdft_init(s, nbits, trans))
 100         av_freep(&s);
 101
 102     return s;
 103 }
 104
 105 void av_rdft_calc(RDFTContext *s, FFTSample *data)
 106 {
 107     s->rdft_calc(s, data);
 108 }
 109
 110 av_cold void av_rdft_end(RDFTContext *s)
 111 {
 112     if (s) {
 113         ff_rdft_end(s);
 114         av_free(s);
 115     }
 116 }
 117
 118 #endif /* CONFIG_RDFT */
 119
 120 #if CONFIG_DCT
 121
 122 DCTContext *av_dct_init(int nbits, enum DCTTransformType inverse)
 123 {
 124     DCTContext *s = av_malloc(sizeof(*s));
 125
 126     if (s && ff_dct_init(s, nbits, inverse))
 127         av_freep(&s);
 128
 129     return s;
 130 }
 131
 132 void av_dct_calc(DCTContext *s, FFTSample *data)
 133 {
 134     s->dct_calc(s, data);
 135 }
 136
 137 av_cold void av_dct_end(DCTContext *s)
 138 {
 139     if (s) {
 140         ff_dct_end(s);
 141         av_free(s);
 142     }
 143 }
 144
 145 #endif /* CONFIG_DCT */