/*
* filter graphs
- * copyright (c) 2008 Vitor Sessak
- * copyright (c) 2007 Bobby Bingham
+ * Copyright (c) 2008 Vitor Sessak
+ * Copyright (c) 2007 Bobby Bingham
*
* This file is part of FFmpeg.
*
#include "avfilter.h"
#include "avfiltergraph.h"
+#include "internal.h"
-/**
- * For use in av_log
- */
-static const char *log_name(void *p)
+AVFilterGraph *avfilter_graph_alloc(void)
+{
+ return av_mallocz(sizeof(AVFilterGraph));
+}
+
+void avfilter_graph_free(AVFilterGraph *graph)
{
- return "Filter parser";
+ for (; graph->filter_count > 0; graph->filter_count --)
+ avfilter_free(graph->filters[graph->filter_count - 1]);
+ av_freep(&graph->scale_sws_opts);
+ av_freep(&graph->filters);
}
-static const AVClass filter_parser_class = {
- "Filter parser",
- log_name
-};
+int avfilter_graph_add_filter(AVFilterGraph *graph, AVFilterContext *filter)
+{
+ AVFilterContext **filters = av_realloc(graph->filters,
+ sizeof(AVFilterContext*) * (graph->filter_count+1));
+ if (!filters)
+ return AVERROR(ENOMEM);
-static const AVClass *log_ctx = &filter_parser_class;
+ graph->filters = filters;
+ graph->filters[graph->filter_count++] = filter;
-void avfilter_destroy_graph(AVFilterGraph *graph)
+ return 0;
+}
+
+int avfilter_graph_create_filter(AVFilterContext **filt_ctx, AVFilter *filt,
+ const char *name, const char *args, void *opaque,
+ AVFilterGraph *graph_ctx)
{
- for(; graph->filter_count > 0; graph->filter_count --)
- avfilter_destroy(graph->filters[graph->filter_count - 1]);
- av_freep(&graph->filters);
+ int ret;
+
+ if ((ret = avfilter_open(filt_ctx, filt, name)) < 0)
+ goto fail;
+ if ((ret = avfilter_init_filter(*filt_ctx, args, opaque)) < 0)
+ goto fail;
+ if ((ret = avfilter_graph_add_filter(graph_ctx, *filt_ctx)) < 0)
+ goto fail;
+ return 0;
+
+fail:
+ if (*filt_ctx)
+ avfilter_free(*filt_ctx);
+ *filt_ctx = NULL;
+ return ret;
+}
+
+int ff_avfilter_graph_check_validity(AVFilterGraph *graph, AVClass *log_ctx)
+{
+ AVFilterContext *filt;
+ int i, j;
+
+ for (i = 0; i < graph->filter_count; i++) {
+ filt = graph->filters[i];
+
+ for (j = 0; j < filt->input_count; j++) {
+ if (!filt->inputs[j] || !filt->inputs[j]->src) {
+ av_log(log_ctx, AV_LOG_ERROR,
+ "Input pad \"%s\" for the filter \"%s\" of type \"%s\" not connected to any source\n",
+ filt->input_pads[j].name, filt->name, filt->filter->name);
+ return -1;
+ }
+ }
+
+ for (j = 0; j < filt->output_count; j++) {
+ if (!filt->outputs[j] || !filt->outputs[j]->dst) {
+ av_log(log_ctx, AV_LOG_ERROR,
+ "Output pad \"%s\" for the filter \"%s\" of type \"%s\" not connected to any destination\n",
+ filt->output_pads[j].name, filt->name, filt->filter->name);
+ return -1;
+ }
+ }
+ }
+
+ return 0;
}
-/* TODO: insert in sorted order */
-void avfilter_graph_add_filter(AVFilterGraph *graph, AVFilterContext *filter)
+int ff_avfilter_graph_config_links(AVFilterGraph *graph, AVClass *log_ctx)
{
- graph->filters = av_realloc(graph->filters,
- sizeof(AVFilterContext*) * ++graph->filter_count);
- graph->filters[graph->filter_count - 1] = filter;
+ AVFilterContext *filt;
+ int i, ret;
+
+ for (i=0; i < graph->filter_count; i++) {
+ filt = graph->filters[i];
+
+ if (!filt->output_count) {
+ if ((ret = avfilter_config_links(filt)))
+ return ret;
+ }
+ }
+
+ return 0;
}
-/* search intelligently, once we insert in order */
AVFilterContext *avfilter_graph_get_filter(AVFilterGraph *graph, char *name)
{
int i;
- if(!name)
- return NULL;
-
- for(i = 0; i < graph->filter_count; i ++)
- if(graph->filters[i]->name && !strcmp(name, graph->filters[i]->name))
+ for (i = 0; i < graph->filter_count; i++)
+ if (graph->filters[i]->name && !strcmp(name, graph->filters[i]->name))
return graph->filters[i];
return NULL;
}
-static int query_formats(AVFilterGraph *graph)
+static int query_formats(AVFilterGraph *graph, AVClass *log_ctx)
{
- int i, j;
+ int i, j, ret;
+ int scaler_count = 0;
+ char inst_name[30];
- /* ask all the sub-filters for their supported colorspaces */
- for(i = 0; i < graph->filter_count; i ++) {
- if(graph->filters[i]->filter->query_formats)
+ /* ask all the sub-filters for their supported media formats */
+ for (i = 0; i < graph->filter_count; i++) {
+ if (graph->filters[i]->filter->query_formats)
graph->filters[i]->filter->query_formats(graph->filters[i]);
else
avfilter_default_query_formats(graph->filters[i]);
}
/* go through and merge as many format lists as possible */
- for(i = 0; i < graph->filter_count; i ++) {
+ for (i = 0; i < graph->filter_count; i++) {
AVFilterContext *filter = graph->filters[i];
- for(j = 0; j < filter->input_count; j ++) {
- AVFilterLink *link;
- if(!(link = filter->inputs[j]))
- continue;
- if(link->in_formats != link->out_formats) {
- if(!avfilter_merge_formats(link->in_formats,
- link->out_formats)) {
- /* couldn't merge format lists. auto-insert scale filter */
+ for (j = 0; j < filter->input_count; j++) {
+ AVFilterLink *link = filter->inputs[j];
+ if (link && link->in_formats != link->out_formats) {
+ if (!avfilter_merge_formats(link->in_formats,
+ link->out_formats)) {
AVFilterContext *scale;
+ char scale_args[256];
+ /* couldn't merge format lists. auto-insert scale filter */
+ snprintf(inst_name, sizeof(inst_name), "auto-inserted scaler %d",
+ scaler_count++);
+ snprintf(scale_args, sizeof(scale_args), "0:0:%s", graph->scale_sws_opts);
+ if ((ret = avfilter_graph_create_filter(&scale, avfilter_get_by_name("scale"),
+ inst_name, scale_args, NULL, graph)) < 0)
+ return ret;
+ if ((ret = avfilter_insert_filter(link, scale, 0, 0)) < 0)
+ return ret;
- if(!(scale =
- avfilter_open(avfilter_get_by_name("scale"), NULL)))
- return -1;
- if(scale->filter->init(scale, NULL, NULL) ||
- avfilter_insert_filter(link, scale, 0, 0)) {
- avfilter_destroy(scale);
- return -1;
- }
-
- avfilter_graph_add_filter(graph, scale);
scale->filter->query_formats(scale);
- if(!avfilter_merge_formats(scale-> inputs[0]->in_formats,
- scale-> inputs[0]->out_formats) ||
- !avfilter_merge_formats(scale->outputs[0]->in_formats,
- scale->outputs[0]->out_formats))
+ if (((link = scale-> inputs[0]) &&
+ !avfilter_merge_formats(link->in_formats, link->out_formats)) ||
+ ((link = scale->outputs[0]) &&
+ !avfilter_merge_formats(link->in_formats, link->out_formats))) {
+ av_log(log_ctx, AV_LOG_ERROR,
+ "Impossible to convert between the formats supported by the filter "
+ "'%s' and the filter '%s'\n", link->src->name, link->dst->name);
return -1;
+ }
}
}
}
static void pick_format(AVFilterLink *link)
{
- if(!link || !link->in_formats)
+ if (!link || !link->in_formats)
return;
link->in_formats->format_count = 1;
{
int i, j;
- for(i = 0; i < graph->filter_count; i ++) {
+ for (i = 0; i < graph->filter_count; i++) {
AVFilterContext *filter = graph->filters[i];
- for(j = 0; j < filter->input_count; j ++)
+ for (j = 0; j < filter->input_count; j++)
pick_format(filter->inputs[j]);
- for(j = 0; j < filter->output_count; j ++)
+ for (j = 0; j < filter->output_count; j++)
pick_format(filter->outputs[j]);
}
}
-int avfilter_graph_config_formats(AVFilterGraph *graph)
+int ff_avfilter_graph_config_formats(AVFilterGraph *graph, AVClass *log_ctx)
{
/* find supported formats from sub-filters, and merge along links */
- if(query_formats(graph))
+ if (query_formats(graph, log_ctx))
return -1;
/* Once everything is merged, it's possible that we'll still have
- * multiple valid colorspace choices. We pick the first one. */
+ * multiple valid media format choices. We pick the first one. */
pick_formats(graph);
return 0;
}
-static int create_filter(AVFilterGraph *ctx, int index, char *name,
- char *args)
+int avfilter_graph_config(AVFilterGraph *graphctx, AVClass *log_ctx)
{
- AVFilterContext *filt;
+ int ret;
- AVFilter *filterdef;
- char tmp[20];
-
- snprintf(tmp, 20, "%d", index);
- if(!(filterdef = avfilter_get_by_name(name)) ||
- !(filt = avfilter_open(filterdef, tmp))) {
- av_log(&log_ctx, AV_LOG_ERROR,
- "error creating filter '%s'\n", name);
- return -1;
- }
- avfilter_graph_add_filter(ctx, filt);
- if(avfilter_init_filter(filt, args, NULL)) {
- av_log(&log_ctx, AV_LOG_ERROR,
- "error initializing filter '%s'\n", name);
- return -1;
- }
+ if ((ret = ff_avfilter_graph_check_validity(graphctx, log_ctx)))
+ return ret;
+ if ((ret = ff_avfilter_graph_config_formats(graphctx, log_ctx)))
+ return ret;
+ if ((ret = ff_avfilter_graph_config_links(graphctx, log_ctx)))
+ return ret;
return 0;
}
-
-static int link_filter(AVFilterGraph *ctx, int src, int srcpad,
- int dst, int dstpad)
-{
- AVFilterContext *filt, *filtb;
-
- char tmp[20];
-
- snprintf(tmp, 20, "%d", src);
- if(!(filt = avfilter_graph_get_filter(ctx, tmp))) {
- av_log(&log_ctx, AV_LOG_ERROR, "link source does not exist in graph\n");
- return -1;
- }
- snprintf(tmp, 20, "%d", dst);
- if(!(filtb = avfilter_graph_get_filter(ctx, tmp))) {
- av_log(&log_ctx, AV_LOG_ERROR, "link destination does not exist in graph\n");
- return -1;
- }
- if(avfilter_link(filt, srcpad, filtb, dstpad)) {
- av_log(&log_ctx, AV_LOG_ERROR, "cannot create link between source and destination filters\n");
- return -1;
- }
-
- return 0;
-}
-
-static void consume_whitespace(const char **buf)
-{
- *buf += strspn(*buf, " \n\t");
-}
-
-/**
- * Copy the first size bytes of input string to a null-terminated string,
- * removing any control character. Ex: "aaa'bb'c\'c\\" -> "aaabbc'c\"
- */
-static void copy_unquoted(char *out, const char *in, int size)
-{
- int i;
- for (i=0; i < size; i++) {
- if (in[i] == '\'')
- continue;
- else if (in[i] == '\\') {
- if (i+1 == size) {
- *out = 0;
- return;
- }
- i++;
- }
- *out++ = in[i];
- }
- *out=0;
-}
-
-/**
- * Consumes a string from *buf.
- * @return a copy of the consumed string, which should be free'd after use
- */
-static char *consume_string(const char **buf)
-{
- const char *start;
- char *ret;
- int size;
-
- consume_whitespace(buf);
-
- if (!(**buf))
- return av_mallocz(1);
-
- start = *buf;
-
- while(1) {
- *buf += strcspn(*buf, " ()=,'\\");
- if (**buf == '\\')
- *buf+=2;
- else
- break;
- }
-
- if (**buf == '\'') {
- const char *p = *buf;
- do {
- p++;
- p = strchr(p, '\'');
- } while (p && p[-1] == '\\');
- if (p)
- *buf = p + 1;
- else
- *buf += strlen(*buf); // Move the pointer to the null end byte
- }
-
- size = *buf - start + 1;
- ret = av_malloc(size);
- copy_unquoted(ret, start, size-1);
-
- return ret;
-}
-
-/**
- * Parse "(linkname)"
- * @arg name a pointer (that need to be free'd after use) to the name between
- * parenthesis
- */
-static void parse_link_name(const char **buf, char **name)
-{
- (*buf)++;
-
- *name = consume_string(buf);
-
- if (!*name[0])
- goto fail;
-
- if (*(*buf)++ != ')')
- goto fail;
-
- return;
- fail:
- av_freep(name);
- av_log(&log_ctx, AV_LOG_ERROR, "Could not parse link name!\n");
-}
-
-/**
- * Parse "filter=params"
- * @arg name a pointer (that need to be free'd after use) to the name of the
- * filter
- * @arg ars a pointer (that need to be free'd after use) to the args of the
- * filter
- */
-static int parse_filter(const char **buf, AVFilterGraph *graph, int index)
-{
- char *name, *opts;
- name = consume_string(buf);
-
- if (**buf == '=') {
- (*buf)++;
- opts = consume_string(buf);
- } else {
- opts = NULL;
- }
-
- return create_filter(graph, index, name, opts);
-}
-
-enum LinkType {
- LinkTypeIn,
- LinkTypeOut,
-};
-
-/**
- * A linked-list of the inputs/outputs of the filter chain.
- */
-typedef struct AVFilterInOut {
- enum LinkType type;
- char *name;
- int instance;
- int pad_idx;
-
- struct AVFilterInOut *next;
-} AVFilterInOut;
-
-static void free_inout(AVFilterInOut *head)
-{
- while (head) {
- AVFilterInOut *next;
- next = head->next;
- av_free(head);
- head = next;
- }
-}
-
-/**
- * Parse "(a1)(link2) ... (etc)"
- */
-static int parse_inouts(const char **buf, AVFilterInOut **inout, int firstpad,
- enum LinkType type, int instance)
-{
- int pad = firstpad;
- while (**buf == '(') {
- AVFilterInOut *inoutn = av_malloc(sizeof(AVFilterInOut));
- parse_link_name(buf, &inoutn->name);
- inoutn->type = type;
- inoutn->instance = instance;
- inoutn->pad_idx = pad++;
- inoutn->next = *inout;
- *inout = inoutn;
- }
- return pad;
-}
-
-/**
- * Parse a string describing a filter graph.
- */
-int avfilter_graph_parse_chain(AVFilterGraph *graph, const char *filters, AVFilterContext *in, int inpad, AVFilterContext *out, int outpad)
-{
- AVFilterInOut *inout=NULL;
- AVFilterInOut *head=NULL;
-
- int index = 0;
- char chr = 0;
- int pad = 0;
- int has_out = 0;
-
- char tmp[20];
- AVFilterContext *filt;
-
- consume_whitespace(&filters);
-
- do {
- int oldpad = pad;
-
- pad = parse_inouts(&filters, &inout, chr == ',', LinkTypeIn, index);
-
- if (parse_filter(&filters, graph, index) < 0)
- goto fail;
-
- // If the first filter has an input and none was given, it is
- // implicitly the input of the whole graph.
- if (pad == 0 && graph->filters[graph->filter_count-1]->input_count == 1) {
- snprintf(tmp, 20, "%d", index);
- if(!(filt = avfilter_graph_get_filter(graph, tmp))) {
- av_log(&log_ctx, AV_LOG_ERROR, "filter owning exported pad does not exist\n");
- goto fail;
- }
- if(avfilter_link(in, inpad, filt, 0)) {
- av_log(&log_ctx, AV_LOG_ERROR, "cannot create link between source and destination filters\n");
- goto fail;
- }
- }
-
- if(chr == ',') {
- if (link_filter(graph, index-1, oldpad, index, 0) < 0)
- goto fail;
-
- }
- pad = parse_inouts(&filters, &inout, 0, LinkTypeOut, index);
- chr = *filters++;
- index++;
- } while (chr == ',' || chr == ';');
-
- head = inout;
- for (; inout != NULL; inout = inout->next) {
- if (inout->instance == -1)
- continue; // Already processed
-
- if (!strcmp(inout->name, "in")) {
- snprintf(tmp, 20, "%d", inout->instance);
- if(!(filt = avfilter_graph_get_filter(graph, tmp))) {
- av_log(&log_ctx, AV_LOG_ERROR, "filter owning exported pad does not exist\n");
- goto fail;
- }
- if(avfilter_link(in, inpad, filt, inout->pad_idx)) {
- av_log(&log_ctx, AV_LOG_ERROR, "cannot create link between source and destination filters\n");
- goto fail;
- }
- } else if (!strcmp(inout->name, "out")) {
- has_out = 1;
- snprintf(tmp, 20, "%d", inout->instance);
- if(!(filt = avfilter_graph_get_filter(graph, tmp))) {
- av_log(&log_ctx, AV_LOG_ERROR, "filter owning exported pad does not exist\n");
- goto fail;
- }
-
- if(avfilter_link(filt, inout->pad_idx, out, outpad)) {
- av_log(&log_ctx, AV_LOG_ERROR, "cannot create link between source and destination filters\n");
- goto fail;
- }
-
- } else {
- AVFilterInOut *p, *src, *dst;
- for (p = inout->next;
- p && strcmp(p->name,inout->name); p = p->next);
-
- if (!p) {
- av_log(&log_ctx, AV_LOG_ERROR, "Unmatched link: %s.\n",
- inout->name);
- goto fail;
- }
-
- if (p->type == LinkTypeIn && inout->type == LinkTypeOut) {
- src = inout;
- dst = p;
- } else if (p->type == LinkTypeOut && inout->type == LinkTypeIn) {
- src = p;
- dst = inout;
- } else {
- av_log(&log_ctx, AV_LOG_ERROR, "Two links named '%s' are either both input or both output\n",
- inout->name);
- goto fail;
- }
-
- if (link_filter(graph, src->instance, src->pad_idx, dst->instance, dst->pad_idx) < 0)
- goto fail;
-
- src->instance = -1;
- dst->instance = -1;
- }
- }
-
- free_inout(head);
-
- if (!has_out) {
- snprintf(tmp, 20, "%d", index-1);
- if(!(filt = avfilter_graph_get_filter(graph, tmp))) {
- av_log(&log_ctx, AV_LOG_ERROR, "filter owning exported pad does not exist\n");
- goto fail;
- }
-
- if(avfilter_link(filt, pad, out, outpad)) {
- av_log(&log_ctx, AV_LOG_ERROR, "cannot create link between source and destination filters\n");
- goto fail;
- }
-
- }
-
- return 0;
-
- fail:
- free_inout(head);
- avfilter_destroy_graph(graph);
- return -1;
-}