2 * Copyright (c) 2013 Nicolas George
4 * This file is part of FFmpeg.
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public License
8 * as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
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
14 * GNU Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public License
17 * along with FFmpeg; if not, write to the Free Software Foundation, Inc.,
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 #include "libavutil/avassert.h"
23 #include "bufferqueue.h"
24 #include "framesync.h"
27 #define OFFSET(member) offsetof(FFFrameSync, member)
29 static const char *framesync_name(void *ptr)
34 static const AVClass framesync_class = {
35 .version = LIBAVUTIL_VERSION_INT,
36 .class_name = "framesync",
37 .item_name = framesync_name,
38 .category = AV_CLASS_CATEGORY_FILTER,
40 .parent_log_context_offset = OFFSET(parent),
49 int ff_framesync_init(FFFrameSync *fs, void *parent, unsigned nb_in)
51 fs->class = &framesync_class;
55 fs->in = av_calloc(nb_in, sizeof(*fs->in));
57 return AVERROR(ENOMEM);
61 static void framesync_sync_level_update(FFFrameSync *fs)
63 unsigned i, level = 0;
65 for (i = 0; i < fs->nb_in; i++)
66 if (fs->in[i].state != STATE_EOF)
67 level = FFMAX(level, fs->in[i].sync);
68 av_assert0(level <= fs->sync_level);
69 if (level < fs->sync_level)
70 av_log(fs, AV_LOG_VERBOSE, "Sync level %u\n", level);
72 fs->sync_level = level;
77 int ff_framesync_configure(FFFrameSync *fs)
82 if (!fs->time_base.num) {
83 for (i = 0; i < fs->nb_in; i++) {
85 if (fs->time_base.num) {
86 gcd = av_gcd(fs->time_base.den, fs->in[i].time_base.den);
87 lcm = (fs->time_base.den / gcd) * fs->in[i].time_base.den;
88 if (lcm < AV_TIME_BASE / 2) {
89 fs->time_base.den = lcm;
90 fs->time_base.num = av_gcd(fs->time_base.num,
91 fs->in[i].time_base.num);
93 fs->time_base.num = 1;
94 fs->time_base.den = AV_TIME_BASE;
98 fs->time_base = fs->in[i].time_base;
102 if (!fs->time_base.num) {
103 av_log(fs, AV_LOG_ERROR, "Impossible to set time base\n");
104 return AVERROR(EINVAL);
106 av_log(fs, AV_LOG_VERBOSE, "Selected %d/%d time base\n",
107 fs->time_base.num, fs->time_base.den);
110 for (i = 0; i < fs->nb_in; i++)
111 fs->in[i].pts = fs->in[i].pts_next = AV_NOPTS_VALUE;
112 fs->sync_level = UINT_MAX;
113 framesync_sync_level_update(fs);
118 static void framesync_advance(FFFrameSync *fs)
126 while (!fs->frame_ready) {
128 for (i = 0; i < fs->nb_in; i++) {
129 if (!fs->in[i].have_next) {
130 if (latest < 0 || fs->in[i].pts < fs->in[latest].pts)
135 fs->in_request = latest;
139 pts = fs->in[0].pts_next;
140 for (i = 1; i < fs->nb_in; i++)
141 if (fs->in[i].pts_next < pts)
142 pts = fs->in[i].pts_next;
143 if (pts == INT64_MAX) {
147 for (i = 0; i < fs->nb_in; i++) {
148 if (fs->in[i].pts_next == pts ||
149 (fs->in[i].before == EXT_INFINITY &&
150 fs->in[i].state == STATE_BOF)) {
151 av_frame_free(&fs->in[i].frame);
152 fs->in[i].frame = fs->in[i].frame_next;
153 fs->in[i].pts = fs->in[i].pts_next;
154 fs->in[i].frame_next = NULL;
155 fs->in[i].pts_next = AV_NOPTS_VALUE;
156 fs->in[i].have_next = 0;
157 fs->in[i].state = fs->in[i].frame ? STATE_RUN : STATE_EOF;
158 if (fs->in[i].sync == fs->sync_level && fs->in[i].frame)
160 if (fs->in[i].state == STATE_EOF &&
161 fs->in[i].after == EXT_STOP)
168 for (i = 0; i < fs->nb_in; i++)
169 if ((fs->in[i].state == STATE_BOF &&
170 fs->in[i].before == EXT_STOP))
176 static int64_t framesync_pts_extrapolate(FFFrameSync *fs, unsigned in,
179 /* Possible enhancement: use the link's frame rate */
183 static void framesync_inject_frame(FFFrameSync *fs, unsigned in, AVFrame *frame)
187 av_assert0(!fs->in[in].have_next);
189 pts = av_rescale_q(frame->pts, fs->in[in].time_base, fs->time_base);
192 pts = fs->in[in].state != STATE_RUN || fs->in[in].after == EXT_INFINITY
193 ? INT64_MAX : framesync_pts_extrapolate(fs, in, fs->in[in].pts);
195 framesync_sync_level_update(fs);
197 fs->in[in].frame_next = frame;
198 fs->in[in].pts_next = pts;
199 fs->in[in].have_next = 1;
202 int ff_framesync_add_frame(FFFrameSync *fs, unsigned in, AVFrame *frame)
204 av_assert1(in < fs->nb_in);
205 if (!fs->in[in].have_next)
206 framesync_inject_frame(fs, in, frame);
208 ff_bufqueue_add(fs, &fs->in[in].queue, frame);
212 void ff_framesync_next(FFFrameSync *fs)
216 av_assert0(!fs->frame_ready);
217 for (i = 0; i < fs->nb_in; i++)
218 if (!fs->in[i].have_next && fs->in[i].queue.available)
219 framesync_inject_frame(fs, i, ff_bufqueue_get(&fs->in[i].queue));
221 framesync_advance(fs);
224 void ff_framesync_drop(FFFrameSync *fs)
229 int ff_framesync_get_frame(FFFrameSync *fs, unsigned in, AVFrame **rframe,
233 unsigned need_copy = 0, i;
237 if (!fs->in[in].frame) {
241 frame = fs->in[in].frame;
243 /* Find out if we need to copy the frame: is there another sync
244 stream, and do we know if its current frame will outlast this one? */
245 pts_next = fs->in[in].have_next ? fs->in[in].pts_next : INT64_MAX;
246 for (i = 0; i < fs->nb_in && !need_copy; i++)
247 if (i != in && fs->in[i].sync &&
248 (!fs->in[i].have_next || fs->in[i].pts_next < pts_next))
251 if (!(frame = av_frame_clone(frame)))
252 return AVERROR(ENOMEM);
253 if ((ret = av_frame_make_writable(frame)) < 0) {
254 av_frame_free(&frame);
258 fs->in[in].frame = NULL;
266 void ff_framesync_uninit(FFFrameSync *fs)
270 for (i = 0; i < fs->nb_in; i++) {
271 av_frame_free(&fs->in[i].frame);
272 av_frame_free(&fs->in[i].frame_next);
273 ff_bufqueue_discard_all(&fs->in[i].queue);
279 int ff_framesync_process_frame(FFFrameSync *fs, unsigned all)
283 av_assert0(fs->on_event);
285 ff_framesync_next(fs);
286 if (fs->eof || !fs->frame_ready)
288 if ((ret = fs->on_event(fs)) < 0)
290 ff_framesync_drop(fs);
295 if (!count && fs->eof)
300 int ff_framesync_filter_frame(FFFrameSync *fs, AVFilterLink *inlink,
305 if ((ret = ff_framesync_process_frame(fs, 1)) < 0)
307 if ((ret = ff_framesync_add_frame(fs, FF_INLINK_IDX(inlink), in)) < 0)
309 if ((ret = ff_framesync_process_frame(fs, 0)) < 0)
314 int ff_framesync_request_frame(FFFrameSync *fs, AVFilterLink *outlink)
316 AVFilterContext *ctx = outlink->src;
319 if ((ret = ff_framesync_process_frame(fs, 0)) < 0)
325 input = fs->in_request;
326 ret = ff_request_frame(ctx->inputs[input]);
327 if (ret == AVERROR_EOF) {
328 if ((ret = ff_framesync_add_frame(fs, input, NULL)) < 0)
330 if ((ret = ff_framesync_process_frame(fs, 0)) < 0)