1 /*
2 * Copyright (c) 2013 Nicolas George
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public License
8 * as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public License
17 * along with FFmpeg; if not, write to the Free Software Foundation, Inc.,
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #include "libavutil/adler32.h"
22 #include "libavutil/avassert.h"
23 #include "libavutil/bprint.h"
24 #include "libavutil/imgutils.h"
25 #include "libavutil/pixdesc.h"
26 #include "libavformat/mux.h"
27 #include "avformat.h"
28 #include "internal.h"
29
30 /* Identical to Adler32 when the type is uint8_t. */
31 #define DEFINE_CKSUM_LINE(name, type, conv) \
32 static void cksum_line_ ## name(unsigned *cksum, void *data, unsigned size) \
33 { \
34 type *p = data; \
35 unsigned a = *cksum & 0xFFFF, b = *cksum >> 16; \
36 for (; size > 0; size--, p++) { \
37 a = (a + (unsigned)(conv)) % 65521; \
38 b = (b + a) % 65521; \
39 } \
40 *cksum = a | (b << 16); \
41 }
42
43 DEFINE_CKSUM_LINE(u8, uint8_t, *p)
44 DEFINE_CKSUM_LINE(s16, int16_t, *p + 0x8000)
45 DEFINE_CKSUM_LINE(s32, int32_t, *p + 0x80000000)
46 DEFINE_CKSUM_LINE(flt, float, *p * 0x80000000 + 0x80000000)
47 DEFINE_CKSUM_LINE(dbl, double, *p * 0x80000000 + 0x80000000)
48
video_frame_cksum(AVBPrint *bp, AVFrame *frame)49 static void video_frame_cksum(AVBPrint *bp, AVFrame *frame)
50 {
51 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(frame->format);
52 int i, y;
53 uint8_t *data;
54 int linesize[5] = { 0 };
55
56 av_bprintf(bp, ", %d x %d", frame->width, frame->height);
57 if (!desc) {
58 av_bprintf(bp, ", unknown");
59 return;
60 }
61 if (av_image_fill_linesizes(linesize, frame->format, frame->width) < 0)
62 return;
63 av_bprintf(bp, ", %s", desc->name);
64 for (i = 0; linesize[i]; i++) {
65 unsigned cksum = 0;
66 int h = frame->height;
67 if ((i == 1 || i == 2) && desc->nb_components >= 3)
68 h = AV_CEIL_RSHIFT(h, desc->log2_chroma_h);
69 data = frame->data[i];
70 for (y = 0; y < h; y++) {
71 cksum = av_adler32_update(cksum, data, linesize[i]);
72 data += frame->linesize[i];
73 }
74 av_bprintf(bp, ", 0x%08x", cksum);
75 }
76 }
77
audio_frame_cksum(AVBPrint *bp, AVFrame *frame)78 static void audio_frame_cksum(AVBPrint *bp, AVFrame *frame)
79 {
80 int nb_planes, nb_samples, p;
81 const char *name;
82
83 nb_planes = frame->ch_layout.nb_channels;
84 nb_samples = frame->nb_samples;
85 if (!av_sample_fmt_is_planar(frame->format)) {
86 nb_samples *= nb_planes;
87 nb_planes = 1;
88 }
89 name = av_get_sample_fmt_name(frame->format);
90 av_bprintf(bp, ", %d samples", frame->nb_samples);
91 av_bprintf(bp, ", %s", name ? name : "unknown");
92 for (p = 0; p < nb_planes; p++) {
93 uint32_t cksum = 0;
94 void *d = frame->extended_data[p];
95 switch (frame->format) {
96 case AV_SAMPLE_FMT_U8:
97 case AV_SAMPLE_FMT_U8P:
98 cksum_line_u8(&cksum, d, nb_samples);
99 break;
100 case AV_SAMPLE_FMT_S16:
101 case AV_SAMPLE_FMT_S16P:
102 cksum_line_s16(&cksum, d, nb_samples);
103 break;
104 case AV_SAMPLE_FMT_S32:
105 case AV_SAMPLE_FMT_S32P:
106 cksum_line_s32(&cksum, d, nb_samples);
107 break;
108 case AV_SAMPLE_FMT_FLT:
109 case AV_SAMPLE_FMT_FLTP:
110 cksum_line_flt(&cksum, d, nb_samples);
111 break;
112 case AV_SAMPLE_FMT_DBL:
113 case AV_SAMPLE_FMT_DBLP:
114 cksum_line_dbl(&cksum, d, nb_samples);
115 break;
116 default:
117 av_assert0(!"reached");
118 }
119 av_bprintf(bp, ", 0x%08"PRIx32, cksum);
120 }
121 }
122
write_header(struct AVFormatContext *s)123 static int write_header(struct AVFormatContext *s)
124 {
125 return ff_framehash_write_header(s);
126 }
127
write_frame(struct AVFormatContext *s, int stream_index, AVFrame **frame, unsigned flags)128 static int write_frame(struct AVFormatContext *s, int stream_index,
129 AVFrame **frame, unsigned flags)
130 {
131 AVBPrint bp;
132 int ret = 0;
133 enum AVMediaType type;
134 const char *type_name;
135
136 if ((flags & AV_WRITE_UNCODED_FRAME_QUERY))
137 return 0;
138
139 av_bprint_init(&bp, 0, AV_BPRINT_SIZE_UNLIMITED);
140 av_bprintf(&bp, "%d, %10"PRId64"",
141 stream_index, (*frame)->pts);
142 type = s->streams[stream_index]->codecpar->codec_type;
143 type_name = av_get_media_type_string(type);
144 av_bprintf(&bp, ", %s", type_name ? type_name : "unknown");
145 switch (type) {
146 case AVMEDIA_TYPE_VIDEO:
147 video_frame_cksum(&bp, *frame);
148 break;
149 case AVMEDIA_TYPE_AUDIO:
150 audio_frame_cksum(&bp, *frame);
151 break;
152 }
153
154 av_bprint_chars(&bp, '\n', 1);
155 if (av_bprint_is_complete(&bp))
156 avio_write(s->pb, bp.str, bp.len);
157 else
158 ret = AVERROR(ENOMEM);
159 av_bprint_finalize(&bp, NULL);
160 return ret;
161 }
162
write_packet(struct AVFormatContext *s, AVPacket *pkt)163 static int write_packet(struct AVFormatContext *s, AVPacket *pkt)
164 {
165 return AVERROR(ENOSYS);
166 }
167
168 const AVOutputFormat ff_uncodedframecrc_muxer = {
169 .name = "uncodedframecrc",
170 .long_name = NULL_IF_CONFIG_SMALL("uncoded framecrc testing"),
171 .audio_codec = AV_CODEC_ID_PCM_S16LE,
172 .video_codec = AV_CODEC_ID_RAWVIDEO,
173 .write_header = write_header,
174 .write_packet = write_packet,
175 .write_uncoded_frame = write_frame,
176 .flags = AVFMT_VARIABLE_FPS | AVFMT_TS_NONSTRICT |
177 AVFMT_TS_NEGATIVE,
178 };
179