FFmpeg
av_tx.c
Go to the documentation of this file.
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.0005
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 
42 static const int check_lens[] = {
43  2, 4, 8, 16, 32, 64, 120, 960, 1024, 1920, 16384,
44 };
45 
46 static AVTXContext *tx_refs[AV_TX_NB][2 /* Direction */][FF_ARRAY_ELEMS(check_lens)] = { 0 };
47 static int init = 0;
48 
49 static 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  for (int k = 0; k < FF_ARRAY_ELEMS(**tx_refs); k++)
54  av_tx_uninit(&tx_refs[i][j][k]);
55 }
56 
57 #define CHECK_TEMPLATE(PREFIX, TYPE, DIR, DATA_TYPE, SCALE_TYPE, LENGTHS, CHECK_EXPRESSION) \
58  do { \
59  int err; \
60  AVTXContext *tx; \
61  av_tx_fn fn; \
62  int num_checks = 0; \
63  int last_check = 0; \
64  \
65  for (int i = 0; i < FF_ARRAY_ELEMS(LENGTHS); i++) { \
66  int len = LENGTHS[i]; \
67  const SCALE_TYPE scale = 1.0 / len; \
68  \
69  if ((err = av_tx_init(&tx, &fn, TYPE, DIR, len, &scale, 0x0)) < 0) { \
70  fprintf(stderr, "av_tx: %s\n", av_err2str(err)); \
71  return; \
72  } \
73  \
74  if (check_func(fn, PREFIX "_%i", len)) { \
75  AVTXContext *tx_ref = tx_refs[TYPE][DIR][i]; \
76  if (!tx_ref) \
77  tx_ref = tx; \
78  num_checks++; \
79  last_check = len; \
80  call_ref(tx_ref, out_ref, in, sizeof(DATA_TYPE)); \
81  call_new(tx, out_new, in, sizeof(DATA_TYPE)); \
82  if (CHECK_EXPRESSION) { \
83  fail(); \
84  av_tx_uninit(&tx); \
85  break; \
86  } \
87  bench_new(tx, out_new, in, sizeof(DATA_TYPE)); \
88  av_tx_uninit(&tx_refs[TYPE][DIR][i]); \
89  tx_refs[TYPE][DIR][i] = tx; \
90  } else { \
91  av_tx_uninit(&tx); \
92  } \
93  } \
94  \
95  if (num_checks == 1) \
96  report(PREFIX "_%i", last_check); \
97  else if (num_checks) \
98  report(PREFIX); \
99  } while (0)
100 
102 {
103  declare_func(void, AVTXContext *tx, void *out, void *in, ptrdiff_t stride);
104 
105  void *in = av_malloc(16384*2*8);
106  void *out_ref = av_malloc(16384*2*8);
107  void *out_new = av_malloc(16384*2*8);
108 
110  CHECK_TEMPLATE("float_fft", AV_TX_FLOAT_FFT, 0, AVComplexFloat, float, check_lens,
111  !float_near_abs_eps_array(out_ref, out_new, EPS, len*2));
112 
113  CHECK_TEMPLATE("float_imdct", AV_TX_FLOAT_MDCT, 1, float, float, check_lens,
114  !float_near_abs_eps_array(out_ref, out_new, EPS, len));
115 
117  CHECK_TEMPLATE("double_fft", AV_TX_DOUBLE_FFT, 0, AVComplexDouble, double, check_lens,
118  !double_near_abs_eps_array(out_ref, out_new, EPS, len*2));
119 
120  av_free(in);
121  av_free(out_ref);
122  av_free(out_new);
123 
124  if (!init) {
125  init = 1;
126  atexit(free_tx_refs);
127  }
128 }
tx_refs
static AVTXContext * tx_refs[AV_TX_NB][2][FF_ARRAY_ELEMS(check_lens)]
Definition: av_tx.c:46
mem_internal.h
out
FILE * out
Definition: movenc.c:54
float_near_abs_eps_array
int float_near_abs_eps_array(const float *a, const float *b, float eps, unsigned len)
Definition: checkasm.c:399
AVTXContext
Definition: tx_priv.h:235
AV_TX_DOUBLE_FFT
@ AV_TX_DOUBLE_FFT
Definition: tx.h:48
AVComplexFloat
Definition: tx.h:27
av_malloc
#define av_malloc(s)
Definition: tableprint_vlc.h:30
init
static int init
Definition: av_tx.c:47
checkasm.h
free_tx_refs
static void free_tx_refs(void)
Definition: av_tx.c:49
FF_ARRAY_ELEMS
#define FF_ARRAY_ELEMS(a)
Definition: sinewin_tablegen.c:29
AV_TX_NB
@ AV_TX_NB
Definition: tx.h:133
AV_TX_FLOAT_MDCT
@ AV_TX_FLOAT_MDCT
Standard MDCT with a sample data type of float, double or int32_t, respecively.
Definition: tx.h:68
AV_TX_FLOAT_FFT
@ AV_TX_FLOAT_FFT
Standard complex to complex FFT with sample data type of AVComplexFloat, AVComplexDouble or AVComplex...
Definition: tx.h:47
double_near_abs_eps_array
int double_near_abs_eps_array(const double *a, const double *b, double eps, unsigned len)
Definition: checkasm.c:435
CHECK_TEMPLATE
#define CHECK_TEMPLATE(PREFIX, TYPE, DIR, DATA_TYPE, SCALE_TYPE, LENGTHS, CHECK_EXPRESSION)
Definition: av_tx.c:57
check_lens
static const int check_lens[]
Definition: av_tx.c:42
error.h
EPS
#define EPS
Definition: av_tx.c:27
av_tx_uninit
av_cold void av_tx_uninit(AVTXContext **ctx)
Frees a context and sets *ctx to NULL, does nothing when *ctx == NULL.
Definition: tx.c:294
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:255
SCALE_NOOP
#define SCALE_NOOP(x)
Definition: av_tx.c:29
len
int len
Definition: vorbis_enc_data.h:426
AVComplexDouble
Definition: tx.h:31
stride
#define stride
Definition: h264pred_template.c:537
checkasm_check_av_tx
void checkasm_check_av_tx(void)
Definition: av_tx.c:101
randomize_complex
#define randomize_complex(BUF, LEN, TYPE, SCALE)
Definition: av_tx.c:32
declare_func
#define declare_func(ret,...)
Definition: checkasm.h:174
av_free
#define av_free(p)
Definition: tableprint_vlc.h:33
tx.h