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.h"
20 #include "libavutil/mem_internal.h"
21 #include "libavutil/tx.h"
22 #include "libavutil/error.h"
23 
24 #include "checkasm.h"
25 
26 #include <stdlib.h>
27 
28 #define EPS 0.0005
29 
30 #define SCALE_NOOP(x) (x)
31 #define SCALE_INT20(x) (av_clip64(lrintf((x) * 2147483648.0), INT32_MIN, INT32_MAX) >> 12)
32 
33 #define randomize_complex(BUF, LEN, TYPE, SCALE) \
34  do { \
35  TYPE *buf = (TYPE *)BUF; \
36  for (int i = 0; i < LEN; i++) { \
37  double fre = (double)rnd() / UINT_MAX; \
38  double fim = (double)rnd() / UINT_MAX; \
39  buf[i] = (TYPE){ SCALE(fre), SCALE(fim) }; \
40  } \
41  } while (0)
42 
43 static const int check_lens[] = {
44  2, 4, 8, 16, 32, 64, 120, 960, 1024, 1920, 16384,
45 };
46 
47 static AVTXContext *tx_refs[AV_TX_NB][2 /* Direction */][FF_ARRAY_ELEMS(check_lens)] = { 0 };
48 static int init = 0;
49 
50 static void free_tx_refs(void)
51 {
52  for (int i = 0; i < FF_ARRAY_ELEMS(tx_refs); i++)
53  for (int j = 0; j < FF_ARRAY_ELEMS(*tx_refs); j++)
54  for (int k = 0; k < FF_ARRAY_ELEMS(**tx_refs); k++)
55  av_tx_uninit(&tx_refs[i][j][k]);
56 }
57 
58 #define CHECK_TEMPLATE(PREFIX, TYPE, DIR, DATA_TYPE, SCALE_TYPE, LENGTHS, CHECK_EXPRESSION) \
59  do { \
60  int err; \
61  AVTXContext *tx; \
62  av_tx_fn fn; \
63  int num_checks = 0; \
64  int last_check = 0; \
65  \
66  for (int i = 0; i < FF_ARRAY_ELEMS(LENGTHS); i++) { \
67  int len = LENGTHS[i]; \
68  const SCALE_TYPE scale = 1.0 / len; \
69  \
70  if ((err = av_tx_init(&tx, &fn, TYPE, DIR, len, &scale, 0x0)) < 0) { \
71  fprintf(stderr, "av_tx: %s\n", av_err2str(err)); \
72  return; \
73  } \
74  \
75  if (check_func(fn, PREFIX "_%i", len)) { \
76  AVTXContext *tx_ref = tx_refs[TYPE][DIR][i]; \
77  if (!tx_ref) \
78  tx_ref = tx; \
79  num_checks++; \
80  last_check = len; \
81  call_ref(tx_ref, out_ref, in, sizeof(DATA_TYPE)); \
82  call_new(tx, out_new, in, sizeof(DATA_TYPE)); \
83  if (CHECK_EXPRESSION) { \
84  fail(); \
85  av_tx_uninit(&tx); \
86  break; \
87  } \
88  bench_new(tx, out_new, in, sizeof(DATA_TYPE)); \
89  av_tx_uninit(&tx_refs[TYPE][DIR][i]); \
90  tx_refs[TYPE][DIR][i] = tx; \
91  } else { \
92  av_tx_uninit(&tx); \
93  } \
94  } \
95  \
96  if (num_checks == 1) \
97  report(PREFIX "_%i", last_check); \
98  else if (num_checks) \
99  report(PREFIX); \
100  } while (0)
101 
103 {
104  declare_func(void, AVTXContext *tx, void *out, void *in, ptrdiff_t stride);
105 
106  void *in = av_malloc(16384*2*8);
107  void *out_ref = av_malloc(16384*2*8);
108  void *out_new = av_malloc(16384*2*8);
109 
111  CHECK_TEMPLATE("float_fft", AV_TX_FLOAT_FFT, 0, AVComplexFloat, float, check_lens,
112  !float_near_abs_eps_array(out_ref, out_new, EPS, len*2));
113 
114  CHECK_TEMPLATE("float_imdct", AV_TX_FLOAT_MDCT, 1, float, float, check_lens,
115  !float_near_abs_eps_array(out_ref, out_new, EPS, len));
116 
118  CHECK_TEMPLATE("double_fft", AV_TX_DOUBLE_FFT, 0, AVComplexDouble, double, check_lens,
119  !double_near_abs_eps_array(out_ref, out_new, EPS, len*2));
120 
121  av_free(in);
122  av_free(out_ref);
123  av_free(out_new);
124 
125  if (!init) {
126  init = 1;
127  atexit(free_tx_refs);
128  }
129 }
tx_refs
static AVTXContext * tx_refs[AV_TX_NB][2][FF_ARRAY_ELEMS(check_lens)]
Definition: av_tx.c:47
mem_internal.h
out
FILE * out
Definition: movenc.c:55
float_near_abs_eps_array
int float_near_abs_eps_array(const float *a, const float *b, float eps, unsigned len)
Definition: checkasm.c:408
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:48
checkasm.h
free_tx_refs
static void free_tx_refs(void)
Definition: av_tx.c:50
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:444
CHECK_TEMPLATE
#define CHECK_TEMPLATE(PREFIX, TYPE, DIR, DATA_TYPE, SCALE_TYPE, LENGTHS, CHECK_EXPRESSION)
Definition: av_tx.c:58
check_lens
static const int check_lens[]
Definition: av_tx.c:43
error.h
EPS
#define EPS
Definition: av_tx.c:28
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:295
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:256
SCALE_NOOP
#define SCALE_NOOP(x)
Definition: av_tx.c:30
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:102
randomize_complex
#define randomize_complex(BUF, LEN, TYPE, SCALE)
Definition: av_tx.c:33
mem.h
declare_func
#define declare_func(ret,...)
Definition: checkasm.h:177
av_free
#define av_free(p)
Definition: tableprint_vlc.h:33
tx.h