123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993 |
- /*
- * filter graphs
- * Copyright (c) 2008 Vitor Sessak
- * Copyright (c) 2007 Bobby Bingham
- *
- * This file is part of FFmpeg.
- *
- * FFmpeg is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation; either
- * version 2.1 of the License, or (at your option) any later version.
- *
- * FFmpeg is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public
- * License along with FFmpeg; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
- */
- #include <ctype.h>
- #include <string.h>
- #include "libavutil/audioconvert.h"
- #include "libavutil/avassert.h"
- #include "libavutil/pixdesc.h"
- #include "libavcodec/avcodec.h" // avcodec_find_best_pix_fmt2()
- #include "avfilter.h"
- #include "avfiltergraph.h"
- #include "formats.h"
- #include "internal.h"
- #include "libavutil/audioconvert.h"
- #include "libavutil/avassert.h"
- #include "libavutil/log.h"
- static const AVClass filtergraph_class = {
- .class_name = "AVFilterGraph",
- .item_name = av_default_item_name,
- .version = LIBAVUTIL_VERSION_INT,
- .category = AV_CLASS_CATEGORY_FILTER,
- };
- AVFilterGraph *avfilter_graph_alloc(void)
- {
- AVFilterGraph *ret = av_mallocz(sizeof(AVFilterGraph));
- if (!ret)
- return NULL;
- ret->av_class = &filtergraph_class;
- return ret;
- }
- void avfilter_graph_free(AVFilterGraph **graph)
- {
- if (!*graph)
- return;
- for (; (*graph)->filter_count > 0; (*graph)->filter_count--)
- avfilter_free((*graph)->filters[(*graph)->filter_count - 1]);
- av_freep(&(*graph)->sink_links);
- av_freep(&(*graph)->scale_sws_opts);
- av_freep(&(*graph)->filters);
- av_freep(graph);
- }
- int avfilter_graph_add_filter(AVFilterGraph *graph, AVFilterContext *filter)
- {
- AVFilterContext **filters = av_realloc(graph->filters,
- sizeof(AVFilterContext*) * (graph->filter_count+1));
- if (!filters)
- return AVERROR(ENOMEM);
- graph->filters = filters;
- graph->filters[graph->filter_count++] = filter;
- return 0;
- }
- int avfilter_graph_create_filter(AVFilterContext **filt_ctx, AVFilter *filt,
- const char *name, const char *args, void *opaque,
- AVFilterGraph *graph_ctx)
- {
- int ret;
- if ((ret = avfilter_open(filt_ctx, filt, name)) < 0)
- goto fail;
- if ((ret = avfilter_init_filter(*filt_ctx, args, opaque)) < 0)
- goto fail;
- if ((ret = avfilter_graph_add_filter(graph_ctx, *filt_ctx)) < 0)
- goto fail;
- return 0;
- fail:
- if (*filt_ctx)
- avfilter_free(*filt_ctx);
- *filt_ctx = NULL;
- return ret;
- }
- void avfilter_graph_set_auto_convert(AVFilterGraph *graph, unsigned flags)
- {
- graph->disable_auto_convert = flags;
- }
- /**
- * Check for the validity of graph.
- *
- * A graph is considered valid if all its input and output pads are
- * connected.
- *
- * @return 0 in case of success, a negative value otherwise
- */
- static int graph_check_validity(AVFilterGraph *graph, AVClass *log_ctx)
- {
- AVFilterContext *filt;
- int i, j;
- for (i = 0; i < graph->filter_count; i++) {
- filt = graph->filters[i];
- for (j = 0; j < filt->nb_inputs; j++) {
- if (!filt->inputs[j] || !filt->inputs[j]->src) {
- av_log(log_ctx, AV_LOG_ERROR,
- "Input pad \"%s\" for the filter \"%s\" of type \"%s\" not connected to any source\n",
- filt->input_pads[j].name, filt->name, filt->filter->name);
- return AVERROR(EINVAL);
- }
- }
- for (j = 0; j < filt->nb_outputs; j++) {
- if (!filt->outputs[j] || !filt->outputs[j]->dst) {
- av_log(log_ctx, AV_LOG_ERROR,
- "Output pad \"%s\" for the filter \"%s\" of type \"%s\" not connected to any destination\n",
- filt->output_pads[j].name, filt->name, filt->filter->name);
- return AVERROR(EINVAL);
- }
- }
- }
- return 0;
- }
- /**
- * Configure all the links of graphctx.
- *
- * @return 0 in case of success, a negative value otherwise
- */
- static int graph_config_links(AVFilterGraph *graph, AVClass *log_ctx)
- {
- AVFilterContext *filt;
- int i, ret;
- for (i=0; i < graph->filter_count; i++) {
- filt = graph->filters[i];
- if (!filt->nb_outputs) {
- if ((ret = avfilter_config_links(filt)))
- return ret;
- }
- }
- return 0;
- }
- AVFilterContext *avfilter_graph_get_filter(AVFilterGraph *graph, char *name)
- {
- int i;
- for (i = 0; i < graph->filter_count; i++)
- if (graph->filters[i]->name && !strcmp(name, graph->filters[i]->name))
- return graph->filters[i];
- return NULL;
- }
- static int filter_query_formats(AVFilterContext *ctx)
- {
- int ret;
- AVFilterFormats *formats;
- AVFilterChannelLayouts *chlayouts;
- AVFilterFormats *samplerates;
- enum AVMediaType type = ctx->inputs && ctx->inputs [0] ? ctx->inputs [0]->type :
- ctx->outputs && ctx->outputs[0] ? ctx->outputs[0]->type :
- AVMEDIA_TYPE_VIDEO;
- if ((ret = ctx->filter->query_formats(ctx)) < 0)
- return ret;
- formats = ff_all_formats(type);
- if (!formats)
- return AVERROR(ENOMEM);
- ff_set_common_formats(ctx, formats);
- if (type == AVMEDIA_TYPE_AUDIO) {
- samplerates = ff_all_samplerates();
- if (!samplerates)
- return AVERROR(ENOMEM);
- ff_set_common_samplerates(ctx, samplerates);
- chlayouts = ff_all_channel_layouts();
- if (!chlayouts)
- return AVERROR(ENOMEM);
- ff_set_common_channel_layouts(ctx, chlayouts);
- }
- return 0;
- }
- static int insert_conv_filter(AVFilterGraph *graph, AVFilterLink *link,
- const char *filt_name, const char *filt_args)
- {
- static int auto_count = 0, ret;
- char inst_name[32];
- AVFilterContext *filt_ctx;
- if (graph->disable_auto_convert) {
- av_log(NULL, AV_LOG_ERROR,
- "The filters '%s' and '%s' do not have a common format "
- "and automatic conversion is disabled.\n",
- link->src->name, link->dst->name);
- return AVERROR(EINVAL);
- }
- snprintf(inst_name, sizeof(inst_name), "auto-inserted %s %d",
- filt_name, auto_count++);
- if ((ret = avfilter_graph_create_filter(&filt_ctx,
- avfilter_get_by_name(filt_name),
- inst_name, filt_args, NULL, graph)) < 0)
- return ret;
- if ((ret = avfilter_insert_filter(link, filt_ctx, 0, 0)) < 0)
- return ret;
- filter_query_formats(filt_ctx);
- if ( ((link = filt_ctx-> inputs[0]) &&
- !ff_merge_formats(link->in_formats, link->out_formats)) ||
- ((link = filt_ctx->outputs[0]) &&
- !ff_merge_formats(link->in_formats, link->out_formats))
- ) {
- av_log(NULL, AV_LOG_ERROR,
- "Impossible to convert between the formats supported by the filter "
- "'%s' and the filter '%s'\n", link->src->name, link->dst->name);
- return AVERROR(EINVAL);
- }
- if (link->type == AVMEDIA_TYPE_AUDIO &&
- (((link = filt_ctx-> inputs[0]) &&
- !ff_merge_channel_layouts(link->in_channel_layouts, link->out_channel_layouts)) ||
- ((link = filt_ctx->outputs[0]) &&
- !ff_merge_channel_layouts(link->in_channel_layouts, link->out_channel_layouts)))
- ) {
- av_log(NULL, AV_LOG_ERROR,
- "Impossible to convert between the channel layouts formats supported by the filter "
- "'%s' and the filter '%s'\n", link->src->name, link->dst->name);
- return AVERROR(EINVAL);
- }
- return 0;
- }
- static int query_formats(AVFilterGraph *graph, AVClass *log_ctx)
- {
- int i, j, ret;
- char filt_args[128];
- AVFilterFormats *formats;
- AVFilterChannelLayouts *chlayouts;
- AVFilterFormats *samplerates;
- int scaler_count = 0, resampler_count = 0;
- for (j = 0; j < 2; j++) {
- /* ask all the sub-filters for their supported media formats */
- for (i = 0; i < graph->filter_count; i++) {
- /* Call query_formats on sources first.
- This is a temporary workaround for amerge,
- until format renegociation is implemented. */
- if (!graph->filters[i]->nb_inputs == j)
- continue;
- if (graph->filters[i]->filter->query_formats)
- ret = filter_query_formats(graph->filters[i]);
- else
- ret = ff_default_query_formats(graph->filters[i]);
- if (ret < 0)
- return ret;
- }
- }
- /* go through and merge as many format lists as possible */
- for (i = 0; i < graph->filter_count; i++) {
- AVFilterContext *filter = graph->filters[i];
- for (j = 0; j < filter->nb_inputs; j++) {
- AVFilterLink *link = filter->inputs[j];
- #if 0
- if (!link) continue;
- if (!link->in_formats || !link->out_formats)
- return AVERROR(EINVAL);
- if (link->type == AVMEDIA_TYPE_VIDEO &&
- !ff_merge_formats(link->in_formats, link->out_formats)) {
- /* couldn't merge format lists, auto-insert scale filter */
- snprintf(filt_args, sizeof(filt_args), "0:0:%s",
- graph->scale_sws_opts);
- if (ret = insert_conv_filter(graph, link, "scale", filt_args))
- return ret;
- }
- else if (link->type == AVMEDIA_TYPE_AUDIO) {
- if (!link->in_channel_layouts || !link->out_channel_layouts)
- return AVERROR(EINVAL);
- /* Merge all three list before checking: that way, in all
- * three categories, aconvert will use a common format
- * whenever possible. */
- formats = ff_merge_formats(link->in_formats, link->out_formats);
- chlayouts = ff_merge_channel_layouts(link->in_channel_layouts , link->out_channel_layouts);
- samplerates = ff_merge_samplerates (link->in_samplerates, link->out_samplerates);
- if (!formats || !chlayouts || !samplerates)
- if (ret = insert_conv_filter(graph, link, "aresample", NULL))
- return ret;
- #else
- int convert_needed = 0;
- if (!link)
- continue;
- if (link->in_formats != link->out_formats &&
- !ff_merge_formats(link->in_formats,
- link->out_formats))
- convert_needed = 1;
- if (link->type == AVMEDIA_TYPE_AUDIO) {
- if (link->in_channel_layouts != link->out_channel_layouts &&
- !ff_merge_channel_layouts(link->in_channel_layouts,
- link->out_channel_layouts))
- convert_needed = 1;
- if (link->in_samplerates != link->out_samplerates &&
- !ff_merge_samplerates(link->in_samplerates,
- link->out_samplerates))
- convert_needed = 1;
- }
- if (convert_needed) {
- AVFilterContext *convert;
- AVFilter *filter;
- AVFilterLink *inlink, *outlink;
- char scale_args[256];
- char inst_name[30];
- /* couldn't merge format lists. auto-insert conversion filter */
- switch (link->type) {
- case AVMEDIA_TYPE_VIDEO:
- if (!(filter = avfilter_get_by_name("scale"))) {
- av_log(log_ctx, AV_LOG_ERROR, "'scale' filter "
- "not present, cannot convert pixel formats.\n");
- return AVERROR(EINVAL);
- }
- snprintf(inst_name, sizeof(inst_name), "auto-inserted scaler %d",
- scaler_count++);
- snprintf(scale_args, sizeof(scale_args), "0:0:%s", graph->scale_sws_opts);
- if ((ret = avfilter_graph_create_filter(&convert, filter,
- inst_name, scale_args, NULL,
- graph)) < 0)
- return ret;
- break;
- case AVMEDIA_TYPE_AUDIO:
- if (!(filter = avfilter_get_by_name("aresample"))) {
- av_log(log_ctx, AV_LOG_ERROR, "'resample' filter "
- "not present, cannot convert audio formats.\n");
- return AVERROR(EINVAL);
- }
- snprintf(inst_name, sizeof(inst_name), "auto-inserted resampler %d",
- resampler_count++);
- if ((ret = avfilter_graph_create_filter(&convert, filter,
- inst_name, NULL, NULL, graph)) < 0)
- return ret;
- break;
- default:
- return AVERROR(EINVAL);
- }
- if ((ret = avfilter_insert_filter(link, convert, 0, 0)) < 0)
- return ret;
- filter_query_formats(convert);
- inlink = convert->inputs[0];
- outlink = convert->outputs[0];
- if (!ff_merge_formats( inlink->in_formats, inlink->out_formats) ||
- !ff_merge_formats(outlink->in_formats, outlink->out_formats))
- ret |= AVERROR(ENOSYS);
- if (inlink->type == AVMEDIA_TYPE_AUDIO &&
- (!ff_merge_samplerates(inlink->in_samplerates,
- inlink->out_samplerates) ||
- !ff_merge_channel_layouts(inlink->in_channel_layouts,
- inlink->out_channel_layouts)))
- ret |= AVERROR(ENOSYS);
- if (outlink->type == AVMEDIA_TYPE_AUDIO &&
- (!ff_merge_samplerates(outlink->in_samplerates,
- outlink->out_samplerates) ||
- !ff_merge_channel_layouts(outlink->in_channel_layouts,
- outlink->out_channel_layouts)))
- ret |= AVERROR(ENOSYS);
- if (ret < 0) {
- av_log(log_ctx, AV_LOG_ERROR,
- "Impossible to convert between the formats supported by the filter "
- "'%s' and the filter '%s'\n", link->src->name, link->dst->name);
- return ret;
- }
- #endif
- }
- }
- }
- return 0;
- }
- static int pick_format(AVFilterLink *link, AVFilterLink *ref)
- {
- if (!link || !link->in_formats)
- return 0;
- if (link->type == AVMEDIA_TYPE_VIDEO) {
- if(ref && ref->type == AVMEDIA_TYPE_VIDEO){
- int has_alpha= av_pix_fmt_descriptors[ref->format].nb_components % 2 == 0;
- enum PixelFormat best= PIX_FMT_NONE;
- int i;
- for (i=0; i<link->in_formats->format_count; i++) {
- enum PixelFormat p = link->in_formats->formats[i];
- best= avcodec_find_best_pix_fmt2(best, p, ref->format, has_alpha, NULL);
- }
- link->in_formats->formats[0] = best;
- }
- }
- link->in_formats->format_count = 1;
- link->format = link->in_formats->formats[0];
- if (link->type == AVMEDIA_TYPE_AUDIO) {
- if (!link->in_samplerates->format_count) {
- av_log(link->src, AV_LOG_ERROR, "Cannot select sample rate for"
- " the link between filters %s and %s.\n", link->src->name,
- link->dst->name);
- return AVERROR(EINVAL);
- }
- link->in_samplerates->format_count = 1;
- link->sample_rate = link->in_samplerates->formats[0];
- if (!link->in_channel_layouts->nb_channel_layouts) {
- av_log(link->src, AV_LOG_ERROR, "Cannot select channel layout for"
- "the link between filters %s and %s.\n", link->src->name,
- link->dst->name);
- return AVERROR(EINVAL);
- }
- link->in_channel_layouts->nb_channel_layouts = 1;
- link->channel_layout = link->in_channel_layouts->channel_layouts[0];
- }
- ff_formats_unref(&link->in_formats);
- ff_formats_unref(&link->out_formats);
- ff_formats_unref(&link->in_samplerates);
- ff_formats_unref(&link->out_samplerates);
- ff_channel_layouts_unref(&link->in_channel_layouts);
- ff_channel_layouts_unref(&link->out_channel_layouts);
- return 0;
- }
- #define REDUCE_FORMATS(fmt_type, list_type, list, var, nb, add_format) \
- do { \
- for (i = 0; i < filter->nb_inputs; i++) { \
- AVFilterLink *link = filter->inputs[i]; \
- fmt_type fmt; \
- \
- if (!link->out_ ## list || link->out_ ## list->nb != 1) \
- continue; \
- fmt = link->out_ ## list->var[0]; \
- \
- for (j = 0; j < filter->nb_outputs; j++) { \
- AVFilterLink *out_link = filter->outputs[j]; \
- list_type *fmts; \
- \
- if (link->type != out_link->type || \
- out_link->in_ ## list->nb == 1) \
- continue; \
- fmts = out_link->in_ ## list; \
- \
- if (!out_link->in_ ## list->nb) { \
- add_format(&out_link->in_ ##list, fmt); \
- break; \
- } \
- \
- for (k = 0; k < out_link->in_ ## list->nb; k++) \
- if (fmts->var[k] == fmt) { \
- fmts->var[0] = fmt; \
- fmts->nb = 1; \
- ret = 1; \
- break; \
- } \
- } \
- } \
- } while (0)
- static int reduce_formats_on_filter(AVFilterContext *filter)
- {
- int i, j, k, ret = 0;
- REDUCE_FORMATS(int, AVFilterFormats, formats, formats,
- format_count, ff_add_format);
- REDUCE_FORMATS(int, AVFilterFormats, samplerates, formats,
- format_count, ff_add_format);
- REDUCE_FORMATS(uint64_t, AVFilterChannelLayouts, channel_layouts,
- channel_layouts, nb_channel_layouts, ff_add_channel_layout);
- return ret;
- }
- static void reduce_formats(AVFilterGraph *graph)
- {
- int i, reduced;
- do {
- reduced = 0;
- for (i = 0; i < graph->filter_count; i++)
- reduced |= reduce_formats_on_filter(graph->filters[i]);
- } while (reduced);
- }
- static void swap_samplerates_on_filter(AVFilterContext *filter)
- {
- AVFilterLink *link = NULL;
- int sample_rate;
- int i, j;
- for (i = 0; i < filter->nb_inputs; i++) {
- link = filter->inputs[i];
- if (link->type == AVMEDIA_TYPE_AUDIO &&
- link->out_samplerates->format_count == 1)
- break;
- }
- if (i == filter->nb_inputs)
- return;
- sample_rate = link->out_samplerates->formats[0];
- for (i = 0; i < filter->nb_outputs; i++) {
- AVFilterLink *outlink = filter->outputs[i];
- int best_idx, best_diff = INT_MAX;
- if (outlink->type != AVMEDIA_TYPE_AUDIO ||
- outlink->in_samplerates->format_count < 2)
- continue;
- for (j = 0; j < outlink->in_samplerates->format_count; j++) {
- int diff = abs(sample_rate - outlink->in_samplerates->formats[j]);
- if (diff < best_diff) {
- best_diff = diff;
- best_idx = j;
- }
- }
- FFSWAP(int, outlink->in_samplerates->formats[0],
- outlink->in_samplerates->formats[best_idx]);
- }
- }
- static void swap_samplerates(AVFilterGraph *graph)
- {
- int i;
- for (i = 0; i < graph->filter_count; i++)
- swap_samplerates_on_filter(graph->filters[i]);
- }
- static void swap_channel_layouts_on_filter(AVFilterContext *filter)
- {
- AVFilterLink *link = NULL;
- uint64_t chlayout;
- int i, j;
- for (i = 0; i < filter->nb_inputs; i++) {
- link = filter->inputs[i];
- if (link->type == AVMEDIA_TYPE_AUDIO &&
- link->out_channel_layouts->nb_channel_layouts == 1)
- break;
- }
- if (i == filter->nb_inputs)
- return;
- chlayout = link->out_channel_layouts->channel_layouts[0];
- for (i = 0; i < filter->nb_outputs; i++) {
- AVFilterLink *outlink = filter->outputs[i];
- int best_idx, best_score = INT_MIN;
- if (outlink->type != AVMEDIA_TYPE_AUDIO ||
- outlink->in_channel_layouts->nb_channel_layouts < 2)
- continue;
- for (j = 0; j < outlink->in_channel_layouts->nb_channel_layouts; j++) {
- uint64_t out_chlayout = outlink->in_channel_layouts->channel_layouts[j];
- int matched_channels = av_get_channel_layout_nb_channels(chlayout &
- out_chlayout);
- int extra_channels = av_get_channel_layout_nb_channels(out_chlayout &
- (~chlayout));
- int score = matched_channels - extra_channels;
- if (score > best_score) {
- best_score = score;
- best_idx = j;
- }
- }
- FFSWAP(uint64_t, outlink->in_channel_layouts->channel_layouts[0],
- outlink->in_channel_layouts->channel_layouts[best_idx]);
- }
- }
- static void swap_channel_layouts(AVFilterGraph *graph)
- {
- int i;
- for (i = 0; i < graph->filter_count; i++)
- swap_channel_layouts_on_filter(graph->filters[i]);
- }
- static void swap_sample_fmts_on_filter(AVFilterContext *filter)
- {
- AVFilterLink *link = NULL;
- int format, bps;
- int i, j;
- for (i = 0; i < filter->nb_inputs; i++) {
- link = filter->inputs[i];
- if (link->type == AVMEDIA_TYPE_AUDIO &&
- link->out_formats->format_count == 1)
- break;
- }
- if (i == filter->nb_inputs)
- return;
- format = link->out_formats->formats[0];
- bps = av_get_bytes_per_sample(format);
- for (i = 0; i < filter->nb_outputs; i++) {
- AVFilterLink *outlink = filter->outputs[i];
- int best_idx = -1, best_score = INT_MIN;
- if (outlink->type != AVMEDIA_TYPE_AUDIO ||
- outlink->in_formats->format_count < 2)
- continue;
- for (j = 0; j < outlink->in_formats->format_count; j++) {
- int out_format = outlink->in_formats->formats[j];
- int out_bps = av_get_bytes_per_sample(out_format);
- int score;
- if (av_get_packed_sample_fmt(out_format) == format ||
- av_get_planar_sample_fmt(out_format) == format) {
- best_idx = j;
- break;
- }
- /* for s32 and float prefer double to prevent loss of information */
- if (bps == 4 && out_bps == 8) {
- best_idx = j;
- break;
- }
- /* prefer closest higher or equal bps */
- score = -abs(out_bps - bps);
- if (out_bps >= bps)
- score += INT_MAX/2;
- if (score > best_score) {
- best_score = score;
- best_idx = j;
- }
- }
- av_assert0(best_idx >= 0);
- FFSWAP(int, outlink->in_formats->formats[0],
- outlink->in_formats->formats[best_idx]);
- }
- }
- static void swap_sample_fmts(AVFilterGraph *graph)
- {
- int i;
- for (i = 0; i < graph->filter_count; i++)
- swap_sample_fmts_on_filter(graph->filters[i]);
- }
- static int pick_formats(AVFilterGraph *graph)
- {
- int i, j, ret;
- int change;
- do{
- change = 0;
- for (i = 0; i < graph->filter_count; i++) {
- AVFilterContext *filter = graph->filters[i];
- if (filter->nb_inputs){
- for (j = 0; j < filter->nb_inputs; j++){
- if(filter->inputs[j]->in_formats && filter->inputs[j]->in_formats->format_count == 1) {
- pick_format(filter->inputs[j], NULL);
- change = 1;
- }
- }
- }
- if (filter->nb_outputs){
- for (j = 0; j < filter->nb_outputs; j++){
- if(filter->outputs[j]->in_formats && filter->outputs[j]->in_formats->format_count == 1) {
- pick_format(filter->outputs[j], NULL);
- change = 1;
- }
- }
- }
- if (filter->nb_inputs && filter->nb_outputs && filter->inputs[0]->format>=0) {
- for (j = 0; j < filter->nb_outputs; j++) {
- if(filter->outputs[j]->format<0) {
- pick_format(filter->outputs[j], filter->inputs[0]);
- change = 1;
- }
- }
- }
- }
- }while(change);
- for (i = 0; i < graph->filter_count; i++) {
- AVFilterContext *filter = graph->filters[i];
- for (j = 0; j < filter->nb_inputs; j++)
- if ((ret = pick_format(filter->inputs[j], NULL)) < 0)
- return ret;
- for (j = 0; j < filter->nb_outputs; j++)
- if ((ret = pick_format(filter->outputs[j], NULL)) < 0)
- return ret;
- }
- return 0;
- }
- /**
- * Configure the formats of all the links in the graph.
- */
- static int graph_config_formats(AVFilterGraph *graph, AVClass *log_ctx)
- {
- int ret;
- /* find supported formats from sub-filters, and merge along links */
- if ((ret = query_formats(graph, log_ctx)) < 0)
- return ret;
- /* Once everything is merged, it's possible that we'll still have
- * multiple valid media format choices. We try to minimize the amount
- * of format conversion inside filters */
- reduce_formats(graph);
- /* for audio filters, ensure the best format, sample rate and channel layout
- * is selected */
- swap_sample_fmts(graph);
- swap_samplerates(graph);
- swap_channel_layouts(graph);
- if ((ret = pick_formats(graph)) < 0)
- return ret;
- return 0;
- }
- static int ff_avfilter_graph_config_pointers(AVFilterGraph *graph,
- AVClass *log_ctx)
- {
- unsigned i, j;
- int sink_links_count = 0, n = 0;
- AVFilterContext *f;
- AVFilterLink **sinks;
- for (i = 0; i < graph->filter_count; i++) {
- f = graph->filters[i];
- for (j = 0; j < f->nb_inputs; j++) {
- f->inputs[j]->graph = graph;
- f->inputs[j]->age_index = -1;
- }
- for (j = 0; j < f->nb_outputs; j++) {
- f->outputs[j]->graph = graph;
- f->outputs[j]->age_index= -1;
- }
- if (!f->nb_outputs) {
- if (f->nb_inputs > INT_MAX - sink_links_count)
- return AVERROR(EINVAL);
- sink_links_count += f->nb_inputs;
- }
- }
- sinks = av_calloc(sink_links_count, sizeof(*sinks));
- if (!sinks)
- return AVERROR(ENOMEM);
- for (i = 0; i < graph->filter_count; i++) {
- f = graph->filters[i];
- if (!f->nb_outputs) {
- for (j = 0; j < f->nb_inputs; j++) {
- sinks[n] = f->inputs[j];
- f->inputs[j]->age_index = n++;
- }
- }
- }
- av_assert0(n == sink_links_count);
- graph->sink_links = sinks;
- graph->sink_links_count = sink_links_count;
- return 0;
- }
- static int graph_insert_fifos(AVFilterGraph *graph, AVClass *log_ctx)
- {
- AVFilterContext *f;
- int i, j, ret;
- int fifo_count = 0;
- for (i = 0; i < graph->filter_count; i++) {
- f = graph->filters[i];
- for (j = 0; j < f->nb_inputs; j++) {
- AVFilterLink *link = f->inputs[j];
- AVFilterContext *fifo_ctx;
- AVFilter *fifo;
- char name[32];
- if (!link->dstpad->needs_fifo)
- continue;
- fifo = f->inputs[j]->type == AVMEDIA_TYPE_VIDEO ?
- avfilter_get_by_name("fifo") :
- avfilter_get_by_name("afifo");
- snprintf(name, sizeof(name), "auto-inserted fifo %d", fifo_count++);
- ret = avfilter_graph_create_filter(&fifo_ctx, fifo, name, NULL,
- NULL, graph);
- if (ret < 0)
- return ret;
- ret = avfilter_insert_filter(link, fifo_ctx, 0, 0);
- if (ret < 0)
- return ret;
- }
- }
- return 0;
- }
- int avfilter_graph_config(AVFilterGraph *graphctx, void *log_ctx)
- {
- int ret;
- if ((ret = graph_check_validity(graphctx, log_ctx)))
- return ret;
- if ((ret = graph_insert_fifos(graphctx, log_ctx)) < 0)
- return ret;
- if ((ret = graph_config_formats(graphctx, log_ctx)))
- return ret;
- if ((ret = graph_config_links(graphctx, log_ctx)))
- return ret;
- if ((ret = ff_avfilter_graph_config_pointers(graphctx, log_ctx)))
- return ret;
- return 0;
- }
- int avfilter_graph_send_command(AVFilterGraph *graph, const char *target, const char *cmd, const char *arg, char *res, int res_len, int flags)
- {
- int i, r = AVERROR(ENOSYS);
- if(!graph)
- return r;
- if((flags & AVFILTER_CMD_FLAG_ONE) && !(flags & AVFILTER_CMD_FLAG_FAST)) {
- r=avfilter_graph_send_command(graph, target, cmd, arg, res, res_len, flags | AVFILTER_CMD_FLAG_FAST);
- if(r != AVERROR(ENOSYS))
- return r;
- }
- if(res_len && res)
- res[0]= 0;
- for (i = 0; i < graph->filter_count; i++) {
- AVFilterContext *filter = graph->filters[i];
- if(!strcmp(target, "all") || (filter->name && !strcmp(target, filter->name)) || !strcmp(target, filter->filter->name)){
- r = avfilter_process_command(filter, cmd, arg, res, res_len, flags);
- if(r != AVERROR(ENOSYS)) {
- if((flags & AVFILTER_CMD_FLAG_ONE) || r<0)
- return r;
- }
- }
- }
- return r;
- }
- int avfilter_graph_queue_command(AVFilterGraph *graph, const char *target, const char *command, const char *arg, int flags, double ts)
- {
- int i;
- if(!graph)
- return 0;
- for (i = 0; i < graph->filter_count; i++) {
- AVFilterContext *filter = graph->filters[i];
- if(filter && (!strcmp(target, "all") || !strcmp(target, filter->name) || !strcmp(target, filter->filter->name))){
- AVFilterCommand **que = &filter->command_queue, *next;
- while(*que && (*que)->time <= ts)
- que = &(*que)->next;
- next= *que;
- *que= av_mallocz(sizeof(AVFilterCommand));
- (*que)->command = av_strdup(command);
- (*que)->arg = av_strdup(arg);
- (*que)->time = ts;
- (*que)->flags = flags;
- (*que)->next = next;
- if(flags & AVFILTER_CMD_FLAG_ONE)
- return 0;
- }
- }
- return 0;
- }
- static void heap_bubble_up(AVFilterGraph *graph,
- AVFilterLink *link, int index)
- {
- AVFilterLink **links = graph->sink_links;
- while (index) {
- int parent = (index - 1) >> 1;
- if (links[parent]->current_pts >= link->current_pts)
- break;
- links[index] = links[parent];
- links[index]->age_index = index;
- index = parent;
- }
- links[index] = link;
- link->age_index = index;
- }
- static void heap_bubble_down(AVFilterGraph *graph,
- AVFilterLink *link, int index)
- {
- AVFilterLink **links = graph->sink_links;
- while (1) {
- int child = 2 * index + 1;
- if (child >= graph->sink_links_count)
- break;
- if (child + 1 < graph->sink_links_count &&
- links[child + 1]->current_pts < links[child]->current_pts)
- child++;
- if (link->current_pts < links[child]->current_pts)
- break;
- links[index] = links[child];
- links[index]->age_index = index;
- index = child;
- }
- links[index] = link;
- link->age_index = index;
- }
- void ff_avfilter_graph_update_heap(AVFilterGraph *graph, AVFilterLink *link)
- {
- heap_bubble_up (graph, link, link->age_index);
- heap_bubble_down(graph, link, link->age_index);
- }
- int avfilter_graph_request_oldest(AVFilterGraph *graph)
- {
- while (graph->sink_links_count) {
- AVFilterLink *oldest = graph->sink_links[0];
- int r = ff_request_frame(oldest);
- if (r != AVERROR_EOF)
- return r;
- /* EOF: remove the link from the heap */
- if (oldest->age_index < --graph->sink_links_count)
- heap_bubble_down(graph, graph->sink_links[graph->sink_links_count],
- oldest->age_index);
- oldest->age_index = -1;
- }
- return AVERROR_EOF;
- }
|