Commit | Line | Data |
---|---|---|
de6d9b64 FB |
1 | /* |
2 | * AVI encoder. | |
19720f15 | 3 | * Copyright (c) 2000 Fabrice Bellard. |
de6d9b64 | 4 | * |
19720f15 FB |
5 | * This library is free software; you can redistribute it and/or |
6 | * modify it under the terms of the GNU Lesser General Public | |
7 | * License as published by the Free Software Foundation; either | |
8 | * version 2 of the License, or (at your option) any later version. | |
de6d9b64 | 9 | * |
19720f15 | 10 | * This library is distributed in the hope that it will be useful, |
de6d9b64 | 11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
19720f15 FB |
12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
13 | * Lesser General Public License for more details. | |
de6d9b64 | 14 | * |
19720f15 FB |
15 | * You should have received a copy of the GNU Lesser General Public |
16 | * License along with this library; if not, write to the Free Software | |
17 | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | |
de6d9b64 | 18 | */ |
de6d9b64 FB |
19 | #include "avformat.h" |
20 | #include "avi.h" | |
21 | ||
22 | /* | |
23 | * TODO: | |
24 | * - fill all fields if non streamed (nb_frames for example) | |
25 | */ | |
26 | ||
764ef400 | 27 | #ifdef CONFIG_ENCODERS |
e738cee9 | 28 | typedef struct AVIIentry { |
de6d9b64 | 29 | unsigned int flags, pos, len; |
e738cee9 RS |
30 | } AVIIentry; |
31 | ||
32 | #define AVI_INDEX_CLUSTER_SIZE 16384 | |
33 | ||
34 | typedef struct AVIIndex { | |
35 | offset_t indx_start; | |
36 | int entry; | |
37 | int ents_allocated; | |
38 | AVIIentry** cluster; | |
de6d9b64 FB |
39 | } AVIIndex; |
40 | ||
41 | typedef struct { | |
ce9fce63 RS |
42 | offset_t riff_start, movi_list, odml_list; |
43 | offset_t frames_hdr_all, frames_hdr_strm[MAX_STREAMS]; | |
86bec9a1 | 44 | int audio_strm_length[MAX_STREAMS]; |
e738cee9 | 45 | int riff_id; |
50c3dd32 | 46 | int packet_count[MAX_STREAMS]; |
e738cee9 RS |
47 | |
48 | AVIIndex indexes[MAX_STREAMS]; | |
de6d9b64 FB |
49 | } AVIContext; |
50 | ||
e738cee9 RS |
51 | static inline AVIIentry* avi_get_ientry(AVIIndex* idx, int ent_id) |
52 | { | |
53 | int cl = ent_id / AVI_INDEX_CLUSTER_SIZE; | |
54 | int id = ent_id % AVI_INDEX_CLUSTER_SIZE; | |
55 | return &idx->cluster[cl][id]; | |
56 | } | |
57 | ||
0570bf06 | 58 | offset_t start_tag(ByteIOContext *pb, const char *tag) |
de6d9b64 FB |
59 | { |
60 | put_tag(pb, tag); | |
61 | put_le32(pb, 0); | |
62 | return url_ftell(pb); | |
63 | } | |
64 | ||
65 | void end_tag(ByteIOContext *pb, offset_t start) | |
66 | { | |
67 | offset_t pos; | |
68 | ||
69 | pos = url_ftell(pb); | |
70 | url_fseek(pb, start - 4, SEEK_SET); | |
0c1a9eda | 71 | put_le32(pb, (uint32_t)(pos - start)); |
de6d9b64 FB |
72 | url_fseek(pb, pos, SEEK_SET); |
73 | } | |
764ef400 | 74 | #endif //CONFIG_ENCODERS |
de6d9b64 FB |
75 | |
76 | /* Note: when encoding, the first matching tag is used, so order is | |
77 | important if multiple tags possible for a given codec. */ | |
0570bf06 | 78 | const CodecTag codec_bmp_tags[] = { |
cc15c931 LA |
79 | { CODEC_ID_H264, MKTAG('H', '2', '6', '4') }, |
80 | ||
95c79a24 J |
81 | { CODEC_ID_H263, MKTAG('H', '2', '6', '3') }, |
82 | { CODEC_ID_H263P, MKTAG('H', '2', '6', '3') }, | |
de6d9b64 | 83 | { CODEC_ID_H263I, MKTAG('I', '2', '6', '3') }, /* intel h263 */ |
c931e608 | 84 | { CODEC_ID_H261, MKTAG('H', '2', '6', '1') }, |
a05c8d71 MN |
85 | |
86 | /* added based on MPlayer */ | |
a05c8d71 | 87 | { CODEC_ID_H263P, MKTAG('U', '2', '6', '3') }, |
a05c8d71 MN |
88 | { CODEC_ID_H263P, MKTAG('v', 'i', 'v', '1') }, |
89 | ||
420370d4 | 90 | { CODEC_ID_MPEG4, MKTAG('F', 'M', 'P', '4')}, |
b1d89f82 | 91 | { CODEC_ID_MPEG4, MKTAG('D', 'I', 'V', 'X'), .invalid_asf = 1 }, |
b1d89f82 MR |
92 | { CODEC_ID_MPEG4, MKTAG('D', 'X', '5', '0'), .invalid_asf = 1 }, |
93 | { CODEC_ID_MPEG4, MKTAG('X', 'V', 'I', 'D'), .invalid_asf = 1 }, | |
390dffc5 MN |
94 | { CODEC_ID_MPEG4, MKTAG('M', 'P', '4', 'S') }, |
95 | { CODEC_ID_MPEG4, MKTAG('M', '4', 'S', '2') }, | |
58f26ba9 | 96 | { CODEC_ID_MPEG4, MKTAG(0x04, 0, 0, 0) }, /* some broken avi use this */ |
a05c8d71 MN |
97 | |
98 | /* added based on MPlayer */ | |
99 | { CODEC_ID_MPEG4, MKTAG('D', 'I', 'V', '1') }, | |
a05c8d71 MN |
100 | { CODEC_ID_MPEG4, MKTAG('B', 'L', 'Z', '0') }, |
101 | { CODEC_ID_MPEG4, MKTAG('m', 'p', '4', 'v') }, | |
102 | { CODEC_ID_MPEG4, MKTAG('U', 'M', 'P', '4') }, | |
2d2651f8 | 103 | { CODEC_ID_MPEG4, MKTAG('W', 'V', '1', 'F') }, |
a05c8d71 | 104 | |
b1d89f82 | 105 | { CODEC_ID_MSMPEG4V3, MKTAG('D', 'I', 'V', '3'), .invalid_asf = 1 }, /* default signature when using MSMPEG4 */ |
bb3debab | 106 | { CODEC_ID_MSMPEG4V3, MKTAG('M', 'P', '4', '3') }, |
a05c8d71 MN |
107 | |
108 | /* added based on MPlayer */ | |
109 | { CODEC_ID_MSMPEG4V3, MKTAG('M', 'P', 'G', '3') }, | |
a05c8d71 | 110 | { CODEC_ID_MSMPEG4V3, MKTAG('D', 'I', 'V', '5') }, |
a05c8d71 | 111 | { CODEC_ID_MSMPEG4V3, MKTAG('D', 'I', 'V', '6') }, |
a05c8d71 | 112 | { CODEC_ID_MSMPEG4V3, MKTAG('D', 'I', 'V', '4') }, |
a05c8d71 MN |
113 | { CODEC_ID_MSMPEG4V3, MKTAG('A', 'P', '4', '1') }, |
114 | { CODEC_ID_MSMPEG4V3, MKTAG('C', 'O', 'L', '1') }, | |
a05c8d71 | 115 | { CODEC_ID_MSMPEG4V3, MKTAG('C', 'O', 'L', '0') }, |
a05c8d71 | 116 | |
bb3debab | 117 | { CODEC_ID_MSMPEG4V2, MKTAG('M', 'P', '4', '2') }, |
a05c8d71 MN |
118 | |
119 | /* added based on MPlayer */ | |
120 | { CODEC_ID_MSMPEG4V2, MKTAG('D', 'I', 'V', '2') }, | |
a05c8d71 | 121 | |
9b030d9d | 122 | { CODEC_ID_MSMPEG4V1, MKTAG('M', 'P', 'G', '4') }, |
a05c8d71 | 123 | |
9b030d9d | 124 | { CODEC_ID_WMV1, MKTAG('W', 'M', 'V', '1') }, |
a05c8d71 MN |
125 | |
126 | /* added based on MPlayer */ | |
e23d5712 | 127 | { CODEC_ID_WMV2, MKTAG('W', 'M', 'V', '2') }, |
020fcc94 FB |
128 | { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 's', 'd') }, |
129 | { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 'h', 'd') }, | |
ce9fce63 RS |
130 | { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 's', 'l') }, |
131 | { CODEC_ID_DVVIDEO, MKTAG('d', 'v', '2', '5') }, | |
1cbc2890 FB |
132 | { CODEC_ID_MPEG1VIDEO, MKTAG('m', 'p', 'g', '1') }, |
133 | { CODEC_ID_MPEG1VIDEO, MKTAG('m', 'p', 'g', '2') }, | |
b4c64c50 | 134 | { CODEC_ID_MPEG2VIDEO, MKTAG('m', 'p', 'g', '2') }, |
1cbc2890 | 135 | { CODEC_ID_MPEG1VIDEO, MKTAG('P', 'I', 'M', '1') }, |
6b17c988 | 136 | { CODEC_ID_MPEG1VIDEO, MKTAG('V', 'C', 'R', '2') }, |
c60d6ad4 MN |
137 | { CODEC_ID_MPEG1VIDEO, 0x10000001 }, |
138 | { CODEC_ID_MPEG2VIDEO, 0x10000002 }, | |
afbb5ce4 | 139 | { CODEC_ID_MPEG2VIDEO, MKTAG('D', 'V', 'R', ' ') }, |
bb15f7fd | 140 | { CODEC_ID_MJPEG, MKTAG('M', 'J', 'P', 'G') }, |
ccafe7b1 | 141 | { CODEC_ID_MJPEG, MKTAG('L', 'J', 'P', 'G') }, |
5b9d235f | 142 | { CODEC_ID_LJPEG, MKTAG('L', 'J', 'P', 'G') }, |
ccafe7b1 | 143 | { CODEC_ID_MJPEG, MKTAG('J', 'P', 'G', 'L') }, /* Pegasus lossless JPEG */ |
f560dd82 | 144 | { CODEC_ID_HUFFYUV, MKTAG('H', 'F', 'Y', 'U') }, |
f37b9768 | 145 | { CODEC_ID_FFVHUFF, MKTAG('F', 'F', 'V', 'H') }, |
1e711bd6 | 146 | { CODEC_ID_CYUV, MKTAG('C', 'Y', 'U', 'V') }, |
e8750b00 | 147 | { CODEC_ID_RAWVIDEO, MKTAG('Y', '4', '2', '2') }, |
e8b478c1 | 148 | { CODEC_ID_RAWVIDEO, MKTAG('I', '4', '2', '0') }, |
deabd4fd MM |
149 | { CODEC_ID_INDEO3, MKTAG('I', 'V', '3', '1') }, |
150 | { CODEC_ID_INDEO3, MKTAG('I', 'V', '3', '2') }, | |
d86053a4 | 151 | { CODEC_ID_VP3, MKTAG('V', 'P', '3', '1') }, |
9ca9fce1 | 152 | { CODEC_ID_ASV1, MKTAG('A', 'S', 'V', '1') }, |
8d876a43 | 153 | { CODEC_ID_ASV2, MKTAG('A', 'S', 'V', '2') }, |
be3564ed | 154 | { CODEC_ID_VCR1, MKTAG('V', 'C', 'R', '1') }, |
7408ad05 | 155 | { CODEC_ID_FFV1, MKTAG('F', 'F', 'V', '1') }, |
9e8cd0db | 156 | { CODEC_ID_XAN_WC4, MKTAG('X', 'x', 'a', 'n') }, |
2fdf638b MM |
157 | { CODEC_ID_MSRLE, MKTAG('m', 'r', 'l', 'e') }, |
158 | { CODEC_ID_MSRLE, MKTAG(0x1, 0x0, 0x0, 0x0) }, | |
159 | { CODEC_ID_MSVIDEO1, MKTAG('M', 'S', 'V', 'C') }, | |
160 | { CODEC_ID_MSVIDEO1, MKTAG('m', 's', 'v', 'c') }, | |
161 | { CODEC_ID_MSVIDEO1, MKTAG('C', 'R', 'A', 'M') }, | |
162 | { CODEC_ID_MSVIDEO1, MKTAG('c', 'r', 'a', 'm') }, | |
163 | { CODEC_ID_MSVIDEO1, MKTAG('W', 'H', 'A', 'M') }, | |
164 | { CODEC_ID_MSVIDEO1, MKTAG('w', 'h', 'a', 'm') }, | |
165 | { CODEC_ID_CINEPAK, MKTAG('c', 'v', 'i', 'd') }, | |
b7cbfc23 | 166 | { CODEC_ID_TRUEMOTION1, MKTAG('D', 'U', 'C', 'K') }, |
a273bbfb RT |
167 | { CODEC_ID_MSZH, MKTAG('M', 'S', 'Z', 'H') }, |
168 | { CODEC_ID_ZLIB, MKTAG('Z', 'L', 'I', 'B') }, | |
791e7b83 | 169 | { CODEC_ID_SNOW, MKTAG('S', 'N', 'O', 'W') }, |
7585aa5c | 170 | { CODEC_ID_4XM, MKTAG('4', 'X', 'M', 'V') }, |
ee777235 | 171 | { CODEC_ID_FLV1, MKTAG('F', 'L', 'V', '1') }, |
d9171299 | 172 | { CODEC_ID_SVQ1, MKTAG('s', 'v', 'q', '1') }, |
9d53d58e | 173 | { CODEC_ID_TSCC, MKTAG('t', 's', 'c', 'c') }, |
d0a0bbd2 | 174 | { CODEC_ID_ULTI, MKTAG('U', 'L', 'T', 'I') }, |
ab711b3c | 175 | { CODEC_ID_VIXL, MKTAG('V', 'I', 'X', 'L') }, |
acfd8f0f MM |
176 | { CODEC_ID_QPEG, MKTAG('Q', 'P', 'E', 'G') }, |
177 | { CODEC_ID_QPEG, MKTAG('Q', '1', '.', '0') }, | |
178 | { CODEC_ID_QPEG, MKTAG('Q', '1', '.', '1') }, | |
b636365a | 179 | { CODEC_ID_WMV3, MKTAG('W', 'M', 'V', '3') }, |
9c7fb608 | 180 | { CODEC_ID_LOCO, MKTAG('L', 'O', 'C', 'O') }, |
a8a15e9d | 181 | { CODEC_ID_WNV1, MKTAG('W', 'N', 'V', '1') }, |
589f8220 | 182 | { CODEC_ID_AASC, MKTAG('A', 'A', 'S', 'C') }, |
856dbbff | 183 | { CODEC_ID_INDEO2, MKTAG('R', 'T', '2', '1') }, |
627c50b7 | 184 | { CODEC_ID_RAWVIDEO, 0 }, |
de6d9b64 FB |
185 | { 0, 0 }, |
186 | }; | |
187 | ||
a266644f | 188 | unsigned int codec_get_tag(const CodecTag *tags, int id) |
de6d9b64 FB |
189 | { |
190 | while (tags->id != 0) { | |
191 | if (tags->id == id) | |
192 | return tags->tag; | |
193 | tags++; | |
194 | } | |
195 | return 0; | |
196 | } | |
197 | ||
7fea94ce | 198 | static unsigned int codec_get_asf_tag(const CodecTag *tags, unsigned int id) |
2727c35e PG |
199 | { |
200 | while (tags->id != 0) { | |
201 | if (!tags->invalid_asf && tags->id == id) | |
202 | return tags->tag; | |
203 | tags++; | |
204 | } | |
205 | return 0; | |
206 | } | |
207 | ||
5d234974 | 208 | enum CodecID codec_get_id(const CodecTag *tags, unsigned int tag) |
de6d9b64 FB |
209 | { |
210 | while (tags->id != 0) { | |
77efa159 MN |
211 | if( toupper((tag >> 0)&0xFF) == toupper((tags->tag >> 0)&0xFF) |
212 | && toupper((tag >> 8)&0xFF) == toupper((tags->tag >> 8)&0xFF) | |
213 | && toupper((tag >>16)&0xFF) == toupper((tags->tag >>16)&0xFF) | |
214 | && toupper((tag >>24)&0xFF) == toupper((tags->tag >>24)&0xFF)) | |
de6d9b64 FB |
215 | return tags->id; |
216 | tags++; | |
217 | } | |
5d234974 | 218 | return CODEC_ID_NONE; |
de6d9b64 FB |
219 | } |
220 | ||
221 | unsigned int codec_get_bmp_tag(int id) | |
222 | { | |
223 | return codec_get_tag(codec_bmp_tags, id); | |
224 | } | |
225 | ||
3aa180b8 AB |
226 | unsigned int codec_get_wav_tag(int id) |
227 | { | |
228 | return codec_get_tag(codec_wav_tags, id); | |
229 | } | |
230 | ||
231 | enum CodecID codec_get_bmp_id(unsigned int tag) | |
232 | { | |
233 | return codec_get_id(codec_bmp_tags, tag); | |
234 | } | |
235 | ||
236 | enum CodecID codec_get_wav_id(unsigned int tag) | |
237 | { | |
238 | return codec_get_id(codec_wav_tags, tag); | |
239 | } | |
240 | ||
764ef400 | 241 | #ifdef CONFIG_ENCODERS |
de6d9b64 | 242 | /* BITMAPINFOHEADER header */ |
0570bf06 | 243 | void put_bmp_header(ByteIOContext *pb, AVCodecContext *enc, const CodecTag *tags, int for_asf) |
de6d9b64 | 244 | { |
952c69c4 | 245 | put_le32(pb, 40 + enc->extradata_size); /* size */ |
de6d9b64 FB |
246 | put_le32(pb, enc->width); |
247 | put_le32(pb, enc->height); | |
248 | put_le16(pb, 1); /* planes */ | |
952c69c4 MN |
249 | |
250 | put_le16(pb, enc->bits_per_sample ? enc->bits_per_sample : 24); /* depth */ | |
de6d9b64 | 251 | /* compression type */ |
649b918c | 252 | put_le32(pb, for_asf ? (enc->codec_tag ? enc->codec_tag : codec_get_asf_tag(tags, enc->codec_id)) : enc->codec_tag); // |
de6d9b64 FB |
253 | put_le32(pb, enc->width * enc->height * 3); |
254 | put_le32(pb, 0); | |
255 | put_le32(pb, 0); | |
256 | put_le32(pb, 0); | |
257 | put_le32(pb, 0); | |
952c69c4 MN |
258 | |
259 | put_buffer(pb, enc->extradata, enc->extradata_size); | |
260 | ||
261 | if (enc->extradata_size & 1) | |
262 | put_byte(pb, 0); | |
de6d9b64 FB |
263 | } |
264 | ||
359fa0fe | 265 | static void parse_specific_params(AVCodecContext *stream, int *au_rate, int *au_ssize, int *au_scale) |
86bec9a1 | 266 | { |
359fa0fe MN |
267 | int gcd; |
268 | ||
269 | *au_ssize= stream->block_align; | |
270 | if(stream->frame_size && stream->sample_rate){ | |
271 | *au_scale=stream->frame_size; | |
272 | *au_rate= stream->sample_rate; | |
273 | }else if(stream->codec_type == CODEC_TYPE_VIDEO){ | |
274 | *au_scale= stream->frame_rate_base; | |
275 | *au_rate = stream->frame_rate; | |
276 | }else{ | |
277 | *au_scale= stream->block_align ? stream->block_align*8 : 8; | |
278 | *au_rate = stream->bit_rate; | |
86bec9a1 | 279 | } |
359fa0fe MN |
280 | gcd= ff_gcd(*au_scale, *au_rate); |
281 | *au_scale /= gcd; | |
282 | *au_rate /= gcd; | |
86bec9a1 J |
283 | } |
284 | ||
ce9fce63 RS |
285 | static offset_t avi_start_new_riff(AVIContext *avi, ByteIOContext *pb, |
286 | const char* riff_tag, const char* list_tag) | |
287 | { | |
288 | offset_t loff; | |
e738cee9 RS |
289 | int i; |
290 | ||
291 | avi->riff_id++; | |
292 | for (i=0; i<MAX_STREAMS; i++) | |
293 | avi->indexes[i].entry = 0; | |
294 | ||
ce9fce63 RS |
295 | avi->riff_start = start_tag(pb, "RIFF"); |
296 | put_tag(pb, riff_tag); | |
297 | loff = start_tag(pb, "LIST"); | |
298 | put_tag(pb, list_tag); | |
299 | return loff; | |
300 | } | |
301 | ||
e738cee9 RS |
302 | static unsigned char* avi_stream2fourcc(unsigned char* tag, int index, |
303 | enum CodecType type) | |
304 | { | |
305 | tag[0] = '0'; | |
306 | tag[1] = '0' + index; | |
307 | if (type == CODEC_TYPE_VIDEO) { | |
308 | tag[2] = 'd'; | |
309 | tag[3] = 'c'; | |
310 | } else { | |
311 | tag[2] = 'w'; | |
312 | tag[3] = 'b'; | |
313 | } | |
314 | tag[4] = '\0'; | |
315 | return tag; | |
316 | } | |
317 | ||
de6d9b64 FB |
318 | static int avi_write_header(AVFormatContext *s) |
319 | { | |
c9a65ca8 | 320 | AVIContext *avi = s->priv_data; |
de6d9b64 | 321 | ByteIOContext *pb = &s->pb; |
86bec9a1 | 322 | int bitrate, n, i, nb_frames, au_byterate, au_ssize, au_scale; |
de6d9b64 FB |
323 | AVCodecContext *stream, *video_enc; |
324 | offset_t list1, list2, strh, strf; | |
325 | ||
de6d9b64 | 326 | /* header list */ |
e738cee9 | 327 | avi->riff_id = 0; |
ce9fce63 | 328 | list1 = avi_start_new_riff(avi, pb, "AVI ", "hdrl"); |
de6d9b64 FB |
329 | |
330 | /* avi header */ | |
331 | put_tag(pb, "avih"); | |
332 | put_le32(pb, 14 * 4); | |
333 | bitrate = 0; | |
334 | ||
335 | video_enc = NULL; | |
336 | for(n=0;n<s->nb_streams;n++) { | |
337 | stream = &s->streams[n]->codec; | |
338 | bitrate += stream->bit_rate; | |
339 | if (stream->codec_type == CODEC_TYPE_VIDEO) | |
340 | video_enc = stream; | |
341 | } | |
342 | ||
de6d9b64 FB |
343 | nb_frames = 0; |
344 | ||
850742d7 | 345 | if(video_enc){ |
14bea432 | 346 | put_le32(pb, (uint32_t)(int64_t_C(1000000) * video_enc->frame_rate_base / video_enc->frame_rate)); |
850742d7 AR |
347 | } else { |
348 | put_le32(pb, 0); | |
349 | } | |
de6d9b64 FB |
350 | put_le32(pb, bitrate / 8); /* XXX: not quite exact */ |
351 | put_le32(pb, 0); /* padding */ | |
352 | put_le32(pb, AVIF_TRUSTCKTYPE | AVIF_HASINDEX | AVIF_ISINTERLEAVED); /* flags */ | |
86bec9a1 | 353 | avi->frames_hdr_all = url_ftell(pb); /* remember this offset to fill later */ |
de6d9b64 FB |
354 | put_le32(pb, nb_frames); /* nb frames, filled later */ |
355 | put_le32(pb, 0); /* initial frame */ | |
356 | put_le32(pb, s->nb_streams); /* nb streams */ | |
357 | put_le32(pb, 1024 * 1024); /* suggested buffer size */ | |
850742d7 | 358 | if(video_enc){ |
de6d9b64 FB |
359 | put_le32(pb, video_enc->width); |
360 | put_le32(pb, video_enc->height); | |
850742d7 AR |
361 | } else { |
362 | put_le32(pb, 0); | |
363 | put_le32(pb, 0); | |
364 | } | |
de6d9b64 FB |
365 | put_le32(pb, 0); /* reserved */ |
366 | put_le32(pb, 0); /* reserved */ | |
367 | put_le32(pb, 0); /* reserved */ | |
368 | put_le32(pb, 0); /* reserved */ | |
369 | ||
370 | /* stream list */ | |
371 | for(i=0;i<n;i++) { | |
372 | list2 = start_tag(pb, "LIST"); | |
373 | put_tag(pb, "strl"); | |
374 | ||
375 | stream = &s->streams[i]->codec; | |
376 | ||
e8750b00 FR |
377 | /* FourCC should really be set by the codec itself */ |
378 | if (! stream->codec_tag) { | |
379 | stream->codec_tag = codec_get_bmp_tag(stream->codec_id); | |
380 | } | |
381 | ||
de6d9b64 FB |
382 | /* stream generic header */ |
383 | strh = start_tag(pb, "strh"); | |
384 | switch(stream->codec_type) { | |
385 | case CODEC_TYPE_VIDEO: | |
386 | put_tag(pb, "vids"); | |
e8750b00 | 387 | put_le32(pb, stream->codec_tag); |
de6d9b64 FB |
388 | put_le32(pb, 0); /* flags */ |
389 | put_le16(pb, 0); /* priority */ | |
390 | put_le16(pb, 0); /* language */ | |
391 | put_le32(pb, 0); /* initial frame */ | |
b559b29b | 392 | |
14bea432 MN |
393 | put_le32(pb, stream->frame_rate_base); /* scale */ |
394 | put_le32(pb, stream->frame_rate); /* rate */ | |
50c3dd32 | 395 | av_set_pts_info(s->streams[i], 64, stream->frame_rate_base, stream->frame_rate); |
b559b29b | 396 | |
de6d9b64 | 397 | put_le32(pb, 0); /* start */ |
86bec9a1 | 398 | avi->frames_hdr_strm[i] = url_ftell(pb); /* remember this offset to fill later */ |
de6d9b64 FB |
399 | put_le32(pb, nb_frames); /* length, XXX: fill later */ |
400 | put_le32(pb, 1024 * 1024); /* suggested buffer size */ | |
86bec9a1 | 401 | put_le32(pb, -1); /* quality */ |
4a7d2651 | 402 | put_le32(pb, 0); /* sample size */ |
de6d9b64 FB |
403 | put_le16(pb, 0); |
404 | put_le16(pb, 0); | |
405 | put_le16(pb, stream->width); | |
406 | put_le16(pb, stream->height); | |
407 | break; | |
408 | case CODEC_TYPE_AUDIO: | |
409 | put_tag(pb, "auds"); | |
86bec9a1 | 410 | put_le32(pb, 1); /* tag */ |
de6d9b64 FB |
411 | put_le32(pb, 0); /* flags */ |
412 | put_le16(pb, 0); /* priority */ | |
413 | put_le16(pb, 0); /* language */ | |
414 | put_le32(pb, 0); /* initial frame */ | |
86bec9a1 J |
415 | parse_specific_params(stream, &au_byterate, &au_ssize, &au_scale); |
416 | put_le32(pb, au_scale); /* scale */ | |
417 | put_le32(pb, au_byterate); /* rate */ | |
b6c7f870 | 418 | av_set_pts_info(s->streams[i], 64, au_scale, au_byterate); |
de6d9b64 | 419 | put_le32(pb, 0); /* start */ |
86bec9a1 | 420 | avi->frames_hdr_strm[i] = url_ftell(pb); /* remember this offset to fill later */ |
de6d9b64 FB |
421 | put_le32(pb, 0); /* length, XXX: filled later */ |
422 | put_le32(pb, 12 * 1024); /* suggested buffer size */ | |
423 | put_le32(pb, -1); /* quality */ | |
86bec9a1 | 424 | put_le32(pb, au_ssize); /* sample size */ |
de6d9b64 FB |
425 | put_le32(pb, 0); |
426 | put_le32(pb, 0); | |
427 | break; | |
53cafac0 | 428 | default: |
71c32f19 | 429 | return -1; |
de6d9b64 FB |
430 | } |
431 | end_tag(pb, strh); | |
432 | ||
433 | strf = start_tag(pb, "strf"); | |
434 | switch(stream->codec_type) { | |
435 | case CODEC_TYPE_VIDEO: | |
2727c35e | 436 | put_bmp_header(pb, stream, codec_bmp_tags, 0); |
de6d9b64 FB |
437 | break; |
438 | case CODEC_TYPE_AUDIO: | |
46a3d068 | 439 | if (put_wav_header(pb, stream) < 0) { |
1ea4f593 | 440 | av_free(avi); |
46a3d068 FB |
441 | return -1; |
442 | } | |
de6d9b64 | 443 | break; |
53cafac0 | 444 | default: |
71c32f19 | 445 | return -1; |
de6d9b64 FB |
446 | } |
447 | end_tag(pb, strf); | |
e738cee9 RS |
448 | |
449 | if (!url_is_streamed(pb)) { | |
450 | unsigned char tag[5]; | |
8c0ff5d5 | 451 | int j; |
e738cee9 RS |
452 | |
453 | /* Starting to lay out AVI OpenDML master index. | |
454 | * We want to make it JUNK entry for now, since we'd | |
8c0ff5d5 | 455 | * like to get away without making AVI an OpenDML one |
e738cee9 RS |
456 | * for compatibility reasons. |
457 | */ | |
458 | avi->indexes[i].entry = avi->indexes[i].ents_allocated = 0; | |
459 | avi->indexes[i].indx_start = start_tag(pb, "JUNK"); | |
460 | put_le16(pb, 4); /* wLongsPerEntry */ | |
461 | put_byte(pb, 0); /* bIndexSubType (0 == frame index) */ | |
462 | put_byte(pb, 0); /* bIndexType (0 == AVI_INDEX_OF_INDEXES) */ | |
463 | put_le32(pb, 0); /* nEntriesInUse (will fill out later on) */ | |
464 | put_tag(pb, avi_stream2fourcc(&tag[0], i, stream->codec_type)); | |
465 | /* dwChunkId */ | |
466 | put_le64(pb, 0); /* dwReserved[3] | |
467 | put_le32(pb, 0); Must be 0. */ | |
8c0ff5d5 RS |
468 | for (j=0; j < AVI_MASTER_INDEX_SIZE * 2; j++) |
469 | put_le64(pb, 0); | |
e738cee9 RS |
470 | end_tag(pb, avi->indexes[i].indx_start); |
471 | } | |
472 | ||
de6d9b64 FB |
473 | end_tag(pb, list2); |
474 | } | |
ce9fce63 | 475 | |
e738cee9 RS |
476 | if (!url_is_streamed(pb)) { |
477 | /* AVI could become an OpenDML one, if it grows beyond 2Gb range */ | |
478 | avi->odml_list = start_tag(pb, "JUNK"); | |
479 | put_tag(pb, "odml"); | |
480 | put_tag(pb, "dmlh"); | |
481 | put_le32(pb, 248); | |
482 | for (i = 0; i < 248; i+= 4) | |
483 | put_le32(pb, 0); | |
484 | end_tag(pb, avi->odml_list); | |
485 | } | |
de6d9b64 FB |
486 | |
487 | end_tag(pb, list1); | |
488 | ||
489 | avi->movi_list = start_tag(pb, "LIST"); | |
de6d9b64 FB |
490 | put_tag(pb, "movi"); |
491 | ||
492 | put_flush_packet(pb); | |
493 | ||
494 | return 0; | |
495 | } | |
496 | ||
e738cee9 RS |
497 | static int avi_write_ix(AVFormatContext *s) |
498 | { | |
499 | ByteIOContext *pb = &s->pb; | |
500 | AVIContext *avi = s->priv_data; | |
501 | unsigned char tag[5]; | |
502 | unsigned char ix_tag[] = "ix00"; | |
503 | int i, j; | |
504 | ||
505 | if (avi->riff_id > AVI_MASTER_INDEX_SIZE) | |
506 | return -1; | |
507 | ||
508 | for (i=0;i<s->nb_streams;i++) { | |
509 | offset_t ix, pos; | |
510 | ||
511 | avi_stream2fourcc(&tag[0], i, s->streams[i]->codec.codec_type); | |
512 | ix_tag[3] = '0' + i; | |
513 | ||
514 | /* Writing AVI OpenDML leaf index chunk */ | |
515 | ix = url_ftell(pb); | |
516 | put_tag(pb, &ix_tag[0]); /* ix?? */ | |
517 | put_le32(pb, avi->indexes[i].entry * 8 + 24); | |
518 | /* chunk size */ | |
519 | put_le16(pb, 2); /* wLongsPerEntry */ | |
520 | put_byte(pb, 0); /* bIndexSubType (0 == frame index) */ | |
521 | put_byte(pb, 1); /* bIndexType (1 == AVI_INDEX_OF_CHUNKS) */ | |
522 | put_le32(pb, avi->indexes[i].entry); | |
523 | /* nEntriesInUse */ | |
524 | put_tag(pb, &tag[0]); /* dwChunkId */ | |
525 | put_le64(pb, avi->movi_list);/* qwBaseOffset */ | |
526 | put_le32(pb, 0); /* dwReserved_3 (must be 0) */ | |
527 | ||
528 | for (j=0; j<avi->indexes[i].entry; j++) { | |
529 | AVIIentry* ie = avi_get_ientry(&avi->indexes[i], j); | |
530 | put_le32(pb, ie->pos + 8); | |
531 | put_le32(pb, ((uint32_t)ie->len & ~0x80000000) | | |
532 | (ie->flags & 0x10 ? 0 : 0x80000000)); | |
533 | } | |
534 | put_flush_packet(pb); | |
535 | pos = url_ftell(pb); | |
536 | ||
537 | /* Updating one entry in the AVI OpenDML master index */ | |
538 | url_fseek(pb, avi->indexes[i].indx_start - 8, SEEK_SET); | |
539 | put_tag(pb, "indx"); /* enabling this entry */ | |
540 | url_fskip(pb, 8); | |
541 | put_le32(pb, avi->riff_id); /* nEntriesInUse */ | |
542 | url_fskip(pb, 16*avi->riff_id); | |
543 | put_le64(pb, ix); /* qwOffset */ | |
544 | put_le32(pb, pos - ix); /* dwSize */ | |
545 | put_le32(pb, avi->indexes[i].entry); /* dwDuration */ | |
546 | ||
547 | url_fseek(pb, pos, SEEK_SET); | |
548 | } | |
549 | return 0; | |
550 | } | |
551 | ||
552 | static int avi_write_idx1(AVFormatContext *s) | |
ce9fce63 RS |
553 | { |
554 | ByteIOContext *pb = &s->pb; | |
555 | AVIContext *avi = s->priv_data; | |
556 | offset_t file_size, idx_chunk; | |
e738cee9 | 557 | int i, n, nb_frames, au_byterate, au_ssize, au_scale; |
ce9fce63 | 558 | AVCodecContext *stream; |
e738cee9 | 559 | unsigned char tag[5]; |
ce9fce63 RS |
560 | |
561 | if (!url_is_streamed(pb)) { | |
546031ee | 562 | AVIIentry* ie = 0, *tie; |
e738cee9 | 563 | int entry[MAX_STREAMS]; |
546031ee | 564 | int empty, stream_id = -1; |
e738cee9 RS |
565 | |
566 | idx_chunk = start_tag(pb, "idx1"); | |
567 | memset(&entry[0], 0, sizeof(entry)); | |
568 | do { | |
569 | empty = 1; | |
570 | for (i=0; i<s->nb_streams; i++) { | |
571 | if (avi->indexes[i].entry <= entry[i]) | |
572 | continue; | |
573 | ||
574 | tie = avi_get_ientry(&avi->indexes[i], entry[i]); | |
575 | if (empty || tie->pos < ie->pos) { | |
576 | ie = tie; | |
577 | stream_id = i; | |
578 | } | |
579 | empty = 0; | |
580 | } | |
581 | if (!empty) { | |
582 | avi_stream2fourcc(&tag[0], stream_id, | |
583 | s->streams[stream_id]->codec.codec_type); | |
584 | put_tag(pb, &tag[0]); | |
585 | put_le32(pb, ie->flags); | |
586 | put_le32(pb, ie->pos); | |
587 | put_le32(pb, ie->len); | |
588 | entry[stream_id]++; | |
589 | } | |
590 | } while (!empty); | |
591 | end_tag(pb, idx_chunk); | |
ce9fce63 RS |
592 | |
593 | /* Fill in frame/sample counters */ | |
e738cee9 | 594 | file_size = url_ftell(pb); |
ce9fce63 RS |
595 | nb_frames = 0; |
596 | for(n=0;n<s->nb_streams;n++) { | |
597 | if (avi->frames_hdr_strm[n] != 0) { | |
598 | stream = &s->streams[n]->codec; | |
599 | url_fseek(pb, avi->frames_hdr_strm[n], SEEK_SET); | |
359fa0fe MN |
600 | parse_specific_params(stream, &au_byterate, &au_ssize, &au_scale); |
601 | if (au_ssize == 0) { | |
602 | put_le32(pb, stream->frame_number); | |
603 | nb_frames += stream->frame_number; | |
ce9fce63 | 604 | } else { |
359fa0fe | 605 | put_le32(pb, avi->audio_strm_length[n] / au_ssize); |
ce9fce63 RS |
606 | } |
607 | } | |
608 | } | |
609 | if (avi->frames_hdr_all != 0) { | |
610 | url_fseek(pb, avi->frames_hdr_all, SEEK_SET); | |
611 | put_le32(pb, nb_frames); | |
612 | } | |
613 | url_fseek(pb, file_size, SEEK_SET); | |
614 | } | |
615 | return 0; | |
616 | } | |
617 | ||
e928649b | 618 | static int avi_write_packet(AVFormatContext *s, AVPacket *pkt) |
de6d9b64 FB |
619 | { |
620 | AVIContext *avi = s->priv_data; | |
621 | ByteIOContext *pb = &s->pb; | |
de6d9b64 | 622 | unsigned char tag[5]; |
e928649b | 623 | unsigned int flags=0; |
e928649b | 624 | const int stream_index= pkt->stream_index; |
4904d6c2 | 625 | AVCodecContext *enc= &s->streams[stream_index]->codec; |
e928649b | 626 | int size= pkt->size; |
ce9fce63 | 627 | |
50c3dd32 | 628 | // av_log(s, AV_LOG_DEBUG, "%lld %d %d\n", pkt->dts, avi->packet_count[stream_index], stream_index); |
16543599 | 629 | while(enc->block_align==0 && pkt->dts != AV_NOPTS_VALUE && pkt->dts > avi->packet_count[stream_index]){ |
50c3dd32 MN |
630 | AVPacket empty_packet; |
631 | ||
632 | av_init_packet(&empty_packet); | |
633 | empty_packet.size= 0; | |
634 | empty_packet.data= NULL; | |
635 | empty_packet.stream_index= stream_index; | |
636 | avi_write_packet(s, &empty_packet); | |
637 | // av_log(s, AV_LOG_DEBUG, "dup %lld %d\n", pkt->dts, avi->packet_count[stream_index]); | |
638 | } | |
639 | avi->packet_count[stream_index]++; | |
640 | ||
ce9fce63 | 641 | if (url_ftell(pb) - avi->riff_start > AVI_MAX_RIFF_SIZE) { |
e738cee9 RS |
642 | avi_write_ix(s); |
643 | end_tag(pb, avi->movi_list); | |
644 | ||
645 | if (avi->riff_id == 1) | |
646 | avi_write_idx1(s); | |
647 | ||
648 | end_tag(pb, avi->riff_start); | |
ce9fce63 RS |
649 | avi->movi_list = avi_start_new_riff(avi, pb, "AVIX", "movi"); |
650 | } | |
de6d9b64 | 651 | |
e738cee9 | 652 | avi_stream2fourcc(&tag[0], stream_index, enc->codec_type); |
e928649b MN |
653 | if(pkt->flags&PKT_FLAG_KEY) |
654 | flags = 0x10; | |
e738cee9 | 655 | if (enc->codec_type == CODEC_TYPE_AUDIO) { |
86bec9a1 | 656 | avi->audio_strm_length[stream_index] += size; |
e928649b | 657 | } |
de6d9b64 FB |
658 | |
659 | if (!url_is_streamed(&s->pb)) { | |
e738cee9 RS |
660 | AVIIndex* idx = &avi->indexes[stream_index]; |
661 | int cl = idx->entry / AVI_INDEX_CLUSTER_SIZE; | |
662 | int id = idx->entry % AVI_INDEX_CLUSTER_SIZE; | |
663 | if (idx->ents_allocated <= idx->entry) { | |
664 | idx->cluster = av_realloc(idx->cluster, (cl+1)*sizeof(void*)); | |
665 | if (!idx->cluster) | |
666 | return -1; | |
667 | idx->cluster[cl] = av_malloc(AVI_INDEX_CLUSTER_SIZE*sizeof(AVIIentry)); | |
668 | if (!idx->cluster[cl]) | |
669 | return -1; | |
670 | idx->ents_allocated += AVI_INDEX_CLUSTER_SIZE; | |
671 | } | |
672 | ||
673 | idx->cluster[cl][id].flags = flags; | |
674 | idx->cluster[cl][id].pos = url_ftell(pb) - avi->movi_list; | |
675 | idx->cluster[cl][id].len = size; | |
676 | idx->entry++; | |
de6d9b64 FB |
677 | } |
678 | ||
679 | put_buffer(pb, tag, 4); | |
680 | put_le32(pb, size); | |
e928649b | 681 | put_buffer(pb, pkt->data, size); |
de6d9b64 FB |
682 | if (size & 1) |
683 | put_byte(pb, 0); | |
684 | ||
685 | put_flush_packet(pb); | |
686 | return 0; | |
687 | } | |
688 | ||
689 | static int avi_write_trailer(AVFormatContext *s) | |
690 | { | |
de6d9b64 | 691 | AVIContext *avi = s->priv_data; |
ce9fce63 RS |
692 | ByteIOContext *pb = &s->pb; |
693 | int res = 0; | |
e738cee9 RS |
694 | int i, j, n, nb_frames; |
695 | offset_t file_size; | |
696 | ||
697 | if (avi->riff_id == 1) { | |
698 | end_tag(pb, avi->movi_list); | |
699 | res = avi_write_idx1(s); | |
700 | end_tag(pb, avi->riff_start); | |
701 | } else { | |
702 | avi_write_ix(s); | |
703 | end_tag(pb, avi->movi_list); | |
704 | end_tag(pb, avi->riff_start); | |
de6d9b64 | 705 | |
de6d9b64 | 706 | file_size = url_ftell(pb); |
ce9fce63 RS |
707 | url_fseek(pb, avi->odml_list - 8, SEEK_SET); |
708 | put_tag(pb, "LIST"); /* Making this AVI OpenDML one */ | |
709 | url_fskip(pb, 16); | |
710 | ||
711 | for (n=nb_frames=0;n<s->nb_streams;n++) { | |
712 | AVCodecContext *stream = &s->streams[n]->codec; | |
713 | if (stream->codec_type == CODEC_TYPE_VIDEO) { | |
714 | if (nb_frames < stream->frame_number) | |
715 | nb_frames = stream->frame_number; | |
716 | } else { | |
80783dc2 | 717 | if (stream->codec_id == CODEC_ID_MP2 || stream->codec_id == CODEC_ID_MP3) { |
ce9fce63 | 718 | nb_frames += stream->frame_number; |
86bec9a1 J |
719 | } |
720 | } | |
ce9fce63 RS |
721 | } |
722 | put_le32(pb, nb_frames); | |
ce9fce63 | 723 | url_fseek(pb, file_size, SEEK_SET); |
e738cee9 | 724 | } |
de6d9b64 | 725 | put_flush_packet(pb); |
e738cee9 RS |
726 | |
727 | for (i=0; i<MAX_STREAMS; i++) { | |
728 | for (j=0; j<avi->indexes[i].ents_allocated/AVI_INDEX_CLUSTER_SIZE; j++) | |
729 | av_free(avi->indexes[i].cluster[j]); | |
730 | av_free(avi->indexes[i].cluster); | |
731 | avi->indexes[i].cluster = NULL; | |
732 | avi->indexes[i].ents_allocated = avi->indexes[i].entry = 0; | |
733 | } | |
734 | ||
ce9fce63 | 735 | return res; |
de6d9b64 FB |
736 | } |
737 | ||
c9a65ca8 | 738 | static AVOutputFormat avi_oformat = { |
de6d9b64 FB |
739 | "avi", |
740 | "avi format", | |
741 | "video/x-msvideo", | |
742 | "avi", | |
c9a65ca8 | 743 | sizeof(AVIContext), |
de6d9b64 | 744 | CODEC_ID_MP2, |
22cf0455 | 745 | CODEC_ID_MPEG4, |
de6d9b64 FB |
746 | avi_write_header, |
747 | avi_write_packet, | |
748 | avi_write_trailer, | |
de6d9b64 | 749 | }; |
c9a65ca8 FB |
750 | |
751 | int avienc_init(void) | |
752 | { | |
753 | av_register_output_format(&avi_oformat); | |
754 | return 0; | |
755 | } | |
764ef400 | 756 | #endif //CONFIG_ENCODERS |