oggparseogm.c 5.68 KB
Newer Older
Måns Rullgård's avatar
Måns Rullgård committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
/**
    Copyright (C) 2005  Michael Ahlberg, Måns Rullgård

    Permission is hereby granted, free of charge, to any person
    obtaining a copy of this software and associated documentation
    files (the "Software"), to deal in the Software without
    restriction, including without limitation the rights to use, copy,
    modify, merge, publish, distribute, sublicense, and/or sell copies
    of the Software, and to permit persons to whom the Software is
    furnished to do so, subject to the following conditions:

    The above copyright notice and this permission notice shall be
    included in all copies or substantial portions of the Software.

    THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
    EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
    MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
    NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
    HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
    WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
    OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
    DEALINGS IN THE SOFTWARE.
**/

#include <stdlib.h>
26
#include "libavutil/intreadwrite.h"
27
#include "libavcodec/get_bits.h"
28
#include "libavcodec/bytestream.h"
Måns Rullgård's avatar
Måns Rullgård committed
29
#include "avformat.h"
30
#include "oggdec.h"
31
#include "riff.h"
Måns Rullgård's avatar
Måns Rullgård committed
32 33 34 35

static int
ogm_header(AVFormatContext *s, int idx)
{
36 37
    struct ogg *ogg = s->priv_data;
    struct ogg_stream *os = ogg->streams + idx;
Måns Rullgård's avatar
Måns Rullgård committed
38
    AVStream *st = s->streams[idx];
Baptiste Coudurier's avatar
Baptiste Coudurier committed
39
    const uint8_t *p = os->buf + os->pstart;
Måns Rullgård's avatar
Måns Rullgård committed
40 41 42 43 44 45 46
    uint64_t time_unit;
    uint64_t spu;
    uint32_t default_len;

    if(!(*p & 1))
        return 0;

David Conrad's avatar
David Conrad committed
47 48
    if(*p == 1) {
        p++;
Måns Rullgård's avatar
Måns Rullgård committed
49

David Conrad's avatar
David Conrad committed
50 51
        if(*p == 'v'){
            int tag;
52
            st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
David Conrad's avatar
David Conrad committed
53 54 55 56 57
            p += 8;
            tag = bytestream_get_le32(&p);
            st->codec->codec_id = ff_codec_get_id(ff_codec_bmp_tags, tag);
            st->codec->codec_tag = tag;
        } else if (*p == 't') {
58
            st->codec->codec_type = AVMEDIA_TYPE_SUBTITLE;
David Conrad's avatar
David Conrad committed
59 60 61 62 63
            st->codec->codec_id = CODEC_ID_TEXT;
            p += 12;
        } else {
            uint8_t acid[5];
            int cid;
64
            st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
David Conrad's avatar
David Conrad committed
65 66 67 68 69 70 71
            p += 8;
            bytestream_get_buffer(&p, acid, 4);
            acid[4] = 0;
            cid = strtol(acid, NULL, 16);
            st->codec->codec_id = ff_codec_get_id(ff_codec_wav_tags, cid);
            st->need_parsing = AVSTREAM_PARSE_FULL;
        }
Måns Rullgård's avatar
Måns Rullgård committed
72

David Conrad's avatar
David Conrad committed
73
        p += 4;                     /* useless size field */
Måns Rullgård's avatar
Måns Rullgård committed
74

David Conrad's avatar
David Conrad committed
75 76 77
        time_unit   = bytestream_get_le64(&p);
        spu         = bytestream_get_le64(&p);
        default_len = bytestream_get_le32(&p);
Måns Rullgård's avatar
Måns Rullgård committed
78

David Conrad's avatar
David Conrad committed
79
        p += 8;                     /* buffersize + bits_per_sample */
Måns Rullgård's avatar
Måns Rullgård committed
80

81
        if(st->codec->codec_type == AVMEDIA_TYPE_VIDEO){
David Conrad's avatar
David Conrad committed
82 83 84 85 86 87 88 89 90 91 92 93 94
            st->codec->width = bytestream_get_le32(&p);
            st->codec->height = bytestream_get_le32(&p);
            st->codec->time_base.den = spu * 10000000;
            st->codec->time_base.num = time_unit;
            st->time_base = st->codec->time_base;
        } else {
            st->codec->channels = bytestream_get_le16(&p);
            p += 2;                 /* block_align */
            st->codec->bit_rate = bytestream_get_le32(&p) * 8;
            st->codec->sample_rate = spu * 10000000 / time_unit;
            st->time_base.num = 1;
            st->time_base.den = st->codec->sample_rate;
        }
David Conrad's avatar
David Conrad committed
95 96 97 98
    } else if (*p == 3) {
        if (os->psize > 8)
            ff_vorbis_comment(s, &st->metadata, p+7, os->psize-8);
    }
Måns Rullgård's avatar
Måns Rullgård committed
99 100 101 102 103 104 105

    return 1;
}

static int
ogm_dshow_header(AVFormatContext *s, int idx)
{
106 107
    struct ogg *ogg = s->priv_data;
    struct ogg_stream *os = ogg->streams + idx;
Måns Rullgård's avatar
Måns Rullgård committed
108 109 110 111 112 113 114 115 116
    AVStream *st = s->streams[idx];
    uint8_t *p = os->buf + os->pstart;
    uint32_t t;

    if(!(*p & 1))
        return 0;
    if(*p != 1)
        return 1;

117
    t = AV_RL32(p + 96);
Måns Rullgård's avatar
Måns Rullgård committed
118 119

    if(t == 0x05589f80){
120
        st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
121
        st->codec->codec_id = ff_codec_get_id(ff_codec_bmp_tags, AV_RL32(p + 68));
Måns Rullgård's avatar
Måns Rullgård committed
122
        st->codec->time_base.den = 10000000;
123 124 125
        st->codec->time_base.num = AV_RL64(p + 164);
        st->codec->width = AV_RL32(p + 176);
        st->codec->height = AV_RL32(p + 180);
Måns Rullgård's avatar
Måns Rullgård committed
126
    } else if(t == 0x05589f81){
127
        st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
128
        st->codec->codec_id = ff_codec_get_id(ff_codec_wav_tags, AV_RL16(p + 124));
129 130 131
        st->codec->channels = AV_RL16(p + 126);
        st->codec->sample_rate = AV_RL32(p + 128);
        st->codec->bit_rate = AV_RL32(p + 132) * 8;
Måns Rullgård's avatar
Måns Rullgård committed
132 133 134 135 136 137 138 139
    }

    return 1;
}

static int
ogm_packet(AVFormatContext *s, int idx)
{
140 141
    struct ogg *ogg = s->priv_data;
    struct ogg_stream *os = ogg->streams + idx;
Måns Rullgård's avatar
Måns Rullgård committed
142 143 144
    uint8_t *p = os->buf + os->pstart;
    int lb;

145
    if(*p & 8)
146
        os->pflags |= AV_PKT_FLAG_KEY;
147

Måns Rullgård's avatar
Måns Rullgård committed
148 149 150 151
    lb = ((*p & 2) << 1) | ((*p >> 6) & 3);
    os->pstart += lb + 1;
    os->psize -= lb + 1;

152 153 154
    while (lb--)
        os->pduration += p[lb+1] << (lb*8);

Måns Rullgård's avatar
Måns Rullgård committed
155 156 157
    return 0;
}

158
const struct ogg_codec ff_ogm_video_codec = {
Måns Rullgård's avatar
Måns Rullgård committed
159 160 161
    .magic = "\001video",
    .magicsize = 6,
    .header = ogm_header,
David Conrad's avatar
David Conrad committed
162 163
    .packet = ogm_packet,
    .granule_is_start = 1,
Måns Rullgård's avatar
Måns Rullgård committed
164 165
};

166
const struct ogg_codec ff_ogm_audio_codec = {
Måns Rullgård's avatar
Måns Rullgård committed
167 168 169
    .magic = "\001audio",
    .magicsize = 6,
    .header = ogm_header,
David Conrad's avatar
David Conrad committed
170 171
    .packet = ogm_packet,
    .granule_is_start = 1,
Måns Rullgård's avatar
Måns Rullgård committed
172 173
};

174
const struct ogg_codec ff_ogm_text_codec = {
175 176 177
    .magic = "\001text",
    .magicsize = 5,
    .header = ogm_header,
David Conrad's avatar
David Conrad committed
178 179
    .packet = ogm_packet,
    .granule_is_start = 1,
180 181
};

182
const struct ogg_codec ff_ogm_old_codec = {
Måns Rullgård's avatar
Måns Rullgård committed
183 184 185
    .magic = "\001Direct Show Samples embedded in Ogg",
    .magicsize = 35,
    .header = ogm_dshow_header,
David Conrad's avatar
David Conrad committed
186 187
    .packet = ogm_packet,
    .granule_is_start = 1,
Måns Rullgård's avatar
Måns Rullgård committed
188
};