git.sesse.net Git - ffmpeg/blob - libavformat/avidec.c

   1 /*
   2  * AVI decoder.
   3  * Copyright (c) 2001 Fabrice Bellard.
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18  */
  19 #include "avformat.h"
  20 #include "avi.h"
  21
  22 //#define DEBUG
  23
  24 typedef struct AVIIndex {
  25     unsigned char tag[4];
  26     unsigned int flags, pos, len;
  27     struct AVIIndex *next;
  28 } AVIIndex;
  29
  30 typedef struct {
  31     INT64 movi_end;
  32     offset_t movi_list;
  33     AVIIndex *first, *last;
  34 } AVIContext;
  35
  36 #ifdef DEBUG
  37 static void print_tag(const char *str, unsigned int tag, int size)
  38 {
  39     printf("%s: tag=%c%c%c%c size=0x%x\n",
  40            str, tag & 0xff,
  41            (tag >> 8) & 0xff,
  42            (tag >> 16) & 0xff,
  43            (tag >> 24) & 0xff,
  44            size);
  45 }
  46 #endif
  47
  48 static int avi_read_header(AVFormatContext *s, AVFormatParameters *ap)
  49 {
  50     AVIContext *avi = s->priv_data;
  51     ByteIOContext *pb = &s->pb;
  52     UINT32 tag, tag1;
  53     int codec_type, stream_index, frame_period, bit_rate;
  54     unsigned int size;
  55     int i;
  56     AVStream *st;
  57
  58     /* check RIFF header */
  59     tag = get_le32(pb);
  60
  61     if (tag != MKTAG('R', 'I', 'F', 'F'))
  62         return -1;
  63     get_le32(pb); /* file size */
  64     tag = get_le32(pb);
  65     if (tag != MKTAG('A', 'V', 'I', ' '))
  66         return -1;
  67
  68     /* first list tag */
  69     stream_index = -1;
  70     codec_type = -1;
  71     frame_period = 0;
  72     for(;;) {
  73         if (url_feof(pb))
  74             goto fail;
  75         tag = get_le32(pb);
  76         size = get_le32(pb);
  77 #ifdef DEBUG
  78         print_tag("tag", tag, size);
  79 #endif
  80
  81         switch(tag) {
  82         case MKTAG('L', 'I', 'S', 'T'):
  83             /* ignored, except when start of video packets */
  84             tag1 = get_le32(pb);
  85 #ifdef DEBUG
  86             print_tag("list", tag1, 0);
  87 #endif
  88             if (tag1 == MKTAG('m', 'o', 'v', 'i')) {
  89                 avi->movi_end = url_ftell(pb) + size - 4;
  90 #ifdef DEBUG
  91                 printf("movi end=%Lx\n", avi->movi_end);
  92 #endif
  93                 goto end_of_header;
  94             }
  95             break;
  96         case MKTAG('a', 'v', 'i', 'h'):
  97             /* avi header */
  98             /* using frame_period is bad idea */
  99             frame_period = get_le32(pb);
 100             bit_rate = get_le32(pb) * 8;
 101             url_fskip(pb, 4 * 4);
 102             s->nb_streams = get_le32(pb);
 103             for(i=0;i<s->nb_streams;i++) {
 104                 AVStream *st = av_mallocz(sizeof(AVStream));
 105                 if (!st)
 106                     goto fail;
 107                 avcodec_get_context_defaults(&st->codec);
 108
 109                 s->streams[i] = st;
 110             }
 111             url_fskip(pb, size - 7 * 4);
 112             break;
 113         case MKTAG('s', 't', 'r', 'h'):
 114             /* stream header */
 115             stream_index++;
 116             tag1 = get_le32(pb);
 117             switch(tag1) {
 118             case MKTAG('v', 'i', 'd', 's'):
 119                 codec_type = CODEC_TYPE_VIDEO;
 120                 get_le32(pb); /* codec tag */
 121                 get_le32(pb); /* flags */
 122                 get_le16(pb); /* priority */
 123                 get_le16(pb); /* language */
 124                 get_le32(pb); /* XXX: initial frame ? */
 125                 get_le32(pb); /* scale */
 126                 get_le32(pb); /* rate */
 127                 size -= 6 * 4;
 128                 break;
 129             case MKTAG('a', 'u', 'd', 's'):
 130                 codec_type = CODEC_TYPE_AUDIO;
 131                 /* nothing really useful */
 132             }
 133             url_fskip(pb, size - 4);
 134             break;
 135         case MKTAG('s', 't', 'r', 'f'):
 136             /* stream header */
 137             if (stream_index >= s->nb_streams) {
 138                 url_fskip(pb, size);
 139             } else {
 140                 st = s->streams[stream_index];
 141                 switch(codec_type) {
 142                 case CODEC_TYPE_VIDEO:
 143                     get_le32(pb); /* size */
 144                     st->codec.width = get_le32(pb);
 145                     st->codec.height = get_le32(pb);
 146                     if (frame_period)
 147                         st->codec.frame_rate = (INT64_C(1000000) * FRAME_RATE_BASE) / frame_period;
 148                     else
 149                         st->codec.frame_rate = 25 * FRAME_RATE_BASE;
 150                     get_le16(pb); /* panes */
 151                     get_le16(pb); /* depth */
 152                     tag1 = get_le32(pb);
 153 #ifdef DEBUG
 154                     print_tag("video", tag1, 0);
 155 #endif
 156                     st->codec.codec_type = CODEC_TYPE_VIDEO;
 157                     st->codec.codec_tag = tag1;
 158                     st->codec.codec_id = codec_get_id(codec_bmp_tags, tag1);
 159                     url_fskip(pb, size - 5 * 4);
 160                     break;
 161                 case CODEC_TYPE_AUDIO:
 162                     get_wav_header(pb, &st->codec, (size >= 18));
 163                     if (size%2) /* 2-aligned (fix for Stargate SG-1 - 3x18 - Shades of Grey.avi) */
 164                         url_fskip(pb, 1);
 165                     break;
 166                 default:
 167                     url_fskip(pb, size);
 168                     break;
 169                 }
 170             }
 171             break;
 172         default:
 173             /* skip tag */
 174             size += (size & 1);
 175             url_fskip(pb, size);
 176             break;
 177         }
 178     }
 179  end_of_header:
 180     /* check stream number */
 181     if (stream_index != s->nb_streams - 1) {
 182     fail:
 183         for(i=0;i<s->nb_streams;i++) {
 184             av_freep(&s->streams[i]);
 185         }
 186         return -1;
 187     }
 188
 189     return 0;
 190 }
 191
 192 static int avi_read_packet(AVFormatContext *s, AVPacket *pkt)
 193 {
 194     AVIContext *avi = s->priv_data;
 195     ByteIOContext *pb = &s->pb;
 196     int n, d1, d2, size;
 197
 198     for(;;) {
 199         if (url_feof(pb) || url_ftell(pb) >= avi->movi_end)
 200             return -1;
 201         d1 = get_byte(pb) - '0';
 202         d2 = get_byte(pb) - '0';
 203         if (d1 < 0 || d1 > 9 || d2 < 0 || d2 > 9)
 204             continue;
 205
 206         n = d1 * 10 + d2;
 207         if (n < 0 || n >= s->nb_streams)
 208             continue;
 209
 210         d1 = get_byte(pb);
 211         d2 = get_byte(pb);
 212         if ((d1 == 'd' && d2 == 'c')
 213             || (d1 == 'w' && d2 == 'b'))
 214             break;
 215     }
 216     size = get_le32(pb);
 217     av_new_packet(pkt, size);
 218     pkt->stream_index = n;
 219
 220     get_buffer(pb, pkt->data, pkt->size);
 221
 222     if (size & 1)
 223         get_byte(pb);
 224
 225     return 0;
 226 }
 227
 228 static int avi_read_close(AVFormatContext *s)
 229 {
 230     return 0;
 231 }
 232
 233 static int avi_probe(AVProbeData *p)
 234 {
 235     /* check file header */
 236     if (p->buf_size <= 32)
 237         return 0;
 238     if (p->buf[0] == 'R' && p->buf[1] == 'I' &&
 239         p->buf[2] == 'F' && p->buf[3] == 'F' &&
 240         p->buf[8] == 'A' && p->buf[9] == 'V' &&
 241         p->buf[10] == 'I' && p->buf[11] == ' ')
 242         return AVPROBE_SCORE_MAX;
 243     else
 244         return 0;
 245 }
 246
 247 static AVInputFormat avi_iformat = {
 248     "avi",
 249     "avi format",
 250     sizeof(AVIContext),
 251     avi_probe,
 252     avi_read_header,
 253     avi_read_packet,
 254     avi_read_close,
 255 };
 256
 257 int avidec_init(void)
 258 {
 259     av_register_input_format(&avi_iformat);
 260     return 0;
 261 }