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

   1 /*
   2  * Copyright (c) 2007 Bobby Bingham
   3  *
   4  * This file is part of FFmpeg.
   5  *
   6  * FFmpeg is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * FFmpeg is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with FFmpeg; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19  */
  20
  21 /**
  22  * @file
  23  * audio and video splitter
  24  */
  25
  26 #include <stdio.h>
  27
  28 #include "libavutil/attributes.h"
  29 #include "libavutil/internal.h"
  30 #include "libavutil/mem.h"
  31 #include "libavutil/opt.h"
  32
  33 #include "avfilter.h"
  34 #include "audio.h"
  35 #include "formats.h"
  36 #include "internal.h"
  37 #include "video.h"
  38
  39 typedef struct SplitContext {
  40     const AVClass *class;
  41     int nb_outputs;
  42 } SplitContext;
  43
  44 static av_cold int split_init(AVFilterContext *ctx)
  45 {
  46     SplitContext *s = ctx->priv;
  47     int i;
  48
  49     for (i = 0; i < s->nb_outputs; i++) {
  50         char name[32];
  51         AVFilterPad pad = { 0 };
  52
  53         snprintf(name, sizeof(name), "output%d", i);
  54         pad.type = ctx->filter->inputs[0].type;
  55         pad.name = av_strdup(name);
  56         if (!pad.name)
  57             return AVERROR(ENOMEM);
  58
  59         ff_insert_outpad(ctx, i, &pad);
  60     }
  61
  62     return 0;
  63 }
  64
  65 static av_cold void split_uninit(AVFilterContext *ctx)
  66 {
  67     int i;
  68
  69     for (i = 0; i < ctx->nb_outputs; i++)
  70         av_freep(&ctx->output_pads[i].name);
  71 }
  72
  73 static int filter_frame(AVFilterLink *inlink, AVFrame *frame)
  74 {
  75     AVFilterContext *ctx = inlink->dst;
  76     int i, ret = AVERROR_EOF;
  77
  78     for (i = 0; i < ctx->nb_outputs; i++) {
  79         AVFrame *buf_out;
  80
  81         if (ctx->outputs[i]->status)
  82             continue;
  83         buf_out = av_frame_clone(frame);
  84         if (!buf_out) {
  85             ret = AVERROR(ENOMEM);
  86             break;
  87         }
  88
  89         ret = ff_filter_frame(ctx->outputs[i], buf_out);
  90         if (ret < 0)
  91             break;
  92     }
  93     av_frame_free(&frame);
  94     return ret;
  95 }
  96
  97 #define OFFSET(x) offsetof(SplitContext, x)
  98 #define FLAGS AV_OPT_FLAG_AUDIO_PARAM | AV_OPT_FLAG_VIDEO_PARAM
  99 static const AVOption options[] = {
 100     { "outputs", "set number of outputs", OFFSET(nb_outputs), AV_OPT_TYPE_INT, { .i64 = 2 }, 1, INT_MAX, FLAGS },
 101     { NULL }
 102 };
 103
 104 #define split_options options
 105 AVFILTER_DEFINE_CLASS(split);
 106
 107 #define asplit_options options
 108 AVFILTER_DEFINE_CLASS(asplit);
 109
 110 static const AVFilterPad avfilter_vf_split_inputs[] = {
 111     {
 112         .name         = "default",
 113         .type         = AVMEDIA_TYPE_VIDEO,
 114         .filter_frame = filter_frame,
 115     },
 116     { NULL }
 117 };
 118
 119 AVFilter ff_vf_split = {
 120     .name        = "split",
 121     .description = NULL_IF_CONFIG_SMALL("Pass on the input to N video outputs."),
 122     .priv_size   = sizeof(SplitContext),
 123     .priv_class  = &split_class,
 124     .init        = split_init,
 125     .uninit      = split_uninit,
 126     .inputs      = avfilter_vf_split_inputs,
 127     .outputs     = NULL,
 128     .flags       = AVFILTER_FLAG_DYNAMIC_OUTPUTS,
 129 };
 130
 131 static const AVFilterPad avfilter_af_asplit_inputs[] = {
 132     {
 133         .name         = "default",
 134         .type         = AVMEDIA_TYPE_AUDIO,
 135         .filter_frame = filter_frame,
 136     },
 137     { NULL }
 138 };
 139
 140 AVFilter ff_af_asplit = {
 141     .name        = "asplit",
 142     .description = NULL_IF_CONFIG_SMALL("Pass on the audio input to N audio outputs."),
 143     .priv_size   = sizeof(SplitContext),
 144     .priv_class  = &asplit_class,
 145     .init        = split_init,
 146     .uninit      = split_uninit,
 147     .query_formats = ff_query_formats_all,
 148     .inputs      = avfilter_af_asplit_inputs,
 149     .outputs     = NULL,
 150     .flags       = AVFILTER_FLAG_DYNAMIC_OUTPUTS,
 151 };