[FFmpeg-devel] [PATCH] avfilter: add acontrast filter
Paul B Mahol
onemda at gmail.com
Sat Nov 18 18:03:56 EET 2017
On 11/18/17, Rostislav Pehlivanov <atomnuker at gmail.com> wrote:
> On 18 November 2017 at 10:44, Paul B Mahol <onemda at gmail.com> wrote:
>
>> Signed-off-by: Paul B Mahol <onemda at gmail.com>
>> ---
>> doc/filters.texi | 10 +++
>> libavfilter/Makefile | 1 +
>> libavfilter/af_acontrast.c | 219 ++++++++++++++++++++++++++++++
>> +++++++++++++++
>> libavfilter/allfilters.c | 1 +
>> 4 files changed, 231 insertions(+)
>> create mode 100644 libavfilter/af_acontrast.c
>>
>> diff --git a/doc/filters.texi b/doc/filters.texi
>> index 5d99437871..e35952510b 100644
>> --- a/doc/filters.texi
>> +++ b/doc/filters.texi
>> @@ -429,6 +429,16 @@ How much to use compressed signal in output. Default
>> is 1.
>> Range is between 0 and 1.
>> @end table
>>
>> + at section acontrast
>> +Simple audio dynamic range commpression/expansion filter.
>> +
>> +The filter accepts the following options:
>> +
>> + at table @option
>> + at item c
>> +Set contrast. Default is 33. Allowed range is between 0 and 100.
>> + at end table
>> +
>> @section acopy
>>
>> Copy the input audio source unchanged to the output. This is mainly
>> useful for
>> diff --git a/libavfilter/Makefile b/libavfilter/Makefile
>> index 9acae3ff5b..71c6333a52 100644
>> --- a/libavfilter/Makefile
>> +++ b/libavfilter/Makefile
>> @@ -31,6 +31,7 @@ OBJS-$(CONFIG_QSVVPP) += qsvvpp.o
>> # audio filters
>> OBJS-$(CONFIG_ABENCH_FILTER) += f_bench.o
>> OBJS-$(CONFIG_ACOMPRESSOR_FILTER) += af_sidechaincompress.o
>> +OBJS-$(CONFIG_ACONTRAST_FILTER) += af_acontrast.o
>> OBJS-$(CONFIG_ACOPY_FILTER) += af_acopy.o
>> OBJS-$(CONFIG_ACROSSFADE_FILTER) += af_afade.o
>> OBJS-$(CONFIG_ACRUSHER_FILTER) += af_acrusher.o
>> diff --git a/libavfilter/af_acontrast.c b/libavfilter/af_acontrast.c
>> new file mode 100644
>> index 0000000000..38de08ffe5
>> --- /dev/null
>> +++ b/libavfilter/af_acontrast.c
>> @@ -0,0 +1,219 @@
>> +/*
>> + * Copyright (c) 2008 Rob Sykes
>> + * Copyright (c) 2017 Paul B Mahol
>> + *
>> + * This file is part of FFmpeg.
>> + *
>> + * FFmpeg is free software; you can redistribute it and/or
>> + * modify it under the terms of the GNU Lesser General Public
>> + * License as published by the Free Software Foundation; either
>> + * version 2.1 of the License, or (at your option) any later version.
>> + *
>> + * FFmpeg is distributed in the hope that it will be useful,
>> + * but WITHOUT ANY WARRANTY; without even the implied warranty of
>> + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
>> + * Lesser General Public License for more details.
>> + *
>> + * You should have received a copy of the GNU Lesser General Public
>> + * License along with FFmpeg; if not, write to the Free Software
>> + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
>> 02110-1301 USA
>> + */
>> +
>> +#include "libavutil/channel_layout.h"
>> +#include "libavutil/opt.h"
>> +#include "avfilter.h"
>> +#include "audio.h"
>> +#include "formats.h"
>> +
>> +typedef struct AudioContrastContext {
>> + const AVClass *class;
>> + float contrast;
>> + void (*filter)(void **dst, const void **src,
>> + int nb_samples, int channels, float contrast);
>> +} AudioContrastContext;
>> +
>> +#define OFFSET(x) offsetof(AudioContrastContext, x)
>> +#define A AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
>> +
>> +static const AVOption acontrast_options[] = {
>> + { "c", "set contrast", OFFSET(contrast), AV_OPT_TYPE_FLOAT,
>> {.dbl=33}, 0, 100, A },
>>
>
> "contrast" instead of "c"? Not sure if single letter options are a good
> idea.
>
>
>
>> + { NULL }
>> +};
>> +
>> +AVFILTER_DEFINE_CLASS(acontrast);
>> +
>> +static int query_formats(AVFilterContext *ctx)
>> +{
>> + AVFilterFormats *formats = NULL;
>> + AVFilterChannelLayouts *layouts = NULL;
>> + static const enum AVSampleFormat sample_fmts[] = {
>> + AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_FLTP,
>> + AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_DBLP,
>> + AV_SAMPLE_FMT_NONE
>> + };
>> + int 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;
>> +
>> + 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_all_samplerates();
>> + return ff_set_common_samplerates(ctx, formats);
>> +}
>> +
>> +static void filter_flt(void **d, const void **s,
>> + int nb_samples, int channels,
>> + float contrast)
>> +{
>> + const float *src = s[0];
>> + float *dst = d[0];
>> + int n, c;
>> +
>> + for (n = 0; n < nb_samples; n++) {
>> + for (c = 0; c < channels; c++) {
>> + double d = src[c] * M_PI_2;
>> +
>> + dst[c] = sin(d + contrast * sin(d * 4));
>>
>
> sinf() instead of sin()
ok
>
>
>
>> + }
>> +
>> + dst += c;
>> + src += c;
>> + }
>> +}
>> +
>> +static void filter_dbl(void **d, const void **s,
>> + int nb_samples, int channels,
>> + float contrast)
>> +{
>> + const double *src = s[0];
>> + double *dst = d[0];
>> + int n, c;
>> +
>> + for (n = 0; n < nb_samples; n++) {
>> + for (c = 0; c < channels; c++) {
>> + double d = src[c] * M_PI_2;
>> +
>> + dst[c] = sin(d + contrast * sin(d * 4));
>> + }
>> +
>> + dst += c;
>> + src += c;
>> + }
>> +}
>> +
>> +static void filter_fltp(void **d, const void **s,
>> + int nb_samples, int channels,
>> + float contrast)
>> +{
>> + int n, c;
>> +
>> + for (c = 0; c < channels; c++) {
>> + const float *src = s[c];
>> + float *dst = d[c];
>> +
>> + for (n = 0; n < nb_samples; n++) {
>> + double d = src[n] * M_PI_2;
>> +
>> + dst[n] = sin(d + contrast * sin(d * 4));
>>
>
> sinf() instead of sin()
>
ok
>
>
>> + }
>> + }
>> +}
>> +
>> +static void filter_dblp(void **d, const void **s,
>> + int nb_samples, int channels,
>> + float contrast)
>> +{
>> + int n, c;
>> +
>> + for (c = 0; c < channels; c++) {
>> + const double *src = s[c];
>> + double *dst = d[c];
>> +
>> + for (n = 0; n < nb_samples; n++) {
>> + double d = src[n] * M_PI_2;
>> +
>> + dst[n] = sin(d + contrast * sin(d * 4));
>> + }
>> + }
>> +}
>>
>
> Could you do the filtering in-place? Via av_frame_make_writeable?
>
Both are supported.
>
>
>> +
>> +static int config_input(AVFilterLink *inlink)
>> +{
>> + AVFilterContext *ctx = inlink->dst;
>> + AudioContrastContext *s = ctx->priv;
>> +
>> + switch (inlink->format) {
>> + case AV_SAMPLE_FMT_FLT: s->filter = filter_flt; break;
>> + case AV_SAMPLE_FMT_DBL: s->filter = filter_dbl; break;
>> + case AV_SAMPLE_FMT_FLTP: s->filter = filter_fltp; break;
>> + case AV_SAMPLE_FMT_DBLP: s->filter = filter_dblp; break;
>> + }
>> +
>> + return 0;
>> +}
>> +
>> +static int filter_frame(AVFilterLink *inlink, AVFrame *in)
>> +{
>> + AVFilterContext *ctx = inlink->dst;
>> + AVFilterLink *outlink = ctx->outputs[0];
>> + AudioContrastContext *s = ctx->priv;
>> + AVFrame *out;
>> +
>> + 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);
>> + }
>> +
>> + s->filter((void **)out->extended_data, (const void
>> **)in->extended_data,
>> + in->nb_samples, in->channels, s->contrast / 750);
>>
>
> Divide s->contrast by 750 during init?
Doesn't cost much, and also it is less lines this way.
More information about the ffmpeg-devel
mailing list