FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavfilter/trim.c
Date: 2024-02-16 17:37:06
Exec Total Coverage
Lines: 143 151 94.7%
Functions: 5 5 100.0%
Branches: 116 138 84.1%

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 897 static av_cold int init(AVFilterContext *ctx)
78 {
79 897 TrimContext *s = ctx->priv;
80
81 897 s->first_pts = AV_NOPTS_VALUE;
82
83 897 return 0;
84 }
85
86 #if CONFIG_TRIM_FILTER
87 3907 static int trim_filter_frame(AVFilterLink *inlink, AVFrame *frame)
88 {
89 3907 AVFilterContext *ctx = inlink->dst;
90 3907 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 3907 times.
3907 if (s->eof) {
95 av_frame_free(&frame);
96 return 0;
97 }
98
99
4/4
✓ Branch 0 taken 3885 times.
✓ Branch 1 taken 22 times.
✓ Branch 2 taken 105 times.
✓ Branch 3 taken 3780 times.
3907 if (s->start_frame >= 0 || s->start_pts != AV_NOPTS_VALUE) {
100 127 drop = 1;
101
4/4
✓ Branch 0 taken 22 times.
✓ Branch 1 taken 105 times.
✓ Branch 2 taken 18 times.
✓ Branch 3 taken 4 times.
127 if (s->start_frame >= 0 && s->nb_frames >= s->start_frame)
102 18 drop = 0;
103
3/4
✓ Branch 0 taken 116 times.
✓ Branch 1 taken 11 times.
✓ Branch 2 taken 116 times.
✗ Branch 3 not taken.
127 if (s->start_pts != AV_NOPTS_VALUE && frame->pts != AV_NOPTS_VALUE &&
104
2/2
✓ Branch 0 taken 101 times.
✓ Branch 1 taken 15 times.
116 frame->pts >= s->start_pts)
105 101 drop = 0;
106
2/2
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 113 times.
127 if (drop)
107 14 goto drop;
108 }
109
110
3/4
✓ Branch 0 taken 115 times.
✓ Branch 1 taken 3778 times.
✓ Branch 2 taken 115 times.
✗ Branch 3 not taken.
3893 if (s->first_pts == AV_NOPTS_VALUE && frame->pts != AV_NOPTS_VALUE)
111 115 s->first_pts = frame->pts;
112
113
6/6
✓ Branch 0 taken 3875 times.
✓ Branch 1 taken 18 times.
✓ Branch 2 taken 3872 times.
✓ Branch 3 taken 3 times.
✓ Branch 4 taken 3782 times.
✓ Branch 5 taken 90 times.
3893 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 3785 s->nb_frames++;
134
135 3785 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 12815 static int atrim_filter_frame(AVFilterLink *inlink, AVFrame *frame)
148 {
149 12815 AVFilterContext *ctx = inlink->dst;
150 12815 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 12815 times.
12815 if (s->eof) {
157 av_frame_free(&frame);
158 return 0;
159 }
160
161
1/2
✓ Branch 0 taken 12815 times.
✗ Branch 1 not taken.
12815 if (frame->pts != AV_NOPTS_VALUE)
162 12815 pts = av_rescale_q(frame->pts, inlink->time_base,
163 12815 (AVRational){ 1, inlink->sample_rate });
164 else
165 pts = s->next_pts;
166 12815 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 12807 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 11023 times.
✓ Branch 3 taken 1784 times.
12815 if (s->start_sample < 0 && s->start_pts == AV_NOPTS_VALUE) {
170 11023 start_sample = 0;
171 } else {
172 1792 drop = 1;
173 1792 start_sample = frame->nb_samples;
174
175
2/2
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 1784 times.
1792 if (s->start_sample >= 0 &&
176
2/2
✓ Branch 0 taken 7 times.
✓ Branch 1 taken 1 times.
8 s->nb_samples + frame->nb_samples > s->start_sample) {
177 7 drop = 0;
178 7 start_sample = FFMIN(start_sample, s->start_sample - s->nb_samples);
179 }
180
181
3/4
✓ Branch 0 taken 1790 times.
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 1790 times.
✗ Branch 3 not taken.
1792 if (s->start_pts != AV_NOPTS_VALUE && pts != AV_NOPTS_VALUE &&
182
2/2
✓ Branch 0 taken 1458 times.
✓ Branch 1 taken 332 times.
1790 pts + frame->nb_samples > s->start_pts) {
183 1458 drop = 0;
184 1458 start_sample = FFMIN(start_sample, s->start_pts - pts);
185 }
186
187
2/2
✓ Branch 0 taken 331 times.
✓ Branch 1 taken 1461 times.
1792 if (drop)
188 331 goto drop;
189 }
190
191
2/2
✓ Branch 0 taken 440 times.
✓ Branch 1 taken 12044 times.
12484 if (s->first_pts == AV_NOPTS_VALUE)
192 440 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 9596 times.
✓ Branch 1 taken 2888 times.
✓ Branch 2 taken 9573 times.
✓ Branch 3 taken 23 times.
✓ Branch 4 taken 1446 times.
✓ Branch 5 taken 8127 times.
12484 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 11038 drop = 1;
199 11038 end_sample = 0;
200
201
2/2
✓ Branch 0 taken 2888 times.
✓ Branch 1 taken 8150 times.
11038 if (s->end_sample != INT64_MAX &&
202
2/2
✓ Branch 0 taken 2554 times.
✓ Branch 1 taken 334 times.
2888 s->nb_samples < s->end_sample) {
203 2554 drop = 0;
204 2554 end_sample = FFMAX(end_sample, s->end_sample - s->nb_samples);
205 }
206
207
3/4
✓ Branch 0 taken 28 times.
✓ Branch 1 taken 11010 times.
✓ Branch 2 taken 28 times.
✗ Branch 3 not taken.
11038 if (s->end_pts != AV_NOPTS_VALUE && pts != AV_NOPTS_VALUE &&
208
2/2
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 3 times.
28 pts < s->end_pts) {
209 25 drop = 0;
210 25 end_sample = FFMAX(end_sample, s->end_pts - pts);
211 }
212
213
4/4
✓ Branch 0 taken 8127 times.
✓ Branch 1 taken 2911 times.
✓ Branch 2 taken 8041 times.
✓ Branch 3 taken 86 times.
11038 if (s->duration_tb && pts - s->first_pts < s->duration_tb) {
214 8041 drop = 0;
215 8041 end_sample = FFMAX(end_sample, s->first_pts + s->duration_tb - pts);
216 }
217
218
2/2
✓ Branch 0 taken 422 times.
✓ Branch 1 taken 10616 times.
11038 if (drop) {
219 422 s->eof = 1;
220 422 ff_inlink_set_status(inlink, AVERROR_EOF);
221 422 ff_outlink_set_status(ctx->outputs[0], AVERROR_EOF, frame->pts);
222 422 goto drop;
223 }
224 }
225
226 12062 s->nb_samples += frame->nb_samples;
227 12062 start_sample = FFMAX(0, start_sample);
228 12062 end_sample = FFMIN(frame->nb_samples, end_sample);
229
2/4
✓ Branch 0 taken 12062 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 12062 times.
12062 if (start_sample >= end_sample || !frame->nb_samples)
230 goto drop;
231
232
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 12056 times.
12062 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 12056 frame->nb_samples = end_sample;
251
252 12062 return ff_filter_frame(ctx->outputs[0], frame);
253
254 753 drop:
255
2/2
✓ Branch 0 taken 331 times.
✓ Branch 1 taken 422 times.
753 if (!s->eof)
256 331 ff_filter_set_ready(ctx, 100);
257 753 s->nb_samples += frame->nb_samples;
258 753 av_frame_free(&frame);
259 753 return 0;
260 }
261 #endif // CONFIG_ATRIM_FILTER
262
263 555 static int config_input(AVFilterLink *inlink)
264 {
265 555 AVFilterContext *ctx = inlink->dst;
266 555 TrimContext *s = ctx->priv;
267 115 AVRational tb = (inlink->type == AVMEDIA_TYPE_VIDEO) ?
268
2/2
✓ Branch 0 taken 115 times.
✓ Branch 1 taken 440 times.
555 inlink->time_base : (AVRational){ 1, inlink->sample_rate };
269
270 #if CONFIG_TRIM_FILTER
271
2/2
✓ Branch 0 taken 115 times.
✓ Branch 1 taken 440 times.
555 if (inlink->type == AVMEDIA_TYPE_VIDEO)
272 115 s->filter_frame = trim_filter_frame;
273 #endif
274 #if CONFIG_ATRIM_FILTER
275
2/2
✓ Branch 0 taken 440 times.
✓ Branch 1 taken 115 times.
555 if (inlink->type == AVMEDIA_TYPE_AUDIO)
276 440 s->filter_frame = atrim_filter_frame;
277 #endif
278
2/2
✓ Branch 0 taken 24 times.
✓ Branch 1 taken 531 times.
555 if (s->start_time != INT64_MAX) {
279 24 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 24 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
24 if (s->start_pts == AV_NOPTS_VALUE || start_pts < s->start_pts)
281 24 s->start_pts = start_pts;
282 }
283
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 550 times.
555 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 198 times.
✓ Branch 1 taken 357 times.
555 if (s->duration)
289 198 s->duration_tb = av_rescale_q(s->duration, AV_TIME_BASE_Q, tb);
290
291 555 return 0;
292 }
293
294 31218 static int activate(AVFilterContext *ctx)
295 {
296 31218 TrimContext *s = ctx->priv;
297 31218 AVFilterLink *inlink = ctx->inputs[0];
298 31218 AVFilterLink *outlink = ctx->outputs[0];
299
300
2/2
✓ Branch 1 taken 6 times.
✓ Branch 2 taken 31212 times.
31218 FF_FILTER_FORWARD_STATUS_BACK(outlink, inlink);
301
302
3/4
✓ Branch 0 taken 31212 times.
✗ Branch 1 not taken.
✓ Branch 3 taken 16722 times.
✓ Branch 4 taken 14490 times.
31212 if (!s->eof && ff_inlink_queued_frames(inlink)) {
303 16722 AVFrame *frame = NULL;
304 int ret;
305
306 16722 ret = ff_inlink_consume_frame(inlink, &frame);
307
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16722 times.
16722 if (ret < 0)
308 16722 return ret;
309
1/2
✓ Branch 0 taken 16722 times.
✗ Branch 1 not taken.
16722 if (ret > 0)
310 16722 return s->filter_frame(inlink, frame);
311 }
312
313
2/2
✓ Branch 1 taken 13 times.
✓ Branch 2 taken 14477 times.
14490 FF_FILTER_FORWARD_STATUS(inlink, outlink);
314
2/2
✓ Branch 1 taken 14470 times.
✓ Branch 2 taken 7 times.
14477 FF_FILTER_FORWARD_WANTED(outlink, inlink);
315
316 7 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