Statistics
| Branch: | Revision:

ffmpeg / libavformat / ogg.c @ 756fb7fe

History | View | Annotate | Download (7.74 KB)

1
/*
2
 * Ogg bitstream support
3
 * Mark Hills <mark@pogo.org.uk>
4
 *
5
 * Uses libogg, but requires libvorbisenc to construct correct headers
6
 * when containing Vorbis stream -- currently the only format supported
7
 *
8
 * This file is part of FFmpeg.
9
 *
10
 * FFmpeg is free software; you can redistribute it and/or
11
 * modify it under the terms of the GNU Lesser General Public
12
 * License as published by the Free Software Foundation; either
13
 * version 2.1 of the License, or (at your option) any later version.
14
 *
15
 * FFmpeg is distributed in the hope that it will be useful,
16
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
18
 * Lesser General Public License for more details.
19
 *
20
 * You should have received a copy of the GNU Lesser General Public
21
 * License along with FFmpeg; if not, write to the Free Software
22
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23
 */
24

    
25
#include <stdio.h>
26

    
27
#include <ogg/ogg.h>
28

    
29
#include "avformat.h"
30

    
31
#undef NDEBUG
32
#include <assert.h>
33

    
34
#define DECODER_BUFFER_SIZE 4096
35

    
36

    
37
typedef struct OggContext {
38
    /* output */
39
    ogg_stream_state os ;
40
    int header_handled ;
41
    ogg_packet op;
42

    
43
    /* input */
44
    ogg_sync_state oy ;
45
} OggContext ;
46

    
47

    
48
#ifdef CONFIG_MUXERS
49
static int ogg_write_header(AVFormatContext *avfcontext)
50
{
51
    OggContext *context = avfcontext->priv_data;
52
    ogg_packet *op= &context->op;
53
    int n;
54

    
55
    ogg_stream_init(&context->os, 31415);
56

    
57
    for(n = 0 ; n < avfcontext->nb_streams ; n++) {
58
        AVCodecContext *codec = avfcontext->streams[n]->codec;
59
        uint8_t *headers = codec->extradata;
60
        int headers_len = codec->extradata_size;
61
        uint8_t *header_start[3];
62
        int header_len[3];
63
        int i, j;
64

    
65
        av_set_pts_info(avfcontext->streams[n], 60, 1, AV_TIME_BASE);
66

    
67
        for(j=1,i=0;i<2;++i, ++j) {
68
            header_len[i]=0;
69
            while(j<headers_len && headers[j]==0xff) {
70
                header_len[i]+=0xff;
71
                ++j;
72
            }
73
            header_len[i]+=headers[j];
74
        }
75
        header_len[2]=headers_len-header_len[0]-header_len[1]-j;
76
        headers+=j;
77
        header_start[0] = headers;
78
        header_start[1] = header_start[0] + header_len[0];
79
        header_start[2] = header_start[1] + header_len[1];
80

    
81
        for(i=0; i < 3; ++i){
82
            op->bytes = header_len[i];
83

    
84
            op->packet= header_start[i];
85
            op->b_o_s= op->packetno==0;
86

    
87
            ogg_stream_packetin(&context->os, op);
88

    
89
            op->packetno++; //FIXME multiple streams
90
        }
91

    
92
        context->header_handled = 0 ;
93
    }
94

    
95
    return 0 ;
96
}
97

    
98
static int ogg_write_packet(AVFormatContext *avfcontext, AVPacket *pkt)
99
{
100
    OggContext *context = avfcontext->priv_data ;
101
    AVCodecContext *avctx= avfcontext->streams[pkt->stream_index]->codec;
102
    ogg_packet *op= &context->op;
103
    ogg_page og ;
104
    int64_t pts;
105

    
106
    pts= av_rescale(pkt->pts, avctx->sample_rate, AV_TIME_BASE);
107

    
108
//    av_log(avfcontext, AV_LOG_DEBUG, "M%d\n", size);
109

    
110
    /* flush header packets so audio starts on a new page */
111

    
112
    if(!context->header_handled) {
113
        while(ogg_stream_flush(&context->os, &og)) {
114
            put_buffer(&avfcontext->pb, og.header, og.header_len) ;
115
            put_buffer(&avfcontext->pb, og.body, og.body_len) ;
116
            put_flush_packet(&avfcontext->pb);
117
        }
118
        context->header_handled = 1 ;
119
    }
120

    
121
    op->packet = (uint8_t*) pkt->data;
122
    op->bytes  = pkt->size;
123
    op->b_o_s  = op->packetno == 0;
124
    op->granulepos= pts;
125

    
126
    /* correct the fields in the packet -- essential for streaming */
127

    
128
    ogg_stream_packetin(&context->os, op);
129

    
130
    while(ogg_stream_pageout(&context->os, &og)) {
131
        put_buffer(&avfcontext->pb, og.header, og.header_len);
132
        put_buffer(&avfcontext->pb, og.body, og.body_len);
133
        put_flush_packet(&avfcontext->pb);
134
    }
135
    op->packetno++;
136

    
137
    return 0;
138
}
139

    
140

    
141
static int ogg_write_trailer(AVFormatContext *avfcontext) {
142
    OggContext *context = avfcontext->priv_data ;
143
    ogg_page og ;
144

    
145
    while(ogg_stream_flush(&context->os, &og)) {
146
        put_buffer(&avfcontext->pb, og.header, og.header_len) ;
147
        put_buffer(&avfcontext->pb, og.body, og.body_len) ;
148
        put_flush_packet(&avfcontext->pb);
149
    }
150

    
151
    ogg_stream_clear(&context->os) ;
152
    return 0 ;
153
}
154

    
155

    
156
AVOutputFormat ogg_muxer = {
157
    "ogg",
158
    "Ogg format",
159
    "application/ogg",
160
    "ogg",
161
    sizeof(OggContext),
162
    CODEC_ID_VORBIS,
163
    0,
164
    ogg_write_header,
165
    ogg_write_packet,
166
    ogg_write_trailer,
167
} ;
168
#endif //CONFIG_MUXERS
169

    
170
#if 0
171
static int next_packet(AVFormatContext *avfcontext, ogg_packet *op) {
172
    OggContext *context = avfcontext->priv_data ;
173
    ogg_page og ;
174
    char *buf ;
175

176
    while(ogg_stream_packetout(&context->os, op) != 1) {
177

178
        /* while no pages are available, read in more data to the sync */
179
        while(ogg_sync_pageout(&context->oy, &og) != 1) {
180
            buf = ogg_sync_buffer(&context->oy, DECODER_BUFFER_SIZE) ;
181
            if(get_buffer(&avfcontext->pb, buf, DECODER_BUFFER_SIZE) <= 0)
182
                return 1 ;
183
            ogg_sync_wrote(&context->oy, DECODER_BUFFER_SIZE) ;
184
        }
185

186
        /* got a page. Feed it into the stream and get the packet */
187
        if(ogg_stream_pagein(&context->os, &og) != 0)
188
            return 1 ;
189
    }
190

191
    return 0 ;
192
}
193

194

195
static int ogg_read_header(AVFormatContext *avfcontext, AVFormatParameters *ap)
196
{
197
    OggContext *context = avfcontext->priv_data;
198
    ogg_packet op ;
199
    char *buf ;
200
    ogg_page og ;
201
    AVStream *ast ;
202
    AVCodecContext *codec;
203
    uint8_t *p;
204
    int i;
205

206
    ogg_sync_init(&context->oy) ;
207
    buf = ogg_sync_buffer(&context->oy, DECODER_BUFFER_SIZE) ;
208

209
    if(get_buffer(&avfcontext->pb, buf, DECODER_BUFFER_SIZE) <= 0)
210
        return AVERROR_IO ;
211

212
    ogg_sync_wrote(&context->oy, DECODER_BUFFER_SIZE) ;
213
    ogg_sync_pageout(&context->oy, &og) ;
214
    ogg_stream_init(&context->os, ogg_page_serialno(&og)) ;
215
    ogg_stream_pagein(&context->os, &og) ;
216

217
    /* currently only one vorbis stream supported */
218

219
    ast = av_new_stream(avfcontext, 0) ;
220
    if(!ast)
221
        return AVERROR_NOMEM ;
222
    av_set_pts_info(ast, 60, 1, AV_TIME_BASE);
223

224
    codec= &ast->codec;
225
    codec->codec_type = CODEC_TYPE_AUDIO;
226
    codec->codec_id = CODEC_ID_VORBIS;
227
    for(i=0; i<3; i++){
228
        if(next_packet(avfcontext, &op)){
229
            return -1;
230
        }
231
        if(op.bytes >= (1<<16) || op.bytes < 0)
232
            return -1;
233
        codec->extradata_size+= 2 + op.bytes;
234
        codec->extradata= av_realloc(codec->extradata, codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
235
        memset(codec->extradata + codec->extradata_size, 0, FF_INPUT_BUFFER_PADDING_SIZE);
236
        p= codec->extradata + codec->extradata_size - 2 - op.bytes;
237
        *(p++)= op.bytes>>8;
238
        *(p++)= op.bytes&0xFF;
239
        memcpy(p, op.packet, op.bytes);
240
    }
241

242
    return 0 ;
243
}
244

245

246
static int ogg_read_packet(AVFormatContext *avfcontext, AVPacket *pkt) {
247
    ogg_packet op ;
248

249
    if(next_packet(avfcontext, &op))
250
        return AVERROR_IO ;
251
    if(av_new_packet(pkt, op.bytes) < 0)
252
        return AVERROR_IO ;
253
    pkt->stream_index = 0 ;
254
    memcpy(pkt->data, op.packet, op.bytes);
255
    if(avfcontext->streams[0]->codec.sample_rate && op.granulepos!=-1)
256
        pkt->pts= av_rescale(op.granulepos, AV_TIME_BASE, avfcontext->streams[0]->codec.sample_rate);
257
//    printf("%"PRId64" %d %d\n", pkt->pts, (int)op.granulepos, avfcontext->streams[0]->codec.sample_rate);
258

259
    return op.bytes;
260
}
261

262

263
static int ogg_read_close(AVFormatContext *avfcontext) {
264
    OggContext *context = avfcontext->priv_data ;
265

266
    ogg_stream_clear(&context->os) ;
267
    ogg_sync_clear(&context->oy) ;
268

269
    return 0 ;
270
}
271

272

273
static AVInputFormat ogg_iformat = {
274
    "ogg",
275
    "Ogg Vorbis",
276
    sizeof(OggContext),
277
    NULL,
278
    ogg_read_header,
279
    ogg_read_packet,
280
    ogg_read_close,
281
    .extensions = "ogg",
282
} ;
283
#endif