avienc.c 21.7 KB
Newer Older
Fabrice Bellard's avatar
Fabrice Bellard committed
1
/*
2
 * AVI muxer
3
 * Copyright (c) 2000 Fabrice Bellard
Fabrice Bellard's avatar
Fabrice Bellard committed
4
 *
5 6 7
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
8 9
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
10
 * version 2.1 of the License, or (at your option) any later version.
Fabrice Bellard's avatar
Fabrice Bellard committed
11
 *
12
 * FFmpeg is distributed in the hope that it will be useful,
Fabrice Bellard's avatar
Fabrice Bellard committed
13
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 15
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
Fabrice Bellard's avatar
Fabrice Bellard committed
16
 *
17
 * You should have received a copy of the GNU Lesser General Public
18
 * License along with FFmpeg; if not, write to the Free Software
19
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
Fabrice Bellard's avatar
Fabrice Bellard committed
20
 */
21 22 23

//#define DEBUG

Fabrice Bellard's avatar
Fabrice Bellard committed
24
#include "avformat.h"
25
#include "internal.h"
Fabrice Bellard's avatar
Fabrice Bellard committed
26
#include "avi.h"
27
#include "avio_internal.h"
28
#include "riff.h"
29
#include "libavutil/intreadwrite.h"
30
#include "libavutil/dict.h"
31
#include "libavutil/avassert.h"
32
#include "libavutil/timestamp.h"
Fabrice Bellard's avatar
Fabrice Bellard committed
33 34

/*
35
 * TODO:
Fabrice Bellard's avatar
Fabrice Bellard committed
36 37 38
 *  - fill all fields if non streamed (nb_frames for example)
 */

39
typedef struct AVIIentry {
Fabrice Bellard's avatar
Fabrice Bellard committed
40
    unsigned int flags, pos, len;
41 42 43 44 45
} AVIIentry;

#define AVI_INDEX_CLUSTER_SIZE 16384

typedef struct AVIIndex {
46
    int64_t     indx_start;
47 48 49
    int         entry;
    int         ents_allocated;
    AVIIentry** cluster;
Fabrice Bellard's avatar
Fabrice Bellard committed
50 51 52
} AVIIndex;

typedef struct {
53
    int64_t riff_start, movi_list, odml_list;
54
    int64_t frames_hdr_all;
55
    int riff_id;
Fabrice Bellard's avatar
Fabrice Bellard committed
56 57
} AVIContext;

58 59
typedef struct  {
    int64_t frames_hdr_strm;
60
    int64_t audio_strm_length;
61
    int packet_count;
62
    int entry;
63 64 65 66

    AVIIndex indexes;
} AVIStream ;

67
static inline AVIIentry* avi_get_ientry(AVIIndex* idx, int ent_id)
68 69 70 71 72 73
{
    int cl = ent_id / AVI_INDEX_CLUSTER_SIZE;
    int id = ent_id % AVI_INDEX_CLUSTER_SIZE;
    return &idx->cluster[cl][id];
}

74
static int64_t avi_start_new_riff(AVFormatContext *s, AVIOContext *pb,
75
                                  const char* riff_tag, const char* list_tag)
76
{
77
    AVIContext *avi= s->priv_data;
78
    int64_t loff;
79
    int i;
80

81
    avi->riff_id++;
82 83 84 85
    for (i=0; i<s->nb_streams; i++){
        AVIStream *avist= s->streams[i]->priv_data;
        avist->indexes.entry = 0;
    }
86

87
    avi->riff_start = ff_start_tag(pb, "RIFF");
88
    ffio_wfourcc(pb, riff_tag);
89
    loff = ff_start_tag(pb, "LIST");
90
    ffio_wfourcc(pb, list_tag);
91 92 93
    return loff;
}

94
static char* avi_stream2fourcc(char* tag, int index, enum AVMediaType type)
95
{
96 97
    tag[0] = '0' + index/10;
    tag[1] = '0' + index%10;
98
    if (type == AVMEDIA_TYPE_VIDEO) {
99 100
        tag[2] = 'd';
        tag[3] = 'c';
101
    } else if (type == AVMEDIA_TYPE_SUBTITLE) {
102 103 104
        // note: this is not an official code
        tag[2] = 's';
        tag[3] = 'b';
105 106 107 108 109 110 111 112
    } else {
        tag[2] = 'w';
        tag[3] = 'b';
    }
    tag[4] = '\0';
    return tag;
}

113 114
static int avi_write_counters(AVFormatContext* s, int riff_id)
{
115
    AVIOContext *pb = s->pb;
116 117
    AVIContext *avi = s->priv_data;
    int n, au_byterate, au_ssize, au_scale, nb_frames = 0;
118
    int64_t file_size;
119 120
    AVCodecContext* stream;

121
    file_size = avio_tell(pb);
122
    for(n = 0; n < s->nb_streams; n++) {
123 124
        AVIStream *avist= s->streams[n]->priv_data;

125
        av_assert0(avist->frames_hdr_strm);
126
        stream = s->streams[n]->codec;
127
        avio_seek(pb, avist->frames_hdr_strm, SEEK_SET);
128 129
        ff_parse_specific_params(stream, &au_byterate, &au_ssize, &au_scale);
        if(au_ssize == 0) {
130
            avio_wl32(pb, avist->packet_count);
131
        } else {
132
            avio_wl32(pb, avist->audio_strm_length / au_ssize);
133
        }
134
        if(stream->codec_type == AVMEDIA_TYPE_VIDEO)
135
            nb_frames = FFMAX(nb_frames, avist->packet_count);
136 137
    }
    if(riff_id == 1) {
138
        av_assert0(avi->frames_hdr_all);
139
        avio_seek(pb, avi->frames_hdr_all, SEEK_SET);
140
        avio_wl32(pb, nb_frames);
141
    }
142
    avio_seek(pb, file_size, SEEK_SET);
143 144 145 146

    return 0;
}

Fabrice Bellard's avatar
Fabrice Bellard committed
147 148
static int avi_write_header(AVFormatContext *s)
{
Fabrice Bellard's avatar
Fabrice Bellard committed
149
    AVIContext *avi = s->priv_data;
150
    AVIOContext *pb = s->pb;
151
    int bitrate, n, i, nb_frames, au_byterate, au_ssize, au_scale;
Fabrice Bellard's avatar
Fabrice Bellard committed
152
    AVCodecContext *stream, *video_enc;
153
    int64_t list1, list2, strh, strf;
154
    AVDictionaryEntry *t = NULL;
Fabrice Bellard's avatar
Fabrice Bellard committed
155

156 157 158
    if (s->nb_streams > AVI_MAX_STREAM_COUNT) {
        av_log(s, AV_LOG_ERROR, "AVI does not support >%d streams\n",
               AVI_MAX_STREAM_COUNT);
159
        return AVERROR(EINVAL);
160 161
    }

162 163 164 165 166 167
    for(n=0;n<s->nb_streams;n++) {
        s->streams[n]->priv_data= av_mallocz(sizeof(AVIStream));
        if(!s->streams[n]->priv_data)
            return AVERROR(ENOMEM);
    }

Fabrice Bellard's avatar
Fabrice Bellard committed
168
    /* header list */
169
    avi->riff_id = 0;
170
    list1 = avi_start_new_riff(s, pb, "AVI ", "hdrl");
Fabrice Bellard's avatar
Fabrice Bellard committed
171 172

    /* avi header */
173
    ffio_wfourcc(pb, "avih");
174
    avio_wl32(pb, 14 * 4);
Fabrice Bellard's avatar
Fabrice Bellard committed
175 176 177 178
    bitrate = 0;

    video_enc = NULL;
    for(n=0;n<s->nb_streams;n++) {
179
        stream = s->streams[n]->codec;
Fabrice Bellard's avatar
Fabrice Bellard committed
180
        bitrate += stream->bit_rate;
181
        if (stream->codec_type == AVMEDIA_TYPE_VIDEO)
Fabrice Bellard's avatar
Fabrice Bellard committed
182 183
            video_enc = stream;
    }
184

Fabrice Bellard's avatar
Fabrice Bellard committed
185 186
    nb_frames = 0;

187
    if(video_enc){
188
        avio_wl32(pb, (uint32_t)(INT64_C(1000000) * video_enc->time_base.num / video_enc->time_base.den));
189
    } else {
190
        avio_wl32(pb, 0);
191
    }
192 193
    avio_wl32(pb, bitrate / 8); /* XXX: not quite exact */
    avio_wl32(pb, 0); /* padding */
194
    if (!pb->seekable)
195
        avio_wl32(pb, AVIF_TRUSTCKTYPE | AVIF_ISINTERLEAVED); /* flags */
196
    else
197
        avio_wl32(pb, AVIF_TRUSTCKTYPE | AVIF_HASINDEX | AVIF_ISINTERLEAVED); /* flags */
198
    avi->frames_hdr_all = avio_tell(pb); /* remember this offset to fill later */
199 200 201 202
    avio_wl32(pb, nb_frames); /* nb frames, filled later */
    avio_wl32(pb, 0); /* initial frame */
    avio_wl32(pb, s->nb_streams); /* nb streams */
    avio_wl32(pb, 1024 * 1024); /* suggested buffer size */
203
    if(video_enc){
204 205
        avio_wl32(pb, video_enc->width);
        avio_wl32(pb, video_enc->height);
206
    } else {
207 208
        avio_wl32(pb, 0);
        avio_wl32(pb, 0);
209
    }
210 211 212 213
    avio_wl32(pb, 0); /* reserved */
    avio_wl32(pb, 0); /* reserved */
    avio_wl32(pb, 0); /* reserved */
    avio_wl32(pb, 0); /* reserved */
214

Fabrice Bellard's avatar
Fabrice Bellard committed
215 216
    /* stream list */
    for(i=0;i<n;i++) {
217
        AVIStream *avist= s->streams[i]->priv_data;
218
        list2 = ff_start_tag(pb, "LIST");
219
        ffio_wfourcc(pb, "strl");
220

221
        stream = s->streams[i]->codec;
Fabrice Bellard's avatar
Fabrice Bellard committed
222 223

        /* stream generic header */
224
        strh = ff_start_tag(pb, "strh");
Fabrice Bellard's avatar
Fabrice Bellard committed
225
        switch(stream->codec_type) {
226
        case AVMEDIA_TYPE_SUBTITLE:
227 228
            // XSUB subtitles behave like video tracks, other subtitles
            // are not (yet) supported.
229
            if (stream->codec_id != AV_CODEC_ID_XSUB) {
230 231 232
                av_log(s, AV_LOG_ERROR, "Subtitle streams other than DivX XSUB are not supported by the AVI muxer.\n");
                return AVERROR_PATCHWELCOME;
            }
233 234 235 236
        case AVMEDIA_TYPE_VIDEO: ffio_wfourcc(pb, "vids"); break;
        case AVMEDIA_TYPE_AUDIO: ffio_wfourcc(pb, "auds"); break;
//      case AVMEDIA_TYPE_TEXT : ffio_wfourcc(pb, "txts"); break;
        case AVMEDIA_TYPE_DATA : ffio_wfourcc(pb, "dats"); break;
Fabrice Bellard's avatar
Fabrice Bellard committed
237
        }
238
        if(stream->codec_type == AVMEDIA_TYPE_VIDEO ||
239
           stream->codec_id == AV_CODEC_ID_XSUB)
240
            avio_wl32(pb, stream->codec_tag);
241
        else
242 243 244 245 246
            avio_wl32(pb, 1);
        avio_wl32(pb, 0); /* flags */
        avio_wl16(pb, 0); /* priority */
        avio_wl16(pb, 0); /* language */
        avio_wl32(pb, 0); /* initial frame */
247 248 249

        ff_parse_specific_params(stream, &au_byterate, &au_ssize, &au_scale);

250 251 252 253 254 255
        if (   stream->codec_type == AVMEDIA_TYPE_VIDEO
            && stream->codec_id != AV_CODEC_ID_XSUB
            && au_byterate > 1000LL*au_scale) {
            au_byterate = 600;
            au_scale    = 1;
        }
256
        avpriv_set_pts_info(s->streams[i], 64, au_scale, au_byterate);
257
        if(stream->codec_id == AV_CODEC_ID_XSUB)
258 259
            au_scale = au_byterate = 0;

260 261
        avio_wl32(pb, au_scale); /* scale */
        avio_wl32(pb, au_byterate); /* rate */
262

263
        avio_wl32(pb, 0); /* start */
264
        avist->frames_hdr_strm = avio_tell(pb); /* remember this offset to fill later */
265
        if (!pb->seekable)
266
            avio_wl32(pb, AVI_MAX_RIFF_SIZE); /* FIXME: this may be broken, but who cares */
267
        else
268
            avio_wl32(pb, 0); /* length, XXX: filled later */
269

270
        /* suggested buffer size */ //FIXME set at the end to largest chunk
271
        if(stream->codec_type == AVMEDIA_TYPE_VIDEO)
272
            avio_wl32(pb, 1024 * 1024);
273
        else if(stream->codec_type == AVMEDIA_TYPE_AUDIO)
274
            avio_wl32(pb, 12 * 1024);
275
        else
276 277 278 279 280 281
            avio_wl32(pb, 0);
        avio_wl32(pb, -1); /* quality */
        avio_wl32(pb, au_ssize); /* sample size */
        avio_wl32(pb, 0);
        avio_wl16(pb, stream->width);
        avio_wl16(pb, stream->height);
282
        ff_end_tag(pb, strh);
Fabrice Bellard's avatar
Fabrice Bellard committed
283

284
      if(stream->codec_type != AVMEDIA_TYPE_DATA){
285 286
          int ret;

287
        strf = ff_start_tag(pb, "strf");
Fabrice Bellard's avatar
Fabrice Bellard committed
288
        switch(stream->codec_type) {
289
        case AVMEDIA_TYPE_SUBTITLE:
290 291
            // XSUB subtitles behave like video tracks, other subtitles
            // are not (yet) supported.
292
            if (stream->codec_id != AV_CODEC_ID_XSUB) break;
293
        case AVMEDIA_TYPE_VIDEO:
294
            ff_put_bmp_header(pb, stream, ff_codec_bmp_tags, 0);
Fabrice Bellard's avatar
Fabrice Bellard committed
295
            break;
296
        case AVMEDIA_TYPE_AUDIO:
297 298
            if ((ret = ff_put_wav_header(pb, stream)) < 0) {
                return ret;
299
            }
Fabrice Bellard's avatar
Fabrice Bellard committed
300
            break;
301
        default:
302 303 304 305
            av_log(s, AV_LOG_ERROR,
                   "Invalid or not supported codec type '%s' found in the input\n",
                   (char *)av_x_if_null(av_get_media_type_string(stream->codec_type), "?"));
            return AVERROR(EINVAL);
Fabrice Bellard's avatar
Fabrice Bellard committed
306
        }
307
        ff_end_tag(pb, strf);
308
        if ((t = av_dict_get(s->streams[i]->metadata, "title", NULL, 0))) {
309
            ff_riff_write_info_tag(s->pb, "strn", t->value);
310 311
            t = NULL;
        }
312
      }
313

314
        if (pb->seekable) {
315 316
            unsigned char tag[5];
            int j;
317 318

            /* Starting to lay out AVI OpenDML master index.
319 320 321 322
             * We want to make it JUNK entry for now, since we'd
             * like to get away without making AVI an OpenDML one
             * for compatibility reasons.
             */
323 324
            avist->indexes.entry = avist->indexes.ents_allocated = 0;
            avist->indexes.indx_start = ff_start_tag(pb, "JUNK");
325 326 327 328
            avio_wl16(pb, 4);        /* wLongsPerEntry */
            avio_w8(pb, 0);          /* bIndexSubType (0 == frame index) */
            avio_w8(pb, 0);          /* bIndexType (0 == AVI_INDEX_OF_INDEXES) */
            avio_wl32(pb, 0);        /* nEntriesInUse (will fill out later on) */
329
            ffio_wfourcc(pb, avi_stream2fourcc(tag, i, stream->codec_type));
330
                                    /* dwChunkId */
331 332
            avio_wl64(pb, 0);        /* dwReserved[3]
            avio_wl32(pb, 0);           Must be 0.    */
333
            for (j=0; j < AVI_MASTER_INDEX_SIZE * 2; j++)
334
                 avio_wl64(pb, 0);
335
            ff_end_tag(pb, avist->indexes.indx_start);
336
        }
337

338
        if(   stream->codec_type == AVMEDIA_TYPE_VIDEO
339 340
           && s->streams[i]->sample_aspect_ratio.num>0
           && s->streams[i]->sample_aspect_ratio.den>0){
341
            int vprp= ff_start_tag(pb, "vprp");
342
            AVRational dar = av_mul_q(s->streams[i]->sample_aspect_ratio,
343 344 345 346
                                      (AVRational){stream->width, stream->height});
            int num, den;
            av_reduce(&num, &den, dar.num, dar.den, 0xFFFF);

347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366
            avio_wl32(pb, 0); //video format  = unknown
            avio_wl32(pb, 0); //video standard= unknown
            avio_wl32(pb, lrintf(1.0/av_q2d(stream->time_base)));
            avio_wl32(pb, stream->width );
            avio_wl32(pb, stream->height);
            avio_wl16(pb, den);
            avio_wl16(pb, num);
            avio_wl32(pb, stream->width );
            avio_wl32(pb, stream->height);
            avio_wl32(pb, 1); //progressive FIXME

            avio_wl32(pb, stream->height);
            avio_wl32(pb, stream->width );
            avio_wl32(pb, stream->height);
            avio_wl32(pb, stream->width );
            avio_wl32(pb, 0);
            avio_wl32(pb, 0);

            avio_wl32(pb, 0);
            avio_wl32(pb, 0);
367
            ff_end_tag(pb, vprp);
368 369
        }

370
        ff_end_tag(pb, list2);
Fabrice Bellard's avatar
Fabrice Bellard committed
371
    }
372

373
    if (pb->seekable) {
374
        /* AVI could become an OpenDML one, if it grows beyond 2Gb range */
375
        avi->odml_list = ff_start_tag(pb, "JUNK");
376 377
        ffio_wfourcc(pb, "odml");
        ffio_wfourcc(pb, "dmlh");
378
        avio_wl32(pb, 248);
379
        for (i = 0; i < 248; i+= 4)
380
             avio_wl32(pb, 0);
381
        ff_end_tag(pb, avi->odml_list);
382
    }
Fabrice Bellard's avatar
Fabrice Bellard committed
383

384
    ff_end_tag(pb, list1);
385

386
    ff_riff_write_info(s);
387 388

    /* some padding for easier tag editing */
389
    list2 = ff_start_tag(pb, "JUNK");
390
    for (i = 0; i < 1016; i += 4)
391
        avio_wl32(pb, 0);
392
    ff_end_tag(pb, list2);
393

394
    avi->movi_list = ff_start_tag(pb, "LIST");
395
    ffio_wfourcc(pb, "movi");
Fabrice Bellard's avatar
Fabrice Bellard committed
396

397
    avio_flush(pb);
Fabrice Bellard's avatar
Fabrice Bellard committed
398 399 400 401

    return 0;
}

402 403
static int avi_write_ix(AVFormatContext *s)
{
404
    AVIOContext *pb = s->pb;
405
    AVIContext *avi = s->priv_data;
406 407
    char tag[5];
    char ix_tag[] = "ix00";
408
    int i, j;
409

410
    av_assert0(pb->seekable);
411

412 413 414 415 416
    if (avi->riff_id > AVI_MASTER_INDEX_SIZE) {
        av_log(s, AV_LOG_ERROR, "Invalid riff index %d > %d\n",
               avi->riff_id, AVI_MASTER_INDEX_SIZE);
        return AVERROR(EINVAL);
    }
417

418
    for (i=0;i<s->nb_streams;i++) {
419
        AVIStream *avist= s->streams[i]->priv_data;
420
         int64_t ix, pos;
421

422
         avi_stream2fourcc(tag, i, s->streams[i]->codec->codec_type);
423
         ix_tag[3] = '0' + i;
424

425
         /* Writing AVI OpenDML leaf index chunk */
426
         ix = avio_tell(pb);
427
         ffio_wfourcc(pb, ix_tag);     /* ix?? */
428
         avio_wl32(pb, avist->indexes.entry * 8 + 24);
429
                                      /* chunk size */
430 431 432 433
         avio_wl16(pb, 2);             /* wLongsPerEntry */
         avio_w8(pb, 0);             /* bIndexSubType (0 == frame index) */
         avio_w8(pb, 1);             /* bIndexType (1 == AVI_INDEX_OF_CHUNKS) */
         avio_wl32(pb, avist->indexes.entry);
434
                                      /* nEntriesInUse */
435
         ffio_wfourcc(pb, tag);        /* dwChunkId */
436 437
         avio_wl64(pb, avi->movi_list);/* qwBaseOffset */
         avio_wl32(pb, 0);             /* dwReserved_3 (must be 0) */
438

439 440
         for (j=0; j<avist->indexes.entry; j++) {
             AVIIentry* ie = avi_get_ientry(&avist->indexes, j);
441 442
             avio_wl32(pb, ie->pos + 8);
             avio_wl32(pb, ((uint32_t)ie->len & ~0x80000000) |
443
                          (ie->flags & 0x10 ? 0 : 0x80000000));
444
         }
445
         avio_flush(pb);
446
         pos = avio_tell(pb);
447

448
         /* Updating one entry in the AVI OpenDML master index */
449
         avio_seek(pb, avist->indexes.indx_start - 8, SEEK_SET);
450
         ffio_wfourcc(pb, "indx");            /* enabling this entry */
451
         avio_skip(pb, 8);
452
         avio_wl32(pb, avi->riff_id);         /* nEntriesInUse */
453
         avio_skip(pb, 16*avi->riff_id);
454 455 456
         avio_wl64(pb, ix);                   /* qwOffset */
         avio_wl32(pb, pos - ix);             /* dwSize */
         avio_wl32(pb, avist->indexes.entry); /* dwDuration */
457

458
         avio_seek(pb, pos, SEEK_SET);
459 460 461 462 463
    }
    return 0;
}

static int avi_write_idx1(AVFormatContext *s)
464
{
465
    AVIOContext *pb = s->pb;
466
    AVIContext *avi = s->priv_data;
467
    int64_t idx_chunk;
468
    int i;
469
    char tag[5];
470

471
    if (pb->seekable) {
472
        AVIStream *avist;
473 474 475
        AVIIentry* ie = 0, *tie;
        int empty, stream_id = -1;

476
        idx_chunk = ff_start_tag(pb, "idx1");
477 478 479 480 481
        for(i=0; i<s->nb_streams; i++){
            avist= s->streams[i]->priv_data;
            avist->entry=0;
        }

482 483 484
        do {
            empty = 1;
            for (i=0; i<s->nb_streams; i++) {
485 486
                avist= s->streams[i]->priv_data;
                 if (avist->indexes.entry <= avist->entry)
487 488
                     continue;

489
                 tie = avi_get_ientry(&avist->indexes, avist->entry);
490 491 492 493 494 495 496
                 if (empty || tie->pos < ie->pos) {
                     ie = tie;
                     stream_id = i;
                 }
                 empty = 0;
            }
            if (!empty) {
497
                avist= s->streams[stream_id]->priv_data;
498
                avi_stream2fourcc(tag, stream_id,
499
                                  s->streams[stream_id]->codec->codec_type);
500
                ffio_wfourcc(pb, tag);
501 502 503
                avio_wl32(pb, ie->flags);
                avio_wl32(pb, ie->pos);
                avio_wl32(pb, ie->len);
504
                avist->entry++;
505 506
            }
        } while (!empty);
507
        ff_end_tag(pb, idx_chunk);
508

509
        avi_write_counters(s, avi->riff_id);
510 511 512 513
    }
    return 0;
}

514
static int avi_write_packet(AVFormatContext *s, AVPacket *pkt)
Fabrice Bellard's avatar
Fabrice Bellard committed
515 516
{
    AVIContext *avi = s->priv_data;
517
    AVIOContext *pb = s->pb;
Fabrice Bellard's avatar
Fabrice Bellard committed
518
    unsigned char tag[5];
519 520
    unsigned int flags=0;
    const int stream_index= pkt->stream_index;
521
    AVIStream *avist= s->streams[stream_index]->priv_data;
522
    AVCodecContext *enc= s->streams[stream_index]->codec;
523
    int size= pkt->size;
524

525
    av_dlog(s, "dts:%s packet_count:%d stream_index:%d\n", av_ts2str(pkt->dts), avist->packet_count, stream_index);
526
    while(enc->block_align==0 && pkt->dts != AV_NOPTS_VALUE && pkt->dts > avist->packet_count && enc->codec_id != AV_CODEC_ID_XSUB && avist->packet_count){
527 528
        AVPacket empty_packet;

529
        if(pkt->dts - avist->packet_count > 60000){
530
            av_log(s, AV_LOG_ERROR, "Too large number of skipped frames %"PRId64" > 60000\n", pkt->dts - avist->packet_count);
531 532 533
            return AVERROR(EINVAL);
        }

534 535 536 537 538
        av_init_packet(&empty_packet);
        empty_packet.size= 0;
        empty_packet.data= NULL;
        empty_packet.stream_index= stream_index;
        avi_write_packet(s, &empty_packet);
539
        av_dlog(s, "dup dts:%s packet_count:%d\n", av_ts2str(pkt->dts), avist->packet_count);
540
    }
541
    avist->packet_count++;
542

543
    // Make sure to put an OpenDML chunk when the file size exceeds the limits
544
    if (pb->seekable &&
545
        (avio_tell(pb) - avi->riff_start > AVI_MAX_RIFF_SIZE)) {
546

547
        avi_write_ix(s);
548
        ff_end_tag(pb, avi->movi_list);
549

550 551
        if (avi->riff_id == 1)
            avi_write_idx1(s);
552

553
        ff_end_tag(pb, avi->riff_start);
554
        avi->movi_list = avi_start_new_riff(s, pb, "AVIX", "movi");
555
    }
556

557
    avi_stream2fourcc(tag, stream_index, enc->codec_type);
558
    if(pkt->flags&AV_PKT_FLAG_KEY)
559
        flags = 0x10;
560
    if (enc->codec_type == AVMEDIA_TYPE_AUDIO) {
561
       avist->audio_strm_length += size;
562
    }
Fabrice Bellard's avatar
Fabrice Bellard committed
563

564
    if (s->pb->seekable) {
565
        AVIIndex* idx = &avist->indexes;
566 567
        int cl = idx->entry / AVI_INDEX_CLUSTER_SIZE;
        int id = idx->entry % AVI_INDEX_CLUSTER_SIZE;
568
        if (idx->ents_allocated <= idx->entry) {
569
            idx->cluster = av_realloc_f(idx->cluster, sizeof(void*), cl+1);
570
            if (!idx->cluster)
571
                return AVERROR(ENOMEM);
572
            idx->cluster[cl] = av_malloc(AVI_INDEX_CLUSTER_SIZE*sizeof(AVIIentry));
573
            if (!idx->cluster[cl])
574
                return AVERROR(ENOMEM);
575 576
            idx->ents_allocated += AVI_INDEX_CLUSTER_SIZE;
        }
577

578
        idx->cluster[cl][id].flags = flags;
579
        idx->cluster[cl][id].pos = avio_tell(pb) - avi->movi_list;
580
        idx->cluster[cl][id].len = size;
581
        idx->entry++;
Fabrice Bellard's avatar
Fabrice Bellard committed
582
    }
583

584 585 586
    avio_write(pb, tag, 4);
    avio_wl32(pb, size);
    avio_write(pb, pkt->data, size);
Fabrice Bellard's avatar
Fabrice Bellard committed
587
    if (size & 1)
588
        avio_w8(pb, 0);
Fabrice Bellard's avatar
Fabrice Bellard committed
589 590 591 592 593 594 595

    return 0;
}

static int avi_write_trailer(AVFormatContext *s)
{
    AVIContext *avi = s->priv_data;
596
    AVIOContext *pb = s->pb;
597
    int res = 0;
598
    int i, j, n, nb_frames;
599
    int64_t file_size;
600

601
    if (pb->seekable){
602
        if (avi->riff_id == 1) {
603
            ff_end_tag(pb, avi->movi_list);
604
            res = avi_write_idx1(s);
605
            ff_end_tag(pb, avi->riff_start);
606 607
        } else {
            avi_write_ix(s);
608 609
            ff_end_tag(pb, avi->movi_list);
            ff_end_tag(pb, avi->riff_start);
610

611
            file_size = avio_tell(pb);
612
            avio_seek(pb, avi->odml_list - 8, SEEK_SET);
613
            ffio_wfourcc(pb, "LIST"); /* Making this AVI OpenDML one */
614
            avio_skip(pb, 16);
615 616 617

            for (n=nb_frames=0;n<s->nb_streams;n++) {
                AVCodecContext *stream = s->streams[n]->codec;
618 619
                AVIStream *avist= s->streams[n]->priv_data;

620
                if (stream->codec_type == AVMEDIA_TYPE_VIDEO) {
621 622
                    if (nb_frames < avist->packet_count)
                        nb_frames = avist->packet_count;
623
                } else {
624
                    if (stream->codec_id == AV_CODEC_ID_MP2 || stream->codec_id == AV_CODEC_ID_MP3) {
625
                        nb_frames += avist->packet_count;
626
                    }
627 628
                }
            }
629
            avio_wl32(pb, nb_frames);
630
            avio_seek(pb, file_size, SEEK_SET);
631

632 633
            avi_write_counters(s, avi->riff_id);
        }
634
    }
635

636 637 638 639 640 641
    for (i=0; i<s->nb_streams; i++) {
         AVIStream *avist= s->streams[i]->priv_data;
         for (j=0; j<avist->indexes.ents_allocated/AVI_INDEX_CLUSTER_SIZE; j++)
              av_free(avist->indexes.cluster[j]);
         av_freep(&avist->indexes.cluster);
         avist->indexes.ents_allocated = avist->indexes.entry = 0;
642
    }
643

644
    return res;
Fabrice Bellard's avatar
Fabrice Bellard committed
645 646
}

647
AVOutputFormat ff_avi_muxer = {
648
    .name              = "avi",
649
    .long_name         = NULL_IF_CONFIG_SMALL("AVI (Audio Video Interleaved)"),
650 651 652
    .mime_type         = "video/x-msvideo",
    .extensions        = "avi",
    .priv_data_size    = sizeof(AVIContext),
653 654
    .audio_codec       = CONFIG_LIBMP3LAME ? AV_CODEC_ID_MP3 : AV_CODEC_ID_AC3,
    .video_codec       = AV_CODEC_ID_MPEG4,
655 656 657
    .write_header      = avi_write_header,
    .write_packet      = avi_write_packet,
    .write_trailer     = avi_write_trailer,
658 659 660 661
    .codec_tag         = (const AVCodecTag* const []){
        ff_codec_bmp_tags, ff_codec_wav_tags, 0
    },
    .flags             = AVFMT_VARIABLE_FPS,
Fabrice Bellard's avatar
Fabrice Bellard committed
662
};