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 "internal.h" 22#include "mux.h" 23#include "libavutil/intreadwrite.h" 24 25typedef struct IVFEncContext { 26 unsigned frame_cnt; 27 uint64_t last_pts, sum_delta_pts, last_pkt_duration; 28} IVFEncContext; 29 30static int ivf_init(AVFormatContext *s) 31{ 32 AVCodecParameters *par; 33 34 if (s->nb_streams != 1) { 35 av_log(s, AV_LOG_ERROR, "Format supports only exactly one video stream\n"); 36 return AVERROR(EINVAL); 37 } 38 par = s->streams[0]->codecpar; 39 if (par->codec_type != AVMEDIA_TYPE_VIDEO || 40 !(par->codec_id == AV_CODEC_ID_AV1 || 41 par->codec_id == AV_CODEC_ID_VP8 || 42 par->codec_id == AV_CODEC_ID_VP9)) { 43 av_log(s, AV_LOG_ERROR, "Currently only VP8, VP9 and AV1 are supported!\n"); 44 return AVERROR(EINVAL); 45 } 46 47 if (par->codec_id == AV_CODEC_ID_VP9) { 48 int ret = ff_stream_add_bitstream_filter(s->streams[0], "vp9_superframe", NULL); 49 if (ret < 0) 50 return ret; 51 } else if (par->codec_id == AV_CODEC_ID_AV1) { 52 int ret = ff_stream_add_bitstream_filter(s->streams[0], "av1_metadata", "td=insert"); 53 if (ret < 0) 54 return ret; 55 } 56 57 return 0; 58} 59 60static int ivf_write_header(AVFormatContext *s) 61{ 62 AVCodecParameters *par = s->streams[0]->codecpar; 63 AVIOContext *pb = s->pb; 64 65 avio_write(pb, "DKIF", 4); 66 avio_wl16(pb, 0); // version 67 avio_wl16(pb, 32); // header length 68 avio_wl32(pb, 69 par->codec_id == AV_CODEC_ID_VP9 ? AV_RL32("VP90") : 70 par->codec_id == AV_CODEC_ID_VP8 ? AV_RL32("VP80") : AV_RL32("AV01")); 71 avio_wl16(pb, par->width); 72 avio_wl16(pb, par->height); 73 avio_wl32(pb, s->streams[0]->time_base.den); 74 avio_wl32(pb, s->streams[0]->time_base.num); 75 avio_wl64(pb, 0xFFFFFFFFFFFFFFFFULL); // length is overwritten at the end of muxing 76 77 return 0; 78} 79 80static int ivf_write_packet(AVFormatContext *s, AVPacket *pkt) 81{ 82 AVIOContext *pb = s->pb; 83 IVFEncContext *ctx = s->priv_data; 84 85 avio_wl32(pb, pkt->size); 86 avio_wl64(pb, pkt->pts); 87 avio_write(pb, pkt->data, pkt->size); 88 if (ctx->frame_cnt) 89 ctx->sum_delta_pts += pkt->pts - ctx->last_pts; 90 ctx->last_pkt_duration = pkt->duration; 91 ctx->frame_cnt++; 92 ctx->last_pts = pkt->pts; 93 94 return 0; 95} 96 97static int ivf_write_trailer(AVFormatContext *s) 98{ 99 AVIOContext *pb = s->pb; 100 IVFEncContext *ctx = s->priv_data; 101 102 if ((pb->seekable & AVIO_SEEKABLE_NORMAL) && 103 (ctx->frame_cnt > 1 || (ctx->frame_cnt == 1 && ctx->last_pkt_duration))) { 104 int64_t end = avio_tell(pb); 105 106 avio_seek(pb, 24, SEEK_SET); 107 // overwrite the "length" field (duration) 108 avio_wl32(pb, ctx->last_pkt_duration ? 109 ctx->sum_delta_pts + ctx->last_pkt_duration : 110 ctx->frame_cnt * ctx->sum_delta_pts / (ctx->frame_cnt - 1)); 111 avio_wl32(pb, 0); // zero out unused bytes 112 avio_seek(pb, end, SEEK_SET); 113 } 114 115 return 0; 116} 117 118static const AVCodecTag codec_ivf_tags[] = { 119 { AV_CODEC_ID_VP8, MKTAG('V', 'P', '8', '0') }, 120 { AV_CODEC_ID_VP9, MKTAG('V', 'P', '9', '0') }, 121 { AV_CODEC_ID_AV1, MKTAG('A', 'V', '0', '1') }, 122 { AV_CODEC_ID_NONE, 0 } 123}; 124 125const AVOutputFormat ff_ivf_muxer = { 126 .priv_data_size = sizeof(IVFEncContext), 127 .name = "ivf", 128 .long_name = NULL_IF_CONFIG_SMALL("On2 IVF"), 129 .extensions = "ivf", 130 .audio_codec = AV_CODEC_ID_NONE, 131 .video_codec = AV_CODEC_ID_VP8, 132 .init = ivf_init, 133 .write_header = ivf_write_header, 134 .write_packet = ivf_write_packet, 135 .write_trailer = ivf_write_trailer, 136 .codec_tag = (const AVCodecTag* const []){ codec_ivf_tags, 0 }, 137}; 138