1/* 2 * This file is part of FFmpeg. 3 * 4 * FFmpeg is free software; you can redistribute it and/or modify 5 * it under the terms of the GNU General Public License as published by 6 * the Free Software Foundation; either version 2 of the License, or 7 * (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 12 * GNU General Public License for more details. 13 * 14 * You should have received a copy of the GNU General Public License along 15 * with FFmpeg; if not, write to the Free Software Foundation, Inc., 16 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. 17 */ 18 19#include "libavutil/mem_internal.h" 20#include "libavutil/tx.h" 21#include "libavutil/error.h" 22 23#include "checkasm.h" 24 25#include <stdlib.h> 26 27#define EPS 0.00005 28 29#define SCALE_NOOP(x) (x) 30#define SCALE_INT20(x) (av_clip64(lrintf((x) * 2147483648.0), INT32_MIN, INT32_MAX) >> 12) 31 32#define randomize_complex(BUF, LEN, TYPE, SCALE) \ 33 do { \ 34 TYPE *buf = (TYPE *)BUF; \ 35 for (int i = 0; i < LEN; i++) { \ 36 double fre = (double)rnd() / UINT_MAX; \ 37 double fim = (double)rnd() / UINT_MAX; \ 38 buf[i] = (TYPE){ SCALE(fre), SCALE(fim) }; \ 39 } \ 40 } while (0) 41 42static const int check_lens[] = { 43 2, 4, 8, 16, 32, 64, 1024, 16384, 44}; 45 46static AVTXContext *tx_refs[6 /*AVTXType*/][FF_ARRAY_ELEMS(check_lens)]; 47static int init = 0; 48 49static void free_tx_refs(void) 50{ 51 for (int i = 0; i < FF_ARRAY_ELEMS(tx_refs); i++) 52 for (int j = 0; j < FF_ARRAY_ELEMS(*tx_refs); j++) 53 av_tx_uninit(&tx_refs[i][j]); 54} 55 56#define CHECK_TEMPLATE(PREFIX, TYPE, DATA_TYPE, SCALE, LENGTHS, CHECK_EXPRESSION) \ 57 do { \ 58 int err; \ 59 AVTXContext *tx; \ 60 av_tx_fn fn; \ 61 int num_checks = 0; \ 62 int last_check = 0; \ 63 const void *scale = &SCALE; \ 64 \ 65 for (int i = 0; i < FF_ARRAY_ELEMS(LENGTHS); i++) { \ 66 int len = LENGTHS[i]; \ 67 \ 68 if ((err = av_tx_init(&tx, &fn, TYPE, 0, len, &scale, 0x0)) < 0) { \ 69 fprintf(stderr, "av_tx: %s\n", av_err2str(err)); \ 70 return; \ 71 } \ 72 \ 73 if (check_func(fn, PREFIX "_%i", len)) { \ 74 AVTXContext *tx_ref = tx_refs[TYPE][i]; \ 75 if (!tx_ref) \ 76 tx_ref = tx; \ 77 num_checks++; \ 78 last_check = len; \ 79 call_ref(tx_ref, out_ref, in, sizeof(DATA_TYPE)); \ 80 call_new(tx, out_new, in, sizeof(DATA_TYPE)); \ 81 if (CHECK_EXPRESSION) { \ 82 fail(); \ 83 av_tx_uninit(&tx); \ 84 break; \ 85 } \ 86 bench_new(tx, out_new, in, sizeof(DATA_TYPE)); \ 87 av_tx_uninit(&tx_refs[TYPE][i]); \ 88 tx_refs[TYPE][i] = tx; \ 89 } else { \ 90 av_tx_uninit(&tx); \ 91 } \ 92 } \ 93 \ 94 if (num_checks == 1) \ 95 report(PREFIX "_%i", last_check); \ 96 else if (num_checks) \ 97 report(PREFIX); \ 98 } while (0) 99 100void checkasm_check_av_tx(void) 101{ 102 const float scale_float = 1.0f; 103 const double scale_double = 1.0f; 104 105 declare_func(void, AVTXContext *tx, void *out, void *in, ptrdiff_t stride); 106 107 void *in = av_malloc(16384*2*8); 108 void *out_ref = av_malloc(16384*2*8); 109 void *out_new = av_malloc(16384*2*8); 110 111 randomize_complex(in, 16384, AVComplexFloat, SCALE_NOOP); 112 CHECK_TEMPLATE("float_fft", AV_TX_FLOAT_FFT, AVComplexFloat, scale_float, check_lens, 113 !float_near_abs_eps_array(out_ref, out_new, EPS, len*2)); 114 115 randomize_complex(in, 16384, AVComplexDouble, SCALE_NOOP); 116 CHECK_TEMPLATE("double_fft", AV_TX_DOUBLE_FFT, AVComplexDouble, scale_double, check_lens, 117 !double_near_abs_eps_array(out_ref, out_new, EPS, len*2)); 118 119 av_free(in); 120 av_free(out_ref); 121 av_free(out_new); 122 123 if (!init) { 124 init = 1; 125 atexit(free_tx_refs); 126 } 127} 128