mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-12-28 20:53:54 +02:00
4e0e9ce2dc
With this helper API, filters that used the first framesync helper API can easily be changed to use the new and more extensible design for filters with a single activate() callback.
314 lines
9.4 KiB
C
314 lines
9.4 KiB
C
/*
|
|
* Copyright (c) 2013 Nicolas George
|
|
*
|
|
* This file is part of FFmpeg.
|
|
*
|
|
* FFmpeg is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public License
|
|
* as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* FFmpeg is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public License
|
|
* along with FFmpeg; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include "libavutil/avassert.h"
|
|
#include "avfilter.h"
|
|
#include "filters.h"
|
|
#include "framesync2.h"
|
|
#include "internal.h"
|
|
|
|
#define OFFSET(member) offsetof(FFFrameSync, member)
|
|
|
|
static const char *framesync_name(void *ptr)
|
|
{
|
|
return "framesync";
|
|
}
|
|
|
|
static const AVClass framesync_class = {
|
|
.version = LIBAVUTIL_VERSION_INT,
|
|
.class_name = "framesync",
|
|
.item_name = framesync_name,
|
|
.category = AV_CLASS_CATEGORY_FILTER,
|
|
.option = NULL,
|
|
.parent_log_context_offset = OFFSET(parent),
|
|
};
|
|
|
|
enum {
|
|
STATE_BOF,
|
|
STATE_RUN,
|
|
STATE_EOF,
|
|
};
|
|
|
|
int ff_framesync2_init(FFFrameSync *fs, AVFilterContext *parent, unsigned nb_in)
|
|
{
|
|
/* For filters with several outputs, we will not be able to assume which
|
|
output is relevant for ff_outlink_frame_wanted() and
|
|
ff_outlink_set_status(). To be designed when needed. */
|
|
av_assert0(parent->nb_outputs == 1);
|
|
|
|
fs->class = &framesync_class;
|
|
fs->parent = parent;
|
|
fs->nb_in = nb_in;
|
|
|
|
fs->in = av_calloc(nb_in, sizeof(*fs->in));
|
|
if (!fs->in)
|
|
return AVERROR(ENOMEM);
|
|
return 0;
|
|
}
|
|
|
|
static void framesync_eof(FFFrameSync *fs)
|
|
{
|
|
fs->eof = 1;
|
|
fs->frame_ready = 0;
|
|
ff_outlink_set_status(fs->parent->outputs[0], AVERROR_EOF, AV_NOPTS_VALUE);
|
|
}
|
|
|
|
static void framesync_sync_level_update(FFFrameSync *fs)
|
|
{
|
|
unsigned i, level = 0;
|
|
|
|
for (i = 0; i < fs->nb_in; i++)
|
|
if (fs->in[i].state != STATE_EOF)
|
|
level = FFMAX(level, fs->in[i].sync);
|
|
av_assert0(level <= fs->sync_level);
|
|
if (level < fs->sync_level)
|
|
av_log(fs, AV_LOG_VERBOSE, "Sync level %u\n", level);
|
|
if (level)
|
|
fs->sync_level = level;
|
|
else
|
|
framesync_eof(fs);
|
|
}
|
|
|
|
int ff_framesync2_configure(FFFrameSync *fs)
|
|
{
|
|
unsigned i;
|
|
int64_t gcd, lcm;
|
|
|
|
if (!fs->time_base.num) {
|
|
for (i = 0; i < fs->nb_in; i++) {
|
|
if (fs->in[i].sync) {
|
|
if (fs->time_base.num) {
|
|
gcd = av_gcd(fs->time_base.den, fs->in[i].time_base.den);
|
|
lcm = (fs->time_base.den / gcd) * fs->in[i].time_base.den;
|
|
if (lcm < AV_TIME_BASE / 2) {
|
|
fs->time_base.den = lcm;
|
|
fs->time_base.num = av_gcd(fs->time_base.num,
|
|
fs->in[i].time_base.num);
|
|
} else {
|
|
fs->time_base.num = 1;
|
|
fs->time_base.den = AV_TIME_BASE;
|
|
break;
|
|
}
|
|
} else {
|
|
fs->time_base = fs->in[i].time_base;
|
|
}
|
|
}
|
|
}
|
|
if (!fs->time_base.num) {
|
|
av_log(fs, AV_LOG_ERROR, "Impossible to set time base\n");
|
|
return AVERROR(EINVAL);
|
|
}
|
|
av_log(fs, AV_LOG_VERBOSE, "Selected %d/%d time base\n",
|
|
fs->time_base.num, fs->time_base.den);
|
|
}
|
|
|
|
for (i = 0; i < fs->nb_in; i++)
|
|
fs->in[i].pts = fs->in[i].pts_next = AV_NOPTS_VALUE;
|
|
fs->sync_level = UINT_MAX;
|
|
framesync_sync_level_update(fs);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void framesync_advance(FFFrameSync *fs)
|
|
{
|
|
int latest;
|
|
unsigned i;
|
|
int64_t pts;
|
|
|
|
if (fs->eof)
|
|
return;
|
|
while (!fs->frame_ready) {
|
|
latest = -1;
|
|
for (i = 0; i < fs->nb_in; i++) {
|
|
if (!fs->in[i].have_next) {
|
|
if (latest < 0 || fs->in[i].pts < fs->in[latest].pts)
|
|
latest = i;
|
|
}
|
|
}
|
|
if (latest >= 0) {
|
|
fs->in_request = latest;
|
|
break;
|
|
}
|
|
|
|
pts = fs->in[0].pts_next;
|
|
for (i = 1; i < fs->nb_in; i++)
|
|
if (fs->in[i].pts_next < pts)
|
|
pts = fs->in[i].pts_next;
|
|
if (pts == INT64_MAX) {
|
|
framesync_eof(fs);
|
|
break;
|
|
}
|
|
for (i = 0; i < fs->nb_in; i++) {
|
|
if (fs->in[i].pts_next == pts ||
|
|
(fs->in[i].before == EXT_INFINITY &&
|
|
fs->in[i].state == STATE_BOF)) {
|
|
av_frame_free(&fs->in[i].frame);
|
|
fs->in[i].frame = fs->in[i].frame_next;
|
|
fs->in[i].pts = fs->in[i].pts_next;
|
|
fs->in[i].frame_next = NULL;
|
|
fs->in[i].pts_next = AV_NOPTS_VALUE;
|
|
fs->in[i].have_next = 0;
|
|
fs->in[i].state = fs->in[i].frame ? STATE_RUN : STATE_EOF;
|
|
if (fs->in[i].sync == fs->sync_level && fs->in[i].frame)
|
|
fs->frame_ready = 1;
|
|
if (fs->in[i].state == STATE_EOF &&
|
|
fs->in[i].after == EXT_STOP)
|
|
framesync_eof(fs);
|
|
}
|
|
}
|
|
if (fs->frame_ready)
|
|
for (i = 0; i < fs->nb_in; i++)
|
|
if ((fs->in[i].state == STATE_BOF &&
|
|
fs->in[i].before == EXT_STOP))
|
|
fs->frame_ready = 0;
|
|
fs->pts = pts;
|
|
}
|
|
}
|
|
|
|
static int64_t framesync_pts_extrapolate(FFFrameSync *fs, unsigned in,
|
|
int64_t pts)
|
|
{
|
|
/* Possible enhancement: use the link's frame rate */
|
|
return pts + 1;
|
|
}
|
|
|
|
static void framesync_inject_frame(FFFrameSync *fs, unsigned in, AVFrame *frame)
|
|
{
|
|
int64_t pts;
|
|
|
|
av_assert0(!fs->in[in].have_next);
|
|
av_assert0(frame);
|
|
pts = av_rescale_q(frame->pts, fs->in[in].time_base, fs->time_base);
|
|
frame->pts = pts;
|
|
fs->in[in].frame_next = frame;
|
|
fs->in[in].pts_next = pts;
|
|
fs->in[in].have_next = 1;
|
|
}
|
|
|
|
static void framesync_inject_status(FFFrameSync *fs, unsigned in, int status, int64_t pts)
|
|
{
|
|
av_assert0(!fs->in[in].have_next);
|
|
pts = fs->in[in].state != STATE_RUN || fs->in[in].after == EXT_INFINITY
|
|
? INT64_MAX : framesync_pts_extrapolate(fs, in, fs->in[in].pts);
|
|
fs->in[in].sync = 0;
|
|
framesync_sync_level_update(fs);
|
|
fs->in[in].frame_next = NULL;
|
|
fs->in[in].pts_next = pts;
|
|
fs->in[in].have_next = 1;
|
|
}
|
|
|
|
int ff_framesync2_get_frame(FFFrameSync *fs, unsigned in, AVFrame **rframe,
|
|
unsigned get)
|
|
{
|
|
AVFrame *frame;
|
|
unsigned need_copy = 0, i;
|
|
int64_t pts_next;
|
|
int ret;
|
|
|
|
if (!fs->in[in].frame) {
|
|
*rframe = NULL;
|
|
return 0;
|
|
}
|
|
frame = fs->in[in].frame;
|
|
if (get) {
|
|
/* Find out if we need to copy the frame: is there another sync
|
|
stream, and do we know if its current frame will outlast this one? */
|
|
pts_next = fs->in[in].have_next ? fs->in[in].pts_next : INT64_MAX;
|
|
for (i = 0; i < fs->nb_in && !need_copy; i++)
|
|
if (i != in && fs->in[i].sync &&
|
|
(!fs->in[i].have_next || fs->in[i].pts_next < pts_next))
|
|
need_copy = 1;
|
|
if (need_copy) {
|
|
if (!(frame = av_frame_clone(frame)))
|
|
return AVERROR(ENOMEM);
|
|
if ((ret = av_frame_make_writable(frame)) < 0) {
|
|
av_frame_free(&frame);
|
|
return ret;
|
|
}
|
|
} else {
|
|
fs->in[in].frame = NULL;
|
|
}
|
|
fs->frame_ready = 0;
|
|
}
|
|
*rframe = frame;
|
|
return 0;
|
|
}
|
|
|
|
void ff_framesync2_uninit(FFFrameSync *fs)
|
|
{
|
|
unsigned i;
|
|
|
|
for (i = 0; i < fs->nb_in; i++) {
|
|
av_frame_free(&fs->in[i].frame);
|
|
av_frame_free(&fs->in[i].frame_next);
|
|
}
|
|
|
|
av_freep(&fs->in);
|
|
}
|
|
|
|
int ff_framesync2_activate(FFFrameSync *fs)
|
|
{
|
|
AVFilterContext *ctx = fs->parent;
|
|
AVFrame *frame = NULL;
|
|
int64_t pts;
|
|
unsigned i, nb_active, nb_miss;
|
|
int ret, status;
|
|
|
|
nb_active = nb_miss = 0;
|
|
for (i = 0; i < fs->nb_in; i++) {
|
|
if (fs->in[i].have_next || fs->in[i].state == STATE_EOF)
|
|
continue;
|
|
nb_active++;
|
|
ret = ff_inlink_consume_frame(ctx->inputs[i], &frame);
|
|
if (ret < 0)
|
|
return ret;
|
|
if (ret) {
|
|
av_assert0(frame);
|
|
framesync_inject_frame(fs, i, frame);
|
|
} else {
|
|
ret = ff_inlink_acknowledge_status(ctx->inputs[i], &status, &pts);
|
|
if (ret > 0) {
|
|
framesync_inject_status(fs, i, status, pts);
|
|
} else if (!ret) {
|
|
nb_miss++;
|
|
}
|
|
}
|
|
}
|
|
if (nb_miss) {
|
|
if (nb_miss == nb_active && !ff_outlink_frame_wanted(ctx->outputs[0]))
|
|
return FFERROR_NOT_READY;
|
|
for (i = 0; i < fs->nb_in; i++)
|
|
if (!fs->in[i].have_next && fs->in[i].state != STATE_EOF)
|
|
ff_inlink_request_frame(ctx->inputs[i]);
|
|
return 0;
|
|
}
|
|
|
|
framesync_advance(fs);
|
|
if (fs->eof || !fs->frame_ready)
|
|
return 0;
|
|
ret = fs->on_event(fs);
|
|
if (ret < 0)
|
|
return ret;
|
|
fs->frame_ready = 0;
|
|
|
|
return 0;
|
|
}
|