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