1/* 2 * Copyright (c) 2010 Reimar D��ffinger 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 8 * License 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 GNU 14 * Lesser General Public License for more details. 15 * 16 * You should have received a copy of the GNU Lesser General Public 17 * License along with FFmpeg; if not, write to the Free Software 18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 19 */ 20#include "avformat.h" 21#include "libavutil/intreadwrite.h" 22 23static int ivf_write_header(AVFormatContext *s) 24{ 25 AVCodecContext *ctx; 26 AVIOContext *pb = s->pb; 27 28 if (s->nb_streams != 1) { 29 av_log(s, AV_LOG_ERROR, "Format supports only exactly one video stream\n"); 30 return AVERROR(EINVAL); 31 } 32 ctx = s->streams[0]->codec; 33 if (ctx->codec_type != AVMEDIA_TYPE_VIDEO || 34 !(ctx->codec_id == AV_CODEC_ID_VP8 || ctx->codec_id == AV_CODEC_ID_VP9)) { 35 av_log(s, AV_LOG_ERROR, "Currently only VP8 and VP9 are supported!\n"); 36 return AVERROR(EINVAL); 37 } 38 avio_write(pb, "DKIF", 4); 39 avio_wl16(pb, 0); // version 40 avio_wl16(pb, 32); // header length 41 avio_wl32(pb, ctx->codec_tag ? ctx->codec_tag : AV_RL32("VP80")); 42 avio_wl16(pb, ctx->width); 43 avio_wl16(pb, ctx->height); 44 avio_wl32(pb, s->streams[0]->time_base.den); 45 avio_wl32(pb, s->streams[0]->time_base.num); 46 avio_wl64(pb, s->streams[0]->duration); // TODO: duration or number of frames?!? 47 48 return 0; 49} 50 51static int ivf_write_packet(AVFormatContext *s, AVPacket *pkt) 52{ 53 AVIOContext *pb = s->pb; 54 avio_wl32(pb, pkt->size); 55 avio_wl64(pb, pkt->pts); 56 avio_write(pb, pkt->data, pkt->size); 57 58 return 0; 59} 60 61AVOutputFormat ff_ivf_muxer = { 62 .name = "ivf", 63 .long_name = NULL_IF_CONFIG_SMALL("On2 IVF"), 64 .extensions = "ivf", 65 .audio_codec = AV_CODEC_ID_NONE, 66 .video_codec = AV_CODEC_ID_VP8, 67 .write_header = ivf_write_header, 68 .write_packet = ivf_write_packet, 69}; 70