1/*
2 * Copyright (c) 2010 David Conrad
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21#include "avformat.h"
22#include "internal.h"
23#include "riff.h"
24#include "libavutil/intreadwrite.h"
25
26static int probe(AVProbeData *p)
27{
28    if (AV_RL32(p->buf) == MKTAG('D','K','I','F')
29        && !AV_RL16(p->buf+4) && AV_RL16(p->buf+6) == 32)
30        return AVPROBE_SCORE_MAX-2;
31
32    return 0;
33}
34
35static int read_header(AVFormatContext *s)
36{
37    AVStream *st;
38    AVRational time_base;
39
40    avio_rl32(s->pb); // DKIF
41    avio_rl16(s->pb); // version
42    avio_rl16(s->pb); // header size
43
44    st = avformat_new_stream(s, NULL);
45    if (!st)
46        return AVERROR(ENOMEM);
47
48
49    st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
50    st->codec->codec_tag  = avio_rl32(s->pb);
51    st->codec->codec_id   = ff_codec_get_id(ff_codec_bmp_tags, st->codec->codec_tag);
52    st->codec->width      = avio_rl16(s->pb);
53    st->codec->height     = avio_rl16(s->pb);
54    time_base.den         = avio_rl32(s->pb);
55    time_base.num         = avio_rl32(s->pb);
56    st->duration          = avio_rl64(s->pb);
57
58    st->need_parsing      = AVSTREAM_PARSE_HEADERS;
59
60    if (!time_base.den || !time_base.num) {
61        av_log(s, AV_LOG_ERROR, "Invalid frame rate\n");
62        return AVERROR_INVALIDDATA;
63    }
64
65    avpriv_set_pts_info(st, 64, time_base.num, time_base.den);
66
67    return 0;
68}
69
70static int read_packet(AVFormatContext *s, AVPacket *pkt)
71{
72    int ret, size = avio_rl32(s->pb);
73    int64_t   pts = avio_rl64(s->pb);
74
75    ret = av_get_packet(s->pb, pkt, size);
76    pkt->stream_index = 0;
77    pkt->pts          = pts;
78    pkt->pos         -= 12;
79
80    return ret;
81}
82
83AVInputFormat ff_ivf_demuxer = {
84    .name           = "ivf",
85    .long_name      = NULL_IF_CONFIG_SMALL("On2 IVF"),
86    .read_probe     = probe,
87    .read_header    = read_header,
88    .read_packet    = read_packet,
89    .flags          = AVFMT_GENERIC_INDEX,
90    .codec_tag      = (const AVCodecTag* const []){ ff_codec_bmp_tags, 0 },
91};
92