1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
|
/*
* Matroska file muxer
* Copyright (c) 2007 David Conrad
*
* This file is part of FFmpeg.
*
* FFmpeg is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* FFmpeg is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with FFmpeg; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
#include "avformat.h"
#include "riff.h"
#include "matroska.h"
typedef struct MatroskaMuxContext {
offset_t segment;
} MatroskaMuxContext;
static void put_ebml_id(ByteIOContext *pb, unsigned int id)
{
if (id >= 0x3fffff)
put_byte(pb, id >> 24);
if (id >= 0x7fff)
put_byte(pb, id >> 16);
if (id >= 0xff)
put_byte(pb, id >> 8);
put_byte(pb, id);
}
// XXX: test this thoroughly and get rid of minbytes hack (currently needed to
// use up all of the space reserved in start_ebml_master)
static void put_ebml_size(ByteIOContext *pb, uint64_t size, int minbytes)
{
int bytes = minbytes;
while (size >> (bytes*7 + 7)) bytes++;
// sizes larger than this are currently undefined in EBML
// XXX: error condition?
if (size > (1ULL<<56)-1) return;
put_byte(pb, (0x80 >> bytes) | (size >> bytes*8));
for (bytes -= 1; bytes >= 0; bytes--)
put_byte(pb, size >> bytes*8);
}
static void put_ebml_uint(ByteIOContext *pb, unsigned int elementid, uint64_t val)
{
int bytes = 1;
while (val >> bytes*8) bytes++;
put_ebml_id(pb, elementid);
put_ebml_size(pb, bytes, 0);
for (bytes -= 1; bytes >= 0; bytes--)
put_byte(pb, val >> bytes*8);
}
//static void put_ebml_sint(ByteIOContext *pb, unsigned int elementid, int64_t val)
static void put_ebml_float(ByteIOContext *pb, unsigned int elementid, double val)
{
// XXX: single-precision floats?
put_ebml_id(pb, elementid);
put_ebml_size(pb, 8, 0);
put_be64(pb, av_dbl2int(val));
}
static void put_ebml_binary(ByteIOContext *pb, unsigned int elementid,
const uint8_t *buf, int size)
{
put_ebml_id(pb, elementid);
put_ebml_size(pb, size, 0);
put_buffer(pb, buf, size);
}
static void put_ebml_string(ByteIOContext *pb, unsigned int elementid, const char *str)
{
put_ebml_binary(pb, elementid, str, strlen(str));
}
static offset_t start_ebml_master(ByteIOContext *pb, unsigned int elementid)
{
put_ebml_id(pb, elementid);
// XXX: this always reserves the maximum needed space to store any size value
// we should be smarter (additional parameter for expected size?)
put_ebml_size(pb, (1ULL<<56)-1, 0); // largest unknown size
return url_ftell(pb);
}
static void end_ebml_master(ByteIOContext *pb, offset_t start)
{
offset_t pos = url_ftell(pb);
url_fseek(pb, start - 8, SEEK_SET);
put_ebml_size(pb, pos - start, 7);
url_fseek(pb, pos, SEEK_SET);
}
static int mkv_write_header(AVFormatContext *s)
{
MatroskaMuxContext *mkv = s->priv_data;
ByteIOContext *pb = &s->pb;
offset_t ebml_header, segment_info, tracks;
int i;
ebml_header = start_ebml_master(pb, EBML_ID_HEADER);
put_ebml_uint (pb, EBML_ID_EBMLVERSION , 1);
put_ebml_uint (pb, EBML_ID_EBMLREADVERSION , 1);
put_ebml_uint (pb, EBML_ID_EBMLMAXIDLENGTH , 4);
put_ebml_uint (pb, EBML_ID_EBMLMAXSIZELENGTH , 8);
put_ebml_string (pb, EBML_ID_DOCTYPE , "matroska");
put_ebml_uint (pb, EBML_ID_DOCTYPEVERSION , 1);
put_ebml_uint (pb, EBML_ID_DOCTYPEREADVERSION , 1);
end_ebml_master(pb, ebml_header);
mkv->segment = start_ebml_master(pb, MATROSKA_ID_SEGMENT);
segment_info = start_ebml_master(pb, MATROSKA_ID_INFO);
put_ebml_uint(pb, MATROSKA_ID_TIMECODESCALE, 1000000);
if (strlen(s->title))
put_ebml_string(pb, MATROSKA_ID_TITLE, s->title);
if (!(s->streams[0]->codec->flags & CODEC_FLAG_BITEXACT)) {
put_ebml_string(pb, MATROSKA_ID_MUXINGAPP, LIBAVFORMAT_IDENT);
// XXX: both are required; something better for writing app?
put_ebml_string(pb, MATROSKA_ID_WRITINGAPP, LIBAVFORMAT_IDENT);
}
// XXX: segment UID and duration
end_ebml_master(pb, segment_info);
tracks = start_ebml_master(pb, MATROSKA_ID_TRACKS);
for (i = 0; i < s->nb_streams; i++) {
AVStream *st = s->streams[i];
offset_t track = start_ebml_master(pb, MATROSKA_ID_TRACKENTRY);
end_ebml_master(pb, track);
}
end_ebml_master(pb, tracks);
put_be64(pb, 0xdeadbeef);
return 0;
}
static int mkv_write_packet(AVFormatContext *s, AVPacket *pkt)
{
ByteIOContext *pb = &s->pb;
put_buffer(pb, pkt->data, pkt->size);
return 0;
}
static int mkv_write_trailer(AVFormatContext *s)
{
MatroskaMuxContext *mkv = s->priv_data;
ByteIOContext *pb = &s->pb;
end_ebml_master(pb, mkv->segment);
return 0;
}
AVOutputFormat matroska_muxer = {
"matroska",
"Matroska File Format",
"video/x-matroska",
"mkv",
sizeof(MatroskaMuxContext),
CODEC_ID_MP2,
CODEC_ID_MPEG4,
mkv_write_header,
mkv_write_packet,
mkv_write_trailer,
.codec_tag = (const AVCodecTag*[]){codec_bmp_tags, codec_wav_tags, 0},
};
|
