lavfi: merge start_frame/draw_slice/end_frame
[ffmpeg.git] / libavfilter / vf_showinfo.c
1 /*
2  * Copyright (c) 2011 Stefano Sabatini
3  * This file is part of Libav.
4  *
5  * Libav 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  * Libav 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 Libav; 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 "avfilter.h"
30 #include "internal.h"
31 #include "video.h"
32
33 typedef struct {
34     unsigned int frame;
35 } ShowInfoContext;
36
37 static av_cold int init(AVFilterContext *ctx, const char *args)
38 {
39     ShowInfoContext *showinfo = ctx->priv;
40     showinfo->frame = 0;
41     return 0;
42 }
43
44 static int filter_frame(AVFilterLink *inlink, AVFilterBufferRef *frame)
45 {
46     AVFilterContext *ctx = inlink->dst;
47     ShowInfoContext *showinfo = ctx->priv;
48     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format);
49     uint32_t plane_checksum[4] = {0}, checksum = 0;
50     int i, plane, vsub = desc->log2_chroma_h;
51
52     for (plane = 0; frame->data[plane] && plane < 4; plane++) {
53         size_t linesize = av_image_get_linesize(frame->format, frame->video->w, plane);
54         uint8_t *data = frame->data[plane];
55         int h = plane == 1 || plane == 2 ? inlink->h >> vsub : inlink->h;
56
57         for (i = 0; i < h; i++) {
58             plane_checksum[plane] = av_adler32_update(plane_checksum[plane], data, linesize);
59             checksum = av_adler32_update(checksum, data, linesize);
60             data += frame->linesize[plane];
61         }
62     }
63
64     av_log(ctx, AV_LOG_INFO,
65            "n:%d pts:%"PRId64" pts_time:%f pos:%"PRId64" "
66            "fmt:%s sar:%d/%d s:%dx%d i:%c iskey:%d type:%c "
67            "checksum:%u plane_checksum:[%u %u %u %u]\n",
68            showinfo->frame,
69            frame->pts, frame->pts * av_q2d(inlink->time_base), frame->pos,
70            desc->name,
71            frame->video->pixel_aspect.num, frame->video->pixel_aspect.den,
72            frame->video->w, frame->video->h,
73            !frame->video->interlaced     ? 'P' :         /* Progressive  */
74            frame->video->top_field_first ? 'T' : 'B',    /* Top / Bottom */
75            frame->video->key_frame,
76            av_get_picture_type_char(frame->video->pict_type),
77            checksum, plane_checksum[0], plane_checksum[1], plane_checksum[2], plane_checksum[3]);
78
79     showinfo->frame++;
80     return ff_filter_frame(inlink->dst->outputs[0], frame);
81 }
82
83 static const AVFilterPad avfilter_vf_showinfo_inputs[] = {
84     {
85         .name             = "default",
86         .type             = AVMEDIA_TYPE_VIDEO,
87         .get_video_buffer = ff_null_get_video_buffer,
88         .filter_frame     = filter_frame,
89         .min_perms        = AV_PERM_READ,
90     },
91     { NULL }
92 };
93
94 static const AVFilterPad avfilter_vf_showinfo_outputs[] = {
95     {
96         .name = "default",
97         .type = AVMEDIA_TYPE_VIDEO
98     },
99     { NULL }
100 };
101
102 AVFilter avfilter_vf_showinfo = {
103     .name        = "showinfo",
104     .description = NULL_IF_CONFIG_SMALL("Show textual information for each video frame."),
105
106     .priv_size = sizeof(ShowInfoContext),
107     .init      = init,
108
109     .inputs    = avfilter_vf_showinfo_inputs,
110
111     .outputs   = avfilter_vf_showinfo_outputs,
112 };