FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavfilter/trim.c
Date: 2024-04-25 15:36:26
Exec Total Coverage
Lines: 143 151 94.7%
Functions: 5 5 100.0%
Branches: 115 138 83.3%
Line Branch Exec Source
1 /*
2 * This file is part of FFmpeg.
3 *
4 * FFmpeg is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * FFmpeg is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with FFmpeg; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17 */
18
19 #include <stdint.h>
20
21 #include "config.h"
22 #include "config_components.h"
23
24 #include "libavutil/channel_layout.h"
25 #include "libavutil/common.h"
26 #include "libavutil/log.h"
27 #include "libavutil/mathematics.h"
28 #include "libavutil/opt.h"
29 #include "libavutil/samplefmt.h"
30
31 #include "audio.h"
32 #include "avfilter.h"
33 #include "internal.h"
34 #include "filters.h"
35 #include "video.h"
36
37 typedef struct TrimContext {
38 const AVClass *class;
39
40 /*
41 * AVOptions
42 */
43 int64_t duration;
44 int64_t start_time, end_time;
45 int64_t start_frame, end_frame;
46 /*
47 * in the link timebase for video,
48 * in 1/samplerate for audio
49 */
50 int64_t start_pts, end_pts;
51 int64_t start_sample, end_sample;
52
53 /*
54 * number of video frames that arrived on this filter so far
55 */
56 int64_t nb_frames;
57 /*
58 * number of audio samples that arrived on this filter so far
59 */
60 int64_t nb_samples;
61 /*
62 * timestamp of the first frame in the output, in the timebase units
63 */
64 int64_t first_pts;
65 /*
66 * duration in the timebase units
67 */
68 int64_t duration_tb;
69
70 int64_t next_pts;
71
72 int eof;
73
74 int (*filter_frame)(AVFilterLink *inlink, AVFrame *frame);
75 } TrimContext;
76
77 918 static av_cold int init(AVFilterContext *ctx)
78 {
79 918 TrimContext *s = ctx->priv;
80
81 918 s->first_pts = AV_NOPTS_VALUE;
82
83 918 return 0;
84 }
85
86 #if CONFIG_TRIM_FILTER
87 3957 static int trim_filter_frame(AVFilterLink *inlink, AVFrame *frame)
88 {
89 3957 AVFilterContext *ctx = inlink->dst;
90 3957 TrimContext *s = ctx->priv;
91 int drop;
92
93 /* drop everything if EOF has already been returned */
94
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3957 times.
 3957 if (s->eof) {
95 ✗ av_frame_free(&frame);
96 ✗ return 0;
97 }
98
99
4/4
✓ Branch 0 taken 3935 times.
✓ Branch 1 taken 22 times.
✓ Branch 2 taken 155 times.
✓ Branch 3 taken 3780 times.
 3957 if (s->start_frame >= 0 || s->start_pts != AV_NOPTS_VALUE) {
100 177 drop = 1;
101
4/4
✓ Branch 0 taken 22 times.
✓ Branch 1 taken 155 times.
✓ Branch 2 taken 18 times.
✓ Branch 3 taken 4 times.
 177 if (s->start_frame >= 0 && s->nb_frames >= s->start_frame)
102 18 drop = 0;
103
3/4
✓ Branch 0 taken 166 times.
✓ Branch 1 taken 11 times.
✓ Branch 2 taken 166 times.
✗ Branch 3 not taken.
 177 if (s->start_pts != AV_NOPTS_VALUE && frame->pts != AV_NOPTS_VALUE &&
104
2/2
✓ Branch 0 taken 151 times.
✓ Branch 1 taken 15 times.
 166 frame->pts >= s->start_pts)
105 151 drop = 0;
106
2/2
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 163 times.
 177 if (drop)
107 14 goto drop;
108 }
109
110
3/4
✓ Branch 0 taken 116 times.
✓ Branch 1 taken 3827 times.
✓ Branch 2 taken 116 times.
✗ Branch 3 not taken.
 3943 if (s->first_pts == AV_NOPTS_VALUE && frame->pts != AV_NOPTS_VALUE)
111 116 s->first_pts = frame->pts;
112
113
6/6
✓ Branch 0 taken 3925 times.
✓ Branch 1 taken 18 times.
✓ Branch 2 taken 3922 times.
✓ Branch 3 taken 3 times.
✓ Branch 4 taken 3782 times.
✓ Branch 5 taken 140 times.
 3943 if (s->end_frame != INT64_MAX || s->end_pts != AV_NOPTS_VALUE || s->duration_tb) {
114 3803 drop = 1;
115
116
4/4
✓ Branch 0 taken 18 times.
✓ Branch 1 taken 3785 times.
✓ Branch 2 taken 9 times.
✓ Branch 3 taken 9 times.
 3803 if (s->end_frame != INT64_MAX && s->nb_frames < s->end_frame)
117 9 drop = 0;
118
3/4
✓ Branch 0 taken 13 times.
✓ Branch 1 taken 3790 times.
✓ Branch 2 taken 13 times.
✗ Branch 3 not taken.
 3803 if (s->end_pts != AV_NOPTS_VALUE && frame->pts != AV_NOPTS_VALUE &&
119
2/2
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 2 times.
 13 frame->pts < s->end_pts)
120 11 drop = 0;
121
3/4
✓ Branch 0 taken 3782 times.
✓ Branch 1 taken 21 times.
✓ Branch 2 taken 3782 times.
✗ Branch 3 not taken.
 3803 if (s->duration_tb && frame->pts != AV_NOPTS_VALUE &&
122
2/2
✓ Branch 0 taken 3677 times.
✓ Branch 1 taken 105 times.
 3782 frame->pts - s->first_pts < s->duration_tb)
123 3677 drop = 0;
124
125
2/2
✓ Branch 0 taken 108 times.
✓ Branch 1 taken 3695 times.
 3803 if (drop) {
126 108 s->eof = 1;
127 108 ff_inlink_set_status(inlink, AVERROR_EOF);
128 108 ff_outlink_set_status(ctx->outputs[0], AVERROR_EOF, frame->pts);
129 108 goto drop;
130 }
131 }
132
133 3835 s->nb_frames++;
134
135 3835 return ff_filter_frame(ctx->outputs[0], frame);
136
137 122 drop:
138
2/2
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 108 times.
 122 if (!s->eof)
139 14 ff_filter_set_ready(ctx, 100);
140 122 s->nb_frames++;
141 122 av_frame_free(&frame);
142 122 return 0;
143 }
144 #endif // CONFIG_TRIM_FILTER
145
146 #if CONFIG_ATRIM_FILTER
147 10370 static int atrim_filter_frame(AVFilterLink *inlink, AVFrame *frame)
148 {
149 10370 AVFilterContext *ctx = inlink->dst;
150 10370 TrimContext *s = ctx->priv;
151 int64_t start_sample, end_sample;
152 int64_t pts;
153 int drop;
154
155 /* drop everything if EOF has already been returned */
156
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10370 times.
 10370 if (s->eof) {
157 ✗ av_frame_free(&frame);
158 ✗ return 0;
159 }
160
161
1/2
✓ Branch 0 taken 10370 times.
✗ Branch 1 not taken.
 10370 if (frame->pts != AV_NOPTS_VALUE)
162 10370 pts = av_rescale_q(frame->pts, inlink->time_base,
163 10370 (AVRational){ 1, inlink->sample_rate });
164 else
165 ✗ pts = s->next_pts;
166 10370 s->next_pts = pts + frame->nb_samples;
167
168 /* check if at least a part of the frame is after the start time */
169
4/4
✓ Branch 0 taken 10365 times.
✓ Branch 1 taken 5 times.
✓ Branch 2 taken 8590 times.
✓ Branch 3 taken 1775 times.
 10370 if (s->start_sample < 0 && s->start_pts == AV_NOPTS_VALUE) {
170 8590 start_sample = 0;
171 } else {
172 1780 drop = 1;
173 1780 start_sample = frame->nb_samples;
174
175
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 1775 times.
 1780 if (s->start_sample >= 0 &&
176
1/2
✓ Branch 0 taken 5 times.
✗ Branch 1 not taken.
 5 s->nb_samples + frame->nb_samples > s->start_sample) {
177 5 drop = 0;
178 5 start_sample = FFMIN(start_sample, s->start_sample - s->nb_samples);
179 }
180
181
3/4
✓ Branch 0 taken 1778 times.
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 1778 times.
✗ Branch 3 not taken.
 1780 if (s->start_pts != AV_NOPTS_VALUE && pts != AV_NOPTS_VALUE &&
182
2/2
✓ Branch 0 taken 1454 times.
✓ Branch 1 taken 324 times.
 1778 pts + frame->nb_samples > s->start_pts) {
183 1454 drop = 0;
184 1454 start_sample = FFMIN(start_sample, s->start_pts - pts);
185 }
186
187
2/2
✓ Branch 0 taken 324 times.
✓ Branch 1 taken 1456 times.
 1780 if (drop)
188 324 goto drop;
189 }
190
191
2/2
✓ Branch 0 taken 459 times.
✓ Branch 1 taken 9587 times.
 10046 if (s->first_pts == AV_NOPTS_VALUE)
192 459 s->first_pts = pts + start_sample;
193
194 /* check if at least a part of the frame is before the end time */
195
6/6
✓ Branch 0 taken 4796 times.
✓ Branch 1 taken 5250 times.
✓ Branch 2 taken 4776 times.
✓ Branch 3 taken 20 times.
✓ Branch 4 taken 1446 times.
✓ Branch 5 taken 3330 times.
 10046 if (s->end_sample == INT64_MAX && s->end_pts == AV_NOPTS_VALUE && !s->duration_tb) {
196 1446 end_sample = frame->nb_samples;
197 } else {
198 8600 drop = 1;
199 8600 end_sample = 0;
200
201
2/2
✓ Branch 0 taken 5250 times.
✓ Branch 1 taken 3350 times.
 8600 if (s->end_sample != INT64_MAX &&
202
2/2
✓ Branch 0 taken 4915 times.
✓ Branch 1 taken 335 times.
 5250 s->nb_samples < s->end_sample) {
203 4915 drop = 0;
204 4915 end_sample = FFMAX(end_sample, s->end_sample - s->nb_samples);
205 }
206
207
3/4
✓ Branch 0 taken 23 times.
✓ Branch 1 taken 8577 times.
✓ Branch 2 taken 23 times.
✗ Branch 3 not taken.
 8600 if (s->end_pts != AV_NOPTS_VALUE && pts != AV_NOPTS_VALUE &&
208
2/2
✓ Branch 0 taken 20 times.
✓ Branch 1 taken 3 times.
 23 pts < s->end_pts) {
209 20 drop = 0;
210 20 end_sample = FFMAX(end_sample, s->end_pts - pts);
211 }
212
213
4/4
✓ Branch 0 taken 3330 times.
✓ Branch 1 taken 5270 times.
✓ Branch 2 taken 3226 times.
✓ Branch 3 taken 104 times.
 8600 if (s->duration_tb && pts - s->first_pts < s->duration_tb) {
214 3226 drop = 0;
215 3226 end_sample = FFMAX(end_sample, s->first_pts + s->duration_tb - pts);
216 }
217
218
2/2
✓ Branch 0 taken 441 times.
✓ Branch 1 taken 8159 times.
 8600 if (drop) {
219 441 s->eof = 1;
220 441 ff_inlink_set_status(inlink, AVERROR_EOF);
221 441 ff_outlink_set_status(ctx->outputs[0], AVERROR_EOF, frame->pts);
222 441 goto drop;
223 }
224 }
225
226 9605 s->nb_samples += frame->nb_samples;
227 9605 start_sample = FFMAX(0, start_sample);
228 9605 end_sample = FFMIN(frame->nb_samples, end_sample);
229
2/4
✓ Branch 0 taken 9605 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 9605 times.
 9605 if (start_sample >= end_sample || !frame->nb_samples)
230 ✗ goto drop;
231
232
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 9599 times.
 9605 if (start_sample) {
233 6 AVFrame *out = ff_get_audio_buffer(ctx->outputs[0], end_sample - start_sample);
234
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 if (!out) {
235 ✗ av_frame_free(&frame);
236 ✗ return AVERROR(ENOMEM);
237 }
238
239 6 av_frame_copy_props(out, frame);
240 6 av_samples_copy(out->extended_data, frame->extended_data, 0, start_sample,
241 out->nb_samples, inlink->ch_layout.nb_channels,
242 6 frame->format);
243
1/2
✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
 6 if (out->pts != AV_NOPTS_VALUE)
244 6 out->pts += av_rescale_q(start_sample, (AVRational){ 1, out->sample_rate },
245 inlink->time_base);
246
247 6 av_frame_free(&frame);
248 6 frame = out;
249 } else
250 9599 frame->nb_samples = end_sample;
251
252 9605 return ff_filter_frame(ctx->outputs[0], frame);
253
254 765 drop:
255
2/2
✓ Branch 0 taken 324 times.
✓ Branch 1 taken 441 times.
 765 if (!s->eof)
256 324 ff_filter_set_ready(ctx, 100);
257 765 s->nb_samples += frame->nb_samples;
258 765 av_frame_free(&frame);
259 765 return 0;
260 }
261 #endif // CONFIG_ATRIM_FILTER
262
263 575 static int config_input(AVFilterLink *inlink)
264 {
265 575 AVFilterContext *ctx = inlink->dst;
266 575 TrimContext *s = ctx->priv;
267 116 AVRational tb = (inlink->type == AVMEDIA_TYPE_VIDEO) ?
268
2/2
✓ Branch 0 taken 116 times.
✓ Branch 1 taken 459 times.
 575 inlink->time_base : (AVRational){ 1, inlink->sample_rate };
269
270 #if CONFIG_TRIM_FILTER
271
2/2
✓ Branch 0 taken 116 times.
✓ Branch 1 taken 459 times.
 575 if (inlink->type == AVMEDIA_TYPE_VIDEO)
272 116 s->filter_frame = trim_filter_frame;
273 #endif
274 #if CONFIG_ATRIM_FILTER
275
2/2
✓ Branch 0 taken 459 times.
✓ Branch 1 taken 116 times.
 575 if (inlink->type == AVMEDIA_TYPE_AUDIO)
276 459 s->filter_frame = atrim_filter_frame;
277 #endif
278
2/2
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 550 times.
 575 if (s->start_time != INT64_MAX) {
279 25 int64_t start_pts = av_rescale_q(s->start_time, AV_TIME_BASE_Q, tb);
280
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 25 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 25 if (s->start_pts == AV_NOPTS_VALUE || start_pts < s->start_pts)
281 25 s->start_pts = start_pts;
282 }
283
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 570 times.
 575 if (s->end_time != INT64_MAX) {
284 5 int64_t end_pts = av_rescale_q(s->end_time, AV_TIME_BASE_Q, tb);
285
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 5 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 5 if (s->end_pts == AV_NOPTS_VALUE || end_pts > s->end_pts)
286 5 s->end_pts = end_pts;
287 }
288
2/2
✓ Branch 0 taken 216 times.
✓ Branch 1 taken 359 times.
 575 if (s->duration)
289 216 s->duration_tb = av_rescale_q(s->duration, AV_TIME_BASE_Q, tb);
290
291 575 return 0;
292 }
293
294 27995 static int activate(AVFilterContext *ctx)
295 {
296 27995 TrimContext *s = ctx->priv;
297 27995 AVFilterLink *inlink = ctx->inputs[0];
298 27995 AVFilterLink *outlink = ctx->outputs[0];
299
300
2/2
✓ Branch 1 taken 6 times.
✓ Branch 2 taken 27989 times.
 27995 FF_FILTER_FORWARD_STATUS_BACK(outlink, inlink);
301
302
3/4
✓ Branch 0 taken 27989 times.
✗ Branch 1 not taken.
✓ Branch 3 taken 14327 times.
✓ Branch 4 taken 13662 times.
 27989 if (!s->eof && ff_inlink_queued_frames(inlink)) {
303 14327 AVFrame *frame = NULL;
304 int ret;
305
306 14327 ret = ff_inlink_consume_frame(inlink, &frame);
307
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 14327 times.
 14327 if (ret < 0)
308 14327 return ret;
309
1/2
✓ Branch 0 taken 14327 times.
✗ Branch 1 not taken.
 14327 if (ret > 0)
310 14327 return s->filter_frame(inlink, frame);
311 }
312
313
2/2
✓ Branch 1 taken 14 times.
✓ Branch 2 taken 13648 times.
 13662 FF_FILTER_FORWARD_STATUS(inlink, outlink);
314
2/2
✓ Branch 1 taken 13642 times.
✓ Branch 2 taken 6 times.
 13648 FF_FILTER_FORWARD_WANTED(outlink, inlink);
315
316 6 return FFERROR_NOT_READY;
317 }
318
319 #define OFFSET(x) offsetof(TrimContext, x)
320 #define COMMON_OPTS \
321 { "start", "Timestamp of the first frame that " \
322 "should be passed", OFFSET(start_time), AV_OPT_TYPE_DURATION, { .i64 = INT64_MAX }, INT64_MIN, INT64_MAX, FLAGS }, \
323 { "starti", "Timestamp of the first frame that " \
324 "should be passed", OFFSET(start_time), AV_OPT_TYPE_DURATION, { .i64 = INT64_MAX }, INT64_MIN, INT64_MAX, FLAGS }, \
325 { "end", "Timestamp of the first frame that " \
326 "should be dropped again", OFFSET(end_time), AV_OPT_TYPE_DURATION, { .i64 = INT64_MAX }, INT64_MIN, INT64_MAX, FLAGS }, \
327 { "endi", "Timestamp of the first frame that " \
328 "should be dropped again", OFFSET(end_time), AV_OPT_TYPE_DURATION, { .i64 = INT64_MAX }, INT64_MIN, INT64_MAX, FLAGS }, \
329 { "start_pts", "Timestamp of the first frame that should be " \
330 " passed", OFFSET(start_pts), AV_OPT_TYPE_INT64, { .i64 = AV_NOPTS_VALUE }, INT64_MIN, INT64_MAX, FLAGS }, \
331 { "end_pts", "Timestamp of the first frame that should be " \
332 "dropped again", OFFSET(end_pts), AV_OPT_TYPE_INT64, { .i64 = AV_NOPTS_VALUE }, INT64_MIN, INT64_MAX, FLAGS }, \
333 { "duration", "Maximum duration of the output", OFFSET(duration), AV_OPT_TYPE_DURATION, { .i64 = 0 }, 0, INT64_MAX, FLAGS }, \
334 { "durationi", "Maximum duration of the output", OFFSET(duration), AV_OPT_TYPE_DURATION, { .i64 = 0 }, 0, INT64_MAX, FLAGS },
335
336
337 #if CONFIG_TRIM_FILTER
338
339 #define FLAGS AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_FILTERING_PARAM
340 static const AVOption trim_options[] = {
341 COMMON_OPTS
342 { "start_frame", "Number of the first frame that should be passed "
343 "to the output", OFFSET(start_frame), AV_OPT_TYPE_INT64, { .i64 = -1 }, -1, INT64_MAX, FLAGS },
344 { "end_frame", "Number of the first frame that should be dropped "
345 "again", OFFSET(end_frame), AV_OPT_TYPE_INT64, { .i64 = INT64_MAX }, 0, INT64_MAX, FLAGS },
346 { NULL }
347 };
348 #undef FLAGS
349
350 AVFILTER_DEFINE_CLASS(trim);
351
352 static const AVFilterPad trim_inputs[] = {
353 {
354 .name = "default",
355 .type = AVMEDIA_TYPE_VIDEO,
356 .config_props = config_input,
357 },
358 };
359
360 const AVFilter ff_vf_trim = {
361 .name = "trim",
362 .description = NULL_IF_CONFIG_SMALL("Pick one continuous section from the input, drop the rest."),
363 .init = init,
364 .activate = activate,
365 .priv_size = sizeof(TrimContext),
366 .priv_class = &trim_class,
367 FILTER_INPUTS(trim_inputs),
368 FILTER_OUTPUTS(ff_video_default_filterpad),
369 };
370 #endif // CONFIG_TRIM_FILTER
371
372 #if CONFIG_ATRIM_FILTER
373
374 #define FLAGS AV_OPT_FLAG_AUDIO_PARAM | AV_OPT_FLAG_FILTERING_PARAM
375 static const AVOption atrim_options[] = {
376 COMMON_OPTS
377 { "start_sample", "Number of the first audio sample that should be "
378 "passed to the output", OFFSET(start_sample), AV_OPT_TYPE_INT64, { .i64 = -1 }, -1, INT64_MAX, FLAGS },
379 { "end_sample", "Number of the first audio sample that should be "
380 "dropped again", OFFSET(end_sample), AV_OPT_TYPE_INT64, { .i64 = INT64_MAX }, 0, INT64_MAX, FLAGS },
381 { NULL }
382 };
383 #undef FLAGS
384
385 AVFILTER_DEFINE_CLASS(atrim);
386
387 static const AVFilterPad atrim_inputs[] = {
388 {
389 .name = "default",
390 .type = AVMEDIA_TYPE_AUDIO,
391 .config_props = config_input,
392 },
393 };
394
395 const AVFilter ff_af_atrim = {
396 .name = "atrim",
397 .description = NULL_IF_CONFIG_SMALL("Pick one continuous section from the input, drop the rest."),
398 .init = init,
399 .activate = activate,
400 .priv_size = sizeof(TrimContext),
401 .priv_class = &atrim_class,
402 .flags = AVFILTER_FLAG_METADATA_ONLY,
403 FILTER_INPUTS(atrim_inputs),
404 FILTER_OUTPUTS(ff_audio_default_filterpad),
405 };
406 #endif // CONFIG_ATRIM_FILTER
407