]> git.sesse.net Git - ffmpeg/blobdiff - libavfilter/graphparser.c
Move code out of if
[ffmpeg] / libavfilter / graphparser.c
index 0a77d07ecf623174dfa5dcd345a9ae45ddf9515f..ad1e8d473ae0b1db338fa073262d493a6bea38e2 100644 (file)
 #include "avfilter.h"
 #include "avfiltergraph.h"
 
-/**
- * For use in av_log
- */
-static const char *log_name(void *p)
-{
-    return "Filter parser";
-}
-
-static const AVClass filter_parser_class = {
-    "Filter parser",
-    log_name
-};
-
-static const AVClass *log_ctx = &filter_parser_class;
-
-static int create_filter(AVFilterGraph *ctx, int index, char *name,
-                         char *args)
+static AVFilterContext *create_filter(AVFilterGraph *ctx, int index,
+                                      const char *name, const char *args,
+                                      AVClass *log_ctx)
 {
     AVFilterContext *filt;
 
     AVFilter *filterdef;
-    char tmp[20];
+    char inst_name[30];
+
+    snprintf(inst_name, sizeof(inst_name), "Parsed filter %d", index);
+
+    if(!(filterdef = avfilter_get_by_name(name))) {
+        av_log(log_ctx, AV_LOG_ERROR,
+               "no such filter: '%s'\n", name);
+        return NULL;
+    }
 
-    snprintf(tmp, 20, "%d", index);
-    if(!(filterdef = avfilter_get_by_name(name)) ||
-       !(filt = avfilter_open(filterdef, tmp))) {
-        av_log(&log_ctx, AV_LOG_ERROR,
+    if(!(filt = avfilter_open(filterdef, inst_name))) {
+        av_log(log_ctx, AV_LOG_ERROR,
                "error creating filter '%s'\n", name);
-        return -1;
+        return NULL;
     }
 
-    if (avfilter_graph_add_filter(ctx, filt) < 0)
-        return -1;
+    if(avfilter_graph_add_filter(ctx, filt) < 0)
+        return NULL;
 
     if(avfilter_init_filter(filt, args, NULL)) {
-        av_log(&log_ctx, AV_LOG_ERROR,
-               "error initializing filter '%s'\n", name);
-        return -1;
+        av_log(log_ctx, AV_LOG_ERROR,
+               "error initializing filter '%s' with args '%s'\n", name, args);
+        return NULL;
     }
 
-    return 0;
+    return filt;
 }
 
-static int link_filter(AVFilterGraph *ctx, int src, int srcpad,
-                       int dst, int dstpad)
+static int link_filter(AVFilterContext *src, int srcpad,
+                       AVFilterContext *dst, int dstpad,
+                       AVClass *log_ctx)
 {
-    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");
+    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;
     }
 
@@ -99,117 +80,76 @@ 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;
+    char *out = av_malloc(strlen(*buf) + 1);
+    char *ret = out;
 
     consume_whitespace(buf);
 
-    if (!(**buf))
-        return av_mallocz(1);
-
-    start = *buf;
-
-    while(1) {
-        *buf += strcspn(*buf, " ()=,'\\");
-        if (**buf == '\\')
-            *buf+=2;
-        else
+    do{
+        char c = *(*buf)++;
+        switch (c) {
+        case '\\':
+            *out++= *(*buf)++;
             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
-    }
+        case '\'':
+            while(**buf && **buf != '\'')
+                *out++= *(*buf)++;
+            if(**buf) (*buf)++;
+            break;
+        case 0:
+        case ']':
+        case '[':
+        case '=':
+        case ',':
+        case ';':
+        case ' ':
+        case '\n':
+            *out++= 0;
+            break;
+        default:
+            *out++= c;
+        }
+    } while(out[-1]);
 
-    size = *buf - start + 1;
-    ret = av_malloc(size);
-    copy_unquoted(ret, start, size-1);
+    (*buf)--;
+    consume_whitespace(buf);
 
     return ret;
 }
 
 /**
- * Parse "(linkname)"
+ * 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 parse_link_name(const char **buf, char **name, AVClass *log_ctx)
 {
+    const char *start = *buf;
     (*buf)++;
 
     *name = consume_string(buf);
 
-    if (!*name[0])
-        goto fail;
-
-    if (*(*buf)++ != ')')
+    if(!*name[0]) {
+        av_log(log_ctx, AV_LOG_ERROR,
+               "Bad (empty?) label found in the following: \"%s\".\n", start);
         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);
+    if(*(*buf)++ != ']') {
+        av_log(log_ctx, AV_LOG_ERROR,
+               "Mismatched '[' found in the following: \"%s\".\n", start);
+    fail:
+        av_freep(name);
+    }
 }
 
+
 enum LinkType {
     LinkTypeIn,
     LinkTypeOut,
@@ -220,8 +160,8 @@ enum LinkType {
  */
 typedef struct AVFilterInOut {
     enum LinkType type;
-    char *name;
-    int instance;
+    const char *name;
+    AVFilterContext *filter;
     int pad_idx;
 
     struct AVFilterInOut *next;
@@ -230,161 +170,274 @@ typedef struct AVFilterInOut {
 static void free_inout(AVFilterInOut *head)
 {
     while (head) {
-        AVFilterInOut *next;
-        next = head->next;
+        AVFilterInOut *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)
+static AVFilterInOut *extract_inout(const char *label, AVFilterInOut **links)
+{
+    AVFilterInOut *ret;
+
+    while(*links && strcmp((*links)->name, label))
+        links = &((*links)->next);
+
+    ret = *links;
+
+    if(ret)
+        *links = ret->next;
+
+    return ret;
+}
+
+
+static int link_filter_inouts(AVFilterContext *filter,
+                              AVFilterInOut **currInputs,
+                              AVFilterInOut **openLinks, AVClass *log_ctx)
 {
-    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;
+    int pad = 0;
+
+    pad = filter->input_count;
+    while(pad--) {
+        AVFilterInOut *p= *currInputs;
+        *currInputs = (*currInputs)->next;
+        if(!p) {
+            av_log(log_ctx, AV_LOG_ERROR,
+                   "Not enough inputs specified for the \"%s\" filter.\n",
+                   filter->name);
+            return -1;
+        }
+
+        if(p->filter) {
+            if(link_filter(p->filter, p->pad_idx, filter, pad, log_ctx))
+                return -1;
+            av_free(p);
+        } else {
+            p->filter = filter;
+            p->pad_idx = pad;
+            p->next = *openLinks;
+            *openLinks = p;
+        }
     }
-    return pad;
+
+
+    if(*currInputs) {
+        av_log(log_ctx, AV_LOG_ERROR,
+               "Too many inputs specified for the \"%s\" filter.\n",
+               filter->name);
+        return -1;
+    }
+
+    pad = filter->output_count;
+    while(pad--) {
+        AVFilterInOut *currlinkn = av_malloc(sizeof(AVFilterInOut));
+        currlinkn->name    = NULL;
+        currlinkn->type    = LinkTypeOut;
+        currlinkn->filter  = filter;
+        currlinkn->pad_idx = pad;
+        currlinkn->next    = *currInputs;
+        *currInputs = currlinkn;
+    }
+
+    return 0;
 }
 
 /**
- * Parse a string describing a filter graph.
+ * 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
  */
-int avfilter_graph_parse_chain(AVFilterGraph *graph, const char *filters, AVFilterContext *in, int inpad, AVFilterContext *out, int outpad)
+static AVFilterContext *parse_filter(const char **buf, AVFilterGraph *graph,
+                                     int index, AVClass *log_ctx)
 {
-    AVFilterInOut *inout=NULL;
-    AVFilterInOut  *head=NULL;
+    char *opts;
+    char *name = consume_string(buf);
 
-    int index = 0;
-    char chr = 0;
-    int pad = 0;
-    int has_out = 0;
+    if(**buf == '=') {
+        (*buf)++;
+        opts = consume_string(buf);
+    } else {
+        opts = NULL;
+    }
 
-    char tmp[20];
-    AVFilterContext *filt;
+    return create_filter(graph, index, name, opts, log_ctx);
+}
 
-    consume_whitespace(&filters);
+static int parse_inputs(const char **buf, AVFilterInOut **currInputs,
+                        AVFilterInOut **openLinks, AVClass *log_ctx)
+{
+    int pad = 0;
 
-    do {
-        int oldpad = pad;
+    while (**buf == '[') {
+        char *name;
+        AVFilterInOut *link_to_add;
+        AVFilterInOut *match;
 
-        pad = parse_inouts(&filters, &inout, chr == ',', LinkTypeIn, index);
+        parse_link_name(buf, &name, log_ctx);
 
-        if (parse_filter(&filters, graph, index) < 0)
-            goto fail;
+        if(!name)
+            return -1;
 
-        // 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;
+        /* First check if the label is not in the openLinks list */
+        match = extract_inout(name, openLinks);
+
+        if(match) {
+            /* A label of a open link. Make it one of the inputs of the next
+               filter */
+            if (match->type != LinkTypeOut) {
+                av_log(log_ctx, AV_LOG_ERROR,
+                       "Label \"%s\" appears twice as input!\n", match->name);
+                return -1;
             }
-        }
 
-        if(chr == ',') {
-            if (link_filter(graph, index-1, oldpad, index, 0) < 0)
-                goto fail;
+            link_to_add = match;
+        } else {
+            /* Not in the list, so add it as an input */
+            link_to_add = av_malloc(sizeof(AVFilterInOut));
 
+            link_to_add->name    = name;
+            link_to_add->type    = LinkTypeIn;
+            link_to_add->filter  = NULL;
+            link_to_add->pad_idx = pad;
         }
-        pad = parse_inouts(&filters, &inout, 0, LinkTypeOut, index);
-        chr = *filters++;
-        index++;
-    } while (chr == ',' || chr == ';');
+        link_to_add->next = *currInputs;
+        *currInputs = link_to_add;
+        consume_whitespace(buf);
+        pad++;
+    }
 
-    head = inout;
-    for (; inout != NULL; inout = inout->next) {
-        if (inout->instance == -1)
-            continue; // Already processed
+    return pad;
+}
 
-        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;
-            }
+static int parse_outputs(const char **buf, AVFilterInOut **currInputs,
+                         AVFilterInOut **openLinks, AVClass *log_ctx)
+{
+    int pad = 0;
 
-            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;
-        }
+    while (**buf == '[') {
+        char *name;
+        AVFilterInOut *match;
 
-        } else {
-            AVFilterInOut *p, *src, *dst;
-            for (p = inout->next;
-                 p && strcmp(p->name,inout->name); p = p->next);
+        AVFilterInOut *input = *currInputs;
+        *currInputs = (*currInputs)->next;
 
-            if (!p) {
-                av_log(&log_ctx, AV_LOG_ERROR, "Unmatched link: %s.\n",
-                       inout->name);
-                goto fail;
-            }
+        parse_link_name(buf, &name, log_ctx);
 
-            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(!name)
+            return -1;
 
-            if (link_filter(graph, src->instance, src->pad_idx, dst->instance, dst->pad_idx) < 0)
-                goto fail;
+        /* First check if the label is not in the openLinks list */
+        match = extract_inout(name, openLinks);
 
-            src->instance = -1;
-            dst->instance = -1;
+        if(match) {
+            /* A label of a open link. Link it. */
+            if (match->type != LinkTypeIn) {
+                av_log(log_ctx, AV_LOG_ERROR,
+                       "Label \"%s\" appears twice as output!\n", match->name);
+                return -1;
+            }
+
+            if(link_filter(input->filter, input->pad_idx,
+                           match->filter, match->pad_idx, log_ctx) < 0)
+                return -1;
+            av_free(match);
+            av_free(input);
+        } else {
+            /* Not in the list, so add the first input as a openLink */
+            input->next = *openLinks;
+            input->type = LinkTypeOut;
+            input->name = name;
+            *openLinks = input;
         }
+        consume_whitespace(buf);
+        pad++;
     }
 
-    free_inout(head);
+    return pad;
+}
+
+/**
+ * Parse a string describing a filter graph.
+ */
+int avfilter_parse_graph(AVFilterGraph *graph, const char *filters,
+                         AVFilterContext *in, int inpad,
+                         AVFilterContext *out, int outpad,
+                         AVClass *log_ctx)
+{
+    int index = 0;
+    char chr = 0;
+    int pad = 0;
 
-    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");
+    AVFilterInOut *currInputs=NULL;
+    AVFilterInOut *openLinks  = av_malloc(sizeof(AVFilterInOut));
+
+    openLinks->name    = "in";
+    openLinks->filter  = in;
+    openLinks->type    = LinkTypeOut;
+    openLinks->pad_idx = inpad;
+    openLinks->next    = av_malloc(sizeof(AVFilterInOut));
+
+    openLinks->next->name    = "out";
+    openLinks->next->filter  = out;
+    openLinks->next->type    = LinkTypeIn;
+    openLinks->next->pad_idx = outpad;
+    openLinks->next->next    = NULL;
+
+    do {
+        AVFilterContext *filter;
+        consume_whitespace(&filters);
+
+        pad = parse_inputs(&filters, &currInputs, &openLinks, log_ctx);
+
+        if(pad < 0)
+            goto fail;
+
+        if(!(filter = parse_filter(&filters, graph, index, log_ctx)))
             goto fail;
+
+        if(filter->input_count == 1 && !currInputs && !index) {
+            // First input can be ommitted if it is "[in]"
+            const char *tmp = "[in]";
+            pad = parse_inputs(&tmp, &currInputs, &openLinks, log_ctx);
+            if (pad < 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");
+        if(link_filter_inouts(filter, &currInputs, &openLinks, log_ctx) < 0)
+            goto fail;
+
+        pad = parse_outputs(&filters, &currInputs, &openLinks, log_ctx);
+
+        if(pad < 0)
+            goto fail;
+
+        consume_whitespace(&filters);
+        chr = *filters++;
+
+        if (chr == ';' && currInputs) {
+            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(openLinks && !strcmp(openLinks->name, "out") && currInputs) {
+        // Last output can be ommitted if it is "[out]"
+        const char *tmp = "[out]";
+        if(parse_outputs(&tmp, &currInputs, &openLinks, log_ctx) < 0)
+            goto fail;
     }
 
     return 0;
 
  fail:
-    free_inout(head);
     avfilter_destroy_graph(graph);
+    free_inout(openLinks);
+    free_inout(currInputs);
     return -1;
 }