vf_blackframe: switch to filter_frame

Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
Anton Khirnov 2012-11-29 00:32:08 +01:00 committed by Michael Niedermayer
parent 8f21cfc6b3
commit 0a767ad796

View File

@ -81,54 +81,42 @@ static av_cold int init(AVFilterContext *ctx, const char *args)
return 0; return 0;
} }
static int draw_slice(AVFilterLink *inlink, int y, int h, int slice_dir) static int filter_frame(AVFilterLink *inlink, AVFilterBufferRef *frame)
{ {
AVFilterContext *ctx = inlink->dst; AVFilterContext *ctx = inlink->dst;
BlackFrameContext *blackframe = ctx->priv; BlackFrameContext *blackframe = ctx->priv;
AVFilterBufferRef *picref = inlink->cur_buf;
int x, i; int x, i;
uint8_t *p = picref->data[0] + y * picref->linesize[0]; int pblack = 0;
uint8_t *p = frame->data[0];
for (i = 0; i < h; i++) { for (i = 0; i < frame->video->h; i++) {
for (x = 0; x < inlink->w; x++) for (x = 0; x < inlink->w; x++)
blackframe->nblack += p[x] < blackframe->bthresh; blackframe->nblack += p[x] < blackframe->bthresh;
p += picref->linesize[0]; p += frame->linesize[0];
} }
return ff_draw_slice(ctx->outputs[0], y, h, slice_dir); if (frame->video->key_frame)
}
static int end_frame(AVFilterLink *inlink)
{
AVFilterContext *ctx = inlink->dst;
BlackFrameContext *blackframe = ctx->priv;
AVFilterBufferRef *picref = inlink->cur_buf;
int pblack = 0;
if (picref->video->key_frame)
blackframe->last_keyframe = blackframe->frame; blackframe->last_keyframe = blackframe->frame;
pblack = blackframe->nblack * 100 / (inlink->w * inlink->h); pblack = blackframe->nblack * 100 / (inlink->w * inlink->h);
if (pblack >= blackframe->bamount) if (pblack >= blackframe->bamount)
av_log(ctx, AV_LOG_INFO, "frame:%u pblack:%u pos:%"PRId64" pts:%"PRId64" t:%f " av_log(ctx, AV_LOG_INFO, "frame:%u pblack:%u pos:%"PRId64" pts:%"PRId64" t:%f "
"type:%c last_keyframe:%d\n", "type:%c last_keyframe:%d\n",
blackframe->frame, pblack, picref->pos, picref->pts, blackframe->frame, pblack, frame->pos, frame->pts,
picref->pts == AV_NOPTS_VALUE ? -1 : picref->pts * av_q2d(inlink->time_base), frame->pts == AV_NOPTS_VALUE ? -1 : frame->pts * av_q2d(inlink->time_base),
av_get_picture_type_char(picref->video->pict_type), blackframe->last_keyframe); av_get_picture_type_char(frame->video->pict_type), blackframe->last_keyframe);
blackframe->frame++; blackframe->frame++;
blackframe->nblack = 0; blackframe->nblack = 0;
return ff_end_frame(inlink->dst->outputs[0]); return ff_filter_frame(inlink->dst->outputs[0], frame);
} }
static const AVFilterPad avfilter_vf_blackframe_inputs[] = { static const AVFilterPad avfilter_vf_blackframe_inputs[] = {
{ {
.name = "default", .name = "default",
.type = AVMEDIA_TYPE_VIDEO, .type = AVMEDIA_TYPE_VIDEO,
.draw_slice = draw_slice,
.get_video_buffer = ff_null_get_video_buffer, .get_video_buffer = ff_null_get_video_buffer,
.start_frame = ff_null_start_frame, .filter_frame = filter_frame,
.end_frame = end_frame,
}, },
{ NULL } { NULL }
}; };