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 |
|
|
|
46 |
|
34 |
static av_cold int split_init(AVFilterContext *ctx) |
47 |
|
|
{ |
48 |
|
34 |
SplitContext *s = ctx->priv; |
49 |
|
|
int i, ret; |
50 |
|
|
|
51 |
✓✓ |
102 |
for (i = 0; i < s->nb_outputs; i++) { |
52 |
|
68 |
AVFilterPad pad = { 0 }; |
53 |
|
|
|
54 |
|
68 |
pad.type = ctx->filter->inputs[0].type; |
55 |
|
68 |
pad.name = av_asprintf("output%d", i); |
56 |
✗✓ |
68 |
if (!pad.name) |
57 |
|
|
return AVERROR(ENOMEM); |
58 |
|
|
|
59 |
✗✓ |
68 |
if ((ret = ff_insert_outpad(ctx, i, &pad)) < 0) { |
60 |
|
|
av_freep(&pad.name); |
61 |
|
|
return ret; |
62 |
|
|
} |
63 |
|
|
} |
64 |
|
|
|
65 |
|
34 |
return 0; |
66 |
|
|
} |
67 |
|
|
|
68 |
|
34 |
static av_cold void split_uninit(AVFilterContext *ctx) |
69 |
|
|
{ |
70 |
|
|
int i; |
71 |
|
|
|
72 |
✓✓ |
102 |
for (i = 0; i < ctx->nb_outputs; i++) |
73 |
|
68 |
av_freep(&ctx->output_pads[i].name); |
74 |
|
34 |
} |
75 |
|
|
|
76 |
|
516 |
static int filter_frame(AVFilterLink *inlink, AVFrame *frame) |
77 |
|
|
{ |
78 |
|
516 |
AVFilterContext *ctx = inlink->dst; |
79 |
|
516 |
int i, ret = AVERROR_EOF; |
80 |
|
|
|
81 |
✓✓ |
1548 |
for (i = 0; i < ctx->nb_outputs; i++) { |
82 |
|
|
AVFrame *buf_out; |
83 |
|
|
|
84 |
✗✓ |
1032 |
if (ff_outlink_get_status(ctx->outputs[i])) |
85 |
|
|
continue; |
86 |
|
1032 |
buf_out = av_frame_clone(frame); |
87 |
✗✓ |
1032 |
if (!buf_out) { |
88 |
|
|
ret = AVERROR(ENOMEM); |
89 |
|
|
break; |
90 |
|
|
} |
91 |
|
|
|
92 |
|
1032 |
ret = ff_filter_frame(ctx->outputs[i], buf_out); |
93 |
✗✓ |
1032 |
if (ret < 0) |
94 |
|
|
break; |
95 |
|
|
} |
96 |
|
516 |
av_frame_free(&frame); |
97 |
|
516 |
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 |
|
|
}; |