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 "avformat.h" 27#include "internal.h" 28 29/* Identical to Adler32 when the type is uint8_t. */ 30#define DEFINE_CKSUM_LINE(name, type, conv) \ 31static void cksum_line_ ## name(unsigned *cksum, void *data, unsigned size) \ 32{ \ 33 type *p = data; \ 34 unsigned a = *cksum & 0xFFFF, b = *cksum >> 16; \ 35 for (; size > 0; size--, p++) { \ 36 a = (a + (unsigned)(conv)) % 65521; \ 37 b = (b + a) % 65521; \ 38 } \ 39 *cksum = a | (b << 16); \ 40} 41 42DEFINE_CKSUM_LINE(u8, uint8_t, *p) 43DEFINE_CKSUM_LINE(s16, int16_t, *p + 0x8000) 44DEFINE_CKSUM_LINE(s32, int32_t, *p + 0x80000000) 45DEFINE_CKSUM_LINE(flt, float, *p * 0x80000000 + 0x80000000) 46DEFINE_CKSUM_LINE(dbl, double, *p * 0x80000000 + 0x80000000) 47 48static void video_frame_cksum(AVBPrint *bp, AVFrame *frame) 49{ 50 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(frame->format); 51 int i, y; 52 uint8_t *data; 53 int linesize[5] = { 0 }; 54 55 av_bprintf(bp, ", %d x %d", frame->width, frame->height); 56 if (!desc) { 57 av_bprintf(bp, ", unknown"); 58 return; 59 } 60 if (av_image_fill_linesizes(linesize, frame->format, frame->width) < 0) 61 return; 62 av_bprintf(bp, ", %s", desc->name); 63 for (i = 0; linesize[i]; i++) { 64 unsigned cksum = 0; 65 int h = frame->height; 66 if ((i == 1 || i == 2) && desc->nb_components >= 3) 67 h = -((-h) >> desc->log2_chroma_h); 68 data = frame->data[i]; 69 for (y = 0; y < h; y++) { 70 cksum = av_adler32_update(cksum, data, linesize[i]); 71 data += frame->linesize[i]; 72 } 73 av_bprintf(bp, ", 0x%08x", cksum); 74 } 75} 76 77static void audio_frame_cksum(AVBPrint *bp, AVFrame *frame) 78{ 79 int nb_planes, nb_samples, p; 80 const char *name; 81 82 nb_planes = av_frame_get_channels(frame); 83 nb_samples = frame->nb_samples; 84 if (!av_sample_fmt_is_planar(frame->format)) { 85 nb_samples *= nb_planes; 86 nb_planes = 1; 87 } 88 name = av_get_sample_fmt_name(frame->format); 89 av_bprintf(bp, ", %d samples", frame->nb_samples); 90 av_bprintf(bp, ", %s", name ? name : "unknown"); 91 for (p = 0; p < nb_planes; p++) { 92 uint32_t cksum = 0; 93 void *d = frame->extended_data[p]; 94 switch (frame->format) { 95 case AV_SAMPLE_FMT_U8: 96 case AV_SAMPLE_FMT_U8P: 97 cksum_line_u8(&cksum, d, nb_samples); 98 break; 99 case AV_SAMPLE_FMT_S16: 100 case AV_SAMPLE_FMT_S16P: 101 cksum_line_s16(&cksum, d, nb_samples); 102 break; 103 case AV_SAMPLE_FMT_S32: 104 case AV_SAMPLE_FMT_S32P: 105 cksum_line_s32(&cksum, d, nb_samples); 106 break; 107 case AV_SAMPLE_FMT_FLT: 108 case AV_SAMPLE_FMT_FLTP: 109 cksum_line_flt(&cksum, d, nb_samples); 110 break; 111 case AV_SAMPLE_FMT_DBL: 112 case AV_SAMPLE_FMT_DBLP: 113 cksum_line_dbl(&cksum, d, nb_samples); 114 break; 115 default: 116 av_assert0(!"reached"); 117 } 118 av_bprintf(bp, ", 0x%08x", cksum); 119 } 120} 121 122static int write_frame(struct AVFormatContext *s, int stream_index, 123 AVFrame **frame, unsigned flags) 124{ 125 AVBPrint bp; 126 int ret = 0; 127 enum AVMediaType type; 128 const char *type_name; 129 130 if ((flags & AV_WRITE_UNCODED_FRAME_QUERY)) 131 return 0; 132 133 av_bprint_init(&bp, 0, AV_BPRINT_SIZE_UNLIMITED); 134 av_bprintf(&bp, "%d, %10"PRId64"", 135 stream_index, (*frame)->pts); 136 type = s->streams[stream_index]->codec->codec_type; 137 type_name = av_get_media_type_string(type); 138 av_bprintf(&bp, ", %s", type_name ? type_name : "unknown"); 139 switch (type) { 140 case AVMEDIA_TYPE_VIDEO: 141 video_frame_cksum(&bp, *frame); 142 break; 143 case AVMEDIA_TYPE_AUDIO: 144 audio_frame_cksum(&bp, *frame); 145 break; 146 } 147 148 av_bprint_chars(&bp, '\n', 1); 149 if (av_bprint_is_complete(&bp)) 150 avio_write(s->pb, bp.str, bp.len); 151 else 152 ret = AVERROR(ENOMEM); 153 av_bprint_finalize(&bp, NULL); 154 return ret; 155} 156 157static int write_packet(struct AVFormatContext *s, AVPacket *pkt) 158{ 159 return AVERROR(ENOSYS); 160} 161 162AVOutputFormat ff_uncodedframecrc_muxer = { 163 .name = "uncodedframecrc", 164 .long_name = NULL_IF_CONFIG_SMALL("uncoded framecrc testing"), 165 .audio_codec = AV_CODEC_ID_PCM_S16LE, 166 .video_codec = AV_CODEC_ID_RAWVIDEO, 167 .write_header = ff_framehash_write_header, 168 .write_packet = write_packet, 169 .write_uncoded_frame = write_frame, 170 .flags = AVFMT_VARIABLE_FPS | AVFMT_TS_NONSTRICT | 171 AVFMT_TS_NEGATIVE, 172}; 173