libavformat/ivfenc.c

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