fftools/ffmpeg_filter: add filtergraph private data

Start by moving OutputStream.filtered_frame to it, which really belongs
to the filtergraph rather than the output stream.
This commit is contained in:
Anton Khirnov 2023-04-18 19:44:15 +02:00
parent aee5b774bf
commit 09dd9c5f41
4 changed files with 26 additions and 11 deletions

View File

@ -604,7 +604,6 @@ typedef struct OutputStream {
Encoder *enc;
AVCodecContext *enc_ctx;
AVFrame *filtered_frame;
AVPacket *pkt;
int64_t last_dropped;

View File

@ -38,6 +38,18 @@
#include "libavutil/samplefmt.h"
#include "libavutil/timestamp.h"
typedef struct FilterGraphPriv {
FilterGraph fg;
// frame for temporarily holding output from the filtergraph
AVFrame *frame;
} FilterGraphPriv;
static FilterGraphPriv *fgp_from_fg(FilterGraph *fg)
{
return (FilterGraphPriv*)fg;
}
// FIXME: YUV420P etc. are actually supported with full color range,
// yet the latter information isn't available here.
static const enum AVPixelFormat *get_compliance_normal_pix_fmts(const AVCodec *codec, const enum AVPixelFormat default_formats[])
@ -192,9 +204,11 @@ static OutputFilter *ofilter_alloc(FilterGraph *fg)
void fg_free(FilterGraph **pfg)
{
FilterGraph *fg = *pfg;
FilterGraphPriv *fgp;
if (!fg)
return;
fgp = fgp_from_fg(fg);
avfilter_graph_free(&fg->graph);
for (int j = 0; j < fg->nb_inputs; j++) {
@ -230,17 +244,23 @@ void fg_free(FilterGraph **pfg)
av_freep(&fg->outputs);
av_freep(&fg->graph_desc);
av_frame_free(&fgp->frame);
av_freep(pfg);
}
FilterGraph *fg_create(char *graph_desc)
{
FilterGraph *fg;
FilterGraphPriv *fgp = allocate_array_elem(&filtergraphs, sizeof(*fgp), &nb_filtergraphs);
FilterGraph *fg = &fgp->fg;
fg = ALLOC_ARRAY_ELEM(filtergraphs, nb_filtergraphs);
fg->index = nb_filtergraphs - 1;
fg->graph_desc = graph_desc;
fgp->frame = av_frame_alloc();
if (!fgp->frame)
report_and_exit(AVERROR(ENOMEM));
return fg;
}
@ -1350,18 +1370,19 @@ int filtergraph_is_simple(FilterGraph *fg)
int reap_filters(int flush)
{
AVFrame *filtered_frame = NULL;
/* Reap all buffers present in the buffer sinks */
for (OutputStream *ost = ost_iter(NULL); ost; ost = ost_iter(ost)) {
FilterGraphPriv *fgp;
AVFrame *filtered_frame;
AVFilterContext *filter;
int ret = 0;
if (!ost->filter || !ost->filter->graph->graph)
continue;
filter = ost->filter->filter;
fgp = fgp_from_fg(ost->filter->graph);
filtered_frame = ost->filtered_frame;
filtered_frame = fgp->frame;
while (1) {
ret = av_buffersink_get_frame_flags(filter, filtered_frame,

View File

@ -845,7 +845,6 @@ static void ost_free(OutputStream **post)
av_bsf_free(&ms->bsf_ctx);
av_frame_free(&ost->filtered_frame);
av_packet_free(&ost->pkt);
av_dict_free(&ost->encoder_opts);

View File

@ -1026,10 +1026,6 @@ static OutputStream *ost_add(Muxer *mux, const OptionsContext *o,
av_strlcat(ms->log_name, "/copy", sizeof(ms->log_name));
}
ost->filtered_frame = av_frame_alloc();
if (!ost->filtered_frame)
report_and_exit(AVERROR(ENOMEM));
ost->pkt = av_packet_alloc();
if (!ost->pkt)
report_and_exit(AVERROR(ENOMEM));