00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023 #include "libavcodec/put_bits.h"
00024 #include "libavutil/avassert.h"
00025 #include "avformat.h"
00026 #include "swf.h"
00027
00028 static void put_swf_tag(AVFormatContext *s, int tag)
00029 {
00030 SWFContext *swf = s->priv_data;
00031 AVIOContext *pb = s->pb;
00032
00033 swf->tag_pos = avio_tell(pb);
00034 swf->tag = tag;
00035
00036 if (tag & TAG_LONG) {
00037 avio_wl16(pb, 0);
00038 avio_wl32(pb, 0);
00039 } else {
00040 avio_wl16(pb, 0);
00041 }
00042 }
00043
00044 static void put_swf_end_tag(AVFormatContext *s)
00045 {
00046 SWFContext *swf = s->priv_data;
00047 AVIOContext *pb = s->pb;
00048 int64_t pos;
00049 int tag_len, tag;
00050
00051 pos = avio_tell(pb);
00052 tag_len = pos - swf->tag_pos - 2;
00053 tag = swf->tag;
00054 avio_seek(pb, swf->tag_pos, SEEK_SET);
00055 if (tag & TAG_LONG) {
00056 tag &= ~TAG_LONG;
00057 avio_wl16(pb, (tag << 6) | 0x3f);
00058 avio_wl32(pb, tag_len - 4);
00059 } else {
00060 av_assert0(tag_len < 0x3f);
00061 avio_wl16(pb, (tag << 6) | tag_len);
00062 }
00063 avio_seek(pb, pos, SEEK_SET);
00064 }
00065
00066 static inline void max_nbits(int *nbits_ptr, int val)
00067 {
00068 int n;
00069
00070 if (val == 0)
00071 return;
00072 val = abs(val);
00073 n = 1;
00074 while (val != 0) {
00075 n++;
00076 val >>= 1;
00077 }
00078 if (n > *nbits_ptr)
00079 *nbits_ptr = n;
00080 }
00081
00082 static void put_swf_rect(AVIOContext *pb,
00083 int xmin, int xmax, int ymin, int ymax)
00084 {
00085 PutBitContext p;
00086 uint8_t buf[256];
00087 int nbits, mask;
00088
00089 init_put_bits(&p, buf, sizeof(buf));
00090
00091 nbits = 0;
00092 max_nbits(&nbits, xmin);
00093 max_nbits(&nbits, xmax);
00094 max_nbits(&nbits, ymin);
00095 max_nbits(&nbits, ymax);
00096 mask = (1 << nbits) - 1;
00097
00098
00099 put_bits(&p, 5, nbits);
00100 put_bits(&p, nbits, xmin & mask);
00101 put_bits(&p, nbits, xmax & mask);
00102 put_bits(&p, nbits, ymin & mask);
00103 put_bits(&p, nbits, ymax & mask);
00104
00105 flush_put_bits(&p);
00106 avio_write(pb, buf, put_bits_ptr(&p) - p.buf);
00107 }
00108
00109 static void put_swf_line_edge(PutBitContext *pb, int dx, int dy)
00110 {
00111 int nbits, mask;
00112
00113 put_bits(pb, 1, 1);
00114 put_bits(pb, 1, 1);
00115 nbits = 2;
00116 max_nbits(&nbits, dx);
00117 max_nbits(&nbits, dy);
00118
00119 mask = (1 << nbits) - 1;
00120 put_bits(pb, 4, nbits - 2);
00121 if (dx == 0) {
00122 put_bits(pb, 1, 0);
00123 put_bits(pb, 1, 1);
00124 put_bits(pb, nbits, dy & mask);
00125 } else if (dy == 0) {
00126 put_bits(pb, 1, 0);
00127 put_bits(pb, 1, 0);
00128 put_bits(pb, nbits, dx & mask);
00129 } else {
00130 put_bits(pb, 1, 1);
00131 put_bits(pb, nbits, dx & mask);
00132 put_bits(pb, nbits, dy & mask);
00133 }
00134 }
00135
00136 #define FRAC_BITS 16
00137
00138 static void put_swf_matrix(AVIOContext *pb,
00139 int a, int b, int c, int d, int tx, int ty)
00140 {
00141 PutBitContext p;
00142 uint8_t buf[256];
00143 int nbits;
00144
00145 init_put_bits(&p, buf, sizeof(buf));
00146
00147 put_bits(&p, 1, 1);
00148 nbits = 1;
00149 max_nbits(&nbits, a);
00150 max_nbits(&nbits, d);
00151 put_bits(&p, 5, nbits);
00152 put_bits(&p, nbits, a);
00153 put_bits(&p, nbits, d);
00154
00155 put_bits(&p, 1, 1);
00156 nbits = 1;
00157 max_nbits(&nbits, c);
00158 max_nbits(&nbits, b);
00159 put_bits(&p, 5, nbits);
00160 put_bits(&p, nbits, c);
00161 put_bits(&p, nbits, b);
00162
00163 nbits = 1;
00164 max_nbits(&nbits, tx);
00165 max_nbits(&nbits, ty);
00166 put_bits(&p, 5, nbits);
00167 put_bits(&p, nbits, tx);
00168 put_bits(&p, nbits, ty);
00169
00170 flush_put_bits(&p);
00171 avio_write(pb, buf, put_bits_ptr(&p) - p.buf);
00172 }
00173
00174 static int swf_write_header(AVFormatContext *s)
00175 {
00176 SWFContext *swf = s->priv_data;
00177 AVIOContext *pb = s->pb;
00178 PutBitContext p;
00179 uint8_t buf1[256];
00180 int i, width, height, rate, rate_base;
00181 int version;
00182
00183 swf->sound_samples = 0;
00184 swf->swf_frame_number = 0;
00185 swf->video_frame_number = 0;
00186
00187 for(i=0;i<s->nb_streams;i++) {
00188 AVCodecContext *enc = s->streams[i]->codec;
00189 if (enc->codec_type == AVMEDIA_TYPE_AUDIO) {
00190 if (swf->audio_enc) {
00191 av_log(s, AV_LOG_ERROR, "SWF muxer only supports 1 audio stream\n");
00192 return AVERROR_INVALIDDATA;
00193 }
00194 if (enc->codec_id == AV_CODEC_ID_MP3) {
00195 swf->audio_enc = enc;
00196 swf->audio_fifo= av_fifo_alloc(AUDIO_FIFO_SIZE);
00197 if (!swf->audio_fifo)
00198 return AVERROR(ENOMEM);
00199 } else {
00200 av_log(s, AV_LOG_ERROR, "SWF muxer only supports MP3\n");
00201 return -1;
00202 }
00203 } else {
00204 if (swf->video_enc) {
00205 av_log(s, AV_LOG_ERROR, "SWF muxer only supports 1 video stream\n");
00206 return AVERROR_INVALIDDATA;
00207 }
00208 if (enc->codec_id == AV_CODEC_ID_VP6F ||
00209 enc->codec_id == AV_CODEC_ID_FLV1 ||
00210 enc->codec_id == AV_CODEC_ID_MJPEG) {
00211 swf->video_enc = enc;
00212 } else {
00213 av_log(s, AV_LOG_ERROR, "SWF muxer only supports VP6, FLV1 and MJPEG\n");
00214 return -1;
00215 }
00216 }
00217 }
00218
00219 if (!swf->video_enc) {
00220
00221 width = 320;
00222 height = 200;
00223 rate = 10;
00224 rate_base= 1;
00225 } else {
00226 width = swf->video_enc->width;
00227 height = swf->video_enc->height;
00228 rate = swf->video_enc->time_base.den;
00229 rate_base = swf->video_enc->time_base.num;
00230 }
00231
00232 if (!swf->audio_enc)
00233 swf->samples_per_frame = (44100. * rate_base) / rate;
00234 else
00235 swf->samples_per_frame = (swf->audio_enc->sample_rate * rate_base) / rate;
00236
00237 avio_write(pb, "FWS", 3);
00238
00239 if (!strcmp("avm2", s->oformat->name))
00240 version = 9;
00241 else if (swf->video_enc && swf->video_enc->codec_id == AV_CODEC_ID_VP6F)
00242 version = 8;
00243 else if (swf->video_enc && swf->video_enc->codec_id == AV_CODEC_ID_FLV1)
00244 version = 6;
00245 else
00246 version = 4;
00247 avio_w8(pb, version);
00248
00249 avio_wl32(pb, DUMMY_FILE_SIZE);
00250
00251
00252 put_swf_rect(pb, 0, width * 20, 0, height * 20);
00253 avio_wl16(pb, (rate * 256) / rate_base);
00254 swf->duration_pos = avio_tell(pb);
00255 avio_wl16(pb, (uint16_t)(DUMMY_DURATION * (int64_t)rate / rate_base));
00256
00257
00258 if (version == 9) {
00259 put_swf_tag(s, TAG_FILEATTRIBUTES);
00260 avio_wl32(pb, 1<<3);
00261 put_swf_end_tag(s);
00262 }
00263
00264
00265 if (swf->video_enc && swf->video_enc->codec_id == AV_CODEC_ID_MJPEG) {
00266 put_swf_tag(s, TAG_DEFINESHAPE);
00267
00268 avio_wl16(pb, SHAPE_ID);
00269
00270 put_swf_rect(pb, 0, width, 0, height);
00271
00272 avio_w8(pb, 1);
00273 avio_w8(pb, 0x41);
00274 avio_wl16(pb, BITMAP_ID);
00275
00276 put_swf_matrix(pb, (int)(1.0 * (1 << FRAC_BITS)), 0,
00277 0, (int)(1.0 * (1 << FRAC_BITS)), 0, 0);
00278 avio_w8(pb, 0);
00279
00280
00281 init_put_bits(&p, buf1, sizeof(buf1));
00282 put_bits(&p, 4, 1);
00283 put_bits(&p, 4, 0);
00284
00285 put_bits(&p, 1, 0);
00286 put_bits(&p, 5, FLAG_MOVETO | FLAG_SETFILL0);
00287 put_bits(&p, 5, 1);
00288 put_bits(&p, 1, 0);
00289 put_bits(&p, 1, 0);
00290 put_bits(&p, 1, 1);
00291
00292
00293 put_swf_line_edge(&p, width, 0);
00294 put_swf_line_edge(&p, 0, height);
00295 put_swf_line_edge(&p, -width, 0);
00296 put_swf_line_edge(&p, 0, -height);
00297
00298
00299 put_bits(&p, 1, 0);
00300 put_bits(&p, 5, 0);
00301
00302 flush_put_bits(&p);
00303 avio_write(pb, buf1, put_bits_ptr(&p) - p.buf);
00304
00305 put_swf_end_tag(s);
00306 }
00307
00308 if (swf->audio_enc && swf->audio_enc->codec_id == AV_CODEC_ID_MP3) {
00309 int v = 0;
00310
00311
00312 put_swf_tag(s, TAG_STREAMHEAD2);
00313 switch(swf->audio_enc->sample_rate) {
00314 case 11025: v |= 1 << 2; break;
00315 case 22050: v |= 2 << 2; break;
00316 case 44100: v |= 3 << 2; break;
00317 default:
00318
00319 av_log(s, AV_LOG_ERROR, "swf does not support that sample rate, choose from (44100, 22050, 11025).\n");
00320 return -1;
00321 }
00322 v |= 0x02;
00323 if (swf->audio_enc->channels == 2)
00324 v |= 0x01;
00325 avio_w8(s->pb, v);
00326 v |= 0x20;
00327 avio_w8(s->pb, v);
00328 avio_wl16(s->pb, swf->samples_per_frame);
00329 avio_wl16(s->pb, 0);
00330
00331 put_swf_end_tag(s);
00332 }
00333
00334 avio_flush(s->pb);
00335 return 0;
00336 }
00337
00338 static int swf_write_video(AVFormatContext *s,
00339 AVCodecContext *enc, const uint8_t *buf, int size)
00340 {
00341 SWFContext *swf = s->priv_data;
00342 AVIOContext *pb = s->pb;
00343
00344
00345 if (swf->swf_frame_number == 16000)
00346 av_log(enc, AV_LOG_INFO, "warning: Flash Player limit of 16000 frames reached\n");
00347
00348 if (enc->codec_id == AV_CODEC_ID_VP6F ||
00349 enc->codec_id == AV_CODEC_ID_FLV1) {
00350 if (swf->video_frame_number == 0) {
00351
00352 put_swf_tag(s, TAG_VIDEOSTREAM);
00353 avio_wl16(pb, VIDEO_ID);
00354 swf->vframes_pos = avio_tell(pb);
00355 avio_wl16(pb, 15000);
00356 avio_wl16(pb, enc->width);
00357 avio_wl16(pb, enc->height);
00358 avio_w8(pb, 0);
00359 avio_w8(pb,ff_codec_get_tag(ff_swf_codec_tags, enc->codec_id));
00360 put_swf_end_tag(s);
00361
00362
00363 put_swf_tag(s, TAG_PLACEOBJECT2);
00364 avio_w8(pb, 0x36);
00365 avio_wl16(pb, 1);
00366 avio_wl16(pb, VIDEO_ID);
00367 put_swf_matrix(pb, 1 << FRAC_BITS, 0, 0, 1 << FRAC_BITS, 0, 0);
00368 avio_wl16(pb, swf->video_frame_number);
00369 avio_write(pb, "video", 5);
00370 avio_w8(pb, 0x00);
00371 put_swf_end_tag(s);
00372 } else {
00373
00374 put_swf_tag(s, TAG_PLACEOBJECT2);
00375 avio_w8(pb, 0x11);
00376 avio_wl16(pb, 1);
00377 avio_wl16(pb, swf->video_frame_number);
00378 put_swf_end_tag(s);
00379 }
00380
00381
00382 put_swf_tag(s, TAG_VIDEOFRAME | TAG_LONG);
00383 avio_wl16(pb, VIDEO_ID);
00384 avio_wl16(pb, swf->video_frame_number++);
00385 avio_write(pb, buf, size);
00386 put_swf_end_tag(s);
00387 } else if (enc->codec_id == AV_CODEC_ID_MJPEG) {
00388 if (swf->swf_frame_number > 0) {
00389
00390 put_swf_tag(s, TAG_REMOVEOBJECT);
00391 avio_wl16(pb, SHAPE_ID);
00392 avio_wl16(pb, 1);
00393 put_swf_end_tag(s);
00394
00395
00396 put_swf_tag(s, TAG_FREECHARACTER);
00397 avio_wl16(pb, BITMAP_ID);
00398 put_swf_end_tag(s);
00399 }
00400
00401 put_swf_tag(s, TAG_JPEG2 | TAG_LONG);
00402
00403 avio_wl16(pb, BITMAP_ID);
00404
00405
00406 avio_wb32(pb, 0xffd8ffd9);
00407
00408 avio_write(pb, buf, size);
00409
00410 put_swf_end_tag(s);
00411
00412
00413
00414 put_swf_tag(s, TAG_PLACEOBJECT);
00415 avio_wl16(pb, SHAPE_ID);
00416 avio_wl16(pb, 1);
00417 put_swf_matrix(pb, 20 << FRAC_BITS, 0, 0, 20 << FRAC_BITS, 0, 0);
00418 put_swf_end_tag(s);
00419 }
00420
00421 swf->swf_frame_number++;
00422
00423
00424 if (swf->audio_enc && av_fifo_size(swf->audio_fifo)) {
00425 int frame_size = av_fifo_size(swf->audio_fifo);
00426 put_swf_tag(s, TAG_STREAMBLOCK | TAG_LONG);
00427 avio_wl16(pb, swf->sound_samples);
00428 avio_wl16(pb, 0);
00429 av_fifo_generic_read(swf->audio_fifo, pb, frame_size, (void*)avio_write);
00430 put_swf_end_tag(s);
00431
00432
00433 swf->sound_samples = 0;
00434 }
00435
00436
00437 put_swf_tag(s, TAG_SHOWFRAME);
00438 put_swf_end_tag(s);
00439
00440 avio_flush(s->pb);
00441
00442 return 0;
00443 }
00444
00445 static int swf_write_audio(AVFormatContext *s,
00446 AVCodecContext *enc, uint8_t *buf, int size)
00447 {
00448 SWFContext *swf = s->priv_data;
00449
00450
00451 if (swf->swf_frame_number == 16000)
00452 av_log(enc, AV_LOG_INFO, "warning: Flash Player limit of 16000 frames reached\n");
00453
00454 if (av_fifo_size(swf->audio_fifo) + size > AUDIO_FIFO_SIZE) {
00455 av_log(s, AV_LOG_ERROR, "audio fifo too small to mux audio essence\n");
00456 return -1;
00457 }
00458
00459 av_fifo_generic_write(swf->audio_fifo, buf, size, NULL);
00460 swf->sound_samples += av_get_audio_frame_duration(enc, size);
00461
00462
00463 if (!swf->video_enc)
00464 swf_write_video(s, enc, 0, 0);
00465
00466 return 0;
00467 }
00468
00469 static int swf_write_packet(AVFormatContext *s, AVPacket *pkt)
00470 {
00471 AVCodecContext *codec = s->streams[pkt->stream_index]->codec;
00472 if (codec->codec_type == AVMEDIA_TYPE_AUDIO)
00473 return swf_write_audio(s, codec, pkt->data, pkt->size);
00474 else
00475 return swf_write_video(s, codec, pkt->data, pkt->size);
00476 }
00477
00478 static int swf_write_trailer(AVFormatContext *s)
00479 {
00480 SWFContext *swf = s->priv_data;
00481 AVIOContext *pb = s->pb;
00482 AVCodecContext *enc, *video_enc;
00483 int file_size, i;
00484
00485 video_enc = NULL;
00486 for(i=0;i<s->nb_streams;i++) {
00487 enc = s->streams[i]->codec;
00488 if (enc->codec_type == AVMEDIA_TYPE_VIDEO)
00489 video_enc = enc;
00490 else {
00491 av_fifo_free(swf->audio_fifo);
00492 swf->audio_fifo = NULL;
00493 }
00494 }
00495
00496 put_swf_tag(s, TAG_END);
00497 put_swf_end_tag(s);
00498
00499
00500 if (s->pb->seekable && video_enc) {
00501 file_size = avio_tell(pb);
00502 avio_seek(pb, 4, SEEK_SET);
00503 avio_wl32(pb, file_size);
00504 avio_seek(pb, swf->duration_pos, SEEK_SET);
00505 avio_wl16(pb, swf->video_frame_number);
00506 if (swf->vframes_pos) {
00507 avio_seek(pb, swf->vframes_pos, SEEK_SET);
00508 avio_wl16(pb, swf->video_frame_number);
00509 }
00510 avio_seek(pb, file_size, SEEK_SET);
00511 }
00512 return 0;
00513 }
00514
00515 #if CONFIG_SWF_MUXER
00516 AVOutputFormat ff_swf_muxer = {
00517 .name = "swf",
00518 .long_name = NULL_IF_CONFIG_SMALL("SWF (ShockWave Flash)"),
00519 .mime_type = "application/x-shockwave-flash",
00520 .extensions = "swf",
00521 .priv_data_size = sizeof(SWFContext),
00522 .audio_codec = AV_CODEC_ID_MP3,
00523 .video_codec = AV_CODEC_ID_FLV1,
00524 .write_header = swf_write_header,
00525 .write_packet = swf_write_packet,
00526 .write_trailer = swf_write_trailer,
00527 .flags = AVFMT_TS_NONSTRICT,
00528 };
00529 #endif
00530 #if CONFIG_AVM2_MUXER
00531 AVOutputFormat ff_avm2_muxer = {
00532 .name = "avm2",
00533 .long_name = NULL_IF_CONFIG_SMALL("SWF (ShockWave Flash) (AVM2)"),
00534 .mime_type = "application/x-shockwave-flash",
00535 .priv_data_size = sizeof(SWFContext),
00536 .audio_codec = AV_CODEC_ID_MP3,
00537 .video_codec = AV_CODEC_ID_FLV1,
00538 .write_header = swf_write_header,
00539 .write_packet = swf_write_packet,
00540 .write_trailer = swf_write_trailer,
00541 .flags = AVFMT_TS_NONSTRICT,
00542 };
00543 #endif