mirror of https://github.com/FFmpeg/FFmpeg.git
Also add the public interface libavfilter/buffersink.h. Based on a commit by Stefano Sabatini.pull/30/merge
parent
ab165047a6
commit
ac71230902
10 changed files with 267 additions and 122 deletions
@ -0,0 +1,114 @@ |
|||||||
|
/*
|
||||||
|
* Copyright (c) 2011 Stefano Sabatini |
||||||
|
* |
||||||
|
* This file is part of Libav. |
||||||
|
* |
||||||
|
* Libav 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. |
||||||
|
* |
||||||
|
* Libav 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 Libav; if not, write to the Free Software |
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||||
|
*/ |
||||||
|
|
||||||
|
/**
|
||||||
|
* @file |
||||||
|
* buffer sink |
||||||
|
*/ |
||||||
|
|
||||||
|
#include "libavutil/fifo.h" |
||||||
|
|
||||||
|
#include "avfilter.h" |
||||||
|
#include "buffersink.h" |
||||||
|
|
||||||
|
typedef struct { |
||||||
|
AVFifoBuffer *fifo; ///< FIFO buffer of video frame references
|
||||||
|
} BufferSinkContext; |
||||||
|
|
||||||
|
#define FIFO_INIT_SIZE 8 |
||||||
|
|
||||||
|
static av_cold void uninit(AVFilterContext *ctx) |
||||||
|
{ |
||||||
|
BufferSinkContext *sink = ctx->priv; |
||||||
|
|
||||||
|
while (sink->fifo && av_fifo_size(sink->fifo)) { |
||||||
|
AVFilterBufferRef *buf; |
||||||
|
av_fifo_generic_read(sink->fifo, &buf, sizeof(buf), NULL); |
||||||
|
avfilter_unref_buffer(buf); |
||||||
|
} |
||||||
|
av_fifo_free(sink->fifo); |
||||||
|
} |
||||||
|
|
||||||
|
static av_cold int init(AVFilterContext *ctx, const char *args, void *opaque) |
||||||
|
{ |
||||||
|
BufferSinkContext *sink = ctx->priv; |
||||||
|
|
||||||
|
if (!(sink->fifo = av_fifo_alloc(FIFO_INIT_SIZE*sizeof(AVFilterBufferRef*)))) { |
||||||
|
av_log(ctx, AV_LOG_ERROR, "Failed to allocate fifo\n"); |
||||||
|
return AVERROR(ENOMEM); |
||||||
|
} |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static void end_frame(AVFilterLink *link) |
||||||
|
{ |
||||||
|
AVFilterContext *ctx = link->dst; |
||||||
|
BufferSinkContext *sink = ctx->priv; |
||||||
|
|
||||||
|
if (av_fifo_space(sink->fifo) < sizeof(AVFilterBufferRef *) && |
||||||
|
(av_fifo_realloc2(sink->fifo, av_fifo_size(sink->fifo) * 2) < 0)) { |
||||||
|
av_log(ctx, AV_LOG_ERROR, "Error reallocating the FIFO.\n"); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
av_fifo_generic_write(sink->fifo, &link->cur_buf, sizeof(link->cur_buf), NULL); |
||||||
|
link->cur_buf = NULL; |
||||||
|
} |
||||||
|
|
||||||
|
int av_buffersink_read(AVFilterContext *ctx, AVFilterBufferRef **buf) |
||||||
|
{ |
||||||
|
BufferSinkContext *sink = ctx->priv; |
||||||
|
AVFilterLink *link = ctx->inputs[0]; |
||||||
|
int ret; |
||||||
|
|
||||||
|
if (!buf) { |
||||||
|
if (av_fifo_size(sink->fifo)) |
||||||
|
return av_fifo_size(sink->fifo)/sizeof(*buf); |
||||||
|
else |
||||||
|
return avfilter_poll_frame(ctx->inputs[0]); |
||||||
|
} |
||||||
|
|
||||||
|
if (!av_fifo_size(sink->fifo) && |
||||||
|
(ret = avfilter_request_frame(link)) < 0) |
||||||
|
return ret; |
||||||
|
|
||||||
|
if (!av_fifo_size(sink->fifo)) |
||||||
|
return AVERROR(EINVAL); |
||||||
|
|
||||||
|
av_fifo_generic_read(sink->fifo, buf, sizeof(*buf), NULL); |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
AVFilter avfilter_vsink_buffer = { |
||||||
|
.name = "buffersink", |
||||||
|
.description = NULL_IF_CONFIG_SMALL("Buffer video frames, and make them available to the end of the filter graph."), |
||||||
|
.priv_size = sizeof(BufferSinkContext), |
||||||
|
.init = init, |
||||||
|
.uninit = uninit, |
||||||
|
|
||||||
|
.inputs = (AVFilterPad[]) {{ .name = "default", |
||||||
|
.type = AVMEDIA_TYPE_VIDEO, |
||||||
|
.end_frame = end_frame, |
||||||
|
.min_perms = AV_PERM_READ, }, |
||||||
|
{ .name = NULL }}, |
||||||
|
.outputs = (AVFilterPad[]) {{ .name = NULL }}, |
||||||
|
}; |
@ -0,0 +1,43 @@ |
|||||||
|
/*
|
||||||
|
* This file is part of Libav. |
||||||
|
* |
||||||
|
* Libav 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. |
||||||
|
* |
||||||
|
* Libav 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 Libav; if not, write to the Free Software |
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef AVFILTER_BUFFERSINK_H |
||||||
|
#define AVFILTER_BUFFERSINK_H |
||||||
|
|
||||||
|
/**
|
||||||
|
* @file |
||||||
|
* memory buffer sink API |
||||||
|
*/ |
||||||
|
|
||||||
|
#include "avfilter.h" |
||||||
|
|
||||||
|
/**
|
||||||
|
* Get a buffer with filtered data from sink and put it in buf. |
||||||
|
* |
||||||
|
* @param sink pointer to a context of a buffersink AVFilter. |
||||||
|
* @param buf pointer to the buffer will be written here if buf is non-NULL. buf |
||||||
|
* must be freed by the caller using avfilter_unref_buffer(). |
||||||
|
* Buf may also be NULL to query whether a buffer is ready to be |
||||||
|
* output. |
||||||
|
* |
||||||
|
* @return >= 0 in case of success, a negative AVERROR code in case of |
||||||
|
* failure. |
||||||
|
*/ |
||||||
|
int av_buffersink_read(AVFilterContext *sink, AVFilterBufferRef **buf); |
||||||
|
|
||||||
|
#endif /* AVFILTER_BUFFERSINK_H */ |
Loading…
Reference in new issue