/* * AVI encoder. * Copyright (c) 2000 Fabrice Bellard. * * This library 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 of the License, or (at your option) any later version. * * This library 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 this library; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ #include "avformat.h" #include "avi.h" /* * TODO: * - fill all fields if non streamed (nb_frames for example) */ typedef struct AVIIndex { unsigned char tag[4]; unsigned int flags, pos, len; struct AVIIndex *next; } AVIIndex; typedef struct { offset_t movi_list, frames_hdr_all, frames_hdr_strm[MAX_STREAMS]; int audio_strm_length[MAX_STREAMS]; AVIIndex *first, *last; } AVIContext; offset_t start_tag(ByteIOContext *pb, const char *tag) { put_tag(pb, tag); put_le32(pb, 0); return url_ftell(pb); } void end_tag(ByteIOContext *pb, offset_t start) { offset_t pos; pos = url_ftell(pb); url_fseek(pb, start - 4, SEEK_SET); put_le32(pb, (uint32_t)(pos - start)); url_fseek(pb, pos, SEEK_SET); } /* Note: when encoding, the first matching tag is used, so order is important if multiple tags possible for a given codec. */ const CodecTag codec_bmp_tags[] = { { CODEC_ID_H263, MKTAG('H', '2', '6', '3') }, { CODEC_ID_H263P, MKTAG('H', '2', '6', '3') }, { CODEC_ID_H263I, MKTAG('I', '2', '6', '3') }, /* intel h263 */ /* added based on MPlayer */ { CODEC_ID_H263I, MKTAG('i', '2', '6', '3') }, { CODEC_ID_H263P, MKTAG('U', '2', '6', '3') }, { CODEC_ID_H263P, MKTAG('h', '2', '6', '3') }, { CODEC_ID_H263P, MKTAG('v', 'i', 'v', '1') }, { CODEC_ID_MJPEG, MKTAG('M', 'J', 'P', 'G') }, { CODEC_ID_MPEG4, MKTAG('D', 'I', 'V', 'X'), .invalid_asf = 1 }, { CODEC_ID_MPEG4, MKTAG('d', 'i', 'v', 'x'), .invalid_asf = 1 }, { CODEC_ID_MPEG4, MKTAG('D', 'X', '5', '0'), .invalid_asf = 1 }, { CODEC_ID_MPEG4, MKTAG('X', 'V', 'I', 'D'), .invalid_asf = 1 }, { CODEC_ID_MPEG4, MKTAG('x', 'v', 'i', 'd'), .invalid_asf = 1 }, { CODEC_ID_MPEG4, MKTAG('m', 'p', '4', 's'), .invalid_asf = 1 }, { CODEC_ID_MPEG4, MKTAG('M', 'P', '4', 'S') }, { CODEC_ID_MPEG4, MKTAG('M', '4', 'S', '2') }, { CODEC_ID_MPEG4, MKTAG('m', '4', 's', '2') }, { CODEC_ID_MPEG4, MKTAG(0x04, 0, 0, 0) }, /* some broken avi use this */ /* added based on MPlayer */ { CODEC_ID_MPEG4, MKTAG('D', 'I', 'V', '1') }, { CODEC_ID_MPEG4, MKTAG('d', 'i', 'v', '1') }, { CODEC_ID_MPEG4, MKTAG('X', 'v', 'i', 'D') }, { CODEC_ID_MPEG4, MKTAG('B', 'L', 'Z', '0') }, { CODEC_ID_MPEG4, MKTAG('m', 'p', '4', 'v') }, { CODEC_ID_MPEG4, MKTAG('U', 'M', 'P', '4') }, { CODEC_ID_MSMPEG4V3, MKTAG('D', 'I', 'V', '3'), .invalid_asf = 1 }, /* default signature when using MSMPEG4 */ { CODEC_ID_MSMPEG4V3, MKTAG('d', 'i', 'v', '3'), .invalid_asf = 1 }, { CODEC_ID_MSMPEG4V3, MKTAG('M', 'P', '4', '3') }, /* added based on MPlayer */ { CODEC_ID_MSMPEG4V3, MKTAG('M', 'P', 'G', '3') }, { CODEC_ID_MSMPEG4V3, MKTAG('m', 'p', 'g', '3') }, { CODEC_ID_MSMPEG4V3, MKTAG('m', 'p', '4', '3') }, { CODEC_ID_MSMPEG4V3, MKTAG('D', 'I', 'V', '5') }, { CODEC_ID_MSMPEG4V3, MKTAG('d', 'i', 'v', '5') }, { CODEC_ID_MSMPEG4V3, MKTAG('D', 'I', 'V', '6') }, { CODEC_ID_MSMPEG4V3, MKTAG('d', 'i', 'v', '6') }, { CODEC_ID_MSMPEG4V3, MKTAG('D', 'I', 'V', '4') }, { CODEC_ID_MSMPEG4V3, MKTAG('d', 'i', 'v', '4') }, { CODEC_ID_MSMPEG4V3, MKTAG('A', 'P', '4', '1') }, { CODEC_ID_MSMPEG4V3, MKTAG('C', 'O', 'L', '1') }, { CODEC_ID_MSMPEG4V3, MKTAG('c', 'o', 'l', '1') }, { CODEC_ID_MSMPEG4V3, MKTAG('C', 'O', 'L', '0') }, { CODEC_ID_MSMPEG4V3, MKTAG('c', 'o', 'l', '0') }, { CODEC_ID_MSMPEG4V2, MKTAG('M', 'P', '4', '2') }, /* added based on MPlayer */ { CODEC_ID_MSMPEG4V2, MKTAG('D', 'I', 'V', '2') }, { CODEC_ID_MSMPEG4V2, MKTAG('d', 'i', 'v', '2') }, { CODEC_ID_MSMPEG4V2, MKTAG('m', 'p', '4', '2') }, { CODEC_ID_MSMPEG4V1, MKTAG('M', 'P', 'G', '4') }, /* added based on MPlayer */ { CODEC_ID_MSMPEG4V1, MKTAG('D', 'I', 'V', '4') }, { CODEC_ID_MSMPEG4V1, MKTAG('d', 'i', 'v', '4') }, { CODEC_ID_MSMPEG4V1, MKTAG('m', 'p', 'g', '4') }, { CODEC_ID_WMV1, MKTAG('W', 'M', 'V', '1') }, /* added based on MPlayer */ { CODEC_ID_WMV1, MKTAG('w', 'm', 'v', '1') }, { CODEC_ID_WMV2, MKTAG('W', 'M', 'V', '2') }, { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 's', 'l') }, { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 's', 'd') }, { CODEC_ID_DVVIDEO, MKTAG('D', 'V', 'S', 'D') }, { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 'h', 'd') }, { CODEC_ID_MPEG1VIDEO, MKTAG('m', 'p', 'g', '1') }, { CODEC_ID_MPEG1VIDEO, MKTAG('m', 'p', 'g', '2') }, { CODEC_ID_MPEG1VIDEO, MKTAG('P', 'I', 'M', '1') }, { CODEC_ID_MJPEG, MKTAG('M', 'J', 'P', 'G') }, { CODEC_ID_HUFFYUV, MKTAG('H', 'F', 'Y', 'U') }, { CODEC_ID_HUFFYUV, MKTAG('h', 'f', 'y', 'u') }, { CODEC_ID_CYUV, MKTAG('C', 'Y', 'U', 'V') }, { CODEC_ID_CYUV, MKTAG('c', 'y', 'u', 'v') }, { CODEC_ID_RAWVIDEO, MKTAG('Y', '4', '2', '2') }, { 0, 0 }, }; unsigned int codec_get_tag(const CodecTag *tags, int id) { while (tags->id != 0) { if (tags->id == id) return tags->tag; tags++; } return 0; } static unsigned int codec_get_asf_tag(const CodecTag *tags, int id) { while (tags->id != 0) { if (!tags->invalid_asf && tags->id == id) return tags->tag; tags++; } return 0; } int codec_get_id(const CodecTag *tags, unsigned int tag) { while (tags->id != 0) { if (tags->tag == tag) return tags->id; tags++; } return 0; } unsigned int codec_get_bmp_tag(int id) { return codec_get_tag(codec_bmp_tags, id); } /* BITMAPINFOHEADER header */ void put_bmp_header(ByteIOContext *pb, AVCodecContext *enc, const CodecTag *tags, int for_asf) { put_le32(pb, 40 + enc->extradata_size); /* size */ put_le32(pb, enc->width); put_le32(pb, enc->height); put_le16(pb, 1); /* planes */ put_le16(pb, enc->bits_per_sample ? enc->bits_per_sample : 24); /* depth */ /* compression type */ put_le32(pb, for_asf ? codec_get_asf_tag(tags, enc->codec_id) : codec_get_tag(tags, enc->codec_id)); put_le32(pb, enc->width * enc->height * 3); put_le32(pb, 0); put_le32(pb, 0); put_le32(pb, 0); put_le32(pb, 0); put_buffer(pb, enc->extradata, enc->extradata_size); if (enc->extradata_size & 1) put_byte(pb, 0); } static void parse_specific_params(AVCodecContext *stream, int *au_byterate, int *au_ssize, int *au_scale) { switch(stream->codec_id) { case CODEC_ID_PCM_S16LE: *au_scale = *au_ssize = 2*stream->channels; *au_byterate = *au_ssize * stream->sample_rate; break; case CODEC_ID_PCM_U8: case CODEC_ID_PCM_ALAW: case CODEC_ID_PCM_MULAW: *au_scale = *au_ssize = stream->channels; *au_byterate = *au_ssize * stream->sample_rate; break; case CODEC_ID_MP2: *au_ssize = 1; *au_scale = 1; *au_byterate = stream->bit_rate / 8; case CODEC_ID_MP3LAME: *au_ssize = 1; *au_scale = 1; *au_byterate = stream->bit_rate / 8; default: *au_ssize = 1; *au_scale = 1; *au_byterate = stream->bit_rate / 8; break; } } static int avi_write_header(AVFormatContext *s) { AVIContext *avi = s->priv_data; ByteIOContext *pb = &s->pb; int bitrate, n, i, nb_frames, au_byterate, au_ssize, au_scale; AVCodecContext *stream, *video_enc; offset_t list1, list2, strh, strf; put_tag(pb, "RIFF"); put_le32(pb, 0); /* file length */ put_tag(pb, "AVI "); /* header list */ list1 = start_tag(pb, "LIST"); put_tag(pb, "hdrl"); /* avi header */ put_tag(pb, "avih"); put_le32(pb, 14 * 4); bitrate = 0; video_enc = NULL; for(n=0;nnb_streams;n++) { stream = &s->streams[n]->codec; bitrate += stream->bit_rate; if (stream->codec_type == CODEC_TYPE_VIDEO) video_enc = stream; } /* allowing audio-only AVI file if (!video_enc) { av_free(avi); return -1; } */ nb_frames = 0; if(video_enc){ put_le32(pb, (uint32_t)(int64_t_C(1000000) * video_enc->frame_rate_base / video_enc->frame_rate)); } else { put_le32(pb, 0); } put_le32(pb, bitrate / 8); /* XXX: not quite exact */ put_le32(pb, 0); /* padding */ put_le32(pb, AVIF_TRUSTCKTYPE | AVIF_HASINDEX | AVIF_ISINTERLEAVED); /* flags */ avi->frames_hdr_all = url_ftell(pb); /* remember this offset to fill later */ put_le32(pb, nb_frames); /* nb frames, filled later */ put_le32(pb, 0); /* initial frame */ put_le32(pb, s->nb_streams); /* nb streams */ put_le32(pb, 1024 * 1024); /* suggested buffer size */ if(video_enc){ put_le32(pb, video_enc->width); put_le32(pb, video_enc->height); } else { put_le32(pb, 0); put_le32(pb, 0); } put_le32(pb, 0); /* reserved */ put_le32(pb, 0); /* reserved */ put_le32(pb, 0); /* reserved */ put_le32(pb, 0); /* reserved */ /* stream list */ for(i=0;istreams[i]->codec; /* FourCC should really be set by the codec itself */ if (! stream->codec_tag) { stream->codec_tag = codec_get_bmp_tag(stream->codec_id); } /* stream generic header */ strh = start_tag(pb, "strh"); switch(stream->codec_type) { case CODEC_TYPE_VIDEO: put_tag(pb, "vids"); put_le32(pb, stream->codec_tag); put_le32(pb, 0); /* flags */ put_le16(pb, 0); /* priority */ put_le16(pb, 0); /* language */ put_le32(pb, 0); /* initial frame */ put_le32(pb, stream->frame_rate_base); /* scale */ put_le32(pb, stream->frame_rate); /* rate */ put_le32(pb, 0); /* start */ avi->frames_hdr_strm[i] = url_ftell(pb); /* remember this offset to fill later */ put_le32(pb, nb_frames); /* length, XXX: fill later */ put_le32(pb, 1024 * 1024); /* suggested buffer size */ put_le32(pb, -1); /* quality */ put_le32(pb, stream->width * stream->height * 3); /* sample size */ put_le16(pb, 0); put_le16(pb, 0); put_le16(pb, stream->width); put_le16(pb, stream->height); break; case CODEC_TYPE_AUDIO: put_tag(pb, "auds"); put_le32(pb, 1); /* tag */ put_le32(pb, 0); /* flags */ put_le16(pb, 0); /* priority */ put_le16(pb, 0); /* language */ put_le32(pb, 0); /* initial frame */ parse_specific_params(stream, &au_byterate, &au_ssize, &au_scale); put_le32(pb, au_scale); /* scale */ put_le32(pb, au_byterate); /* rate */ put_le32(pb, 0); /* start */ avi->frames_hdr_strm[i] = url_ftell(pb); /* remember this offset to fill later */ put_le32(pb, 0); /* length, XXX: filled later */ put_le32(pb, 12 * 1024); /* suggested buffer size */ put_le32(pb, -1); /* quality */ put_le32(pb, au_ssize); /* sample size */ put_le32(pb, 0); put_le32(pb, 0); break; default: av_abort(); } end_tag(pb, strh); strf = start_tag(pb, "strf"); switch(stream->codec_type) { case CODEC_TYPE_VIDEO: put_bmp_header(pb, stream, codec_bmp_tags, 0); break; case CODEC_TYPE_AUDIO: if (put_wav_header(pb, stream) < 0) { av_free(avi); return -1; } break; default: av_abort(); } end_tag(pb, strf); end_tag(pb, list2); } end_tag(pb, list1); avi->movi_list = start_tag(pb, "LIST"); avi->first = NULL; avi->last = NULL; put_tag(pb, "movi"); put_flush_packet(pb); return 0; } static int avi_write_packet(AVFormatContext *s, int stream_index, uint8_t *buf, int size, int force_pts) { AVIContext *avi = s->priv_data; ByteIOContext *pb = &s->pb; AVIIndex *idx; unsigned char tag[5]; unsigned int flags; AVCodecContext *enc; enc = &s->streams[stream_index]->codec; tag[0] = '0'; tag[1] = '0' + stream_index; if (enc->codec_type == CODEC_TYPE_VIDEO) { tag[2] = 'd'; tag[3] = 'c'; flags = enc->coded_frame->key_frame ? 0x10 : 0x00; } else { tag[2] = 'w'; tag[3] = 'b'; flags = 0x10; } if (enc->codec_type == CODEC_TYPE_AUDIO) avi->audio_strm_length[stream_index] += size; if (!url_is_streamed(&s->pb)) { idx = av_malloc(sizeof(AVIIndex)); memcpy(idx->tag, tag, 4); idx->flags = flags; idx->pos = url_ftell(pb) - avi->movi_list; idx->len = size; idx->next = NULL; if (!avi->last) avi->first = idx; else avi->last->next = idx; avi->last = idx; } put_buffer(pb, tag, 4); put_le32(pb, size); put_buffer(pb, buf, size); if (size & 1) put_byte(pb, 0); put_flush_packet(pb); return 0; } static int avi_write_trailer(AVFormatContext *s) { ByteIOContext *pb = &s->pb; AVIContext *avi = s->priv_data; offset_t file_size, idx_chunk; int n, nb_frames, au_byterate, au_ssize, au_scale; AVCodecContext *stream; AVIIndex *idx; if (!url_is_streamed(&s->pb)) { end_tag(pb, avi->movi_list); idx_chunk = start_tag(pb, "idx1"); idx = avi->first; while (idx != NULL) { put_buffer(pb, idx->tag, 4); put_le32(pb, idx->flags); put_le32(pb, idx->pos); put_le32(pb, idx->len); idx = idx->next; } end_tag(pb, idx_chunk); /* update file size */ file_size = url_ftell(pb); url_fseek(pb, 4, SEEK_SET); put_le32(pb, (uint32_t)(file_size - 8)); /* Fill in frame/sample counters */ nb_frames = 0; for(n=0;nnb_streams;n++) { if (avi->frames_hdr_strm[n] != 0) { stream = &s->streams[n]->codec; url_fseek(pb, avi->frames_hdr_strm[n], SEEK_SET); if (stream->codec_type == CODEC_TYPE_VIDEO) { put_le32(pb, stream->frame_number); if (nb_frames < stream->frame_number) nb_frames = stream->frame_number; } else { if (stream->codec_id == CODEC_ID_MP2 || stream->codec_id == CODEC_ID_MP3LAME) { put_le32(pb, stream->frame_number); nb_frames += stream->frame_number; } else { parse_specific_params(stream, &au_byterate, &au_ssize, &au_scale); put_le32(pb, avi->audio_strm_length[n] / au_ssize); } } } } if (avi->frames_hdr_all != 0) { url_fseek(pb, avi->frames_hdr_all, SEEK_SET); put_le32(pb, nb_frames); } url_fseek(pb, file_size, SEEK_SET); } put_flush_packet(pb); return 0; } static AVOutputFormat avi_oformat = { "avi", "avi format", "video/x-msvideo", "avi", sizeof(AVIContext), CODEC_ID_MP2, CODEC_ID_MSMPEG4V3, avi_write_header, avi_write_packet, avi_write_trailer, }; int avienc_init(void) { av_register_output_format(&avi_oformat); return 0; }