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 "mux.h"
23 #include "libavutil/intreadwrite.h"
24 
25 typedef struct IVFEncContext {
26  unsigned frame_cnt;
29 
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 
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 
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 
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 
118 static 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 
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 };
ivf_write_trailer
static int ivf_write_trailer(AVFormatContext *s)
Definition: ivfenc.c:97
ivf_init
static int ivf_init(AVFormatContext *s)
Definition: ivfenc.c:30
name
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf default minimum maximum flags name is the option name
Definition: writing_filters.txt:88
AVERROR
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
AVCodecParameters::codec_type
enum AVMediaType codec_type
General type of the encoded data.
Definition: codec_par.h:57
AVCodecParameters
This struct describes the properties of an encoded stream.
Definition: codec_par.h:53
codec_ivf_tags
static const AVCodecTag codec_ivf_tags[]
Definition: ivfenc.c:118
AVPacket::data
uint8_t * data
Definition: packet.h:374
avio_wl64
void avio_wl64(AVIOContext *s, uint64_t val)
Definition: aviobuf.c:454
AVPacket::duration
int64_t duration
Duration of this packet in AVStream->time_base units, 0 if unknown.
Definition: packet.h:392
avio_wl16
void avio_wl16(AVIOContext *s, unsigned int val)
Definition: aviobuf.c:466
avio_tell
static av_always_inline int64_t avio_tell(AVIOContext *s)
ftell() equivalent for AVIOContext.
Definition: avio.h:505
pkt
AVPacket * pkt
Definition: movenc.c:59
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:180
ff_ivf_muxer
const AVOutputFormat ff_ivf_muxer
Definition: ivfenc.c:125
IVFEncContext::frame_cnt
unsigned frame_cnt
Definition: ivfenc.c:26
AVCodecTag
Definition: internal.h:49
intreadwrite.h
s
#define s(width, name)
Definition: cbs_vp9.c:256
AV_CODEC_ID_VP9
@ AV_CODEC_ID_VP9
Definition: codec_id.h:220
AVCodecParameters::width
int width
Video only.
Definition: codec_par.h:127
ctx
AVFormatContext * ctx
Definition: movenc.c:48
IVFEncContext
Definition: ivfenc.c:25
AVFormatContext
Format I/O context.
Definition: avformat.h:1216
internal.h
NULL
#define NULL
Definition: coverity.c:32
IVFEncContext::last_pkt_duration
uint64_t last_pkt_duration
Definition: ivfenc.c:27
AV_CODEC_ID_AV1
@ AV_CODEC_ID_AV1
Definition: codec_id.h:283
AVIOContext
Bytestream IO Context.
Definition: avio.h:162
AVPacket::size
int size
Definition: packet.h:375
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:115
AVIOContext::seekable
int seekable
A combination of AVIO_SEEKABLE_ flags or 0 when the stream is not seekable.
Definition: avio.h:263
IVFEncContext::sum_delta_pts
uint64_t sum_delta_pts
Definition: ivfenc.c:27
ivf_write_header
static int ivf_write_header(AVFormatContext *s)
Definition: ivfenc.c:60
avio_write
void avio_write(AVIOContext *s, const unsigned char *buf, int size)
Definition: aviobuf.c:232
IVFEncContext::last_pts
uint64_t last_pts
Definition: ivfenc.c:27
avio_wl32
void avio_wl32(AVIOContext *s, unsigned int val)
Definition: aviobuf.c:386
AV_CODEC_ID_NONE
@ AV_CODEC_ID_NONE
Definition: codec_id.h:50
AVOutputFormat
Definition: avformat.h:509
AVPacket::pts
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: packet.h:367
AVCodecParameters::height
int height
Definition: codec_par.h:128
ret
ret
Definition: filter_design.txt:187
avio_seek
int64_t avio_seek(AVIOContext *s, int64_t offset, int whence)
fseek() equivalent for AVIOContext.
Definition: aviobuf.c:260
avformat.h
AV_RL32
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_RL32
Definition: bytestream.h:92
ff_stream_add_bitstream_filter
int ff_stream_add_bitstream_filter(AVStream *st, const char *name, const char *args)
Add a bitstream filter to a stream.
Definition: mux.c:1310
AVIO_SEEKABLE_NORMAL
#define AVIO_SEEKABLE_NORMAL
Seeking works like for a local file.
Definition: avio.h:41
ivf_write_packet
static int ivf_write_packet(AVFormatContext *s, AVPacket *pkt)
Definition: ivfenc.c:80
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
AVCodecParameters::codec_id
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: codec_par.h:61
AVPacket
This structure stores compressed data.
Definition: packet.h:351
AV_CODEC_ID_VP8
@ AV_CODEC_ID_VP8
Definition: codec_id.h:192
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
MKTAG
#define MKTAG(a, b, c, d)
Definition: macros.h:55
mux.h
AVOutputFormat::priv_data_size
int priv_data_size
size of private data so that it can be allocated in the wrapper
Definition: avformat.h:550