/* * 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) */#ifdef CONFIG_ENCODERStypedef struct AVIIentry {    unsigned int flags, pos, len;} AVIIentry;#define AVI_INDEX_CLUSTER_SIZE 16384typedef struct AVIIndex {    offset_t    indx_start;    int         entry;    int         ents_allocated;    AVIIentry** cluster;} AVIIndex;typedef struct {    offset_t riff_start, movi_list, odml_list;    offset_t frames_hdr_all, frames_hdr_strm[MAX_STREAMS];    int audio_strm_length[MAX_STREAMS];    int riff_id;    AVIIndex indexes[MAX_STREAMS];} AVIContext;static inline AVIIentry* avi_get_ientry(AVIIndex* idx, int ent_id) {    int cl = ent_id / AVI_INDEX_CLUSTER_SIZE;    int id = ent_id % AVI_INDEX_CLUSTER_SIZE;    return &idx->cluster[cl][id];}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);}#endif //CONFIG_ENCODERS/* 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_H264, MKTAG('H', '2', '6', '4') },    { 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_H263P, MKTAG('U', '2', '6', '3') },    { CODEC_ID_H263P, MKTAG('v', 'i', 'v', '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('M', 'P', '4', 'S') },    { 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('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('M', 'P', '4', '3') },     /* added based on MPlayer */    { CODEC_ID_MSMPEG4V3, MKTAG('M', 'P', 'G', '3') },     { CODEC_ID_MSMPEG4V3, MKTAG('D', 'I', 'V', '5') },     { CODEC_ID_MSMPEG4V3, MKTAG('D', 'I', 'V', '6') },     { 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', '0') },     { CODEC_ID_MSMPEG4V2, MKTAG('M', 'P', '4', '2') },     /* added based on MPlayer */    { CODEC_ID_MSMPEG4V2, MKTAG('D', 'I', 'V', '2') },     { CODEC_ID_MSMPEG4V1, MKTAG('M', 'P', 'G', '4') },     { CODEC_ID_WMV1, MKTAG('W', 'M', 'V', '1') },     /* added based on MPlayer */    { CODEC_ID_WMV2, MKTAG('W', 'M', 'V', '2') },     { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 's', 'd') },     { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 'h', 'd') },     { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 's', 'l') },     { CODEC_ID_DVVIDEO, MKTAG('d', 'v', '2', '5') },    { CODEC_ID_MPEG1VIDEO, MKTAG('m', 'p', 'g', '1') },     { CODEC_ID_MPEG1VIDEO, MKTAG('m', 'p', 'g', '2') },     { CODEC_ID_MPEG2VIDEO, MKTAG('m', 'p', 'g', '2') },     { CODEC_ID_MPEG1VIDEO, MKTAG('P', 'I', 'M', '1') },     { CODEC_ID_MPEG1VIDEO, MKTAG('V', 'C', 'R', '2') },     { CODEC_ID_MJPEG, MKTAG('M', 'J', 'P', 'G') },    { CODEC_ID_MJPEG, MKTAG('L', 'J', 'P', 'G') },    { CODEC_ID_LJPEG, MKTAG('L', 'J', 'P', 'G') },    { CODEC_ID_MJPEG, MKTAG('J', 'P', 'G', 'L') }, /* Pegasus lossless JPEG */    { CODEC_ID_HUFFYUV, MKTAG('H', 'F', 'Y', 'U') },    { CODEC_ID_CYUV, MKTAG('C', 'Y', 'U', 'V') },    { CODEC_ID_RAWVIDEO, MKTAG('Y', '4', '2', '2') },    { CODEC_ID_RAWVIDEO, MKTAG('I', '4', '2', '0') },    { CODEC_ID_INDEO3, MKTAG('I', 'V', '3', '1') },    { CODEC_ID_INDEO3, MKTAG('I', 'V', '3', '2') },    { CODEC_ID_VP3, MKTAG('V', 'P', '3', '1') },    { CODEC_ID_ASV1, MKTAG('A', 'S', 'V', '1') },    { CODEC_ID_ASV2, MKTAG('A', 'S', 'V', '2') },    { CODEC_ID_VCR1, MKTAG('V', 'C', 'R', '1') },    { CODEC_ID_FFV1, MKTAG('F', 'F', 'V', '1') },    { CODEC_ID_XAN_WC4, MKTAG('X', 'x', 'a', 'n') },    { CODEC_ID_MSRLE, MKTAG('m', 'r', 'l', 'e') },    { CODEC_ID_MSRLE, MKTAG(0x1, 0x0, 0x0, 0x0) },    { CODEC_ID_MSVIDEO1, MKTAG('M', 'S', 'V', 'C') },    { CODEC_ID_MSVIDEO1, MKTAG('m', 's', 'v', 'c') },    { CODEC_ID_MSVIDEO1, MKTAG('C', 'R', 'A', 'M') },    { CODEC_ID_MSVIDEO1, MKTAG('c', 'r', 'a', 'm') },    { CODEC_ID_MSVIDEO1, MKTAG('W', 'H', 'A', 'M') },    { CODEC_ID_MSVIDEO1, MKTAG('w', 'h', 'a', 'm') },    { CODEC_ID_CINEPAK, MKTAG('c', 'v', 'i', 'd') },    { CODEC_ID_TRUEMOTION1, MKTAG('D', 'U', 'C', 'K') },    { CODEC_ID_MSZH, MKTAG('M', 'S', 'Z', 'H') },    { CODEC_ID_ZLIB, MKTAG('Z', 'L', 'I', 'B') },    { CODEC_ID_4XM, MKTAG('4', 'X', 'M', 'V') },    { CODEC_ID_FLV1, MKTAG('F', 'L', 'V', '1') },    { 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;}enum CodecID codec_get_id(const CodecTag *tags, unsigned int tag){    while (tags->id != 0) {        if(   toupper((tag >> 0)&0xFF) == toupper((tags->tag >> 0)&0xFF)           && toupper((tag >> 8)&0xFF) == toupper((tags->tag >> 8)&0xFF)           && toupper((tag >>16)&0xFF) == toupper((tags->tag >>16)&0xFF)           && toupper((tag >>24)&0xFF) == toupper((tags->tag >>24)&0xFF))            return tags->id;        tags++;    }    return CODEC_ID_NONE;}unsigned int codec_get_bmp_tag(int id){    return codec_get_tag(codec_bmp_tags, id);}unsigned int codec_get_wav_tag(int id){    return codec_get_tag(codec_wav_tags, id);}enum CodecID codec_get_bmp_id(unsigned int tag){    return codec_get_id(codec_bmp_tags, tag);}enum CodecID codec_get_wav_id(unsigned int tag){    return codec_get_id(codec_wav_tags, tag);}#ifdef CONFIG_ENCODERS/* 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) : enc->codec_tag);    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_MP3:        *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 offset_t avi_start_new_riff(AVIContext *avi, ByteIOContext *pb,                                    const char* riff_tag, const char* list_tag){    offset_t loff;    int i;        avi->riff_id++;    for (i=0; i<MAX_STREAMS; i++)         avi->indexes[i].entry = 0;        avi->riff_start = start_tag(pb, "RIFF");    put_tag(pb, riff_tag);    loff = start_tag(pb, "LIST");    put_tag(pb, list_tag);    return loff;}static unsigned char* avi_stream2fourcc(unsigned char* tag, int index,                                         enum CodecType type){    tag[0] = '0';    tag[1] = '0' + index;    if (type == CODEC_TYPE_VIDEO) {        tag[2] = 'd';        tag[3] = 'c';    } else {        tag[2] = 'w';        tag[3] = 'b';    }    tag[4] = '\0';    return tag;}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;    /* header list */    avi->riff_id = 0;    list1 = avi_start_new_riff(avi, pb, "AVI ", "hdrl");    /* avi header */    put_tag(pb, "avih");    put_le32(pb, 14 * 4);    bitrate = 0;    video_enc = NULL;    for(n=0;n<s->nb_streams;n++) {        stream = &s->streams[n]->codec;        bitrate += stream->bit_rate;        if (stream->codec_type == CODEC_TYPE_VIDEO)            video_enc = stream;    }        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;i<n;i++) {        list2 = start_tag(pb, "LIST");        put_tag(pb, "strl");            stream = &s->streams[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);        }