git.sesse.net Git - ffmpeg/blob - 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