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
   9 #include <stdio.h>
  10
  11 #include <ogg/ogg.h>
  12
  13 #include "avformat.h"
  14
  15 #undef NDEBUG
  16 #include <assert.h>
  17
  18 #define DECODER_BUFFER_SIZE 4096
  19
  20
  21 typedef struct OggContext {
  22     /* output */
  23     ogg_stream_state os ;
  24     int header_handled ;
  25     ogg_packet op;
  26
  27     /* input */
  28     ogg_sync_state oy ;
  29 } OggContext ;
  30
  31
  32 #ifdef CONFIG_ENCODERS
  33 static int ogg_write_header(AVFormatContext *avfcontext)
  34 {
  35     OggContext *context = avfcontext->priv_data;
  36     ogg_packet *op= &context->op;
  37     int n, i;
  38
  39     ogg_stream_init(&context->os, 31415);
  40
  41     for(n = 0 ; n < avfcontext->nb_streams ; n++) {
  42         AVCodecContext *codec = &avfcontext->streams[n]->codec;
  43         uint8_t *p= codec->extradata;
  44
  45         av_set_pts_info(avfcontext->streams[n], 60, 1, AV_TIME_BASE);
  46
  47         for(i=0; i < codec->extradata_size; i+= op->bytes){
  48             op->bytes = p[i++]<<8;
  49             op->bytes+= p[i++];
  50
  51             op->packet= &p[i];
  52             op->b_o_s= op->packetno==0;
  53
  54             ogg_stream_packetin(&context->os, op);
  55
  56             op->packetno++; //FIXME multiple streams
  57         }
  58
  59         context->header_handled = 0 ;
  60     }
  61
  62     return 0 ;
  63 }
  64
  65 static int ogg_write_packet(AVFormatContext *avfcontext, AVPacket *pkt)
  66 {
  67     OggContext *context = avfcontext->priv_data ;
  68     AVCodecContext *avctx= &avfcontext->streams[pkt->stream_index]->codec;
  69     ogg_packet *op= &context->op;
  70     ogg_page og ;
  71     int64_t pts;
  72
  73     pts= av_rescale(pkt->pts, avctx->sample_rate, AV_TIME_BASE);
  74
  75 //    av_log(avfcontext, AV_LOG_DEBUG, "M%d\n", size);
  76
  77     /* flush header packets so audio starts on a new page */
  78
  79     if(!context->header_handled) {
  80         while(ogg_stream_flush(&context->os, &og)) {
  81             put_buffer(&avfcontext->pb, og.header, og.header_len) ;
  82             put_buffer(&avfcontext->pb, og.body, og.body_len) ;
  83             put_flush_packet(&avfcontext->pb);
  84         }
  85         context->header_handled = 1 ;
  86     }
  87
  88     op->packet = (uint8_t*) pkt->data;
  89     op->bytes  = pkt->size;
  90     op->b_o_s  = op->packetno == 0;
  91     op->granulepos= pts;
  92
  93     /* correct the fields in the packet -- essential for streaming */
  94
  95     ogg_stream_packetin(&context->os, op);
  96
  97     while(ogg_stream_pageout(&context->os, &og)) {
  98         put_buffer(&avfcontext->pb, og.header, og.header_len);
  99         put_buffer(&avfcontext->pb, og.body, og.body_len);
 100         put_flush_packet(&avfcontext->pb);
 101     }
 102     op->packetno++;
 103
 104     return 0;
 105 }
 106
 107
 108 static int ogg_write_trailer(AVFormatContext *avfcontext) {
 109     OggContext *context = avfcontext->priv_data ;
 110     ogg_page og ;
 111
 112     while(ogg_stream_flush(&context->os, &og)) {
 113         put_buffer(&avfcontext->pb, og.header, og.header_len) ;
 114         put_buffer(&avfcontext->pb, og.body, og.body_len) ;
 115         put_flush_packet(&avfcontext->pb);
 116     }
 117
 118     ogg_stream_clear(&context->os) ;
 119     return 0 ;
 120 }
 121
 122
 123 static AVOutputFormat ogg_oformat = {
 124     "ogg",
 125     "Ogg Vorbis",
 126     "audio/x-vorbis",
 127     "ogg",
 128     sizeof(OggContext),
 129     CODEC_ID_VORBIS,
 130     0,
 131     ogg_write_header,
 132     ogg_write_packet,
 133     ogg_write_trailer,
 134 } ;
 135 #endif //CONFIG_ENCODERS
 136
 137
 138 static int next_packet(AVFormatContext *avfcontext, ogg_packet *op) {
 139     OggContext *context = avfcontext->priv_data ;
 140     ogg_page og ;
 141     char *buf ;
 142
 143     while(ogg_stream_packetout(&context->os, op) != 1) {
 144
 145         /* while no pages are available, read in more data to the sync */
 146         while(ogg_sync_pageout(&context->oy, &og) != 1) {
 147             buf = ogg_sync_buffer(&context->oy, DECODER_BUFFER_SIZE) ;
 148             if(get_buffer(&avfcontext->pb, buf, DECODER_BUFFER_SIZE) <= 0)
 149                 return 1 ;
 150             ogg_sync_wrote(&context->oy, DECODER_BUFFER_SIZE) ;
 151         }
 152
 153         /* got a page. Feed it into the stream and get the packet */
 154         if(ogg_stream_pagein(&context->os, &og) != 0)
 155             return 1 ;
 156     }
 157
 158     return 0 ;
 159 }
 160
 161
 162 static int ogg_read_header(AVFormatContext *avfcontext, AVFormatParameters *ap)
 163 {
 164     OggContext *context = avfcontext->priv_data;
 165     ogg_packet op ;
 166     char *buf ;
 167     ogg_page og ;
 168     AVStream *ast ;
 169     AVCodecContext *codec;
 170     uint8_t *p;
 171     int i;
 172
 173     ogg_sync_init(&context->oy) ;
 174     buf = ogg_sync_buffer(&context->oy, DECODER_BUFFER_SIZE) ;
 175
 176     if(get_buffer(&avfcontext->pb, buf, DECODER_BUFFER_SIZE) <= 0)
 177         return AVERROR_IO ;
 178
 179     ogg_sync_wrote(&context->oy, DECODER_BUFFER_SIZE) ;
 180     ogg_sync_pageout(&context->oy, &og) ;
 181     ogg_stream_init(&context->os, ogg_page_serialno(&og)) ;
 182     ogg_stream_pagein(&context->os, &og) ;
 183
 184     /* currently only one vorbis stream supported */
 185
 186     ast = av_new_stream(avfcontext, 0) ;
 187     if(!ast)
 188         return AVERROR_NOMEM ;
 189     av_set_pts_info(ast, 60, 1, AV_TIME_BASE);
 190
 191     codec= &ast->codec;
 192     codec->codec_type = CODEC_TYPE_AUDIO;
 193     codec->codec_id = CODEC_ID_VORBIS;
 194     for(i=0; i<3; i++){
 195         if(next_packet(avfcontext, &op)){
 196             return -1;
 197         }
 198         codec->extradata_size+= 2 + op.bytes;
 199         codec->extradata= av_realloc(codec->extradata, codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
 200         p= codec->extradata + codec->extradata_size - 2 - op.bytes;
 201         *(p++)= op.bytes>>8;
 202         *(p++)= op.bytes&0xFF;
 203         memcpy(p, op.packet, op.bytes);
 204     }
 205
 206     return 0 ;
 207 }
 208
 209
 210 static int ogg_read_packet(AVFormatContext *avfcontext, AVPacket *pkt) {
 211     ogg_packet op ;
 212
 213     if(next_packet(avfcontext, &op))
 214         return AVERROR_IO ;
 215     if(av_new_packet(pkt, op.bytes) < 0)
 216         return AVERROR_IO ;
 217     pkt->stream_index = 0 ;
 218     memcpy(pkt->data, op.packet, op.bytes);
 219     if(avfcontext->streams[0]->codec.sample_rate && op.granulepos!=-1)
 220         pkt->pts= av_rescale(op.granulepos, AV_TIME_BASE, avfcontext->streams[0]->codec.sample_rate);
 221 //    printf("%lld %d %d\n", pkt->pts, (int)op.granulepos, avfcontext->streams[0]->codec.sample_rate);
 222
 223     return op.bytes;
 224 }
 225
 226
 227 static int ogg_read_close(AVFormatContext *avfcontext) {
 228     OggContext *context = avfcontext->priv_data ;
 229
 230     ogg_stream_clear(&context->os) ;
 231     ogg_sync_clear(&context->oy) ;
 232     av_freep(&avfcontext->streams[0]->codec.extradata);
 233
 234     return 0 ;
 235 }
 236
 237
 238 static AVInputFormat ogg_iformat = {
 239     "ogg",
 240     "Ogg Vorbis",
 241     sizeof(OggContext),
 242     NULL,
 243     ogg_read_header,
 244     ogg_read_packet,
 245     ogg_read_close,
 246     .extensions = "ogg",
 247 } ;
 248
 249
 250 int ogg_init(void) {
 251 #ifdef CONFIG_ENCODERS
 252     av_register_output_format(&ogg_oformat) ;
 253 #endif
 254     av_register_input_format(&ogg_iformat);
 255     return 0 ;
 256 }