c98bb2273594b1f58a3a6f9265236ab5b76b4900
[ffmpeg.git] / libavformat / ogg.c
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