1 /*
2 * Copyright (C) 2017 Paul B Mahol
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 #include "libavutil/common.h"
22 #include "libavutil/opt.h"
23 #include "internal.h"
24
25 typedef struct VFRDETContext {
26 const AVClass *class;
27
28 int64_t prev_pts;
29 int64_t delta;
30 int64_t min_delta;
31 int64_t max_delta;
32 int64_t avg_delta;
33
34 uint64_t vfr;
35 uint64_t cfr;
36 } VFRDETContext;
37
filter_frame(AVFilterLink * inlink,AVFrame * in)38 static int filter_frame(AVFilterLink *inlink, AVFrame *in)
39 {
40 AVFilterContext *ctx = inlink->dst;
41 VFRDETContext *s = ctx->priv;
42
43 if (s->prev_pts != AV_NOPTS_VALUE) {
44 int64_t delta = in->pts - s->prev_pts;
45
46 if (s->delta == AV_NOPTS_VALUE) {
47 s->delta = delta;
48 s->min_delta = delta;
49 s->max_delta = delta;
50 }
51
52 if (s->delta != delta) {
53 s->vfr++;
54 s->delta = delta;
55 s->min_delta = FFMIN(delta, s->min_delta);
56 s->max_delta = FFMAX(delta, s->max_delta);
57 s->avg_delta += delta;
58 } else {
59 s->cfr++;
60 }
61 }
62
63 s->prev_pts = in->pts;
64
65 return ff_filter_frame(ctx->outputs[0], in);
66 }
67
init(AVFilterContext * ctx)68 static av_cold int init(AVFilterContext *ctx)
69 {
70 VFRDETContext *s = ctx->priv;
71
72 s->prev_pts = AV_NOPTS_VALUE;
73 s->delta = AV_NOPTS_VALUE;
74 s->min_delta = INT64_MAX;
75 s->max_delta = INT64_MIN;
76
77 return 0;
78 }
79
uninit(AVFilterContext * ctx)80 static av_cold void uninit(AVFilterContext *ctx)
81 {
82 VFRDETContext *s = ctx->priv;
83
84 av_log(ctx, AV_LOG_INFO, "VFR:%f (%"PRIu64"/%"PRIu64")", s->vfr / (float)(s->vfr + s->cfr), s->vfr, s->cfr);
85 if (s->vfr)
86 av_log(ctx, AV_LOG_INFO, " min: %"PRId64" max: %"PRId64" avg: %"PRId64, s->min_delta, s->max_delta, s->avg_delta / s->vfr);
87 av_log(ctx, AV_LOG_INFO, "\n");
88 }
89
90 static const AVFilterPad vfrdet_inputs[] = {
91 {
92 .name = "default",
93 .type = AVMEDIA_TYPE_VIDEO,
94 .filter_frame = filter_frame,
95 },
96 { NULL }
97 };
98
99 static const AVFilterPad vfrdet_outputs[] = {
100 {
101 .name = "default",
102 .type = AVMEDIA_TYPE_VIDEO,
103 },
104 { NULL }
105 };
106
107 AVFilter ff_vf_vfrdet = {
108 .name = "vfrdet",
109 .description = NULL_IF_CONFIG_SMALL("Variable frame rate detect filter."),
110 .priv_size = sizeof(VFRDETContext),
111 .init = init,
112 .uninit = uninit,
113 .inputs = vfrdet_inputs,
114 .outputs = vfrdet_outputs,
115 };
116