Mercurial > audlegacy-plugins
comparison src/ffmpeg/libavformat/asf-enc.c @ 808:e8776388b02a trunk
[svn] - add ffmpeg
author | nenolod |
---|---|
date | Mon, 12 Mar 2007 11:18:54 -0700 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
807:0f9c8d4d3ac4 | 808:e8776388b02a |
---|---|
1 /* | |
2 * Adaptive stream format encoder | |
3 * Copyright (c) 2000, 2001 Fabrice Bellard. | |
4 * | |
5 * This file is part of FFmpeg. | |
6 * | |
7 * FFmpeg is free software; you can redistribute it and/or | |
8 * modify it under the terms of the GNU Lesser General Public | |
9 * License as published by the Free Software Foundation; either | |
10 * version 2.1 of the License, or (at your option) any later version. | |
11 * | |
12 * FFmpeg is distributed in the hope that it will be useful, | |
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 * Lesser General Public License for more details. | |
16 * | |
17 * You should have received a copy of the GNU Lesser General Public | |
18 * License along with FFmpeg; if not, write to the Free Software | |
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
20 */ | |
21 #include "avformat.h" | |
22 #include "riff.h" | |
23 #include "asf.h" | |
24 | |
25 #undef NDEBUG | |
26 #include <assert.h> | |
27 | |
28 | |
29 #define ASF_INDEXED_INTERVAL 10000000 | |
30 #define ASF_INDEX_BLOCK 600 | |
31 | |
32 #define ASF_PACKET_ERROR_CORRECTION_DATA_SIZE 0x2 | |
33 #define ASF_PACKET_ERROR_CORRECTION_FLAGS (\ | |
34 ASF_PACKET_FLAG_ERROR_CORRECTION_PRESENT | \ | |
35 ASF_PACKET_ERROR_CORRECTION_DATA_SIZE\ | |
36 ) | |
37 | |
38 #if (ASF_PACKET_ERROR_CORRECTION_FLAGS != 0) | |
39 # define ASF_PACKET_ERROR_CORRECTION_FLAGS_FIELD_SIZE 1 | |
40 #else | |
41 # define ASF_PACKET_ERROR_CORRECTION_FLAGS_FIELD_SIZE 0 | |
42 #endif | |
43 | |
44 #define ASF_PPI_PROPERTY_FLAGS (\ | |
45 ASF_PL_FLAG_REPLICATED_DATA_LENGTH_FIELD_IS_BYTE | \ | |
46 ASF_PL_FLAG_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_IS_DWORD | \ | |
47 ASF_PL_FLAG_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_IS_BYTE | \ | |
48 ASF_PL_FLAG_STREAM_NUMBER_LENGTH_FIELD_IS_BYTE \ | |
49 ) | |
50 | |
51 #define ASF_PPI_LENGTH_TYPE_FLAGS 0 | |
52 | |
53 #define ASF_PAYLOAD_FLAGS ASF_PL_FLAG_PAYLOAD_LENGTH_FIELD_IS_WORD | |
54 | |
55 #if (ASF_PPI_FLAG_SEQUENCE_FIELD_IS_BYTE == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_SEQUENCE_FIELD_SIZE)) | |
56 # define ASF_PPI_SEQUENCE_FIELD_SIZE 1 | |
57 #endif | |
58 #if (ASF_PPI_FLAG_SEQUENCE_FIELD_IS_WORD == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_SEQUENCE_FIELD_SIZE)) | |
59 # define ASF_PPI_SEQUENCE_FIELD_SIZE 2 | |
60 #endif | |
61 #if (ASF_PPI_FLAG_SEQUENCE_FIELD_IS_DWORD == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_SEQUENCE_FIELD_SIZE)) | |
62 # define ASF_PPI_SEQUENCE_FIELD_SIZE 4 | |
63 #endif | |
64 #ifndef ASF_PPI_SEQUENCE_FIELD_SIZE | |
65 # define ASF_PPI_SEQUENCE_FIELD_SIZE 0 | |
66 #endif | |
67 | |
68 | |
69 #if (ASF_PPI_FLAG_PACKET_LENGTH_FIELD_IS_BYTE == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_PACKET_LENGTH_FIELD_SIZE)) | |
70 # define ASF_PPI_PACKET_LENGTH_FIELD_SIZE 1 | |
71 #endif | |
72 #if (ASF_PPI_FLAG_PACKET_LENGTH_FIELD_IS_WORD == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_PACKET_LENGTH_FIELD_SIZE)) | |
73 # define ASF_PPI_PACKET_LENGTH_FIELD_SIZE 2 | |
74 #endif | |
75 #if (ASF_PPI_FLAG_PACKET_LENGTH_FIELD_IS_DWORD == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_PACKET_LENGTH_FIELD_SIZE)) | |
76 # define ASF_PPI_PACKET_LENGTH_FIELD_SIZE 4 | |
77 #endif | |
78 #ifndef ASF_PPI_PACKET_LENGTH_FIELD_SIZE | |
79 # define ASF_PPI_PACKET_LENGTH_FIELD_SIZE 0 | |
80 #endif | |
81 | |
82 #if (ASF_PPI_FLAG_PADDING_LENGTH_FIELD_IS_BYTE == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_PADDING_LENGTH_FIELD_SIZE)) | |
83 # define ASF_PPI_PADDING_LENGTH_FIELD_SIZE 1 | |
84 #endif | |
85 #if (ASF_PPI_FLAG_PADDING_LENGTH_FIELD_IS_WORD == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_PADDING_LENGTH_FIELD_SIZE)) | |
86 # define ASF_PPI_PADDING_LENGTH_FIELD_SIZE 2 | |
87 #endif | |
88 #if (ASF_PPI_FLAG_PADDING_LENGTH_FIELD_IS_DWORD == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_PADDING_LENGTH_FIELD_SIZE)) | |
89 # define ASF_PPI_PADDING_LENGTH_FIELD_SIZE 4 | |
90 #endif | |
91 #ifndef ASF_PPI_PADDING_LENGTH_FIELD_SIZE | |
92 # define ASF_PPI_PADDING_LENGTH_FIELD_SIZE 0 | |
93 #endif | |
94 | |
95 #if (ASF_PL_FLAG_REPLICATED_DATA_LENGTH_FIELD_IS_BYTE == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_REPLICATED_DATA_LENGTH_FIELD_SIZE)) | |
96 # define ASF_PAYLOAD_REPLICATED_DATA_LENGTH_FIELD_SIZE 1 | |
97 #endif | |
98 #if (ASF_PL_FLAG_REPLICATED_DATA_LENGTH_FIELD_IS_WORD == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_REPLICATED_DATA_LENGTH_FIELD_SIZE)) | |
99 # define ASF_PAYLOAD_REPLICATED_DATA_LENGTH_FIELD_SIZE 2 | |
100 #endif | |
101 #if (ASF_PL_FLAG_REPLICATED_DATA_LENGTH_FIELD_IS_DWORD == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_REPLICATED_DATA_LENGTH_FIELD_SIZE)) | |
102 # define ASF_PAYLOAD_REPLICATED_DATA_LENGTH_FIELD_SIZE 4 | |
103 #endif | |
104 #ifndef ASF_PAYLOAD_REPLICATED_DATA_LENGTH_FIELD_SIZE | |
105 # define ASF_PAYLOAD_REPLICATED_DATA_LENGTH_FIELD_SIZE 0 | |
106 #endif | |
107 | |
108 #if (ASF_PL_FLAG_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_IS_BYTE == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_SIZE)) | |
109 # define ASF_PAYLOAD_OFFSET_INTO_MEDIA_OBJECT_FIELD_SIZE 1 | |
110 #endif | |
111 #if (ASF_PL_FLAG_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_IS_WORD == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_SIZE)) | |
112 # define ASF_PAYLOAD_OFFSET_INTO_MEDIA_OBJECT_FIELD_SIZE 2 | |
113 #endif | |
114 #if (ASF_PL_FLAG_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_IS_DWORD == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_SIZE)) | |
115 # define ASF_PAYLOAD_OFFSET_INTO_MEDIA_OBJECT_FIELD_SIZE 4 | |
116 #endif | |
117 #ifndef ASF_PAYLOAD_OFFSET_INTO_MEDIA_OBJECT_FIELD_SIZE | |
118 # define ASF_PAYLOAD_OFFSET_INTO_MEDIA_OBJECT_FIELD_SIZE 0 | |
119 #endif | |
120 | |
121 #if (ASF_PL_FLAG_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_IS_BYTE == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_SIZE)) | |
122 # define ASF_PAYLOAD_MEDIA_OBJECT_NUMBER_FIELD_SIZE 1 | |
123 #endif | |
124 #if (ASF_PL_FLAG_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_IS_WORD == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_SIZE)) | |
125 # define ASF_PAYLOAD_MEDIA_OBJECT_NUMBER_FIELD_SIZE 2 | |
126 #endif | |
127 #if (ASF_PL_FLAG_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_IS_DWORD == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_SIZE)) | |
128 # define ASF_PAYLOAD_MEDIA_OBJECT_NUMBER_FIELD_SIZE 4 | |
129 #endif | |
130 #ifndef ASF_PAYLOAD_MEDIA_OBJECT_NUMBER_FIELD_SIZE | |
131 # define ASF_PAYLOAD_MEDIA_OBJECT_NUMBER_FIELD_SIZE 0 | |
132 #endif | |
133 | |
134 #if (ASF_PL_FLAG_PAYLOAD_LENGTH_FIELD_IS_BYTE == (ASF_PAYLOAD_FLAGS & ASF_PL_MASK_PAYLOAD_LENGTH_FIELD_SIZE)) | |
135 # define ASF_PAYLOAD_LENGTH_FIELD_SIZE 1 | |
136 #endif | |
137 #if (ASF_PL_FLAG_PAYLOAD_LENGTH_FIELD_IS_WORD == (ASF_PAYLOAD_FLAGS & ASF_PL_MASK_PAYLOAD_LENGTH_FIELD_SIZE)) | |
138 # define ASF_PAYLOAD_LENGTH_FIELD_SIZE 2 | |
139 #endif | |
140 #ifndef ASF_PAYLOAD_LENGTH_FIELD_SIZE | |
141 # define ASF_PAYLOAD_LENGTH_FIELD_SIZE 0 | |
142 #endif | |
143 | |
144 #define PACKET_HEADER_MIN_SIZE (\ | |
145 ASF_PACKET_ERROR_CORRECTION_FLAGS_FIELD_SIZE + \ | |
146 ASF_PACKET_ERROR_CORRECTION_DATA_SIZE + \ | |
147 1 + /*Length Type Flags*/ \ | |
148 1 + /*Property Flags*/ \ | |
149 ASF_PPI_PACKET_LENGTH_FIELD_SIZE + \ | |
150 ASF_PPI_SEQUENCE_FIELD_SIZE + \ | |
151 ASF_PPI_PADDING_LENGTH_FIELD_SIZE + \ | |
152 4 + /*Send Time Field*/ \ | |
153 2 /*Duration Field*/ \ | |
154 ) | |
155 | |
156 | |
157 // Replicated Data shall be at least 8 bytes long. | |
158 #define ASF_PAYLOAD_REPLICATED_DATA_LENGTH 0x08 | |
159 | |
160 #define PAYLOAD_HEADER_SIZE_SINGLE_PAYLOAD (\ | |
161 1 + /*Stream Number*/ \ | |
162 ASF_PAYLOAD_MEDIA_OBJECT_NUMBER_FIELD_SIZE + \ | |
163 ASF_PAYLOAD_OFFSET_INTO_MEDIA_OBJECT_FIELD_SIZE + \ | |
164 ASF_PAYLOAD_REPLICATED_DATA_LENGTH_FIELD_SIZE + \ | |
165 ASF_PAYLOAD_REPLICATED_DATA_LENGTH \ | |
166 ) | |
167 | |
168 #define PAYLOAD_HEADER_SIZE_MULTIPLE_PAYLOADS (\ | |
169 1 + /*Stream Number*/ \ | |
170 ASF_PAYLOAD_MEDIA_OBJECT_NUMBER_FIELD_SIZE + \ | |
171 ASF_PAYLOAD_OFFSET_INTO_MEDIA_OBJECT_FIELD_SIZE + \ | |
172 ASF_PAYLOAD_REPLICATED_DATA_LENGTH_FIELD_SIZE + \ | |
173 ASF_PAYLOAD_REPLICATED_DATA_LENGTH + \ | |
174 ASF_PAYLOAD_LENGTH_FIELD_SIZE \ | |
175 ) | |
176 | |
177 #define SINGLE_PAYLOAD_DATA_LENGTH (\ | |
178 PACKET_SIZE - \ | |
179 PACKET_HEADER_MIN_SIZE - \ | |
180 PAYLOAD_HEADER_SIZE_SINGLE_PAYLOAD \ | |
181 ) | |
182 | |
183 #define MULTI_PAYLOAD_CONSTANT (\ | |
184 PACKET_SIZE - \ | |
185 PACKET_HEADER_MIN_SIZE - \ | |
186 1 - /*Payload Flags*/ \ | |
187 2*PAYLOAD_HEADER_SIZE_MULTIPLE_PAYLOADS \ | |
188 ) | |
189 | |
190 static int preroll_time = 2000; | |
191 | |
192 static const uint8_t error_spread_ADPCM_G726[] = { 0x01, 0x90, 0x01, 0x90, 0x01, 0x01, 0x00, 0x00 }; | |
193 | |
194 static void put_guid(ByteIOContext *s, const GUID *g) | |
195 { | |
196 int i; | |
197 | |
198 put_le32(s, g->v1); | |
199 put_le16(s, g->v2); | |
200 put_le16(s, g->v3); | |
201 for(i=0;i<8;i++) | |
202 put_byte(s, g->v4[i]); | |
203 } | |
204 | |
205 static void put_str16(ByteIOContext *s, const char *tag) | |
206 { | |
207 int c; | |
208 | |
209 put_le16(s,strlen(tag) + 1); | |
210 for(;;) { | |
211 c = (uint8_t)*tag++; | |
212 put_le16(s, c); | |
213 if (c == '\0') | |
214 break; | |
215 } | |
216 } | |
217 | |
218 static void put_str16_nolen(ByteIOContext *s, const char *tag) | |
219 { | |
220 int c; | |
221 | |
222 for(;;) { | |
223 c = (uint8_t)*tag++; | |
224 put_le16(s, c); | |
225 if (c == '\0') | |
226 break; | |
227 } | |
228 } | |
229 | |
230 static int64_t put_header(ByteIOContext *pb, const GUID *g) | |
231 { | |
232 int64_t pos; | |
233 | |
234 pos = url_ftell(pb); | |
235 put_guid(pb, g); | |
236 put_le64(pb, 24); | |
237 return pos; | |
238 } | |
239 | |
240 /* update header size */ | |
241 static void end_header(ByteIOContext *pb, int64_t pos) | |
242 { | |
243 int64_t pos1; | |
244 | |
245 pos1 = url_ftell(pb); | |
246 url_fseek(pb, pos + 16, SEEK_SET); | |
247 put_le64(pb, pos1 - pos); | |
248 url_fseek(pb, pos1, SEEK_SET); | |
249 } | |
250 | |
251 /* write an asf chunk (only used in streaming case) */ | |
252 static void put_chunk(AVFormatContext *s, int type, int payload_length, int flags) | |
253 { | |
254 ASFContext *asf = s->priv_data; | |
255 ByteIOContext *pb = &s->pb; | |
256 int length; | |
257 | |
258 length = payload_length + 8; | |
259 put_le16(pb, type); | |
260 put_le16(pb, length); //size | |
261 put_le32(pb, asf->seqno);//sequence number | |
262 put_le16(pb, flags); /* unknown bytes */ | |
263 put_le16(pb, length); //size_confirm | |
264 asf->seqno++; | |
265 } | |
266 | |
267 /* convert from unix to windows time */ | |
268 static int64_t unix_to_file_time(int ti) | |
269 { | |
270 int64_t t; | |
271 | |
272 t = ti * int64_t_C(10000000); | |
273 t += int64_t_C(116444736000000000); | |
274 return t; | |
275 } | |
276 | |
277 /* write the header (used two times if non streamed) */ | |
278 static int asf_write_header1(AVFormatContext *s, int64_t file_size, int64_t data_chunk_size) | |
279 { | |
280 ASFContext *asf = s->priv_data; | |
281 ByteIOContext *pb = &s->pb; | |
282 int header_size, n, extra_size, extra_size2, wav_extra_size, file_time; | |
283 int has_title; | |
284 AVCodecContext *enc; | |
285 int64_t header_offset, cur_pos, hpos; | |
286 int bit_rate; | |
287 int64_t duration; | |
288 | |
289 duration = asf->duration + preroll_time * 10000; | |
290 has_title = (s->title[0] || s->author[0] || s->copyright[0] || s->comment[0]); | |
291 | |
292 bit_rate = 0; | |
293 for(n=0;n<s->nb_streams;n++) { | |
294 enc = s->streams[n]->codec; | |
295 | |
296 av_set_pts_info(s->streams[n], 32, 1, 1000); /* 32 bit pts in ms */ | |
297 | |
298 bit_rate += enc->bit_rate; | |
299 } | |
300 | |
301 if (asf->is_streamed) { | |
302 put_chunk(s, 0x4824, 0, 0xc00); /* start of stream (length will be patched later) */ | |
303 } | |
304 | |
305 put_guid(pb, &asf_header); | |
306 put_le64(pb, -1); /* header length, will be patched after */ | |
307 put_le32(pb, 3 + has_title + s->nb_streams); /* number of chunks in header */ | |
308 put_byte(pb, 1); /* ??? */ | |
309 put_byte(pb, 2); /* ??? */ | |
310 | |
311 /* file header */ | |
312 header_offset = url_ftell(pb); | |
313 hpos = put_header(pb, &file_header); | |
314 put_guid(pb, &my_guid); | |
315 put_le64(pb, file_size); | |
316 file_time = 0; | |
317 put_le64(pb, unix_to_file_time(file_time)); | |
318 put_le64(pb, asf->nb_packets); /* number of packets */ | |
319 put_le64(pb, duration); /* end time stamp (in 100ns units) */ | |
320 put_le64(pb, duration); /* duration (in 100ns units) */ | |
321 put_le32(pb, preroll_time); /* start time stamp */ | |
322 put_le32(pb, 0); /* ??? */ | |
323 put_le32(pb, asf->is_streamed ? 1 : 0); /* ??? */ | |
324 put_le32(pb, asf->packet_size); /* packet size */ | |
325 put_le32(pb, asf->packet_size); /* packet size */ | |
326 put_le32(pb, bit_rate); /* Nominal data rate in bps */ | |
327 end_header(pb, hpos); | |
328 | |
329 /* unknown headers */ | |
330 hpos = put_header(pb, &head1_guid); | |
331 put_guid(pb, &head2_guid); | |
332 put_le32(pb, 6); | |
333 put_le16(pb, 0); | |
334 end_header(pb, hpos); | |
335 | |
336 /* title and other infos */ | |
337 if (has_title) { | |
338 hpos = put_header(pb, &comment_header); | |
339 if ( s->title[0] ) { put_le16(pb, 2 * (strlen(s->title ) + 1)); } else { put_le16(pb, 0); } | |
340 if ( s->author[0] ) { put_le16(pb, 2 * (strlen(s->author ) + 1)); } else { put_le16(pb, 0); } | |
341 if ( s->copyright[0] ) { put_le16(pb, 2 * (strlen(s->copyright) + 1)); } else { put_le16(pb, 0); } | |
342 if ( s->comment[0] ) { put_le16(pb, 2 * (strlen(s->comment ) + 1)); } else { put_le16(pb, 0); } | |
343 put_le16(pb, 0); | |
344 if ( s->title[0] ) put_str16_nolen(pb, s->title); | |
345 if ( s->author[0] ) put_str16_nolen(pb, s->author); | |
346 if ( s->copyright[0] ) put_str16_nolen(pb, s->copyright); | |
347 if ( s->comment[0] ) put_str16_nolen(pb, s->comment); | |
348 end_header(pb, hpos); | |
349 } | |
350 | |
351 /* stream headers */ | |
352 for(n=0;n<s->nb_streams;n++) { | |
353 int64_t es_pos; | |
354 const uint8_t *er_spr = NULL; | |
355 int er_spr_len = 0; | |
356 // ASFStream *stream = &asf->streams[n]; | |
357 | |
358 enc = s->streams[n]->codec; | |
359 asf->streams[n].num = n + 1; | |
360 asf->streams[n].seq = 0; | |
361 | |
362 | |
363 if (enc->codec_type == CODEC_TYPE_AUDIO) { | |
364 if (enc->codec_id == CODEC_ID_ADPCM_G726) { | |
365 er_spr = error_spread_ADPCM_G726; | |
366 er_spr_len = sizeof(error_spread_ADPCM_G726); | |
367 } | |
368 } | |
369 | |
370 switch(enc->codec_type) { | |
371 case CODEC_TYPE_AUDIO: | |
372 wav_extra_size = 0; | |
373 extra_size = 18 + wav_extra_size; | |
374 extra_size2 = er_spr_len; | |
375 break; | |
376 default: | |
377 case CODEC_TYPE_VIDEO: | |
378 wav_extra_size = enc->extradata_size; | |
379 extra_size = 0x33 + wav_extra_size; | |
380 extra_size2 = 0; | |
381 break; | |
382 } | |
383 | |
384 hpos = put_header(pb, &stream_header); | |
385 if (enc->codec_type == CODEC_TYPE_AUDIO) { | |
386 put_guid(pb, &audio_stream); | |
387 if ((er_spr != NULL) && (er_spr_len != 0)) { | |
388 put_guid(pb, &audio_conceal_spread); | |
389 } else { | |
390 put_guid(pb, &video_conceal_none); | |
391 } | |
392 } else { | |
393 put_guid(pb, &video_stream); | |
394 put_guid(pb, &video_conceal_none); | |
395 } | |
396 put_le64(pb, 0); /* ??? */ | |
397 es_pos = url_ftell(pb); | |
398 put_le32(pb, extra_size); /* wav header len */ | |
399 put_le32(pb, extra_size2); /* additional data len */ | |
400 put_le16(pb, n + 1); /* stream number */ | |
401 put_le32(pb, 0); /* ??? */ | |
402 | |
403 if (enc->codec_type == CODEC_TYPE_AUDIO) { | |
404 /* WAVEFORMATEX header */ | |
405 int wavsize = put_wav_header(pb, enc); | |
406 if ((enc->codec_id != CODEC_ID_MP3) && (enc->codec_id != CODEC_ID_MP2) && (enc->codec_id != CODEC_ID_ADPCM_IMA_WAV) && (enc->extradata_size==0)) { | |
407 wavsize += 2; | |
408 put_le16(pb, 0); | |
409 } | |
410 | |
411 if (wavsize < 0) | |
412 return -1; | |
413 if (wavsize != extra_size) { | |
414 cur_pos = url_ftell(pb); | |
415 url_fseek(pb, es_pos, SEEK_SET); | |
416 put_le32(pb, wavsize); /* wav header len */ | |
417 url_fseek(pb, cur_pos, SEEK_SET); | |
418 } | |
419 /* ERROR Correction */ | |
420 if ((er_spr != NULL) && (er_spr_len != 0)) | |
421 put_buffer(pb, er_spr, er_spr_len); | |
422 } else { | |
423 put_le32(pb, enc->width); | |
424 put_le32(pb, enc->height); | |
425 put_byte(pb, 2); /* ??? */ | |
426 put_le16(pb, 40 + enc->extradata_size); /* size */ | |
427 | |
428 /* BITMAPINFOHEADER header */ | |
429 put_bmp_header(pb, enc, codec_bmp_tags, 1); | |
430 } | |
431 end_header(pb, hpos); | |
432 } | |
433 | |
434 /* media comments */ | |
435 | |
436 hpos = put_header(pb, &codec_comment_header); | |
437 put_guid(pb, &codec_comment1_header); | |
438 put_le32(pb, s->nb_streams); | |
439 for(n=0;n<s->nb_streams;n++) { | |
440 AVCodec *p; | |
441 | |
442 enc = s->streams[n]->codec; | |
443 p = avcodec_find_encoder(enc->codec_id); | |
444 | |
445 put_le16(pb, asf->streams[n].num); | |
446 put_str16(pb, p ? p->name : enc->codec_name); | |
447 put_le16(pb, 0); /* no parameters */ | |
448 | |
449 | |
450 /* id */ | |
451 if (enc->codec_type == CODEC_TYPE_AUDIO) { | |
452 put_le16(pb, 2); | |
453 if(!enc->codec_tag) | |
454 enc->codec_tag = codec_get_tag(codec_wav_tags, enc->codec_id); | |
455 if(!enc->codec_tag) | |
456 return -1; | |
457 put_le16(pb, enc->codec_tag); | |
458 } else { | |
459 put_le16(pb, 4); | |
460 if(!enc->codec_tag) | |
461 enc->codec_tag = codec_get_tag(codec_bmp_tags, enc->codec_id); | |
462 if(!enc->codec_tag) | |
463 return -1; | |
464 put_le32(pb, enc->codec_tag); | |
465 } | |
466 } | |
467 end_header(pb, hpos); | |
468 | |
469 /* patch the header size fields */ | |
470 | |
471 cur_pos = url_ftell(pb); | |
472 header_size = cur_pos - header_offset; | |
473 if (asf->is_streamed) { | |
474 header_size += 8 + 30 + 50; | |
475 | |
476 url_fseek(pb, header_offset - 10 - 30, SEEK_SET); | |
477 put_le16(pb, header_size); | |
478 url_fseek(pb, header_offset - 2 - 30, SEEK_SET); | |
479 put_le16(pb, header_size); | |
480 | |
481 header_size -= 8 + 30 + 50; | |
482 } | |
483 header_size += 24 + 6; | |
484 url_fseek(pb, header_offset - 14, SEEK_SET); | |
485 put_le64(pb, header_size); | |
486 url_fseek(pb, cur_pos, SEEK_SET); | |
487 | |
488 /* movie chunk, followed by packets of packet_size */ | |
489 asf->data_offset = cur_pos; | |
490 put_guid(pb, &data_header); | |
491 put_le64(pb, data_chunk_size); | |
492 put_guid(pb, &my_guid); | |
493 put_le64(pb, asf->nb_packets); /* nb packets */ | |
494 put_byte(pb, 1); /* ??? */ | |
495 put_byte(pb, 1); /* ??? */ | |
496 return 0; | |
497 } | |
498 | |
499 static int asf_write_header(AVFormatContext *s) | |
500 { | |
501 ASFContext *asf = s->priv_data; | |
502 | |
503 asf->packet_size = PACKET_SIZE; | |
504 asf->nb_packets = 0; | |
505 | |
506 asf->last_indexed_pts = 0; | |
507 asf->index_ptr = (ASFIndex*)av_malloc( sizeof(ASFIndex) * ASF_INDEX_BLOCK ); | |
508 asf->nb_index_memory_alloc = ASF_INDEX_BLOCK; | |
509 asf->nb_index_count = 0; | |
510 asf->maximum_packet = 0; | |
511 | |
512 if (asf_write_header1(s, 0, 50) < 0) { | |
513 //av_free(asf); | |
514 return -1; | |
515 } | |
516 | |
517 put_flush_packet(&s->pb); | |
518 | |
519 asf->packet_nb_payloads = 0; | |
520 asf->prev_packet_sent_time = 0; | |
521 asf->packet_timestamp_start = -1; | |
522 asf->packet_timestamp_end = -1; | |
523 init_put_byte(&asf->pb, asf->packet_buf, asf->packet_size, 1, | |
524 NULL, NULL, NULL, NULL); | |
525 | |
526 return 0; | |
527 } | |
528 | |
529 static int asf_write_stream_header(AVFormatContext *s) | |
530 { | |
531 ASFContext *asf = s->priv_data; | |
532 | |
533 asf->is_streamed = 1; | |
534 | |
535 return asf_write_header(s); | |
536 } | |
537 | |
538 static int put_payload_parsing_info( | |
539 AVFormatContext *s, | |
540 unsigned int sendtime, | |
541 unsigned int duration, | |
542 int nb_payloads, | |
543 int padsize | |
544 ) | |
545 { | |
546 ASFContext *asf = s->priv_data; | |
547 ByteIOContext *pb = &s->pb; | |
548 int ppi_size, i; | |
549 unsigned char *start_ppi_ptr = pb->buf_ptr; | |
550 | |
551 int iLengthTypeFlags = ASF_PPI_LENGTH_TYPE_FLAGS; | |
552 | |
553 put_byte(pb, ASF_PACKET_ERROR_CORRECTION_FLAGS); | |
554 for (i = 0; i < ASF_PACKET_ERROR_CORRECTION_DATA_SIZE; i++){ | |
555 put_byte(pb, 0x0); | |
556 } | |
557 | |
558 if (asf->multi_payloads_present) | |
559 iLengthTypeFlags |= ASF_PPI_FLAG_MULTIPLE_PAYLOADS_PRESENT; | |
560 | |
561 if (padsize > 0) { | |
562 if (padsize < 256) | |
563 iLengthTypeFlags |= ASF_PPI_FLAG_PADDING_LENGTH_FIELD_IS_BYTE; | |
564 else | |
565 iLengthTypeFlags |= ASF_PPI_FLAG_PADDING_LENGTH_FIELD_IS_WORD; | |
566 } | |
567 put_byte(pb, iLengthTypeFlags); | |
568 | |
569 put_byte(pb, ASF_PPI_PROPERTY_FLAGS); | |
570 | |
571 if (iLengthTypeFlags & ASF_PPI_FLAG_PADDING_LENGTH_FIELD_IS_WORD) | |
572 put_le16(pb, padsize - 2); | |
573 if (iLengthTypeFlags & ASF_PPI_FLAG_PADDING_LENGTH_FIELD_IS_BYTE) | |
574 put_byte(pb, padsize - 1); | |
575 | |
576 put_le32(pb, sendtime); | |
577 put_le16(pb, duration); | |
578 if (asf->multi_payloads_present) | |
579 put_byte(pb, nb_payloads | ASF_PAYLOAD_FLAGS); | |
580 | |
581 ppi_size = pb->buf_ptr - start_ppi_ptr; | |
582 | |
583 return ppi_size; | |
584 } | |
585 | |
586 static void flush_packet(AVFormatContext *s) | |
587 { | |
588 ASFContext *asf = s->priv_data; | |
589 int packet_hdr_size, packet_filled_size; | |
590 | |
591 if (asf->is_streamed) { | |
592 put_chunk(s, 0x4424, asf->packet_size, 0); | |
593 } | |
594 | |
595 packet_hdr_size = put_payload_parsing_info( | |
596 s, | |
597 asf->packet_timestamp_start, | |
598 asf->packet_timestamp_end - asf->packet_timestamp_start, | |
599 asf->packet_nb_payloads, | |
600 asf->packet_size_left | |
601 ); | |
602 | |
603 packet_filled_size = PACKET_SIZE - packet_hdr_size - asf->packet_size_left; | |
604 memset(asf->packet_buf + packet_filled_size, 0, asf->packet_size_left); | |
605 | |
606 put_buffer(&s->pb, asf->packet_buf, asf->packet_size - packet_hdr_size); | |
607 | |
608 put_flush_packet(&s->pb); | |
609 asf->nb_packets++; | |
610 asf->packet_nb_payloads = 0; | |
611 asf->prev_packet_sent_time = asf->packet_timestamp_start; | |
612 asf->packet_timestamp_start = -1; | |
613 asf->packet_timestamp_end = -1; | |
614 init_put_byte(&asf->pb, asf->packet_buf, asf->packet_size, 1, | |
615 NULL, NULL, NULL, NULL); | |
616 } | |
617 | |
618 static void put_payload_header( | |
619 AVFormatContext *s, | |
620 ASFStream *stream, | |
621 int presentation_time, | |
622 int m_obj_size, | |
623 int m_obj_offset, | |
624 int payload_len, | |
625 int flags | |
626 ) | |
627 { | |
628 ASFContext *asf = s->priv_data; | |
629 ByteIOContext *pb = &asf->pb; | |
630 int val; | |
631 | |
632 val = stream->num; | |
633 if (flags & PKT_FLAG_KEY) | |
634 val |= ASF_PL_FLAG_KEY_FRAME; | |
635 put_byte(pb, val); | |
636 | |
637 put_byte(pb, stream->seq); //Media object number | |
638 put_le32(pb, m_obj_offset); //Offset Into Media Object | |
639 | |
640 // Replicated Data shall be at least 8 bytes long. | |
641 // The first 4 bytes of data shall contain the | |
642 // Size of the Media Object that the payload belongs to. | |
643 // The next 4 bytes of data shall contain the | |
644 // Presentation Time for the media object that the payload belongs to. | |
645 put_byte(pb, ASF_PAYLOAD_REPLICATED_DATA_LENGTH); | |
646 | |
647 put_le32(pb, m_obj_size); //Replicated Data - Media Object Size | |
648 put_le32(pb, presentation_time);//Replicated Data - Presentation Time | |
649 | |
650 if (asf->multi_payloads_present){ | |
651 put_le16(pb, payload_len); //payload length | |
652 } | |
653 } | |
654 | |
655 static void put_frame( | |
656 AVFormatContext *s, | |
657 ASFStream *stream, | |
658 int timestamp, | |
659 const uint8_t *buf, | |
660 int m_obj_size, | |
661 int flags | |
662 ) | |
663 { | |
664 ASFContext *asf = s->priv_data; | |
665 int m_obj_offset, payload_len, frag_len1; | |
666 | |
667 m_obj_offset = 0; | |
668 while (m_obj_offset < m_obj_size) { | |
669 payload_len = m_obj_size - m_obj_offset; | |
670 if (asf->packet_timestamp_start == -1) { | |
671 asf->multi_payloads_present = (payload_len < MULTI_PAYLOAD_CONSTANT); | |
672 | |
673 if (asf->multi_payloads_present){ | |
674 asf->packet_size_left = PACKET_SIZE; //For debug | |
675 asf->packet_size_left = PACKET_SIZE - PACKET_HEADER_MIN_SIZE - 1; | |
676 frag_len1 = MULTI_PAYLOAD_CONSTANT - 1; | |
677 } | |
678 else { | |
679 asf->packet_size_left = PACKET_SIZE - PACKET_HEADER_MIN_SIZE; | |
680 frag_len1 = SINGLE_PAYLOAD_DATA_LENGTH; | |
681 } | |
682 if (asf->prev_packet_sent_time > timestamp) | |
683 asf->packet_timestamp_start = asf->prev_packet_sent_time; | |
684 else | |
685 asf->packet_timestamp_start = timestamp; | |
686 } | |
687 else { | |
688 // multi payloads | |
689 frag_len1 = asf->packet_size_left - PAYLOAD_HEADER_SIZE_MULTIPLE_PAYLOADS; | |
690 | |
691 if (asf->prev_packet_sent_time > timestamp) | |
692 asf->packet_timestamp_start = asf->prev_packet_sent_time; | |
693 else if (asf->packet_timestamp_start >= timestamp) | |
694 asf->packet_timestamp_start = timestamp; | |
695 } | |
696 if (frag_len1 > 0) { | |
697 if (payload_len > frag_len1) | |
698 payload_len = frag_len1; | |
699 else if (payload_len == (frag_len1 - 1)) | |
700 payload_len = frag_len1 - 2; //additional byte need to put padding length | |
701 | |
702 put_payload_header(s, stream, timestamp+preroll_time, m_obj_size, m_obj_offset, payload_len, flags); | |
703 put_buffer(&asf->pb, buf, payload_len); | |
704 | |
705 if (asf->multi_payloads_present) | |
706 asf->packet_size_left -= (payload_len + PAYLOAD_HEADER_SIZE_MULTIPLE_PAYLOADS); | |
707 else | |
708 asf->packet_size_left -= (payload_len + PAYLOAD_HEADER_SIZE_SINGLE_PAYLOAD); | |
709 asf->packet_timestamp_end = timestamp; | |
710 | |
711 asf->packet_nb_payloads++; | |
712 } else { | |
713 payload_len = 0; | |
714 } | |
715 m_obj_offset += payload_len; | |
716 buf += payload_len; | |
717 | |
718 if (!asf->multi_payloads_present) | |
719 flush_packet(s); | |
720 else if (asf->packet_size_left <= (PAYLOAD_HEADER_SIZE_MULTIPLE_PAYLOADS + 1)) | |
721 flush_packet(s); | |
722 } | |
723 stream->seq++; | |
724 } | |
725 | |
726 static int asf_write_packet(AVFormatContext *s, AVPacket *pkt) | |
727 { | |
728 ASFContext *asf = s->priv_data; | |
729 ASFStream *stream; | |
730 int64_t duration; | |
731 AVCodecContext *codec; | |
732 int64_t packet_st,pts; | |
733 int start_sec,i; | |
734 | |
735 codec = s->streams[pkt->stream_index]->codec; | |
736 stream = &asf->streams[pkt->stream_index]; | |
737 | |
738 //XXX /FIXME use duration from AVPacket (quick hack by) | |
739 pts = (pkt->pts != AV_NOPTS_VALUE) ? pkt->pts : pkt->dts; | |
740 if (pts == AV_NOPTS_VALUE) { | |
741 if (codec->codec_type == CODEC_TYPE_AUDIO) { | |
742 duration = (codec->frame_number * (int64_t)codec->frame_size * int64_t_C(10000000)) / | |
743 codec->sample_rate; | |
744 } else { | |
745 duration = av_rescale(codec->frame_number * (int64_t)codec->time_base.num, 10000000, codec->time_base.den); | |
746 } | |
747 } else { | |
748 duration = pts * 10000; | |
749 } | |
750 if (duration > asf->duration) | |
751 asf->duration = duration; | |
752 | |
753 packet_st = asf->nb_packets; | |
754 put_frame(s, stream, pkt->pts, pkt->data, pkt->size, pkt->flags); | |
755 | |
756 /* check index */ | |
757 if ((!asf->is_streamed) && (codec->codec_type == CODEC_TYPE_VIDEO) && (pkt->flags & PKT_FLAG_KEY)) { | |
758 start_sec = (int)(duration / int64_t_C(10000000)); | |
759 if (start_sec != (int)(asf->last_indexed_pts / int64_t_C(10000000))) { | |
760 for(i=asf->nb_index_count;i<start_sec;i++) { | |
761 if (i>=asf->nb_index_memory_alloc) { | |
762 asf->nb_index_memory_alloc += ASF_INDEX_BLOCK; | |
763 asf->index_ptr = (ASFIndex*)av_realloc( asf->index_ptr, sizeof(ASFIndex) * asf->nb_index_memory_alloc ); | |
764 } | |
765 // store | |
766 asf->index_ptr[i].packet_number = (uint32_t)packet_st; | |
767 asf->index_ptr[i].packet_count = (uint16_t)(asf->nb_packets-packet_st); | |
768 if (asf->maximum_packet < (uint16_t)(asf->nb_packets-packet_st)) | |
769 asf->maximum_packet = (uint16_t)(asf->nb_packets-packet_st); | |
770 } | |
771 asf->nb_index_count = start_sec; | |
772 asf->last_indexed_pts = duration; | |
773 } | |
774 } | |
775 return 0; | |
776 } | |
777 | |
778 // | |
779 static int asf_write_index(AVFormatContext *s, ASFIndex *index, uint16_t max, uint32_t count) | |
780 { | |
781 ByteIOContext *pb = &s->pb; | |
782 int i; | |
783 | |
784 put_guid(pb, &simple_index_header); | |
785 put_le64(pb, 24 + 16 + 8 + 4 + 4 + (4 + 2)*count); | |
786 put_guid(pb, &my_guid); | |
787 put_le64(pb, ASF_INDEXED_INTERVAL); | |
788 put_le32(pb, max); | |
789 put_le32(pb, count); | |
790 for(i=0; i<count; i++) { | |
791 put_le32(pb, index[i].packet_number); | |
792 put_le16(pb, index[i].packet_count); | |
793 } | |
794 | |
795 return 0; | |
796 } | |
797 | |
798 static int asf_write_trailer(AVFormatContext *s) | |
799 { | |
800 ASFContext *asf = s->priv_data; | |
801 int64_t file_size,data_size; | |
802 | |
803 /* flush the current packet */ | |
804 if (asf->pb.buf_ptr > asf->pb.buffer) | |
805 flush_packet(s); | |
806 | |
807 /* write index */ | |
808 data_size = url_ftell(&s->pb); | |
809 if ((!asf->is_streamed) && (asf->nb_index_count != 0)) { | |
810 asf_write_index(s, asf->index_ptr, asf->maximum_packet, asf->nb_index_count); | |
811 } | |
812 put_flush_packet(&s->pb); | |
813 | |
814 if (asf->is_streamed) { | |
815 put_chunk(s, 0x4524, 0, 0); /* end of stream */ | |
816 } else { | |
817 /* rewrite an updated header */ | |
818 file_size = url_ftell(&s->pb); | |
819 url_fseek(&s->pb, 0, SEEK_SET); | |
820 asf_write_header1(s, file_size, data_size - asf->data_offset); | |
821 } | |
822 | |
823 put_flush_packet(&s->pb); | |
824 av_free(asf->index_ptr); | |
825 return 0; | |
826 } | |
827 | |
828 #ifdef CONFIG_ASF_MUXER | |
829 AVOutputFormat asf_muxer = { | |
830 "asf", | |
831 "asf format", | |
832 "video/x-ms-asf", | |
833 "asf,wmv,wma", | |
834 sizeof(ASFContext), | |
835 #ifdef CONFIG_MP3LAME | |
836 CODEC_ID_MP3, | |
837 #else | |
838 CODEC_ID_MP2, | |
839 #endif | |
840 CODEC_ID_MSMPEG4V3, | |
841 asf_write_header, | |
842 asf_write_packet, | |
843 asf_write_trailer, | |
844 .flags = AVFMT_GLOBALHEADER, | |
845 }; | |
846 #endif | |
847 | |
848 #ifdef CONFIG_ASF_STREAM_MUXER | |
849 AVOutputFormat asf_stream_muxer = { | |
850 "asf_stream", | |
851 "asf format", | |
852 "video/x-ms-asf", | |
853 "asf,wmv,wma", | |
854 sizeof(ASFContext), | |
855 #ifdef CONFIG_MP3LAME | |
856 CODEC_ID_MP3, | |
857 #else | |
858 CODEC_ID_MP2, | |
859 #endif | |
860 CODEC_ID_MSMPEG4V3, | |
861 asf_write_stream_header, | |
862 asf_write_packet, | |
863 asf_write_trailer, | |
864 .flags = AVFMT_GLOBALHEADER, | |
865 }; | |
866 #endif //CONFIG_ASF_STREAM_MUXER |