avfft.c 3.69 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
/*
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

19
#include "libavutil/attributes.h"
20 21 22
#include "libavutil/mem.h"
#include "avfft.h"
#include "fft.h"
23 24
#include "rdft.h"
#include "dct.h"
25 26 27 28 29

/* FFT */

FFTContext *av_fft_init(int nbits, int inverse)
{
30
    FFTContext *s = av_mallocz(sizeof(*s));
31

32
    if (s && ff_fft_init(s, nbits, inverse))
33
        av_freep(&s);
34 35 36 37 38 39 40 41 42 43 44 45 46 47

    return s;
}

void av_fft_permute(FFTContext *s, FFTComplex *z)
{
    s->fft_permute(s, z);
}

void av_fft_calc(FFTContext *s, FFTComplex *z)
{
    s->fft_calc(s, z);
}

48
av_cold void av_fft_end(FFTContext *s)
49 50 51 52 53 54 55 56 57 58 59 60 61
{
    if (s) {
        ff_fft_end(s);
        av_free(s);
    }
}

#if CONFIG_MDCT

FFTContext *av_mdct_init(int nbits, int inverse, double scale)
{
    FFTContext *s = av_malloc(sizeof(*s));

62
    if (s && ff_mdct_init(s, nbits, inverse, scale))
63
        av_freep(&s);
64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82

    return s;
}

void av_imdct_calc(FFTContext *s, FFTSample *output, const FFTSample *input)
{
    s->imdct_calc(s, output, input);
}

void av_imdct_half(FFTContext *s, FFTSample *output, const FFTSample *input)
{
    s->imdct_half(s, output, input);
}

void av_mdct_calc(FFTContext *s, FFTSample *output, const FFTSample *input)
{
    s->mdct_calc(s, output, input);
}

83
av_cold void av_mdct_end(FFTContext *s)
84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
{
    if (s) {
        ff_mdct_end(s);
        av_free(s);
    }
}

#endif /* CONFIG_MDCT */

#if CONFIG_RDFT

RDFTContext *av_rdft_init(int nbits, enum RDFTransformType trans)
{
    RDFTContext *s = av_malloc(sizeof(*s));

99
    if (s && ff_rdft_init(s, nbits, trans))
100
        av_freep(&s);
101 102 103 104 105 106

    return s;
}

void av_rdft_calc(RDFTContext *s, FFTSample *data)
{
107
    s->rdft_calc(s, data);
108 109
}

110
av_cold void av_rdft_end(RDFTContext *s)
111 112 113 114 115 116 117 118 119 120 121
{
    if (s) {
        ff_rdft_end(s);
        av_free(s);
    }
}

#endif /* CONFIG_RDFT */

#if CONFIG_DCT

122
DCTContext *av_dct_init(int nbits, enum DCTTransformType inverse)
123 124 125
{
    DCTContext *s = av_malloc(sizeof(*s));

126
    if (s && ff_dct_init(s, nbits, inverse))
127
        av_freep(&s);
128 129 130 131 132 133

    return s;
}

void av_dct_calc(DCTContext *s, FFTSample *data)
{
134
    s->dct_calc(s, data);
135 136
}

137
av_cold void av_dct_end(DCTContext *s)
138 139 140 141 142 143 144
{
    if (s) {
        ff_dct_end(s);
        av_free(s);
    }
}

145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178
#ifdef TEST
int main(int argc, char **argv)
{
    int i;
#define LEN 1024
    FFTSample *ref  = av_malloc_array(LEN, sizeof(*ref));
    FFTSample *data = av_malloc_array(LEN, sizeof(*data));
    RDFTContext *rdft_context  = av_rdft_init(10, DFT_R2C);
    RDFTContext *irdft_context = av_rdft_init(10, IDFT_C2R);

    if (!ref || !data || !rdft_context || !irdft_context)
        return 2;
    for (i=0; i<LEN; i++) {
        ref[i] = data[i] = i*456 + 123 + i*i;
    }
    av_rdft_calc(rdft_context, data);
    av_rdft_calc(irdft_context, data);

    for (i=0; i<LEN; i++) {
        if (fabs(ref[i] - data[i]/LEN*2) > 1) {
            fprintf(stderr, "Failed at %d (%f %f)\n", i, ref[i], data[i]/LEN*2);
            return 1;
        }
    }

    av_rdft_end(rdft_context);
    av_rdft_end(irdft_context);
    av_free(data);
    av_free(ref);

    return 0;
}
#endif

179
#endif /* CONFIG_DCT */