Mercurial > libavformat.hg
annotate swfenc.c @ 3375:9877da0d8be3 libavformat
use AVFifoBuffer
author | bcoudurier |
---|---|
date | Tue, 27 May 2008 04:33:48 +0000 |
parents | 67b57431ac20 |
children | b3827117c786 |
rev | line source |
---|---|
0 | 1 /* |
3356 | 2 * Flash Compatible Streaming Format muxer |
0 | 3 * Copyright (c) 2000 Fabrice Bellard. |
359 | 4 * Copyright (c) 2003 Tinic Uro. |
0 | 5 * |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1305
diff
changeset
|
6 * This file is part of FFmpeg. |
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1305
diff
changeset
|
7 * |
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1305
diff
changeset
|
8 * FFmpeg is free software; you can redistribute it and/or |
0 | 9 * modify it under the terms of the GNU Lesser General Public |
10 * License as published by the Free Software Foundation; either | |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1305
diff
changeset
|
11 * version 2.1 of the License, or (at your option) any later version. |
0 | 12 * |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1305
diff
changeset
|
13 * FFmpeg is distributed in the hope that it will be useful, |
0 | 14 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
16 * Lesser General Public License for more details. | |
17 * | |
18 * You should have received a copy of the GNU Lesser General Public | |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1305
diff
changeset
|
19 * License along with FFmpeg; if not, write to the Free Software |
896
edbe5c3717f9
Update licensing information: The FSF changed postal address.
diego
parents:
887
diff
changeset
|
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
0 | 21 */ |
3286 | 22 |
23 #include "libavcodec/bitstream.h" | |
0 | 24 #include "avformat.h" |
3302 | 25 #include "swf.h" |
0 | 26 |
27 static void put_swf_tag(AVFormatContext *s, int tag) | |
28 { | |
29 SWFContext *swf = s->priv_data; | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
30 ByteIOContext *pb = s->pb; |
0 | 31 |
32 swf->tag_pos = url_ftell(pb); | |
33 swf->tag = tag; | |
34 /* reserve some room for the tag */ | |
35 if (tag & TAG_LONG) { | |
36 put_le16(pb, 0); | |
37 put_le32(pb, 0); | |
38 } else { | |
39 put_le16(pb, 0); | |
40 } | |
41 } | |
42 | |
43 static void put_swf_end_tag(AVFormatContext *s) | |
44 { | |
45 SWFContext *swf = s->priv_data; | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
46 ByteIOContext *pb = s->pb; |
0 | 47 offset_t pos; |
48 int tag_len, tag; | |
49 | |
50 pos = url_ftell(pb); | |
51 tag_len = pos - swf->tag_pos - 2; | |
52 tag = swf->tag; | |
53 url_fseek(pb, swf->tag_pos, SEEK_SET); | |
54 if (tag & TAG_LONG) { | |
55 tag &= ~TAG_LONG; | |
56 put_le16(pb, (tag << 6) | 0x3f); | |
57 put_le32(pb, tag_len - 4); | |
58 } else { | |
59 assert(tag_len < 0x3f); | |
60 put_le16(pb, (tag << 6) | tag_len); | |
61 } | |
62 url_fseek(pb, pos, SEEK_SET); | |
63 } | |
64 | |
65 static inline void max_nbits(int *nbits_ptr, int val) | |
66 { | |
67 int n; | |
68 | |
69 if (val == 0) | |
70 return; | |
71 val = abs(val); | |
72 n = 1; | |
73 while (val != 0) { | |
74 n++; | |
75 val >>= 1; | |
76 } | |
77 if (n > *nbits_ptr) | |
78 *nbits_ptr = n; | |
79 } | |
80 | |
885 | 81 static void put_swf_rect(ByteIOContext *pb, |
0 | 82 int xmin, int xmax, int ymin, int ymax) |
83 { | |
84 PutBitContext p; | |
65 | 85 uint8_t buf[256]; |
0 | 86 int nbits, mask; |
87 | |
276 | 88 init_put_bits(&p, buf, sizeof(buf)); |
885 | 89 |
0 | 90 nbits = 0; |
91 max_nbits(&nbits, xmin); | |
92 max_nbits(&nbits, xmax); | |
93 max_nbits(&nbits, ymin); | |
94 max_nbits(&nbits, ymax); | |
95 mask = (1 << nbits) - 1; | |
96 | |
97 /* rectangle info */ | |
98 put_bits(&p, 5, nbits); | |
99 put_bits(&p, nbits, xmin & mask); | |
100 put_bits(&p, nbits, xmax & mask); | |
101 put_bits(&p, nbits, ymin & mask); | |
102 put_bits(&p, nbits, ymax & mask); | |
885 | 103 |
0 | 104 flush_put_bits(&p); |
105 put_buffer(pb, buf, pbBufPtr(&p) - p.buf); | |
106 } | |
107 | |
108 static void put_swf_line_edge(PutBitContext *pb, int dx, int dy) | |
109 { | |
110 int nbits, mask; | |
111 | |
112 put_bits(pb, 1, 1); /* edge */ | |
113 put_bits(pb, 1, 1); /* line select */ | |
114 nbits = 2; | |
115 max_nbits(&nbits, dx); | |
116 max_nbits(&nbits, dy); | |
117 | |
118 mask = (1 << nbits) - 1; | |
119 put_bits(pb, 4, nbits - 2); /* 16 bits precision */ | |
120 if (dx == 0) { | |
2308 | 121 put_bits(pb, 1, 0); |
122 put_bits(pb, 1, 1); | |
123 put_bits(pb, nbits, dy & mask); | |
0 | 124 } else if (dy == 0) { |
2308 | 125 put_bits(pb, 1, 0); |
126 put_bits(pb, 1, 0); | |
127 put_bits(pb, nbits, dx & mask); | |
0 | 128 } else { |
2308 | 129 put_bits(pb, 1, 1); |
130 put_bits(pb, nbits, dx & mask); | |
131 put_bits(pb, nbits, dy & mask); | |
0 | 132 } |
133 } | |
134 | |
135 #define FRAC_BITS 16 | |
136 | |
137 static void put_swf_matrix(ByteIOContext *pb, | |
138 int a, int b, int c, int d, int tx, int ty) | |
139 { | |
140 PutBitContext p; | |
65 | 141 uint8_t buf[256]; |
359 | 142 int nbits; |
0 | 143 |
276 | 144 init_put_bits(&p, buf, sizeof(buf)); |
885 | 145 |
0 | 146 put_bits(&p, 1, 1); /* a, d present */ |
359 | 147 nbits = 1; |
148 max_nbits(&nbits, a); | |
149 max_nbits(&nbits, d); | |
150 put_bits(&p, 5, nbits); /* nb bits */ | |
151 put_bits(&p, nbits, a); | |
152 put_bits(&p, nbits, d); | |
885 | 153 |
0 | 154 put_bits(&p, 1, 1); /* b, c present */ |
359 | 155 nbits = 1; |
156 max_nbits(&nbits, c); | |
157 max_nbits(&nbits, b); | |
158 put_bits(&p, 5, nbits); /* nb bits */ | |
159 put_bits(&p, nbits, c); | |
160 put_bits(&p, nbits, b); | |
0 | 161 |
359 | 162 nbits = 1; |
163 max_nbits(&nbits, tx); | |
164 max_nbits(&nbits, ty); | |
165 put_bits(&p, 5, nbits); /* nb bits */ | |
166 put_bits(&p, nbits, tx); | |
167 put_bits(&p, nbits, ty); | |
0 | 168 |
169 flush_put_bits(&p); | |
170 put_buffer(pb, buf, pbBufPtr(&p) - p.buf); | |
171 } | |
172 | |
173 static int swf_write_header(AVFormatContext *s) | |
174 { | |
1623 | 175 SWFContext *swf = s->priv_data; |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
176 ByteIOContext *pb = s->pb; |
0 | 177 AVCodecContext *enc, *audio_enc, *video_enc; |
178 PutBitContext p; | |
65 | 179 uint8_t buf1[256]; |
85
25062c9b1f86
per context frame_rate_base, this should finally fix frame_rate related av sync issues
michaelni
parents:
65
diff
changeset
|
180 int i, width, height, rate, rate_base; |
3372 | 181 int version; |
0 | 182 |
359 | 183 swf->sound_samples = 0; |
184 swf->swf_frame_number = 0; | |
185 swf->video_frame_number = 0; | |
186 | |
0 | 187 video_enc = NULL; |
188 audio_enc = NULL; | |
189 for(i=0;i<s->nb_streams;i++) { | |
820
feca73904e67
changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents:
807
diff
changeset
|
190 enc = s->streams[i]->codec; |
1854 | 191 if (enc->codec_type == CODEC_TYPE_AUDIO) { |
192 if (enc->codec_id == CODEC_ID_MP3) { | |
193 if (!enc->frame_size) { | |
194 av_log(s, AV_LOG_ERROR, "audio frame size not set\n"); | |
195 return -1; | |
196 } | |
197 audio_enc = enc; | |
3375 | 198 av_fifo_init(&swf->audio_fifo, AUDIO_FIFO_SIZE); |
1854 | 199 } else { |
1865 | 200 av_log(s, AV_LOG_ERROR, "SWF muxer only supports MP3\n"); |
1854 | 201 return -1; |
202 } | |
203 } else { | |
2309 | 204 if (enc->codec_id == CODEC_ID_VP6F || |
205 enc->codec_id == CODEC_ID_FLV1 || | |
206 enc->codec_id == CODEC_ID_MJPEG) { | |
359 | 207 video_enc = enc; |
208 } else { | |
1865 | 209 av_log(s, AV_LOG_ERROR, "SWF muxer only supports VP6, FLV1 and MJPEG\n"); |
359 | 210 return -1; |
211 } | |
212 } | |
0 | 213 } |
214 | |
215 if (!video_enc) { | |
2289 | 216 /* currently, cannot work correctly if audio only */ |
359 | 217 swf->video_type = 0; |
0 | 218 width = 320; |
219 height = 200; | |
85
25062c9b1f86
per context frame_rate_base, this should finally fix frame_rate related av sync issues
michaelni
parents:
65
diff
changeset
|
220 rate = 10; |
25062c9b1f86
per context frame_rate_base, this should finally fix frame_rate related av sync issues
michaelni
parents:
65
diff
changeset
|
221 rate_base= 1; |
0 | 222 } else { |
359 | 223 swf->video_type = video_enc->codec_id; |
0 | 224 width = video_enc->width; |
225 height = video_enc->height; | |
743 | 226 rate = video_enc->time_base.den; |
227 rate_base = video_enc->time_base.num; | |
0 | 228 } |
229 | |
2309 | 230 if (!audio_enc) { |
359 | 231 swf->audio_type = 0; |
2309 | 232 swf->samples_per_frame = (44100. * rate_base) / rate; |
359 | 233 } else { |
234 swf->audio_type = audio_enc->codec_id; | |
2309 | 235 swf->samples_per_frame = (audio_enc->sample_rate * rate_base) / rate; |
359 | 236 } |
237 | |
3372 | 238 put_tag(pb, "FWS"); |
2955
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
239 |
3372 | 240 if (!strcmp("avm2", s->oformat->name)) |
241 version = 9; | |
3369 | 242 else if (video_enc && video_enc->codec_id == CODEC_ID_VP6F) |
3372 | 243 version = 8; /* version 8 and above support VP6 codec */ |
3369 | 244 else if (video_enc && video_enc->codec_id == CODEC_ID_FLV1) |
3372 | 245 version = 6; /* version 6 and above support FLV1 codec */ |
3369 | 246 else |
3372 | 247 version = 4; /* version 4 for mpeg audio support */ |
248 put_byte(pb, version); | |
249 | |
885 | 250 put_le32(pb, DUMMY_FILE_SIZE); /* dummy size |
251 (will be patched if not streamed) */ | |
0 | 252 |
359 | 253 put_swf_rect(pb, 0, width * 20, 0, height * 20); |
85
25062c9b1f86
per context frame_rate_base, this should finally fix frame_rate related av sync issues
michaelni
parents:
65
diff
changeset
|
254 put_le16(pb, (rate * 256) / rate_base); /* frame rate */ |
0 | 255 swf->duration_pos = url_ftell(pb); |
85
25062c9b1f86
per context frame_rate_base, this should finally fix frame_rate related av sync issues
michaelni
parents:
65
diff
changeset
|
256 put_le16(pb, (uint16_t)(DUMMY_DURATION * (int64_t)rate / rate_base)); /* frame count */ |
885 | 257 |
2955
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
258 /* avm2/swf v9 (also v8?) files require a file attribute tag */ |
3372 | 259 if (version == 9) { |
2955
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
260 put_swf_tag(s, TAG_FILEATTRIBUTES); |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
261 put_le32(pb, 1<<3); /* set ActionScript v3/AVM2 flag */ |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
262 put_swf_end_tag(s); |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
263 } |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
264 |
0 | 265 /* define a shape with the jpeg inside */ |
3369 | 266 if (video_enc && video_enc->codec_id == CODEC_ID_MJPEG) { |
359 | 267 put_swf_tag(s, TAG_DEFINESHAPE); |
0 | 268 |
359 | 269 put_le16(pb, SHAPE_ID); /* ID of shape */ |
270 /* bounding rectangle */ | |
271 put_swf_rect(pb, 0, width, 0, height); | |
272 /* style info */ | |
273 put_byte(pb, 1); /* one fill style */ | |
274 put_byte(pb, 0x41); /* clipped bitmap fill */ | |
275 put_le16(pb, BITMAP_ID); /* bitmap ID */ | |
276 /* position of the bitmap */ | |
885 | 277 put_swf_matrix(pb, (int)(1.0 * (1 << FRAC_BITS)), 0, |
2308 | 278 0, (int)(1.0 * (1 << FRAC_BITS)), 0, 0); |
359 | 279 put_byte(pb, 0); /* no line style */ |
885 | 280 |
359 | 281 /* shape drawing */ |
282 init_put_bits(&p, buf1, sizeof(buf1)); | |
283 put_bits(&p, 4, 1); /* one fill bit */ | |
284 put_bits(&p, 4, 0); /* zero line bit */ | |
885 | 285 |
359 | 286 put_bits(&p, 1, 0); /* not an edge */ |
287 put_bits(&p, 5, FLAG_MOVETO | FLAG_SETFILL0); | |
288 put_bits(&p, 5, 1); /* nbits */ | |
289 put_bits(&p, 1, 0); /* X */ | |
290 put_bits(&p, 1, 0); /* Y */ | |
291 put_bits(&p, 1, 1); /* set fill style 1 */ | |
885 | 292 |
359 | 293 /* draw the rectangle ! */ |
294 put_swf_line_edge(&p, width, 0); | |
295 put_swf_line_edge(&p, 0, height); | |
296 put_swf_line_edge(&p, -width, 0); | |
297 put_swf_line_edge(&p, 0, -height); | |
885 | 298 |
359 | 299 /* end of shape */ |
300 put_bits(&p, 1, 0); /* not an edge */ | |
301 put_bits(&p, 5, 0); | |
0 | 302 |
359 | 303 flush_put_bits(&p); |
304 put_buffer(pb, buf1, pbBufPtr(&p) - p.buf); | |
0 | 305 |
359 | 306 put_swf_end_tag(s); |
307 } | |
885 | 308 |
2309 | 309 if (audio_enc && audio_enc->codec_id == CODEC_ID_MP3) { |
3373 | 310 int v = 0; |
0 | 311 |
312 /* start sound */ | |
359 | 313 put_swf_tag(s, TAG_STREAMHEAD2); |
0 | 314 switch(audio_enc->sample_rate) { |
3374 | 315 case 11025: v |= 1 << 2; break; |
316 case 22050: v |= 2 << 2; break; | |
317 case 44100: v |= 3 << 2; break; | |
0 | 318 default: |
319 /* not supported */ | |
2164 | 320 av_log(s, AV_LOG_ERROR, "swf does not support that sample rate, choose from (44100, 22050, 11025).\n"); |
0 | 321 return -1; |
322 } | |
359 | 323 v |= 0x02; /* 16 bit playback */ |
0 | 324 if (audio_enc->channels == 2) |
359 | 325 v |= 0x01; /* stereo playback */ |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
326 put_byte(s->pb, v); |
0 | 327 v |= 0x20; /* mp3 compressed */ |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
328 put_byte(s->pb, v); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
329 put_le16(s->pb, swf->samples_per_frame); /* avg samples per frame */ |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
330 put_le16(s->pb, 0); |
885 | 331 |
0 | 332 put_swf_end_tag(s); |
333 } | |
334 | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
335 put_flush_packet(s->pb); |
0 | 336 return 0; |
337 } | |
338 | |
885 | 339 static int swf_write_video(AVFormatContext *s, |
241 | 340 AVCodecContext *enc, const uint8_t *buf, int size) |
0 | 341 { |
359 | 342 SWFContext *swf = s->priv_data; |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
343 ByteIOContext *pb = s->pb; |
885 | 344 |
359 | 345 /* Flash Player limit */ |
3369 | 346 if (swf->swf_frame_number == 16000) |
370
845f9de2c883
av_log() patch by (Michel Bardiaux <mbardiaux at peaktime dot be>)
michael
parents:
360
diff
changeset
|
347 av_log(enc, AV_LOG_INFO, "warning: Flash Player limit of 16000 frames reached\n"); |
0 | 348 |
2309 | 349 if (swf->video_type == CODEC_ID_VP6F || |
350 swf->video_type == CODEC_ID_FLV1) { | |
351 if (swf->video_frame_number == 0) { | |
2308 | 352 /* create a new video object */ |
353 put_swf_tag(s, TAG_VIDEOSTREAM); | |
354 put_le16(pb, VIDEO_ID); | |
2309 | 355 put_le16(pb, 15000); /* hard flash player limit */ |
2308 | 356 put_le16(pb, enc->width); |
357 put_le16(pb, enc->height); | |
358 put_byte(pb, 0); | |
359 put_byte(pb,codec_get_tag(swf_codec_tags,swf->video_type)); | |
360 put_swf_end_tag(s); | |
885 | 361 |
2308 | 362 /* place the video object for the first time */ |
363 put_swf_tag(s, TAG_PLACEOBJECT2); | |
364 put_byte(pb, 0x36); | |
365 put_le16(pb, 1); | |
366 put_le16(pb, VIDEO_ID); | |
367 put_swf_matrix(pb, 1 << FRAC_BITS, 0, 0, 1 << FRAC_BITS, 0, 0); | |
2309 | 368 put_le16(pb, swf->video_frame_number); |
3370 | 369 put_tag(pb, "video"); |
2308 | 370 put_byte(pb, 0x00); |
371 put_swf_end_tag(s); | |
372 } else { | |
373 /* mark the character for update */ | |
374 put_swf_tag(s, TAG_PLACEOBJECT2); | |
375 put_byte(pb, 0x11); | |
376 put_le16(pb, 1); | |
2309 | 377 put_le16(pb, swf->video_frame_number); |
2308 | 378 put_swf_end_tag(s); |
379 } | |
885 | 380 |
2308 | 381 /* set video frame data */ |
382 put_swf_tag(s, TAG_VIDEOFRAME | TAG_LONG); | |
383 put_le16(pb, VIDEO_ID); | |
2309 | 384 put_le16(pb, swf->video_frame_number++); |
2308 | 385 put_buffer(pb, buf, size); |
386 put_swf_end_tag(s); | |
2309 | 387 } else if (swf->video_type == CODEC_ID_MJPEG) { |
2308 | 388 if (swf->swf_frame_number > 0) { |
389 /* remove the shape */ | |
390 put_swf_tag(s, TAG_REMOVEOBJECT); | |
391 put_le16(pb, SHAPE_ID); /* shape ID */ | |
392 put_le16(pb, 1); /* depth */ | |
393 put_swf_end_tag(s); | |
885 | 394 |
2308 | 395 /* free the bitmap */ |
396 put_swf_tag(s, TAG_FREECHARACTER); | |
397 put_le16(pb, BITMAP_ID); | |
398 put_swf_end_tag(s); | |
399 } | |
885 | 400 |
2308 | 401 put_swf_tag(s, TAG_JPEG2 | TAG_LONG); |
885 | 402 |
2308 | 403 put_le16(pb, BITMAP_ID); /* ID of the image */ |
885 | 404 |
2308 | 405 /* a dummy jpeg header seems to be required */ |
3371 | 406 put_be32(pb, 0xffd8ffd9); |
2308 | 407 /* write the jpeg image */ |
408 put_buffer(pb, buf, size); | |
885 | 409 |
2308 | 410 put_swf_end_tag(s); |
885 | 411 |
2308 | 412 /* draw the shape */ |
885 | 413 |
2308 | 414 put_swf_tag(s, TAG_PLACEOBJECT); |
415 put_le16(pb, SHAPE_ID); /* shape ID */ | |
416 put_le16(pb, 1); /* depth */ | |
417 put_swf_matrix(pb, 20 << FRAC_BITS, 0, 0, 20 << FRAC_BITS, 0, 0); | |
418 put_swf_end_tag(s); | |
419 } | |
885 | 420 |
2308 | 421 swf->swf_frame_number ++; |
0 | 422 |
359 | 423 /* streaming sound always should be placed just before showframe tags */ |
3375 | 424 if (swf->audio_type && av_fifo_size(&swf->audio_fifo)) { |
425 int frame_size = av_fifo_size(&swf->audio_fifo); | |
359 | 426 put_swf_tag(s, TAG_STREAMBLOCK | TAG_LONG); |
1854 | 427 put_le16(pb, swf->sound_samples); |
428 put_le16(pb, 0); // seek samples | |
3375 | 429 av_fifo_generic_read(&swf->audio_fifo, frame_size, &put_buffer, pb); |
359 | 430 put_swf_end_tag(s); |
885 | 431 |
359 | 432 /* update FIFO */ |
1854 | 433 swf->sound_samples = 0; |
359 | 434 } |
435 | |
0 | 436 /* output the frame */ |
437 put_swf_tag(s, TAG_SHOWFRAME); | |
438 put_swf_end_tag(s); | |
885 | 439 |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
440 put_flush_packet(s->pb); |
885 | 441 |
0 | 442 return 0; |
443 } | |
444 | |
885 | 445 static int swf_write_audio(AVFormatContext *s, |
359 | 446 AVCodecContext *enc, const uint8_t *buf, int size) |
0 | 447 { |
359 | 448 SWFContext *swf = s->priv_data; |
449 | |
450 /* Flash Player limit */ | |
3369 | 451 if (swf->swf_frame_number == 16000) |
370
845f9de2c883
av_log() patch by (Michel Bardiaux <mbardiaux at peaktime dot be>)
michael
parents:
360
diff
changeset
|
452 av_log(enc, AV_LOG_INFO, "warning: Flash Player limit of 16000 frames reached\n"); |
0 | 453 |
3375 | 454 if (av_fifo_size(&swf->audio_fifo) + size > AUDIO_FIFO_SIZE) { |
1854 | 455 av_log(s, AV_LOG_ERROR, "audio fifo too small to mux audio essence\n"); |
456 return -1; | |
359 | 457 } |
0 | 458 |
3375 | 459 av_fifo_generic_write(&swf->audio_fifo, buf, size, NULL); |
1854 | 460 swf->sound_samples += enc->frame_size; |
461 | |
359 | 462 /* if audio only stream make sure we add swf frames */ |
3369 | 463 if (swf->video_type == 0) |
359 | 464 swf_write_video(s, enc, 0, 0); |
465 | |
0 | 466 return 0; |
467 } | |
468 | |
468 | 469 static int swf_write_packet(AVFormatContext *s, AVPacket *pkt) |
0 | 470 { |
820
feca73904e67
changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents:
807
diff
changeset
|
471 AVCodecContext *codec = s->streams[pkt->stream_index]->codec; |
0 | 472 if (codec->codec_type == CODEC_TYPE_AUDIO) |
468 | 473 return swf_write_audio(s, codec, pkt->data, pkt->size); |
0 | 474 else |
468 | 475 return swf_write_video(s, codec, pkt->data, pkt->size); |
0 | 476 } |
477 | |
478 static int swf_write_trailer(AVFormatContext *s) | |
479 { | |
480 SWFContext *swf = s->priv_data; | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
481 ByteIOContext *pb = s->pb; |
0 | 482 AVCodecContext *enc, *video_enc; |
483 int file_size, i; | |
484 | |
485 video_enc = NULL; | |
486 for(i=0;i<s->nb_streams;i++) { | |
820
feca73904e67
changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents:
807
diff
changeset
|
487 enc = s->streams[i]->codec; |
0 | 488 if (enc->codec_type == CODEC_TYPE_VIDEO) |
489 video_enc = enc; | |
3375 | 490 else |
491 av_fifo_free(&swf->audio_fifo); | |
0 | 492 } |
493 | |
494 put_swf_tag(s, TAG_END); | |
495 put_swf_end_tag(s); | |
885 | 496 |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
497 put_flush_packet(s->pb); |
0 | 498 |
499 /* patch file size and number of frames if not streamed */ | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
500 if (!url_is_streamed(s->pb) && video_enc) { |
0 | 501 file_size = url_ftell(pb); |
502 url_fseek(pb, 4, SEEK_SET); | |
503 put_le32(pb, file_size); | |
504 url_fseek(pb, swf->duration_pos, SEEK_SET); | |
505 put_le16(pb, video_enc->frame_number); | |
1643
20c25a594c49
seek back at the end of file after updating header
bcoudurier
parents:
1642
diff
changeset
|
506 url_fseek(pb, file_size, SEEK_SET); |
0 | 507 } |
508 return 0; | |
509 } | |
510 | |
1169 | 511 #ifdef CONFIG_SWF_MUXER |
512 AVOutputFormat swf_muxer = { | |
0 | 513 "swf", |
514 "Flash format", | |
515 "application/x-shockwave-flash", | |
516 "swf", | |
517 sizeof(SWFContext), | |
359 | 518 CODEC_ID_MP3, |
519 CODEC_ID_FLV1, | |
0 | 520 swf_write_header, |
521 swf_write_packet, | |
522 swf_write_trailer, | |
523 }; | |
1169 | 524 #endif |
2955
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
525 #ifdef CONFIG_AVM2_MUXER |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
526 AVOutputFormat avm2_muxer = { |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
527 "avm2", |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
528 "Flash 9 (AVM2) format", |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
529 "application/x-shockwave-flash", |
2959
1b7bf70aab74
unset extension, so code path, and guess format do not choose
bcoudurier
parents:
2955
diff
changeset
|
530 NULL, |
2955
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
531 sizeof(SWFContext), |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
532 CODEC_ID_MP3, |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
533 CODEC_ID_FLV1, |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
534 swf_write_header, |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
535 swf_write_packet, |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
536 swf_write_trailer, |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
537 }; |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
538 #endif |