avfilter/af_acrossover: add option to adjust input gain

1 files changed, 20 insertions, 7 deletions

diff --git a/libavfilter/af_acrossover.c b/libavfilter/af_acrossover.c
index 33a0812c9f..ae6173f088 100644
--- a/libavfilter/af_acrossover.c
+++ b/libavfilter/af_acrossover.c
@@ -27,6 +27,7 @@
 #include "libavutil/avstring.h"
 #include "libavutil/channel_layout.h"
 #include "libavutil/eval.h"
+#include "libavutil/float_dsp.h"
 #include "libavutil/internal.h"
 #include "libavutil/opt.h"
 
@@ -55,6 +56,7 @@ typedef struct AudioCrossoverContext {
 
     char *splits_str;
     int order_opt;
+    float level_in;
 
     int order;
     int filter_count;
@@ -69,6 +71,8 @@ typedef struct AudioCrossoverContext {
     AVFrame *frames[MAX_BANDS];
 
     int (*filter_channels)(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs);
+
+    AVFloatDSPContext *fdsp;
 } AudioCrossoverContext;
 
 #define OFFSET(x) offsetof(AudioCrossoverContext, x)
@@ -87,6 +91,7 @@ static const AVOption acrossover_options[] = {
     { "16th",  "16th order",            0,                  AV_OPT_TYPE_CONST,  {.i64=7},     0, 0, AF, "m" },
     { "18th",  "18th order",            0,                  AV_OPT_TYPE_CONST,  {.i64=8},     0, 0, AF, "m" },
     { "20th",  "20th order",            0,                  AV_OPT_TYPE_CONST,  {.i64=9},     0, 0, AF, "m" },
+    { "level", "set input gain",        OFFSET(level_in),   AV_OPT_TYPE_FLOAT,  {.dbl=1},     0, 1, AF },
     { NULL }
 };
 
@@ -98,6 +103,10 @@ static av_cold int init(AVFilterContext *ctx)
     char *p, *arg, *saveptr = NULL;
     int i, ret = 0;
 
+    s->fdsp = avpriv_float_dsp_alloc(0);
+    if (!s->fdsp)
+        return AVERROR(ENOMEM);
+
     s->splits = av_calloc(MAX_SPLITS, sizeof(*s->splits));
     if (!s->splits)
         return AVERROR(ENOMEM);
@@ -288,7 +297,7 @@ static void biquad_process_## name(BiquadContext *b,           \
 BIQUAD_PROCESS(fltp, float)
 BIQUAD_PROCESS(dblp, double)
 
-#define XOVER_PROCESS(name, type, one)                         \
+#define XOVER_PROCESS(name, type, one, ff)                                                  \
 static int filter_channels_## name(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) \
 {                                                                                           \
     AudioCrossoverContext *s = ctx->priv;                                                   \
@@ -299,23 +308,26 @@ static int filter_channels_## name(AVFilterContext *ctx, void *arg, int jobnr, i
     const int nb_samples = in->nb_samples;                                                  \
                                                                                             \
     for (int ch = start; ch < end; ch++) {                                                  \
+        const type *src = (const type *)in->extended_data[ch];                              \
         CrossoverChannel *xover = &s->xover[ch];                                            \
                                                                                             \
+        s->fdsp->vector_## ff ##mul_scalar((type *)frames[0]->extended_data[ch], src,       \
+                                    s->level_in, nb_samples);                               \
+        emms_c();                                                                           \
+                                                                                            \
         for (int band = 0; band < ctx->nb_outputs; band++) {                                \
             for (int f = 0; band + 1 < ctx->nb_outputs && f < s->filter_count; f++) {       \
-                const type *src = (const type *)in->extended_data[ch];                      \
                 const type *prv = (const type *)frames[band]->extended_data[ch];            \
                 type *dst = (type *)frames[band + 1]->extended_data[ch];                    \
-                const type *hsrc = (band == 0 && f == 0) ? src : f == 0 ? prv : dst;        \
+                const type *hsrc = f == 0 ? prv : dst;                                      \
                 BiquadContext *hp = &xover->hp[band][f];                                    \
                                                                                             \
                 biquad_process_## name(hp, dst, hsrc, nb_samples);                          \
             }                                                                               \
                                                                                             \
             for (int f = 0; band + 1 < ctx->nb_outputs && f < s->filter_count; f++) {       \
-                const type *src = (const type *)in->extended_data[ch];                      \
                 type *dst = (type *)frames[band]->extended_data[ch];                        \
-                const type *lsrc = (band == 0 && f == 0) ? src : dst;                       \
+                const type *lsrc = dst;                                                     \
                 BiquadContext *lp = &xover->lp[band][f];                                    \
                                                                                             \
                 biquad_process_## name(lp, dst, lsrc, nb_samples);                          \
@@ -353,8 +365,8 @@ static int filter_channels_## name(AVFilterContext *ctx, void *arg, int jobnr, i
     return 0;                                                                               \
 }
 
-XOVER_PROCESS(fltp, float, 1.f)
-XOVER_PROCESS(dblp, double, 1.0)
+XOVER_PROCESS(fltp, float, 1.f, f)
+XOVER_PROCESS(dblp, double, 1.0, d)
 
 static int config_input(AVFilterLink *inlink)
 {
@@ -453,6 +465,7 @@ static av_cold void uninit(AVFilterContext *ctx)
     AudioCrossoverContext *s = ctx->priv;
     int i;
 
+    av_freep(&s->fdsp);
     av_freep(&s->splits);
     av_freep(&s->xover);