yading@11: /* yading@11: * "Real" compatible muxer. yading@11: * Copyright (c) 2000, 2001 Fabrice Bellard yading@11: * yading@11: * This file is part of FFmpeg. yading@11: * yading@11: * FFmpeg is free software; you can redistribute it and/or yading@11: * modify it under the terms of the GNU Lesser General Public yading@11: * License as published by the Free Software Foundation; either yading@11: * version 2.1 of the License, or (at your option) any later version. yading@11: * yading@11: * FFmpeg is distributed in the hope that it will be useful, yading@11: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@11: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@11: * Lesser General Public License for more details. yading@11: * yading@11: * You should have received a copy of the GNU Lesser General Public yading@11: * License along with FFmpeg; if not, write to the Free Software yading@11: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@11: */ yading@11: #include "avformat.h" yading@11: #include "avio_internal.h" yading@11: #include "rm.h" yading@11: #include "libavutil/dict.h" yading@11: yading@11: typedef struct { yading@11: int nb_packets; yading@11: int packet_total_size; yading@11: int packet_max_size; yading@11: /* codec related output */ yading@11: int bit_rate; yading@11: float frame_rate; yading@11: int nb_frames; /* current frame number */ yading@11: int total_frames; /* total number of frames */ yading@11: int num; yading@11: AVCodecContext *enc; yading@11: } StreamInfo; yading@11: yading@11: typedef struct { yading@11: StreamInfo streams[2]; yading@11: StreamInfo *audio_stream, *video_stream; yading@11: int data_pos; /* position of the data after the header */ yading@11: } RMMuxContext; yading@11: yading@11: /* in ms */ yading@11: #define BUFFER_DURATION 0 yading@11: yading@11: yading@11: static void put_str(AVIOContext *s, const char *tag) yading@11: { yading@11: avio_wb16(s,strlen(tag)); yading@11: while (*tag) { yading@11: avio_w8(s, *tag++); yading@11: } yading@11: } yading@11: yading@11: static void put_str8(AVIOContext *s, const char *tag) yading@11: { yading@11: avio_w8(s, strlen(tag)); yading@11: while (*tag) { yading@11: avio_w8(s, *tag++); yading@11: } yading@11: } yading@11: yading@11: static int rv10_write_header(AVFormatContext *ctx, yading@11: int data_size, int index_pos) yading@11: { yading@11: RMMuxContext *rm = ctx->priv_data; yading@11: AVIOContext *s = ctx->pb; yading@11: StreamInfo *stream; yading@11: unsigned char *data_offset_ptr, *start_ptr; yading@11: const char *desc, *mimetype; yading@11: int nb_packets, packet_total_size, packet_max_size, size, packet_avg_size, i; yading@11: int bit_rate, v, duration, flags, data_pos; yading@11: AVDictionaryEntry *tag; yading@11: yading@11: start_ptr = s->buf_ptr; yading@11: yading@11: ffio_wfourcc(s, ".RMF"); yading@11: avio_wb32(s,18); /* header size */ yading@11: avio_wb16(s,0); yading@11: avio_wb32(s,0); yading@11: avio_wb32(s,4 + ctx->nb_streams); /* num headers */ yading@11: yading@11: ffio_wfourcc(s,"PROP"); yading@11: avio_wb32(s, 50); yading@11: avio_wb16(s, 0); yading@11: packet_max_size = 0; yading@11: packet_total_size = 0; yading@11: nb_packets = 0; yading@11: bit_rate = 0; yading@11: duration = 0; yading@11: for(i=0;inb_streams;i++) { yading@11: StreamInfo *stream = &rm->streams[i]; yading@11: bit_rate += stream->bit_rate; yading@11: if (stream->packet_max_size > packet_max_size) yading@11: packet_max_size = stream->packet_max_size; yading@11: nb_packets += stream->nb_packets; yading@11: packet_total_size += stream->packet_total_size; yading@11: /* select maximum duration */ yading@11: v = (int) (1000.0 * (float)stream->total_frames / stream->frame_rate); yading@11: if (v > duration) yading@11: duration = v; yading@11: } yading@11: avio_wb32(s, bit_rate); /* max bit rate */ yading@11: avio_wb32(s, bit_rate); /* avg bit rate */ yading@11: avio_wb32(s, packet_max_size); /* max packet size */ yading@11: if (nb_packets > 0) yading@11: packet_avg_size = packet_total_size / nb_packets; yading@11: else yading@11: packet_avg_size = 0; yading@11: avio_wb32(s, packet_avg_size); /* avg packet size */ yading@11: avio_wb32(s, nb_packets); /* num packets */ yading@11: avio_wb32(s, duration); /* duration */ yading@11: avio_wb32(s, BUFFER_DURATION); /* preroll */ yading@11: avio_wb32(s, index_pos); /* index offset */ yading@11: /* computation of data the data offset */ yading@11: data_offset_ptr = s->buf_ptr; yading@11: avio_wb32(s, 0); /* data offset : will be patched after */ yading@11: avio_wb16(s, ctx->nb_streams); /* num streams */ yading@11: flags = 1 | 2; /* save allowed & perfect play */ yading@11: if (!s->seekable) yading@11: flags |= 4; /* live broadcast */ yading@11: avio_wb16(s, flags); yading@11: yading@11: /* comments */ yading@11: yading@11: ffio_wfourcc(s,"CONT"); yading@11: size = 4 * 2 + 10; yading@11: for(i=0; imetadata, ff_rm_metadata[i], NULL, 0); yading@11: if(tag) size += strlen(tag->value); yading@11: } yading@11: avio_wb32(s,size); yading@11: avio_wb16(s,0); yading@11: for(i=0; imetadata, ff_rm_metadata[i], NULL, 0); yading@11: put_str(s, tag ? tag->value : ""); yading@11: } yading@11: yading@11: for(i=0;inb_streams;i++) { yading@11: int codec_data_size; yading@11: yading@11: stream = &rm->streams[i]; yading@11: yading@11: if (stream->enc->codec_type == AVMEDIA_TYPE_AUDIO) { yading@11: desc = "The Audio Stream"; yading@11: mimetype = "audio/x-pn-realaudio"; yading@11: codec_data_size = 73; yading@11: } else { yading@11: desc = "The Video Stream"; yading@11: mimetype = "video/x-pn-realvideo"; yading@11: codec_data_size = 34; yading@11: } yading@11: yading@11: ffio_wfourcc(s,"MDPR"); yading@11: size = 10 + 9 * 4 + strlen(desc) + strlen(mimetype) + codec_data_size; yading@11: avio_wb32(s, size); yading@11: avio_wb16(s, 0); yading@11: yading@11: avio_wb16(s, i); /* stream number */ yading@11: avio_wb32(s, stream->bit_rate); /* max bit rate */ yading@11: avio_wb32(s, stream->bit_rate); /* avg bit rate */ yading@11: avio_wb32(s, stream->packet_max_size); /* max packet size */ yading@11: if (stream->nb_packets > 0) yading@11: packet_avg_size = stream->packet_total_size / yading@11: stream->nb_packets; yading@11: else yading@11: packet_avg_size = 0; yading@11: avio_wb32(s, packet_avg_size); /* avg packet size */ yading@11: avio_wb32(s, 0); /* start time */ yading@11: avio_wb32(s, BUFFER_DURATION); /* preroll */ yading@11: /* duration */ yading@11: if (!s->seekable || !stream->total_frames) yading@11: avio_wb32(s, (int)(3600 * 1000)); yading@11: else yading@11: avio_wb32(s, (int)(stream->total_frames * 1000 / stream->frame_rate)); yading@11: put_str8(s, desc); yading@11: put_str8(s, mimetype); yading@11: avio_wb32(s, codec_data_size); yading@11: yading@11: if (stream->enc->codec_type == AVMEDIA_TYPE_AUDIO) { yading@11: int coded_frame_size, fscode, sample_rate; yading@11: sample_rate = stream->enc->sample_rate; yading@11: coded_frame_size = (stream->enc->bit_rate * yading@11: stream->enc->frame_size) / (8 * sample_rate); yading@11: /* audio codec info */ yading@11: avio_write(s, ".ra", 3); yading@11: avio_w8(s, 0xfd); yading@11: avio_wb32(s, 0x00040000); /* version */ yading@11: ffio_wfourcc(s, ".ra4"); yading@11: avio_wb32(s, 0x01b53530); /* stream length */ yading@11: avio_wb16(s, 4); /* unknown */ yading@11: avio_wb32(s, 0x39); /* header size */ yading@11: yading@11: switch(sample_rate) { yading@11: case 48000: yading@11: case 24000: yading@11: case 12000: yading@11: fscode = 1; yading@11: break; yading@11: default: yading@11: case 44100: yading@11: case 22050: yading@11: case 11025: yading@11: fscode = 2; yading@11: break; yading@11: case 32000: yading@11: case 16000: yading@11: case 8000: yading@11: fscode = 3; yading@11: } yading@11: avio_wb16(s, fscode); /* codec additional info, for AC-3, seems yading@11: to be a frequency code */ yading@11: /* special hack to compensate rounding errors... */ yading@11: if (coded_frame_size == 557) yading@11: coded_frame_size--; yading@11: avio_wb32(s, coded_frame_size); /* frame length */ yading@11: avio_wb32(s, 0x51540); /* unknown */ yading@11: avio_wb32(s, stream->enc->bit_rate / 8 * 60); /* bytes per minute */ yading@11: avio_wb32(s, stream->enc->bit_rate / 8 * 60); /* bytes per minute */ yading@11: avio_wb16(s, 0x01); yading@11: /* frame length : seems to be very important */ yading@11: avio_wb16(s, coded_frame_size); yading@11: avio_wb32(s, 0); /* unknown */ yading@11: avio_wb16(s, stream->enc->sample_rate); /* sample rate */ yading@11: avio_wb32(s, 0x10); /* unknown */ yading@11: avio_wb16(s, stream->enc->channels); yading@11: put_str8(s, "Int0"); /* codec name */ yading@11: if (stream->enc->codec_tag) { yading@11: avio_w8(s, 4); /* tag length */ yading@11: avio_wl32(s, stream->enc->codec_tag); yading@11: } else { yading@11: av_log(ctx, AV_LOG_ERROR, "Invalid codec tag\n"); yading@11: return -1; yading@11: } yading@11: avio_wb16(s, 0); /* title length */ yading@11: avio_wb16(s, 0); /* author length */ yading@11: avio_wb16(s, 0); /* copyright length */ yading@11: avio_w8(s, 0); /* end of header */ yading@11: } else { yading@11: /* video codec info */ yading@11: avio_wb32(s,34); /* size */ yading@11: ffio_wfourcc(s, "VIDO"); yading@11: if(stream->enc->codec_id == AV_CODEC_ID_RV10) yading@11: ffio_wfourcc(s,"RV10"); yading@11: else yading@11: ffio_wfourcc(s,"RV20"); yading@11: avio_wb16(s, stream->enc->width); yading@11: avio_wb16(s, stream->enc->height); yading@11: avio_wb16(s, (int) stream->frame_rate); /* frames per seconds ? */ yading@11: avio_wb32(s,0); /* unknown meaning */ yading@11: avio_wb16(s, (int) stream->frame_rate); /* unknown meaning */ yading@11: avio_wb32(s,0); /* unknown meaning */ yading@11: avio_wb16(s, 8); /* unknown meaning */ yading@11: /* Seems to be the codec version: only use basic H263. The next yading@11: versions seems to add a diffential DC coding as in yading@11: MPEG... nothing new under the sun */ yading@11: if(stream->enc->codec_id == AV_CODEC_ID_RV10) yading@11: avio_wb32(s,0x10000000); yading@11: else yading@11: avio_wb32(s,0x20103001); yading@11: //avio_wb32(s,0x10003000); yading@11: } yading@11: } yading@11: yading@11: /* patch data offset field */ yading@11: data_pos = s->buf_ptr - start_ptr; yading@11: rm->data_pos = data_pos; yading@11: data_offset_ptr[0] = data_pos >> 24; yading@11: data_offset_ptr[1] = data_pos >> 16; yading@11: data_offset_ptr[2] = data_pos >> 8; yading@11: data_offset_ptr[3] = data_pos; yading@11: yading@11: /* data stream */ yading@11: ffio_wfourcc(s, "DATA"); yading@11: avio_wb32(s,data_size + 10 + 8); yading@11: avio_wb16(s,0); yading@11: yading@11: avio_wb32(s, nb_packets); /* number of packets */ yading@11: avio_wb32(s,0); /* next data header */ yading@11: return 0; yading@11: } yading@11: yading@11: static void write_packet_header(AVFormatContext *ctx, StreamInfo *stream, yading@11: int length, int key_frame) yading@11: { yading@11: int timestamp; yading@11: AVIOContext *s = ctx->pb; yading@11: yading@11: stream->nb_packets++; yading@11: stream->packet_total_size += length; yading@11: if (length > stream->packet_max_size) yading@11: stream->packet_max_size = length; yading@11: yading@11: avio_wb16(s,0); /* version */ yading@11: avio_wb16(s,length + 12); yading@11: avio_wb16(s, stream->num); /* stream number */ yading@11: timestamp = (1000 * (float)stream->nb_frames) / stream->frame_rate; yading@11: avio_wb32(s, timestamp); /* timestamp */ yading@11: avio_w8(s, 0); /* reserved */ yading@11: avio_w8(s, key_frame ? 2 : 0); /* flags */ yading@11: } yading@11: yading@11: static int rm_write_header(AVFormatContext *s) yading@11: { yading@11: RMMuxContext *rm = s->priv_data; yading@11: StreamInfo *stream; yading@11: int n; yading@11: AVCodecContext *codec; yading@11: yading@11: if (s->nb_streams > 2) { yading@11: av_log(s, AV_LOG_ERROR, "At most 2 streams are currently supported for muxing in RM\n"); yading@11: return AVERROR_PATCHWELCOME; yading@11: } yading@11: yading@11: for(n=0;nnb_streams;n++) { yading@11: s->streams[n]->id = n; yading@11: codec = s->streams[n]->codec; yading@11: stream = &rm->streams[n]; yading@11: memset(stream, 0, sizeof(StreamInfo)); yading@11: stream->num = n; yading@11: stream->bit_rate = codec->bit_rate; yading@11: stream->enc = codec; yading@11: yading@11: switch(codec->codec_type) { yading@11: case AVMEDIA_TYPE_AUDIO: yading@11: rm->audio_stream = stream; yading@11: stream->frame_rate = (float)codec->sample_rate / (float)codec->frame_size; yading@11: /* XXX: dummy values */ yading@11: stream->packet_max_size = 1024; yading@11: stream->nb_packets = 0; yading@11: stream->total_frames = stream->nb_packets; yading@11: break; yading@11: case AVMEDIA_TYPE_VIDEO: yading@11: rm->video_stream = stream; yading@11: stream->frame_rate = (float)codec->time_base.den / (float)codec->time_base.num; yading@11: /* XXX: dummy values */ yading@11: stream->packet_max_size = 4096; yading@11: stream->nb_packets = 0; yading@11: stream->total_frames = stream->nb_packets; yading@11: break; yading@11: default: yading@11: return -1; yading@11: } yading@11: } yading@11: yading@11: if (rv10_write_header(s, 0, 0)) yading@11: return AVERROR_INVALIDDATA; yading@11: avio_flush(s->pb); yading@11: return 0; yading@11: } yading@11: yading@11: static int rm_write_audio(AVFormatContext *s, const uint8_t *buf, int size, int flags) yading@11: { yading@11: uint8_t *buf1; yading@11: RMMuxContext *rm = s->priv_data; yading@11: AVIOContext *pb = s->pb; yading@11: StreamInfo *stream = rm->audio_stream; yading@11: int i; yading@11: yading@11: /* XXX: suppress this malloc */ yading@11: buf1 = av_malloc(size * sizeof(uint8_t)); yading@11: yading@11: write_packet_header(s, stream, size, !!(flags & AV_PKT_FLAG_KEY)); yading@11: yading@11: if (stream->enc->codec_id == AV_CODEC_ID_AC3) { yading@11: /* for AC-3, the words seem to be reversed */ yading@11: for(i=0;inb_frames++; yading@11: av_free(buf1); yading@11: return 0; yading@11: } yading@11: yading@11: static int rm_write_video(AVFormatContext *s, const uint8_t *buf, int size, int flags) yading@11: { yading@11: RMMuxContext *rm = s->priv_data; yading@11: AVIOContext *pb = s->pb; yading@11: StreamInfo *stream = rm->video_stream; yading@11: int key_frame = !!(flags & AV_PKT_FLAG_KEY); yading@11: yading@11: /* XXX: this is incorrect: should be a parameter */ yading@11: yading@11: /* Well, I spent some time finding the meaning of these bits. I am yading@11: not sure I understood everything, but it works !! */ yading@11: #if 1 yading@11: write_packet_header(s, stream, size + 7 + (size >= 0x4000)*4, key_frame); yading@11: /* bit 7: '1' if final packet of a frame converted in several packets */ yading@11: avio_w8(pb, 0x81); yading@11: /* bit 7: '1' if I frame. bits 6..0 : sequence number in current yading@11: frame starting from 1 */ yading@11: if (key_frame) { yading@11: avio_w8(pb, 0x81); yading@11: } else { yading@11: avio_w8(pb, 0x01); yading@11: } yading@11: if(size >= 0x4000){ yading@11: avio_wb32(pb, size); /* total frame size */ yading@11: avio_wb32(pb, size); /* offset from the start or the end */ yading@11: }else{ yading@11: avio_wb16(pb, 0x4000 | size); /* total frame size */ yading@11: avio_wb16(pb, 0x4000 | size); /* offset from the start or the end */ yading@11: } yading@11: #else yading@11: /* full frame */ yading@11: write_packet_header(s, size + 6); yading@11: avio_w8(pb, 0xc0); yading@11: avio_wb16(pb, 0x4000 + size); /* total frame size */ yading@11: avio_wb16(pb, 0x4000 + packet_number * 126); /* position in stream */ yading@11: #endif yading@11: avio_w8(pb, stream->nb_frames & 0xff); yading@11: yading@11: avio_write(pb, buf, size); yading@11: yading@11: stream->nb_frames++; yading@11: return 0; yading@11: } yading@11: yading@11: static int rm_write_packet(AVFormatContext *s, AVPacket *pkt) yading@11: { yading@11: if (s->streams[pkt->stream_index]->codec->codec_type == yading@11: AVMEDIA_TYPE_AUDIO) yading@11: return rm_write_audio(s, pkt->data, pkt->size, pkt->flags); yading@11: else yading@11: return rm_write_video(s, pkt->data, pkt->size, pkt->flags); yading@11: } yading@11: yading@11: static int rm_write_trailer(AVFormatContext *s) yading@11: { yading@11: RMMuxContext *rm = s->priv_data; yading@11: int data_size, index_pos, i; yading@11: AVIOContext *pb = s->pb; yading@11: yading@11: if (s->pb->seekable) { yading@11: /* end of file: finish to write header */ yading@11: index_pos = avio_tell(pb); yading@11: data_size = index_pos - rm->data_pos; yading@11: yading@11: /* FIXME: write index */ yading@11: yading@11: /* undocumented end header */ yading@11: avio_wb32(pb, 0); yading@11: avio_wb32(pb, 0); yading@11: yading@11: avio_seek(pb, 0, SEEK_SET); yading@11: for(i=0;inb_streams;i++) yading@11: rm->streams[i].total_frames = rm->streams[i].nb_frames; yading@11: rv10_write_header(s, data_size, 0); yading@11: } else { yading@11: /* undocumented end header */ yading@11: avio_wb32(pb, 0); yading@11: avio_wb32(pb, 0); yading@11: } yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: yading@11: AVOutputFormat ff_rm_muxer = { yading@11: .name = "rm", yading@11: .long_name = NULL_IF_CONFIG_SMALL("RealMedia"), yading@11: .mime_type = "application/vnd.rn-realmedia", yading@11: .extensions = "rm,ra", yading@11: .priv_data_size = sizeof(RMMuxContext), yading@11: .audio_codec = AV_CODEC_ID_AC3, yading@11: .video_codec = AV_CODEC_ID_RV10, yading@11: .write_header = rm_write_header, yading@11: .write_packet = rm_write_packet, yading@11: .write_trailer = rm_write_trailer, yading@11: .codec_tag = (const AVCodecTag* const []){ ff_rm_codec_tags, 0 }, yading@11: };