X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=libavfilter%2Faf_sidechaincompress.c;h=dac4605d0ce630d29cab93b6807cf0167b0c3be7;hb=89bbf01978194ee1354bb3feef139a648bc1903b;hp=29b3753691d22493aff679dde813a5b6e60338f1;hpb=3e0daf0716888d166329cffa554893846d6ec3be;p=ffmpeg diff --git a/libavfilter/af_sidechaincompress.c b/libavfilter/af_sidechaincompress.c index 29b3753691d..dac4605d0ce 100644 --- a/libavfilter/af_sidechaincompress.c +++ b/libavfilter/af_sidechaincompress.c @@ -21,7 +21,7 @@ /** * @file - * Sidechain compressor filter + * Audio (Sidechain) Compressor filter */ #include "libavutil/avassert.h" @@ -38,17 +38,21 @@ typedef struct SidechainCompressContext { const AVClass *class; + double level_in; + double level_sc; double attack, attack_coeff; double release, release_coeff; double lin_slope; double ratio; double threshold; double makeup; + double mix; double thres; double knee; double knee_start; double knee_stop; double lin_knee_start; + double adj_knee_start; double compressed_knee_stop; int link; int detection; @@ -60,37 +64,28 @@ typedef struct SidechainCompressContext { #define A AV_OPT_FLAG_AUDIO_PARAM #define F AV_OPT_FLAG_FILTERING_PARAM -static const AVOption sidechaincompress_options[] = { - { "threshold", "set threshold", OFFSET(threshold), AV_OPT_TYPE_DOUBLE, {.dbl=0.125}, 0.000976563, 1, A|F }, - { "ratio", "set ratio", OFFSET(ratio), AV_OPT_TYPE_DOUBLE, {.dbl=2}, 1, 20, A|F }, - { "attack", "set attack", OFFSET(attack), AV_OPT_TYPE_DOUBLE, {.dbl=20}, 0.01, 2000, A|F }, - { "release", "set release", OFFSET(release), AV_OPT_TYPE_DOUBLE, {.dbl=250}, 0.01, 9000, A|F }, - { "makeup", "set make up gain", OFFSET(makeup), AV_OPT_TYPE_DOUBLE, {.dbl=2}, 1, 64, A|F }, - { "knee", "set knee", OFFSET(knee), AV_OPT_TYPE_DOUBLE, {.dbl=2.82843}, 1, 8, A|F }, - { "link", "set link type", OFFSET(link), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, A|F, "link" }, - { "average", 0, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, A|F, "link" }, - { "maximum", 0, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, A|F, "link" }, - { "detection", "set detection", OFFSET(detection), AV_OPT_TYPE_INT, {.i64=1}, 0, 1, A|F, "detection" }, - { "peak", 0, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, A|F, "detection" }, - { "rms", 0, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, A|F, "detection" }, +static const AVOption options[] = { + { "level_in", "set input gain", OFFSET(level_in), AV_OPT_TYPE_DOUBLE, {.dbl=1}, 0.015625, 64, A|F }, + { "threshold", "set threshold", OFFSET(threshold), AV_OPT_TYPE_DOUBLE, {.dbl=0.125}, 0.000976563, 1, A|F }, + { "ratio", "set ratio", OFFSET(ratio), AV_OPT_TYPE_DOUBLE, {.dbl=2}, 1, 20, A|F }, + { "attack", "set attack", OFFSET(attack), AV_OPT_TYPE_DOUBLE, {.dbl=20}, 0.01, 2000, A|F }, + { "release", "set release", OFFSET(release), AV_OPT_TYPE_DOUBLE, {.dbl=250}, 0.01, 9000, A|F }, + { "makeup", "set make up gain", OFFSET(makeup), AV_OPT_TYPE_DOUBLE, {.dbl=2}, 1, 64, A|F }, + { "knee", "set knee", OFFSET(knee), AV_OPT_TYPE_DOUBLE, {.dbl=2.82843}, 1, 8, A|F }, + { "link", "set link type", OFFSET(link), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, A|F, "link" }, + { "average", 0, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, A|F, "link" }, + { "maximum", 0, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, A|F, "link" }, + { "detection", "set detection", OFFSET(detection), AV_OPT_TYPE_INT, {.i64=1}, 0, 1, A|F, "detection" }, + { "peak", 0, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, A|F, "detection" }, + { "rms", 0, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, A|F, "detection" }, + { "level_sc", "set sidechain gain", OFFSET(level_sc), AV_OPT_TYPE_DOUBLE, {.dbl=1}, 0.015625, 64, A|F }, + { "mix", "set mix", OFFSET(mix), AV_OPT_TYPE_DOUBLE, {.dbl=1}, 0, 1, A|F }, { NULL } }; +#define sidechaincompress_options options AVFILTER_DEFINE_CLASS(sidechaincompress); -static av_cold int init(AVFilterContext *ctx) -{ - SidechainCompressContext *s = ctx->priv; - - s->thres = log(s->threshold); - s->lin_knee_start = s->threshold / sqrt(s->knee); - s->knee_start = log(s->lin_knee_start); - s->knee_stop = log(s->threshold * sqrt(s->knee)); - s->compressed_knee_stop = (s->knee_stop - s->thres) / s->ratio + s->thres; - - return 0; -} - // A fake infinity value (because real infinity may break some hosts) #define FAKE_INFINITY (65536.0 * 65536.0) @@ -124,44 +119,44 @@ static double output_gain(double lin_slope, double ratio, double thres, return exp(gain - slope); } -static int filter_frame(AVFilterLink *link, AVFrame *frame) +static int compressor_config_output(AVFilterLink *outlink) { - AVFilterContext *ctx = link->dst; + AVFilterContext *ctx = outlink->src; SidechainCompressContext *s = ctx->priv; - AVFilterLink *sclink = ctx->inputs[1]; - AVFilterLink *outlink = ctx->outputs[0]; - const double makeup = s->makeup; - const double *scsrc; - double *sample; - int nb_samples; - int ret, i, c; - for (i = 0; i < 2; i++) - if (link == ctx->inputs[i]) - break; - av_assert0(i < 2 && !s->input_frame[i]); - s->input_frame[i] = frame; + s->thres = log(s->threshold); + s->lin_knee_start = s->threshold / sqrt(s->knee); + s->adj_knee_start = s->lin_knee_start * s->lin_knee_start; + s->knee_start = log(s->lin_knee_start); + s->knee_stop = log(s->threshold * sqrt(s->knee)); + s->compressed_knee_stop = (s->knee_stop - s->thres) / s->ratio + s->thres; - if (!s->input_frame[0] || !s->input_frame[1]) - return 0; + s->attack_coeff = FFMIN(1., 1. / (s->attack * outlink->sample_rate / 4000.)); + s->release_coeff = FFMIN(1., 1. / (s->release * outlink->sample_rate / 4000.)); - nb_samples = FFMIN(s->input_frame[0]->nb_samples, - s->input_frame[1]->nb_samples); + return 0; +} - sample = (double *)s->input_frame[0]->data[0]; - scsrc = (const double *)s->input_frame[1]->data[0]; +static void compressor(SidechainCompressContext *s, + const double *src, double *dst, const double *scsrc, int nb_samples, + double level_in, double level_sc, + AVFilterLink *inlink, AVFilterLink *sclink) +{ + const double makeup = s->makeup; + const double mix = s->mix; + int i, c; for (i = 0; i < nb_samples; i++) { double abs_sample, gain = 1.0; - abs_sample = fabs(scsrc[0]); + abs_sample = fabs(scsrc[0] * level_sc); if (s->link == 1) { for (c = 1; c < sclink->channels; c++) - abs_sample = FFMAX(fabs(scsrc[c]), abs_sample); + abs_sample = FFMAX(fabs(scsrc[c] * level_sc), abs_sample); } else { for (c = 1; c < sclink->channels; c++) - abs_sample += fabs(scsrc[c]); + abs_sample += fabs(scsrc[c] * level_sc); abs_sample /= sclink->channels; } @@ -171,18 +166,49 @@ static int filter_frame(AVFilterLink *link, AVFrame *frame) s->lin_slope += (abs_sample - s->lin_slope) * (abs_sample > s->lin_slope ? s->attack_coeff : s->release_coeff); - if (s->lin_slope > 0.0 && s->lin_slope > s->lin_knee_start) + if (s->lin_slope > 0.0 && s->lin_slope > (s->detection ? s->adj_knee_start : s->lin_knee_start)) gain = output_gain(s->lin_slope, s->ratio, s->thres, s->knee, s->knee_start, s->knee_stop, s->compressed_knee_stop, s->detection); - for (c = 0; c < outlink->channels; c++) - sample[c] *= gain * makeup; + for (c = 0; c < inlink->channels; c++) + dst[c] = src[c] * level_in * (gain * makeup * mix + (1. - mix)); - sample += outlink->channels; + src += inlink->channels; + dst += inlink->channels; scsrc += sclink->channels; } +} +#if CONFIG_SIDECHAINCOMPRESS_FILTER +static int filter_frame(AVFilterLink *link, AVFrame *frame) +{ + AVFilterContext *ctx = link->dst; + SidechainCompressContext *s = ctx->priv; + AVFilterLink *outlink = ctx->outputs[0]; + const double *scsrc; + double *sample; + int nb_samples; + int ret, i; + + for (i = 0; i < 2; i++) + if (link == ctx->inputs[i]) + break; + av_assert0(i < 2 && !s->input_frame[i]); + s->input_frame[i] = frame; + + if (!s->input_frame[0] || !s->input_frame[1]) + return 0; + + nb_samples = FFMIN(s->input_frame[0]->nb_samples, + s->input_frame[1]->nb_samples); + + sample = (double *)s->input_frame[0]->data[0]; + scsrc = (const double *)s->input_frame[1]->data[0]; + + compressor(s, sample, sample, scsrc, nb_samples, + s->level_in, s->level_sc, + ctx->inputs[0], ctx->inputs[1]); ret = ff_filter_frame(outlink, s->input_frame[0]); s->input_frame[0] = NULL; @@ -250,7 +276,6 @@ static int query_formats(AVFilterContext *ctx) static int config_output(AVFilterLink *outlink) { AVFilterContext *ctx = outlink->src; - SidechainCompressContext *s = ctx->priv; if (ctx->inputs[0]->sample_rate != ctx->inputs[1]->sample_rate) { av_log(ctx, AV_LOG_ERROR, @@ -265,8 +290,7 @@ static int config_output(AVFilterLink *outlink) outlink->channel_layout = ctx->inputs[0]->channel_layout; outlink->channels = ctx->inputs[0]->channels; - s->attack_coeff = FFMIN(1., 1. / (s->attack * outlink->sample_rate / 4000.)); - s->release_coeff = FFMIN(1., 1. / (s->release * outlink->sample_rate / 4000.)); + compressor_config_output(outlink); return 0; } @@ -302,8 +326,101 @@ AVFilter ff_af_sidechaincompress = { .description = NULL_IF_CONFIG_SMALL("Sidechain compressor."), .priv_size = sizeof(SidechainCompressContext), .priv_class = &sidechaincompress_class, - .init = init, .query_formats = query_formats, .inputs = sidechaincompress_inputs, .outputs = sidechaincompress_outputs, }; +#endif /* CONFIG_SIDECHAINCOMPRESS_FILTER */ + +#if CONFIG_ACOMPRESSOR_FILTER +static int acompressor_filter_frame(AVFilterLink *inlink, AVFrame *in) +{ + const double *src = (const double *)in->data[0]; + AVFilterContext *ctx = inlink->dst; + SidechainCompressContext *s = ctx->priv; + AVFilterLink *outlink = ctx->outputs[0]; + AVFrame *out; + double *dst; + + if (av_frame_is_writable(in)) { + out = in; + } else { + out = ff_get_audio_buffer(inlink, in->nb_samples); + if (!out) { + av_frame_free(&in); + return AVERROR(ENOMEM); + } + av_frame_copy_props(out, in); + } + dst = (double *)out->data[0]; + + compressor(s, src, dst, src, in->nb_samples, + s->level_in, s->level_in, + inlink, inlink); + + if (out != in) + av_frame_free(&in); + return ff_filter_frame(outlink, out); +} + +static int acompressor_query_formats(AVFilterContext *ctx) +{ + AVFilterFormats *formats; + AVFilterChannelLayouts *layouts; + static const enum AVSampleFormat sample_fmts[] = { + AV_SAMPLE_FMT_DBL, + AV_SAMPLE_FMT_NONE + }; + int ret; + + layouts = ff_all_channel_counts(); + if (!layouts) + return AVERROR(ENOMEM); + ret = ff_set_common_channel_layouts(ctx, layouts); + if (ret < 0) + return ret; + + formats = ff_make_format_list(sample_fmts); + if (!formats) + return AVERROR(ENOMEM); + ret = ff_set_common_formats(ctx, formats); + if (ret < 0) + return ret; + + formats = ff_all_samplerates(); + if (!formats) + return AVERROR(ENOMEM); + return ff_set_common_samplerates(ctx, formats); +} + +#define acompressor_options options +AVFILTER_DEFINE_CLASS(acompressor); + +static const AVFilterPad acompressor_inputs[] = { + { + .name = "default", + .type = AVMEDIA_TYPE_AUDIO, + .filter_frame = acompressor_filter_frame, + }, + { NULL } +}; + +static const AVFilterPad acompressor_outputs[] = { + { + .name = "default", + .type = AVMEDIA_TYPE_AUDIO, + .config_props = compressor_config_output, + }, + { NULL } +}; + +AVFilter ff_af_acompressor = { + .name = "acompressor", + .description = NULL_IF_CONFIG_SMALL("Audio compressor."), + .priv_size = sizeof(SidechainCompressContext), + .priv_class = &acompressor_class, + .query_formats = acompressor_query_formats, + .inputs = acompressor_inputs, + .outputs = acompressor_outputs, +}; +#endif /* CONFIG_ACOMPRESSOR_FILTER */