git.sesse.net Git - ffmpeg/blob - libavfilter/avfiltergraph.c

   1 /*
   2  * filter graphs
   3  * Copyright (c) 2008 Vitor Sessak
   4  * Copyright (c) 2007 Bobby Bingham
   5  *
   6  * This file is part of FFmpeg.
   7  *
   8  * FFmpeg is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2.1 of the License, or (at your option) any later version.
  12  *
  13  * FFmpeg is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with FFmpeg; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21  */
  22
  23 #include "config.h"
  24
  25 #include <string.h>
  26
  27 #include "libavutil/avassert.h"
  28 #include "libavutil/avstring.h"
  29 #include "libavutil/bprint.h"
  30 #include "libavutil/channel_layout.h"
  31 #include "libavutil/imgutils.h"
  32 #include "libavutil/internal.h"
  33 #include "libavutil/opt.h"
  34 #include "libavutil/pixdesc.h"
  35
  36 #define FF_INTERNAL_FIELDS 1
  37 #include "framequeue.h"
  38
  39 #include "avfilter.h"
  40 #include "buffersink.h"
  41 #include "formats.h"
  42 #include "internal.h"
  43 #include "thread.h"
  44
  45 #define OFFSET(x) offsetof(AVFilterGraph, x)
  46 #define F AV_OPT_FLAG_FILTERING_PARAM
  47 #define V AV_OPT_FLAG_VIDEO_PARAM
  48 #define A AV_OPT_FLAG_AUDIO_PARAM
  49 static const AVOption filtergraph_options[] = {
  50     { "thread_type", "Allowed thread types", OFFSET(thread_type), AV_OPT_TYPE_FLAGS,
  51         { .i64 = AVFILTER_THREAD_SLICE }, 0, INT_MAX, F|V|A, "thread_type" },
  52         { "slice", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = AVFILTER_THREAD_SLICE }, .flags = F|V|A, .unit = "thread_type" },
  53     { "threads",     "Maximum number of threads", OFFSET(nb_threads),
  54         AV_OPT_TYPE_INT,   { .i64 = 0 }, 0, INT_MAX, F|V|A },
  55     {"scale_sws_opts"       , "default scale filter options"        , OFFSET(scale_sws_opts)        ,
  56         AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, F|V },
  57     {"aresample_swr_opts"   , "default aresample filter options"    , OFFSET(aresample_swr_opts)    ,
  58         AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, F|A },
  59     { NULL },
  60 };
  61
  62 static const AVClass filtergraph_class = {
  63     .class_name = "AVFilterGraph",
  64     .item_name  = av_default_item_name,
  65     .version    = LIBAVUTIL_VERSION_INT,
  66     .option     = filtergraph_options,
  67     .category   = AV_CLASS_CATEGORY_FILTER,
  68 };
  69
  70 #if !HAVE_THREADS
  71 void ff_graph_thread_free(AVFilterGraph *graph)
  72 {
  73 }
  74
  75 int ff_graph_thread_init(AVFilterGraph *graph)
  76 {
  77     graph->thread_type = 0;
  78     graph->nb_threads  = 1;
  79     return 0;
  80 }
  81 #endif
  82
  83 AVFilterGraph *avfilter_graph_alloc(void)
  84 {
  85     AVFilterGraph *ret = av_mallocz(sizeof(*ret));
  86     if (!ret)
  87         return NULL;
  88
  89     ret->internal = av_mallocz(sizeof(*ret->internal));
  90     if (!ret->internal) {
  91         av_freep(&ret);
  92         return NULL;
  93     }
  94
  95     ret->av_class = &filtergraph_class;
  96     av_opt_set_defaults(ret);
  97     ff_framequeue_global_init(&ret->internal->frame_queues);
  98
  99     return ret;
 100 }
 101
 102 void ff_filter_graph_remove_filter(AVFilterGraph *graph, AVFilterContext *filter)
 103 {
 104     int i, j;
 105     for (i = 0; i < graph->nb_filters; i++) {
 106         if (graph->filters[i] == filter) {
 107             FFSWAP(AVFilterContext*, graph->filters[i],
 108                    graph->filters[graph->nb_filters - 1]);
 109             graph->nb_filters--;
 110             filter->graph = NULL;
 111             for (j = 0; j<filter->nb_outputs; j++)
 112                 if (filter->outputs[j])
 113                     filter->outputs[j]->graph = NULL;
 114
 115             return;
 116         }
 117     }
 118 }
 119
 120 void avfilter_graph_free(AVFilterGraph **graph)
 121 {
 122     if (!*graph)
 123         return;
 124
 125     while ((*graph)->nb_filters)
 126         avfilter_free((*graph)->filters[0]);
 127
 128     ff_graph_thread_free(*graph);
 129
 130     av_freep(&(*graph)->sink_links);
 131
 132     av_freep(&(*graph)->scale_sws_opts);
 133     av_freep(&(*graph)->aresample_swr_opts);
 134 #if FF_API_LAVR_OPTS
 135     av_freep(&(*graph)->resample_lavr_opts);
 136 #endif
 137     av_freep(&(*graph)->filters);
 138     av_freep(&(*graph)->internal);
 139     av_freep(graph);
 140 }
 141
 142 int avfilter_graph_create_filter(AVFilterContext **filt_ctx, const AVFilter *filt,
 143                                  const char *name, const char *args, void *opaque,
 144                                  AVFilterGraph *graph_ctx)
 145 {
 146     int ret;
 147
 148     *filt_ctx = avfilter_graph_alloc_filter(graph_ctx, filt, name);
 149     if (!*filt_ctx)
 150         return AVERROR(ENOMEM);
 151
 152     ret = avfilter_init_str(*filt_ctx, args);
 153     if (ret < 0)
 154         goto fail;
 155
 156     return 0;
 157
 158 fail:
 159     if (*filt_ctx)
 160         avfilter_free(*filt_ctx);
 161     *filt_ctx = NULL;
 162     return ret;
 163 }
 164
 165 void avfilter_graph_set_auto_convert(AVFilterGraph *graph, unsigned flags)
 166 {
 167     graph->disable_auto_convert = flags;
 168 }
 169
 170 AVFilterContext *avfilter_graph_alloc_filter(AVFilterGraph *graph,
 171                                              const AVFilter *filter,
 172                                              const char *name)
 173 {
 174     AVFilterContext **filters, *s;
 175
 176     if (graph->thread_type && !graph->internal->thread_execute) {
 177         if (graph->execute) {
 178             graph->internal->thread_execute = graph->execute;
 179         } else {
 180             int ret = ff_graph_thread_init(graph);
 181             if (ret < 0) {
 182                 av_log(graph, AV_LOG_ERROR, "Error initializing threading: %s.\n", av_err2str(ret));
 183                 return NULL;
 184             }
 185         }
 186     }
 187
 188     s = ff_filter_alloc(filter, name);
 189     if (!s)
 190         return NULL;
 191
 192     filters = av_realloc(graph->filters, sizeof(*filters) * (graph->nb_filters + 1));
 193     if (!filters) {
 194         avfilter_free(s);
 195         return NULL;
 196     }
 197
 198     graph->filters = filters;
 199     graph->filters[graph->nb_filters++] = s;
 200
 201     s->graph = graph;
 202
 203     return s;
 204 }
 205
 206 /**
 207  * Check for the validity of graph.
 208  *
 209  * A graph is considered valid if all its input and output pads are
 210  * connected.
 211  *
 212  * @return >= 0 in case of success, a negative value otherwise
 213  */
 214 static int graph_check_validity(AVFilterGraph *graph, AVClass *log_ctx)
 215 {
 216     AVFilterContext *filt;
 217     int i, j;
 218
 219     for (i = 0; i < graph->nb_filters; i++) {
 220         const AVFilterPad *pad;
 221         filt = graph->filters[i];
 222
 223         for (j = 0; j < filt->nb_inputs; j++) {
 224             if (!filt->inputs[j] || !filt->inputs[j]->src) {
 225                 pad = &filt->input_pads[j];
 226                 av_log(log_ctx, AV_LOG_ERROR,
 227                        "Input pad \"%s\" with type %s of the filter instance \"%s\" of %s not connected to any source\n",
 228                        pad->name, av_get_media_type_string(pad->type), filt->name, filt->filter->name);
 229                 return AVERROR(EINVAL);
 230             }
 231         }
 232
 233         for (j = 0; j < filt->nb_outputs; j++) {
 234             if (!filt->outputs[j] || !filt->outputs[j]->dst) {
 235                 pad = &filt->output_pads[j];
 236                 av_log(log_ctx, AV_LOG_ERROR,
 237                        "Output pad \"%s\" with type %s of the filter instance \"%s\" of %s not connected to any destination\n",
 238                        pad->name, av_get_media_type_string(pad->type), filt->name, filt->filter->name);
 239                 return AVERROR(EINVAL);
 240             }
 241         }
 242     }
 243
 244     return 0;
 245 }
 246
 247 /**
 248  * Configure all the links of graphctx.
 249  *
 250  * @return >= 0 in case of success, a negative value otherwise
 251  */
 252 static int graph_config_links(AVFilterGraph *graph, AVClass *log_ctx)
 253 {
 254     AVFilterContext *filt;
 255     int i, ret;
 256
 257     for (i = 0; i < graph->nb_filters; i++) {
 258         filt = graph->filters[i];
 259
 260         if (!filt->nb_outputs) {
 261             if ((ret = avfilter_config_links(filt)))
 262                 return ret;
 263         }
 264     }
 265
 266     return 0;
 267 }
 268
 269 static int graph_check_links(AVFilterGraph *graph, AVClass *log_ctx)
 270 {
 271     AVFilterContext *f;
 272     AVFilterLink *l;
 273     unsigned i, j;
 274     int ret;
 275
 276     for (i = 0; i < graph->nb_filters; i++) {
 277         f = graph->filters[i];
 278         for (j = 0; j < f->nb_outputs; j++) {
 279             l = f->outputs[j];
 280             if (l->type == AVMEDIA_TYPE_VIDEO) {
 281                 ret = av_image_check_size2(l->w, l->h, INT64_MAX, l->format, 0, f);
 282                 if (ret < 0)
 283                     return ret;
 284             }
 285         }
 286     }
 287     return 0;
 288 }
 289
 290 AVFilterContext *avfilter_graph_get_filter(AVFilterGraph *graph, const char *name)
 291 {
 292     int i;
 293
 294     for (i = 0; i < graph->nb_filters; i++)
 295         if (graph->filters[i]->name && !strcmp(name, graph->filters[i]->name))
 296             return graph->filters[i];
 297
 298     return NULL;
 299 }
 300
 301 static void sanitize_channel_layouts(void *log, AVFilterChannelLayouts *l)
 302 {
 303     if (!l)
 304         return;
 305     if (l->nb_channel_layouts) {
 306         if (l->all_layouts || l->all_counts)
 307             av_log(log, AV_LOG_WARNING, "All layouts set on non-empty list\n");
 308         l->all_layouts = l->all_counts = 0;
 309     } else {
 310         if (l->all_counts && !l->all_layouts)
 311             av_log(log, AV_LOG_WARNING, "All counts without all layouts\n");
 312         l->all_layouts = 1;
 313     }
 314 }
 315
 316 static int filter_query_formats(AVFilterContext *ctx)
 317 {
 318     int ret, i;
 319     AVFilterFormats *formats;
 320     AVFilterChannelLayouts *chlayouts;
 321     AVFilterFormats *samplerates;
 322     enum AVMediaType type = ctx->inputs  && ctx->inputs [0] ? ctx->inputs [0]->type :
 323                             ctx->outputs && ctx->outputs[0] ? ctx->outputs[0]->type :
 324                             AVMEDIA_TYPE_VIDEO;
 325
 326     if ((ret = ctx->filter->query_formats(ctx)) < 0) {
 327         if (ret != AVERROR(EAGAIN))
 328             av_log(ctx, AV_LOG_ERROR, "Query format failed for '%s': %s\n",
 329                    ctx->name, av_err2str(ret));
 330         return ret;
 331     }
 332
 333     for (i = 0; i < ctx->nb_inputs; i++)
 334         sanitize_channel_layouts(ctx, ctx->inputs[i]->outcfg.channel_layouts);
 335     for (i = 0; i < ctx->nb_outputs; i++)
 336         sanitize_channel_layouts(ctx, ctx->outputs[i]->incfg.channel_layouts);
 337
 338     formats = ff_all_formats(type);
 339     if ((ret = ff_set_common_formats(ctx, formats)) < 0)
 340         return ret;
 341     if (type == AVMEDIA_TYPE_AUDIO) {
 342         samplerates = ff_all_samplerates();
 343         if ((ret = ff_set_common_samplerates(ctx, samplerates)) < 0)
 344             return ret;
 345         chlayouts = ff_all_channel_layouts();
 346         if ((ret = ff_set_common_channel_layouts(ctx, chlayouts)) < 0)
 347             return ret;
 348     }
 349     return 0;
 350 }
 351
 352 static int formats_declared(AVFilterContext *f)
 353 {
 354     int i;
 355
 356     for (i = 0; i < f->nb_inputs; i++) {
 357         if (!f->inputs[i]->outcfg.formats)
 358             return 0;
 359         if (f->inputs[i]->type == AVMEDIA_TYPE_AUDIO &&
 360             !(f->inputs[i]->outcfg.samplerates &&
 361               f->inputs[i]->outcfg.channel_layouts))
 362             return 0;
 363     }
 364     for (i = 0; i < f->nb_outputs; i++) {
 365         if (!f->outputs[i]->incfg.formats)
 366             return 0;
 367         if (f->outputs[i]->type == AVMEDIA_TYPE_AUDIO &&
 368             !(f->outputs[i]->incfg.samplerates &&
 369               f->outputs[i]->incfg.channel_layouts))
 370             return 0;
 371     }
 372     return 1;
 373 }
 374
 375 /**
 376  * Perform one round of query_formats() and merging formats lists on the
 377  * filter graph.
 378  * @return  >=0 if all links formats lists could be queried and merged;
 379  *          AVERROR(EAGAIN) some progress was made in the queries or merging
 380  *          and a later call may succeed;
 381  *          AVERROR(EIO) (may be changed) plus a log message if no progress
 382  *          was made and the negotiation is stuck;
 383  *          a negative error code if some other error happened
 384  */
 385 static int query_formats(AVFilterGraph *graph, AVClass *log_ctx)
 386 {
 387     int i, j, ret;
 388     int scaler_count = 0, resampler_count = 0;
 389     int count_queried = 0;        /* successful calls to query_formats() */
 390     int count_merged = 0;         /* successful merge of formats lists */
 391     int count_already_merged = 0; /* lists already merged */
 392     int count_delayed = 0;        /* lists that need to be merged later */
 393
 394     for (i = 0; i < graph->nb_filters; i++) {
 395         AVFilterContext *f = graph->filters[i];
 396         if (formats_declared(f))
 397             continue;
 398         if (f->filter->query_formats)
 399             ret = filter_query_formats(f);
 400         else
 401             ret = ff_default_query_formats(f);
 402         if (ret < 0 && ret != AVERROR(EAGAIN))
 403             return ret;
 404         /* note: EAGAIN could indicate a partial success, not counted yet */
 405         count_queried += ret >= 0;
 406     }
 407
 408     /* go through and merge as many format lists as possible */
 409     for (i = 0; i < graph->nb_filters; i++) {
 410         AVFilterContext *filter = graph->filters[i];
 411
 412         for (j = 0; j < filter->nb_inputs; j++) {
 413             AVFilterLink *link = filter->inputs[j];
 414             int convert_needed = 0;
 415
 416             if (!link)
 417                 continue;
 418
 419             if (link->incfg.formats != link->outcfg.formats
 420                 && link->incfg.formats && link->outcfg.formats)
 421                 if (!ff_can_merge_formats(link->incfg.formats, link->outcfg.formats,
 422                                           link->type))
 423                     convert_needed = 1;
 424             if (link->type == AVMEDIA_TYPE_AUDIO) {
 425                 if (link->incfg.samplerates != link->outcfg.samplerates
 426                     && link->incfg.samplerates && link->outcfg.samplerates)
 427                     if (!ff_can_merge_samplerates(link->incfg.samplerates,
 428                                                   link->outcfg.samplerates))
 429                         convert_needed = 1;
 430             }
 431
 432 #define CHECKED_MERGE(field, ...) ((ret = ff_merge_ ## field(__VA_ARGS__)) <= 0)
 433 #define MERGE_DISPATCH(field, ...)                                           \
 434             if (!(link->incfg.field && link->outcfg.field)) {                \
 435                 count_delayed++;                                             \
 436             } else if (link->incfg.field == link->outcfg.field) {            \
 437                 count_already_merged++;                                      \
 438             } else if (!convert_needed) {                                    \
 439                 count_merged++;                                              \
 440                 if (CHECKED_MERGE(field, __VA_ARGS__)) {                     \
 441                     if (ret < 0)                                             \
 442                         return ret;                                          \
 443                     convert_needed = 1;                                      \
 444                 }                                                            \
 445             }
 446
 447             if (link->type == AVMEDIA_TYPE_AUDIO) {
 448                 MERGE_DISPATCH(channel_layouts, link->incfg.channel_layouts,
 449                                                 link->outcfg.channel_layouts)
 450                 MERGE_DISPATCH(samplerates, link->incfg.samplerates,
 451                                             link->outcfg.samplerates)
 452             }
 453             MERGE_DISPATCH(formats, link->incfg.formats,
 454                            link->outcfg.formats, link->type)
 455 #undef MERGE_DISPATCH
 456
 457             if (convert_needed) {
 458                 AVFilterContext *convert;
 459                 const AVFilter *filter;
 460                 AVFilterLink *inlink, *outlink;
 461                 char inst_name[30];
 462
 463                 if (graph->disable_auto_convert) {
 464                     av_log(log_ctx, AV_LOG_ERROR,
 465                            "The filters '%s' and '%s' do not have a common format "
 466                            "and automatic conversion is disabled.\n",
 467                            link->src->name, link->dst->name);
 468                     return AVERROR(EINVAL);
 469                 }
 470
 471                 /* couldn't merge format lists. auto-insert conversion filter */
 472                 switch (link->type) {
 473                 case AVMEDIA_TYPE_VIDEO:
 474                     if (!(filter = avfilter_get_by_name("scale"))) {
 475                         av_log(log_ctx, AV_LOG_ERROR, "'scale' filter "
 476                                "not present, cannot convert pixel formats.\n");
 477                         return AVERROR(EINVAL);
 478                     }
 479
 480                     snprintf(inst_name, sizeof(inst_name), "auto_scaler_%d",
 481                              scaler_count++);
 482
 483                     if ((ret = avfilter_graph_create_filter(&convert, filter,
 484                                                             inst_name, graph->scale_sws_opts, NULL,
 485                                                             graph)) < 0)
 486                         return ret;
 487                     break;
 488                 case AVMEDIA_TYPE_AUDIO:
 489                     if (!(filter = avfilter_get_by_name("aresample"))) {
 490                         av_log(log_ctx, AV_LOG_ERROR, "'aresample' filter "
 491                                "not present, cannot convert audio formats.\n");
 492                         return AVERROR(EINVAL);
 493                     }
 494
 495                     snprintf(inst_name, sizeof(inst_name), "auto_resampler_%d",
 496                              resampler_count++);
 497                     if ((ret = avfilter_graph_create_filter(&convert, filter,
 498                                                             inst_name, graph->aresample_swr_opts,
 499                                                             NULL, graph)) < 0)
 500                         return ret;
 501                     break;
 502                 default:
 503                     return AVERROR(EINVAL);
 504                 }
 505
 506                 if ((ret = avfilter_insert_filter(link, convert, 0, 0)) < 0)
 507                     return ret;
 508
 509                 if ((ret = filter_query_formats(convert)) < 0)
 510                     return ret;
 511
 512                 inlink  = convert->inputs[0];
 513                 outlink = convert->outputs[0];
 514                 av_assert0( inlink->incfg.formats->refcount > 0);
 515                 av_assert0( inlink->outcfg.formats->refcount > 0);
 516                 av_assert0(outlink->incfg.formats->refcount > 0);
 517                 av_assert0(outlink->outcfg.formats->refcount > 0);
 518                 if (outlink->type == AVMEDIA_TYPE_AUDIO) {
 519                     av_assert0( inlink-> incfg.samplerates->refcount > 0);
 520                     av_assert0( inlink->outcfg.samplerates->refcount > 0);
 521                     av_assert0(outlink-> incfg.samplerates->refcount > 0);
 522                     av_assert0(outlink->outcfg.samplerates->refcount > 0);
 523                     av_assert0( inlink-> incfg.channel_layouts->refcount > 0);
 524                     av_assert0( inlink->outcfg.channel_layouts->refcount > 0);
 525                     av_assert0(outlink-> incfg.channel_layouts->refcount > 0);
 526                     av_assert0(outlink->outcfg.channel_layouts->refcount > 0);
 527                 }
 528                 if (CHECKED_MERGE(formats, inlink->incfg.formats,
 529                                   inlink->outcfg.formats, inlink->type)         ||
 530                     CHECKED_MERGE(formats, outlink->incfg.formats,
 531                                   outlink->outcfg.formats, outlink->type)       ||
 532                     inlink->type == AVMEDIA_TYPE_AUDIO &&
 533                     (CHECKED_MERGE(samplerates, inlink->incfg.samplerates,
 534                                                 inlink->outcfg.samplerates)  ||
 535                      CHECKED_MERGE(channel_layouts, inlink->incfg.channel_layouts,
 536                                    inlink->outcfg.channel_layouts))             ||
 537                     outlink->type == AVMEDIA_TYPE_AUDIO &&
 538                     (CHECKED_MERGE(samplerates, outlink->incfg.samplerates,
 539                                                 outlink->outcfg.samplerates) ||
 540                      CHECKED_MERGE(channel_layouts, outlink->incfg.channel_layouts,
 541                                                     outlink->outcfg.channel_layouts))) {
 542                     if (ret < 0)
 543                         return ret;
 544                     av_log(log_ctx, AV_LOG_ERROR,
 545                            "Impossible to convert between the formats supported by the filter "
 546                            "'%s' and the filter '%s'\n", link->src->name, link->dst->name);
 547                     return AVERROR(ENOSYS);
 548                 }
 549             }
 550         }
 551     }
 552
 553     av_log(graph, AV_LOG_DEBUG, "query_formats: "
 554            "%d queried, %d merged, %d already done, %d delayed\n",
 555            count_queried, count_merged, count_already_merged, count_delayed);
 556     if (count_delayed) {
 557         AVBPrint bp;
 558
 559         /* if count_queried > 0, one filter at least did set its formats,
 560            that will give additional information to its neighbour;
 561            if count_merged > 0, one pair of formats lists at least was merged,
 562            that will give additional information to all connected filters;
 563            in both cases, progress was made and a new round must be done */
 564         if (count_queried || count_merged)
 565             return AVERROR(EAGAIN);
 566         av_bprint_init(&bp, 0, AV_BPRINT_SIZE_AUTOMATIC);
 567         for (i = 0; i < graph->nb_filters; i++)
 568             if (!formats_declared(graph->filters[i]))
 569                 av_bprintf(&bp, "%s%s", bp.len ? ", " : "",
 570                           graph->filters[i]->name);
 571         av_log(graph, AV_LOG_ERROR,
 572                "The following filters could not choose their formats: %s\n"
 573                "Consider inserting the (a)format filter near their input or "
 574                "output.\n", bp.str);
 575         return AVERROR(EIO);
 576     }
 577     return 0;
 578 }
 579
 580 static int get_fmt_score(enum AVSampleFormat dst_fmt, enum AVSampleFormat src_fmt)
 581 {
 582     int score = 0;
 583
 584     if (av_sample_fmt_is_planar(dst_fmt) != av_sample_fmt_is_planar(src_fmt))
 585         score ++;
 586
 587     if (av_get_bytes_per_sample(dst_fmt) < av_get_bytes_per_sample(src_fmt)) {
 588         score += 100 * (av_get_bytes_per_sample(src_fmt) - av_get_bytes_per_sample(dst_fmt));
 589     }else
 590         score += 10  * (av_get_bytes_per_sample(dst_fmt) - av_get_bytes_per_sample(src_fmt));
 591
 592     if (av_get_packed_sample_fmt(dst_fmt) == AV_SAMPLE_FMT_S32 &&
 593         av_get_packed_sample_fmt(src_fmt) == AV_SAMPLE_FMT_FLT)
 594         score += 20;
 595
 596     if (av_get_packed_sample_fmt(dst_fmt) == AV_SAMPLE_FMT_FLT &&
 597         av_get_packed_sample_fmt(src_fmt) == AV_SAMPLE_FMT_S32)
 598         score += 2;
 599
 600     return score;
 601 }
 602
 603 static enum AVSampleFormat find_best_sample_fmt_of_2(enum AVSampleFormat dst_fmt1, enum AVSampleFormat dst_fmt2,
 604                                                      enum AVSampleFormat src_fmt)
 605 {
 606     int score1, score2;
 607
 608     score1 = get_fmt_score(dst_fmt1, src_fmt);
 609     score2 = get_fmt_score(dst_fmt2, src_fmt);
 610
 611     return score1 < score2 ? dst_fmt1 : dst_fmt2;
 612 }
 613
 614 static int pick_format(AVFilterLink *link, AVFilterLink *ref)
 615 {
 616     if (!link || !link->incfg.formats)
 617         return 0;
 618
 619     if (link->type == AVMEDIA_TYPE_VIDEO) {
 620         if(ref && ref->type == AVMEDIA_TYPE_VIDEO){
 621             //FIXME: This should check for AV_PIX_FMT_FLAG_ALPHA after PAL8 pixel format without alpha is implemented
 622             int has_alpha= av_pix_fmt_desc_get(ref->format)->nb_components % 2 == 0;
 623             enum AVPixelFormat best= AV_PIX_FMT_NONE;
 624             int i;
 625             for (i = 0; i < link->incfg.formats->nb_formats; i++) {
 626                 enum AVPixelFormat p = link->incfg.formats->formats[i];
 627                 best= av_find_best_pix_fmt_of_2(best, p, ref->format, has_alpha, NULL);
 628             }
 629             av_log(link->src,AV_LOG_DEBUG, "picking %s out of %d ref:%s alpha:%d\n",
 630                    av_get_pix_fmt_name(best), link->incfg.formats->nb_formats,
 631                    av_get_pix_fmt_name(ref->format), has_alpha);
 632             link->incfg.formats->formats[0] = best;
 633         }
 634     } else if (link->type == AVMEDIA_TYPE_AUDIO) {
 635         if(ref && ref->type == AVMEDIA_TYPE_AUDIO){
 636             enum AVSampleFormat best= AV_SAMPLE_FMT_NONE;
 637             int i;
 638             for (i = 0; i < link->incfg.formats->nb_formats; i++) {
 639                 enum AVSampleFormat p = link->incfg.formats->formats[i];
 640                 best = find_best_sample_fmt_of_2(best, p, ref->format);
 641             }
 642             av_log(link->src,AV_LOG_DEBUG, "picking %s out of %d ref:%s\n",
 643                    av_get_sample_fmt_name(best), link->incfg.formats->nb_formats,
 644                    av_get_sample_fmt_name(ref->format));
 645             link->incfg.formats->formats[0] = best;
 646         }
 647     }
 648
 649     link->incfg.formats->nb_formats = 1;
 650     link->format = link->incfg.formats->formats[0];
 651
 652     if (link->type == AVMEDIA_TYPE_AUDIO) {
 653         if (!link->incfg.samplerates->nb_formats) {
 654             av_log(link->src, AV_LOG_ERROR, "Cannot select sample rate for"
 655                    " the link between filters %s and %s.\n", link->src->name,
 656                    link->dst->name);
 657             return AVERROR(EINVAL);
 658         }
 659         link->incfg.samplerates->nb_formats = 1;
 660         link->sample_rate = link->incfg.samplerates->formats[0];
 661
 662         if (link->incfg.channel_layouts->all_layouts) {
 663             av_log(link->src, AV_LOG_ERROR, "Cannot select channel layout for"
 664                    " the link between filters %s and %s.\n", link->src->name,
 665                    link->dst->name);
 666             if (!link->incfg.channel_layouts->all_counts)
 667                 av_log(link->src, AV_LOG_ERROR, "Unknown channel layouts not "
 668                        "supported, try specifying a channel layout using "
 669                        "'aformat=channel_layouts=something'.\n");
 670             return AVERROR(EINVAL);
 671         }
 672         link->incfg.channel_layouts->nb_channel_layouts = 1;
 673         link->channel_layout = link->incfg.channel_layouts->channel_layouts[0];
 674         if ((link->channels = FF_LAYOUT2COUNT(link->channel_layout)))
 675             link->channel_layout = 0;
 676         else
 677             link->channels = av_get_channel_layout_nb_channels(link->channel_layout);
 678     }
 679
 680     ff_formats_unref(&link->incfg.formats);
 681     ff_formats_unref(&link->outcfg.formats);
 682     ff_formats_unref(&link->incfg.samplerates);
 683     ff_formats_unref(&link->outcfg.samplerates);
 684     ff_channel_layouts_unref(&link->incfg.channel_layouts);
 685     ff_channel_layouts_unref(&link->outcfg.channel_layouts);
 686
 687     return 0;
 688 }
 689
 690 #define REDUCE_FORMATS(fmt_type, list_type, list, var, nb, add_format) \
 691 do {                                                                   \
 692     for (i = 0; i < filter->nb_inputs; i++) {                          \
 693         AVFilterLink *link = filter->inputs[i];                        \
 694         fmt_type fmt;                                                  \
 695                                                                        \
 696         if (!link->outcfg.list || link->outcfg.list->nb != 1)          \
 697             continue;                                                  \
 698         fmt = link->outcfg.list->var[0];                               \
 699                                                                        \
 700         for (j = 0; j < filter->nb_outputs; j++) {                     \
 701             AVFilterLink *out_link = filter->outputs[j];               \
 702             list_type *fmts;                                           \
 703                                                                        \
 704             if (link->type != out_link->type ||                        \
 705                 out_link->incfg.list->nb == 1)                         \
 706                 continue;                                              \
 707             fmts = out_link->incfg.list;                               \
 708                                                                        \
 709             if (!out_link->incfg.list->nb) {                           \
 710                 if ((ret = add_format(&out_link->incfg.list, fmt)) < 0)\
 711                     return ret;                                        \
 712                 ret = 1;                                               \
 713                 break;                                                 \
 714             }                                                          \
 715                                                                        \
 716             for (k = 0; k < out_link->incfg.list->nb; k++)             \
 717                 if (fmts->var[k] == fmt) {                             \
 718                     fmts->var[0]  = fmt;                               \
 719                     fmts->nb = 1;                                      \
 720                     ret = 1;                                           \
 721                     break;                                             \
 722                 }                                                      \
 723         }                                                              \
 724     }                                                                  \
 725 } while (0)
 726
 727 static int reduce_formats_on_filter(AVFilterContext *filter)
 728 {
 729     int i, j, k, ret = 0;
 730
 731     REDUCE_FORMATS(int,      AVFilterFormats,        formats,         formats,
 732                    nb_formats, ff_add_format);
 733     REDUCE_FORMATS(int,      AVFilterFormats,        samplerates,     formats,
 734                    nb_formats, ff_add_format);
 735
 736     /* reduce channel layouts */
 737     for (i = 0; i < filter->nb_inputs; i++) {
 738         AVFilterLink *inlink = filter->inputs[i];
 739         uint64_t fmt;
 740
 741         if (!inlink->outcfg.channel_layouts ||
 742             inlink->outcfg.channel_layouts->nb_channel_layouts != 1)
 743             continue;
 744         fmt = inlink->outcfg.channel_layouts->channel_layouts[0];
 745
 746         for (j = 0; j < filter->nb_outputs; j++) {
 747             AVFilterLink *outlink = filter->outputs[j];
 748             AVFilterChannelLayouts *fmts;
 749
 750             fmts = outlink->incfg.channel_layouts;
 751             if (inlink->type != outlink->type || fmts->nb_channel_layouts == 1)
 752                 continue;
 753
 754             if (fmts->all_layouts &&
 755                 (!FF_LAYOUT2COUNT(fmt) || fmts->all_counts)) {
 756                 /* Turn the infinite list into a singleton */
 757                 fmts->all_layouts = fmts->all_counts  = 0;
 758                 if (ff_add_channel_layout(&outlink->incfg.channel_layouts, fmt) < 0)
 759                     ret = 1;
 760                 break;
 761             }
 762
 763             for (k = 0; k < outlink->incfg.channel_layouts->nb_channel_layouts; k++) {
 764                 if (fmts->channel_layouts[k] == fmt) {
 765                     fmts->channel_layouts[0]  = fmt;
 766                     fmts->nb_channel_layouts = 1;
 767                     ret = 1;
 768                     break;
 769                 }
 770             }
 771         }
 772     }
 773
 774     return ret;
 775 }
 776
 777 static int reduce_formats(AVFilterGraph *graph)
 778 {
 779     int i, reduced, ret;
 780
 781     do {
 782         reduced = 0;
 783
 784         for (i = 0; i < graph->nb_filters; i++) {
 785             if ((ret = reduce_formats_on_filter(graph->filters[i])) < 0)
 786                 return ret;
 787             reduced |= ret;
 788         }
 789     } while (reduced);
 790
 791     return 0;
 792 }
 793
 794 static void swap_samplerates_on_filter(AVFilterContext *filter)
 795 {
 796     AVFilterLink *link = NULL;
 797     int sample_rate;
 798     int i, j;
 799
 800     for (i = 0; i < filter->nb_inputs; i++) {
 801         link = filter->inputs[i];
 802
 803         if (link->type == AVMEDIA_TYPE_AUDIO &&
 804             link->outcfg.samplerates->nb_formats== 1)
 805             break;
 806     }
 807     if (i == filter->nb_inputs)
 808         return;
 809
 810     sample_rate = link->outcfg.samplerates->formats[0];
 811
 812     for (i = 0; i < filter->nb_outputs; i++) {
 813         AVFilterLink *outlink = filter->outputs[i];
 814         int best_idx, best_diff = INT_MAX;
 815
 816         if (outlink->type != AVMEDIA_TYPE_AUDIO ||
 817             outlink->incfg.samplerates->nb_formats < 2)
 818             continue;
 819
 820         for (j = 0; j < outlink->incfg.samplerates->nb_formats; j++) {
 821             int diff = abs(sample_rate - outlink->incfg.samplerates->formats[j]);
 822
 823             av_assert0(diff < INT_MAX); // This would lead to the use of uninitialized best_diff but is only possible with invalid sample rates
 824
 825             if (diff < best_diff) {
 826                 best_diff = diff;
 827                 best_idx  = j;
 828             }
 829         }
 830         FFSWAP(int, outlink->incfg.samplerates->formats[0],
 831                outlink->incfg.samplerates->formats[best_idx]);
 832     }
 833 }
 834
 835 static void swap_samplerates(AVFilterGraph *graph)
 836 {
 837     int i;
 838
 839     for (i = 0; i < graph->nb_filters; i++)
 840         swap_samplerates_on_filter(graph->filters[i]);
 841 }
 842
 843 #define CH_CENTER_PAIR (AV_CH_FRONT_LEFT_OF_CENTER | AV_CH_FRONT_RIGHT_OF_CENTER)
 844 #define CH_FRONT_PAIR  (AV_CH_FRONT_LEFT           | AV_CH_FRONT_RIGHT)
 845 #define CH_STEREO_PAIR (AV_CH_STEREO_LEFT          | AV_CH_STEREO_RIGHT)
 846 #define CH_WIDE_PAIR   (AV_CH_WIDE_LEFT            | AV_CH_WIDE_RIGHT)
 847 #define CH_SIDE_PAIR   (AV_CH_SIDE_LEFT            | AV_CH_SIDE_RIGHT)
 848 #define CH_DIRECT_PAIR (AV_CH_SURROUND_DIRECT_LEFT | AV_CH_SURROUND_DIRECT_RIGHT)
 849 #define CH_BACK_PAIR   (AV_CH_BACK_LEFT            | AV_CH_BACK_RIGHT)
 850
 851 /* allowable substitutions for channel pairs when comparing layouts,
 852  * ordered by priority for both values */
 853 static const uint64_t ch_subst[][2] = {
 854     { CH_FRONT_PAIR,      CH_CENTER_PAIR     },
 855     { CH_FRONT_PAIR,      CH_WIDE_PAIR       },
 856     { CH_FRONT_PAIR,      AV_CH_FRONT_CENTER },
 857     { CH_CENTER_PAIR,     CH_FRONT_PAIR      },
 858     { CH_CENTER_PAIR,     CH_WIDE_PAIR       },
 859     { CH_CENTER_PAIR,     AV_CH_FRONT_CENTER },
 860     { CH_WIDE_PAIR,       CH_FRONT_PAIR      },
 861     { CH_WIDE_PAIR,       CH_CENTER_PAIR     },
 862     { CH_WIDE_PAIR,       AV_CH_FRONT_CENTER },
 863     { AV_CH_FRONT_CENTER, CH_FRONT_PAIR      },
 864     { AV_CH_FRONT_CENTER, CH_CENTER_PAIR     },
 865     { AV_CH_FRONT_CENTER, CH_WIDE_PAIR       },
 866     { CH_SIDE_PAIR,       CH_DIRECT_PAIR     },
 867     { CH_SIDE_PAIR,       CH_BACK_PAIR       },
 868     { CH_SIDE_PAIR,       AV_CH_BACK_CENTER  },
 869     { CH_BACK_PAIR,       CH_DIRECT_PAIR     },
 870     { CH_BACK_PAIR,       CH_SIDE_PAIR       },
 871     { CH_BACK_PAIR,       AV_CH_BACK_CENTER  },
 872     { AV_CH_BACK_CENTER,  CH_BACK_PAIR       },
 873     { AV_CH_BACK_CENTER,  CH_DIRECT_PAIR     },
 874     { AV_CH_BACK_CENTER,  CH_SIDE_PAIR       },
 875 };
 876
 877 static void swap_channel_layouts_on_filter(AVFilterContext *filter)
 878 {
 879     AVFilterLink *link = NULL;
 880     int i, j, k;
 881
 882     for (i = 0; i < filter->nb_inputs; i++) {
 883         link = filter->inputs[i];
 884
 885         if (link->type == AVMEDIA_TYPE_AUDIO &&
 886             link->outcfg.channel_layouts->nb_channel_layouts == 1)
 887             break;
 888     }
 889     if (i == filter->nb_inputs)
 890         return;
 891
 892     for (i = 0; i < filter->nb_outputs; i++) {
 893         AVFilterLink *outlink = filter->outputs[i];
 894         int best_idx = -1, best_score = INT_MIN, best_count_diff = INT_MAX;
 895
 896         if (outlink->type != AVMEDIA_TYPE_AUDIO ||
 897             outlink->incfg.channel_layouts->nb_channel_layouts < 2)
 898             continue;
 899
 900         for (j = 0; j < outlink->incfg.channel_layouts->nb_channel_layouts; j++) {
 901             uint64_t  in_chlayout = link->outcfg.channel_layouts->channel_layouts[0];
 902             uint64_t out_chlayout = outlink->incfg.channel_layouts->channel_layouts[j];
 903             int  in_channels      = av_get_channel_layout_nb_channels(in_chlayout);
 904             int out_channels      = av_get_channel_layout_nb_channels(out_chlayout);
 905             int count_diff        = out_channels - in_channels;
 906             int matched_channels, extra_channels;
 907             int score = 100000;
 908
 909             if (FF_LAYOUT2COUNT(in_chlayout) || FF_LAYOUT2COUNT(out_chlayout)) {
 910                 /* Compute score in case the input or output layout encodes
 911                    a channel count; in this case the score is not altered by
 912                    the computation afterwards, as in_chlayout and
 913                    out_chlayout have both been set to 0 */
 914                 if (FF_LAYOUT2COUNT(in_chlayout))
 915                     in_channels = FF_LAYOUT2COUNT(in_chlayout);
 916                 if (FF_LAYOUT2COUNT(out_chlayout))
 917                     out_channels = FF_LAYOUT2COUNT(out_chlayout);
 918                 score -= 10000 + FFABS(out_channels - in_channels) +
 919                          (in_channels > out_channels ? 10000 : 0);
 920                 in_chlayout = out_chlayout = 0;
 921                 /* Let the remaining computation run, even if the score
 922                    value is not altered */
 923             }
 924
 925             /* channel substitution */
 926             for (k = 0; k < FF_ARRAY_ELEMS(ch_subst); k++) {
 927                 uint64_t cmp0 = ch_subst[k][0];
 928                 uint64_t cmp1 = ch_subst[k][1];
 929                 if (( in_chlayout & cmp0) && (!(out_chlayout & cmp0)) &&
 930                     (out_chlayout & cmp1) && (!( in_chlayout & cmp1))) {
 931                     in_chlayout  &= ~cmp0;
 932                     out_chlayout &= ~cmp1;
 933                     /* add score for channel match, minus a deduction for
 934                        having to do the substitution */
 935                     score += 10 * av_get_channel_layout_nb_channels(cmp1) - 2;
 936                 }
 937             }
 938
 939             /* no penalty for LFE channel mismatch */
 940             if ( (in_chlayout & AV_CH_LOW_FREQUENCY) &&
 941                 (out_chlayout & AV_CH_LOW_FREQUENCY))
 942                 score += 10;
 943             in_chlayout  &= ~AV_CH_LOW_FREQUENCY;
 944             out_chlayout &= ~AV_CH_LOW_FREQUENCY;
 945
 946             matched_channels = av_get_channel_layout_nb_channels(in_chlayout &
 947                                                                  out_chlayout);
 948             extra_channels   = av_get_channel_layout_nb_channels(out_chlayout &
 949                                                                  (~in_chlayout));
 950             score += 10 * matched_channels - 5 * extra_channels;
 951
 952             if (score > best_score ||
 953                 (count_diff < best_count_diff && score == best_score)) {
 954                 best_score = score;
 955                 best_idx   = j;
 956                 best_count_diff = count_diff;
 957             }
 958         }
 959         av_assert0(best_idx >= 0);
 960         FFSWAP(uint64_t, outlink->incfg.channel_layouts->channel_layouts[0],
 961                outlink->incfg.channel_layouts->channel_layouts[best_idx]);
 962     }
 963
 964 }
 965
 966 static void swap_channel_layouts(AVFilterGraph *graph)
 967 {
 968     int i;
 969
 970     for (i = 0; i < graph->nb_filters; i++)
 971         swap_channel_layouts_on_filter(graph->filters[i]);
 972 }
 973
 974 static void swap_sample_fmts_on_filter(AVFilterContext *filter)
 975 {
 976     AVFilterLink *link = NULL;
 977     int format, bps;
 978     int i, j;
 979
 980     for (i = 0; i < filter->nb_inputs; i++) {
 981         link = filter->inputs[i];
 982
 983         if (link->type == AVMEDIA_TYPE_AUDIO &&
 984             link->outcfg.formats->nb_formats == 1)
 985             break;
 986     }
 987     if (i == filter->nb_inputs)
 988         return;
 989
 990     format = link->outcfg.formats->formats[0];
 991     bps    = av_get_bytes_per_sample(format);
 992
 993     for (i = 0; i < filter->nb_outputs; i++) {
 994         AVFilterLink *outlink = filter->outputs[i];
 995         int best_idx = -1, best_score = INT_MIN;
 996
 997         if (outlink->type != AVMEDIA_TYPE_AUDIO ||
 998             outlink->incfg.formats->nb_formats < 2)
 999             continue;
1000
1001         for (j = 0; j < outlink->incfg.formats->nb_formats; j++) {
1002             int out_format = outlink->incfg.formats->formats[j];
1003             int out_bps    = av_get_bytes_per_sample(out_format);
1004             int score;
1005
1006             if (av_get_packed_sample_fmt(out_format) == format ||
1007                 av_get_planar_sample_fmt(out_format) == format) {
1008                 best_idx   = j;
1009                 break;
1010             }
1011
1012             /* for s32 and float prefer double to prevent loss of information */
1013             if (bps == 4 && out_bps == 8) {
1014                 best_idx = j;
1015                 break;
1016             }
1017
1018             /* prefer closest higher or equal bps */
1019             score = -abs(out_bps - bps);
1020             if (out_bps >= bps)
1021                 score += INT_MAX/2;
1022
1023             if (score > best_score) {
1024                 best_score = score;
1025                 best_idx   = j;
1026             }
1027         }
1028         av_assert0(best_idx >= 0);
1029         FFSWAP(int, outlink->incfg.formats->formats[0],
1030                outlink->incfg.formats->formats[best_idx]);
1031     }
1032 }
1033
1034 static void swap_sample_fmts(AVFilterGraph *graph)
1035 {
1036     int i;
1037
1038     for (i = 0; i < graph->nb_filters; i++)
1039         swap_sample_fmts_on_filter(graph->filters[i]);
1040
1041 }
1042
1043 static int pick_formats(AVFilterGraph *graph)
1044 {
1045     int i, j, ret;
1046     int change;
1047
1048     do{
1049         change = 0;
1050         for (i = 0; i < graph->nb_filters; i++) {
1051             AVFilterContext *filter = graph->filters[i];
1052             if (filter->nb_inputs){
1053                 for (j = 0; j < filter->nb_inputs; j++){
1054                     if (filter->inputs[j]->incfg.formats && filter->inputs[j]->incfg.formats->nb_formats == 1) {
1055                         if ((ret = pick_format(filter->inputs[j], NULL)) < 0)
1056                             return ret;
1057                         change = 1;
1058                     }
1059                 }
1060             }
1061             if (filter->nb_outputs){
1062                 for (j = 0; j < filter->nb_outputs; j++){
1063                     if (filter->outputs[j]->incfg.formats && filter->outputs[j]->incfg.formats->nb_formats == 1) {
1064                         if ((ret = pick_format(filter->outputs[j], NULL)) < 0)
1065                             return ret;
1066                         change = 1;
1067                     }
1068                 }
1069             }
1070             if (filter->nb_inputs && filter->nb_outputs && filter->inputs[0]->format>=0) {
1071                 for (j = 0; j < filter->nb_outputs; j++) {
1072                     if (filter->outputs[j]->format<0) {
1073                         if ((ret = pick_format(filter->outputs[j], filter->inputs[0])) < 0)
1074                             return ret;
1075                         change = 1;
1076                     }
1077                 }
1078             }
1079         }
1080     }while(change);
1081
1082     for (i = 0; i < graph->nb_filters; i++) {
1083         AVFilterContext *filter = graph->filters[i];
1084
1085         for (j = 0; j < filter->nb_inputs; j++)
1086             if ((ret = pick_format(filter->inputs[j], NULL)) < 0)
1087                 return ret;
1088         for (j = 0; j < filter->nb_outputs; j++)
1089             if ((ret = pick_format(filter->outputs[j], NULL)) < 0)
1090                 return ret;
1091     }
1092     return 0;
1093 }
1094
1095 /**
1096  * Configure the formats of all the links in the graph.
1097  */
1098 static int graph_config_formats(AVFilterGraph *graph, AVClass *log_ctx)
1099 {
1100     int ret;
1101
1102     /* find supported formats from sub-filters, and merge along links */
1103     while ((ret = query_formats(graph, log_ctx)) == AVERROR(EAGAIN))
1104         av_log(graph, AV_LOG_DEBUG, "query_formats not finished\n");
1105     if (ret < 0)
1106         return ret;
1107
1108     /* Once everything is merged, it's possible that we'll still have
1109      * multiple valid media format choices. We try to minimize the amount
1110      * of format conversion inside filters */
1111     if ((ret = reduce_formats(graph)) < 0)
1112         return ret;
1113
1114     /* for audio filters, ensure the best format, sample rate and channel layout
1115      * is selected */
1116     swap_sample_fmts(graph);
1117     swap_samplerates(graph);
1118     swap_channel_layouts(graph);
1119
1120     if ((ret = pick_formats(graph)) < 0)
1121         return ret;
1122
1123     return 0;
1124 }
1125
1126 static int graph_config_pointers(AVFilterGraph *graph,
1127                                              AVClass *log_ctx)
1128 {
1129     unsigned i, j;
1130     int sink_links_count = 0, n = 0;
1131     AVFilterContext *f;
1132     AVFilterLink **sinks;
1133
1134     for (i = 0; i < graph->nb_filters; i++) {
1135         f = graph->filters[i];
1136         for (j = 0; j < f->nb_inputs; j++) {
1137             f->inputs[j]->graph     = graph;
1138             f->inputs[j]->age_index = -1;
1139         }
1140         for (j = 0; j < f->nb_outputs; j++) {
1141             f->outputs[j]->graph    = graph;
1142             f->outputs[j]->age_index= -1;
1143         }
1144         if (!f->nb_outputs) {
1145             if (f->nb_inputs > INT_MAX - sink_links_count)
1146                 return AVERROR(EINVAL);
1147             sink_links_count += f->nb_inputs;
1148         }
1149     }
1150     sinks = av_calloc(sink_links_count, sizeof(*sinks));
1151     if (!sinks)
1152         return AVERROR(ENOMEM);
1153     for (i = 0; i < graph->nb_filters; i++) {
1154         f = graph->filters[i];
1155         if (!f->nb_outputs) {
1156             for (j = 0; j < f->nb_inputs; j++) {
1157                 sinks[n] = f->inputs[j];
1158                 f->inputs[j]->age_index = n++;
1159             }
1160         }
1161     }
1162     av_assert0(n == sink_links_count);
1163     graph->sink_links       = sinks;
1164     graph->sink_links_count = sink_links_count;
1165     return 0;
1166 }
1167
1168 int avfilter_graph_config(AVFilterGraph *graphctx, void *log_ctx)
1169 {
1170     int ret;
1171
1172     if ((ret = graph_check_validity(graphctx, log_ctx)))
1173         return ret;
1174     if ((ret = graph_config_formats(graphctx, log_ctx)))
1175         return ret;
1176     if ((ret = graph_config_links(graphctx, log_ctx)))
1177         return ret;
1178     if ((ret = graph_check_links(graphctx, log_ctx)))
1179         return ret;
1180     if ((ret = graph_config_pointers(graphctx, log_ctx)))
1181         return ret;
1182
1183     return 0;
1184 }
1185
1186 int avfilter_graph_send_command(AVFilterGraph *graph, const char *target, const char *cmd, const char *arg, char *res, int res_len, int flags)
1187 {
1188     int i, r = AVERROR(ENOSYS);
1189
1190     if (!graph)
1191         return r;
1192
1193     if ((flags & AVFILTER_CMD_FLAG_ONE) && !(flags & AVFILTER_CMD_FLAG_FAST)) {
1194         r = avfilter_graph_send_command(graph, target, cmd, arg, res, res_len, flags | AVFILTER_CMD_FLAG_FAST);
1195         if (r != AVERROR(ENOSYS))
1196             return r;
1197     }
1198
1199     if (res_len && res)
1200         res[0] = 0;
1201
1202     for (i = 0; i < graph->nb_filters; i++) {
1203         AVFilterContext *filter = graph->filters[i];
1204         if (!strcmp(target, "all") || (filter->name && !strcmp(target, filter->name)) || !strcmp(target, filter->filter->name)) {
1205             r = avfilter_process_command(filter, cmd, arg, res, res_len, flags);
1206             if (r != AVERROR(ENOSYS)) {
1207                 if ((flags & AVFILTER_CMD_FLAG_ONE) || r < 0)
1208                     return r;
1209             }
1210         }
1211     }
1212
1213     return r;
1214 }
1215
1216 int avfilter_graph_queue_command(AVFilterGraph *graph, const char *target, const char *command, const char *arg, int flags, double ts)
1217 {
1218     int i;
1219
1220     if(!graph)
1221         return 0;
1222
1223     for (i = 0; i < graph->nb_filters; i++) {
1224         AVFilterContext *filter = graph->filters[i];
1225         if(filter && (!strcmp(target, "all") || !strcmp(target, filter->name) || !strcmp(target, filter->filter->name))){
1226             AVFilterCommand **queue = &filter->command_queue, *next;
1227             while (*queue && (*queue)->time <= ts)
1228                 queue = &(*queue)->next;
1229             next = *queue;
1230             *queue = av_mallocz(sizeof(AVFilterCommand));
1231             if (!*queue)
1232                 return AVERROR(ENOMEM);
1233
1234             (*queue)->command = av_strdup(command);
1235             (*queue)->arg     = av_strdup(arg);
1236             (*queue)->time    = ts;
1237             (*queue)->flags   = flags;
1238             (*queue)->next    = next;
1239             if(flags & AVFILTER_CMD_FLAG_ONE)
1240                 return 0;
1241         }
1242     }
1243
1244     return 0;
1245 }
1246
1247 static void heap_bubble_up(AVFilterGraph *graph,
1248                            AVFilterLink *link, int index)
1249 {
1250     AVFilterLink **links = graph->sink_links;
1251
1252     av_assert0(index >= 0);
1253
1254     while (index) {
1255         int parent = (index - 1) >> 1;
1256         if (links[parent]->current_pts_us >= link->current_pts_us)
1257             break;
1258         links[index] = links[parent];
1259         links[index]->age_index = index;
1260         index = parent;
1261     }
1262     links[index] = link;
1263     link->age_index = index;
1264 }
1265
1266 static void heap_bubble_down(AVFilterGraph *graph,
1267                              AVFilterLink *link, int index)
1268 {
1269     AVFilterLink **links = graph->sink_links;
1270
1271     av_assert0(index >= 0);
1272
1273     while (1) {
1274         int child = 2 * index + 1;
1275         if (child >= graph->sink_links_count)
1276             break;
1277         if (child + 1 < graph->sink_links_count &&
1278             links[child + 1]->current_pts_us < links[child]->current_pts_us)
1279             child++;
1280         if (link->current_pts_us < links[child]->current_pts_us)
1281             break;
1282         links[index] = links[child];
1283         links[index]->age_index = index;
1284         index = child;
1285     }
1286     links[index] = link;
1287     link->age_index = index;
1288 }
1289
1290 void ff_avfilter_graph_update_heap(AVFilterGraph *graph, AVFilterLink *link)
1291 {
1292     heap_bubble_up  (graph, link, link->age_index);
1293     heap_bubble_down(graph, link, link->age_index);
1294 }
1295
1296 int avfilter_graph_request_oldest(AVFilterGraph *graph)
1297 {
1298     AVFilterLink *oldest = graph->sink_links[0];
1299     int64_t frame_count;
1300     int r;
1301
1302     while (graph->sink_links_count) {
1303         oldest = graph->sink_links[0];
1304         if (oldest->dst->filter->activate) {
1305             /* For now, buffersink is the only filter implementing activate. */
1306             r = av_buffersink_get_frame_flags(oldest->dst, NULL,
1307                                               AV_BUFFERSINK_FLAG_PEEK);
1308             if (r != AVERROR_EOF)
1309                 return r;
1310         } else {
1311             r = ff_request_frame(oldest);
1312         }
1313         if (r != AVERROR_EOF)
1314             break;
1315         av_log(oldest->dst, AV_LOG_DEBUG, "EOF on sink link %s:%s.\n",
1316                oldest->dst ? oldest->dst->name : "unknown",
1317                oldest->dstpad ? oldest->dstpad->name : "unknown");
1318         /* EOF: remove the link from the heap */
1319         if (oldest->age_index < --graph->sink_links_count)
1320             heap_bubble_down(graph, graph->sink_links[graph->sink_links_count],
1321                              oldest->age_index);
1322         oldest->age_index = -1;
1323     }
1324     if (!graph->sink_links_count)
1325         return AVERROR_EOF;
1326     av_assert1(!oldest->dst->filter->activate);
1327     av_assert1(oldest->age_index >= 0);
1328     frame_count = oldest->frame_count_out;
1329     while (frame_count == oldest->frame_count_out) {
1330         r = ff_filter_graph_run_once(graph);
1331         if (r == AVERROR(EAGAIN) &&
1332             !oldest->frame_wanted_out && !oldest->frame_blocked_in &&
1333             !oldest->status_in)
1334             ff_request_frame(oldest);
1335         else if (r < 0)
1336             return r;
1337     }
1338     return 0;
1339 }
1340
1341 int ff_filter_graph_run_once(AVFilterGraph *graph)
1342 {
1343     AVFilterContext *filter;
1344     unsigned i;
1345
1346     av_assert0(graph->nb_filters);
1347     filter = graph->filters[0];
1348     for (i = 1; i < graph->nb_filters; i++)
1349         if (graph->filters[i]->ready > filter->ready)
1350             filter = graph->filters[i];
1351     if (!filter->ready)
1352         return AVERROR(EAGAIN);
1353     return ff_filter_activate(filter);
1354 }