FFmpeg
ivfenc.c
Go to the documentation of this file.
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 "libavutil/intreadwrite.h"
23 
24 typedef struct IVFEncContext {
25  unsigned frame_cnt;
28 
30 {
31  AVCodecParameters *par;
32  AVIOContext *pb = s->pb;
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  avio_write(pb, "DKIF", 4);
47  avio_wl16(pb, 0); // version
48  avio_wl16(pb, 32); // header length
49  avio_wl32(pb,
50  par->codec_id == AV_CODEC_ID_VP9 ? AV_RL32("VP90") :
51  par->codec_id == AV_CODEC_ID_VP8 ? AV_RL32("VP80") : AV_RL32("AV01"));
52  avio_wl16(pb, par->width);
53  avio_wl16(pb, par->height);
54  avio_wl32(pb, s->streams[0]->time_base.den);
55  avio_wl32(pb, s->streams[0]->time_base.num);
56  avio_wl64(pb, 0xFFFFFFFFFFFFFFFFULL);
57 
58  return 0;
59 }
60 
62 {
63  AVIOContext *pb = s->pb;
65 
66  avio_wl32(pb, pkt->size);
67  avio_wl64(pb, pkt->pts);
68  avio_write(pb, pkt->data, pkt->size);
69  if (ctx->frame_cnt)
70  ctx->sum_delta_pts += pkt->pts - ctx->last_pts;
71  ctx->frame_cnt++;
72  ctx->last_pts = pkt->pts;
73 
74  return 0;
75 }
76 
78 {
79  AVIOContext *pb = s->pb;
81 
82  if ((pb->seekable & AVIO_SEEKABLE_NORMAL) && ctx->frame_cnt > 1) {
83  size_t end = avio_tell(pb);
84 
85  avio_seek(pb, 24, SEEK_SET);
86  avio_wl64(pb, ctx->frame_cnt * ctx->sum_delta_pts / (ctx->frame_cnt - 1));
87  avio_seek(pb, end, SEEK_SET);
88  }
89 
90  return 0;
91 }
92 
93 static int ivf_check_bitstream(struct AVFormatContext *s, const AVPacket *pkt)
94 {
95  int ret = 1;
96  AVStream *st = s->streams[pkt->stream_index];
97 
98  if (st->codecpar->codec_id == AV_CODEC_ID_VP9)
99  ret = ff_stream_add_bitstream_filter(st, "vp9_superframe", NULL);
100  else if (st->codecpar->codec_id == AV_CODEC_ID_AV1)
101  ret = ff_stream_add_bitstream_filter(st, "av1_metadata", "td=insert");
102 
103  return ret;
104 }
105 
106 static const AVCodecTag codec_ivf_tags[] = {
107  { AV_CODEC_ID_VP8, MKTAG('V', 'P', '8', '0') },
108  { AV_CODEC_ID_VP9, MKTAG('V', 'P', '9', '0') },
109  { AV_CODEC_ID_AV1, MKTAG('A', 'V', '0', '1') },
110  { AV_CODEC_ID_NONE, 0 }
111 };
112 
114  .priv_data_size = sizeof(IVFEncContext),
115  .name = "ivf",
116  .long_name = NULL_IF_CONFIG_SMALL("On2 IVF"),
117  .extensions = "ivf",
118  .audio_codec = AV_CODEC_ID_NONE,
119  .video_codec = AV_CODEC_ID_VP8,
120  .write_header = ivf_write_header,
121  .write_packet = ivf_write_packet,
122  .write_trailer = ivf_write_trailer,
123  .check_bitstream = ivf_check_bitstream,
124  .codec_tag = (const AVCodecTag* const []){ codec_ivf_tags, 0 },
125 };
#define NULL
Definition: coverity.c:32
void avio_wl16(AVIOContext *s, unsigned int val)
Definition: aviobuf.c:469
Bytestream IO Context.
Definition: avio.h:161
static int ivf_write_header(AVFormatContext *s)
Definition: ivfenc.c:29
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: avcodec.h:3953
int num
Numerator.
Definition: rational.h:59
int size
Definition: avcodec.h:1478
int64_t avio_seek(AVIOContext *s, int64_t offset, int whence)
fseek() equivalent for AVIOContext.
Definition: aviobuf.c:246
AVOutputFormat ff_ivf_muxer
Definition: ivfenc.c:113
static AVPacket pkt
This struct describes the properties of an encoded stream.
Definition: avcodec.h:3945
static int ivf_write_packet(AVFormatContext *s, AVPacket *pkt)
Definition: ivfenc.c:61
Format I/O context.
Definition: avformat.h:1358
void avio_wl32(AVIOContext *s, unsigned int val)
Definition: aviobuf.c:369
int width
Video only.
Definition: avcodec.h:4019
int priv_data_size
size of private data so that it can be allocated in the wrapper
Definition: avformat.h:555
static av_cold int end(AVCodecContext *avctx)
Definition: avrndec.c:90
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1426
uint8_t * data
Definition: avcodec.h:1477
static av_always_inline int64_t avio_tell(AVIOContext *s)
ftell() equivalent for AVIOContext.
Definition: avio.h:557
void avio_write(AVIOContext *s, const unsigned char *buf, int size)
Definition: aviobuf.c:218
static const AVCodecTag codec_ivf_tags[]
Definition: ivfenc.c:106
#define av_log(a,...)
void avio_wl64(AVIOContext *s, uint64_t val)
Definition: aviobuf.c:457
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:186
enum AVMediaType codec_type
General type of the encoded data.
Definition: avcodec.h:3949
unsigned int nb_streams
Number of elements in AVFormatContext.streams.
Definition: avformat.h:1414
int seekable
A combination of AVIO_SEEKABLE_ flags or 0 when the stream is not seekable.
Definition: avio.h:260
AVFormatContext * ctx
Definition: movenc.c:48
#define s(width, name)
Definition: cbs_vp9.c:257
static int ivf_write_trailer(AVFormatContext *s)
Definition: ivfenc.c:77
Stream structure.
Definition: avformat.h:881
uint64_t sum_delta_pts
Definition: ivfenc.c:26
#define AVIO_SEEKABLE_NORMAL
Seeking works like for a local file.
Definition: avio.h:40
AVIOContext * pb
I/O context.
Definition: avformat.h:1400
unsigned frame_cnt
Definition: ivfenc.c:25
Main libavformat public API header.
int den
Denominator.
Definition: rational.h:60
uint64_t last_pts
Definition: ivfenc.c:26
int ff_stream_add_bitstream_filter(AVStream *st, const char *name, const char *args)
Add a bitstream filter to a stream.
Definition: utils.c:5512
void * priv_data
Format private data.
Definition: avformat.h:1386
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:1028
int stream_index
Definition: avcodec.h:1479
AVRational time_base
This is the fundamental unit of time (in seconds) in terms of which frame timestamps are represented...
Definition: avformat.h:910
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later.That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another.Frame references ownership and permissions
#define MKTAG(a, b, c, d)
Definition: common.h:366
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_RL32
Definition: bytestream.h:87
This structure stores compressed data.
Definition: avcodec.h:1454
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: avcodec.h:1470
static int ivf_check_bitstream(struct AVFormatContext *s, const AVPacket *pkt)
Definition: ivfenc.c:93
const char * name
Definition: opengl_enc.c:102