Subversion Repositories Kolibri OS

Rev

Blame | Last modification | View Log | RSS feed

  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.  
  23. typedef struct IVFEncContext {
  24.     unsigned frame_cnt;
  25.     uint64_t last_pts, sum_delta_pts;
  26. } IVFEncContext;
  27.  
  28. static int ivf_write_header(AVFormatContext *s)
  29. {
  30.     AVCodecContext *ctx;
  31.     AVIOContext *pb = s->pb;
  32.  
  33.     if (s->nb_streams != 1) {
  34.         av_log(s, AV_LOG_ERROR, "Format supports only exactly one video stream\n");
  35.         return AVERROR(EINVAL);
  36.     }
  37.     ctx = s->streams[0]->codec;
  38.     if (ctx->codec_type != AVMEDIA_TYPE_VIDEO ||
  39.         !(ctx->codec_id == AV_CODEC_ID_VP8 || ctx->codec_id == AV_CODEC_ID_VP9)) {
  40.         av_log(s, AV_LOG_ERROR, "Currently only VP8 and VP9 are supported!\n");
  41.         return AVERROR(EINVAL);
  42.     }
  43.     avio_write(pb, "DKIF", 4);
  44.     avio_wl16(pb, 0); // version
  45.     avio_wl16(pb, 32); // header length
  46.     avio_wl32(pb, ctx->codec_tag ? ctx->codec_tag : ctx->codec_id == AV_CODEC_ID_VP9 ? AV_RL32("VP90") : AV_RL32("VP80"));
  47.     avio_wl16(pb, ctx->width);
  48.     avio_wl16(pb, ctx->height);
  49.     avio_wl32(pb, s->streams[0]->time_base.den);
  50.     avio_wl32(pb, s->streams[0]->time_base.num);
  51.     avio_wl64(pb, 0xFFFFFFFFFFFFFFFFULL);
  52.  
  53.     return 0;
  54. }
  55.  
  56. static int ivf_write_packet(AVFormatContext *s, AVPacket *pkt)
  57. {
  58.     AVIOContext *pb = s->pb;
  59.     IVFEncContext *ctx = s->priv_data;
  60.  
  61.     avio_wl32(pb, pkt->size);
  62.     avio_wl64(pb, pkt->pts);
  63.     avio_write(pb, pkt->data, pkt->size);
  64.     if (ctx->frame_cnt)
  65.         ctx->sum_delta_pts += pkt->pts - ctx->last_pts;
  66.     ctx->frame_cnt++;
  67.     ctx->last_pts = pkt->pts;
  68.  
  69.     return 0;
  70. }
  71.  
  72. static int ivf_write_trailer(AVFormatContext *s)
  73. {
  74.     AVIOContext *pb = s->pb;
  75.     IVFEncContext *ctx = s->priv_data;
  76.  
  77.     if (pb->seekable && ctx->frame_cnt > 1) {
  78.         size_t end = avio_tell(pb);
  79.  
  80.         avio_seek(pb, 24, SEEK_SET);
  81.         avio_wl64(pb, ctx->frame_cnt * ctx->sum_delta_pts / (ctx->frame_cnt - 1));
  82.         avio_seek(pb, end, SEEK_SET);
  83.     }
  84.  
  85.     return 0;
  86. }
  87.  
  88. AVOutputFormat ff_ivf_muxer = {
  89.     .priv_data_size = sizeof(IVFEncContext),
  90.     .name         = "ivf",
  91.     .long_name    = NULL_IF_CONFIG_SMALL("On2 IVF"),
  92.     .extensions   = "ivf",
  93.     .audio_codec  = AV_CODEC_ID_NONE,
  94.     .video_codec  = AV_CODEC_ID_VP8,
  95.     .write_header = ivf_write_header,
  96.     .write_packet = ivf_write_packet,
  97.     .write_trailer = ivf_write_trailer,
  98. };
  99.