avidec.c 35.1 KB
Newer Older
Fabrice Bellard's avatar
Fabrice Bellard committed
1
/*
2
 * AVI demuxer
3
 * Copyright (c) 2001 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
 */
#include "avformat.h"
#include "avi.h"
23
#include "dv.h"
24
#include "riff.h"
Fabrice Bellard's avatar
Fabrice Bellard committed
25

26 27 28
#undef NDEBUG
#include <assert.h>

Fabrice Bellard's avatar
Fabrice Bellard committed
29
//#define DEBUG
Fabrice Bellard's avatar
Fabrice Bellard committed
30 31 32
//#define DEBUG_SEEK

typedef struct AVIStream {
33
    int64_t frame_offset; /* current frame (video) or byte (audio) counter
Fabrice Bellard's avatar
Fabrice Bellard committed
34
                         (used to compute the pts) */
35 36 37
    int remaining;
    int packet_size;

Fabrice Bellard's avatar
Fabrice Bellard committed
38
    int scale;
39
    int rate;
40
    int sample_size; /* size of one sample (or packet) (in the rate/scale sense) in bytes */
41

42
    int64_t cum_len; /* temporary storage (used during seek) */
43

44 45
    int prefix;                       ///< normally 'd'<<8 + 'c' or 'w'<<8 + 'b'
    int prefix_count;
Fabrice Bellard's avatar
Fabrice Bellard committed
46
} AVIStream;
Fabrice Bellard's avatar
Fabrice Bellard committed
47 48

typedef struct {
49 50
    int64_t  riff_end;
    int64_t  movi_end;
51
    int64_t  fsize;
Fabrice Bellard's avatar
Fabrice Bellard committed
52
    offset_t movi_list;
Fabrice Bellard's avatar
Fabrice Bellard committed
53
    int index_loaded;
54
    int is_odml;
55 56
    int non_interleaved;
    int stream_index;
57
    DVDemuxContext* dv_demux;
Fabrice Bellard's avatar
Fabrice Bellard committed
58 59
} AVIContext;

60 61 62 63
static const char avi_headers[][8] = {
    { 'R', 'I', 'F', 'F',    'A', 'V', 'I', ' ' },
    { 'R', 'I', 'F', 'F',    'A', 'V', 'I', 'X' },
    { 'R', 'I', 'F', 'F',    'A', 'V', 'I', 0x19},
64
    { 'O', 'N', '2', ' ',    'O', 'N', '2', 'f' },
65
    { 'R', 'I', 'F', 'F',    'A', 'M', 'V', ' ' },
66 67 68
    { 0 }
};

69
static int avi_load_index(AVFormatContext *s);
70
static int guess_ni_flag(AVFormatContext *s);
71

Fabrice Bellard's avatar
Fabrice Bellard committed
72
#ifdef DEBUG
73
static void print_tag(const char *str, unsigned int tag, int size)
Fabrice Bellard's avatar
Fabrice Bellard committed
74 75 76 77 78 79 80 81 82 83
{
    printf("%s: tag=%c%c%c%c size=0x%x\n",
           str, tag & 0xff,
           (tag >> 8) & 0xff,
           (tag >> 16) & 0xff,
           (tag >> 24) & 0xff,
           size);
}
#endif

84 85
static int get_riff(AVIContext *avi, ByteIOContext *pb)
{
86 87
    char header[8];
    int i;
88

89 90
    /* check RIFF header */
    get_buffer(pb, header, 4);
91 92
    avi->riff_end = get_le32(pb);   /* RIFF chunk size */
    avi->riff_end += url_ftell(pb); /* RIFF chunk end */
93 94 95 96 97 98
    get_buffer(pb, header+4, 4);

    for(i=0; avi_headers[i][0]; i++)
        if(!memcmp(header, avi_headers[i], 8))
            break;
    if(!avi_headers[i][0])
99
        return -1;
100

101 102 103
    if(header[7] == 0x19)
        av_log(NULL, AV_LOG_INFO, "file has been generated with a totally broken muxer\n");

104 105 106
    return 0;
}

107
static int read_braindead_odml_indx(AVFormatContext *s, int frame_num){
108
    AVIContext *avi = s->priv_data;
109
    ByteIOContext *pb = s->pb;
110 111 112 113 114 115 116 117 118 119
    int longs_pre_entry= get_le16(pb);
    int index_sub_type = get_byte(pb);
    int index_type     = get_byte(pb);
    int entries_in_use = get_le32(pb);
    int chunk_id       = get_le32(pb);
    int64_t base       = get_le64(pb);
    int stream_id= 10*((chunk_id&0xFF) - '0') + (((chunk_id>>8)&0xFF) - '0');
    AVStream *st;
    AVIStream *ast;
    int i;
120
    int64_t last_pos= -1;
121
    int64_t filesize= url_fsize(s->pb);
122

123
#ifdef DEBUG_SEEK
124
    av_log(s, AV_LOG_ERROR, "longs_pre_entry:%d index_type:%d entries_in_use:%d chunk_id:%X base:%16"PRIX64"\n",
125 126
        longs_pre_entry,index_type, entries_in_use, chunk_id, base);
#endif
127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142

    if(stream_id > s->nb_streams || stream_id < 0)
        return -1;
    st= s->streams[stream_id];
    ast = st->priv_data;

    if(index_sub_type)
        return -1;

    get_le32(pb);

    if(index_type && longs_pre_entry != 2)
        return -1;
    if(index_type>1)
        return -1;

143 144 145 146 147 148 149 150
    if(filesize > 0 && base >= filesize){
        av_log(s, AV_LOG_ERROR, "ODML index invalid\n");
        if(base>>32 == (base & 0xFFFFFFFF) && (base & 0xFFFFFFFF) < filesize && filesize <= 0xFFFFFFFF)
            base &= 0xFFFFFFFF;
        else
            return -1;
    }

151 152
    for(i=0; i<entries_in_use; i++){
        if(index_type){
153
            int64_t pos= get_le32(pb) + base - 8;
154
            int len    = get_le32(pb);
155
            int key= len >= 0;
156 157
            len &= 0x7FFFFFFF;

158
#ifdef DEBUG_SEEK
159
            av_log(s, AV_LOG_ERROR, "pos:%"PRId64", len:%X\n", pos, len);
160
#endif
161 162 163
            if(last_pos == pos || pos == base - 8)
                avi->non_interleaved= 1;
            else
164
                av_add_index_entry(st, pos, ast->cum_len / FFMAX(1, ast->sample_size), len, 0, key ? AVINDEX_KEYFRAME : 0);
165

166
            if(ast->sample_size)
167
                ast->cum_len += len;
168 169
            else
                ast->cum_len ++;
170
            last_pos= pos;
171
        }else{
Måns Rullgård's avatar
Måns Rullgård committed
172 173 174 175 176 177
            int64_t offset, pos;
            int duration;
            offset = get_le64(pb);
            get_le32(pb);       /* size */
            duration = get_le32(pb);
            pos = url_ftell(pb);
178 179 180 181 182 183 184 185

            url_fseek(pb, offset+8, SEEK_SET);
            read_braindead_odml_indx(s, frame_num);
            frame_num += duration;

            url_fseek(pb, pos, SEEK_SET);
        }
    }
186
    avi->index_loaded=1;
187 188 189
    return 0;
}

190
static void clean_index(AVFormatContext *s){
191 192
    int i;
    int64_t j;
193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215

    for(i=0; i<s->nb_streams; i++){
        AVStream *st = s->streams[i];
        AVIStream *ast = st->priv_data;
        int n= st->nb_index_entries;
        int max= ast->sample_size;
        int64_t pos, size, ts;

        if(n != 1 || ast->sample_size==0)
            continue;

        while(max < 1024) max+=max;

        pos= st->index_entries[0].pos;
        size= st->index_entries[0].size;
        ts= st->index_entries[0].timestamp;

        for(j=0; j<size; j+=max){
            av_add_index_entry(st, pos+j, ts + j/ast->sample_size, FFMIN(max, size-j), 0, AVINDEX_KEYFRAME);
        }
    }
}

216 217 218 219 220 221 222 223 224
static int avi_read_tag(ByteIOContext *pb, char *buf, int maxlen,  unsigned int size)
{
    offset_t i = url_ftell(pb);
    size += (size & 1);
    get_strz(pb, buf, maxlen);
    url_fseek(pb, i+size, SEEK_SET);
    return 0;
}

225
static int avi_read_header(AVFormatContext *s, AVFormatParameters *ap)
Fabrice Bellard's avatar
Fabrice Bellard committed
226
{
Fabrice Bellard's avatar
Fabrice Bellard committed
227
    AVIContext *avi = s->priv_data;
228
    ByteIOContext *pb = s->pb;
229
    uint32_t tag, tag1, handler;
230
    int codec_type, stream_index, frame_period, bit_rate;
231
    unsigned int size, nb_frames;
232
    int i;
Fabrice Bellard's avatar
Fabrice Bellard committed
233
    AVStream *st;
Måns Rullgård's avatar
Måns Rullgård committed
234
    AVIStream *ast = NULL;
235
    char str_track[4];
236 237
    int avih_width=0, avih_height=0;
    int amv_file_format=0;
Fabrice Bellard's avatar
Fabrice Bellard committed
238

239
    avi->stream_index= -1;
240

241
    if (get_riff(avi, pb) < 0)
Fabrice Bellard's avatar
Fabrice Bellard committed
242
        return -1;
243

244 245 246 247
    avi->fsize = url_fsize(pb);
    if(avi->fsize<=0)
        avi->fsize= avi->riff_end;

Fabrice Bellard's avatar
Fabrice Bellard committed
248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268
    /* first list tag */
    stream_index = -1;
    codec_type = -1;
    frame_period = 0;
    for(;;) {
        if (url_feof(pb))
            goto fail;
        tag = get_le32(pb);
        size = get_le32(pb);
#ifdef DEBUG
        print_tag("tag", tag, size);
#endif

        switch(tag) {
        case MKTAG('L', 'I', 'S', 'T'):
            /* ignored, except when start of video packets */
            tag1 = get_le32(pb);
#ifdef DEBUG
            print_tag("list", tag1, 0);
#endif
            if (tag1 == MKTAG('m', 'o', 'v', 'i')) {
Fabrice Bellard's avatar
Fabrice Bellard committed
269
                avi->movi_list = url_ftell(pb) - 4;
270
                if(size) avi->movi_end = avi->movi_list + size + (size & 1);
271
                else     avi->movi_end = url_fsize(pb);
Fabrice Bellard's avatar
Fabrice Bellard committed
272
#ifdef DEBUG
273
                printf("movi end=%"PRIx64"\n", avi->movi_end);
Fabrice Bellard's avatar
Fabrice Bellard committed
274 275 276 277
#endif
                goto end_of_header;
            }
            break;
278
        case MKTAG('d', 'm', 'l', 'h'):
279 280 281
            avi->is_odml = 1;
            url_fskip(pb, size + (size & 1));
            break;
282 283
        case MKTAG('a', 'm', 'v', 'h'):
            amv_file_format=1;
Fabrice Bellard's avatar
Fabrice Bellard committed
284
        case MKTAG('a', 'v', 'i', 'h'):
285
            /* avi header */
286
            /* using frame_period is bad idea */
Fabrice Bellard's avatar
Fabrice Bellard committed
287 288
            frame_period = get_le32(pb);
            bit_rate = get_le32(pb) * 8;
289 290 291 292
            get_le32(pb);
            avi->non_interleaved |= get_le32(pb) & AVIF_MUSTUSEINDEX;

            url_fskip(pb, 2 * 4);
293
            get_le32(pb);
294 295 296
            get_le32(pb);
            avih_width=get_le32(pb);
            avih_height=get_le32(pb);
297

298
            url_fskip(pb, size - 10 * 4);
299 300 301 302 303 304 305 306 307 308 309 310 311
            break;
        case MKTAG('s', 't', 'r', 'h'):
            /* stream header */

            tag1 = get_le32(pb);
            handler = get_le32(pb); /* codec tag */

            if(tag1 == MKTAG('p', 'a', 'd', 's')){
                url_fskip(pb, size - 8);
                break;
            }else{
                stream_index++;
                st = av_new_stream(s, stream_index);
Fabrice Bellard's avatar
Fabrice Bellard committed
312 313
                if (!st)
                    goto fail;
314

Fabrice Bellard's avatar
Fabrice Bellard committed
315 316 317 318
                ast = av_mallocz(sizeof(AVIStream));
                if (!ast)
                    goto fail;
                st->priv_data = ast;
319
            }
320 321
            if(amv_file_format)
                tag1 = stream_index ? MKTAG('a','u','d','s') : MKTAG('v','i','d','s');
322

323
#ifdef DEBUG
324
            print_tag("strh", tag1, -1);
325
#endif
326
            if(tag1 == MKTAG('i', 'a', 'v', 's') || tag1 == MKTAG('i', 'v', 'a', 's')){
327 328
                int64_t dv_dur;

329
                /*
330 331 332 333 334 335 336 337 338 339 340 341 342 343 344
                 * After some consideration -- I don't think we
                 * have to support anything but DV in a type1 AVIs.
                 */
                if (s->nb_streams != 1)
                    goto fail;

                if (handler != MKTAG('d', 'v', 's', 'd') &&
                    handler != MKTAG('d', 'v', 'h', 'd') &&
                    handler != MKTAG('d', 'v', 's', 'l'))
                   goto fail;

                ast = s->streams[0]->priv_data;
                av_freep(&s->streams[0]->codec->extradata);
                av_freep(&s->streams[0]);
                s->nb_streams = 0;
345
                if (ENABLE_DV_DEMUXER) {
346 347 348
                    avi->dv_demux = dv_init_demux(s);
                    if (!avi->dv_demux)
                        goto fail;
349
                }
350 351 352 353
                s->streams[0]->priv_data = ast;
                url_fskip(pb, 3 * 4);
                ast->scale = get_le32(pb);
                ast->rate = get_le32(pb);
354 355 356 357 358 359 360 361 362 363 364 365
                url_fskip(pb, 4);  /* start time */

                dv_dur = get_le32(pb);
                if (ast->scale > 0 && ast->rate > 0 && dv_dur > 0) {
                    dv_dur *= AV_TIME_BASE;
                    s->duration = av_rescale(dv_dur, ast->scale, ast->rate);
                }
                /*
                 * else, leave duration alone; timing estimation in utils.c
                 *      will make a guess based on bit rate.
                 */

366
                stream_index = s->nb_streams - 1;
367
                url_fskip(pb, size - 9*4);
368 369
                break;
            }
370

371
            assert(stream_index < s->nb_streams);
372
            st->codec->stream_codec_tag= handler;
373

374 375 376 377 378 379 380 381 382 383 384 385 386 387 388
            get_le32(pb); /* flags */
            get_le16(pb); /* priority */
            get_le16(pb); /* language */
            get_le32(pb); /* initial frame */
            ast->scale = get_le32(pb);
            ast->rate = get_le32(pb);
            if(ast->scale && ast->rate){
            }else if(frame_period){
                ast->rate = 1000000;
                ast->scale = frame_period;
            }else{
                ast->rate = 25;
                ast->scale = 1;
            }
            av_set_pts_info(st, 64, ast->scale, ast->rate);
389

Michael Niedermayer's avatar
Michael Niedermayer committed
390
            ast->cum_len=get_le32(pb); /* start */
391
            nb_frames = get_le32(pb);
392

393
            st->start_time = 0;
394
            st->duration = nb_frames;
395 396 397
            get_le32(pb); /* buffer size */
            get_le32(pb); /* quality */
            ast->sample_size = get_le32(pb); /* sample ssize */
398
            ast->cum_len *= FFMAX(1, ast->sample_size);
399
//            av_log(NULL, AV_LOG_DEBUG, "%d %d %d %d\n", ast->rate, ast->scale, ast->start, ast->sample_size);
400

401
            switch(tag1) {
402
            case MKTAG('v', 'i', 'd', 's'):
403
                codec_type = CODEC_TYPE_VIDEO;
404

405 406 407 408
                ast->sample_size = 0;
                break;
            case MKTAG('a', 'u', 'd', 's'):
                codec_type = CODEC_TYPE_AUDIO;
409
                break;
410
            case MKTAG('t', 'x', 't', 's'):
411
                //FIXME
412 413
                codec_type = CODEC_TYPE_DATA; //CODEC_TYPE_SUB ?  FIXME
                break;
414
            default:
Michael Niedermayer's avatar
Michael Niedermayer committed
415
                av_log(s, AV_LOG_ERROR, "unknown stream type %X\n", tag1);
416
                goto fail;
Fabrice Bellard's avatar
Fabrice Bellard committed
417
            }
418
            ast->frame_offset= ast->cum_len;
419
            url_fskip(pb, size - 12 * 4);
Fabrice Bellard's avatar
Fabrice Bellard committed
420 421 422
            break;
        case MKTAG('s', 't', 'r', 'f'):
            /* stream header */
423
            if (stream_index >= (unsigned)s->nb_streams || avi->dv_demux) {
Fabrice Bellard's avatar
Fabrice Bellard committed
424 425 426 427 428
                url_fskip(pb, size);
            } else {
                st = s->streams[stream_index];
                switch(codec_type) {
                case CODEC_TYPE_VIDEO:
429 430 431 432 433 434 435 436
                    if(amv_file_format){
                        st->codec->width=avih_width;
                        st->codec->height=avih_height;
                        st->codec->codec_type = CODEC_TYPE_VIDEO;
                        st->codec->codec_id = CODEC_ID_AMV;
                        url_fskip(pb, size);
                        break;
                    }
Fabrice Bellard's avatar
Fabrice Bellard committed
437
                    get_le32(pb); /* size */
438 439
                    st->codec->width = get_le32(pb);
                    st->codec->height = get_le32(pb);
Fabrice Bellard's avatar
Fabrice Bellard committed
440
                    get_le16(pb); /* panes */
441
                    st->codec->bits_per_sample= get_le16(pb); /* depth */
Fabrice Bellard's avatar
Fabrice Bellard committed
442
                    tag1 = get_le32(pb);
443 444 445 446 447 448
                    get_le32(pb); /* ImageSize */
                    get_le32(pb); /* XPelsPerMeter */
                    get_le32(pb); /* YPelsPerMeter */
                    get_le32(pb); /* ClrUsed */
                    get_le32(pb); /* ClrImportant */

449 450 451 452 453 454 455
                    if (tag1 == MKTAG('D', 'X', 'S', 'B')) {
                        st->codec->codec_type = CODEC_TYPE_SUBTITLE;
                        st->codec->codec_tag = tag1;
                        st->codec->codec_id = CODEC_ID_XSUB;
                        break;
                    }

456 457 458 459 460
                    if(size > 10*4 && size<(1<<30)){
                        st->codec->extradata_size= size - 10*4;
                        st->codec->extradata= av_malloc(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
                        get_buffer(pb, st->codec->extradata, st->codec->extradata_size);
                    }
461

462
                    if(st->codec->extradata_size & 1) //FIXME check if the encoder really did this correctly
463
                        get_byte(pb);
464

465 466 467
                    /* Extract palette from extradata if bpp <= 8 */
                    /* This code assumes that extradata contains only palette */
                    /* This is true for all paletted codecs implemented in ffmpeg */
468 469
                    if (st->codec->extradata_size && (st->codec->bits_per_sample <= 8)) {
                        st->codec->palctrl = av_mallocz(sizeof(AVPaletteControl));
470
#ifdef WORDS_BIGENDIAN
471 472
                        for (i = 0; i < FFMIN(st->codec->extradata_size, AVPALETTE_SIZE)/4; i++)
                            st->codec->palctrl->palette[i] = bswap_32(((uint32_t*)st->codec->extradata)[i]);
473
#else
474 475
                        memcpy(st->codec->palctrl->palette, st->codec->extradata,
                               FFMIN(st->codec->extradata_size, AVPALETTE_SIZE));
476
#endif
477
                        st->codec->palctrl->palette_changed = 1;
478 479
                    }

Fabrice Bellard's avatar
Fabrice Bellard committed
480 481 482
#ifdef DEBUG
                    print_tag("video", tag1, 0);
#endif
483 484 485
                    st->codec->codec_type = CODEC_TYPE_VIDEO;
                    st->codec->codec_tag = tag1;
                    st->codec->codec_id = codec_get_id(codec_bmp_tags, tag1);
486
                    st->need_parsing = AVSTREAM_PARSE_HEADERS; // this is needed to get the pict type which is needed for generating correct pts
487
//                    url_fskip(pb, size - 5 * 4);
Fabrice Bellard's avatar
Fabrice Bellard committed
488
                    break;
489
                case CODEC_TYPE_AUDIO:
490
                    get_wav_header(pb, st->codec, size);
491 492
                    if(ast->sample_size && st->codec->block_align && ast->sample_size % st->codec->block_align)
                        av_log(s, AV_LOG_DEBUG, "invalid sample size or block align detected\n");
493 494
                    if (size%2) /* 2-aligned (fix for Stargate SG-1 - 3x18 - Shades of Grey.avi) */
                        url_fskip(pb, 1);
Diego Biurrun's avatar
Diego Biurrun committed
495
                    /* Force parsing as several audio frames can be in
496 497
                     * one packet and timestamps refer to packet start*/
                    st->need_parsing = AVSTREAM_PARSE_TIMESTAMPS;
498 499
                    /* ADTS header is in extradata, AAC without header must be stored as exact frames, parser not needed and it will fail */
                    if (st->codec->codec_id == CODEC_ID_AAC && st->codec->extradata_size)
500
                        st->need_parsing = AVSTREAM_PARSE_NONE;
501 502 503 504 505 506
                    /* AVI files with Xan DPCM audio (wrongly) declare PCM
                     * audio in the header but have Axan as stream_code_tag. */
                    if (st->codec->stream_codec_tag == ff_get_fourcc("Axan")){
                        st->codec->codec_id  = CODEC_ID_XAN_DPCM;
                        st->codec->codec_tag = 0;
                    }
507 508
                    if (amv_file_format)
                        st->codec->codec_id  = CODEC_ID_ADPCM_IMA_AMV;
Fabrice Bellard's avatar
Fabrice Bellard committed
509 510
                    break;
                default:
511 512 513
                    st->codec->codec_type = CODEC_TYPE_DATA;
                    st->codec->codec_id= CODEC_ID_NONE;
                    st->codec->codec_tag= 0;
Fabrice Bellard's avatar
Fabrice Bellard committed
514 515 516 517 518
                    url_fskip(pb, size);
                    break;
                }
            }
            break;
519 520
        case MKTAG('i', 'n', 'd', 'x'):
            i= url_ftell(pb);
521
            if(!url_is_streamed(pb) && !(s->flags & AVFMT_FLAG_IGNIDX)){
522 523
                read_braindead_odml_indx(s, 0);
            }
524 525
            url_fseek(pb, i+size, SEEK_SET);
            break;
526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550
        case MKTAG('v', 'p', 'r', 'p'):
            if(stream_index < (unsigned)s->nb_streams && size > 9*4){
                AVRational active, active_aspect;

                st = s->streams[stream_index];
                get_le32(pb);
                get_le32(pb);
                get_le32(pb);
                get_le32(pb);
                get_le32(pb);

                active_aspect.num= get_le16(pb);
                active_aspect.den= get_le16(pb);
                active.num       = get_le32(pb);
                active.den       = get_le32(pb);
                get_le32(pb); //nbFieldsPerFrame

                if(active_aspect.num && active_aspect.den && active.num && active.den){
                    st->codec->sample_aspect_ratio= av_div_q(active_aspect, active);
//av_log(s, AV_LOG_ERROR, "vprp %d/%d %d/%d\n", active_aspect.num, active_aspect.den, active.num, active.den);
                }
                size -= 9*4;
            }
            url_fseek(pb, size, SEEK_CUR);
            break;
551 552 553 554 555 556 557 558 559 560 561 562 563 564 565
        case MKTAG('I', 'N', 'A', 'M'):
            avi_read_tag(pb, s->title, sizeof(s->title), size);
            break;
        case MKTAG('I', 'A', 'R', 'T'):
            avi_read_tag(pb, s->author, sizeof(s->author), size);
            break;
        case MKTAG('I', 'C', 'O', 'P'):
            avi_read_tag(pb, s->copyright, sizeof(s->copyright), size);
            break;
        case MKTAG('I', 'C', 'M', 'T'):
            avi_read_tag(pb, s->comment, sizeof(s->comment), size);
            break;
        case MKTAG('I', 'G', 'N', 'R'):
            avi_read_tag(pb, s->genre, sizeof(s->genre), size);
            break;
Panagiotis Issaris's avatar
Panagiotis Issaris committed
566 567 568
        case MKTAG('I', 'P', 'R', 'D'):
            avi_read_tag(pb, s->album, sizeof(s->album), size);
            break;
569 570 571 572
        case MKTAG('I', 'P', 'R', 'T'):
            avi_read_tag(pb, str_track, sizeof(str_track), size);
            sscanf(str_track, "%d", &s->track);
            break;
Fabrice Bellard's avatar
Fabrice Bellard committed
573
        default:
574 575 576 577 578 579 580
            if(size > 1000000){
                av_log(s, AV_LOG_ERROR, "well something went wrong during header parsing, "
                                        "ill ignore it and try to continue anyway\n");
                avi->movi_list = url_ftell(pb) - 4;
                avi->movi_end  = url_fsize(pb);
                goto end_of_header;
            }
Fabrice Bellard's avatar
Fabrice Bellard committed
581 582 583 584 585 586 587 588 589 590 591
            /* skip tag */
            size += (size & 1);
            url_fskip(pb, size);
            break;
        }
    }
 end_of_header:
    /* check stream number */
    if (stream_index != s->nb_streams - 1) {
    fail:
        for(i=0;i<s->nb_streams;i++) {
592
            av_freep(&s->streams[i]->codec->extradata);
Michael Niedermayer's avatar
Michael Niedermayer committed
593
            av_freep(&s->streams[i]);
Fabrice Bellard's avatar
Fabrice Bellard committed
594 595 596
        }
        return -1;
    }
597

598
    if(!avi->index_loaded && !url_is_streamed(pb))
599
        avi_load_index(s);
600
    avi->index_loaded = 1;
601
    avi->non_interleaved |= guess_ni_flag(s);
602 603
    if(avi->non_interleaved)
        clean_index(s);
604

Fabrice Bellard's avatar
Fabrice Bellard committed
605 606 607
    return 0;
}

608
static int avi_read_packet(AVFormatContext *s, AVPacket *pkt)
Fabrice Bellard's avatar
Fabrice Bellard committed
609 610
{
    AVIContext *avi = s->priv_data;
611
    ByteIOContext *pb = s->pb;
612
    int n, d[8], size;
613
    offset_t i, sync;
614
    void* dstr;
615

616
    if (ENABLE_DV_DEMUXER && avi->dv_demux) {
617
        size = dv_get_packet(avi->dv_demux, pkt);
618 619
        if (size >= 0)
            return size;
620
    }
621

622
    if(avi->non_interleaved){
623
        int best_stream_index = 0;
624 625 626 627
        AVStream *best_st= NULL;
        AVIStream *best_ast;
        int64_t best_ts= INT64_MAX;
        int i;
628

629 630 631 632 633 634 635 636 637
        for(i=0; i<s->nb_streams; i++){
            AVStream *st = s->streams[i];
            AVIStream *ast = st->priv_data;
            int64_t ts= ast->frame_offset;

            if(ast->sample_size)
                ts /= ast->sample_size;
            ts= av_rescale(ts, AV_TIME_BASE * (int64_t)st->time_base.num, st->time_base.den);

638
//            av_log(NULL, AV_LOG_DEBUG, "%"PRId64" %d/%d %"PRId64"\n", ts, st->time_base.num, st->time_base.den, ast->frame_offset);
639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654
            if(ts < best_ts){
                best_ts= ts;
                best_st= st;
                best_stream_index= i;
            }
        }
        best_ast = best_st->priv_data;
        best_ts= av_rescale(best_ts, best_st->time_base.den, AV_TIME_BASE * (int64_t)best_st->time_base.num); //FIXME a little ugly
        if(best_ast->remaining)
            i= av_index_search_timestamp(best_st, best_ts, AVSEEK_FLAG_ANY | AVSEEK_FLAG_BACKWARD);
        else
            i= av_index_search_timestamp(best_st, best_ts, AVSEEK_FLAG_ANY);

//        av_log(NULL, AV_LOG_DEBUG, "%d\n", i);
        if(i>=0){
            int64_t pos= best_st->index_entries[i].pos;
655
            pos += best_ast->packet_size - best_ast->remaining;
656
            url_fseek(s->pb, pos + 8, SEEK_SET);
657
//        av_log(NULL, AV_LOG_DEBUG, "pos=%"PRId64"\n", pos);
658

659 660
            assert(best_ast->remaining <= best_ast->packet_size);

661 662
            avi->stream_index= best_stream_index;
            if(!best_ast->remaining)
663
                best_ast->packet_size=
664
                best_ast->remaining= best_st->index_entries[i].size;
665 666
        }
    }
667

668
resync:
669 670 671 672
    if(avi->stream_index >= 0){
        AVStream *st= s->streams[ avi->stream_index ];
        AVIStream *ast= st->priv_data;
        int size;
673

674
        if(ast->sample_size <= 1) // minorityreport.AVI block_align=1024 sample_size=1 IMA-ADPCM
675
            size= INT_MAX;
676
        else if(ast->sample_size < 32)
677 678 679 680 681 682
            size= 64*ast->sample_size;
        else
            size= ast->sample_size;

        if(size > ast->remaining)
            size= ast->remaining;
Michael Niedermayer's avatar
Michael Niedermayer committed
683
        av_get_packet(pb, pkt, size);
684

685
        if (ENABLE_DV_DEMUXER && avi->dv_demux) {
686 687 688 689 690 691 692 693 694 695 696
            dstr = pkt->destruct;
            size = dv_produce_packet(avi->dv_demux, pkt,
                                    pkt->data, pkt->size);
            pkt->destruct = dstr;
            pkt->flags |= PKT_FLAG_KEY;
        } else {
            /* XXX: how to handle B frames in avi ? */
            pkt->dts = ast->frame_offset;
//                pkt->dts += ast->start;
            if(ast->sample_size)
                pkt->dts /= ast->sample_size;
697
//av_log(NULL, AV_LOG_DEBUG, "dts:%"PRId64" offset:%"PRId64" %d/%d smpl_siz:%d base:%d st:%d size:%d\n", pkt->dts, ast->frame_offset, ast->scale, ast->rate, ast->sample_size, AV_TIME_BASE, avi->stream_index, size);
698 699
            pkt->stream_index = avi->stream_index;

700
            if (st->codec->codec_type == CODEC_TYPE_VIDEO) {
Michael Niedermayer's avatar
Michael Niedermayer committed
701 702
                AVIndexEntry *e;
                int index;
703
                assert(st->index_entries);
704

Michael Niedermayer's avatar
Michael Niedermayer committed
705 706
                index= av_index_search_timestamp(st, pkt->dts, 0);
                e= &st->index_entries[index];
707

Michael Niedermayer's avatar
Michael Niedermayer committed
708 709 710 711
                if(index >= 0 && e->timestamp == ast->frame_offset){
                    if (e->flags & AVINDEX_KEYFRAME)
                        pkt->flags |= PKT_FLAG_KEY;
                }
712
            } else {
713
                pkt->flags |= PKT_FLAG_KEY;
714 715 716 717 718 719 720 721 722 723 724 725 726 727 728
            }
            if(ast->sample_size)
                ast->frame_offset += pkt->size;
            else
                ast->frame_offset++;
        }
        ast->remaining -= size;
        if(!ast->remaining){
            avi->stream_index= -1;
            ast->packet_size= 0;
        }

        return size;
    }

729 730
    memset(d, -1, sizeof(int)*8);
    for(i=sync=url_ftell(pb); !url_feof(pb); i++) {
731
        int j;
732

733 734 735
        for(j=0; j<7; j++)
            d[j]= d[j+1];
        d[7]= get_byte(pb);
736

737
        size= d[4] + (d[5]<<8) + (d[6]<<16) + (d[7]<<24);
738

739
        if(    d[2] >= '0' && d[2] <= '9'
740 741 742 743
            && d[3] >= '0' && d[3] <= '9'){
            n= (d[2] - '0') * 10 + (d[3] - '0');
        }else{
            n= 100; //invalid stream id
744
        }
745
//av_log(NULL, AV_LOG_DEBUG, "%X %X %X %X %X %X %X %X %"PRId64" %d %d\n", d[0], d[1], d[2], d[3], d[4], d[5], d[6], d[7], i, size, n);
746
        if(i + size > avi->fsize || d[0]<0)
747
            continue;
748

749 750
        //parse ix##
        if(  (d[0] == 'i' && d[1] == 'x' && n < s->nb_streams)
751
        //parse JUNK
752 753
           ||(d[0] == 'J' && d[1] == 'U' && d[2] == 'N' && d[3] == 'K')
           ||(d[0] == 'i' && d[1] == 'd' && d[2] == 'x' && d[3] == '1')){
754
            url_fskip(pb, size);
755
//av_log(NULL, AV_LOG_DEBUG, "SKIP\n");
756
            goto resync;
757
        }
758

759
        if(    d[0] >= '0' && d[0] <= '9'
760 761 762 763 764
            && d[1] >= '0' && d[1] <= '9'){
            n= (d[0] - '0') * 10 + (d[1] - '0');
        }else{
            n= 100; //invalid stream id
        }
765

766
        //parse ##dc/##wb
767
        if(n < s->nb_streams){
768 769 770 771
          AVStream *st;
          AVIStream *ast;
          st = s->streams[n];
          ast = st->priv_data;
772

773 774 775 776 777
          if(   (st->discard >= AVDISCARD_DEFAULT && size==0)
             /*|| (st->discard >= AVDISCARD_NONKEY && !(pkt->flags & PKT_FLAG_KEY))*/ //FIXME needs a little reordering
             || st->discard >= AVDISCARD_ALL){
                if(ast->sample_size) ast->frame_offset += pkt->size;
                else                 ast->frame_offset++;
778 779 780
                url_fskip(pb, size);
                goto resync;
          }
781

782
          if(   ((ast->prefix_count<5 || sync+9 > i) && d[2]<128 && d[3]<128) ||
783
                d[2]*256+d[3] == ast->prefix /*||
784
                (d[2] == 'd' && d[3] == 'c') ||
785
                (d[2] == 'w' && d[3] == 'b')*/) {
786 787 788 789 790 791 792 793 794

//av_log(NULL, AV_LOG_DEBUG, "OK\n");
            if(d[2]*256+d[3] == ast->prefix)
                ast->prefix_count++;
            else{
                ast->prefix= d[2]*256+d[3];
                ast->prefix_count= 0;
            }

795 796 797
            avi->stream_index= n;
            ast->packet_size= size + 8;
            ast->remaining= size;
798 799 800 801 802 803 804

            {
                uint64_t pos= url_ftell(pb) - 8;
                if(!st->index_entries || !st->nb_index_entries || st->index_entries[st->nb_index_entries - 1].pos < pos){
                    av_add_index_entry(st, pos, ast->frame_offset / FFMAX(1, ast->sample_size), size, 0, AVINDEX_KEYFRAME);
                }
            }
805
            goto resync;
806
          }
807
        }
808 809 810 811
        /* palette changed chunk */
        if (   d[0] >= '0' && d[0] <= '9'
            && d[1] >= '0' && d[1] <= '9'
            && ((d[2] == 'p' && d[3] == 'c'))
812
            && n < s->nb_streams && i + size <= avi->fsize) {
813 814 815 816 817 818 819 820

            AVStream *st;
            int first, clr, flags, k, p;

            st = s->streams[n];

            first = get_byte(pb);
            clr = get_byte(pb);
821 822
            if(!clr) /* all 256 colors used */
                clr = 256;
823 824 825 826 827 828 829 830
            flags = get_le16(pb);
            p = 4;
            for (k = first; k < clr + first; k++) {
                int r, g, b;
                r = get_byte(pb);
                g = get_byte(pb);
                b = get_byte(pb);
                    get_byte(pb);
831
                st->codec->palctrl->palette[k] = b + (g << 8) + (r << 16);
832
            }
833
            st->codec->palctrl->palette_changed = 1;
834
            goto resync;
835 836
        }

837
    }
838

839
    return -1;
Fabrice Bellard's avatar
Fabrice Bellard committed
840 841
}

Fabrice Bellard's avatar
Fabrice Bellard committed
842 843 844 845
/* XXX: we make the implicit supposition that the position are sorted
   for each stream */
static int avi_read_idx1(AVFormatContext *s, int size)
{
846
    AVIContext *avi = s->priv_data;
847
    ByteIOContext *pb = s->pb;
Fabrice Bellard's avatar
Fabrice Bellard committed
848 849 850 851
    int nb_index_entries, i;
    AVStream *st;
    AVIStream *ast;
    unsigned int index, tag, flags, pos, len;
852
    unsigned last_pos= -1;
853

Fabrice Bellard's avatar
Fabrice Bellard committed
854 855 856 857 858 859 860 861 862 863
    nb_index_entries = size / 16;
    if (nb_index_entries <= 0)
        return -1;

    /* read the entries and sort them in each stream component */
    for(i = 0; i < nb_index_entries; i++) {
        tag = get_le32(pb);
        flags = get_le32(pb);
        pos = get_le32(pb);
        len = get_le32(pb);
864
#if defined(DEBUG_SEEK)
865
        av_log(NULL, AV_LOG_DEBUG, "%d: tag=0x%x flags=0x%x pos=0x%x len=%d/",
Fabrice Bellard's avatar
Fabrice Bellard committed
866 867
               i, tag, flags, pos, len);
#endif
868 869
        if(i==0 && pos > avi->movi_list)
            avi->movi_list= 0; //FIXME better check
870
        pos += avi->movi_list;
871

Fabrice Bellard's avatar
Fabrice Bellard committed
872 873 874 875 876 877
        index = ((tag & 0xff) - '0') * 10;
        index += ((tag >> 8) & 0xff) - '0';
        if (index >= s->nb_streams)
            continue;
        st = s->streams[index];
        ast = st->priv_data;
878

879
#if defined(DEBUG_SEEK)
880
        av_log(NULL, AV_LOG_DEBUG, "%d cum_len=%"PRId64"\n", len, ast->cum_len);
881
#endif
882 883 884
        if(last_pos == pos)
            avi->non_interleaved= 1;
        else
885
            av_add_index_entry(st, pos, ast->cum_len / FFMAX(1, ast->sample_size), len, 0, (flags&AVIIF_INDEX) ? AVINDEX_KEYFRAME : 0);
886
        if(ast->sample_size)
887
            ast->cum_len += len;
888 889 890
        else
            ast->cum_len ++;
        last_pos= pos;
Fabrice Bellard's avatar
Fabrice Bellard committed
891 892 893 894
    }
    return 0;
}

895 896 897 898
static int guess_ni_flag(AVFormatContext *s){
    int i;
    int64_t last_start=0;
    int64_t first_end= INT64_MAX;
899

900 901 902 903 904 905 906 907 908 909 910 911 912 913 914
    for(i=0; i<s->nb_streams; i++){
        AVStream *st = s->streams[i];
        int n= st->nb_index_entries;

        if(n <= 0)
            continue;

        if(st->index_entries[0].pos > last_start)
            last_start= st->index_entries[0].pos;
        if(st->index_entries[n-1].pos < first_end)
            first_end= st->index_entries[n-1].pos;
    }
    return last_start > first_end;
}

Fabrice Bellard's avatar
Fabrice Bellard committed
915 916 917
static int avi_load_index(AVFormatContext *s)
{
    AVIContext *avi = s->priv_data;
918
    ByteIOContext *pb = s->pb;
Fabrice Bellard's avatar
Fabrice Bellard committed
919
    uint32_t tag, size;
920
    offset_t pos= url_ftell(pb);
921

Fabrice Bellard's avatar
Fabrice Bellard committed
922 923
    url_fseek(pb, avi->movi_end, SEEK_SET);
#ifdef DEBUG_SEEK
924
    printf("movi_end=0x%"PRIx64"\n", avi->movi_end);
Fabrice Bellard's avatar
Fabrice Bellard committed
925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953
#endif
    for(;;) {
        if (url_feof(pb))
            break;
        tag = get_le32(pb);
        size = get_le32(pb);
#ifdef DEBUG_SEEK
        printf("tag=%c%c%c%c size=0x%x\n",
               tag & 0xff,
               (tag >> 8) & 0xff,
               (tag >> 16) & 0xff,
               (tag >> 24) & 0xff,
               size);
#endif
        switch(tag) {
        case MKTAG('i', 'd', 'x', '1'):
            if (avi_read_idx1(s, size) < 0)
                goto skip;
            else
                goto the_end;
            break;
        default:
        skip:
            size += (size & 1);
            url_fskip(pb, size);
            break;
        }
    }
 the_end:
954
    url_fseek(pb, pos, SEEK_SET);
Fabrice Bellard's avatar
Fabrice Bellard committed
955 956 957
    return 0;
}

958
static int avi_read_seek(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
Fabrice Bellard's avatar
Fabrice Bellard committed
959 960 961
{
    AVIContext *avi = s->priv_data;
    AVStream *st;
962
    int i, index;
Fabrice Bellard's avatar
Fabrice Bellard committed
963 964 965 966 967 968 969
    int64_t pos;

    if (!avi->index_loaded) {
        /* we only load the index on demand */
        avi_load_index(s);
        avi->index_loaded = 1;
    }
970
    assert(stream_index>= 0);
Fabrice Bellard's avatar
Fabrice Bellard committed
971 972

    st = s->streams[stream_index];
973 974
    index= av_index_search_timestamp(st, timestamp, flags);
    if(index<0)
Fabrice Bellard's avatar
Fabrice Bellard committed
975
        return -1;
976

Fabrice Bellard's avatar
Fabrice Bellard committed
977
    /* find the position */
978 979 980
    pos = st->index_entries[index].pos;
    timestamp = st->index_entries[index].timestamp;

981
//    av_log(NULL, AV_LOG_DEBUG, "XX %"PRId64" %d %"PRId64"\n", timestamp, index, st->index_entries[index].timestamp);
Fabrice Bellard's avatar
Fabrice Bellard committed
982

983 984 985 986 987 988 989 990 991 992
    if (ENABLE_DV_DEMUXER && avi->dv_demux) {
        /* One and only one real stream for DV in AVI, and it has video  */
        /* offsets. Calling with other stream indices should have failed */
        /* the av_index_search_timestamp call above.                     */
        assert(stream_index == 0);

        /* Feed the DV video stream version of the timestamp to the */
        /* DV demux so it can synth correct timestamps              */
        dv_offset_reset(avi->dv_demux, timestamp);

993
        url_fseek(s->pb, pos, SEEK_SET);
994 995 996 997
        avi->stream_index= -1;
        return 0;
    }

Fabrice Bellard's avatar
Fabrice Bellard committed
998
    for(i = 0; i < s->nb_streams; i++) {
999 1000
        AVStream *st2 = s->streams[i];
        AVIStream *ast2 = st2->priv_data;
1001 1002 1003 1004

        ast2->packet_size=
        ast2->remaining= 0;

1005 1006
        if (st2->nb_index_entries <= 0)
            continue;
1007

1008
//        assert(st2->codec->block_align);
1009 1010 1011
        assert(st2->time_base.den == ast2->rate);
        assert(st2->time_base.num == ast2->scale);
        index = av_index_search_timestamp(
1012
                st2,
1013 1014 1015 1016
                av_rescale(timestamp, st2->time_base.den*(int64_t)st->time_base.num, st->time_base.den * (int64_t)st2->time_base.num),
                flags | AVSEEK_FLAG_BACKWARD);
        if(index<0)
            index=0;
1017

1018 1019 1020 1021 1022 1023 1024
        if(!avi->non_interleaved){
            while(index>0 && st2->index_entries[index].pos > pos)
                index--;
            while(index+1 < st2->nb_index_entries && st2->index_entries[index].pos < pos)
                index++;
        }

1025
//        av_log(NULL, AV_LOG_DEBUG, "%"PRId64" %d %"PRId64"\n", timestamp, index, st2->index_entries[index].timestamp);
1026 1027 1028 1029
        /* extract the current frame number */
        ast2->frame_offset = st2->index_entries[index].timestamp;
        if(ast2->sample_size)
            ast2->frame_offset *=ast2->sample_size;
Fabrice Bellard's avatar
Fabrice Bellard committed
1030
    }
1031

Fabrice Bellard's avatar
Fabrice Bellard committed
1032
    /* do the seek */
1033
    url_fseek(s->pb, pos, SEEK_SET);
1034
    avi->stream_index= -1;
Fabrice Bellard's avatar
Fabrice Bellard committed
1035 1036 1037
    return 0;
}

1038
static int avi_read_close(AVFormatContext *s)
Fabrice Bellard's avatar
Fabrice Bellard committed
1039
{
Michael Niedermayer's avatar
Michael Niedermayer committed
1040 1041 1042 1043 1044
    int i;
    AVIContext *avi = s->priv_data;

    for(i=0;i<s->nb_streams;i++) {
        AVStream *st = s->streams[i];
Fabrice Bellard's avatar
Fabrice Bellard committed
1045
        AVIStream *ast = st->priv_data;
1046
        av_free(ast);
1047
        av_free(st->codec->palctrl);
Michael Niedermayer's avatar
Michael Niedermayer committed
1048 1049
    }

1050 1051 1052
    if (avi->dv_demux)
        av_free(avi->dv_demux);

Fabrice Bellard's avatar
Fabrice Bellard committed
1053 1054 1055 1056 1057
    return 0;
}

static int avi_probe(AVProbeData *p)
{
1058 1059
    int i;

Fabrice Bellard's avatar
Fabrice Bellard committed
1060
    /* check file header */
1061 1062 1063 1064 1065 1066
    for(i=0; avi_headers[i][0]; i++)
        if(!memcmp(p->buf  , avi_headers[i]  , 4) &&
           !memcmp(p->buf+8, avi_headers[i]+4, 4))
            return AVPROBE_SCORE_MAX;

    return 0;
Fabrice Bellard's avatar
Fabrice Bellard committed
1067 1068
}

1069
AVInputFormat avi_demuxer = {
Fabrice Bellard's avatar
Fabrice Bellard committed
1070 1071 1072 1073 1074 1075 1076
    "avi",
    "avi format",
    sizeof(AVIContext),
    avi_probe,
    avi_read_header,
    avi_read_packet,
    avi_read_close,
Fabrice Bellard's avatar
Fabrice Bellard committed
1077
    avi_read_seek,
Fabrice Bellard's avatar
Fabrice Bellard committed
1078
};