/*
- * Filter graphs
- * copyright (c) 2007 Bobby Bingham
+ * filter graphs
+ * Copyright (c) 2008 Vitor Sessak
+ * Copyright (c) 2007 Bobby Bingham
*
* This file is part of FFmpeg.
*
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
+#include <ctype.h>
#include <string.h>
-#include <stddef.h>
-#include "avstring.h"
#include "avfilter.h"
#include "avfiltergraph.h"
+#include "internal.h"
-typedef struct AVFilterGraph {
- unsigned filter_count;
- AVFilterContext **filters;
-} GraphContext;
-
-static void uninit(AVFilterContext *ctx)
+AVFilterGraph *avfilter_graph_alloc(void)
{
- GraphContext *graph = ctx->priv;
+ return av_mallocz(sizeof(AVFilterGraph));
+}
- for(; graph->filter_count > 0; graph->filter_count --)
- avfilter_destroy(graph->filters[graph->filter_count - 1]);
+void avfilter_graph_free(AVFilterGraph *graph)
+{
+ 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);
}
-void avfilter_graph_add_filter(AVFilterContext *graphctx, AVFilterContext *filter)
+int avfilter_graph_add_filter(AVFilterGraph *graph, AVFilterContext *filter)
{
- GraphContext *graph = graphctx->priv;
+ AVFilterContext **filters = av_realloc(graph->filters,
+ sizeof(AVFilterContext*) * (graph->filter_count+1));
+ if (!filters)
+ return AVERROR(ENOMEM);
- graph->filters = av_realloc(graph->filters,
- sizeof(AVFilterContext*) * ++graph->filter_count);
- graph->filters[graph->filter_count - 1] = filter;
+ graph->filters = filters;
+ graph->filters[graph->filter_count++] = filter;
+
+ return 0;
}
-static AVFilterContext *create_filter_with_args(const char *filt, void *opaque)
+int avfilter_graph_create_filter(AVFilterContext **filt_ctx, AVFilter *filt,
+ const char *name, const char *args, void *opaque,
+ AVFilterGraph *graph_ctx)
{
- AVFilterContext *ret;
- char *filter = av_strdup(filt); /* copy - don't mangle the input string */
- char *name, *args;
+ 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;
- name = filter;
- if((args = strchr(filter, '='))) {
- /* ensure we at least have a name */
- if(args == filter)
- goto fail;
+fail:
+ if (*filt_ctx)
+ avfilter_free(*filt_ctx);
+ *filt_ctx = NULL;
+ return ret;
+}
- *args ++ = 0;
+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;
+ }
+ }
}
- av_log(NULL, AV_LOG_INFO, "creating filter \"%s\" with args \"%s\"\n",
- name, args ? args : "(none)");
+ return 0;
+}
- if((ret = avfilter_create_by_name(name, NULL))) {
- if(avfilter_init_filter(ret, args, opaque)) {
- av_log(NULL, AV_LOG_ERROR, "error initializing filter!\n");
- avfilter_destroy(ret);
- goto fail;
+int ff_avfilter_graph_config_links(AVFilterGraph *graph, AVClass *log_ctx)
+{
+ 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;
}
- } else av_log(NULL, AV_LOG_ERROR, "error creating filter!\n");
+ }
- av_free(filter);
+ return 0;
+}
- return ret;
+AVFilterContext *avfilter_graph_get_filter(AVFilterGraph *graph, char *name)
+{
+ int i;
+
+ for (i = 0; i < graph->filter_count; i++)
+ if (graph->filters[i]->name && !strcmp(name, graph->filters[i]->name))
+ return graph->filters[i];
-fail:
- av_free(filter);
return NULL;
}
-static int graph_load_chain(AVFilterContext *graphctx,
- unsigned count, char **filter_list, void **opaque,
- AVFilterContext **first, AVFilterContext **last)
+static int query_formats(AVFilterGraph *graph, AVClass *log_ctx)
{
- unsigned i;
- AVFilterContext *filters[2] = {NULL,NULL};
-
- for(i = 0; i < count; i ++) {
- void *op;
-
- if(opaque) op = opaque[i];
- else op = NULL;
-
- if(!(filters[1] = create_filter_with_args(filter_list[i], op)))
- goto fail;
- if(i == 0) {
- if(first) *first = filters[1];
- } else {
- if(avfilter_link(filters[0], 0, filters[1], 0)) {
- av_log(NULL, AV_LOG_ERROR, "error linking filters!\n");
- goto fail;
+ int i, j, ret;
+ int scaler_count = 0;
+ char inst_name[30];
+
+ /* 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++) {
+ AVFilterContext *filter = graph->filters[i];
+
+ 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;
+
+ scale->filter->query_formats(scale);
+ 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;
+ }
+ }
}
}
- avfilter_graph_add_filter(graphctx, filters[1]);
- filters[0] = filters[1];
}
- if(last) *last = filters[1];
return 0;
-
-fail:
- uninit(graphctx);
- if(first) *first = NULL;
- if(last) *last = NULL;
- return -1;
}
-static int graph_load_chain_from_string(AVFilterContext *ctx, const char *str,
- AVFilterContext **first,
- AVFilterContext **last)
+static void pick_format(AVFilterLink *link)
{
- int count, ret = 0;
- char **strings;
- char *filt;
-
- strings = av_malloc(sizeof(char *));
- strings[0] = av_strdup(str);
-
- filt = strchr(strings[0], ',');
- for(count = 1; filt; count ++) {
- if(filt == strings[count-1]) {
- ret = -1;
- goto done;
- }
-
- strings = av_realloc(strings, sizeof(char *) * (count+1));
- strings[count] = filt + 1;
- *filt = '\0';
- filt = strchr(strings[count], ',');
- }
+ if (!link || !link->in_formats)
+ return;
- ret = graph_load_chain(ctx, count, strings, NULL, first, last);
+ link->in_formats->format_count = 1;
+ link->format = link->in_formats->formats[0];
-done:
- av_free(strings[0]);
- av_free(strings);
-
- return ret;
+ avfilter_formats_unref(&link->in_formats);
+ avfilter_formats_unref(&link->out_formats);
}
-static int init(AVFilterContext *ctx, const char *args, void *opaque)
+static void pick_formats(AVFilterGraph *graph)
{
- AVFilterContext **filters = opaque;
+ int i, j;
- if(!args)
- return 0;
- if(!opaque)
- return -1;
+ for (i = 0; i < graph->filter_count; i++) {
+ AVFilterContext *filter = graph->filters[i];
- return graph_load_chain_from_string(ctx, args, filters, filters + 1);
+ for (j = 0; j < filter->input_count; j++)
+ pick_format(filter->inputs[j]);
+ for (j = 0; j < filter->output_count; j++)
+ pick_format(filter->outputs[j]);
+ }
}
-AVFilter vf_graph =
+int ff_avfilter_graph_config_formats(AVFilterGraph *graph, AVClass *log_ctx)
{
- .name = "graph",
- .author = "Bobby Bingham",
+ /* find supported formats from sub-filters, and merge along links */
+ if (query_formats(graph, log_ctx))
+ return -1;
- .priv_size = sizeof(GraphContext),
+ /* Once everything is merged, it's possible that we'll still have
+ * multiple valid media format choices. We pick the first one. */
+ pick_formats(graph);
- .init = init,
- .uninit = uninit,
+ return 0;
+}
+
+int avfilter_graph_config(AVFilterGraph *graphctx, AVClass *log_ctx)
+{
+ int ret;
- .inputs = (AVFilterPad[]) {{ .name = NULL, }},
- .outputs = (AVFilterPad[]) {{ .name = NULL, }},
-};
+ 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;
+}
projects / ffmpeg / blobdiff