#include <ctype.h>
#include <string.h>
+#include "graphparser.h"
#include "avfilter.h"
#include "avfiltergraph.h"
+#include "parseutils.h"
-/**
- * For use in av_log
- */
-static const char *log_name(void *p)
-{
- return "Filter parser";
-}
+#define WHITESPACES " \n\t"
-static const AVClass filter_parser_class = {
- "Filter parser",
- log_name
-};
+static int link_filter(AVFilterContext *src, int srcpad,
+ AVFilterContext *dst, int dstpad,
+ AVClass *log_ctx)
+{
+ if(avfilter_link(src, srcpad, dst, dstpad)) {
+ av_log(log_ctx, AV_LOG_ERROR,
+ "cannot create the link %s:%d -> %s:%d\n",
+ src->filter->name, srcpad, dst->filter->name, dstpad);
+ return -1;
+ }
-static const AVClass *log_ctx = &filter_parser_class;
+ return 0;
+}
-static int create_filter(AVFilterGraph *ctx, int index, char *name,
- char *args)
+/**
+ * Parse "[linkname]"
+ * @param name a pointer (that need to be free'd after use) to the name between
+ * parenthesis
+ */
+static char *parse_link_name(const char **buf, AVClass *log_ctx)
{
- AVFilterContext *filt;
+ const char *start = *buf;
+ char *name;
+ (*buf)++;
- AVFilter *filterdef;
- char tmp[20];
+ name = av_get_token(buf, "]");
- 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;
+ if(!name[0]) {
+ av_log(log_ctx, AV_LOG_ERROR,
+ "Bad (empty?) label found in the following: \"%s\".\n", start);
+ goto fail;
}
- if (avfilter_graph_add_filter(ctx, filt) < 0)
- return -1;
-
- if(avfilter_init_filter(filt, args, NULL)) {
- av_log(&log_ctx, AV_LOG_ERROR,
- "error initializing filter '%s'\n", name);
- return -1;
+ if(*(*buf)++ != ']') {
+ av_log(log_ctx, AV_LOG_ERROR,
+ "Mismatched '[' found in the following: \"%s\".\n", start);
+ fail:
+ av_freep(&name);
}
- return 0;
+ return name;
}
-static int link_filter(AVFilterGraph *ctx, int src, int srcpad,
- int dst, int dstpad)
+static AVFilterContext *create_filter(AVFilterGraph *ctx, int index,
+ const char *filt_name, const char *args,
+ AVClass *log_ctx)
{
- AVFilterContext *filt, *filtb;
+ AVFilterContext *filt_ctx;
- char tmp[20];
+ AVFilter *filt;
+ char inst_name[30];
- 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(inst_name, sizeof(inst_name), "Parsed filter %d", index);
+
+ filt = avfilter_get_by_name(filt_name);
+
+ if(!filt) {
+ av_log(log_ctx, AV_LOG_ERROR,
+ "no such filter: '%s'\n", filt_name);
+ return NULL;
}
- 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;
+
+ filt_ctx = avfilter_open(filt, inst_name);
+ if(!filt_ctx) {
+ av_log(log_ctx, AV_LOG_ERROR,
+ "error creating filter '%s'\n", filt_name);
+ return NULL;
}
- 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;
+
+ if(avfilter_graph_add_filter(ctx, filt_ctx) < 0) {
+ avfilter_destroy(filt_ctx);
+ return NULL;
}
- return 0;
-}
+ if(avfilter_init_filter(filt_ctx, args, NULL)) {
+ av_log(log_ctx, AV_LOG_ERROR,
+ "error initializing filter '%s' with args '%s'\n", filt_name, args);
+ return NULL;
+ }
-static void consume_whitespace(const char **buf)
-{
- *buf += strspn(*buf, " \n\t");
+ return filt_ctx;
}
/**
- * Copy the first size bytes of input string to a null-terminated string,
- * removing any control character. Ex: "aaa'bb'c\'c\\" -> "aaabbc'c\"
+ * Parse "filter=params"
*/
-static void copy_unquoted(char *out, const char *in, int size)
+static AVFilterContext *parse_filter(const char **buf, AVFilterGraph *graph,
+ int index, AVClass *log_ctx)
{
- 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];
+ char *opts = NULL;
+ char *name = av_get_token(buf, "=,[");
+ AVFilterContext *ret;
+
+ if(**buf == '=') {
+ (*buf)++;
+ opts = av_get_token(buf, "[],\n");
}
- *out=0;
+
+ ret = create_filter(graph, index, name, opts, log_ctx);
+ av_free(name);
+ av_free(opts);
+ return ret;
}
-/**
- * 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)
+static void free_inout(AVFilterInOut *head)
{
- const char *start;
- char *ret;
- int size;
-
- consume_whitespace(buf);
-
- if (!(**buf))
- return av_mallocz(1);
+ while(head) {
+ AVFilterInOut *next = head->next;
+ av_free(head->name);
+ av_free(head);
+ head = next;
+ }
+}
- start = *buf;
+static AVFilterInOut *extract_inout(const char *label, AVFilterInOut **links)
+{
+ AVFilterInOut *ret;
- while(1) {
- *buf += strcspn(*buf, " ()=,'\\");
- if (**buf == '\\')
- *buf+=2;
- else
- break;
- }
+ while(*links && strcmp((*links)->name, label))
+ links = &((*links)->next);
- 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
- }
+ ret = *links;
- size = *buf - start + 1;
- ret = av_malloc(size);
- copy_unquoted(ret, start, size-1);
+ if(ret)
+ *links = ret->next;
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)
+static void insert_inout(AVFilterInOut **inouts, AVFilterInOut *element)
{
- (*buf)++;
+ element->next = *inouts;
+ *inouts = element;
+}
- *name = consume_string(buf);
+static int link_filter_inouts(AVFilterContext *filter,
+ AVFilterInOut **curr_inputs,
+ AVFilterInOut **open_inputs, AVClass *log_ctx)
+{
+ int pad = filter->input_count;
+
+ while(pad--) {
+ AVFilterInOut *p = *curr_inputs;
+ if(!p) {
+ av_log(log_ctx, AV_LOG_ERROR,
+ "Not enough inputs specified for the \"%s\" filter.\n",
+ filter->filter->name);
+ return -1;
+ }
- if (!*name[0])
- goto fail;
+ *curr_inputs = (*curr_inputs)->next;
- if (*(*buf)++ != ')')
- goto fail;
+ if(p->filter) {
+ if(link_filter(p->filter, p->pad_idx, filter, pad, log_ctx))
+ return -1;
+ av_free(p->name);
+ av_free(p);
+ } else {
+ p->filter = filter;
+ p->pad_idx = pad;
+ insert_inout(open_inputs, p);
+ }
+ }
- return;
- fail:
- av_freep(name);
- av_log(&log_ctx, AV_LOG_ERROR, "Could not parse link name!\n");
+ if(*curr_inputs) {
+ av_log(log_ctx, AV_LOG_ERROR,
+ "Too many inputs specified for the \"%s\" filter.\n",
+ filter->filter->name);
+ return -1;
+ }
+
+ pad = filter->output_count;
+ while(pad--) {
+ AVFilterInOut *currlinkn = av_mallocz(sizeof(AVFilterInOut));
+ currlinkn->filter = filter;
+ currlinkn->pad_idx = pad;
+ insert_inout(curr_inputs, currlinkn);
+ }
+
+ return 0;
}
-/**
- * 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)
+static int parse_inputs(const char **buf, AVFilterInOut **curr_inputs,
+ AVFilterInOut **open_outputs, AVClass *log_ctx)
{
- char *name, *opts;
- name = consume_string(buf);
+ int pad = 0;
- if (**buf == '=') {
- (*buf)++;
- opts = consume_string(buf);
- } else {
- opts = NULL;
- }
+ while(**buf == '[') {
+ char *name = parse_link_name(buf, log_ctx);
+ AVFilterInOut *match;
- return create_filter(graph, index, name, opts);
-}
+ if(!name)
+ return -1;
-enum LinkType {
- LinkTypeIn,
- LinkTypeOut,
-};
+ /* First check if the label is not in the open_outputs list */
+ match = extract_inout(name, open_outputs);
-/**
- * A linked-list of the inputs/outputs of the filter chain.
- */
-typedef struct AVFilterInOut {
- enum LinkType type;
- char *name;
- int instance;
- int pad_idx;
+ if(match) {
+ av_free(name);
+ } else {
+ /* Not in the list, so add it as an input */
+ match = av_mallocz(sizeof(AVFilterInOut));
+ match->name = name;
+ match->pad_idx = pad;
+ }
- struct AVFilterInOut *next;
-} AVFilterInOut;
+ insert_inout(curr_inputs, match);
-static void free_inout(AVFilterInOut *head)
-{
- while (head) {
- AVFilterInOut *next;
- next = head->next;
- av_free(head);
- head = next;
+ *buf += strspn(*buf, WHITESPACES);
+ pad++;
}
-}
-/**
- * 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)
+static int parse_outputs(const char **buf, AVFilterInOut **curr_inputs,
+ AVFilterInOut **open_inputs,
+ AVFilterInOut **open_outputs, AVClass *log_ctx)
{
- AVFilterInOut *inout=NULL;
- AVFilterInOut *head=NULL;
-
- int index = 0;
- char chr = 0;
int pad = 0;
- int has_out = 0;
- char tmp[20];
- AVFilterContext *filt;
+ while(**buf == '[') {
+ char *name = parse_link_name(buf, log_ctx);
+ AVFilterInOut *match;
- consume_whitespace(&filters);
-
- do {
- int oldpad = pad;
+ AVFilterInOut *input = *curr_inputs;
+ *curr_inputs = (*curr_inputs)->next;
- pad = parse_inouts(&filters, &inout, chr == ',', LinkTypeIn, index);
+ if(!name)
+ return -1;
- if (parse_filter(&filters, graph, index) < 0)
- goto fail;
+ /* First check if the label is not in the open_inputs list */
+ match = extract_inout(name, open_inputs);
- // 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(match) {
+ if(link_filter(input->filter, input->pad_idx,
+ match->filter, match->pad_idx, log_ctx) < 0)
+ return -1;
+ av_free(match->name);
+ av_free(name);
+ av_free(match);
+ av_free(input);
+ } else {
+ /* Not in the list, so add the first input as a open_output */
+ input->name = name;
+ insert_inout(open_outputs, input);
}
+ *buf += strspn(*buf, WHITESPACES);
+ pad++;
+ }
- if(chr == ',') {
- if (link_filter(graph, index-1, oldpad, index, 0) < 0)
- goto fail;
+ return pad;
+}
- }
- pad = parse_inouts(&filters, &inout, 0, LinkTypeOut, index);
- chr = *filters++;
- index++;
- } while (chr == ',' || chr == ';');
+int avfilter_graph_parse(AVFilterGraph *graph, const char *filters,
+ AVFilterInOut *open_inputs,
+ AVFilterInOut *open_outputs, AVClass *log_ctx)
+{
+ int index = 0;
+ char chr = 0;
- head = inout;
- for (; inout != NULL; inout = inout->next) {
- if (inout->instance == -1)
- continue; // Already processed
+ AVFilterInOut *curr_inputs = NULL;
- 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;
- }
+ do {
+ AVFilterContext *filter;
+ filters += strspn(filters, WHITESPACES);
- 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;
- }
+ if(parse_inputs(&filters, &curr_inputs, &open_outputs, log_ctx) < 0)
+ goto fail;
- } else {
- AVFilterInOut *p, *src, *dst;
- for (p = inout->next;
- p && strcmp(p->name,inout->name); p = p->next);
+ filter = parse_filter(&filters, graph, index, log_ctx);
- 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(!filter)
+ goto fail;
- if (link_filter(graph, src->instance, src->pad_idx, dst->instance, dst->pad_idx) < 0)
+ if(filter->input_count == 1 && !curr_inputs && !index) {
+ /* First input can be omitted if it is "[in]" */
+ const char *tmp = "[in]";
+ if(parse_inputs(&tmp, &curr_inputs, &open_outputs, log_ctx) < 0)
goto fail;
-
- src->instance = -1;
- dst->instance = -1;
}
- }
- free_inout(head);
+ if(link_filter_inouts(filter, &curr_inputs, &open_inputs, log_ctx) < 0)
+ goto fail;
- 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");
+ if(parse_outputs(&filters, &curr_inputs, &open_inputs, &open_outputs,
+ log_ctx) < 0)
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");
+ filters += strspn(filters, WHITESPACES);
+ chr = *filters++;
+
+ if(chr == ';' && curr_inputs) {
+ av_log(log_ctx, AV_LOG_ERROR,
+ "Could not find a output to link when parsing \"%s\"\n",
+ filters - 1);
goto fail;
}
+ index++;
+ } while(chr == ',' || chr == ';');
+ if (chr) {
+ av_log(log_ctx, AV_LOG_ERROR,
+ "Unable to parse graph description substring: \"%s\"\n",
+ filters - 1);
+ goto fail;
+ }
+
+ if(open_inputs && !strcmp(open_inputs->name, "out") && curr_inputs) {
+ /* Last output can be omitted if it is "[out]" */
+ const char *tmp = "[out]";
+ if(parse_outputs(&tmp, &curr_inputs, &open_inputs,
+ &open_outputs, log_ctx) < 0)
+ goto fail;
}
return 0;
fail:
- free_inout(head);
- avfilter_destroy_graph(graph);
+ avfilter_graph_destroy(graph);
+ free_inout(open_inputs);
+ free_inout(open_outputs);
+ free_inout(curr_inputs);
return -1;
}