diff options
Diffstat (limited to 'audio/filter/af_lavfi.c')
-rw-r--r-- | audio/filter/af_lavfi.c | 413 |
1 files changed, 0 insertions, 413 deletions
diff --git a/audio/filter/af_lavfi.c b/audio/filter/af_lavfi.c deleted file mode 100644 index ab8a026de7..0000000000 --- a/audio/filter/af_lavfi.c +++ /dev/null @@ -1,413 +0,0 @@ -/* - * This file is part of mpv. - * - * Filter graph creation code taken from FFmpeg ffplay.c (LGPL 2.1 or later) - * - * mpv 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. - * - * 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 Lesser General Public License for more details. - * - * You should have received a copy of the GNU Lesser General Public - * License along with mpv. If not, see <http://www.gnu.org/licenses/>. - */ - -#include <stdlib.h> -#include <stdint.h> -#include <stdio.h> -#include <assert.h> - -#include <libavutil/avstring.h> -#include <libavutil/mem.h> -#include <libavutil/mathematics.h> -#include <libavutil/rational.h> -#include <libavutil/samplefmt.h> -#include <libavutil/time.h> -#include <libavutil/opt.h> -#include <libavfilter/avfilter.h> -#include <libavfilter/buffersink.h> -#include <libavfilter/buffersrc.h> - -#include "config.h" - -#include "audio/format.h" -#include "audio/fmt-conversion.h" -#include "af.h" - -#include "common/av_common.h" -#include "common/tags.h" - -#include "options/m_option.h" - -// FFmpeg and Libav have slightly different APIs, just enough to cause us -// unnecessary pain. <Expletive deleted.> -#if LIBAVFILTER_VERSION_MICRO < 100 -#define graph_parse(graph, filters, inputs, outputs, log_ctx) \ - avfilter_graph_parse(graph, filters, inputs, outputs, log_ctx) -#define avfilter_graph_send_command(a, b, c, d, e, f, g) -1 -#else -#define graph_parse(graph, filters, inputs, outputs, log_ctx) \ - avfilter_graph_parse_ptr(graph, filters, &(inputs), &(outputs), log_ctx) -#endif - -struct priv { - // Single filter bridge, instead of a graph. - bool is_bridge; - - AVFilterGraph *graph; - AVFilterContext *in; - AVFilterContext *out; - - int64_t samples_in; - - AVRational timebase_out; - - bool eof; - - struct mp_tags *metadata; - - // options - char *cfg_graph; - char **cfg_avopts; - char *cfg_filter_name; - char **cfg_filter_opts; -}; - -static void destroy_graph(struct af_instance *af) -{ - struct priv *p = af->priv; - avfilter_graph_free(&p->graph); - p->in = p->out = NULL; - p->samples_in = 0; - p->eof = false; -} - -static bool recreate_graph(struct af_instance *af, struct mp_audio *config) -{ - void *tmp = talloc_new(NULL); - struct priv *p = af->priv; - AVFilterContext *in = NULL, *out = NULL; - bool ok = false; - - if (!p->is_bridge && bstr0(p->cfg_graph).len == 0) { - MP_FATAL(af, "lavfi: no filter graph set\n"); - return false; - } - - destroy_graph(af); - - AVFilterGraph *graph = avfilter_graph_alloc(); - if (!graph) - goto error; - - if (mp_set_avopts(af->log, graph, p->cfg_avopts) < 0) - goto error; - - AVFilterInOut *outputs = avfilter_inout_alloc(); - AVFilterInOut *inputs = avfilter_inout_alloc(); - if (!outputs || !inputs) - goto error; - - char *src_args = talloc_asprintf(tmp, - "sample_rate=%d:sample_fmt=%s:time_base=%d/%d:" - "channel_layout=0x%"PRIx64, config->rate, - av_get_sample_fmt_name(af_to_avformat(config->format)), - 1, config->rate, mp_chmap_to_lavc(&config->channels)); - - if (avfilter_graph_create_filter(&in, avfilter_get_by_name("abuffer"), - "src", src_args, NULL, graph) < 0) - goto error; - - if (avfilter_graph_create_filter(&out, avfilter_get_by_name("abuffersink"), - "out", NULL, NULL, graph) < 0) - goto error; - - if (p->is_bridge) { - AVFilterContext *filter = avfilter_graph_alloc_filter(graph, - avfilter_get_by_name(p->cfg_filter_name), "filter"); - if (!filter) - goto error; - - if (mp_set_avopts(af->log, filter->priv, p->cfg_filter_opts) < 0) - goto error; - - if (avfilter_init_str(filter, NULL) < 0) - goto error; - - // Yep, we have to manually link those filters. - if (filter->nb_inputs != 1 || - avfilter_pad_get_type(filter->input_pads, 0) != AVMEDIA_TYPE_AUDIO || - filter->nb_outputs != 1 || - avfilter_pad_get_type(filter->output_pads, 0) != AVMEDIA_TYPE_AUDIO) - { - MP_ERR(af, "The filter is required to have 1 audio input pad and " - "1 audio output pad.\n"); - goto error; - } - if (avfilter_link(in, 0, filter, 0) < 0 || - avfilter_link(filter, 0, out, 0) < 0) - { - MP_ERR(af, "Failed to link filter.\n"); - goto error; - } - } else { - MP_VERBOSE(af, "lavfi: create graph: '%s'\n", p->cfg_graph); - - outputs->name = av_strdup("in"); - outputs->filter_ctx = in; - - inputs->name = av_strdup("out"); - inputs->filter_ctx = out; - - if (graph_parse(graph, p->cfg_graph, inputs, outputs, NULL) < 0) - goto error; - } - - if (avfilter_graph_config(graph, NULL) < 0) - goto error; - - p->in = in; - p->out = out; - p->graph = graph; - - assert(out->nb_inputs == 1); - assert(in->nb_outputs == 1); - - ok = true; -error: - - if (!ok) { - MP_FATAL(af, "Can't configure libavfilter graph.\n"); - avfilter_graph_free(&graph); - } - avfilter_inout_free(&inputs); - avfilter_inout_free(&outputs); - talloc_free(tmp); - return ok; -} - -static void reset(struct af_instance *af) -{ - if (!recreate_graph(af, &af->fmt_in)) - MP_FATAL(af, "Can't recreate libavfilter filter after a seek reset.\n"); -} - -static int control(struct af_instance *af, int cmd, void *arg) -{ - struct priv *p = 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 (af_to_avformat(in->format) == AV_SAMPLE_FMT_NONE) - mp_audio_set_format(in, AF_FORMAT_FLOAT); - - // Removing this requires fixing AVFrame.data vs. AVFrame.extended_data - if (in->channels.num > AV_NUM_DATA_POINTERS) - return AF_ERROR; - - if (!mp_chmap_is_lavc(&in->channels)) - mp_chmap_reorder_to_lavc(&in->channels); // will always work - - if (!recreate_graph(af, in)) - return AF_ERROR; - - AVFilterLink *l_out = p->out->inputs[0]; - - out->rate = l_out->sample_rate; - - mp_audio_set_format(out, af_from_avformat(l_out->format)); - - struct mp_chmap out_cm; - mp_chmap_from_lavc(&out_cm, l_out->channel_layout); - mp_audio_set_channels(out, &out_cm); - - if (!mp_audio_config_valid(out) || out->channels.num > AV_NUM_DATA_POINTERS) - return AF_ERROR; - - p->timebase_out = l_out->time_base; - - return mp_audio_config_equals(in, &orig_in) ? AF_OK : AF_FALSE; - } - case AF_CONTROL_COMMAND: { - if (!p->graph) - break; - char **args = arg; - return avfilter_graph_send_command(p->graph, "all", - args[0], args[1], &(char){0}, 0, 0) - >= 0 ? CONTROL_OK : CONTROL_ERROR; - } - case AF_CONTROL_GET_METADATA: - if (p->metadata) { - *(struct mp_tags *)arg = *p->metadata; - return CONTROL_OK; - } - return CONTROL_NA; - case AF_CONTROL_RESET: - reset(af); - return AF_OK; - } - return AF_UNKNOWN; -} - -static void get_metadata_from_av_frame(struct af_instance *af, AVFrame *frame) -{ -#if LIBAVUTIL_VERSION_MICRO >= 100 - struct priv *p = af->priv; - if (!p->metadata) - p->metadata = talloc_zero(p, struct mp_tags); - - mp_tags_copy_from_av_dictionary(p->metadata, frame->metadata); -#endif -} - -static int filter_frame(struct af_instance *af, struct mp_audio *data) -{ - struct priv *p = af->priv; - AVFrame *frame = NULL; - - if (p->eof && data) - reset(af); - - if (!p->graph) - goto error; - - if (!data) { - if (p->eof) - return 0; - p->eof = true; - } - - if (data) { - frame = mp_audio_to_avframe_and_unref(data); - data = NULL; - if (!frame) - goto error; - - // Timebase is 1/sample_rate - frame->pts = p->samples_in; - p->samples_in += frame->nb_samples; - } - - if (av_buffersrc_add_frame(p->in, frame) < 0) - goto error; - - av_frame_free(&frame); - talloc_free(data); - return 0; -error: - av_frame_free(&frame); - talloc_free(data); - return -1; -} - -static int filter_out(struct af_instance *af) -{ - struct priv *p = af->priv; - - if (!p->graph) - goto error; - - AVFrame *frame = av_frame_alloc(); - if (!frame) - goto error; - - int err = av_buffersink_get_frame(p->out, frame); - if (err == AVERROR(EAGAIN) || err == AVERROR_EOF) { - // Not an error situation - no more output buffers in queue. - // AVERROR_EOF means we shouldn't even give the filter more - // input, but we don't handle that completely correctly. - av_frame_free(&frame); - p->eof |= err == AVERROR_EOF; - return 0; - } - - struct mp_audio *out = mp_audio_from_avframe(frame); - if (!out) - goto error; - - mp_audio_copy_config(out, af->data); - - if (frame->pts != AV_NOPTS_VALUE) { - double in_time = p->samples_in / (double)af->fmt_in.rate; - double out_time = frame->pts * av_q2d(p->timebase_out); - // Need pts past the last output sample. - out_time += out->samples / (double)out->rate; - - af->delay = in_time - out_time; - } - - get_metadata_from_av_frame(af, frame); - af_add_output_frame(af, out); - av_frame_free(&frame); - return 0; -error: - av_frame_free(&frame); - return -1; -} - -static void uninit(struct af_instance *af) -{ - destroy_graph(af); -} - -static int af_open(struct af_instance *af) -{ - struct priv *p = af->priv; - - af->control = control; - af->uninit = uninit; - af->filter_frame = filter_frame; - af->filter_out = filter_out; - - if (p->is_bridge) { - if (!p->cfg_filter_name) { - MP_ERR(af, "Filter name not set!\n"); - return 0; - } - if (!avfilter_get_by_name(p->cfg_filter_name)) { - MP_ERR(af, "libavfilter filter '%s' not found!\n", p->cfg_filter_name); - return 0; - } - } - return AF_OK; -} - -#define OPT_BASE_STRUCT struct priv - -const struct af_info af_info_lavfi = { - .info = "libavfilter bridge", - .name = "lavfi", - .open = af_open, - .priv_size = sizeof(struct priv), - .options = (const struct m_option[]) { - OPT_STRING("graph", cfg_graph, 0), - OPT_KEYVALUELIST("o", cfg_avopts, 0), - {0} - }, -}; - -const struct af_info af_info_lavfi_bridge = { - .info = "libavfilter bridge (explicit options)", - .name = "lavfi-bridge", - .open = af_open, - .priv_size = sizeof(struct priv), - .priv_defaults = &(const struct priv){ - .is_bridge = true, - }, - .options = (const struct m_option[]) { - OPT_STRING("name", cfg_filter_name, M_OPT_MIN, .min = 1), - OPT_KEYVALUELIST("opts", cfg_filter_opts, 0), - OPT_STRING("graph", cfg_graph, 0), - OPT_KEYVALUELIST("o", cfg_avopts, 0), - {0} - }, -}; |