FFmpeg
af_biquads.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2013 Paul B Mahol
3  * Copyright (c) 2006-2008 Rob Sykes <robs@users.sourceforge.net>
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 /*
23  * 2-pole filters designed by Robert Bristow-Johnson <rbj@audioimagination.com>
24  * see http://www.musicdsp.org/files/Audio-EQ-Cookbook.txt
25  *
26  * 1-pole filters based on code (c) 2000 Chris Bagwell <cbagwell@sprynet.com>
27  * Algorithms: Recursive single pole low/high pass filter
28  * Reference: The Scientist and Engineer's Guide to Digital Signal Processing
29  *
30  * low-pass: output[N] = input[N] * A + output[N-1] * B
31  * X = exp(-2.0 * pi * Fc)
32  * A = 1 - X
33  * B = X
34  * Fc = cutoff freq / sample rate
35  *
36  * Mimics an RC low-pass filter:
37  *
38  * ---/\/\/\/\----------->
39  * |
40  * --- C
41  * ---
42  * |
43  * |
44  * V
45  *
46  * high-pass: output[N] = A0 * input[N] + A1 * input[N-1] + B1 * output[N-1]
47  * X = exp(-2.0 * pi * Fc)
48  * A0 = (1 + X) / 2
49  * A1 = -(1 + X) / 2
50  * B1 = X
51  * Fc = cutoff freq / sample rate
52  *
53  * Mimics an RC high-pass filter:
54  *
55  * || C
56  * ----||--------->
57  * || |
58  * <
59  * > R
60  * <
61  * |
62  * V
63  */
64 
65 #include "config_components.h"
66 
67 #include "libavutil/avassert.h"
68 #include "libavutil/channel_layout.h"
69 #include "libavutil/ffmath.h"
70 #include "libavutil/mem.h"
71 #include "libavutil/opt.h"
72 #include "audio.h"
73 #include "avfilter.h"
74 #include "filters.h"
75 #include "formats.h"
76 
77 enum FilterType {
78  biquad,
79  equalizer,
80  bass,
81  treble,
82  bandpass,
83  bandreject,
84  allpass,
85  highpass,
86  lowpass,
87  lowshelf,
88  highshelf,
89  tiltshelf,
90 };
91 
92 enum WidthType {
93  NONE,
94  HERTZ,
95  OCTAVE,
96  QFACTOR,
97  SLOPE,
98  KHERTZ,
99  NB_WTYPE,
100 };
101 
102 enum TransformType {
103  DI,
104  DII,
105  TDI,
106  TDII,
107  LATT,
108  SVF,
109  ZDF,
110  NB_TTYPE,
111 };
112 
113 typedef struct BiquadsContext {
114  const AVClass *class;
115 
116  enum FilterType filter_type;
117  int width_type;
118  int poles;
119  int csg;
120  int transform_type;
121  int precision;
122  int block_samples;
123 
124  int bypass;
125 
126  double gain;
127  double frequency;
128  double width;
129  double mix;
130  char *ch_layout_str;
131  AVChannelLayout ch_layout;
132  int normalize;
133  int order;
134 
135  double a_double[3];
136  double b_double[3];
137 
138  float a_float[3];
139  float b_float[3];
140 
141  double oa[3];
142  double ob[3];
143 
144  AVFrame *block[3];
145 
146  int *clip;
147  AVFrame *cache[2];
148  int block_align;
149 
150  int64_t pts;
151  int nb_samples;
152 
153  void (*filter)(struct BiquadsContext *s, const void *ibuf, void *obuf, int len,
154  void *cache, int *clip, int disabled);
155 } BiquadsContext;
156 
157 static int query_formats(const AVFilterContext *ctx,
158  AVFilterFormatsConfig **cfg_in,
159  AVFilterFormatsConfig **cfg_out)
160 {
161  const BiquadsContext *s = ctx->priv;
162  static const enum AVSampleFormat auto_sample_fmts[] = {
163  AV_SAMPLE_FMT_S16P,
164  AV_SAMPLE_FMT_S32P,
165  AV_SAMPLE_FMT_FLTP,
166  AV_SAMPLE_FMT_DBLP,
167  AV_SAMPLE_FMT_NONE
168  };
169  enum AVSampleFormat sample_fmts[] = {
170  AV_SAMPLE_FMT_S16P,
171  AV_SAMPLE_FMT_NONE
172  };
173  const enum AVSampleFormat *sample_fmts_list = sample_fmts;
174  int ret;
175 
176  switch (s->precision) {
177  case 0:
178  sample_fmts[0] = AV_SAMPLE_FMT_S16P;
179  break;
180  case 1:
181  sample_fmts[0] = AV_SAMPLE_FMT_S32P;
182  break;
183  case 2:
184  sample_fmts[0] = AV_SAMPLE_FMT_FLTP;
185  break;
186  case 3:
187  sample_fmts[0] = AV_SAMPLE_FMT_DBLP;
188  break;
189  default:
190  sample_fmts_list = auto_sample_fmts;
191  break;
192  }
193  ret = ff_set_common_formats_from_list2(ctx, cfg_in, cfg_out, sample_fmts_list);
194  if (ret < 0)
195  return ret;
196 
197  return 0;
198 }
199 
200 #define BIQUAD_FILTER(name, type, ftype, min, max, need_clipping) \
201 static void biquad_## name (BiquadsContext *s, \
202  const void *input, void *output, int len, \
203  void *cache, int *clippings, int disabled) \
204 { \
205  const type *ibuf = input; \
206  type *obuf = output; \
207  ftype *fcache = cache; \
208  ftype i1 = fcache[0], i2 = fcache[1], o1 = fcache[2], o2 = fcache[3]; \
209  ftype *a = s->a_##ftype; \
210  ftype *b = s->b_##ftype; \
211  ftype a1 = -a[1]; \
212  ftype a2 = -a[2]; \
213  ftype b0 = b[0]; \
214  ftype b1 = b[1]; \
215  ftype b2 = b[2]; \
216  ftype wet = s->mix; \
217  ftype dry = 1. - wet; \
218  ftype out; \
219  int i; \
220  \
221  for (i = 0; i+1 < len; i++) { \
222  o2 = i2 * b2 + i1 * b1 + ibuf[i] * b0 + o2 * a2 + o1 * a1; \
223  i2 = ibuf[i]; \
224  out = o2 * wet + i2 * dry; \
225  if (disabled) { \
226  obuf[i] = i2; \
227  } else if (need_clipping && out < min) { \
228  (*clippings)++; \
229  obuf[i] = min; \
230  } else if (need_clipping && out > max) { \
231  (*clippings)++; \
232  obuf[i] = max; \
233  } else { \
234  obuf[i] = out; \
235  } \
236  i++; \
237  o1 = i1 * b2 + i2 * b1 + ibuf[i] * b0 + o1 * a2 + o2 * a1; \
238  i1 = ibuf[i]; \
239  out = o1 * wet + i1 * dry; \
240  if (disabled) { \
241  obuf[i] = i1; \
242  } else if (need_clipping && out < min) { \
243  (*clippings)++; \
244  obuf[i] = min; \
245  } else if (need_clipping && out > max) { \
246  (*clippings)++; \
247  obuf[i] = max; \
248  } else { \
249  obuf[i] = out; \
250  } \
251  } \
252  if (i < len) { \
253  ftype o0 = ibuf[i] * b0 + i1 * b1 + i2 * b2 + o1 * a1 + o2 * a2; \
254  i2 = i1; \
255  i1 = ibuf[i]; \
256  o2 = o1; \
257  o1 = o0; \
258  out = o0 * wet + i1 * dry; \
259  if (disabled) { \
260  obuf[i] = i1; \
261  } else if (need_clipping && out < min) { \
262  (*clippings)++; \
263  obuf[i] = min; \
264  } else if (need_clipping && out > max) { \
265  (*clippings)++; \
266  obuf[i] = max; \
267  } else { \
268  obuf[i] = out; \
269  } \
270  } \
271  fcache[0] = i1; \
272  fcache[1] = i2; \
273  fcache[2] = o1; \
274  fcache[3] = o2; \
275 }
276 
277 BIQUAD_FILTER(s16, int16_t, float, INT16_MIN, INT16_MAX, 1)
278 BIQUAD_FILTER(s32, int32_t, double, INT32_MIN, INT32_MAX, 1)
279 BIQUAD_FILTER(flt, float, float, -1.f, 1.f, 0)
280 BIQUAD_FILTER(dbl, double, double, -1., 1., 0)
281 
282 #define BIQUAD_DII_FILTER(name, type, ftype, min, max, need_clipping) \
283 static void biquad_dii_## name (BiquadsContext *s, \
284  const void *input, void *output, int len, \
285  void *cache, int *clippings, int disabled) \
286 { \
287  const type *ibuf = input; \
288  type *obuf = output; \
289  ftype *fcache = cache; \
290  ftype *a = s->a_##ftype; \
291  ftype *b = s->b_##ftype; \
292  ftype a1 = -a[1]; \
293  ftype a2 = -a[2]; \
294  ftype b0 = b[0]; \
295  ftype b1 = b[1]; \
296  ftype b2 = b[2]; \
297  ftype w1 = fcache[0]; \
298  ftype w2 = fcache[1]; \
299  ftype wet = s->mix; \
300  ftype dry = 1. - wet; \
301  ftype in, out, w0; \
302  \
303  for (int i = 0; i < len; i++) { \
304  in = ibuf[i]; \
305  w0 = in + a1 * w1 + a2 * w2; \
306  out = b0 * w0 + b1 * w1 + b2 * w2; \
307  w2 = w1; \
308  w1 = w0; \
309  out = out * wet + in * dry; \
310  if (disabled) { \
311  obuf[i] = in; \
312  } else if (need_clipping && out < min) { \
313  (*clippings)++; \
314  obuf[i] = min; \
315  } else if (need_clipping && out > max) { \
316  (*clippings)++; \
317  obuf[i] = max; \
318  } else { \
319  obuf[i] = out; \
320  } \
321  } \
322  fcache[0] = w1; \
323  fcache[1] = w2; \
324 }
325 
326 BIQUAD_DII_FILTER(s16, int16_t, float, INT16_MIN, INT16_MAX, 1)
327 BIQUAD_DII_FILTER(s32, int32_t, double, INT32_MIN, INT32_MAX, 1)
328 BIQUAD_DII_FILTER(flt, float, float, -1.f, 1.f, 0)
329 BIQUAD_DII_FILTER(dbl, double, double, -1., 1., 0)
330 
331 #define BIQUAD_TDI_FILTER(name, type, ftype, min, max, need_clipping) \
332 static void biquad_tdi_## name (BiquadsContext *s, \
333  const void *input, void *output, int len, \
334  void *cache, int *clippings, int disabled) \
335 { \
336  const type *ibuf = input; \
337  type *obuf = output; \
338  ftype *fcache = cache; \
339  ftype *a = s->a_##ftype; \
340  ftype *b = s->b_##ftype; \
341  ftype a1 = -a[1]; \
342  ftype a2 = -a[2]; \
343  ftype b0 = b[0]; \
344  ftype b1 = b[1]; \
345  ftype b2 = b[2]; \
346  ftype s1 = fcache[0]; \
347  ftype s2 = fcache[1]; \
348  ftype s3 = fcache[2]; \
349  ftype s4 = fcache[3]; \
350  ftype wet = s->mix; \
351  ftype dry = 1. - wet; \
352  ftype in, out; \
353  \
354  for (int i = 0; i < len; i++) { \
355  ftype t1, t2, t3, t4; \
356  in = ibuf[i] + s1; \
357  t1 = in * a1 + s2; \
358  t2 = in * a2; \
359  t3 = in * b1 + s4; \
360  t4 = in * b2; \
361  out = b0 * in + s3; \
362  out = out * wet + in * dry; \
363  s1 = t1; s2 = t2; s3 = t3; s4 = t4; \
364  if (disabled) { \
365  obuf[i] = in; \
366  } else if (need_clipping && out < min) { \
367  (*clippings)++; \
368  obuf[i] = min; \
369  } else if (need_clipping && out > max) { \
370  (*clippings)++; \
371  obuf[i] = max; \
372  } else { \
373  obuf[i] = out; \
374  } \
375  } \
376  \
377  fcache[0] = s1; \
378  fcache[1] = s2; \
379  fcache[2] = s3; \
380  fcache[3] = s4; \
381 }
382 
383 BIQUAD_TDI_FILTER(s16, int16_t, float, INT16_MIN, INT16_MAX, 1)
384 BIQUAD_TDI_FILTER(s32, int32_t, double, INT32_MIN, INT32_MAX, 1)
385 BIQUAD_TDI_FILTER(flt, float, float, -1.f, 1.f, 0)
386 BIQUAD_TDI_FILTER(dbl, double, double, -1., 1., 0)
387 
388 #define BIQUAD_TDII_FILTER(name, type, ftype, min, max, need_clipping) \
389 static void biquad_tdii_## name (BiquadsContext *s, \
390  const void *input, void *output, int len, \
391  void *cache, int *clippings, int disabled) \
392 { \
393  const type *ibuf = input; \
394  type *obuf = output; \
395  ftype *fcache = cache; \
396  ftype *a = s->a_##ftype; \
397  ftype *b = s->b_##ftype; \
398  ftype a1 = -a[1]; \
399  ftype a2 = -a[2]; \
400  ftype b0 = b[0]; \
401  ftype b1 = b[1]; \
402  ftype b2 = b[2]; \
403  ftype w1 = fcache[0]; \
404  ftype w2 = fcache[1]; \
405  ftype wet = s->mix; \
406  ftype dry = 1. - wet; \
407  ftype in, out; \
408  \
409  for (int i = 0; i < len; i++) { \
410  in = ibuf[i]; \
411  out = b0 * in + w1; \
412  w1 = b1 * in + w2 + a1 * out; \
413  w2 = b2 * in + a2 * out; \
414  out = out * wet + in * dry; \
415  if (disabled) { \
416  obuf[i] = in; \
417  } else if (need_clipping && out < min) { \
418  (*clippings)++; \
419  obuf[i] = min; \
420  } else if (need_clipping && out > max) { \
421  (*clippings)++; \
422  obuf[i] = max; \
423  } else { \
424  obuf[i] = out; \
425  } \
426  } \
427  fcache[0] = w1; \
428  fcache[1] = w2; \
429 }
430 
431 BIQUAD_TDII_FILTER(s16, int16_t, float, INT16_MIN, INT16_MAX, 1)
432 BIQUAD_TDII_FILTER(s32, int32_t, double, INT32_MIN, INT32_MAX, 1)
433 BIQUAD_TDII_FILTER(flt, float, float, -1.f, 1.f, 0)
434 BIQUAD_TDII_FILTER(dbl, double, double, -1., 1., 0)
435 
436 #define BIQUAD_LATT_FILTER(name, type, ftype, min, max, need_clipping) \
437 static void biquad_latt_## name (BiquadsContext *s, \
438  const void *input, void *output, int len, \
439  void *cache, int *clippings, int disabled) \
440 { \
441  const type *ibuf = input; \
442  type *obuf = output; \
443  ftype *fcache = cache; \
444  ftype *a = s->a_##ftype; \
445  ftype *b = s->b_##ftype; \
446  ftype k0 = a[1]; \
447  ftype k1 = a[2]; \
448  ftype v0 = b[0]; \
449  ftype v1 = b[1]; \
450  ftype v2 = b[2]; \
451  ftype s0 = fcache[0]; \
452  ftype s1 = fcache[1]; \
453  ftype wet = s->mix; \
454  ftype dry = 1. - wet; \
455  ftype in, out; \
456  ftype t0, t1; \
457  \
458  for (int i = 0; i < len; i++) { \
459  out = 0.; \
460  in = ibuf[i]; \
461  t0 = in - k1 * s0; \
462  t1 = t0 * k1 + s0; \
463  out += t1 * v2; \
464  \
465  t0 = t0 - k0 * s1; \
466  t1 = t0 * k0 + s1; \
467  out += t1 * v1; \
468  \
469  out += t0 * v0; \
470  s0 = t1; \
471  s1 = t0; \
472  \
473  out = out * wet + in * dry; \
474  if (disabled) { \
475  obuf[i] = in; \
476  } else if (need_clipping && out < min) { \
477  (*clippings)++; \
478  obuf[i] = min; \
479  } else if (need_clipping && out > max) { \
480  (*clippings)++; \
481  obuf[i] = max; \
482  } else { \
483  obuf[i] = out; \
484  } \
485  } \
486  fcache[0] = s0; \
487  fcache[1] = s1; \
488 }
489 
490 BIQUAD_LATT_FILTER(s16, int16_t, float, INT16_MIN, INT16_MAX, 1)
491 BIQUAD_LATT_FILTER(s32, int32_t, double, INT32_MIN, INT32_MAX, 1)
492 BIQUAD_LATT_FILTER(flt, float, float, -1.f, 1.f, 0)
493 BIQUAD_LATT_FILTER(dbl, double, double, -1., 1., 0)
494 
495 #define BIQUAD_SVF_FILTER(name, type, ftype, min, max, need_clipping) \
496 static void biquad_svf_## name (BiquadsContext *s, \
497  const void *input, void *output, int len, \
498  void *cache, int *clippings, int disabled) \
499 { \
500  const type *ibuf = input; \
501  type *obuf = output; \
502  ftype *fcache = cache; \
503  ftype *a = s->a_##ftype; \
504  ftype *b = s->b_##ftype; \
505  ftype a1 = a[1]; \
506  ftype a2 = a[2]; \
507  ftype b0 = b[0]; \
508  ftype b1 = b[1]; \
509  ftype b2 = b[2]; \
510  ftype s0 = fcache[0]; \
511  ftype s1 = fcache[1]; \
512  ftype wet = s->mix; \
513  ftype dry = 1. - wet; \
514  ftype in, out; \
515  ftype t0, t1; \
516  \
517  for (int i = 0; i < len; i++) { \
518  in = ibuf[i]; \
519  out = b2 * in + s0; \
520  t0 = b0 * in + a1 * s0 + s1; \
521  t1 = b1 * in + a2 * s0; \
522  s0 = t0; \
523  s1 = t1; \
524  \
525  out = out * wet + in * dry; \
526  if (disabled) { \
527  obuf[i] = in; \
528  } else if (need_clipping && out < min) { \
529  (*clippings)++; \
530  obuf[i] = min; \
531  } else if (need_clipping && out > max) { \
532  (*clippings)++; \
533  obuf[i] = max; \
534  } else { \
535  obuf[i] = out; \
536  } \
537  } \
538  fcache[0] = s0; \
539  fcache[1] = s1; \
540 }
541 
542 BIQUAD_SVF_FILTER(s16, int16_t, float, INT16_MIN, INT16_MAX, 1)
543 BIQUAD_SVF_FILTER(s32, int32_t, double, INT32_MIN, INT32_MAX, 1)
544 BIQUAD_SVF_FILTER(flt, float, float, -1.f, 1.f, 0)
545 BIQUAD_SVF_FILTER(dbl, double, double, -1., 1., 0)
546 
547 #define BIQUAD_ZDF_FILTER(name, type, ftype, min, max, need_clipping, two) \
548 static void biquad_zdf_## name (BiquadsContext *s, \
549  const void *input, void *output, int len, \
550  void *cache, int *clippings, int disabled) \
551 { \
552  const type *ibuf = input; \
553  type *obuf = output; \
554  ftype *fcache = cache; \
555  ftype *a = s->a_##ftype; \
556  ftype *b = s->b_##ftype; \
557  ftype m0 = b[0]; \
558  ftype m1 = b[1]; \
559  ftype m2 = b[2]; \
560  ftype a0 = a[0]; \
561  ftype a1 = a[1]; \
562  ftype a2 = a[2]; \
563  ftype b0 = fcache[0]; \
564  ftype b1 = fcache[1]; \
565  ftype wet = s->mix; \
566  ftype dry = 1. - wet; \
567  ftype out; \
568  \
569  for (int i = 0; i < len; i++) { \
570  const ftype in = ibuf[i]; \
571  const ftype v0 = in; \
572  const ftype v3 = v0 - b1; \
573  const ftype v1 = a0 * b0 + a1 * v3; \
574  const ftype v2 = b1 + a1 * b0 + a2 * v3; \
575  \
576  b0 = two * v1 - b0; \
577  b1 = two * v2 - b1; \
578  \
579  out = m0 * v0 + m1 * v1 + m2 * v2; \
580  out = out * wet + in * dry; \
581  if (disabled) { \
582  obuf[i] = in; \
583  } else if (need_clipping && out < min) { \
584  (*clippings)++; \
585  obuf[i] = min; \
586  } else if (need_clipping && out > max) { \
587  (*clippings)++; \
588  obuf[i] = max; \
589  } else { \
590  obuf[i] = out; \
591  } \
592  } \
593  fcache[0] = b0; \
594  fcache[1] = b1; \
595 }
596 
597 BIQUAD_ZDF_FILTER(s16, int16_t, float, INT16_MIN, INT16_MAX, 1, 2.f)
598 BIQUAD_ZDF_FILTER(s32, int32_t, double, INT32_MIN, INT32_MAX, 1, 2.0)
599 BIQUAD_ZDF_FILTER(flt, float, float, -1.f, 1.f, 0, 2.f)
600 BIQUAD_ZDF_FILTER(dbl, double, double, -1., 1., 0, 2.0)
601 
602 static void convert_dir2latt(BiquadsContext *s)
603 {
604  double k0, k1, v0, v1, v2;
605 
606  k1 = s->a_double[2];
607  k0 = s->a_double[1] / (1. + k1);
608  v2 = s->b_double[2];
609  v1 = s->b_double[1] - v2 * s->a_double[1];
610  v0 = s->b_double[0] - v1 * k0 - v2 * k1;
611 
612  s->a_double[1] = k0;
613  s->a_double[2] = k1;
614  s->b_double[0] = v0;
615  s->b_double[1] = v1;
616  s->b_double[2] = v2;
617 }
618 
619 static void convert_dir2svf(BiquadsContext *s)
620 {
621  double a[2];
622  double b[3];
623 
624  a[0] = -s->a_double[1];
625  a[1] = -s->a_double[2];
626  b[0] = s->b_double[1] - s->a_double[1] * s->b_double[0];
627  b[1] = s->b_double[2] - s->a_double[2] * s->b_double[0];
628  b[2] = s->b_double[0];
629 
630  s->a_double[1] = a[0];
631  s->a_double[2] = a[1];
632  s->b_double[0] = b[0];
633  s->b_double[1] = b[1];
634  s->b_double[2] = b[2];
635 }
636 
637 static double convert_width2qfactor(double width,
638  double frequency,
639  double gain,
640  double sample_rate,
641  int width_type)
642 {
643  double w0 = 2. * M_PI * frequency / sample_rate;
644  double A = ff_exp10(gain / 40.);
645  double ret;
646 
647  switch (width_type) {
648  case NONE:
649  case QFACTOR:
650  ret = width;
651  break;
652  case HERTZ:
653  ret = frequency / width;
654  break;
655  case KHERTZ:
656  ret = frequency / (width * 1000.);
657  break;
658  case OCTAVE:
659  ret = 1. / (2. * sinh(log(2.) / 2. * width * w0 / sin(w0)));
660  break;
661  case SLOPE:
662  ret = 1. / sqrt((A + 1. / A) * (1. / width - 1.) + 2.);
663  break;
664  default:
665  av_assert0(0);
666  break;
667  }
668 
669  return ret;
670 }
671 
672 static void convert_dir2zdf(BiquadsContext *s, int sample_rate)
673 {
674  double Q = convert_width2qfactor(s->width, s->frequency, s->gain, sample_rate, s->width_type);
675  double g, k, A;
676  double a[3];
677  double m[3];
678 
679  switch (s->filter_type) {
680  case biquad:
681  a[0] = s->oa[0];
682  a[1] = s->oa[1];
683  a[2] = s->oa[2];
684  m[0] = s->ob[0];
685  m[1] = s->ob[1];
686  m[2] = s->ob[2];
687  break;
688  case equalizer:
689  A = ff_exp10(s->gain / 40.);
690  g = tan(M_PI * s->frequency / sample_rate);
691  k = 1. / (Q * A);
692  a[0] = 1. / (1. + g * (g + k));
693  a[1] = g * a[0];
694  a[2] = g * a[1];
695  m[0] = 1.;
696  m[1] = k * (A * A - 1.);
697  m[2] = 0.;
698  break;
699  case bass:
700  case lowshelf:
701  A = ff_exp10(s->gain / 40.);
702  g = tan(M_PI * s->frequency / sample_rate) / sqrt(A);
703  k = 1. / Q;
704  a[0] = 1. / (1. + g * (g + k));
705  a[1] = g * a[0];
706  a[2] = g * a[1];
707  m[0] = 1.;
708  m[1] = k * (A - 1.);
709  m[2] = A * A - 1.;
710  break;
711  case tiltshelf:
712  A = ff_exp10(s->gain / 20.);
713  g = tan(M_PI * s->frequency / sample_rate) / sqrt(A);
714  k = 1. / Q;
715  a[0] = 1. / (1. + g * (g + k));
716  a[1] = g * a[0];
717  a[2] = g * a[1];
718  m[0] = 1./ A;
719  m[1] = k * (A - 1.) / A;
720  m[2] = (A * A - 1.) / A;
721  break;
722  case treble:
723  case highshelf:
724  A = ff_exp10(s->gain / 40.);
725  g = tan(M_PI * s->frequency / sample_rate) * sqrt(A);
726  k = 1. / Q;
727  a[0] = 1. / (1. + g * (g + k));
728  a[1] = g * a[0];
729  a[2] = g * a[1];
730  m[0] = A * A;
731  m[1] = k * (1. - A) * A;
732  m[2] = 1. - A * A;
733  break;
734  case bandpass:
735  g = tan(M_PI * s->frequency / sample_rate);
736  k = 1. / Q;
737  a[0] = 1. / (1. + g * (g + k));
738  a[1] = g * a[0];
739  a[2] = g * a[1];
740  m[0] = 0.;
741  m[1] = s->csg ? 1. : k;
742  m[2] = 0.;
743  break;
744  case bandreject:
745  g = tan(M_PI * s->frequency / sample_rate);
746  k = 1. / Q;
747  a[0] = 1. / (1. + g * (g + k));
748  a[1] = g * a[0];
749  a[2] = g * a[1];
750  m[0] = 1.;
751  m[1] = -k;
752  m[2] = 0.;
753  break;
754  case lowpass:
755  g = tan(M_PI * s->frequency / sample_rate);
756  k = 1. / Q;
757  a[0] = 1. / (1. + g * (g + k));
758  a[1] = g * a[0];
759  a[2] = g * a[1];
760  m[0] = 0.;
761  m[1] = 0.;
762  m[2] = 1.;
763  break;
764  case highpass:
765  g = tan(M_PI * s->frequency / sample_rate);
766  k = 1. / Q;
767  a[0] = 1. / (1. + g * (g + k));
768  a[1] = g * a[0];
769  a[2] = g * a[1];
770  m[0] = 1.;
771  m[1] = -k;
772  m[2] = -1.;
773  break;
774  case allpass:
775  g = tan(M_PI * s->frequency / sample_rate);
776  k = 1. / Q;
777  a[0] = 1. / (1. + g * (g + k));
778  a[1] = g * a[0];
779  a[2] = g * a[1];
780  m[0] = 1.;
781  m[1] = -2. * k;
782  m[2] = 0.;
783  break;
784  default:
785  av_assert0(0);
786  }
787 
788  s->a_double[0] = a[0];
789  s->a_double[1] = a[1];
790  s->a_double[2] = a[2];
791  s->b_double[0] = m[0];
792  s->b_double[1] = m[1];
793  s->b_double[2] = m[2];
794 }
795 
796 static int config_filter(AVFilterLink *outlink, int reset)
797 {
798  AVFilterContext *ctx = outlink->src;
799  BiquadsContext *s = ctx->priv;
800  AVFilterLink *inlink = ctx->inputs[0];
801  double gain = s->gain * ((s->filter_type == tiltshelf) + 1.);
802  double A = ff_exp10(gain / 40);
803  double w0 = 2 * M_PI * s->frequency / inlink->sample_rate;
804  double K = tan(w0 / 2.);
805  double alpha, beta;
806 
807  s->bypass = (((w0 > M_PI || w0 <= 0.) && reset) || (s->width <= 0.)) && (s->filter_type != biquad);
808  if (s->bypass) {
809  av_log(ctx, AV_LOG_WARNING, "Invalid frequency and/or width!\n");
810  return 0;
811  }
812 
813  if ((w0 > M_PI || w0 <= 0.) && (s->filter_type != biquad))
814  return AVERROR(EINVAL);
815 
816  switch (s->width_type) {
817  case NONE:
818  alpha = 0.0;
819  break;
820  case HERTZ:
821  alpha = sin(w0) / (2 * s->frequency / s->width);
822  break;
823  case KHERTZ:
824  alpha = sin(w0) / (2 * s->frequency / (s->width * 1000));
825  break;
826  case OCTAVE:
827  alpha = sin(w0) * sinh(log(2.) / 2 * s->width * w0 / sin(w0));
828  break;
829  case QFACTOR:
830  alpha = sin(w0) / (2 * s->width);
831  break;
832  case SLOPE:
833  alpha = sin(w0) / 2 * sqrt((A + 1 / A) * (1 / s->width - 1) + 2);
834  break;
835  default:
836  av_assert0(0);
837  }
838 
839  beta = 2 * sqrt(A);
840 
841  switch (s->filter_type) {
842  case biquad:
843  s->a_double[0] = s->oa[0];
844  s->a_double[1] = s->oa[1];
845  s->a_double[2] = s->oa[2];
846  s->b_double[0] = s->ob[0];
847  s->b_double[1] = s->ob[1];
848  s->b_double[2] = s->ob[2];
849  break;
850  case equalizer:
851  s->a_double[0] = 1 + alpha / A;
852  s->a_double[1] = -2 * cos(w0);
853  s->a_double[2] = 1 - alpha / A;
854  s->b_double[0] = 1 + alpha * A;
855  s->b_double[1] = -2 * cos(w0);
856  s->b_double[2] = 1 - alpha * A;
857  break;
858  case bass:
859  beta = sqrt((A * A + 1) - (A - 1) * (A - 1));
860  case tiltshelf:
861  case lowshelf:
862  if (s->poles == 1) {
863  double A = ff_exp10(gain / 20);
864  double ro = -sin(w0 / 2. - M_PI_4) / sin(w0 / 2. + M_PI_4);
865  double n = (A + 1) / (A - 1);
866  double alpha1 = A == 1. ? 0. : n - FFSIGN(n) * sqrt(n * n - 1);
867  double beta0 = ((1 + A) + (1 - A) * alpha1) * 0.5;
868  double beta1 = ((1 - A) + (1 + A) * alpha1) * 0.5;
869 
870  s->a_double[0] = 1 + ro * alpha1;
871  s->a_double[1] = -ro - alpha1;
872  s->a_double[2] = 0;
873  s->b_double[0] = beta0 + ro * beta1;
874  s->b_double[1] = -beta1 - ro * beta0;
875  s->b_double[2] = 0;
876  } else {
877  s->a_double[0] = (A + 1) + (A - 1) * cos(w0) + beta * alpha;
878  s->a_double[1] = -2 * ((A - 1) + (A + 1) * cos(w0));
879  s->a_double[2] = (A + 1) + (A - 1) * cos(w0) - beta * alpha;
880  s->b_double[0] = A * ((A + 1) - (A - 1) * cos(w0) + beta * alpha);
881  s->b_double[1] = 2 * A * ((A - 1) - (A + 1) * cos(w0));
882  s->b_double[2] = A * ((A + 1) - (A - 1) * cos(w0) - beta * alpha);
883  }
884  break;
885  case treble:
886  beta = sqrt((A * A + 1) - (A - 1) * (A - 1));
887  case highshelf:
888  if (s->poles == 1) {
889  double A = ff_exp10(gain / 20);
890  double ro = sin(w0 / 2. - M_PI_4) / sin(w0 / 2. + M_PI_4);
891  double n = (A + 1) / (A - 1);
892  double alpha1 = A == 1. ? 0. : n - FFSIGN(n) * sqrt(n * n - 1);
893  double beta0 = ((1 + A) + (1 - A) * alpha1) * 0.5;
894  double beta1 = ((1 - A) + (1 + A) * alpha1) * 0.5;
895 
896  s->a_double[0] = 1 + ro * alpha1;
897  s->a_double[1] = ro + alpha1;
898  s->a_double[2] = 0;
899  s->b_double[0] = beta0 + ro * beta1;
900  s->b_double[1] = beta1 + ro * beta0;
901  s->b_double[2] = 0;
902  } else {
903  s->a_double[0] = (A + 1) - (A - 1) * cos(w0) + beta * alpha;
904  s->a_double[1] = 2 * ((A - 1) - (A + 1) * cos(w0));
905  s->a_double[2] = (A + 1) - (A - 1) * cos(w0) - beta * alpha;
906  s->b_double[0] = A * ((A + 1) + (A - 1) * cos(w0) + beta * alpha);
907  s->b_double[1] =-2 * A * ((A - 1) + (A + 1) * cos(w0));
908  s->b_double[2] = A * ((A + 1) + (A - 1) * cos(w0) - beta * alpha);
909  }
910  break;
911  case bandpass:
912  if (s->csg) {
913  s->a_double[0] = 1 + alpha;
914  s->a_double[1] = -2 * cos(w0);
915  s->a_double[2] = 1 - alpha;
916  s->b_double[0] = sin(w0) / 2;
917  s->b_double[1] = 0;
918  s->b_double[2] = -sin(w0) / 2;
919  } else {
920  s->a_double[0] = 1 + alpha;
921  s->a_double[1] = -2 * cos(w0);
922  s->a_double[2] = 1 - alpha;
923  s->b_double[0] = alpha;
924  s->b_double[1] = 0;
925  s->b_double[2] = -alpha;
926  }
927  break;
928  case bandreject:
929  s->a_double[0] = 1 + alpha;
930  s->a_double[1] = -2 * cos(w0);
931  s->a_double[2] = 1 - alpha;
932  s->b_double[0] = 1;
933  s->b_double[1] = -2 * cos(w0);
934  s->b_double[2] = 1;
935  break;
936  case lowpass:
937  if (s->poles == 1) {
938  s->a_double[0] = 1;
939  s->a_double[1] = -exp(-w0);
940  s->a_double[2] = 0;
941  s->b_double[0] = 1 + s->a_double[1];
942  s->b_double[1] = 0;
943  s->b_double[2] = 0;
944  } else {
945  s->a_double[0] = 1 + alpha;
946  s->a_double[1] = -2 * cos(w0);
947  s->a_double[2] = 1 - alpha;
948  s->b_double[0] = (1 - cos(w0)) / 2;
949  s->b_double[1] = 1 - cos(w0);
950  s->b_double[2] = (1 - cos(w0)) / 2;
951  }
952  break;
953  case highpass:
954  if (s->poles == 1) {
955  s->a_double[0] = 1;
956  s->a_double[1] = -exp(-w0);
957  s->a_double[2] = 0;
958  s->b_double[0] = (1 - s->a_double[1]) / 2;
959  s->b_double[1] = -s->b_double[0];
960  s->b_double[2] = 0;
961  } else {
962  s->a_double[0] = 1 + alpha;
963  s->a_double[1] = -2 * cos(w0);
964  s->a_double[2] = 1 - alpha;
965  s->b_double[0] = (1 + cos(w0)) / 2;
966  s->b_double[1] = -(1 + cos(w0));
967  s->b_double[2] = (1 + cos(w0)) / 2;
968  }
969  break;
970  case allpass:
971  switch (s->order) {
972  case 1:
973  s->a_double[0] = 1.;
974  s->a_double[1] = -(1. - K) / (1. + K);
975  s->a_double[2] = 0.;
976  s->b_double[0] = s->a_double[1];
977  s->b_double[1] = s->a_double[0];
978  s->b_double[2] = 0.;
979  break;
980  case 2:
981  s->a_double[0] = 1 + alpha;
982  s->a_double[1] = -2 * cos(w0);
983  s->a_double[2] = 1 - alpha;
984  s->b_double[0] = 1 - alpha;
985  s->b_double[1] = -2 * cos(w0);
986  s->b_double[2] = 1 + alpha;
987  break;
988  }
989  break;
990  default:
991  av_assert0(0);
992  }
993 
994  av_log(ctx, AV_LOG_VERBOSE, "a=%f %f %f:b=%f %f %f\n",
995  s->a_double[0], s->a_double[1], s->a_double[2],
996  s->b_double[0], s->b_double[1], s->b_double[2]);
997 
998  s->a_double[1] /= s->a_double[0];
999  s->a_double[2] /= s->a_double[0];
1000  s->b_double[0] /= s->a_double[0];
1001  s->b_double[1] /= s->a_double[0];
1002  s->b_double[2] /= s->a_double[0];
1003  s->a_double[0] /= s->a_double[0];
1004 
1005  if (s->normalize && fabs(s->b_double[0] + s->b_double[1] + s->b_double[2]) > 1e-6) {
1006  double factor = (s->a_double[0] + s->a_double[1] + s->a_double[2]) /
1007  (s->b_double[0] + s->b_double[1] + s->b_double[2]);
1008 
1009  s->b_double[0] *= factor;
1010  s->b_double[1] *= factor;
1011  s->b_double[2] *= factor;
1012  }
1013 
1014  switch (s->filter_type) {
1015  case tiltshelf:
1016  s->b_double[0] /= A;
1017  s->b_double[1] /= A;
1018  s->b_double[2] /= A;
1019  break;
1020  }
1021 
1022  if (!s->cache[0])
1023  s->cache[0] = ff_get_audio_buffer(outlink, 4 * sizeof(double));
1024  if (!s->clip)
1025  s->clip = av_calloc(outlink->ch_layout.nb_channels, sizeof(*s->clip));
1026  if (!s->cache[0] || !s->clip)
1027  return AVERROR(ENOMEM);
1028  if (reset) {
1029  av_samples_set_silence(s->cache[0]->extended_data, 0, s->cache[0]->nb_samples,
1030  s->cache[0]->ch_layout.nb_channels, s->cache[0]->format);
1031  }
1032 
1033  if (reset && s->block_samples > 0) {
1034  if (!s->cache[1])
1035  s->cache[1] = ff_get_audio_buffer(outlink, 4 * sizeof(double));
1036  if (!s->cache[1])
1037  return AVERROR(ENOMEM);
1038  av_samples_set_silence(s->cache[1]->extended_data, 0, s->cache[1]->nb_samples,
1039  s->cache[1]->ch_layout.nb_channels, s->cache[1]->format);
1040  for (int i = 0; i < 3; i++) {
1041  if (!s->block[i])
1042  s->block[i] = ff_get_audio_buffer(outlink, s->block_samples * 2);
1043  if (!s->block[i])
1044  return AVERROR(ENOMEM);
1045  av_samples_set_silence(s->block[i]->extended_data, 0, s->block_samples * 2,
1046  s->block[i]->ch_layout.nb_channels, s->block[i]->format);
1047  }
1048  }
1049 
1050  switch (s->transform_type) {
1051  case DI:
1052  switch (inlink->format) {
1053  case AV_SAMPLE_FMT_S16P:
1054  s->filter = biquad_s16;
1055  break;
1056  case AV_SAMPLE_FMT_S32P:
1057  s->filter = biquad_s32;
1058  break;
1059  case AV_SAMPLE_FMT_FLTP:
1060  s->filter = biquad_flt;
1061  break;
1062  case AV_SAMPLE_FMT_DBLP:
1063  s->filter = biquad_dbl;
1064  break;
1065  default: av_assert0(0);
1066  }
1067  break;
1068  case DII:
1069  switch (inlink->format) {
1070  case AV_SAMPLE_FMT_S16P:
1071  s->filter = biquad_dii_s16;
1072  break;
1073  case AV_SAMPLE_FMT_S32P:
1074  s->filter = biquad_dii_s32;
1075  break;
1076  case AV_SAMPLE_FMT_FLTP:
1077  s->filter = biquad_dii_flt;
1078  break;
1079  case AV_SAMPLE_FMT_DBLP:
1080  s->filter = biquad_dii_dbl;
1081  break;
1082  default: av_assert0(0);
1083  }
1084  break;
1085  case TDI:
1086  switch (inlink->format) {
1087  case AV_SAMPLE_FMT_S16P:
1088  s->filter = biquad_tdi_s16;
1089  break;
1090  case AV_SAMPLE_FMT_S32P:
1091  s->filter = biquad_tdi_s32;
1092  break;
1093  case AV_SAMPLE_FMT_FLTP:
1094  s->filter = biquad_tdi_flt;
1095  break;
1096  case AV_SAMPLE_FMT_DBLP:
1097  s->filter = biquad_tdi_dbl;
1098  break;
1099  default: av_assert0(0);
1100  }
1101  break;
1102  case TDII:
1103  switch (inlink->format) {
1104  case AV_SAMPLE_FMT_S16P:
1105  s->filter = biquad_tdii_s16;
1106  break;
1107  case AV_SAMPLE_FMT_S32P:
1108  s->filter = biquad_tdii_s32;
1109  break;
1110  case AV_SAMPLE_FMT_FLTP:
1111  s->filter = biquad_tdii_flt;
1112  break;
1113  case AV_SAMPLE_FMT_DBLP:
1114  s->filter = biquad_tdii_dbl;
1115  break;
1116  default: av_assert0(0);
1117  }
1118  break;
1119  case LATT:
1120  switch (inlink->format) {
1121  case AV_SAMPLE_FMT_S16P:
1122  s->filter = biquad_latt_s16;
1123  break;
1124  case AV_SAMPLE_FMT_S32P:
1125  s->filter = biquad_latt_s32;
1126  break;
1127  case AV_SAMPLE_FMT_FLTP:
1128  s->filter = biquad_latt_flt;
1129  break;
1130  case AV_SAMPLE_FMT_DBLP:
1131  s->filter = biquad_latt_dbl;
1132  break;
1133  default: av_assert0(0);
1134  }
1135  break;
1136  case SVF:
1137  switch (inlink->format) {
1138  case AV_SAMPLE_FMT_S16P:
1139  s->filter = biquad_svf_s16;
1140  break;
1141  case AV_SAMPLE_FMT_S32P:
1142  s->filter = biquad_svf_s32;
1143  break;
1144  case AV_SAMPLE_FMT_FLTP:
1145  s->filter = biquad_svf_flt;
1146  break;
1147  case AV_SAMPLE_FMT_DBLP:
1148  s->filter = biquad_svf_dbl;
1149  break;
1150  default: av_assert0(0);
1151  }
1152  break;
1153  case ZDF:
1154  switch (inlink->format) {
1155  case AV_SAMPLE_FMT_S16P:
1156  s->filter = biquad_zdf_s16;
1157  break;
1158  case AV_SAMPLE_FMT_S32P:
1159  s->filter = biquad_zdf_s32;
1160  break;
1161  case AV_SAMPLE_FMT_FLTP:
1162  s->filter = biquad_zdf_flt;
1163  break;
1164  case AV_SAMPLE_FMT_DBLP:
1165  s->filter = biquad_zdf_dbl;
1166  break;
1167  default: av_assert0(0);
1168  }
1169  break;
1170  default:
1171  av_assert0(0);
1172  }
1173 
1174  s->block_align = av_get_bytes_per_sample(inlink->format);
1175 
1176  if (s->transform_type == LATT)
1177  convert_dir2latt(s);
1178  else if (s->transform_type == SVF)
1179  convert_dir2svf(s);
1180  else if (s->transform_type == ZDF)
1181  convert_dir2zdf(s, inlink->sample_rate);
1182 
1183  s->a_float[0] = s->a_double[0];
1184  s->a_float[1] = s->a_double[1];
1185  s->a_float[2] = s->a_double[2];
1186  s->b_float[0] = s->b_double[0];
1187  s->b_float[1] = s->b_double[1];
1188  s->b_float[2] = s->b_double[2];
1189 
1190  return 0;
1191 }
1192 
1193 static int config_output(AVFilterLink *outlink)
1194 {
1195  return config_filter(outlink, 1);
1196 }
1197 
1198 typedef struct ThreadData {
1199  AVFrame *in, *out;
1200  int eof;
1201 } ThreadData;
1202 
1203 static void reverse_samples(AVFrame *out, AVFrame *in, int p,
1204  int oo, int io, int nb_samples)
1205 {
1206  switch (out->format) {
1207  case AV_SAMPLE_FMT_S16P: {
1208  const int16_t *src = ((const int16_t *)in->extended_data[p]) + io;
1209  int16_t *dst = ((int16_t *)out->extended_data[p]) + oo;
1210  for (int i = 0, j = nb_samples - 1; i < nb_samples; i++, j--)
1211  dst[i] = src[j];
1212  }
1213  break;
1214  case AV_SAMPLE_FMT_S32P: {
1215  const int32_t *src = ((const int32_t *)in->extended_data[p]) + io;
1216  int32_t *dst = ((int32_t *)out->extended_data[p]) + oo;
1217  for (int i = 0, j = nb_samples - 1; i < nb_samples; i++, j--)
1218  dst[i] = src[j];
1219  }
1220  break;
1221  case AV_SAMPLE_FMT_FLTP: {
1222  const float *src = ((const float *)in->extended_data[p]) + io;
1223  float *dst = ((float *)out->extended_data[p]) + oo;
1224  for (int i = 0, j = nb_samples - 1; i < nb_samples; i++, j--)
1225  dst[i] = src[j];
1226  }
1227  break;
1228  case AV_SAMPLE_FMT_DBLP: {
1229  const double *src = ((const double *)in->extended_data[p]) + io;
1230  double *dst = ((double *)out->extended_data[p]) + oo;
1231  for (int i = 0, j = nb_samples - 1; i < nb_samples; i++, j--)
1232  dst[i] = src[j];
1233  }
1234  break;
1235  }
1236 }
1237 
1238 static int filter_channel(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
1239 {
1240  AVFilterLink *inlink = ctx->inputs[0];
1241  ThreadData *td = arg;
1242  AVFrame *buf = td->in;
1243  AVFrame *out_buf = td->out;
1244  BiquadsContext *s = ctx->priv;
1245  const int start = (buf->ch_layout.nb_channels * jobnr) / nb_jobs;
1246  const int end = (buf->ch_layout.nb_channels * (jobnr+1)) / nb_jobs;
1247  int ch;
1248 
1249  for (ch = start; ch < end; ch++) {
1250  enum AVChannel channel = av_channel_layout_channel_from_index(&inlink->ch_layout, ch);
1251 
1252  if (av_channel_layout_index_from_channel(&s->ch_layout, channel) < 0) {
1253  if (buf != out_buf)
1254  memcpy(out_buf->extended_data[ch], buf->extended_data[ch],
1255  buf->nb_samples * s->block_align);
1256  continue;
1257  }
1258 
1259  if (!s->block_samples) {
1260  s->filter(s, buf->extended_data[ch], out_buf->extended_data[ch], buf->nb_samples,
1261  s->cache[0]->extended_data[ch], s->clip+ch, ctx->is_disabled);
1262  } else if (td->eof) {
1263  memcpy(out_buf->extended_data[ch], s->block[1]->extended_data[ch] + s->block_align * s->block_samples,
1264  s->nb_samples * s->block_align);
1265  } else {
1266  memcpy(s->block[0]->extended_data[ch] + s->block_align * s->block_samples, buf->extended_data[ch],
1267  buf->nb_samples * s->block_align);
1268  memset(s->block[0]->extended_data[ch] + s->block_align * (s->block_samples + buf->nb_samples),
1269  0, (s->block_samples - buf->nb_samples) * s->block_align);
1270  s->filter(s, s->block[0]->extended_data[ch], s->block[1]->extended_data[ch], s->block_samples,
1271  s->cache[0]->extended_data[ch], s->clip+ch, ctx->is_disabled);
1272  av_samples_copy(s->cache[1]->extended_data, s->cache[0]->extended_data, 0, 0,
1273  s->cache[0]->nb_samples, s->cache[0]->ch_layout.nb_channels,
1274  s->cache[0]->format);
1275  s->filter(s, s->block[0]->extended_data[ch] + s->block_samples * s->block_align,
1276  s->block[1]->extended_data[ch] + s->block_samples * s->block_align,
1277  s->block_samples, s->cache[1]->extended_data[ch], s->clip+ch,
1278  ctx->is_disabled);
1279  reverse_samples(s->block[2], s->block[1], ch, 0, 0, 2 * s->block_samples);
1280  av_samples_set_silence(s->cache[1]->extended_data, 0, s->cache[1]->nb_samples,
1281  s->cache[1]->ch_layout.nb_channels, s->cache[1]->format);
1282  s->filter(s, s->block[2]->extended_data[ch], s->block[2]->extended_data[ch], 2 * s->block_samples,
1283  s->cache[1]->extended_data[ch], s->clip+ch, ctx->is_disabled);
1284  reverse_samples(s->block[1], s->block[2], ch, 0, 0, 2 * s->block_samples);
1285  memcpy(out_buf->extended_data[ch], s->block[1]->extended_data[ch],
1286  s->block_samples * s->block_align);
1287  memmove(s->block[0]->extended_data[ch], s->block[0]->extended_data[ch] + s->block_align * s->block_samples,
1288  s->block_samples * s->block_align);
1289  }
1290  }
1291 
1292  return 0;
1293 }
1294 
1295 static int filter_frame(AVFilterLink *inlink, AVFrame *buf, int eof)
1296 {
1297  AVFilterContext *ctx = inlink->dst;
1298  BiquadsContext *s = ctx->priv;
1299  AVFilterLink *outlink = ctx->outputs[0];
1300  AVFrame *out_buf;
1301  ThreadData td;
1302  int ch, ret, drop = 0;
1303 
1304  if (s->bypass)
1305  return ff_filter_frame(outlink, buf);
1306 
1307  ret = av_channel_layout_copy(&s->ch_layout, &inlink->ch_layout);
1308  if (ret < 0) {
1309  av_frame_free(&buf);
1310  return ret;
1311  }
1312  if (strcmp(s->ch_layout_str, "all"))
1313  av_channel_layout_from_string(&s->ch_layout,
1314  s->ch_layout_str);
1315 
1316  if (av_frame_is_writable(buf) && s->block_samples == 0) {
1317  out_buf = buf;
1318  } else {
1319  out_buf = ff_get_audio_buffer(outlink, s->block_samples > 0 ? s->block_samples : buf->nb_samples);
1320  if (!out_buf) {
1321  av_frame_free(&buf);
1322  return AVERROR(ENOMEM);
1323  }
1324  av_frame_copy_props(out_buf, buf);
1325  }
1326 
1327  if (s->block_samples > 0 && s->pts == AV_NOPTS_VALUE)
1328  drop = 1;
1329  td.in = buf;
1330  td.out = out_buf;
1331  td.eof = eof;
1332  ff_filter_execute(ctx, filter_channel, &td, NULL,
1333  FFMIN(outlink->ch_layout.nb_channels, ff_filter_get_nb_threads(ctx)));
1334 
1335  for (ch = 0; ch < outlink->ch_layout.nb_channels; ch++) {
1336  if (s->clip[ch] > 0)
1337  av_log(ctx, AV_LOG_WARNING, "Channel %d clipping %d times. Please reduce gain.\n",
1338  ch, s->clip[ch]);
1339  s->clip[ch] = 0;
1340  }
1341 
1342  if (s->block_samples > 0) {
1343  int nb_samples = buf->nb_samples;
1344  int64_t pts = buf->pts;
1345 
1346  out_buf->pts = s->pts;
1347  out_buf->nb_samples = s->nb_samples;
1348  s->pts = pts;
1349  s->nb_samples = nb_samples;
1350  }
1351 
1352  if (buf != out_buf)
1353  av_frame_free(&buf);
1354 
1355  if (!drop)
1356  return ff_filter_frame(outlink, out_buf);
1357  else {
1358  av_frame_free(&out_buf);
1359  ff_filter_set_ready(ctx, 10);
1360  return 0;
1361  }
1362 }
1363 
1364 static int activate(AVFilterContext *ctx)
1365 {
1366  AVFilterLink *inlink = ctx->inputs[0];
1367  AVFilterLink *outlink = ctx->outputs[0];
1368  BiquadsContext *s = ctx->priv;
1369  AVFrame *in = NULL;
1370  int64_t pts;
1371  int status;
1372  int ret;
1373 
1374  FF_FILTER_FORWARD_STATUS_BACK(outlink, inlink);
1375 
1376  if (s->block_samples > 0) {
1377  ret = ff_inlink_consume_samples(inlink, s->block_samples, s->block_samples, &in);
1378  } else {
1379  ret = ff_inlink_consume_frame(inlink, &in);
1380  }
1381  if (ret < 0)
1382  return ret;
1383  if (ret > 0)
1384  return filter_frame(inlink, in, 0);
1385 
1386  if (s->block_samples > 0 && ff_inlink_queued_samples(inlink) >= s->block_samples) {
1387  ff_filter_set_ready(ctx, 10);
1388  return 0;
1389  }
1390 
1391  if (ff_inlink_acknowledge_status(inlink, &status, &pts)) {
1392  if (s->block_samples > 0) {
1393  AVFrame *in = ff_get_audio_buffer(outlink, s->block_samples);
1394  if (!in)
1395  return AVERROR(ENOMEM);
1396 
1397  ret = filter_frame(inlink, in, 1);
1398  }
1399 
1400  ff_outlink_set_status(outlink, status, pts);
1401 
1402  return ret;
1403  }
1404 
1405  FF_FILTER_FORWARD_WANTED(outlink, inlink);
1406 
1407  return FFERROR_NOT_READY;
1408 }
1409 
1410 static int process_command(AVFilterContext *ctx, const char *cmd, const char *args,
1411  char *res, int res_len, int flags)
1412 {
1413  AVFilterLink *outlink = ctx->outputs[0];
1414  int ret;
1415 
1416  ret = ff_filter_process_command(ctx, cmd, args, res, res_len, flags);
1417  if (ret < 0)
1418  return ret;
1419 
1420  return config_filter(outlink, 0);
1421 }
1422 
1423 static av_cold void uninit(AVFilterContext *ctx)
1424 {
1425  BiquadsContext *s = ctx->priv;
1426 
1427  for (int i = 0; i < 3; i++)
1428  av_frame_free(&s->block[i]);
1429  av_frame_free(&s->cache[0]);
1430  av_frame_free(&s->cache[1]);
1431  av_freep(&s->clip);
1432  av_channel_layout_uninit(&s->ch_layout);
1433 }
1434 
1435 static const AVFilterPad outputs[] = {
1436  {
1437  .name = "default",
1438  .type = AVMEDIA_TYPE_AUDIO,
1439  .config_props = config_output,
1440  },
1441 };
1442 
1443 #define OFFSET(x) offsetof(BiquadsContext, x)
1444 #define FLAGS AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_RUNTIME_PARAM
1445 #define AF AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
1446 
1447 #define DEFINE_BIQUAD_FILTER_2(name_, description_, priv_class_) \
1448 static av_cold int name_##_init(AVFilterContext *ctx) \
1449 { \
1450  BiquadsContext *s = ctx->priv; \
1451  s->filter_type = name_; \
1452  s->pts = AV_NOPTS_VALUE; \
1453  return 0; \
1454 } \
1455  \
1456 const AVFilter ff_af_##name_ = { \
1457  .name = #name_, \
1458  .description = NULL_IF_CONFIG_SMALL(description_), \
1459  .priv_class = &priv_class_##_class, \
1460  .priv_size = sizeof(BiquadsContext), \
1461  .init = name_##_init, \
1462  .activate = activate, \
1463  .uninit = uninit, \
1464  FILTER_INPUTS(ff_audio_default_filterpad), \
1465  FILTER_OUTPUTS(outputs), \
1466  FILTER_QUERY_FUNC2(query_formats), \
1467  .process_command = process_command, \
1468  .flags = AVFILTER_FLAG_SLICE_THREADS | AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL, \
1469 }
1470 
1471 #define DEFINE_BIQUAD_FILTER(name, description) \
1472  AVFILTER_DEFINE_CLASS(name); \
1473  DEFINE_BIQUAD_FILTER_2(name, description, name)
1474 
1475 #define WIDTH_OPTION(x) \
1476  {"width", "set width", OFFSET(width), AV_OPT_TYPE_DOUBLE, {.dbl=x}, 0, 99999, FLAGS}, \
1477  {"w", "set width", OFFSET(width), AV_OPT_TYPE_DOUBLE, {.dbl=x}, 0, 99999, FLAGS}
1478 
1479 #define WIDTH_TYPE_OPTION(x) \
1480  {"width_type", "set filter-width type", OFFSET(width_type), AV_OPT_TYPE_INT, {.i64=x}, HERTZ, NB_WTYPE-1, FLAGS, .unit = "width_type"}, \
1481  {"t", "set filter-width type", OFFSET(width_type), AV_OPT_TYPE_INT, {.i64=x}, HERTZ, NB_WTYPE-1, FLAGS, .unit = "width_type"}, \
1482  {"h", "Hz", 0, AV_OPT_TYPE_CONST, {.i64=HERTZ}, 0, 0, FLAGS, .unit = "width_type"}, \
1483  {"q", "Q-Factor", 0, AV_OPT_TYPE_CONST, {.i64=QFACTOR}, 0, 0, FLAGS, .unit = "width_type"}, \
1484  {"o", "octave", 0, AV_OPT_TYPE_CONST, {.i64=OCTAVE}, 0, 0, FLAGS, .unit = "width_type"}, \
1485  {"s", "slope", 0, AV_OPT_TYPE_CONST, {.i64=SLOPE}, 0, 0, FLAGS, .unit = "width_type"}, \
1486  {"k", "kHz", 0, AV_OPT_TYPE_CONST, {.i64=KHERTZ}, 0, 0, FLAGS, .unit = "width_type"}
1487 
1488 #define MIX_CHANNELS_NORMALIZE_OPTION(x, y, z) \
1489  {"mix", "set mix", OFFSET(mix), AV_OPT_TYPE_DOUBLE, {.dbl=x}, 0, 1, FLAGS}, \
1490  {"m", "set mix", OFFSET(mix), AV_OPT_TYPE_DOUBLE, {.dbl=x}, 0, 1, FLAGS}, \
1491  {"channels", "set channels to filter", OFFSET(ch_layout_str), AV_OPT_TYPE_STRING, {.str=y}, 0, 0, FLAGS}, \
1492  {"c", "set channels to filter", OFFSET(ch_layout_str), AV_OPT_TYPE_STRING, {.str=y}, 0, 0, FLAGS}, \
1493  {"normalize", "normalize coefficients", OFFSET(normalize), AV_OPT_TYPE_BOOL, {.i64=z}, 0, 1, FLAGS}, \
1494  {"n", "normalize coefficients", OFFSET(normalize), AV_OPT_TYPE_BOOL, {.i64=z}, 0, 1, FLAGS}
1495 
1496 #define TRANSFORM_OPTION(x) \
1497  {"transform", "set transform type", OFFSET(transform_type), AV_OPT_TYPE_INT, {.i64=x}, 0, NB_TTYPE-1, AF, .unit = "transform_type"}, \
1498  {"a", "set transform type", OFFSET(transform_type), AV_OPT_TYPE_INT, {.i64=x}, 0, NB_TTYPE-1, AF, .unit = "transform_type"}, \
1499  {"di", "direct form I", 0, AV_OPT_TYPE_CONST, {.i64=DI}, 0, 0, AF, .unit = "transform_type"}, \
1500  {"dii", "direct form II", 0, AV_OPT_TYPE_CONST, {.i64=DII}, 0, 0, AF, .unit = "transform_type"}, \
1501  {"tdi", "transposed direct form I", 0, AV_OPT_TYPE_CONST, {.i64=TDI}, 0, 0, AF, .unit = "transform_type"}, \
1502  {"tdii", "transposed direct form II", 0, AV_OPT_TYPE_CONST, {.i64=TDII}, 0, 0, AF, .unit = "transform_type"}, \
1503  {"latt", "lattice-ladder form", 0, AV_OPT_TYPE_CONST, {.i64=LATT}, 0, 0, AF, .unit = "transform_type"}, \
1504  {"svf", "state variable filter form", 0, AV_OPT_TYPE_CONST, {.i64=SVF}, 0, 0, AF, .unit = "transform_type"}, \
1505  {"zdf", "zero-delay filter form", 0, AV_OPT_TYPE_CONST, {.i64=ZDF}, 0, 0, AF, .unit = "transform_type"}
1506 
1507 #define PRECISION_OPTION(x) \
1508  {"precision", "set filtering precision", OFFSET(precision), AV_OPT_TYPE_INT, {.i64=x}, -1, 3, AF, .unit = "precision"}, \
1509  {"r", "set filtering precision", OFFSET(precision), AV_OPT_TYPE_INT, {.i64=x}, -1, 3, AF, .unit = "precision"}, \
1510  {"auto", "automatic", 0, AV_OPT_TYPE_CONST, {.i64=-1}, 0, 0, AF, .unit = "precision"}, \
1511  {"s16", "signed 16-bit", 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, AF, .unit = "precision"}, \
1512  {"s32", "signed 32-bit", 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, AF, .unit = "precision"}, \
1513  {"f32", "floating-point single", 0, AV_OPT_TYPE_CONST, {.i64=2}, 0, 0, AF, .unit = "precision"}, \
1514  {"f64", "floating-point double", 0, AV_OPT_TYPE_CONST, {.i64=3}, 0, 0, AF, .unit = "precision"}
1515 
1516 #define BLOCKSIZE_OPTION(x) \
1517  {"blocksize", "set the block size", OFFSET(block_samples), AV_OPT_TYPE_INT, {.i64=x}, 0, 32768, AF}, \
1518  {"b", "set the block size", OFFSET(block_samples), AV_OPT_TYPE_INT, {.i64=x}, 0, 32768, AF}
1519 
1520 #if CONFIG_EQUALIZER_FILTER
1521 static const AVOption equalizer_options[] = {
1522  {"frequency", "set central frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=0}, 0, 999999, FLAGS},
1523  {"f", "set central frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=0}, 0, 999999, FLAGS},
1524  WIDTH_TYPE_OPTION(QFACTOR),
1525  WIDTH_OPTION(1.0),
1526  {"gain", "set gain", OFFSET(gain), AV_OPT_TYPE_DOUBLE, {.dbl=0}, -900, 900, FLAGS},
1527  {"g", "set gain", OFFSET(gain), AV_OPT_TYPE_DOUBLE, {.dbl=0}, -900, 900, FLAGS},
1528  MIX_CHANNELS_NORMALIZE_OPTION(1, "all", 0),
1529  TRANSFORM_OPTION(DI),
1530  PRECISION_OPTION(-1),
1531  BLOCKSIZE_OPTION(0),
1532  {NULL}
1533 };
1534 
1535 DEFINE_BIQUAD_FILTER(equalizer, "Apply two-pole peaking equalization (EQ) filter.");
1536 #endif /* CONFIG_EQUALIZER_FILTER */
1537 #if CONFIG_BASS_FILTER || CONFIG_LOWSHELF_FILTER
1538 static const AVOption bass_lowshelf_options[] = {
1539  {"frequency", "set central frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=100}, 0, 999999, FLAGS},
1540  {"f", "set central frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=100}, 0, 999999, FLAGS},
1541  WIDTH_TYPE_OPTION(QFACTOR),
1542  WIDTH_OPTION(0.5),
1543  {"gain", "set gain", OFFSET(gain), AV_OPT_TYPE_DOUBLE, {.dbl=0}, -900, 900, FLAGS},
1544  {"g", "set gain", OFFSET(gain), AV_OPT_TYPE_DOUBLE, {.dbl=0}, -900, 900, FLAGS},
1545  {"poles", "set number of poles", OFFSET(poles), AV_OPT_TYPE_INT, {.i64=2}, 1, 2, AF},
1546  {"p", "set number of poles", OFFSET(poles), AV_OPT_TYPE_INT, {.i64=2}, 1, 2, AF},
1547  MIX_CHANNELS_NORMALIZE_OPTION(1, "all", 0),
1548  TRANSFORM_OPTION(DI),
1549  PRECISION_OPTION(-1),
1550  BLOCKSIZE_OPTION(0),
1551  {NULL}
1552 };
1553 
1554 AVFILTER_DEFINE_CLASS_EXT(bass_lowshelf, "bass/lowshelf", bass_lowshelf_options);
1555 #if CONFIG_BASS_FILTER
1556 DEFINE_BIQUAD_FILTER_2(bass, "Boost or cut lower frequencies.", bass_lowshelf);
1557 #endif /* CONFIG_BASS_FILTER */
1558 
1559 #if CONFIG_LOWSHELF_FILTER
1560 DEFINE_BIQUAD_FILTER_2(lowshelf, "Apply a low shelf filter.", bass_lowshelf);
1561 #endif /* CONFIG_LOWSHELF_FILTER */
1562 #endif /* CONFIG_BASS_FILTER || CONFIG LOWSHELF_FILTER */
1563 #if CONFIG_TREBLE_FILTER || CONFIG_HIGHSHELF_FILTER || CONFIG_TILTSHELF_FILTER
1564 static const AVOption treble_highshelf_options[] = {
1565  {"frequency", "set central frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=3000}, 0, 999999, FLAGS},
1566  {"f", "set central frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=3000}, 0, 999999, FLAGS},
1567  WIDTH_TYPE_OPTION(QFACTOR),
1568  WIDTH_OPTION(0.5),
1569  {"gain", "set gain", OFFSET(gain), AV_OPT_TYPE_DOUBLE, {.dbl=0}, -900, 900, FLAGS},
1570  {"g", "set gain", OFFSET(gain), AV_OPT_TYPE_DOUBLE, {.dbl=0}, -900, 900, FLAGS},
1571  {"poles", "set number of poles", OFFSET(poles), AV_OPT_TYPE_INT, {.i64=2}, 1, 2, AF},
1572  {"p", "set number of poles", OFFSET(poles), AV_OPT_TYPE_INT, {.i64=2}, 1, 2, AF},
1573  MIX_CHANNELS_NORMALIZE_OPTION(1, "all", 0),
1574  TRANSFORM_OPTION(DI),
1575  PRECISION_OPTION(-1),
1576  BLOCKSIZE_OPTION(0),
1577  {NULL}
1578 };
1579 
1580 AVFILTER_DEFINE_CLASS_EXT(treble_highshelf, "treble/high/tiltshelf",
1581  treble_highshelf_options);
1582 
1583 #if CONFIG_TREBLE_FILTER
1584 DEFINE_BIQUAD_FILTER_2(treble, "Boost or cut upper frequencies.", treble_highshelf);
1585 #endif /* CONFIG_TREBLE_FILTER */
1586 
1587 #if CONFIG_HIGHSHELF_FILTER
1588 DEFINE_BIQUAD_FILTER_2(highshelf, "Apply a high shelf filter.", treble_highshelf);
1589 #endif /* CONFIG_HIGHSHELF_FILTER */
1590 
1591 #if CONFIG_TILTSHELF_FILTER
1592 DEFINE_BIQUAD_FILTER_2(tiltshelf, "Apply a tilt shelf filter.", treble_highshelf);
1593 #endif
1594 #endif /* CONFIG_TREBLE_FILTER || CONFIG_HIGHSHELF_FILTER || CONFIG_TILTSHELF_FILTER */
1595 
1596 #if CONFIG_BANDPASS_FILTER
1597 static const AVOption bandpass_options[] = {
1598  {"frequency", "set central frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=3000}, 0, 999999, FLAGS},
1599  {"f", "set central frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=3000}, 0, 999999, FLAGS},
1600  WIDTH_TYPE_OPTION(QFACTOR),
1601  WIDTH_OPTION(0.5),
1602  {"csg", "use constant skirt gain", OFFSET(csg), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS},
1603  MIX_CHANNELS_NORMALIZE_OPTION(1, "all", 0),
1604  TRANSFORM_OPTION(DI),
1605  PRECISION_OPTION(-1),
1606  BLOCKSIZE_OPTION(0),
1607  {NULL}
1608 };
1609 
1610 DEFINE_BIQUAD_FILTER(bandpass, "Apply a two-pole Butterworth band-pass filter.");
1611 #endif /* CONFIG_BANDPASS_FILTER */
1612 #if CONFIG_BANDREJECT_FILTER
1613 static const AVOption bandreject_options[] = {
1614  {"frequency", "set central frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=3000}, 0, 999999, FLAGS},
1615  {"f", "set central frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=3000}, 0, 999999, FLAGS},
1616  WIDTH_TYPE_OPTION(QFACTOR),
1617  WIDTH_OPTION(0.5),
1618  MIX_CHANNELS_NORMALIZE_OPTION(1, "all", 0),
1619  TRANSFORM_OPTION(DI),
1620  PRECISION_OPTION(-1),
1621  BLOCKSIZE_OPTION(0),
1622  {NULL}
1623 };
1624 
1625 DEFINE_BIQUAD_FILTER(bandreject, "Apply a two-pole Butterworth band-reject filter.");
1626 #endif /* CONFIG_BANDREJECT_FILTER */
1627 #if CONFIG_LOWPASS_FILTER
1628 static const AVOption lowpass_options[] = {
1629  {"frequency", "set frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=500}, 0, 999999, FLAGS},
1630  {"f", "set frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=500}, 0, 999999, FLAGS},
1631  WIDTH_TYPE_OPTION(QFACTOR),
1632  WIDTH_OPTION(0.707),
1633  {"poles", "set number of poles", OFFSET(poles), AV_OPT_TYPE_INT, {.i64=2}, 1, 2, AF},
1634  {"p", "set number of poles", OFFSET(poles), AV_OPT_TYPE_INT, {.i64=2}, 1, 2, AF},
1635  MIX_CHANNELS_NORMALIZE_OPTION(1, "all", 0),
1636  TRANSFORM_OPTION(DI),
1637  PRECISION_OPTION(-1),
1638  BLOCKSIZE_OPTION(0),
1639  {NULL}
1640 };
1641 
1642 DEFINE_BIQUAD_FILTER(lowpass, "Apply a low-pass filter with 3dB point frequency.");
1643 #endif /* CONFIG_LOWPASS_FILTER */
1644 #if CONFIG_HIGHPASS_FILTER
1645 static const AVOption highpass_options[] = {
1646  {"frequency", "set frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=3000}, 0, 999999, FLAGS},
1647  {"f", "set frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=3000}, 0, 999999, FLAGS},
1648  WIDTH_TYPE_OPTION(QFACTOR),
1649  WIDTH_OPTION(0.707),
1650  {"poles", "set number of poles", OFFSET(poles), AV_OPT_TYPE_INT, {.i64=2}, 1, 2, AF},
1651  {"p", "set number of poles", OFFSET(poles), AV_OPT_TYPE_INT, {.i64=2}, 1, 2, AF},
1652  MIX_CHANNELS_NORMALIZE_OPTION(1, "all", 0),
1653  TRANSFORM_OPTION(DI),
1654  PRECISION_OPTION(-1),
1655  BLOCKSIZE_OPTION(0),
1656  {NULL}
1657 };
1658 
1659 DEFINE_BIQUAD_FILTER(highpass, "Apply a high-pass filter with 3dB point frequency.");
1660 #endif /* CONFIG_HIGHPASS_FILTER */
1661 #if CONFIG_ALLPASS_FILTER
1662 static const AVOption allpass_options[] = {
1663  {"frequency", "set central frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=3000}, 0, 999999, FLAGS},
1664  {"f", "set central frequency", OFFSET(frequency), AV_OPT_TYPE_DOUBLE, {.dbl=3000}, 0, 999999, FLAGS},
1665  WIDTH_TYPE_OPTION(QFACTOR),
1666  WIDTH_OPTION(0.707),
1667  MIX_CHANNELS_NORMALIZE_OPTION(1, "all", 0),
1668  {"order", "set filter order", OFFSET(order), AV_OPT_TYPE_INT, {.i64=2}, 1, 2, FLAGS},
1669  {"o", "set filter order", OFFSET(order), AV_OPT_TYPE_INT, {.i64=2}, 1, 2, FLAGS},
1670  TRANSFORM_OPTION(DI),
1671  PRECISION_OPTION(-1),
1672  {NULL}
1673 };
1674 
1675 DEFINE_BIQUAD_FILTER(allpass, "Apply a two-pole all-pass filter.");
1676 #endif /* CONFIG_ALLPASS_FILTER */
1677 #if CONFIG_BIQUAD_FILTER
1678 static const AVOption biquad_options[] = {
1679  {"a0", NULL, OFFSET(oa[0]), AV_OPT_TYPE_DOUBLE, {.dbl=1}, INT32_MIN, INT32_MAX, FLAGS},
1680  {"a1", NULL, OFFSET(oa[1]), AV_OPT_TYPE_DOUBLE, {.dbl=0}, INT32_MIN, INT32_MAX, FLAGS},
1681  {"a2", NULL, OFFSET(oa[2]), AV_OPT_TYPE_DOUBLE, {.dbl=0}, INT32_MIN, INT32_MAX, FLAGS},
1682  {"b0", NULL, OFFSET(ob[0]), AV_OPT_TYPE_DOUBLE, {.dbl=0}, INT32_MIN, INT32_MAX, FLAGS},
1683  {"b1", NULL, OFFSET(ob[1]), AV_OPT_TYPE_DOUBLE, {.dbl=0}, INT32_MIN, INT32_MAX, FLAGS},
1684  {"b2", NULL, OFFSET(ob[2]), AV_OPT_TYPE_DOUBLE, {.dbl=0}, INT32_MIN, INT32_MAX, FLAGS},
1685  MIX_CHANNELS_NORMALIZE_OPTION(1, "all", 0),
1686  TRANSFORM_OPTION(DI),
1687  PRECISION_OPTION(-1),
1688  BLOCKSIZE_OPTION(0),
1689  {NULL}
1690 };
1691 
1692 DEFINE_BIQUAD_FILTER(biquad, "Apply a biquad IIR filter with the given coefficients.");
1693 #endif /* CONFIG_BIQUAD_FILTER */
A
#define A(x)
Definition: vpx_arith.h:28
BLOCKSIZE_OPTION
#define BLOCKSIZE_OPTION(x)
Definition: af_biquads.c:1516
av_samples_copy
int av_samples_copy(uint8_t *const *dst, uint8_t *const *src, int dst_offset, int src_offset, int nb_samples, int nb_channels, enum AVSampleFormat sample_fmt)
Copy samples from src to dst.
Definition: samplefmt.c:222
ff_get_audio_buffer
AVFrame * ff_get_audio_buffer(AVFilterLink *link, int nb_samples)
Request an audio samples buffer with a specific set of permissions.
Definition: audio.c:98
AV_SAMPLE_FMT_FLTP
@ AV_SAMPLE_FMT_FLTP
float, planar
Definition: samplefmt.h:66
AV_LOG_WARNING
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:215
ff_exp10
static av_always_inline double ff_exp10(double x)
Compute 10^x for floating point values.
Definition: ffmath.h:42
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
opt.h
BiquadsContext::normalize
int normalize
Definition: af_biquads.c:132
out
FILE * out
Definition: movenc.c:55
DII
@ DII
Definition: af_biquads.c:104
BiquadsContext
Definition: af_biquads.c:113
BiquadsContext::block_align
int block_align
Definition: af_biquads.c:148
BiquadsContext::width_type
int width_type
Definition: af_biquads.c:117
ff_filter_frame
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:1062
sample_fmts
static enum AVSampleFormat sample_fmts[]
Definition: adpcmenc.c:948
FFERROR_NOT_READY
return FFERROR_NOT_READY
Definition: filter_design.txt:204
BIQUAD_LATT_FILTER
#define BIQUAD_LATT_FILTER(name, type, ftype, min, max, need_clipping)
Definition: af_biquads.c:436
NB_TTYPE
@ NB_TTYPE
Definition: af_biquads.c:110
DEFINE_BIQUAD_FILTER
#define DEFINE_BIQUAD_FILTER(name, description)
Definition: af_biquads.c:1471
int64_t
long long int64_t
Definition: coverity.c:34
inlink
The exact code depends on how similar the blocks are and how related they are to the and needs to apply these operations to the correct inlink or outlink if there are several Macros are available to factor that when no extra processing is inlink
Definition: filter_design.txt:212
SLOPE
@ SLOPE
Definition: af_biquads.c:97
normalize.log
log
Definition: normalize.py:21
av_frame_free
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:162
AVFILTER_DEFINE_CLASS_EXT
#define AVFILTER_DEFINE_CLASS_EXT(name, desc, options)
Definition: filters.h:265
reverse_samples
static void reverse_samples(AVFrame *out, AVFrame *in, int p, int oo, int io, int nb_samples)
Definition: af_biquads.c:1203
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:389
AVFrame::pts
int64_t pts
Presentation timestamp in time_base units (time when frame should be shown to user).
Definition: frame.h:501
av_channel_layout_channel_from_index
enum AVChannel av_channel_layout_channel_from_index(const AVChannelLayout *channel_layout, unsigned int idx)
Get the channel with the given index in a channel layout.
Definition: channel_layout.c:671
av_samples_set_silence
int av_samples_set_silence(uint8_t *const *audio_data, int offset, int nb_samples, int nb_channels, enum AVSampleFormat sample_fmt)
Fill an audio buffer with silence.
Definition: samplefmt.c:246
AVOption
AVOption.
Definition: opt.h:429
b
#define b
Definition: input.c:41
BiquadsContext::csg
int csg
Definition: af_biquads.c:119
AV_SAMPLE_FMT_S32P
@ AV_SAMPLE_FMT_S32P
signed 32 bits, planar
Definition: samplefmt.h:65
AV_LOG_VERBOSE
#define AV_LOG_VERBOSE
Detailed information.
Definition: log.h:225
bandreject
@ bandreject
Definition: af_biquads.c:83
BiquadsContext::block_samples
int block_samples
Definition: af_biquads.c:122
BiquadsContext::a_float
float a_float[3]
Definition: af_biquads.c:138
ThreadData::out
AVFrame * out
Definition: af_adeclick.c:526
BiquadsContext::block
AVFrame * block[3]
Definition: af_biquads.c:144
AVChannelLayout::nb_channels
int nb_channels
Number of channels in this layout.
Definition: channel_layout.h:327
ThreadData::in
AVFrame * in
Definition: af_adecorrelate.c:155
FF_FILTER_FORWARD_STATUS_BACK
#define FF_FILTER_FORWARD_STATUS_BACK(outlink, inlink)
Forward the status on an output link to an input link.
Definition: filters.h:434
formats.h
ff_inlink_consume_frame
int ff_inlink_consume_frame(AVFilterLink *link, AVFrame **rframe)
Take a frame from the link's FIFO and update the link's stats.
Definition: avfilter.c:1491
convert_dir2latt
static void convert_dir2latt(BiquadsContext *s)
Definition: af_biquads.c:602
query_formats
static int query_formats(const AVFilterContext *ctx, AVFilterFormatsConfig **cfg_in, AVFilterFormatsConfig **cfg_out)
Definition: af_biquads.c:157
BiquadsContext::b_float
float b_float[3]
Definition: af_biquads.c:139
outputs
static const AVFilterPad outputs[]
Definition: af_biquads.c:1435
FFSIGN
#define FFSIGN(a)
Definition: common.h:75
AVFrame::ch_layout
AVChannelLayout ch_layout
Channel layout of the audio data.
Definition: frame.h:790
BiquadsContext::transform_type
int transform_type
Definition: af_biquads.c:120
pts
static int64_t pts
Definition: transcode_aac.c:644
BiquadsContext::ch_layout
AVChannelLayout ch_layout
Definition: af_biquads.c:131
AVFilterPad
A filter pad used for either input or output.
Definition: filters.h:38
BiquadsContext::bypass
int bypass
Definition: af_biquads.c:124
ZDF
@ ZDF
Definition: af_biquads.c:109
FilterType
FilterType
Definition: af_adenorm.c:26
avassert.h
BiquadsContext::ob
double ob[3]
Definition: af_biquads.c:142
activate
static int activate(AVFilterContext *ctx)
Definition: af_biquads.c:1364
av_cold
#define av_cold
Definition: attributes.h:90
convert_dir2zdf
static void convert_dir2zdf(BiquadsContext *s, int sample_rate)
Definition: af_biquads.c:672
ff_outlink_set_status
static void ff_outlink_set_status(AVFilterLink *link, int status, int64_t pts)
Set the status field of a link from the source filter.
Definition: filters.h:424
s
#define s(width, name)
Definition: cbs_vp9.c:198
BIQUAD_ZDF_FILTER
#define BIQUAD_ZDF_FILTER(name, type, ftype, min, max, need_clipping, two)
Definition: af_biquads.c:547
TransformType
TransformType
Definition: webp.c:113
BiquadsContext::ch_layout_str
char * ch_layout_str
Definition: af_biquads.c:130
BIQUAD_TDI_FILTER
#define BIQUAD_TDI_FILTER(name, type, ftype, min, max, need_clipping)
Definition: af_biquads.c:331
BIQUAD_TDII_FILTER
#define BIQUAD_TDII_FILTER(name, type, ftype, min, max, need_clipping)
Definition: af_biquads.c:388
convert_dir2svf
static void convert_dir2svf(BiquadsContext *s)
Definition: af_biquads.c:619
g
const char * g
Definition: vf_curves.c:128
AV_OPT_TYPE_DOUBLE
@ AV_OPT_TYPE_DOUBLE
Underlying C type is double.
Definition: opt.h:267
AVMEDIA_TYPE_AUDIO
@ AVMEDIA_TYPE_AUDIO
Definition: avutil.h:202
av_assert0
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:40
highpass
@ highpass
Definition: af_biquads.c:85
filters.h
BiquadsContext::precision
int precision
Definition: af_biquads.c:121
NB_WTYPE
@ NB_WTYPE
Definition: af_biquads.c:99
ctx
AVFormatContext * ctx
Definition: movenc.c:49
BiquadsContext::width
double width
Definition: af_biquads.c:128
BiquadsContext::poles
int poles
Definition: af_biquads.c:118
BiquadsContext::a_double
double a_double[3]
Definition: af_biquads.c:135
arg
const char * arg
Definition: jacosubdec.c:67
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:75
fabs
static __device__ float fabs(float a)
Definition: cuda_runtime.h:182
ff_inlink_consume_samples
int ff_inlink_consume_samples(AVFilterLink *link, unsigned min, unsigned max, AVFrame **rframe)
Take samples from the link's FIFO and update the link's stats.
Definition: avfilter.c:1511
NULL
#define NULL
Definition: coverity.c:32
biquad
@ biquad
Definition: af_biquads.c:78
av_frame_copy_props
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
Definition: frame.c:713
filter_frame
static int filter_frame(AVFilterLink *inlink, AVFrame *buf, int eof)
Definition: af_biquads.c:1295
allpass
@ allpass
Definition: af_biquads.c:84
WIDTH_OPTION
#define WIDTH_OPTION(x)
Definition: af_biquads.c:1475
BIQUAD_FILTER
#define BIQUAD_FILTER(name, type, ftype, min, max, need_clipping)
Definition: af_biquads.c:200
MIX_CHANNELS_NORMALIZE_OPTION
#define MIX_CHANNELS_NORMALIZE_OPTION(x, y, z)
Definition: af_biquads.c:1488
exp
int8_t exp
Definition: eval.c:73
ff_inlink_acknowledge_status
int ff_inlink_acknowledge_status(AVFilterLink *link, int *rstatus, int64_t *rpts)
Test and acknowledge the change of status on the link.
Definition: avfilter.c:1438
process_command
static int process_command(AVFilterContext *ctx, const char *cmd, const char *args, char *res, int res_len, int flags)
Definition: af_biquads.c:1410
AVFilterFormatsConfig
Lists of formats / etc.
Definition: avfilter.h:111
BiquadsContext::order
int order
Definition: af_biquads.c:133
KHERTZ
@ KHERTZ
Definition: af_biquads.c:98
lowpass
@ lowpass
Definition: af_biquads.c:86
AF
#define AF
Definition: af_biquads.c:1445
f
f
Definition: af_crystalizer.c:122
filter_channel
static int filter_channel(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
Definition: af_biquads.c:1238
AVChannelLayout
An AVChannelLayout holds information about the channel layout of audio data.
Definition: channel_layout.h:317
NONE
@ NONE
Definition: af_biquads.c:93
dst
uint8_t ptrdiff_t const uint8_t ptrdiff_t int intptr_t intptr_t int int16_t * dst
Definition: dsp.h:83
AV_SAMPLE_FMT_NONE
@ AV_SAMPLE_FMT_NONE
Definition: samplefmt.h:56
highshelf
@ highshelf
Definition: af_biquads.c:88
AV_NOPTS_VALUE
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:248
av_frame_is_writable
int av_frame_is_writable(AVFrame *frame)
Check if the frame data is writable.
Definition: frame.c:649
bass
@ bass
Definition: af_biquads.c:80
BiquadsContext::cache
AVFrame * cache[2]
Definition: af_biquads.c:147
ff_filter_process_command
int ff_filter_process_command(AVFilterContext *ctx, const char *cmd, const char *arg, char *res, int res_len, int flags)
Generic processing of user supplied commands that are set in the same way as the filter options.
Definition: avfilter.c:901
BIQUAD_SVF_FILTER
#define BIQUAD_SVF_FILTER(name, type, ftype, min, max, need_clipping)
Definition: af_biquads.c:495
config_output
static int config_output(AVFilterLink *outlink)
Definition: af_biquads.c:1193
a
The reader does not expect b to be semantically here and if the code is changed by maybe adding a a division or other the signedness will almost certainly be mistaken To avoid this confusion a new type was SUINT is the C unsigned type but it holds a signed int to use the same example SUINT a
Definition: undefined.txt:41
BiquadsContext::nb_samples
int nb_samples
Definition: af_biquads.c:151
FF_FILTER_FORWARD_WANTED
FF_FILTER_FORWARD_WANTED(outlink, inlink)
M_PI
#define M_PI
Definition: mathematics.h:67
AV_SAMPLE_FMT_S16P
@ AV_SAMPLE_FMT_S16P
signed 16 bits, planar
Definition: samplefmt.h:64
LATT
@ LATT
Definition: af_biquads.c:107
BiquadsContext::mix
double mix
Definition: af_biquads.c:129
QFACTOR
@ QFACTOR
Definition: af_biquads.c:96
AVChannel
AVChannel
Definition: channel_layout.h:47
av_channel_layout_from_string
int av_channel_layout_from_string(AVChannelLayout *channel_layout, const char *str)
Initialize a channel layout from a given string description.
Definition: channel_layout.c:310
AVFrame::nb_samples
int nb_samples
number of audio samples (per channel) described by this frame
Definition: frame.h:469
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:256
av_get_bytes_per_sample
int av_get_bytes_per_sample(enum AVSampleFormat sample_fmt)
Return number of bytes per sample.
Definition: samplefmt.c:108
BiquadsContext::oa
double oa[3]
Definition: af_biquads.c:141
AVFrame::extended_data
uint8_t ** extended_data
pointers to the data planes/channels.
Definition: frame.h:450
ff_filter_get_nb_threads
int ff_filter_get_nb_threads(AVFilterContext *ctx)
Get number of threads for current filter instance.
Definition: avfilter.c:841
AVSampleFormat
AVSampleFormat
Audio sample formats.
Definition: samplefmt.h:55
M_PI_4
#define M_PI_4
Definition: mathematics.h:79
ThreadData
Used for passing data between threads.
Definition: dsddec.c:71
FFMIN
#define FFMIN(a, b)
Definition: macros.h:49
TDII
@ TDII
Definition: af_biquads.c:106
config_filter
static int config_filter(AVFilterLink *outlink, int reset)
Definition: af_biquads.c:796
tiltshelf
@ tiltshelf
Definition: af_biquads.c:89
len
int len
Definition: vorbis_enc_data.h:426
AVFilterPad::name
const char * name
Pad name.
Definition: filters.h:44
ff_inlink_queued_samples
int ff_inlink_queued_samples(AVFilterLink *link)
Definition: avfilter.c:1466
av_calloc
void * av_calloc(size_t nmemb, size_t size)
Definition: mem.c:264
ThreadData::eof
int eof
Definition: af_biquads.c:1200
PRECISION_OPTION
#define PRECISION_OPTION(x)
Definition: af_biquads.c:1507
TRANSFORM_OPTION
#define TRANSFORM_OPTION(x)
Definition: af_biquads.c:1496
treble
@ treble
Definition: af_biquads.c:81
OFFSET
#define OFFSET(x)
Definition: af_biquads.c:1443
ret
ret
Definition: filter_design.txt:187
BIQUAD_DII_FILTER
#define BIQUAD_DII_FILTER(name, type, ftype, min, max, need_clipping)
Definition: af_biquads.c:282
WidthType
WidthType
Definition: af_biquads.c:92
BiquadsContext::filter
void(* filter)(struct BiquadsContext *s, const void *ibuf, void *obuf, int len, void *cache, int *clip, int disabled)
Definition: af_biquads.c:153
BiquadsContext::filter_type
enum FilterType filter_type
Definition: af_biquads.c:116
status
ov_status_e status
Definition: dnn_backend_openvino.c:100
ff_set_common_formats_from_list2
int ff_set_common_formats_from_list2(const AVFilterContext *ctx, AVFilterFormatsConfig **cfg_in, AVFilterFormatsConfig **cfg_out, const int *fmts)
Definition: formats.c:1016
channel_layout.h
ff_filter_execute
int ff_filter_execute(AVFilterContext *ctx, avfilter_action_func *func, void *arg, int *ret, int nb_jobs)
Definition: avfilter.c:1667
HERTZ
@ HERTZ
Definition: af_biquads.c:94
av_channel_layout_index_from_channel
int av_channel_layout_index_from_channel(const AVChannelLayout *channel_layout, enum AVChannel channel)
Get the index of a given channel in a channel layout.
Definition: channel_layout.c:711
BiquadsContext::gain
double gain
Definition: af_biquads.c:126
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Underlying C type is int.
Definition: opt.h:259
avfilter.h
av_channel_layout_uninit
void av_channel_layout_uninit(AVChannelLayout *channel_layout)
Free any allocated data in the channel layout and reset the channel count to 0.
Definition: channel_layout.c:440
AV_SAMPLE_FMT_DBLP
@ AV_SAMPLE_FMT_DBLP
double, planar
Definition: samplefmt.h:67
DEFINE_BIQUAD_FILTER_2
#define DEFINE_BIQUAD_FILTER_2(name_, description_, priv_class_)
Definition: af_biquads.c:1447
WIDTH_TYPE_OPTION
#define WIDTH_TYPE_OPTION(x)
Definition: af_biquads.c:1479
uninit
static av_cold void uninit(AVFilterContext *ctx)
Definition: af_biquads.c:1423
ffmath.h
AVFilterContext
An instance of a filter.
Definition: avfilter.h:457
factor
static const int factor[16]
Definition: vf_pp7.c:80
av_channel_layout_copy
int av_channel_layout_copy(AVChannelLayout *dst, const AVChannelLayout *src)
Make a copy of a channel layout.
Definition: channel_layout.c:447
FLAGS
#define FLAGS
Definition: af_biquads.c:1444
DI
@ DI
Definition: af_biquads.c:103
mem.h
audio.h
TDI
@ TDI
Definition: af_biquads.c:105
SVF
@ SVF
Definition: af_biquads.c:108
lowshelf
@ lowshelf
Definition: af_biquads.c:87
alpha
static const int16_t alpha[]
Definition: ilbcdata.h:55
AV_OPT_TYPE_BOOL
@ AV_OPT_TYPE_BOOL
Underlying C type is int.
Definition: opt.h:327
equalizer
@ equalizer
Definition: af_biquads.c:79
av_freep
#define av_freep(p)
Definition: tableprint_vlc.h:34
convert_width2qfactor
static double convert_width2qfactor(double width, double frequency, double gain, double sample_rate, int width_type)
Definition: af_biquads.c:637
K
#define K
Definition: palette.c:25
OCTAVE
@ OCTAVE
Definition: af_biquads.c:95
int32_t
int32_t
Definition: audioconvert.c:56
Q
#define Q(x)
Definition: filter_template.c:423
flags
#define flags(name, subs,...)
Definition: cbs_av1.c:482
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
BiquadsContext::clip
int * clip
Definition: af_biquads.c:146
width
#define width
Definition: dsp.h:85
BiquadsContext::b_double
double b_double[3]
Definition: af_biquads.c:136
BiquadsContext::frequency
double frequency
Definition: af_biquads.c:127
BiquadsContext::pts
int64_t pts
Definition: af_biquads.c:150
src
#define src
Definition: vp8dsp.c:248
channel
channel
Definition: ebur128.h:39
bandpass
@ bandpass
Definition: af_biquads.c:82
ff_filter_set_ready
void ff_filter_set_ready(AVFilterContext *filter, unsigned priority)
Mark a filter ready and schedule it for activation.
Definition: avfilter.c:239
Generated on Wed Nov 20 2024 19:22:16 for FFmpeg by   doxygen 1.8.17