00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022 #include "libavutil/intreadwrite.h"
00023 #include "libavutil/intfloat.h"
00024 #include "libavutil/avassert.h"
00025 #include "libavutil/parseutils.h"
00026 #include "avformat.h"
00027 #include "internal.h"
00028 #include "ffm.h"
00029
00030 static void flush_packet(AVFormatContext *s)
00031 {
00032 FFMContext *ffm = s->priv_data;
00033 int fill_size, h;
00034 AVIOContext *pb = s->pb;
00035
00036 fill_size = ffm->packet_end - ffm->packet_ptr;
00037 memset(ffm->packet_ptr, 0, fill_size);
00038
00039 av_assert1(avio_tell(pb) % ffm->packet_size == 0);
00040
00041
00042 avio_wb16(pb, PACKET_ID);
00043 avio_wb16(pb, fill_size);
00044 avio_wb64(pb, ffm->dts);
00045 h = ffm->frame_offset;
00046 if (ffm->first_packet)
00047 h |= 0x8000;
00048 avio_wb16(pb, h);
00049 avio_write(pb, ffm->packet, ffm->packet_end - ffm->packet);
00050 avio_flush(pb);
00051
00052
00053 ffm->frame_offset = 0;
00054 ffm->packet_ptr = ffm->packet;
00055 ffm->first_packet = 0;
00056 }
00057
00058
00059 static void ffm_write_data(AVFormatContext *s,
00060 const uint8_t *buf, int size,
00061 int64_t dts, int header)
00062 {
00063 FFMContext *ffm = s->priv_data;
00064 int len;
00065
00066 if (header && ffm->frame_offset == 0) {
00067 ffm->frame_offset = ffm->packet_ptr - ffm->packet + FFM_HEADER_SIZE;
00068 ffm->dts = dts;
00069 }
00070
00071
00072 while (size > 0) {
00073 len = ffm->packet_end - ffm->packet_ptr;
00074 if (len > size)
00075 len = size;
00076 memcpy(ffm->packet_ptr, buf, len);
00077
00078 ffm->packet_ptr += len;
00079 buf += len;
00080 size -= len;
00081 if (ffm->packet_ptr >= ffm->packet_end)
00082 flush_packet(s);
00083 }
00084 }
00085
00086 static void write_header_chunk(AVIOContext *pb, AVIOContext *dpb, unsigned id)
00087 {
00088 uint8_t *dyn_buf;
00089 int dyn_size= avio_close_dyn_buf(dpb, &dyn_buf);
00090 avio_wb32(pb, id);
00091 avio_wb32(pb, dyn_size);
00092 avio_write(pb, dyn_buf, dyn_size);
00093 av_free(dyn_buf);
00094 }
00095
00096 static int ffm_write_header(AVFormatContext *s)
00097 {
00098 FFMContext *ffm = s->priv_data;
00099 AVDictionaryEntry *t;
00100 AVStream *st;
00101 AVIOContext *pb = s->pb;
00102 AVCodecContext *codec;
00103 int bit_rate, i;
00104
00105 if (t = av_dict_get(s->metadata, "creation_time", NULL, 0)) {
00106 int ret = av_parse_time(&ffm->start_time, t->value, 0);
00107 if (ret < 0)
00108 return ret;
00109 }
00110
00111 ffm->packet_size = FFM_PACKET_SIZE;
00112
00113
00114 avio_wl32(pb, MKTAG('F', 'F', 'M', '2'));
00115 avio_wb32(pb, ffm->packet_size);
00116 avio_wb64(pb, 0);
00117
00118 if(avio_open_dyn_buf(&pb) < 0)
00119 return AVERROR(ENOMEM);
00120
00121 avio_wb32(pb, s->nb_streams);
00122 bit_rate = 0;
00123 for(i=0;i<s->nb_streams;i++) {
00124 st = s->streams[i];
00125 bit_rate += st->codec->bit_rate;
00126 }
00127 avio_wb32(pb, bit_rate);
00128
00129 write_header_chunk(s->pb, pb, MKBETAG('M', 'A', 'I', 'N'));
00130
00131
00132 for(i=0;i<s->nb_streams;i++) {
00133 st = s->streams[i];
00134 avpriv_set_pts_info(st, 64, 1, 1000000);
00135 if(avio_open_dyn_buf(&pb) < 0)
00136 return AVERROR(ENOMEM);
00137
00138 codec = st->codec;
00139
00140 avio_wb32(pb, codec->codec_id);
00141 avio_w8(pb, codec->codec_type);
00142 avio_wb32(pb, codec->bit_rate);
00143 avio_wb32(pb, codec->flags);
00144 avio_wb32(pb, codec->flags2);
00145 avio_wb32(pb, codec->debug);
00146 if (codec->flags & CODEC_FLAG_GLOBAL_HEADER) {
00147 avio_wb32(pb, codec->extradata_size);
00148 avio_write(pb, codec->extradata, codec->extradata_size);
00149 }
00150 write_header_chunk(s->pb, pb, MKBETAG('C', 'O', 'M', 'M'));
00151 if(avio_open_dyn_buf(&pb) < 0)
00152 return AVERROR(ENOMEM);
00153
00154 switch(codec->codec_type) {
00155 case AVMEDIA_TYPE_VIDEO:
00156 avio_wb32(pb, codec->time_base.num);
00157 avio_wb32(pb, codec->time_base.den);
00158 avio_wb16(pb, codec->width);
00159 avio_wb16(pb, codec->height);
00160 avio_wb16(pb, codec->gop_size);
00161 avio_wb32(pb, codec->pix_fmt);
00162 avio_w8(pb, codec->qmin);
00163 avio_w8(pb, codec->qmax);
00164 avio_w8(pb, codec->max_qdiff);
00165 avio_wb16(pb, (int) (codec->qcompress * 10000.0));
00166 avio_wb16(pb, (int) (codec->qblur * 10000.0));
00167 avio_wb32(pb, codec->bit_rate_tolerance);
00168 avio_put_str(pb, codec->rc_eq ? codec->rc_eq : "tex^qComp");
00169 avio_wb32(pb, codec->rc_max_rate);
00170 avio_wb32(pb, codec->rc_min_rate);
00171 avio_wb32(pb, codec->rc_buffer_size);
00172 avio_wb64(pb, av_double2int(codec->i_quant_factor));
00173 avio_wb64(pb, av_double2int(codec->b_quant_factor));
00174 avio_wb64(pb, av_double2int(codec->i_quant_offset));
00175 avio_wb64(pb, av_double2int(codec->b_quant_offset));
00176 avio_wb32(pb, codec->dct_algo);
00177 avio_wb32(pb, codec->strict_std_compliance);
00178 avio_wb32(pb, codec->max_b_frames);
00179 avio_wb32(pb, codec->mpeg_quant);
00180 avio_wb32(pb, codec->intra_dc_precision);
00181 avio_wb32(pb, codec->me_method);
00182 avio_wb32(pb, codec->mb_decision);
00183 avio_wb32(pb, codec->nsse_weight);
00184 avio_wb32(pb, codec->frame_skip_cmp);
00185 avio_wb64(pb, av_double2int(codec->rc_buffer_aggressivity));
00186 avio_wb32(pb, codec->codec_tag);
00187 avio_w8(pb, codec->thread_count);
00188 avio_wb32(pb, codec->coder_type);
00189 avio_wb32(pb, codec->me_cmp);
00190 avio_wb32(pb, codec->me_subpel_quality);
00191 avio_wb32(pb, codec->me_range);
00192 avio_wb32(pb, codec->keyint_min);
00193 avio_wb32(pb, codec->scenechange_threshold);
00194 avio_wb32(pb, codec->b_frame_strategy);
00195 avio_wb64(pb, av_double2int(codec->qcompress));
00196 avio_wb64(pb, av_double2int(codec->qblur));
00197 avio_wb32(pb, codec->max_qdiff);
00198 avio_wb32(pb, codec->refs);
00199 write_header_chunk(s->pb, pb, MKBETAG('S', 'T', 'V', 'I'));
00200 break;
00201 case AVMEDIA_TYPE_AUDIO:
00202 avio_wb32(pb, codec->sample_rate);
00203 avio_wl16(pb, codec->channels);
00204 avio_wl16(pb, codec->frame_size);
00205 write_header_chunk(s->pb, pb, MKBETAG('S', 'T', 'A', 'U'));
00206 break;
00207 default:
00208 return -1;
00209 }
00210 }
00211 pb = s->pb;
00212
00213 avio_wb64(pb, 0);
00214
00215
00216 while ((avio_tell(pb) % ffm->packet_size) != 0)
00217 avio_w8(pb, 0);
00218
00219 avio_flush(pb);
00220
00221
00222 ffm->packet_ptr = ffm->packet;
00223 ffm->packet_end = ffm->packet + ffm->packet_size - FFM_HEADER_SIZE;
00224 av_assert0(ffm->packet_end >= ffm->packet);
00225 ffm->frame_offset = 0;
00226 ffm->dts = 0;
00227 ffm->first_packet = 1;
00228
00229 return 0;
00230 }
00231
00232 static int ffm_write_packet(AVFormatContext *s, AVPacket *pkt)
00233 {
00234 FFMContext *ffm = s->priv_data;
00235 int64_t dts;
00236 uint8_t header[FRAME_HEADER_SIZE+4];
00237 int header_size = FRAME_HEADER_SIZE;
00238
00239 dts = ffm->start_time + pkt->dts;
00240
00241 header[0] = pkt->stream_index;
00242 header[1] = 0;
00243 if (pkt->flags & AV_PKT_FLAG_KEY)
00244 header[1] |= FLAG_KEY_FRAME;
00245 AV_WB24(header+2, pkt->size);
00246 AV_WB24(header+5, pkt->duration);
00247 AV_WB64(header+8, ffm->start_time + pkt->pts);
00248 if (pkt->pts != pkt->dts) {
00249 header[1] |= FLAG_DTS;
00250 AV_WB32(header+16, pkt->pts - pkt->dts);
00251 header_size += 4;
00252 }
00253 ffm_write_data(s, header, header_size, dts, 1);
00254 ffm_write_data(s, pkt->data, pkt->size, dts, 0);
00255
00256 return 0;
00257 }
00258
00259 static int ffm_write_trailer(AVFormatContext *s)
00260 {
00261 FFMContext *ffm = s->priv_data;
00262
00263
00264 if (ffm->packet_ptr > ffm->packet)
00265 flush_packet(s);
00266
00267 return 0;
00268 }
00269
00270 AVOutputFormat ff_ffm_muxer = {
00271 .name = "ffm",
00272 .long_name = NULL_IF_CONFIG_SMALL("FFM (FFserver live feed)"),
00273 .extensions = "ffm",
00274 .priv_data_size = sizeof(FFMContext),
00275 .audio_codec = AV_CODEC_ID_MP2,
00276 .video_codec = AV_CODEC_ID_MPEG1VIDEO,
00277 .write_header = ffm_write_header,
00278 .write_packet = ffm_write_packet,
00279 .write_trailer = ffm_write_trailer,
00280 };