vf_showinfo: switch to filter_frame
[ffmpeg.git] / libavfilter / vf_showinfo.c
1 /*
2  * Copyright (c) 2011 Stefano Sabatini
3  * This file is part of FFmpeg.
4  *
5  * FFmpeg is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2.1 of the License, or (at your option) any later version.
9  *
10  * FFmpeg is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with FFmpeg; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18  */
19
20 /**
21  * @file
22  * filter for showing textual video frame information
23  */
24
25 #include "libavutil/adler32.h"
26 #include "libavutil/imgutils.h"
27 #include "libavutil/internal.h"
28 #include "libavutil/pixdesc.h"
29 #include "libavutil/timestamp.h"
30 #include "avfilter.h"
31 #include "internal.h"
32 #include "video.h"
33
34 typedef struct {
35     unsigned int frame;
36 } ShowInfoContext;
37
38 static av_cold int init(AVFilterContext *ctx, const char *args)
39 {
40     ShowInfoContext *showinfo = ctx->priv;
41     showinfo->frame = 0;
42     return 0;
43 }
44
45 static int filter_frame(AVFilterLink *inlink, AVFilterBufferRef *frame)
46 {
47     AVFilterContext *ctx = inlink->dst;
48     ShowInfoContext *showinfo = ctx->priv;
49     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format);
50     uint32_t plane_checksum[4] = {0}, checksum = 0;
51     int i, plane, vsub = desc->log2_chroma_h;
52
53     for (plane = 0; frame->data[plane] && plane < 4; plane++) {
54         int64_t linesize = av_image_get_linesize(frame->format, frame->video->w, plane);
55         uint8_t *data = frame->data[plane];
56         int h = plane == 1 || plane == 2 ? inlink->h >> vsub : inlink->h;
57
58         if (linesize < 0)
59             return linesize;
60
61         for (i = 0; i < h; i++) {
62             plane_checksum[plane] = av_adler32_update(plane_checksum[plane], data, linesize);
63             checksum = av_adler32_update(checksum, data, linesize);
64             data += frame->linesize[plane];
65         }
66     }
67
68     av_log(ctx, AV_LOG_INFO,
69            "n:%d pts:%s pts_time:%s pos:%"PRId64" "
70            "fmt:%s sar:%d/%d s:%dx%d i:%c iskey:%d type:%c "
71            "checksum:%08X plane_checksum:[%08X",
72            showinfo->frame,
73            av_ts2str(frame->pts), av_ts2timestr(frame->pts, &inlink->time_base), frame->pos,
74            desc->name,
75            frame->video->sample_aspect_ratio.num, frame->video->sample_aspect_ratio.den,
76            frame->video->w, frame->video->h,
77            !frame->video->interlaced     ? 'P' :         /* Progressive  */
78            frame->video->top_field_first ? 'T' : 'B',    /* Top / Bottom */
79            frame->video->key_frame,
80            av_get_picture_type_char(frame->video->pict_type),
81            checksum, plane_checksum[0]);
82
83     for (plane = 1; frame->data[plane] && plane < 4; plane++)
84         av_log(ctx, AV_LOG_INFO, " %08X", plane_checksum[plane]);
85     av_log(ctx, AV_LOG_INFO, "]\n");
86
87     showinfo->frame++;
88     return ff_filter_frame(inlink->dst->outputs[0], frame);
89 }
90
91 static const AVFilterPad avfilter_vf_showinfo_inputs[] = {
92     {
93         .name             = "default",
94         .type             = AVMEDIA_TYPE_VIDEO,
95         .get_video_buffer = ff_null_get_video_buffer,
96         .filter_frame     = filter_frame,
97         .min_perms        = AV_PERM_READ,
98     },
99     { NULL }
100 };
101
102 static const AVFilterPad avfilter_vf_showinfo_outputs[] = {
103     {
104         .name = "default",
105         .type = AVMEDIA_TYPE_VIDEO
106     },
107     { NULL }
108 };
109
110 AVFilter avfilter_vf_showinfo = {
111     .name        = "showinfo",
112     .description = NULL_IF_CONFIG_SMALL("Show textual information for each video frame."),
113
114     .priv_size = sizeof(ShowInfoContext),
115     .init      = init,
116
117     .inputs    = avfilter_vf_showinfo_inputs,
118
119     .outputs   = avfilter_vf_showinfo_outputs,
120 };