Mercurial > libavformat.hg
annotate swfenc.c @ 6048:e507a21a9566 libavformat
matroskaenc: Write codec time base as default duration for video tracks.
This isn't exactly semantically equivalent, but the field has already been
long abused to mean this, and writing it helps in determining a decent cfr
time base when transcoding from a mkv where the video codec stores none (VP8).
author | conrad |
---|---|
date | Mon, 24 May 2010 08:58:19 +0000 |
parents | 536e5527c1e0 |
children |
rev | line source |
---|---|
0 | 1 /* |
3356 | 2 * Flash Compatible Streaming Format muxer |
4251
77e0c7511d41
cosmetics: Remove pointless period after copyright statement non-sentences.
diego
parents:
4206
diff
changeset
|
3 * Copyright (c) 2000 Fabrice Bellard |
77e0c7511d41
cosmetics: Remove pointless period after copyright statement non-sentences.
diego
parents:
4206
diff
changeset
|
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 |
4864
7aa7c5853bb6
Split bitstream.h, put the bitstream writer stuff in the new file
stefano
parents:
4689
diff
changeset
|
23 #include "libavcodec/put_bits.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; |
3973
549a09cf23fe
Remove offset_t typedef and use int64_t directly instead.
diego
parents:
3591
diff
changeset
|
47 int64_t pos; |
0 | 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); |
4873 | 105 put_buffer(pb, buf, put_bits_ptr(&p) - p.buf); |
0 | 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); | |
4873 | 170 put_buffer(pb, buf, put_bits_ptr(&p) - p.buf); |
0 | 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 PutBitContext p; |
65 | 178 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
|
179 int i, width, height, rate, rate_base; |
3372 | 180 int version; |
0 | 181 |
359 | 182 swf->sound_samples = 0; |
183 swf->swf_frame_number = 0; | |
184 swf->video_frame_number = 0; | |
185 | |
0 | 186 for(i=0;i<s->nb_streams;i++) { |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
187 AVCodecContext *enc = s->streams[i]->codec; |
5910
536e5527c1e0
Define AVMediaType enum, and use it instead of enum CodecType, which
stefano
parents:
5058
diff
changeset
|
188 if (enc->codec_type == AVMEDIA_TYPE_AUDIO) { |
1854 | 189 if (enc->codec_id == CODEC_ID_MP3) { |
190 if (!enc->frame_size) { | |
191 av_log(s, AV_LOG_ERROR, "audio frame size not set\n"); | |
192 return -1; | |
193 } | |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
194 swf->audio_enc = enc; |
4669
d6eb19c43e99
Allocate AVFifoBuffer through the fifo API to reduce future API/ABI issues.
michael
parents:
4398
diff
changeset
|
195 swf->audio_fifo= av_fifo_alloc(AUDIO_FIFO_SIZE); |
4947 | 196 if (!swf->audio_fifo) |
197 return AVERROR(ENOMEM); | |
1854 | 198 } else { |
1865 | 199 av_log(s, AV_LOG_ERROR, "SWF muxer only supports MP3\n"); |
1854 | 200 return -1; |
201 } | |
202 } else { | |
2309 | 203 if (enc->codec_id == CODEC_ID_VP6F || |
204 enc->codec_id == CODEC_ID_FLV1 || | |
205 enc->codec_id == CODEC_ID_MJPEG) { | |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
206 swf->video_enc = enc; |
359 | 207 } else { |
1865 | 208 av_log(s, AV_LOG_ERROR, "SWF muxer only supports VP6, FLV1 and MJPEG\n"); |
359 | 209 return -1; |
210 } | |
211 } | |
0 | 212 } |
213 | |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
214 if (!swf->video_enc) { |
2289 | 215 /* currently, cannot work correctly if audio only */ |
0 | 216 width = 320; |
217 height = 200; | |
85
25062c9b1f86
per context frame_rate_base, this should finally fix frame_rate related av sync issues
michaelni
parents:
65
diff
changeset
|
218 rate = 10; |
25062c9b1f86
per context frame_rate_base, this should finally fix frame_rate related av sync issues
michaelni
parents:
65
diff
changeset
|
219 rate_base= 1; |
0 | 220 } else { |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
221 width = swf->video_enc->width; |
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
222 height = swf->video_enc->height; |
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
223 rate = swf->video_enc->time_base.den; |
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
224 rate_base = swf->video_enc->time_base.num; |
0 | 225 } |
226 | |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
227 if (!swf->audio_enc) |
2309 | 228 swf->samples_per_frame = (44100. * rate_base) / rate; |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
229 else |
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
230 swf->samples_per_frame = (swf->audio_enc->sample_rate * rate_base) / rate; |
359 | 231 |
3372 | 232 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
|
233 |
3372 | 234 if (!strcmp("avm2", s->oformat->name)) |
235 version = 9; | |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
236 else if (swf->video_enc && swf->video_enc->codec_id == CODEC_ID_VP6F) |
3372 | 237 version = 8; /* version 8 and above support VP6 codec */ |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
238 else if (swf->video_enc && swf->video_enc->codec_id == CODEC_ID_FLV1) |
3372 | 239 version = 6; /* version 6 and above support FLV1 codec */ |
3369 | 240 else |
3372 | 241 version = 4; /* version 4 for mpeg audio support */ |
242 put_byte(pb, version); | |
243 | |
885 | 244 put_le32(pb, DUMMY_FILE_SIZE); /* dummy size |
245 (will be patched if not streamed) */ | |
0 | 246 |
359 | 247 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
|
248 put_le16(pb, (rate * 256) / rate_base); /* frame rate */ |
0 | 249 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
|
250 put_le16(pb, (uint16_t)(DUMMY_DURATION * (int64_t)rate / rate_base)); /* frame count */ |
885 | 251 |
2955
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
252 /* avm2/swf v9 (also v8?) files require a file attribute tag */ |
3372 | 253 if (version == 9) { |
2955
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
254 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
|
255 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
|
256 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
|
257 } |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
258 |
0 | 259 /* define a shape with the jpeg inside */ |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
260 if (swf->video_enc && swf->video_enc->codec_id == CODEC_ID_MJPEG) { |
359 | 261 put_swf_tag(s, TAG_DEFINESHAPE); |
0 | 262 |
359 | 263 put_le16(pb, SHAPE_ID); /* ID of shape */ |
264 /* bounding rectangle */ | |
265 put_swf_rect(pb, 0, width, 0, height); | |
266 /* style info */ | |
267 put_byte(pb, 1); /* one fill style */ | |
268 put_byte(pb, 0x41); /* clipped bitmap fill */ | |
269 put_le16(pb, BITMAP_ID); /* bitmap ID */ | |
270 /* position of the bitmap */ | |
885 | 271 put_swf_matrix(pb, (int)(1.0 * (1 << FRAC_BITS)), 0, |
2308 | 272 0, (int)(1.0 * (1 << FRAC_BITS)), 0, 0); |
359 | 273 put_byte(pb, 0); /* no line style */ |
885 | 274 |
359 | 275 /* shape drawing */ |
276 init_put_bits(&p, buf1, sizeof(buf1)); | |
277 put_bits(&p, 4, 1); /* one fill bit */ | |
278 put_bits(&p, 4, 0); /* zero line bit */ | |
885 | 279 |
359 | 280 put_bits(&p, 1, 0); /* not an edge */ |
281 put_bits(&p, 5, FLAG_MOVETO | FLAG_SETFILL0); | |
282 put_bits(&p, 5, 1); /* nbits */ | |
283 put_bits(&p, 1, 0); /* X */ | |
284 put_bits(&p, 1, 0); /* Y */ | |
285 put_bits(&p, 1, 1); /* set fill style 1 */ | |
885 | 286 |
359 | 287 /* draw the rectangle ! */ |
288 put_swf_line_edge(&p, width, 0); | |
289 put_swf_line_edge(&p, 0, height); | |
290 put_swf_line_edge(&p, -width, 0); | |
291 put_swf_line_edge(&p, 0, -height); | |
885 | 292 |
359 | 293 /* end of shape */ |
294 put_bits(&p, 1, 0); /* not an edge */ | |
295 put_bits(&p, 5, 0); | |
0 | 296 |
359 | 297 flush_put_bits(&p); |
4873 | 298 put_buffer(pb, buf1, put_bits_ptr(&p) - p.buf); |
0 | 299 |
359 | 300 put_swf_end_tag(s); |
301 } | |
885 | 302 |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
303 if (swf->audio_enc && swf->audio_enc->codec_id == CODEC_ID_MP3) { |
3373 | 304 int v = 0; |
0 | 305 |
306 /* start sound */ | |
359 | 307 put_swf_tag(s, TAG_STREAMHEAD2); |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
308 switch(swf->audio_enc->sample_rate) { |
3374 | 309 case 11025: v |= 1 << 2; break; |
310 case 22050: v |= 2 << 2; break; | |
311 case 44100: v |= 3 << 2; break; | |
0 | 312 default: |
313 /* not supported */ | |
2164 | 314 av_log(s, AV_LOG_ERROR, "swf does not support that sample rate, choose from (44100, 22050, 11025).\n"); |
0 | 315 return -1; |
316 } | |
359 | 317 v |= 0x02; /* 16 bit playback */ |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
318 if (swf->audio_enc->channels == 2) |
359 | 319 v |= 0x01; /* stereo playback */ |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
320 put_byte(s->pb, v); |
0 | 321 v |= 0x20; /* mp3 compressed */ |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
322 put_byte(s->pb, v); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
323 put_le16(s->pb, swf->samples_per_frame); /* avg samples per frame */ |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
324 put_le16(s->pb, 0); |
885 | 325 |
0 | 326 put_swf_end_tag(s); |
327 } | |
328 | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
329 put_flush_packet(s->pb); |
0 | 330 return 0; |
331 } | |
332 | |
885 | 333 static int swf_write_video(AVFormatContext *s, |
241 | 334 AVCodecContext *enc, const uint8_t *buf, int size) |
0 | 335 { |
359 | 336 SWFContext *swf = s->priv_data; |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
337 ByteIOContext *pb = s->pb; |
885 | 338 |
359 | 339 /* Flash Player limit */ |
3369 | 340 if (swf->swf_frame_number == 16000) |
370
845f9de2c883
av_log() patch by (Michel Bardiaux <mbardiaux at peaktime dot be>)
michael
parents:
360
diff
changeset
|
341 av_log(enc, AV_LOG_INFO, "warning: Flash Player limit of 16000 frames reached\n"); |
0 | 342 |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
343 if (enc->codec_id == CODEC_ID_VP6F || |
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
344 enc->codec_id == CODEC_ID_FLV1) { |
2309 | 345 if (swf->video_frame_number == 0) { |
2308 | 346 /* create a new video object */ |
347 put_swf_tag(s, TAG_VIDEOSTREAM); | |
348 put_le16(pb, VIDEO_ID); | |
3591
aa6e8ff72d9e
update swf video frame number when muxing done, fix #439
bcoudurier
parents:
3424
diff
changeset
|
349 swf->vframes_pos = url_ftell(pb); |
2309 | 350 put_le16(pb, 15000); /* hard flash player limit */ |
2308 | 351 put_le16(pb, enc->width); |
352 put_le16(pb, enc->height); | |
353 put_byte(pb, 0); | |
5058
33a244b7ca65
Add ff_ prefixes to exported symbols in libavformat/riff.h.
diego
parents:
4947
diff
changeset
|
354 put_byte(pb,ff_codec_get_tag(swf_codec_tags,enc->codec_id)); |
2308 | 355 put_swf_end_tag(s); |
885 | 356 |
2308 | 357 /* place the video object for the first time */ |
358 put_swf_tag(s, TAG_PLACEOBJECT2); | |
359 put_byte(pb, 0x36); | |
360 put_le16(pb, 1); | |
361 put_le16(pb, VIDEO_ID); | |
362 put_swf_matrix(pb, 1 << FRAC_BITS, 0, 0, 1 << FRAC_BITS, 0, 0); | |
2309 | 363 put_le16(pb, swf->video_frame_number); |
3370 | 364 put_tag(pb, "video"); |
2308 | 365 put_byte(pb, 0x00); |
366 put_swf_end_tag(s); | |
367 } else { | |
368 /* mark the character for update */ | |
369 put_swf_tag(s, TAG_PLACEOBJECT2); | |
370 put_byte(pb, 0x11); | |
371 put_le16(pb, 1); | |
2309 | 372 put_le16(pb, swf->video_frame_number); |
2308 | 373 put_swf_end_tag(s); |
374 } | |
885 | 375 |
2308 | 376 /* set video frame data */ |
377 put_swf_tag(s, TAG_VIDEOFRAME | TAG_LONG); | |
378 put_le16(pb, VIDEO_ID); | |
2309 | 379 put_le16(pb, swf->video_frame_number++); |
2308 | 380 put_buffer(pb, buf, size); |
381 put_swf_end_tag(s); | |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
382 } else if (enc->codec_id == CODEC_ID_MJPEG) { |
2308 | 383 if (swf->swf_frame_number > 0) { |
384 /* remove the shape */ | |
385 put_swf_tag(s, TAG_REMOVEOBJECT); | |
386 put_le16(pb, SHAPE_ID); /* shape ID */ | |
387 put_le16(pb, 1); /* depth */ | |
388 put_swf_end_tag(s); | |
885 | 389 |
2308 | 390 /* free the bitmap */ |
391 put_swf_tag(s, TAG_FREECHARACTER); | |
392 put_le16(pb, BITMAP_ID); | |
393 put_swf_end_tag(s); | |
394 } | |
885 | 395 |
2308 | 396 put_swf_tag(s, TAG_JPEG2 | TAG_LONG); |
885 | 397 |
2308 | 398 put_le16(pb, BITMAP_ID); /* ID of the image */ |
885 | 399 |
2308 | 400 /* a dummy jpeg header seems to be required */ |
3371 | 401 put_be32(pb, 0xffd8ffd9); |
2308 | 402 /* write the jpeg image */ |
403 put_buffer(pb, buf, size); | |
885 | 404 |
2308 | 405 put_swf_end_tag(s); |
885 | 406 |
2308 | 407 /* draw the shape */ |
885 | 408 |
2308 | 409 put_swf_tag(s, TAG_PLACEOBJECT); |
410 put_le16(pb, SHAPE_ID); /* shape ID */ | |
411 put_le16(pb, 1); /* depth */ | |
412 put_swf_matrix(pb, 20 << FRAC_BITS, 0, 0, 20 << FRAC_BITS, 0, 0); | |
413 put_swf_end_tag(s); | |
414 } | |
885 | 415 |
3381 | 416 swf->swf_frame_number++; |
0 | 417 |
359 | 418 /* streaming sound always should be placed just before showframe tags */ |
4669
d6eb19c43e99
Allocate AVFifoBuffer through the fifo API to reduce future API/ABI issues.
michael
parents:
4398
diff
changeset
|
419 if (swf->audio_enc && av_fifo_size(swf->audio_fifo)) { |
d6eb19c43e99
Allocate AVFifoBuffer through the fifo API to reduce future API/ABI issues.
michael
parents:
4398
diff
changeset
|
420 int frame_size = av_fifo_size(swf->audio_fifo); |
359 | 421 put_swf_tag(s, TAG_STREAMBLOCK | TAG_LONG); |
1854 | 422 put_le16(pb, swf->sound_samples); |
423 put_le16(pb, 0); // seek samples | |
4689
fc0a165de804
Reorder arguments for av_fifo_generic_read to be more logical and
reimar
parents:
4669
diff
changeset
|
424 av_fifo_generic_read(swf->audio_fifo, pb, frame_size, &put_buffer); |
359 | 425 put_swf_end_tag(s); |
885 | 426 |
359 | 427 /* update FIFO */ |
1854 | 428 swf->sound_samples = 0; |
359 | 429 } |
430 | |
0 | 431 /* output the frame */ |
432 put_swf_tag(s, TAG_SHOWFRAME); | |
433 put_swf_end_tag(s); | |
885 | 434 |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
435 put_flush_packet(s->pb); |
885 | 436 |
0 | 437 return 0; |
438 } | |
439 | |
885 | 440 static int swf_write_audio(AVFormatContext *s, |
4398
043d314bb216
Remove const qualifier from function argument to eliminate the warning
diego
parents:
4251
diff
changeset
|
441 AVCodecContext *enc, uint8_t *buf, int size) |
0 | 442 { |
359 | 443 SWFContext *swf = s->priv_data; |
444 | |
445 /* Flash Player limit */ | |
3369 | 446 if (swf->swf_frame_number == 16000) |
370
845f9de2c883
av_log() patch by (Michel Bardiaux <mbardiaux at peaktime dot be>)
michael
parents:
360
diff
changeset
|
447 av_log(enc, AV_LOG_INFO, "warning: Flash Player limit of 16000 frames reached\n"); |
0 | 448 |
4669
d6eb19c43e99
Allocate AVFifoBuffer through the fifo API to reduce future API/ABI issues.
michael
parents:
4398
diff
changeset
|
449 if (av_fifo_size(swf->audio_fifo) + size > AUDIO_FIFO_SIZE) { |
1854 | 450 av_log(s, AV_LOG_ERROR, "audio fifo too small to mux audio essence\n"); |
451 return -1; | |
359 | 452 } |
0 | 453 |
4669
d6eb19c43e99
Allocate AVFifoBuffer through the fifo API to reduce future API/ABI issues.
michael
parents:
4398
diff
changeset
|
454 av_fifo_generic_write(swf->audio_fifo, buf, size, NULL); |
1854 | 455 swf->sound_samples += enc->frame_size; |
456 | |
359 | 457 /* if audio only stream make sure we add swf frames */ |
3379
b3827117c786
simplify, use pointer to codec context in struct instead of only id
bcoudurier
parents:
3375
diff
changeset
|
458 if (!swf->video_enc) |
359 | 459 swf_write_video(s, enc, 0, 0); |
460 | |
0 | 461 return 0; |
462 } | |
463 | |
468 | 464 static int swf_write_packet(AVFormatContext *s, AVPacket *pkt) |
0 | 465 { |
820
feca73904e67
changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents:
807
diff
changeset
|
466 AVCodecContext *codec = s->streams[pkt->stream_index]->codec; |
5910
536e5527c1e0
Define AVMediaType enum, and use it instead of enum CodecType, which
stefano
parents:
5058
diff
changeset
|
467 if (codec->codec_type == AVMEDIA_TYPE_AUDIO) |
468 | 468 return swf_write_audio(s, codec, pkt->data, pkt->size); |
0 | 469 else |
468 | 470 return swf_write_video(s, codec, pkt->data, pkt->size); |
0 | 471 } |
472 | |
473 static int swf_write_trailer(AVFormatContext *s) | |
474 { | |
475 SWFContext *swf = s->priv_data; | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
476 ByteIOContext *pb = s->pb; |
0 | 477 AVCodecContext *enc, *video_enc; |
478 int file_size, i; | |
479 | |
480 video_enc = NULL; | |
481 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
|
482 enc = s->streams[i]->codec; |
5910
536e5527c1e0
Define AVMediaType enum, and use it instead of enum CodecType, which
stefano
parents:
5058
diff
changeset
|
483 if (enc->codec_type == AVMEDIA_TYPE_VIDEO) |
0 | 484 video_enc = enc; |
3375 | 485 else |
4669
d6eb19c43e99
Allocate AVFifoBuffer through the fifo API to reduce future API/ABI issues.
michael
parents:
4398
diff
changeset
|
486 av_fifo_free(swf->audio_fifo); |
0 | 487 } |
488 | |
489 put_swf_tag(s, TAG_END); | |
490 put_swf_end_tag(s); | |
885 | 491 |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
492 put_flush_packet(s->pb); |
0 | 493 |
494 /* patch file size and number of frames if not streamed */ | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2402
diff
changeset
|
495 if (!url_is_streamed(s->pb) && video_enc) { |
0 | 496 file_size = url_ftell(pb); |
497 url_fseek(pb, 4, SEEK_SET); | |
498 put_le32(pb, file_size); | |
499 url_fseek(pb, swf->duration_pos, SEEK_SET); | |
3382 | 500 put_le16(pb, swf->video_frame_number); |
3591
aa6e8ff72d9e
update swf video frame number when muxing done, fix #439
bcoudurier
parents:
3424
diff
changeset
|
501 url_fseek(pb, swf->vframes_pos, SEEK_SET); |
aa6e8ff72d9e
update swf video frame number when muxing done, fix #439
bcoudurier
parents:
3424
diff
changeset
|
502 put_le16(pb, swf->video_frame_number); |
1643
20c25a594c49
seek back at the end of file after updating header
bcoudurier
parents:
1642
diff
changeset
|
503 url_fseek(pb, file_size, SEEK_SET); |
0 | 504 } |
505 return 0; | |
506 } | |
507 | |
4206 | 508 #if CONFIG_SWF_MUXER |
1169 | 509 AVOutputFormat swf_muxer = { |
0 | 510 "swf", |
3424
7a0230981402
Make long_names in lavf/lavdev optional depending on CONFIG_SMALL.
diego
parents:
3382
diff
changeset
|
511 NULL_IF_CONFIG_SMALL("Flash format"), |
0 | 512 "application/x-shockwave-flash", |
513 "swf", | |
514 sizeof(SWFContext), | |
359 | 515 CODEC_ID_MP3, |
516 CODEC_ID_FLV1, | |
0 | 517 swf_write_header, |
518 swf_write_packet, | |
519 swf_write_trailer, | |
520 }; | |
1169 | 521 #endif |
4206 | 522 #if CONFIG_AVM2_MUXER |
2955
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
523 AVOutputFormat avm2_muxer = { |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
524 "avm2", |
3424
7a0230981402
Make long_names in lavf/lavdev optional depending on CONFIG_SMALL.
diego
parents:
3382
diff
changeset
|
525 NULL_IF_CONFIG_SMALL("Flash 9 (AVM2) format"), |
2955
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
526 "application/x-shockwave-flash", |
2959
1b7bf70aab74
unset extension, so code path, and guess format do not choose
bcoudurier
parents:
2955
diff
changeset
|
527 NULL, |
2955
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
528 sizeof(SWFContext), |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
529 CODEC_ID_MP3, |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
530 CODEC_ID_FLV1, |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
531 swf_write_header, |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
532 swf_write_packet, |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
533 swf_write_trailer, |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
534 }; |
b2d1cd7ab383
new avm2 (flash 9) muxer, patch by Paul Egan, paulegan at mail dot com
bcoudurier
parents:
2913
diff
changeset
|
535 #endif |