har0ke/FFmpeg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

lavfi/testsrc: extend logic in request_frame, support static image output

Browse Source
This commit is contained in:
Stefano Sabatini 2012-07-31 21:14:49 +02:00
parent 20e940e768
commit 33474eb1c6
1 changed files with 32 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
libavfilter/vsrc_testsrc.c
View File

@ -50,6 +50,8 @@ typedef struct {
char *duration; ///< total duration of the generated video char *duration; ///< total duration of the generated video
AVRational sar; ///< sample aspect ratio AVRational sar; ///< sample aspect ratio
int nb_decimals; int nb_decimals;
int draw_once; ///< draw only the first frame, always put out the same picture
AVFilterBufferRef *picref; ///< cached reference containing the painted picture
void (* fill_picture_fn)(AVFilterContext *ctx, AVFilterBufferRef *picref); void (* fill_picture_fn)(AVFilterContext *ctx, AVFilterBufferRef *picref);
@ -123,6 +125,7 @@ static av_cold void uninit(AVFilterContext *ctx)
TestSourceContext *test = ctx->priv; TestSourceContext *test = ctx->priv;
av_opt_free(test); av_opt_free(test);
avfilter_unref_bufferp(&test->picref);
} }
static int config_props(AVFilterLink *outlink) static int config_props(AVFilterLink *outlink)
@ -140,25 +143,40 @@ static int config_props(AVFilterLink *outlink)
static int request_frame(AVFilterLink *outlink) static int request_frame(AVFilterLink *outlink)
{ {
TestSourceContext *test = outlink->src->priv; TestSourceContext *test = outlink->src->priv;
AVFilterBufferRef *picref; AVFilterBufferRef *outpicref;
int ret; int ret = 0;
if (test->max_pts >= 0 && test->pts >= test->max_pts) if (test->max_pts >= 0 && test->pts >= test->max_pts)
return AVERROR_EOF; return AVERROR_EOF;
picref = ff_get_video_buffer(outlink, AV_PERM_WRITE, test->w, test->h);
if (!picref)
return AVERROR(ENOMEM);
picref->pts = test->pts++; if (test->draw_once) {
picref->pos = -1; if (!test->picref) {
picref->video->key_frame = 1; test->picref =
picref->video->interlaced = 0; ff_get_video_buffer(outlink, AV_PERM_WRITE|AV_PERM_PRESERVE|AV_PERM_REUSE,
picref->video->pict_type = AV_PICTURE_TYPE_I; test->w, test->h);
picref->video->sample_aspect_ratio = test->sar; if (!test->picref)
test->fill_picture_fn(outlink->src, picref); return AVERROR(ENOMEM);
test->fill_picture_fn(outlink->src, test->picref);
}
outpicref = avfilter_ref_buffer(test->picref, ~AV_PERM_WRITE);
} else
outpicref = ff_get_video_buffer(outlink, AV_PERM_WRITE, test->w, test->h);
if (!outpicref)
return AVERROR(ENOMEM);
outpicref->pts = test->pts;
outpicref->pos = -1;
outpicref->video->key_frame = 1;
outpicref->video->interlaced = 0;
outpicref->video->pict_type = AV_PICTURE_TYPE_I;
outpicref->video->sample_aspect_ratio = test->sar;
if (!test->draw_once)
test->fill_picture_fn(outlink->src, outpicref);
test->pts++;
test->nb_frame++; test->nb_frame++;
if ((ret = ff_start_frame(outlink, picref)) < 0 || if ((ret = ff_start_frame(outlink, outpicref)) < 0 ||
(ret = ff_draw_slice(outlink, 0, test->h, 1)) < 0 || (ret = ff_draw_slice(outlink, 0, test->h, 1)) < 0 ||
(ret = ff_end_frame(outlink)) < 0) (ret = ff_end_frame(outlink)) < 0)
return ret; return ret;
@ -493,6 +511,7 @@ static av_cold int rgbtest_init(AVFilterContext *ctx, const char *args)
{ {
TestSourceContext *test = ctx->priv; TestSourceContext *test = ctx->priv;
test->draw_once = 1;
test->class = &rgbtestsrc_class; test->class = &rgbtestsrc_class;
test->fill_picture_fn = rgbtest_fill_picture; test->fill_picture_fn = rgbtest_fill_picture;
return init(ctx, args); return init(ctx, args);