git.sesse.net Git - ffmpeg/blob - libavformat/flvenc.c

   1 /*
   2  * FLV encoder.
   3  * Copyright (c) 2003 The FFmpeg Project.
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  18  */
  19 #include "avformat.h"
  20
  21 #undef NDEBUG
  22 #include <assert.h>
  23
  24 typedef struct FLVContext {
  25     int hasAudio;
  26     int hasVideo;
  27     int reserved;
  28 } FLVContext;
  29
  30 static int get_audio_flags(AVCodecContext *enc){
  31     int flags = (enc->bits_per_sample == 16) ? 0x2 : 0x0;
  32
  33     switch (enc->sample_rate) {
  34         case    44100:
  35             flags |= 0x0C;
  36             break;
  37         case    22050:
  38             flags |= 0x08;
  39             break;
  40         case    11025:
  41             flags |= 0x04;
  42             break;
  43         case     8000: //nellymoser only
  44         case     5512: //not mp3
  45             flags |= 0x00;
  46             break;
  47         default:
  48             av_log(enc, AV_LOG_ERROR, "flv doesnt support that sample rate, choose from (44100, 22050, 11025)\n");
  49             return -1;
  50     }
  51
  52     if (enc->channels > 1) {
  53         flags |= 0x01;
  54     }
  55
  56     switch(enc->codec_id){
  57     case CODEC_ID_MP3:
  58         flags |= 0x20 | 0x2;
  59         break;
  60     case CODEC_ID_PCM_S8:
  61         break;
  62     case CODEC_ID_PCM_S16BE:
  63         flags |= 0x60 | 0x2;
  64         break;
  65     case CODEC_ID_PCM_S16LE:
  66         flags |= 0x2;
  67         break;
  68     case CODEC_ID_ADPCM_SWF:
  69         flags |= 0x10;
  70         break;
  71     case 0:
  72         flags |= enc->codec_tag<<4;
  73         break;
  74     default:
  75         av_log(enc, AV_LOG_ERROR, "codec not compatible with flv\n");
  76         return -1;
  77     }
  78
  79     return flags;
  80 }
  81
  82 #define AMF_DOUBLE  0
  83 #define AMF_BOOLEAN 1
  84 #define AMF_STRING  2
  85 #define AMF_OBJECT  3
  86 #define AMF_MIXED_ARRAY 8
  87 #define AMF_ARRAY  10
  88 #define AMF_DATE   11
  89
  90 static void put_amf_string(ByteIOContext *pb, const char *str)
  91 {
  92     size_t len = strlen(str);
  93     put_be16(pb, len);
  94     put_buffer(pb, str, len);
  95 }
  96
  97 static void put_amf_double(ByteIOContext *pb, double d)
  98 {
  99     put_byte(pb, AMF_DOUBLE);
 100     put_be64(pb, av_dbl2int(d));
 101 }
 102
 103 static int flv_write_header(AVFormatContext *s)
 104 {
 105     ByteIOContext *pb = &s->pb;
 106     FLVContext *flv = s->priv_data;
 107     int i, width, height, samplerate;
 108     double framerate = 0.0;
 109     int metadata_size_pos, data_size;
 110
 111     flv->hasAudio = 0;
 112     flv->hasVideo = 0;
 113
 114     put_tag(pb,"FLV");
 115     put_byte(pb,1);
 116     put_byte(pb,0); // delayed write
 117     put_be32(pb,9);
 118     put_be32(pb,0);
 119
 120     for(i=0; i<s->nb_streams; i++){
 121         AVCodecContext *enc = s->streams[i]->codec;
 122         if (enc->codec_type == CODEC_TYPE_VIDEO) {
 123             width = enc->width;
 124             height = enc->height;
 125             if (s->streams[i]->r_frame_rate.den && s->streams[i]->r_frame_rate.num) {
 126                 framerate = av_q2d(s->streams[i]->r_frame_rate);
 127             } else {
 128                 framerate = 1/av_q2d(s->streams[i]->codec->time_base);
 129             }
 130             flv->hasVideo=1;
 131         } else {
 132             flv->hasAudio=1;
 133             samplerate = enc->sample_rate;
 134         }
 135         av_set_pts_info(s->streams[i], 24, 1, 1000); /* 24 bit pts in ms */
 136         if(enc->codec_tag == 5){
 137             put_byte(pb,8); // message type
 138             put_be24(pb,0); // include flags
 139             put_be24(pb,0); // time stamp
 140             put_be32(pb,0); // reserved
 141             put_be32(pb,11); // size
 142             flv->reserved=5;
 143         }
 144         if(enc->codec_type == CODEC_TYPE_AUDIO && get_audio_flags(enc)<0)
 145             return -1;
 146     }
 147
 148     /* write meta_tag */
 149     put_byte(pb, 18);         // tag type META
 150     metadata_size_pos= url_ftell(pb);
 151     put_be24(pb, 0);          // size of data part (sum of all parts below)
 152     put_be24(pb, 0);          // time stamp
 153     put_be32(pb, 0);          // reserved
 154
 155     /* now data of data_size size */
 156
 157     /* first event name as a string */
 158     put_byte(pb, AMF_STRING); // 1 byte
 159     put_amf_string(pb, "onMetaData"); // 12 bytes
 160
 161     /* mixed array (hash) with size and string/type/data tuples */
 162     put_byte(pb, AMF_MIXED_ARRAY);
 163     put_be32(pb, 4*flv->hasVideo + flv->hasAudio + (!!s->file_size) + (s->duration != AV_NOPTS_VALUE && s->duration));
 164
 165     if (s->duration != AV_NOPTS_VALUE && s->duration) {
 166         put_amf_string(pb, "duration");
 167         put_amf_double(pb, s->duration / (double)AV_TIME_BASE);
 168     }
 169
 170     if(flv->hasVideo){
 171         put_amf_string(pb, "width");
 172         put_amf_double(pb, width);
 173
 174         put_amf_string(pb, "height");
 175         put_amf_double(pb, height);
 176
 177         put_amf_string(pb, "videodatarate");
 178         put_amf_double(pb, s->bit_rate / 1024.0);
 179
 180         put_amf_string(pb, "framerate");
 181         put_amf_double(pb, framerate);
 182     }
 183
 184     if(flv->hasAudio){
 185         put_amf_string(pb, "audiosamplerate");
 186         put_amf_double(pb, samplerate);
 187     }
 188
 189     if(s->file_size){
 190         put_amf_string(pb, "filesize");
 191         put_amf_double(pb, s->file_size);
 192     }
 193
 194     put_amf_string(pb, "");
 195     put_byte(pb, 9); // end marker 1 byte
 196
 197     /* write total size of tag */
 198     data_size= url_ftell(pb) - metadata_size_pos - 10;
 199     url_fseek(pb, metadata_size_pos, SEEK_SET);
 200     put_be24(pb, data_size);
 201     url_fseek(pb, data_size + 10 - 3, SEEK_CUR);
 202     put_be32(pb, data_size + 11);
 203
 204     return 0;
 205 }
 206
 207 static int flv_write_trailer(AVFormatContext *s)
 208 {
 209     int64_t file_size;
 210     int flags = 0;
 211
 212     ByteIOContext *pb = &s->pb;
 213     FLVContext *flv = s->priv_data;
 214
 215     file_size = url_ftell(pb);
 216     flags |= flv->hasAudio ? 4 : 0;
 217     flags |= flv->hasVideo ? 1 : 0;
 218     url_fseek(pb, 4, SEEK_SET);
 219     put_byte(pb,flags);
 220     url_fseek(pb, file_size, SEEK_SET);
 221     return 0;
 222 }
 223
 224 static int flv_write_packet(AVFormatContext *s, AVPacket *pkt)
 225 {
 226     ByteIOContext *pb = &s->pb;
 227     AVCodecContext *enc = s->streams[pkt->stream_index]->codec;
 228     FLVContext *flv = s->priv_data;
 229     int size= pkt->size;
 230     int flags;
 231
 232 //    av_log(s, AV_LOG_DEBUG, "type:%d pts: %lld size:%d\n", enc->codec_type, timestamp, size);
 233
 234     if (enc->codec_type == CODEC_TYPE_VIDEO) {
 235         put_byte(pb, 9);
 236         flags = 2; // choose h263
 237         flags |= pkt->flags & PKT_FLAG_KEY ? 0x10 : 0x20; // add keyframe indicator
 238     } else {
 239         assert(enc->codec_type == CODEC_TYPE_AUDIO);
 240         flags = get_audio_flags(enc);
 241
 242         assert(size);
 243
 244         put_byte(pb, 8);
 245     }
 246
 247     put_be24(pb,size+1); // include flags
 248     put_be24(pb,pkt->pts);
 249     put_be32(pb,flv->reserved);
 250     put_byte(pb,flags);
 251     put_buffer(pb, pkt->data, size);
 252     put_be32(pb,size+1+11); // previous tag size
 253
 254     put_flush_packet(pb);
 255     return 0;
 256 }
 257
 258 AVOutputFormat flv_muxer = {
 259     "flv",
 260     "flv format",
 261     "video/x-flv",
 262     "flv",
 263     sizeof(FLVContext),
 264 #ifdef CONFIG_MP3LAME
 265     CODEC_ID_MP3,
 266 #else // CONFIG_MP3LAME
 267     CODEC_ID_NONE,
 268 #endif // CONFIG_MP3LAME
 269     CODEC_ID_FLV1,
 270     flv_write_header,
 271     flv_write_packet,
 272     flv_write_trailer,
 273 };