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/avstring.h"
30 #include "libavutil/internal.h"
31 #include "libavutil/mem.h"
32 #include "libavutil/opt.h"
33
34 #include "avfilter.h"
35 #include "audio.h"
36 #include "filters.h"
37 #include "formats.h"
38 #include "internal.h"
39 #include "video.h"
40
41 typedef struct SplitContext {
42 const AVClass *class;
43 int nb_outputs;
44 } SplitContext;
45
split_init(AVFilterContext * ctx)46 static av_cold int split_init(AVFilterContext *ctx)
47 {
48 SplitContext *s = ctx->priv;
49 int i, ret;
50
51 for (i = 0; i < s->nb_outputs; i++) {
52 AVFilterPad pad = { 0 };
53
54 pad.type = ctx->filter->inputs[0].type;
55 pad.name = av_asprintf("output%d", i);
56 if (!pad.name)
57 return AVERROR(ENOMEM);
58
59 if ((ret = ff_insert_outpad(ctx, i, &pad)) < 0) {
60 av_freep(&pad.name);
61 return ret;
62 }
63 }
64
65 return 0;
66 }
67
split_uninit(AVFilterContext * ctx)68 static av_cold void split_uninit(AVFilterContext *ctx)
69 {
70 int i;
71
72 for (i = 0; i < ctx->nb_outputs; i++)
73 av_freep(&ctx->output_pads[i].name);
74 }
75
filter_frame(AVFilterLink * inlink,AVFrame * frame)76 static int filter_frame(AVFilterLink *inlink, AVFrame *frame)
77 {
78 AVFilterContext *ctx = inlink->dst;
79 int i, ret = AVERROR_EOF;
80
81 for (i = 0; i < ctx->nb_outputs; i++) {
82 AVFrame *buf_out;
83
84 if (ff_outlink_get_status(ctx->outputs[i]))
85 continue;
86 buf_out = av_frame_clone(frame);
87 if (!buf_out) {
88 ret = AVERROR(ENOMEM);
89 break;
90 }
91
92 ret = ff_filter_frame(ctx->outputs[i], buf_out);
93 if (ret < 0)
94 break;
95 }
96 av_frame_free(&frame);
97 return ret;
98 }
99
100 #define OFFSET(x) offsetof(SplitContext, x)
101 #define FLAGS (AV_OPT_FLAG_AUDIO_PARAM | AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_FILTERING_PARAM)
102 static const AVOption options[] = {
103 { "outputs", "set number of outputs", OFFSET(nb_outputs), AV_OPT_TYPE_INT, { .i64 = 2 }, 1, INT_MAX, FLAGS },
104 { NULL }
105 };
106
107 #define split_options options
108 AVFILTER_DEFINE_CLASS(split);
109
110 #define asplit_options options
111 AVFILTER_DEFINE_CLASS(asplit);
112
113 static const AVFilterPad avfilter_vf_split_inputs[] = {
114 {
115 .name = "default",
116 .type = AVMEDIA_TYPE_VIDEO,
117 .filter_frame = filter_frame,
118 },
119 { NULL }
120 };
121
122 AVFilter ff_vf_split = {
123 .name = "split",
124 .description = NULL_IF_CONFIG_SMALL("Pass on the input to N video outputs."),
125 .priv_size = sizeof(SplitContext),
126 .priv_class = &split_class,
127 .init = split_init,
128 .uninit = split_uninit,
129 .inputs = avfilter_vf_split_inputs,
130 .outputs = NULL,
131 .flags = AVFILTER_FLAG_DYNAMIC_OUTPUTS,
132 };
133
134 static const AVFilterPad avfilter_af_asplit_inputs[] = {
135 {
136 .name = "default",
137 .type = AVMEDIA_TYPE_AUDIO,
138 .filter_frame = filter_frame,
139 },
140 { NULL }
141 };
142
143 AVFilter ff_af_asplit = {
144 .name = "asplit",
145 .description = NULL_IF_CONFIG_SMALL("Pass on the audio input to N audio outputs."),
146 .priv_size = sizeof(SplitContext),
147 .priv_class = &asplit_class,
148 .init = split_init,
149 .uninit = split_uninit,
150 .inputs = avfilter_af_asplit_inputs,
151 .outputs = NULL,
152 .flags = AVFILTER_FLAG_DYNAMIC_OUTPUTS,
153 };
154