diff options
Diffstat (limited to 'libavcodec/fft-test.c')
-rw-r--r-- | libavcodec/fft-test.c | 62 |
1 files changed, 34 insertions, 28 deletions
diff --git a/libavcodec/fft-test.c b/libavcodec/fft-test.c index 2f2f2e9..bb28ffc 100644 --- a/libavcodec/fft-test.c +++ b/libavcodec/fft-test.c @@ -1,20 +1,20 @@ /* * (c) 2002 Fabrice Bellard * - * This file is part of Libav. + * This file is part of FFmpeg. * - * Libav is free software; you can redistribute it and/or + * 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. * - * Libav is distributed in the hope that it will be useful, + * 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 Libav; if not, write to the Free Software + * License along with FFmpeg; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ @@ -53,6 +53,10 @@ # define RANGE 1.0 # define REF_SCALE(x, bits) (x) # define FMT "%10.6f" +#elif FFT_FIXED_32 +# define RANGE 8388608 +# define REF_SCALE(x, bits) (x) +# define FMT "%6d" #else # define RANGE 16384 # define REF_SCALE(x, bits) ((x) / (1<<(bits))) @@ -69,7 +73,7 @@ static void fft_ref_init(int nbits, int inverse) double c1, s1, alpha; n = 1 << nbits; - exptab = av_malloc((n / 2) * sizeof(*exptab)); + exptab = av_malloc_array((n / 2), sizeof(*exptab)); for (i = 0; i < (n/2); i++) { alpha = 2 * M_PI * (float)i / (float)n; @@ -150,7 +154,7 @@ static void mdct_ref(FFTSample *output, FFTSample *input, int nbits) #if FFT_FLOAT #if CONFIG_DCT -static void idct_ref(float *output, float *input, int nbits) +static void idct_ref(FFTSample *output, FFTSample *input, int nbits) { int n = 1<<nbits; int k, i; @@ -166,7 +170,7 @@ static void idct_ref(float *output, float *input, int nbits) output[i] = 2 * s / n; } } -static void dct_ref(float *output, float *input, int nbits) +static void dct_ref(FFTSample *output, FFTSample *input, int nbits) { int n = 1<<nbits; int k, i; @@ -208,7 +212,7 @@ static int check_diff(FFTSample *tab1, FFTSample *tab2, int n, double scale) error+= e*e; if(e>max) max= e; } - av_log(NULL, AV_LOG_INFO, "max:%f e:%g\n", max, sqrt(error)/n); + av_log(NULL, AV_LOG_INFO, "max:%f e:%g\n", max, sqrt(error/n)); return err; } @@ -291,19 +295,21 @@ int main(int argc, char **argv) scale = atof(optarg); break; case 'c': - cpuflags = av_parse_cpu_flags(optarg); - if (cpuflags < 0) + cpuflags = av_get_cpu_flags(); + + if (av_parse_cpu_caps(&cpuflags, optarg) < 0) return 1; - av_set_cpu_flags_mask(cpuflags); + + av_force_cpu_flags(cpuflags); break; } } fft_size = 1 << fft_nbits; - tab = av_malloc(fft_size * sizeof(FFTComplex)); - tab1 = av_malloc(fft_size * sizeof(FFTComplex)); - tab_ref = av_malloc(fft_size * sizeof(FFTComplex)); - tab2 = av_malloc(fft_size * sizeof(FFTSample)); + tab = av_malloc_array(fft_size, sizeof(FFTComplex)); + tab1 = av_malloc_array(fft_size, sizeof(FFTComplex)); + tab_ref = av_malloc_array(fft_size, sizeof(FFTComplex)); + tab2 = av_malloc_array(fft_size, sizeof(FFTSample)); switch (transform) { #if CONFIG_MDCT @@ -325,7 +331,7 @@ int main(int argc, char **argv) fft_ref_init(fft_nbits, do_inverse); break; #if FFT_FLOAT -#if CONFIG_RDFT +# if CONFIG_RDFT case TRANSFORM_RDFT: if (do_inverse) av_log(NULL, AV_LOG_INFO,"IDFT_C2R"); @@ -334,8 +340,8 @@ int main(int argc, char **argv) ff_rdft_init(r, fft_nbits, do_inverse ? IDFT_C2R : DFT_R2C); fft_ref_init(fft_nbits, do_inverse); break; -#endif /* CONFIG_RDFT */ -#if CONFIG_DCT +# endif /* CONFIG_RDFT */ +# if CONFIG_DCT case TRANSFORM_DCT: if (do_inverse) av_log(NULL, AV_LOG_INFO,"DCT_III"); @@ -343,7 +349,7 @@ int main(int argc, char **argv) av_log(NULL, AV_LOG_INFO,"DCT_II"); ff_dct_init(d, fft_nbits, do_inverse ? DCT_III : DCT_II); break; -#endif /* CONFIG_DCT */ +# endif /* CONFIG_DCT */ #endif /* FFT_FLOAT */ default: av_log(NULL, AV_LOG_ERROR, "Requested transform not supported\n"); @@ -422,11 +428,11 @@ int main(int argc, char **argv) #if CONFIG_DCT case TRANSFORM_DCT: memcpy(tab, tab1, fft_size * sizeof(FFTComplex)); - d->dct_calc(d, tab); + d->dct_calc(d, (FFTSample *)tab); if (do_inverse) { - idct_ref(tab_ref, tab1, fft_nbits); + idct_ref((FFTSample*)tab_ref, (FFTSample *)tab1, fft_nbits); } else { - dct_ref(tab_ref, tab1, fft_nbits); + dct_ref((FFTSample*)tab_ref, (FFTSample *)tab1, fft_nbits); } err = check_diff((float *)tab_ref, (float *)tab, fft_size, 1.0); break; @@ -444,7 +450,7 @@ int main(int argc, char **argv) /* we measure during about 1 seconds */ nb_its = 1; for(;;) { - time_start = av_gettime(); + time_start = av_gettime_relative(); for (it = 0; it < nb_its; it++) { switch (transform) { case TRANSFORM_MDCT: @@ -470,7 +476,7 @@ int main(int argc, char **argv) #endif /* FFT_FLOAT */ } } - duration = av_gettime() - time_start; + duration = av_gettime_relative() - time_start; if (duration >= 1000000) break; nb_its *= 2; @@ -491,16 +497,16 @@ int main(int argc, char **argv) ff_fft_end(s); break; #if FFT_FLOAT -#if CONFIG_RDFT +# if CONFIG_RDFT case TRANSFORM_RDFT: ff_rdft_end(r); break; -#endif /* CONFIG_RDFT */ -#if CONFIG_DCT +# endif /* CONFIG_RDFT */ +# if CONFIG_DCT case TRANSFORM_DCT: ff_dct_end(d); break; -#endif /* CONFIG_DCT */ +# endif /* CONFIG_DCT */ #endif /* FFT_FLOAT */ } |