lavfi: add aformat filter

Signed-off-by: Stefano Sabatini <stefano.sabatini-lala@poste.it>

1 files changed, 114 insertions, 0 deletions

diff --git a/libavfilter/af_aformat.c b/libavfilter/af_aformat.c
new file mode 100644
index 0000000000..f4ea0dd887
--- /dev/null
+++ b/libavfilter/af_aformat.c
@@ -0,0 +1,114 @@
+/*
+ * Copyright (c) 2011 Mina Nagy Zaki
+ *
+ * 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
+ */
+
+/**
+ * @file
+ * format audio filter
+ */
+
+#include "libavutil/audioconvert.h"
+#include "avfilter.h"
+#include "internal.h"
+
+typedef struct {
+    AVFilterFormats *formats, *chlayouts, *packing;
+} AFormatContext;
+
+static av_cold int init(AVFilterContext *ctx, const char *args, void *opaque)
+{
+    AFormatContext * const aformat = ctx->priv;
+    char *arg, *fmt_str;
+    int64_t fmt;
+    int ret;
+
+    arg = strsep(&args, ":");
+    if (!arg) goto arg_fail;
+    if (!strcmp(arg, "all")) {
+        aformat->formats = avfilter_all_formats(AVMEDIA_TYPE_AUDIO);
+    } else {
+        while (fmt_str = strsep(&arg, ",")) {
+            if ((ret = ff_parse_sample_format((int*)&fmt, fmt_str, ctx)) < 0)
+                return ret;
+            avfilter_add_format(&aformat->formats, fmt);
+        }
+    }
+
+    arg = strsep(&args, ":");
+    if (!arg) goto arg_fail;
+    if (!strcmp(arg, "all")) {
+        aformat->chlayouts = avfilter_all_channel_layouts();
+    } else {
+        while (fmt_str = strsep(&arg, ",")) {
+            if ((ret = ff_parse_channel_layout(&fmt, fmt_str, ctx)) < 0)
+                return ret;
+            avfilter_add_format(&aformat->chlayouts, fmt);
+        }
+    }
+
+    arg = strsep(&args, ":");
+    if (!arg) goto arg_fail;
+    if (!strcmp(arg, "all")) {
+        aformat->packing = avfilter_all_packing_formats();
+    } else {
+        while (fmt_str = strsep(&arg, ",")) {
+            if ((ret = ff_parse_packing_format((int*)&fmt, fmt_str, ctx)) < 0)
+                return ret;
+            avfilter_add_format(&aformat->packing, fmt);
+        }
+    }
+
+    return 0;
+
+arg_fail:
+    av_log(ctx, AV_LOG_ERROR, "Invalid arguments, they must be of the form "
+                              "sample_fmts:channel_layouts:packing_fmts\n");
+    return AVERROR(EINVAL);
+}
+
+static int query_formats(AVFilterContext *ctx)
+{
+    AFormatContext * const aformat = ctx->priv;
+
+    avfilter_set_common_sample_formats (ctx, aformat->formats);
+    avfilter_set_common_channel_layouts(ctx, aformat->chlayouts);
+    avfilter_set_common_packing_formats(ctx, aformat->packing);
+    return 0;
+}
+
+static void filter_samples(AVFilterLink *inlink, AVFilterBufferRef *insamplesref)
+{
+    avfilter_filter_samples(inlink->dst->outputs[0], insamplesref);
+}
+
+AVFilter avfilter_af_aformat = {
+    .name          = "aformat",
+    .description   = NULL_IF_CONFIG_SMALL("Convert the input audio to one of the specified formats."),
+    .init          = init,
+    .query_formats = query_formats,
+    .priv_size     = sizeof(AFormatContext),
+
+    .inputs        = (AVFilterPad[]) {{ .name            = "default",
+                                        .type            = AVMEDIA_TYPE_AUDIO,
+                                        .filter_samples  = filter_samples},
+                                      { .name = NULL}},
+    .outputs       = (AVFilterPad[]) {{ .name            = "default",
+                                        .type            = AVMEDIA_TYPE_AUDIO},
+                                      { .name = NULL}},
+};