• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (c) 2017 Paul B Mahol
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20 
21 /**
22  * @file
23  * An arbitrary audio FIR filter
24  */
25 
26 #include <float.h>
27 
28 #include "libavutil/tx.h"
29 #include "libavutil/avstring.h"
30 #include "libavutil/channel_layout.h"
31 #include "libavutil/common.h"
32 #include "libavutil/float_dsp.h"
33 #include "libavutil/frame.h"
34 #include "libavutil/intreadwrite.h"
35 #include "libavutil/log.h"
36 #include "libavutil/opt.h"
37 #include "libavutil/rational.h"
38 #include "libavutil/xga_font_data.h"
39 
40 #include "audio.h"
41 #include "avfilter.h"
42 #include "filters.h"
43 #include "formats.h"
44 #include "internal.h"
45 #include "af_afir.h"
46 #include "af_afirdsp.h"
47 
drawtext(AVFrame * pic,int x,int y,const char * txt,uint32_t color)48 static void drawtext(AVFrame *pic, int x, int y, const char *txt, uint32_t color)
49 {
50     const uint8_t *font;
51     int font_height;
52     int i;
53 
54     font = avpriv_cga_font, font_height = 8;
55 
56     for (i = 0; txt[i]; i++) {
57         int char_y, mask;
58 
59         uint8_t *p = pic->data[0] + y * pic->linesize[0] + (x + i * 8) * 4;
60         for (char_y = 0; char_y < font_height; char_y++) {
61             for (mask = 0x80; mask; mask >>= 1) {
62                 if (font[txt[i] * font_height + char_y] & mask)
63                     AV_WL32(p, color);
64                 p += 4;
65             }
66             p += pic->linesize[0] - 8 * 4;
67         }
68     }
69 }
70 
draw_line(AVFrame * out,int x0,int y0,int x1,int y1,uint32_t color)71 static void draw_line(AVFrame *out, int x0, int y0, int x1, int y1, uint32_t color)
72 {
73     int dx = FFABS(x1-x0);
74     int dy = FFABS(y1-y0), sy = y0 < y1 ? 1 : -1;
75     int err = (dx>dy ? dx : -dy) / 2, e2;
76 
77     for (;;) {
78         AV_WL32(out->data[0] + y0 * out->linesize[0] + x0 * 4, color);
79 
80         if (x0 == x1 && y0 == y1)
81             break;
82 
83         e2 = err;
84 
85         if (e2 >-dx) {
86             err -= dy;
87             x0--;
88         }
89 
90         if (e2 < dy) {
91             err += dx;
92             y0 += sy;
93         }
94     }
95 }
96 
97 #define DEPTH 32
98 #include "afir_template.c"
99 
100 #undef DEPTH
101 #define DEPTH 64
102 #include "afir_template.c"
103 
fir_channel(AVFilterContext * ctx,AVFrame * out,int ch)104 static int fir_channel(AVFilterContext *ctx, AVFrame *out, int ch)
105 {
106     AudioFIRContext *s = ctx->priv;
107 
108     for (int offset = 0; offset < out->nb_samples; offset += s->min_part_size) {
109         switch (s->format) {
110         case AV_SAMPLE_FMT_FLTP:
111             fir_quantum_float(ctx, out, ch, offset);
112             break;
113         case AV_SAMPLE_FMT_DBLP:
114             fir_quantum_double(ctx, out, ch, offset);
115             break;
116         }
117     }
118 
119     return 0;
120 }
121 
fir_channels(AVFilterContext * ctx,void * arg,int jobnr,int nb_jobs)122 static int fir_channels(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
123 {
124     AVFrame *out = arg;
125     const int start = (out->ch_layout.nb_channels * jobnr) / nb_jobs;
126     const int end = (out->ch_layout.nb_channels * (jobnr+1)) / nb_jobs;
127 
128     for (int ch = start; ch < end; ch++) {
129         fir_channel(ctx, out, ch);
130     }
131 
132     return 0;
133 }
134 
fir_frame(AudioFIRContext * s,AVFrame * in,AVFilterLink * outlink)135 static int fir_frame(AudioFIRContext *s, AVFrame *in, AVFilterLink *outlink)
136 {
137     AVFilterContext *ctx = outlink->src;
138     AVFrame *out = NULL;
139 
140     out = ff_get_audio_buffer(outlink, in->nb_samples);
141     if (!out) {
142         av_frame_free(&in);
143         return AVERROR(ENOMEM);
144     }
145     out->pts = in->pts;
146 
147     s->in = in;
148     ff_filter_execute(ctx, fir_channels, out, NULL,
149                       FFMIN(outlink->ch_layout.nb_channels, ff_filter_get_nb_threads(ctx)));
150 
151     av_frame_free(&in);
152     s->in = NULL;
153 
154     return ff_filter_frame(outlink, out);
155 }
156 
init_segment(AVFilterContext * ctx,AudioFIRSegment * seg,int offset,int nb_partitions,int part_size)157 static int init_segment(AVFilterContext *ctx, AudioFIRSegment *seg,
158                         int offset, int nb_partitions, int part_size)
159 {
160     AudioFIRContext *s = ctx->priv;
161 
162     seg->tx  = av_calloc(ctx->inputs[0]->ch_layout.nb_channels, sizeof(*seg->tx));
163     seg->itx = av_calloc(ctx->inputs[0]->ch_layout.nb_channels, sizeof(*seg->itx));
164     if (!seg->tx || !seg->itx)
165         return AVERROR(ENOMEM);
166 
167     seg->fft_length    = part_size * 2 + 1;
168     seg->part_size     = part_size;
169     seg->block_size    = FFALIGN(seg->fft_length, 32);
170     seg->coeff_size    = FFALIGN(seg->part_size + 1, 32);
171     seg->nb_partitions = nb_partitions;
172     seg->input_size    = offset + s->min_part_size;
173     seg->input_offset  = offset;
174 
175     seg->part_index    = av_calloc(ctx->inputs[0]->ch_layout.nb_channels, sizeof(*seg->part_index));
176     seg->output_offset = av_calloc(ctx->inputs[0]->ch_layout.nb_channels, sizeof(*seg->output_offset));
177     if (!seg->part_index || !seg->output_offset)
178         return AVERROR(ENOMEM);
179 
180     for (int ch = 0; ch < ctx->inputs[0]->ch_layout.nb_channels && part_size >= 8; ch++) {
181         double dscale = 1.0, idscale = 1.0 / part_size;
182         float fscale = 1.f, ifscale = 1.f / part_size;
183 
184         switch (s->format) {
185         case AV_SAMPLE_FMT_FLTP:
186             av_tx_init(&seg->tx[ch],  &seg->tx_fn,  AV_TX_FLOAT_RDFT, 0, 2 * part_size, &fscale,  0);
187             av_tx_init(&seg->itx[ch], &seg->itx_fn, AV_TX_FLOAT_RDFT, 1, 2 * part_size, &ifscale, 0);
188             break;
189         case AV_SAMPLE_FMT_DBLP:
190             av_tx_init(&seg->tx[ch],  &seg->tx_fn,  AV_TX_DOUBLE_RDFT, 0, 2 * part_size, &dscale,  0);
191             av_tx_init(&seg->itx[ch], &seg->itx_fn, AV_TX_DOUBLE_RDFT, 1, 2 * part_size, &idscale, 0);
192             break;
193         }
194 
195         if (!seg->tx[ch] || !seg->itx[ch])
196             return AVERROR(ENOMEM);
197     }
198 
199     seg->sumin  = ff_get_audio_buffer(ctx->inputs[0], seg->fft_length);
200     seg->sumout = ff_get_audio_buffer(ctx->inputs[0], seg->fft_length);
201     seg->blockin  = ff_get_audio_buffer(ctx->inputs[0], seg->nb_partitions * seg->block_size);
202     seg->blockout = ff_get_audio_buffer(ctx->inputs[0], seg->nb_partitions * seg->block_size);
203     seg->buffer = ff_get_audio_buffer(ctx->inputs[0], seg->part_size);
204     seg->coeff  = ff_get_audio_buffer(ctx->inputs[1 + s->selir], seg->nb_partitions * seg->coeff_size * 2);
205     seg->input  = ff_get_audio_buffer(ctx->inputs[0], seg->input_size);
206     seg->output = ff_get_audio_buffer(ctx->inputs[0], seg->part_size);
207     if (!seg->buffer || !seg->sumin || !seg->sumout || !seg->blockin || !seg->blockout || !seg->coeff || !seg->input || !seg->output)
208         return AVERROR(ENOMEM);
209 
210     return 0;
211 }
212 
uninit_segment(AVFilterContext * ctx,AudioFIRSegment * seg)213 static void uninit_segment(AVFilterContext *ctx, AudioFIRSegment *seg)
214 {
215     AudioFIRContext *s = ctx->priv;
216 
217     if (seg->tx) {
218         for (int ch = 0; ch < s->nb_channels; ch++) {
219             av_tx_uninit(&seg->tx[ch]);
220         }
221     }
222     av_freep(&seg->tx);
223 
224     if (seg->itx) {
225         for (int ch = 0; ch < s->nb_channels; ch++) {
226             av_tx_uninit(&seg->itx[ch]);
227         }
228     }
229     av_freep(&seg->itx);
230 
231     av_freep(&seg->output_offset);
232     av_freep(&seg->part_index);
233 
234     av_frame_free(&seg->blockin);
235     av_frame_free(&seg->blockout);
236     av_frame_free(&seg->sumin);
237     av_frame_free(&seg->sumout);
238     av_frame_free(&seg->buffer);
239     av_frame_free(&seg->coeff);
240     av_frame_free(&seg->input);
241     av_frame_free(&seg->output);
242     seg->input_size = 0;
243 }
244 
convert_coeffs(AVFilterContext * ctx)245 static int convert_coeffs(AVFilterContext *ctx)
246 {
247     AudioFIRContext *s = ctx->priv;
248     int ret, i, cur_nb_taps;
249 
250     if (!s->nb_taps) {
251         int part_size, max_part_size;
252         int left, offset = 0;
253 
254         s->nb_taps = ff_inlink_queued_samples(ctx->inputs[1 + s->selir]);
255         if (s->nb_taps <= 0)
256             return AVERROR(EINVAL);
257 
258         if (s->minp > s->maxp) {
259             s->maxp = s->minp;
260         }
261 
262         left = s->nb_taps;
263         part_size = 1 << av_log2(s->minp);
264         max_part_size = 1 << av_log2(s->maxp);
265 
266         s->min_part_size = part_size;
267 
268         for (i = 0; left > 0; i++) {
269             int step = part_size == max_part_size ? INT_MAX : 1 + (i == 0);
270             int nb_partitions = FFMIN(step, (left + part_size - 1) / part_size);
271 
272             s->nb_segments = i + 1;
273             ret = init_segment(ctx, &s->seg[i], offset, nb_partitions, part_size);
274             if (ret < 0)
275                 return ret;
276             offset += nb_partitions * part_size;
277             left -= nb_partitions * part_size;
278             part_size *= 2;
279             part_size = FFMIN(part_size, max_part_size);
280         }
281     }
282 
283     if (!s->ir[s->selir]) {
284         ret = ff_inlink_consume_samples(ctx->inputs[1 + s->selir], s->nb_taps, s->nb_taps, &s->ir[s->selir]);
285         if (ret < 0)
286             return ret;
287         if (ret == 0)
288             return AVERROR_BUG;
289     }
290 
291     if (s->response) {
292         switch (s->format) {
293         case AV_SAMPLE_FMT_FLTP:
294             draw_response_float(ctx, s->video);
295             break;
296         case AV_SAMPLE_FMT_DBLP:
297             draw_response_double(ctx, s->video);
298             break;
299         }
300     }
301 
302     s->gain = 1;
303     cur_nb_taps = s->ir[s->selir]->nb_samples;
304 
305     switch (s->format) {
306     case AV_SAMPLE_FMT_FLTP:
307         ret = get_power_float(ctx, s, cur_nb_taps);
308         break;
309     case AV_SAMPLE_FMT_DBLP:
310         ret = get_power_double(ctx, s, cur_nb_taps);
311         break;
312     }
313 
314     if (ret < 0)
315         return ret;
316 
317     av_log(ctx, AV_LOG_DEBUG, "nb_taps: %d\n", cur_nb_taps);
318     av_log(ctx, AV_LOG_DEBUG, "nb_segments: %d\n", s->nb_segments);
319 
320     switch (s->format) {
321     case AV_SAMPLE_FMT_FLTP:
322         convert_channels_float(ctx, s);
323         break;
324     case AV_SAMPLE_FMT_DBLP:
325         convert_channels_double(ctx, s);
326         break;
327     }
328 
329     s->have_coeffs = 1;
330 
331     return 0;
332 }
333 
check_ir(AVFilterLink * link)334 static int check_ir(AVFilterLink *link)
335 {
336     AVFilterContext *ctx = link->dst;
337     AudioFIRContext *s = ctx->priv;
338     int nb_taps, max_nb_taps;
339 
340     nb_taps = ff_inlink_queued_samples(link);
341     max_nb_taps = s->max_ir_len * ctx->outputs[0]->sample_rate;
342     if (nb_taps > max_nb_taps) {
343         av_log(ctx, AV_LOG_ERROR, "Too big number of coefficients: %d > %d.\n", nb_taps, max_nb_taps);
344         return AVERROR(EINVAL);
345     }
346 
347     return 0;
348 }
349 
activate(AVFilterContext * ctx)350 static int activate(AVFilterContext *ctx)
351 {
352     AudioFIRContext *s = ctx->priv;
353     AVFilterLink *outlink = ctx->outputs[0];
354     int ret, status, available, wanted;
355     AVFrame *in = NULL;
356     int64_t pts;
357 
358     FF_FILTER_FORWARD_STATUS_BACK_ALL(ctx->outputs[0], ctx);
359     if (s->response)
360         FF_FILTER_FORWARD_STATUS_BACK_ALL(ctx->outputs[1], ctx);
361     if (!s->eof_coeffs[s->selir]) {
362         ret = check_ir(ctx->inputs[1 + s->selir]);
363         if (ret < 0)
364             return ret;
365 
366         if (ff_outlink_get_status(ctx->inputs[1 + s->selir]) == AVERROR_EOF)
367             s->eof_coeffs[s->selir] = 1;
368 
369         if (!s->eof_coeffs[s->selir]) {
370             if (ff_outlink_frame_wanted(ctx->outputs[0]))
371                 ff_inlink_request_frame(ctx->inputs[1 + s->selir]);
372             else if (s->response && ff_outlink_frame_wanted(ctx->outputs[1]))
373                 ff_inlink_request_frame(ctx->inputs[1 + s->selir]);
374             return 0;
375         }
376     }
377 
378     if (!s->have_coeffs && s->eof_coeffs[s->selir]) {
379         ret = convert_coeffs(ctx);
380         if (ret < 0)
381             return ret;
382     }
383 
384     available = ff_inlink_queued_samples(ctx->inputs[0]);
385     wanted = FFMAX(s->min_part_size, (available / s->min_part_size) * s->min_part_size);
386     ret = ff_inlink_consume_samples(ctx->inputs[0], wanted, wanted, &in);
387     if (ret > 0)
388         ret = fir_frame(s, in, outlink);
389 
390     if (ret < 0)
391         return ret;
392 
393     if (s->response && s->have_coeffs) {
394         int64_t old_pts = s->video->pts;
395         int64_t new_pts = av_rescale_q(s->pts, ctx->inputs[0]->time_base, ctx->outputs[1]->time_base);
396 
397         if (ff_outlink_frame_wanted(ctx->outputs[1]) && old_pts < new_pts) {
398             AVFrame *clone;
399             s->video->pts = new_pts;
400             clone = av_frame_clone(s->video);
401             if (!clone)
402                 return AVERROR(ENOMEM);
403             return ff_filter_frame(ctx->outputs[1], clone);
404         }
405     }
406 
407     if (ff_inlink_queued_samples(ctx->inputs[0]) >= s->min_part_size) {
408         ff_filter_set_ready(ctx, 10);
409         return 0;
410     }
411 
412     if (ff_inlink_acknowledge_status(ctx->inputs[0], &status, &pts)) {
413         if (status == AVERROR_EOF) {
414             ff_outlink_set_status(ctx->outputs[0], status, pts);
415             if (s->response)
416                 ff_outlink_set_status(ctx->outputs[1], status, pts);
417             return 0;
418         }
419     }
420 
421     if (ff_outlink_frame_wanted(ctx->outputs[0]) &&
422         !ff_outlink_get_status(ctx->inputs[0])) {
423         ff_inlink_request_frame(ctx->inputs[0]);
424         return 0;
425     }
426 
427     if (s->response &&
428         ff_outlink_frame_wanted(ctx->outputs[1]) &&
429         !ff_outlink_get_status(ctx->inputs[0])) {
430         ff_inlink_request_frame(ctx->inputs[0]);
431         return 0;
432     }
433 
434     return FFERROR_NOT_READY;
435 }
436 
query_formats(AVFilterContext * ctx)437 static int query_formats(AVFilterContext *ctx)
438 {
439     AudioFIRContext *s = ctx->priv;
440     static const enum AVSampleFormat sample_fmts[3][3] = {
441         { AV_SAMPLE_FMT_FLTP, AV_SAMPLE_FMT_DBLP, AV_SAMPLE_FMT_NONE },
442         { AV_SAMPLE_FMT_FLTP, AV_SAMPLE_FMT_NONE },
443         { AV_SAMPLE_FMT_DBLP, AV_SAMPLE_FMT_NONE },
444     };
445     static const enum AVPixelFormat pix_fmts[] = {
446         AV_PIX_FMT_RGB0,
447         AV_PIX_FMT_NONE
448     };
449     int ret;
450 
451     if (s->response) {
452         AVFilterLink *videolink = ctx->outputs[1];
453         AVFilterFormats *formats = ff_make_format_list(pix_fmts);
454         if ((ret = ff_formats_ref(formats, &videolink->incfg.formats)) < 0)
455             return ret;
456     }
457 
458     if (s->ir_format) {
459         ret = ff_set_common_all_channel_counts(ctx);
460         if (ret < 0)
461             return ret;
462     } else {
463         AVFilterChannelLayouts *mono = NULL;
464         AVFilterChannelLayouts *layouts = ff_all_channel_counts();
465 
466         if ((ret = ff_channel_layouts_ref(layouts, &ctx->inputs[0]->outcfg.channel_layouts)) < 0)
467             return ret;
468         if ((ret = ff_channel_layouts_ref(layouts, &ctx->outputs[0]->incfg.channel_layouts)) < 0)
469             return ret;
470 
471         ret = ff_add_channel_layout(&mono, &(AVChannelLayout)AV_CHANNEL_LAYOUT_MONO);
472         if (ret)
473             return ret;
474         for (int i = 1; i < ctx->nb_inputs; i++) {
475             if ((ret = ff_channel_layouts_ref(mono, &ctx->inputs[i]->outcfg.channel_layouts)) < 0)
476                 return ret;
477         }
478     }
479 
480     if ((ret = ff_set_common_formats_from_list(ctx, sample_fmts[s->precision])) < 0)
481         return ret;
482 
483     return ff_set_common_all_samplerates(ctx);
484 }
485 
config_output(AVFilterLink * outlink)486 static int config_output(AVFilterLink *outlink)
487 {
488     AVFilterContext *ctx = outlink->src;
489     AudioFIRContext *s = ctx->priv;
490     int ret;
491 
492     s->one2many = ctx->inputs[1 + s->selir]->ch_layout.nb_channels == 1;
493     outlink->sample_rate = ctx->inputs[0]->sample_rate;
494     outlink->time_base   = ctx->inputs[0]->time_base;
495 #if FF_API_OLD_CHANNEL_LAYOUT
496 FF_DISABLE_DEPRECATION_WARNINGS
497     outlink->channel_layout = ctx->inputs[0]->channel_layout;
498 FF_ENABLE_DEPRECATION_WARNINGS
499 #endif
500     if ((ret = av_channel_layout_copy(&outlink->ch_layout, &ctx->inputs[0]->ch_layout)) < 0)
501         return ret;
502     outlink->ch_layout.nb_channels = ctx->inputs[0]->ch_layout.nb_channels;
503 
504     s->nb_channels = outlink->ch_layout.nb_channels;
505     s->nb_coef_channels = ctx->inputs[1 + s->selir]->ch_layout.nb_channels;
506     s->format = outlink->format;
507 
508     return 0;
509 }
510 
uninit(AVFilterContext * ctx)511 static av_cold void uninit(AVFilterContext *ctx)
512 {
513     AudioFIRContext *s = ctx->priv;
514 
515     for (int i = 0; i < s->nb_segments; i++) {
516         uninit_segment(ctx, &s->seg[i]);
517     }
518 
519     av_freep(&s->fdsp);
520 
521     for (int i = 0; i < s->nb_irs; i++) {
522         av_frame_free(&s->ir[i]);
523     }
524 
525     av_frame_free(&s->video);
526 }
527 
config_video(AVFilterLink * outlink)528 static int config_video(AVFilterLink *outlink)
529 {
530     AVFilterContext *ctx = outlink->src;
531     AudioFIRContext *s = ctx->priv;
532 
533     outlink->sample_aspect_ratio = (AVRational){1,1};
534     outlink->w = s->w;
535     outlink->h = s->h;
536     outlink->frame_rate = s->frame_rate;
537     outlink->time_base = av_inv_q(outlink->frame_rate);
538 
539     av_frame_free(&s->video);
540     s->video = ff_get_video_buffer(outlink, outlink->w, outlink->h);
541     if (!s->video)
542         return AVERROR(ENOMEM);
543 
544     return 0;
545 }
546 
init(AVFilterContext * ctx)547 static av_cold int init(AVFilterContext *ctx)
548 {
549     AudioFIRContext *s = ctx->priv;
550     AVFilterPad pad, vpad;
551     int ret;
552 
553     pad = (AVFilterPad) {
554         .name = "main",
555         .type = AVMEDIA_TYPE_AUDIO,
556     };
557 
558     ret = ff_append_inpad(ctx, &pad);
559     if (ret < 0)
560         return ret;
561 
562     for (int n = 0; n < s->nb_irs; n++) {
563         pad = (AVFilterPad) {
564             .name = av_asprintf("ir%d", n),
565             .type = AVMEDIA_TYPE_AUDIO,
566         };
567 
568         if (!pad.name)
569             return AVERROR(ENOMEM);
570 
571         ret = ff_append_inpad_free_name(ctx, &pad);
572         if (ret < 0)
573             return ret;
574     }
575 
576     pad = (AVFilterPad) {
577         .name          = "default",
578         .type          = AVMEDIA_TYPE_AUDIO,
579         .config_props  = config_output,
580     };
581 
582     ret = ff_append_outpad(ctx, &pad);
583     if (ret < 0)
584         return ret;
585 
586     if (s->response) {
587         vpad = (AVFilterPad){
588             .name         = "filter_response",
589             .type         = AVMEDIA_TYPE_VIDEO,
590             .config_props = config_video,
591         };
592 
593         ret = ff_append_outpad(ctx, &vpad);
594         if (ret < 0)
595             return ret;
596     }
597 
598     s->fdsp = avpriv_float_dsp_alloc(0);
599     if (!s->fdsp)
600         return AVERROR(ENOMEM);
601 
602     ff_afir_init(&s->afirdsp);
603 
604     return 0;
605 }
606 
process_command(AVFilterContext * ctx,const char * cmd,const char * arg,char * res,int res_len,int flags)607 static int process_command(AVFilterContext *ctx,
608                            const char *cmd,
609                            const char *arg,
610                            char *res,
611                            int res_len,
612                            int flags)
613 {
614     AudioFIRContext *s = ctx->priv;
615     int prev_ir = s->selir;
616     int ret = ff_filter_process_command(ctx, cmd, arg, res, res_len, flags);
617 
618     if (ret < 0)
619         return ret;
620 
621     s->selir = FFMIN(s->nb_irs - 1, s->selir);
622 
623     if (prev_ir != s->selir) {
624         s->have_coeffs = 0;
625     }
626 
627     return 0;
628 }
629 
630 #define AF AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
631 #define AFR AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_RUNTIME_PARAM
632 #define VF AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
633 #define OFFSET(x) offsetof(AudioFIRContext, x)
634 
635 static const AVOption afir_options[] = {
636     { "dry",    "set dry gain",      OFFSET(dry_gain),   AV_OPT_TYPE_FLOAT, {.dbl=1},    0, 10, AF },
637     { "wet",    "set wet gain",      OFFSET(wet_gain),   AV_OPT_TYPE_FLOAT, {.dbl=1},    0, 10, AF },
638     { "length", "set IR length",     OFFSET(length),     AV_OPT_TYPE_FLOAT, {.dbl=1},    0,  1, AF },
639     { "gtype",  "set IR auto gain type",OFFSET(gtype),   AV_OPT_TYPE_INT,   {.i64=0},   -1,  2, AF, "gtype" },
640     {  "none",  "without auto gain", 0,                  AV_OPT_TYPE_CONST, {.i64=-1},   0,  0, AF, "gtype" },
641     {  "peak",  "peak gain",         0,                  AV_OPT_TYPE_CONST, {.i64=0},    0,  0, AF, "gtype" },
642     {  "dc",    "DC gain",           0,                  AV_OPT_TYPE_CONST, {.i64=1},    0,  0, AF, "gtype" },
643     {  "gn",    "gain to noise",     0,                  AV_OPT_TYPE_CONST, {.i64=2},    0,  0, AF, "gtype" },
644     { "irgain", "set IR gain",       OFFSET(ir_gain),    AV_OPT_TYPE_FLOAT, {.dbl=1},    0,  1, AF },
645     { "irfmt",  "set IR format",     OFFSET(ir_format),  AV_OPT_TYPE_INT,   {.i64=1},    0,  1, AF, "irfmt" },
646     {  "mono",  "single channel",    0,                  AV_OPT_TYPE_CONST, {.i64=0},    0,  0, AF, "irfmt" },
647     {  "input", "same as input",     0,                  AV_OPT_TYPE_CONST, {.i64=1},    0,  0, AF, "irfmt" },
648     { "maxir",  "set max IR length", OFFSET(max_ir_len), AV_OPT_TYPE_FLOAT, {.dbl=30}, 0.1, 60, AF },
649     { "response", "show IR frequency response", OFFSET(response), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, VF },
650     { "channel", "set IR channel to display frequency response", OFFSET(ir_channel), AV_OPT_TYPE_INT, {.i64=0}, 0, 1024, VF },
651     { "size",   "set video size",    OFFSET(w),          AV_OPT_TYPE_IMAGE_SIZE, {.str = "hd720"}, 0, 0, VF },
652     { "rate",   "set video rate",    OFFSET(frame_rate), AV_OPT_TYPE_VIDEO_RATE, {.str = "25"}, 0, INT32_MAX, VF },
653     { "minp",   "set min partition size", OFFSET(minp),  AV_OPT_TYPE_INT,   {.i64=8192}, 1, 32768, AF },
654     { "maxp",   "set max partition size", OFFSET(maxp),  AV_OPT_TYPE_INT,   {.i64=8192}, 8, 32768, AF },
655     { "nbirs",  "set number of input IRs",OFFSET(nb_irs),AV_OPT_TYPE_INT,   {.i64=1},    1,    32, AF },
656     { "ir",     "select IR",              OFFSET(selir), AV_OPT_TYPE_INT,   {.i64=0},    0,    31, AFR },
657     { "precision", "set processing precision",    OFFSET(precision), AV_OPT_TYPE_INT,   {.i64=0}, 0, 2, AF, "precision" },
658     {  "auto", "set auto processing precision",                   0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, AF, "precision" },
659     {  "float", "set single-floating point processing precision", 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, AF, "precision" },
660     {  "double","set double-floating point processing precision", 0, AV_OPT_TYPE_CONST, {.i64=2}, 0, 0, AF, "precision" },
661     { NULL }
662 };
663 
664 AVFILTER_DEFINE_CLASS(afir);
665 
666 const AVFilter ff_af_afir = {
667     .name          = "afir",
668     .description   = NULL_IF_CONFIG_SMALL("Apply Finite Impulse Response filter with supplied coefficients in additional stream(s)."),
669     .priv_size     = sizeof(AudioFIRContext),
670     .priv_class    = &afir_class,
671     FILTER_QUERY_FUNC(query_formats),
672     .init          = init,
673     .activate      = activate,
674     .uninit        = uninit,
675     .process_command = process_command,
676     .flags         = AVFILTER_FLAG_DYNAMIC_INPUTS  |
677                      AVFILTER_FLAG_DYNAMIC_OUTPUTS |
678                      AVFILTER_FLAG_SLICE_THREADS,
679 };
680