1 files changed, 141 insertions, 0 deletions
diff --git a/audio/filter/af_format.c b/audio/filter/af_format.c
new file mode 100644
index 0000000000..21af663e81
--- /dev/null
+++ b/audio/filter/af_format.c
@@ -0,0 +1,141 @@
+/*
+ * This file is part of mpv.
+ *
+ * mpv is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * mpv 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with mpv.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <stdlib.h>
+
+#include <libavutil/common.h>
+
+#include "mpvcore/m_option.h"
+
+#include "audio/format.h"
+#include "af.h"
+
+struct priv {
+    struct m_config *config;
+
+    int in_format;
+    int in_srate;
+    struct mp_chmap in_channels;
+    int out_format;
+    int out_srate;
+    struct mp_chmap out_channels;
+
+    int fail;
+
+    struct mp_audio data;
+    struct mp_audio temp;
+};
+
+static int control(struct af_instance *af, int cmd, void *arg)
+{
+    struct priv *priv = af->priv;
+
+    switch (cmd) {
+    case AF_CONTROL_REINIT: {
+        struct mp_audio *in = arg;
+        struct mp_audio orig_in = *in;
+        struct mp_audio *out = af->data;
+
+        if (priv->in_format != AF_FORMAT_UNKNOWN)
+            mp_audio_set_format(in, priv->in_format);
+
+        if (priv->in_channels.num)
+            mp_audio_set_channels(in, &priv->in_channels);
+
+        if (priv->in_srate)
+            in->rate = priv->in_srate;
+
+        mp_audio_copy_config(out, in);
+
+        if (priv->out_format != AF_FORMAT_UNKNOWN)
+            mp_audio_set_format(out, priv->out_format);
+
+        if (priv->out_channels.num)
+            mp_audio_set_channels(out, &priv->out_channels);
+
+        if (priv->out_srate)
+            out->rate = priv->out_srate;
+
+        if (in->nch != out->nch || in->bps != out->bps) {
+            mp_msg(MSGT_AFILTER, MSGL_ERR,
+                   "[af_format] Forced input/output formats are incompatible.\n");
+            return AF_ERROR;
+        }
+
+        if (priv->fail) {
+            mp_msg(MSGT_AFILTER, MSGL_ERR, "[af_format] Failing on purpose.\n");
+            return AF_ERROR;
+        }
+
+        return mp_audio_config_equals(in, &orig_in) ? AF_OK : AF_FALSE;
+    }
+    }
+    return AF_UNKNOWN;
+}
+
+static struct mp_audio *play(struct af_instance *af, struct mp_audio *data)
+{
+    struct priv *priv = af->priv;
+    struct mp_audio *r = &priv->temp;
+
+    *r = *af->data;
+    r->audio = data->audio;
+    r->len = data->len;
+
+    return r;
+}
+
+static void uninit(struct af_instance *af)
+{
+}
+
+static int af_open(struct af_instance *af)
+{
+    af->control = control;
+    af->uninit = uninit;
+    af->play = play;
+    af->mul = 1;
+    struct priv *priv = af->priv;
+    af->data = &priv->data;
+    return AF_OK;
+}
+
+#define OPT_BASE_STRUCT struct priv
+
+struct af_info af_info_format = {
+    "Force audio format",
+    "format",
+    "",
+    "",
+    0,
+    af_open,
+    .priv_size = sizeof(struct priv),
+    .priv_defaults = &(const struct priv) {
+        .in_format = AF_FORMAT_UNKNOWN,
+        .out_format = AF_FORMAT_UNKNOWN,
+    },
+    .options = (const struct m_option[]) {
+        OPT_AUDIOFORMAT("format", in_format, 0),
+        OPT_INTRANGE("srate", in_srate, 0, 1000, 8*48000),
+        OPT_CHMAP("channels", in_channels, CONF_MIN, .min = 0),
+        OPT_AUDIOFORMAT("out-format", out_format, 0),
+        OPT_INTRANGE("out-srate", out_srate, 0, 1000, 8*48000),
+        OPT_CHMAP("out-channels", out_channels, CONF_MIN, .min = 0),
+        OPT_FLAG("fail", fail, 0),
+        {0}
+    },
+};