/*
- * Filter graphs
+ * 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 "avfilter.h"
#include "avfiltergraph.h"
-struct AVFilterGraph {
- unsigned filter_count;
- AVFilterContext **filters;
-};
-
-AVFilterGraph *avfilter_create_graph(void)
-{
- return av_mallocz(sizeof(AVFilterGraph));
-}
-
-static void destroy_graph_filters(AVFilterGraph *graph)
+void avfilter_graph_destroy(AVFilterGraph *graph)
{
for(; graph->filter_count > 0; graph->filter_count --)
avfilter_destroy(graph->filters[graph->filter_count - 1]);
+ av_freep(&graph->scale_sws_opts);
av_freep(&graph->filters);
}
-void avfilter_destroy_graph(AVFilterGraph *graph)
-{
- destroy_graph_filters(graph);
- av_free(graph);
-}
-
-void avfilter_graph_add_filter(AVFilterGraph *graph, AVFilterContext *filter)
+int avfilter_graph_add_filter(AVFilterGraph *graph, AVFilterContext *filter)
{
graph->filters = av_realloc(graph->filters,
sizeof(AVFilterContext*) * ++graph->filter_count);
+
+ if (!graph->filters)
+ return -1;
+
graph->filters[graph->filter_count - 1] = filter;
+
+ return 0;
}
-static AVFilterContext *create_filter_with_args(char *filter)
+int avfilter_graph_check_validity(AVFilterGraph *graph, AVClass *log_ctx)
{
- AVFilterContext *ret;
- char *name, *args;
-
- name = filter;
- if((args = strchr(filter, '='))) {
- /* ensure we at least have a name */
- if(args == filter)
- return NULL;
+ 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;
+ }
+ }
- *args ++ = 0;
+ 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)) {
- av_log(NULL, AV_LOG_ERROR, "error initializing filter!\n");
- avfilter_destroy(ret);
- ret = NULL;
+int 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");
+ }
+
+ return 0;
+}
+
+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];
- return ret;
+ return NULL;
}
-int avfilter_graph_load_chain(AVFilterGraph *graph,
- unsigned count, char **filter_list,
- AVFilterContext **first, AVFilterContext **last)
+static int query_formats(AVFilterGraph *graph)
{
- unsigned i;
- AVFilterContext *filters[2] = {NULL,NULL};
-
- for(i = 0; i < count; i ++) {
- if(!(filters[1] = create_filter_with_args(filter_list[i])))
- 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;
+ 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)
+ 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);
+ scale =
+ avfilter_open(avfilter_get_by_name("scale"),inst_name);
+
+ snprintf(scale_args, sizeof(scale_args), "0:0:%s", graph->scale_sws_opts);
+ if(!scale || scale->filter->init(scale, scale_args, NULL) ||
+ avfilter_insert_filter(link, scale, 0, 0)) {
+ avfilter_destroy(scale);
+ return -1;
+ }
+
+ if (avfilter_graph_add_filter(graph, scale) < 0)
+ return -1;
+
+ 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))
+ return -1;
+ }
}
}
- avfilter_graph_add_filter(graph, filters[1]);
- filters[0] = filters[1];
}
- if(last) *last = filters[1];
return 0;
+}
+
+static void pick_format(AVFilterLink *link)
+{
+ if(!link || !link->in_formats)
+ return;
+
+ link->in_formats->format_count = 1;
+ link->format = link->in_formats->formats[0];
-fail:
- destroy_graph_filters(graph);
- if(first) *first = NULL;
- if(last) *last = NULL;
- return -1;
+ avfilter_formats_unref(&link->in_formats);
+ avfilter_formats_unref(&link->out_formats);
+}
+
+static void pick_formats(AVFilterGraph *graph)
+{
+ int i, j;
+
+ for(i = 0; i < graph->filter_count; i ++) {
+ AVFilterContext *filter = graph->filters[i];
+
+ 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]);
+ }
+}
+
+int avfilter_graph_config_formats(AVFilterGraph *graph)
+{
+ /* find supported formats from sub-filters, and merge along links */
+ if(query_formats(graph))
+ return -1;
+
+ /* Once everything is merged, it's possible that we'll still have
+ * multiple valid colorspace choices. We pick the first one. */
+ pick_formats(graph);
+
+ return 0;
}