in a common flv.h. patch by Allan Hsu allan _at_ counterpop.net Originally committed as revision 7231 to svn://svn.ffmpeg.org/ffmpeg/trunktags/v0.5
| @@ -0,0 +1,108 @@ | |||
| /** | |||
| * @file flv.h | |||
| * FLV common header | |||
| * | |||
| * Copyright (c) 2006 The FFmpeg Project. | |||
| * | |||
| * This file is part of FFmpeg. | |||
| * | |||
| * FFmpeg is free software; you can redistribute it and/or | |||
| * modify it under the terms of the GNU Lesser General Public | |||
| * License as published by the Free Software Foundation; either | |||
| * version 2.1 of the License, or (at your option) any later version. | |||
| * | |||
| * FFmpeg is distributed in the hope that it will be useful, | |||
| * but WITHOUT ANY WARRANTY; without even the implied warranty of | |||
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |||
| * Lesser General Public License for more details. | |||
| * | |||
| * You should have received a copy of the GNU Lesser General Public | |||
| * License along with FFmpeg; if not, write to the Free Software | |||
| * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
| */ | |||
| #ifndef FLV_H | |||
| #define FLV_H | |||
| /* offsets for packed values */ | |||
| #define FLV_AUDIO_SAMPLESSIZE_OFFSET 1 | |||
| #define FLV_AUDIO_SAMPLERATE_OFFSET 2 | |||
| #define FLV_AUDIO_CODECID_OFFSET 4 | |||
| #define FLV_VIDEO_FRAMETYPE_OFFSET 4 | |||
| /* bitmasks to isolate specific values */ | |||
| #define FLV_AUDIO_CHANNEL_MASK 0x01 | |||
| #define FLV_AUDIO_SAMPLESIZE_MASK 0x02 | |||
| #define FLV_AUDIO_SAMPLERATE_MASK 0x0c | |||
| #define FLV_AUDIO_CODECID_MASK 0xf0 | |||
| #define FLV_VIDEO_CODECID_MASK 0x0f | |||
| #define FLV_VIDEO_FRAMETYPE_MASK 0xf0 | |||
| #define AMF_END_OF_OBJECT 0x09 | |||
| enum { | |||
| FLV_HEADER_FLAG_HASVIDEO = 1, | |||
| FLV_HEADER_FLAG_HASAUDIO = 4, | |||
| }; | |||
| enum { | |||
| FLV_TAG_TYPE_AUDIO = 0x08, | |||
| FLV_TAG_TYPE_VIDEO = 0x09, | |||
| FLV_TAG_TYPE_META = 0x12, | |||
| }; | |||
| enum { | |||
| FLV_MONO = 0, | |||
| FLV_STEREO = 1, | |||
| }; | |||
| enum { | |||
| FLV_SAMPLESSIZE_8BIT = 0, | |||
| FLV_SAMPLESSIZE_16BIT = 1 << FLV_AUDIO_SAMPLESSIZE_OFFSET, | |||
| }; | |||
| enum { | |||
| FLV_SAMPLERATE_SPECIAL = 0, /**< signifies 5512Hz and 8000Hz in the case of NELLYMOSER */ | |||
| FLV_SAMPLERATE_11025HZ = 1 << FLV_AUDIO_SAMPLERATE_OFFSET, | |||
| FLV_SAMPLERATE_22050HZ = 2 << FLV_AUDIO_SAMPLERATE_OFFSET, | |||
| FLV_SAMPLERATE_44100HZ = 3 << FLV_AUDIO_SAMPLERATE_OFFSET, | |||
| }; | |||
| enum { | |||
| FLV_CODECID_PCM_BE = 0, | |||
| FLV_CODECID_ADPCM = 1 << FLV_AUDIO_CODECID_OFFSET, | |||
| FLV_CODECID_MP3 = 2 << FLV_AUDIO_CODECID_OFFSET, | |||
| FLV_CODECID_PCM_LE = 3 << FLV_AUDIO_CODECID_OFFSET, | |||
| FLV_CODECID_NELLYMOSER_8HZ_MONO = 5 << FLV_AUDIO_CODECID_OFFSET, | |||
| FLV_CODECID_NELLYMOSER = 6 << FLV_AUDIO_CODECID_OFFSET, | |||
| }; | |||
| enum { | |||
| FLV_CODECID_H263 = 2, | |||
| FLV_CODECID_SCREEN = 3, | |||
| FLV_CODECID_VP6 = 4, | |||
| }; | |||
| enum { | |||
| FLV_FRAME_KEY = 1 << FLV_VIDEO_FRAMETYPE_OFFSET, | |||
| FLV_FRAME_INTER = 2 << FLV_VIDEO_FRAMETYPE_OFFSET, | |||
| FLV_FRAME_DISP_INTER = 3 << FLV_VIDEO_FRAMETYPE_OFFSET, | |||
| }; | |||
| typedef enum { | |||
| AMF_DATA_TYPE_NUMBER = 0x00, | |||
| AMF_DATA_TYPE_BOOL = 0x01, | |||
| AMF_DATA_TYPE_STRING = 0x02, | |||
| AMF_DATA_TYPE_OBJECT = 0x03, | |||
| AMF_DATA_TYPE_NULL = 0x05, | |||
| AMF_DATA_TYPE_UNDEFINED = 0x06, | |||
| AMF_DATA_TYPE_REFERENCE = 0x07, | |||
| AMF_DATA_TYPE_MIXEDARRAY = 0x08, | |||
| AMF_DATA_TYPE_ARRAY = 0x0a, | |||
| AMF_DATA_TYPE_DATE = 0x0b, | |||
| AMF_DATA_TYPE_UNSUPPORTED = 0x0d, | |||
| } AMFDataType; | |||
| #endif /* FLV_H */ | |||
| @@ -25,6 +25,7 @@ | |||
| * - lower 4bits: difference between encoded height and visible height | |||
| */ | |||
| #include "avformat.h" | |||
| #include "flv.h" | |||
| static int flv_probe(AVProbeData *p) | |||
| { | |||
| @@ -90,13 +91,13 @@ static int flv_read_packet(AVFormatContext *s, AVPacket *pkt) | |||
| next= size + url_ftell(&s->pb); | |||
| if (type == 8) { | |||
| if (type == FLV_TAG_TYPE_AUDIO) { | |||
| is_audio=1; | |||
| flags = get_byte(&s->pb); | |||
| } else if (type == 9) { | |||
| } else if (type == FLV_TAG_TYPE_VIDEO) { | |||
| is_audio=0; | |||
| flags = get_byte(&s->pb); | |||
| } else if (type == 18 && size > 13+1+4) { | |||
| } else if (type == FLV_TAG_TYPE_META && size > 13+1+4) { | |||
| url_fskip(&s->pb, 13); //onMetaData blah | |||
| if(get_byte(&s->pb) == 8){ | |||
| url_fskip(&s->pb, 4); | |||
| @@ -113,17 +114,17 @@ static int flv_read_packet(AVFormatContext *s, AVPacket *pkt) | |||
| tmp[len]=0; | |||
| type= get_byte(&s->pb); | |||
| if(type==0){ | |||
| if(type == AMF_DATA_TYPE_NUMBER){ | |||
| d= av_int2dbl(get_be64(&s->pb)); | |||
| }else if(type==2){ | |||
| }else if(type == AMF_DATA_TYPE_STRING){ | |||
| len= get_be16(&s->pb); | |||
| if(len >= sizeof(tmp)) | |||
| break; | |||
| url_fskip(&s->pb, len); | |||
| }else if(type==8){ | |||
| }else if(type == AMF_DATA_TYPE_MIXEDARRAY){ | |||
| //array | |||
| break; | |||
| }else if(type==11){ | |||
| }else if(type == AMF_DATA_TYPE_DATE){ | |||
| d= av_int2dbl(get_be64(&s->pb)); | |||
| get_be16(&s->pb); | |||
| } | |||
| @@ -158,14 +159,14 @@ static int flv_read_packet(AVFormatContext *s, AVPacket *pkt) | |||
| st->codec->time_base= (AVRational){1,1000}; | |||
| } | |||
| // av_log(NULL, AV_LOG_DEBUG, "%d %X %d \n", is_audio, flags, st->discard); | |||
| if( (st->discard >= AVDISCARD_NONKEY && !((flags >> 4)==1 || is_audio)) | |||
| ||(st->discard >= AVDISCARD_BIDIR && ((flags >> 4)==3 && !is_audio)) | |||
| if( (st->discard >= AVDISCARD_NONKEY && !((flags & FLV_VIDEO_FRAMETYPE_MASK) == FLV_FRAME_KEY || is_audio)) | |||
| ||(st->discard >= AVDISCARD_BIDIR && ((flags & FLV_VIDEO_FRAMETYPE_MASK) == FLV_FRAME_DISP_INTER && !is_audio)) | |||
| || st->discard >= AVDISCARD_ALL | |||
| ){ | |||
| url_fseek(&s->pb, next, SEEK_SET); | |||
| continue; | |||
| } | |||
| if ((flags >> 4)==1) | |||
| if ((flags & FLV_VIDEO_FRAMETYPE_MASK) == FLV_FRAME_KEY) | |||
| av_add_index_entry(st, pos, pts, size, 0, AVINDEX_KEYFRAME); | |||
| break; | |||
| } | |||
| @@ -173,31 +174,31 @@ static int flv_read_packet(AVFormatContext *s, AVPacket *pkt) | |||
| if(is_audio){ | |||
| if(st->codec->sample_rate == 0){ | |||
| st->codec->codec_type = CODEC_TYPE_AUDIO; | |||
| st->codec->channels = (flags&1)+1; | |||
| if((flags >> 4) == 5) | |||
| st->codec->channels = (flags & FLV_AUDIO_CHANNEL_MASK) == FLV_STEREO ? 2 : 1; | |||
| if((flags & FLV_AUDIO_CODECID_MASK) == FLV_CODECID_NELLYMOSER_8HZ_MONO) | |||
| st->codec->sample_rate= 8000; | |||
| else | |||
| st->codec->sample_rate = (44100<<((flags>>2)&3))>>3; | |||
| switch(flags >> 4){/* 0: uncompressed 1: ADPCM 2: mp3 5: Nellymoser 8kHz mono 6: Nellymoser*/ | |||
| case 0: if (flags&2) st->codec->codec_id = CODEC_ID_PCM_S16BE; | |||
| st->codec->sample_rate = (44100 << ((flags & FLV_AUDIO_SAMPLERATE_MASK) >> FLV_AUDIO_SAMPLERATE_OFFSET) >> 3); | |||
| switch(flags & FLV_AUDIO_CODECID_MASK) { | |||
| case FLV_CODECID_PCM_BE: if (flags & FLV_AUDIO_SAMPLESIZE_MASK) st->codec->codec_id = CODEC_ID_PCM_S16BE; | |||
| else st->codec->codec_id = CODEC_ID_PCM_S8; break; | |||
| case 1: st->codec->codec_id = CODEC_ID_ADPCM_SWF; break; | |||
| case 2: st->codec->codec_id = CODEC_ID_MP3; st->need_parsing = 1; break; | |||
| case FLV_CODECID_ADPCM : st->codec->codec_id = CODEC_ID_ADPCM_SWF; break; | |||
| case FLV_CODECID_MP3 : st->codec->codec_id = CODEC_ID_MP3 ; st->need_parsing = 1; break; | |||
| // this is not listed at FLV but at SWF, strange... | |||
| case 3: if (flags&2) st->codec->codec_id = CODEC_ID_PCM_S16LE; | |||
| case FLV_CODECID_PCM_LE: if (flags & FLV_AUDIO_SAMPLESIZE_MASK) st->codec->codec_id = CODEC_ID_PCM_S16LE; | |||
| else st->codec->codec_id = CODEC_ID_PCM_S8; break; | |||
| default: | |||
| av_log(s, AV_LOG_INFO, "Unsupported audio codec (%x)\n", flags >> 4); | |||
| st->codec->codec_tag= (flags >> 4); | |||
| } | |||
| st->codec->bits_per_sample = (flags & 2) ? 16 : 8; | |||
| st->codec->bits_per_sample = (flags & FLV_AUDIO_SAMPLESIZE_MASK) ? 16 : 8; | |||
| } | |||
| }else{ | |||
| st->codec->codec_type = CODEC_TYPE_VIDEO; | |||
| switch(flags & 0xF){ | |||
| case 2: st->codec->codec_id = CODEC_ID_FLV1; break; | |||
| case 3: st->codec->codec_id = CODEC_ID_FLASHSV; break; | |||
| case 4: | |||
| switch(flags & FLV_VIDEO_CODECID_MASK){ | |||
| case FLV_CODECID_H263 : st->codec->codec_id = CODEC_ID_FLV1 ; break; | |||
| case FLV_CODECID_SCREEN: st->codec->codec_id = CODEC_ID_FLASHSV; break; | |||
| case FLV_CODECID_VP6 : | |||
| st->codec->codec_id = CODEC_ID_VP6F; | |||
| if (st->codec->extradata_size != 1) { | |||
| st->codec->extradata_size = 1; | |||
| @@ -208,8 +209,8 @@ static int flv_read_packet(AVFormatContext *s, AVPacket *pkt) | |||
| size--; | |||
| break; | |||
| default: | |||
| av_log(s, AV_LOG_INFO, "Unsupported video codec (%x)\n", flags & 0xf); | |||
| st->codec->codec_tag= flags & 0xF; | |||
| av_log(s, AV_LOG_INFO, "Unsupported video codec (%x)\n", flags & FLV_VIDEO_CODECID_MASK); | |||
| st->codec->codec_tag = flags & FLV_VIDEO_CODECID_MASK; | |||
| } | |||
| } | |||
| @@ -223,7 +224,7 @@ static int flv_read_packet(AVFormatContext *s, AVPacket *pkt) | |||
| pkt->pts = pts; | |||
| pkt->stream_index = st->index; | |||
| if (is_audio || ((flags >> 4)==1)) | |||
| if (is_audio || ((flags & FLV_VIDEO_FRAMETYPE_MASK) == FLV_FRAME_KEY)) | |||
| pkt->flags |= PKT_FLAG_KEY; | |||
| return ret; | |||
| @@ -19,6 +19,7 @@ | |||
| * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
| */ | |||
| #include "avformat.h" | |||
| #include "flv.h" | |||
| #undef NDEBUG | |||
| #include <assert.h> | |||
| @@ -33,21 +34,21 @@ typedef struct FLVContext { | |||
| } FLVContext; | |||
| static int get_audio_flags(AVCodecContext *enc){ | |||
| int flags = (enc->bits_per_sample == 16) ? 0x2 : 0x0; | |||
| int flags = (enc->bits_per_sample == 16) ? FLV_SAMPLESSIZE_16BIT : FLV_SAMPLESSIZE_8BIT; | |||
| switch (enc->sample_rate) { | |||
| case 44100: | |||
| flags |= 0x0C; | |||
| flags |= FLV_SAMPLERATE_44100HZ; | |||
| break; | |||
| case 22050: | |||
| flags |= 0x08; | |||
| flags |= FLV_SAMPLERATE_22050HZ; | |||
| break; | |||
| case 11025: | |||
| flags |= 0x04; | |||
| flags |= FLV_SAMPLERATE_11025HZ; | |||
| break; | |||
| case 8000: //nellymoser only | |||
| case 5512: //not mp3 | |||
| flags |= 0x00; | |||
| flags |= FLV_SAMPLERATE_SPECIAL; | |||
| break; | |||
| default: | |||
| av_log(enc, AV_LOG_ERROR, "flv doesnt support that sample rate, choose from (44100, 22050, 11025)\n"); | |||
| @@ -55,23 +56,24 @@ static int get_audio_flags(AVCodecContext *enc){ | |||
| } | |||
| if (enc->channels > 1) { | |||
| flags |= 0x01; | |||
| flags |= FLV_STEREO; | |||
| } | |||
| switch(enc->codec_id){ | |||
| case CODEC_ID_MP3: | |||
| flags |= 0x20 | 0x2; | |||
| flags |= FLV_CODECID_MP3 | FLV_SAMPLESSIZE_16BIT; | |||
| break; | |||
| case CODEC_ID_PCM_S8: | |||
| flags |= FLV_CODECID_PCM_BE | FLV_SAMPLESSIZE_8BIT; | |||
| break; | |||
| case CODEC_ID_PCM_S16BE: | |||
| flags |= 0x2; | |||
| flags |= FLV_CODECID_PCM_BE | FLV_SAMPLESSIZE_16BIT; | |||
| break; | |||
| case CODEC_ID_PCM_S16LE: | |||
| flags |= 0x30 | 0x2; | |||
| flags |= FLV_CODECID_PCM_LE | FLV_SAMPLESSIZE_16BIT; | |||
| break; | |||
| case CODEC_ID_ADPCM_SWF: | |||
| flags |= 0x10; | |||
| flags |= FLV_CODECID_ADPCM; | |||
| break; | |||
| case 0: | |||
| flags |= enc->codec_tag<<4; | |||
| @@ -84,14 +86,6 @@ static int get_audio_flags(AVCodecContext *enc){ | |||
| return flags; | |||
| } | |||
| #define AMF_DOUBLE 0 | |||
| #define AMF_BOOLEAN 1 | |||
| #define AMF_STRING 2 | |||
| #define AMF_OBJECT 3 | |||
| #define AMF_MIXED_ARRAY 8 | |||
| #define AMF_ARRAY 10 | |||
| #define AMF_DATE 11 | |||
| static void put_amf_string(ByteIOContext *pb, const char *str) | |||
| { | |||
| size_t len = strlen(str); | |||
| @@ -101,7 +95,7 @@ static void put_amf_string(ByteIOContext *pb, const char *str) | |||
| static void put_amf_double(ByteIOContext *pb, double d) | |||
| { | |||
| put_byte(pb, AMF_DOUBLE); | |||
| put_byte(pb, AMF_DATA_TYPE_NUMBER); | |||
| put_be64(pb, av_dbl2int(d)); | |||
| } | |||
| @@ -160,11 +154,11 @@ static int flv_write_header(AVFormatContext *s) | |||
| /* now data of data_size size */ | |||
| /* first event name as a string */ | |||
| put_byte(pb, AMF_STRING); // 1 byte | |||
| put_byte(pb, AMF_DATA_TYPE_STRING); | |||
| put_amf_string(pb, "onMetaData"); // 12 bytes | |||
| /* mixed array (hash) with size and string/type/data tuples */ | |||
| put_byte(pb, AMF_MIXED_ARRAY); | |||
| put_byte(pb, AMF_DATA_TYPE_MIXEDARRAY); | |||
| put_be32(pb, 4*flv->hasVideo + flv->hasAudio + 2); // +2 for duration and file size | |||
| put_amf_string(pb, "duration"); | |||
| @@ -195,7 +189,7 @@ static int flv_write_header(AVFormatContext *s) | |||
| put_amf_double(pb, 0); // delayed write | |||
| put_amf_string(pb, ""); | |||
| put_byte(pb, 9); // end marker 1 byte | |||
| put_byte(pb, AMF_END_OF_OBJECT); | |||
| /* write total size of tag */ | |||
| data_size= url_ftell(pb) - metadata_size_pos - 10; | |||
| @@ -216,8 +210,8 @@ static int flv_write_trailer(AVFormatContext *s) | |||
| FLVContext *flv = s->priv_data; | |||
| file_size = url_ftell(pb); | |||
| flags |= flv->hasAudio ? 4 : 0; | |||
| flags |= flv->hasVideo ? 1 : 0; | |||
| flags |= flv->hasAudio ? FLV_HEADER_FLAG_HASAUDIO : 0; | |||
| flags |= flv->hasVideo ? FLV_HEADER_FLAG_HASVIDEO : 0; | |||
| url_fseek(pb, 4, SEEK_SET); | |||
| put_byte(pb,flags); | |||
| @@ -242,16 +236,16 @@ static int flv_write_packet(AVFormatContext *s, AVPacket *pkt) | |||
| // av_log(s, AV_LOG_DEBUG, "type:%d pts: %"PRId64" size:%d\n", enc->codec_type, timestamp, size); | |||
| if (enc->codec_type == CODEC_TYPE_VIDEO) { | |||
| put_byte(pb, 9); | |||
| flags = 2; // choose h263 | |||
| flags |= pkt->flags & PKT_FLAG_KEY ? 0x10 : 0x20; // add keyframe indicator | |||
| put_byte(pb, FLV_TAG_TYPE_VIDEO); | |||
| flags = FLV_CODECID_H263; | |||
| flags |= pkt->flags & PKT_FLAG_KEY ? FLV_FRAME_KEY : FLV_FRAME_INTER; | |||
| } else { | |||
| assert(enc->codec_type == CODEC_TYPE_AUDIO); | |||
| flags = get_audio_flags(enc); | |||
| assert(size); | |||
| put_byte(pb, 8); | |||
| put_byte(pb, FLV_TAG_TYPE_AUDIO); | |||
| } | |||
| put_be24(pb,size+1); // include flags | |||