git.sesse.net Git - vlc/blob - modules/demux/ogg.c

   1 /*****************************************************************************
   2  * ogg.c : ogg stream input module for vlc
   3  *****************************************************************************
   4  * Copyright (C) 2001 VideoLAN
   5  * $Id: ogg.c,v 1.13 2002/11/26 17:38:33 gbazin Exp $
   6  *
   7  * Authors: Gildas Bazin <gbazin@netcourrier.com>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  22  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27 #include <stdlib.h>                                      /* malloc(), free() */
  28 #include <string.h>
  29
  30 #include <vlc/vlc.h>
  31 #include <vlc/input.h>
  32
  33 #include <sys/types.h>
  34
  35 #include <ogg/ogg.h>
  36
  37 #include <codecs.h>                        /* BITMAPINFOHEADER, WAVEFORMATEX */
  38
  39 #define OGG_BLOCK_SIZE 4096
  40 #define PAGES_READ_ONCE 1
  41
  42 /*****************************************************************************
  43  * Definitions of structures and functions used by this plugins
  44  *****************************************************************************/
  45 typedef struct logical_stream_s
  46 {
  47     ogg_stream_state os;                        /* logical stream of packets */
  48
  49     int              i_serial_no;
  50     int              i_cat;                            /* AUDIO_ES, VIDEO_ES */
  51     int              i_activated;
  52     vlc_fourcc_t     i_fourcc;
  53     vlc_fourcc_t     i_codec;
  54
  55     es_descriptor_t  *p_es;
  56     int              b_selected;                           /* newly selected */
  57
  58     /* the header of some logical streams (eg vorbis) contain essential
  59      * data for the decoder. We back them up here in case we need to re-feed
  60      * them to the decoder. */
  61     int              b_force_backup;
  62     int              i_packets_backup;
  63     ogg_packet       *p_packets_backup;
  64
  65     /* program clock reference (in units of 90kHz) derived from the previous
  66      * granulepos */
  67     mtime_t          i_pcr;
  68     mtime_t          i_interpolated_pcr;
  69
  70     /* info from logical streams */
  71     double f_rate;
  72     int i_bitrate;
  73     int b_reinit;
  74
  75     /* codec specific stuff */
  76     BITMAPINFOHEADER *p_bih;
  77     WAVEFORMATEX *p_wf;
  78     int i_theora_keyframe_granule_shift;
  79
  80 } logical_stream_t;
  81
  82 struct demux_sys_t
  83 {
  84     ogg_sync_state oy;        /* sync and verify incoming physical bitstream */
  85
  86     int i_streams;                           /* number of logical bitstreams */
  87     logical_stream_t **pp_stream;  /* pointer to an array of logical streams */
  88
  89     /* current audio and video es */
  90     logical_stream_t *p_stream_video;
  91     logical_stream_t *p_stream_audio;
  92     logical_stream_t *p_stream_spu;
  93
  94     /* program clock reference (in units of 90kHz) derived from the pcr of
  95      * the sub-streams */
  96     mtime_t i_pcr;
  97
  98     mtime_t i_length;
  99     int     b_seekable;
 100     int     b_reinit;
 101 };
 102
 103 /* OggDS headers for the new header format (used in ogm files) */
 104 typedef struct stream_header_video
 105 {
 106     ogg_int32_t width;
 107     ogg_int32_t height;
 108 } stream_header_video;
 109
 110 typedef struct stream_header_audio
 111 {
 112     ogg_int16_t channels;
 113     ogg_int16_t blockalign;
 114     ogg_int32_t avgbytespersec;
 115 } stream_header_audio;
 116
 117 typedef struct stream_header
 118 {
 119     char        streamtype[8];
 120     char        subtype[4];
 121
 122     ogg_int32_t size;                               /* size of the structure */
 123
 124     ogg_int64_t time_unit;                              /* in reference time */
 125     ogg_int64_t samples_per_unit;
 126     ogg_int32_t default_len;                                /* in media time */
 127
 128     ogg_int32_t buffersize;
 129     ogg_int16_t bits_per_sample;
 130
 131     union
 132     {
 133         /* Video specific */
 134         stream_header_video video;
 135         /* Audio specific */
 136         stream_header_audio audio;
 137     } sh;
 138 } stream_header;
 139
 140 /* Some defines from OggDS */
 141 #define PACKET_TYPE_HEADER   0x01
 142 #define PACKET_TYPE_BITS     0x07
 143 #define PACKET_LEN_BITS01    0xc0
 144 #define PACKET_LEN_BITS2     0x02
 145 #define PACKET_IS_SYNCPOINT  0x08
 146
 147 /* Some functions to manipulate memory */
 148 static uint16_t GetWLE( uint8_t *p_buff )
 149 {
 150     return( (p_buff[0]) + ( p_buff[1] <<8 ) );
 151 }
 152
 153 static uint32_t GetDWLE( uint8_t *p_buff )
 154 {
 155     return( p_buff[0] + ( p_buff[1] <<8 ) +
 156             ( p_buff[2] <<16 ) + ( p_buff[3] <<24 ) );
 157 }
 158
 159 static uint64_t GetQWLE( uint8_t *p_buff )
 160 {
 161     return( GetDWLE( p_buff ) + ( ((uint64_t)GetDWLE( p_buff + 4 )) << 32 ) );
 162 }
 163 /*****************************************************************************
 164  * Local prototypes
 165  *****************************************************************************/
 166 static int  Activate  ( vlc_object_t * );
 167 static void Deactivate( vlc_object_t * );
 168 static int  Demux     ( input_thread_t * );
 169
 170 /* Stream managment */
 171 static int  Ogg_StreamStart  ( input_thread_t *, demux_sys_t *, int );
 172 static void Ogg_StreamStop   ( input_thread_t *, demux_sys_t *, int );
 173
 174 /* Bitstream manipulation */
 175 static int  Ogg_Check        ( input_thread_t *p_input );
 176 static int  Ogg_ReadPage     ( input_thread_t *, demux_sys_t *, ogg_page * );
 177 static void Ogg_UpdatePCR    ( logical_stream_t *, ogg_packet * );
 178 static void Ogg_DecodePacket ( input_thread_t *p_input,
 179                                logical_stream_t *p_stream, ogg_packet * );
 180 static int  Ogg_FindLogicalStreams( input_thread_t *p_input,
 181                                     demux_sys_t *p_ogg );
 182
 183 /*****************************************************************************
 184  * Module descriptor
 185  *****************************************************************************/
 186 vlc_module_begin();
 187     set_description( _("ogg stream demux" ) );
 188     set_capability( "demux", 50 );
 189     set_callbacks( Activate, Deactivate );
 190     add_shortcut( "ogg" );
 191 vlc_module_end();
 192
 193 /*****************************************************************************
 194  * Stream managment
 195  *****************************************************************************/
 196 static int Ogg_StreamStart( input_thread_t *p_input,
 197                             demux_sys_t *p_ogg, int i_stream )
 198 {
 199 #define p_stream p_ogg->pp_stream[i_stream]
 200     if( !p_stream->p_es )
 201     {
 202         msg_Warn( p_input, "stream[%d] unselectable", i_stream );
 203         return( 0 );
 204     }
 205     if( p_stream->i_activated )
 206     {
 207         msg_Warn( p_input, "stream[%d] already selected", i_stream );
 208         return( 1 );
 209     }
 210
 211     if( !p_stream->p_es->p_decoder_fifo )
 212     {
 213         vlc_mutex_lock( &p_input->stream.stream_lock );
 214         input_SelectES( p_input, p_stream->p_es );
 215         vlc_mutex_unlock( &p_input->stream.stream_lock );
 216     }
 217     p_stream->i_activated = p_stream->p_es->p_decoder_fifo ? 1 : 0;
 218
 219     /* Feed the backup header to the decoder */
 220     if( !p_stream->b_force_backup )
 221     {
 222         int i;
 223         for( i = 0; i < p_stream->i_packets_backup; i++ )
 224         {
 225             Ogg_DecodePacket( p_input, p_stream,
 226                               &p_stream->p_packets_backup[i] );
 227         }
 228     }
 229
 230     return( p_stream->i_activated );
 231 #undef  p_stream
 232 }
 233
 234 static void Ogg_StreamStop( input_thread_t *p_input,
 235                             demux_sys_t *p_ogg, int i_stream )
 236 {
 237 #define p_stream    p_ogg->pp_stream[i_stream]
 238
 239     if( !p_stream->i_activated )
 240     {
 241         msg_Warn( p_input, "stream[%d] already unselected", i_stream );
 242         return;
 243     }
 244
 245     if( p_stream->p_es->p_decoder_fifo )
 246     {
 247         vlc_mutex_lock( &p_input->stream.stream_lock );
 248         input_UnselectES( p_input, p_stream->p_es );
 249         vlc_mutex_unlock( &p_input->stream.stream_lock );
 250     }
 251
 252     p_stream->i_activated = 0;
 253
 254 #undef  p_stream
 255 }
 256
 257 /****************************************************************************
 258  * Ogg_Check: Check we are dealing with an ogg stream.
 259  ****************************************************************************/
 260 static int Ogg_Check( input_thread_t *p_input )
 261 {
 262     u8 *p_peek;
 263     int i_size = input_Peek( p_input, &p_peek, 4 );
 264
 265     /* Check for the Ogg capture pattern */
 266     if( !(i_size>3) || !(p_peek[0] == 'O') || !(p_peek[1] == 'g') ||
 267         !(p_peek[2] == 'g') || !(p_peek[3] == 'S') )
 268         return VLC_EGENERIC;
 269
 270     /* FIXME: Capture pattern might not be enough so we can also check for the
 271      * the first complete page */
 272
 273     return VLC_SUCCESS;
 274 }
 275
 276 /****************************************************************************
 277  * Ogg_ReadPage: Read a full Ogg page from the physical bitstream.
 278  ****************************************************************************
 279  * Returns VLC_SUCCESS if a page has been read. An error might happen if we
 280  * are at the end of stream.
 281  ****************************************************************************/
 282 static int Ogg_ReadPage( input_thread_t *p_input, demux_sys_t *p_ogg,
 283                          ogg_page *p_oggpage )
 284 {
 285     int i_read = 0;
 286     data_packet_t *p_data;
 287     byte_t *p_buffer;
 288
 289     while( ogg_sync_pageout( &p_ogg->oy, p_oggpage ) != 1 )
 290     {
 291         i_read = input_SplitBuffer( p_input, &p_data, OGG_BLOCK_SIZE );
 292         if( i_read <= 0 )
 293             return VLC_EGENERIC;
 294
 295         p_buffer = ogg_sync_buffer( &p_ogg->oy, i_read );
 296         p_input->p_vlc->pf_memcpy( p_buffer, p_data->p_payload_start, i_read );
 297         ogg_sync_wrote( &p_ogg->oy, i_read );
 298         input_DeletePacket( p_input->p_method_data, p_data );
 299     }
 300
 301     return VLC_SUCCESS;
 302 }
 303
 304 /****************************************************************************
 305  * Ogg_UpdatePCR: update the PCR (90kHz program clock reference) for the
 306  *                current stream.
 307  ****************************************************************************/
 308 static void Ogg_UpdatePCR( logical_stream_t *p_stream,
 309                            ogg_packet *p_oggpacket )
 310 {
 311
 312     /* Convert the next granulepos into a pcr */
 313     if( p_oggpacket->granulepos >= 0 )
 314     {
 315         if( p_stream->i_fourcc != VLC_FOURCC( 't','h','e','o' ) )
 316         {
 317             p_stream->i_pcr = p_oggpacket->granulepos * 90000
 318                               / p_stream->f_rate;
 319         }
 320         else
 321         {
 322             ogg_int64_t iframe = p_oggpacket->granulepos >>
 323               p_stream->i_theora_keyframe_granule_shift;
 324             ogg_int64_t pframe = p_oggpacket->granulepos -
 325               ( iframe << p_stream->i_theora_keyframe_granule_shift );
 326
 327             p_stream->i_pcr = ( iframe + pframe ) * 90000
 328                               / p_stream->f_rate;
 329         }
 330
 331         p_stream->i_interpolated_pcr = p_stream->i_pcr;
 332     }
 333     else
 334     {
 335         /* FIXME: ffmpeg doesn't like null pts */
 336         if( p_stream->i_cat == VIDEO_ES )
 337             /* 1 frame per packet */
 338             p_stream->i_pcr += (90000 / p_stream->f_rate);
 339         else
 340             p_stream->i_pcr = -1;
 341
 342         /* no granulepos available, try to interpolate the pcr.
 343          * If we can't then don't touch the old value. */
 344         if( p_stream->i_bitrate )
 345             p_stream->i_interpolated_pcr += ( p_oggpacket->bytes * 90000
 346                                               / p_stream->i_bitrate / 8 );
 347     }
 348 }
 349
 350 /****************************************************************************
 351  * Ogg_DecodePacket: Decode an Ogg packet.
 352  ****************************************************************************/
 353 static void Ogg_DecodePacket( input_thread_t *p_input,
 354                               logical_stream_t *p_stream,
 355                               ogg_packet *p_oggpacket )
 356 {
 357     pes_packet_t  *p_pes;
 358     data_packet_t *p_data;
 359     vlc_bool_t b_trash = VLC_FALSE;
 360     int i_header_len = 0;
 361
 362     if( p_stream->b_force_backup )
 363     {
 364         /* Backup the ogg packet (likely an header packet) */
 365         ogg_packet *p_packet_backup;
 366         p_stream->i_packets_backup++;
 367         p_stream->p_packets_backup =
 368             realloc( p_stream->p_packets_backup, p_stream->i_packets_backup *
 369                      sizeof(ogg_packet) );
 370
 371         p_packet_backup =
 372             &p_stream->p_packets_backup[p_stream->i_packets_backup - 1];
 373
 374         p_packet_backup->bytes = p_oggpacket->bytes;
 375         p_packet_backup->granulepos = p_oggpacket->granulepos;
 376         p_packet_backup->packet = malloc( p_oggpacket->bytes );
 377         if( !p_packet_backup->packet ) return;
 378         memcpy( p_packet_backup->packet, p_oggpacket->packet,
 379                 p_oggpacket->bytes );
 380
 381         switch( p_stream->i_fourcc )
 382         {
 383         case VLC_FOURCC( 'v','o','r','b' ):
 384           if( p_stream->i_packets_backup == 3 ) p_stream->b_force_backup = 0;
 385           break;
 386
 387         default:
 388           p_stream->b_force_backup = 0;
 389           break;
 390         }
 391     }
 392
 393     vlc_mutex_lock( &p_input->stream.control.control_lock );
 394     if( p_stream->i_cat == AUDIO_ES && p_input->stream.control.b_mute )
 395     {
 396         b_trash = VLC_TRUE;
 397     }
 398     vlc_mutex_unlock( &p_input->stream.control.control_lock );
 399
 400     if( !p_stream->p_es->p_decoder_fifo || b_trash )
 401     {
 402         /* This stream isn't currently selected so we don't need to decode it,
 403          * but we do need to store its pcr as it might be selected later on. */
 404         Ogg_UpdatePCR( p_stream, p_oggpacket );
 405
 406         return;
 407     }
 408
 409     if( !( p_pes = input_NewPES( p_input->p_method_data ) ) )
 410     {
 411         return;
 412     }
 413     if( !( p_data = input_NewPacket( p_input->p_method_data,
 414                                      p_oggpacket->bytes ) ) )
 415     {
 416         input_DeletePES( p_input->p_method_data, p_pes );
 417         return;
 418     }
 419
 420     /* Convert the pcr into a pts */
 421     if( p_stream->i_cat != SPU_ES )
 422     {
 423         p_pes->i_pts = ( p_stream->i_pcr < 0 ) ? 0 :
 424             input_ClockGetTS( p_input, p_input->stream.p_selected_program,
 425                               p_stream->i_pcr );
 426     }
 427     else
 428     {
 429         /* Of course subtitles had to be different! */
 430         p_pes->i_pts = ( p_oggpacket->granulepos < 0 ) ? 0 :
 431             input_ClockGetTS( p_input, p_input->stream.p_selected_program,
 432                               p_oggpacket->granulepos * 90000 /
 433                               p_stream->f_rate );
 434     }
 435
 436     /* Convert the next granulepos into a pcr */
 437     Ogg_UpdatePCR( p_stream, p_oggpacket );
 438
 439     p_pes->i_nb_data = 1;
 440     p_pes->i_dts = p_oggpacket->granulepos;
 441     p_pes->p_first = p_pes->p_last = p_data;
 442     p_pes->i_pes_size = p_oggpacket->bytes;
 443
 444     if( p_stream->i_fourcc != VLC_FOURCC( 'v','o','r','b' ) &&
 445         p_stream->i_fourcc != VLC_FOURCC( 't','a','r','k' ) &&
 446         p_stream->i_fourcc != VLC_FOURCC( 't','h','e','o' ) )
 447     {
 448         /* Remove the header from the packet */
 449         i_header_len = (*p_oggpacket->packet & PACKET_LEN_BITS01) >> 6;
 450         i_header_len |= (*p_oggpacket->packet & PACKET_LEN_BITS2) << 1;
 451         i_header_len++;
 452
 453         p_pes->i_pes_size -= i_header_len;
 454         p_pes->i_dts = 0;
 455     }
 456
 457     if( p_stream->i_fourcc == VLC_FOURCC( 't','a','r','k' ) )
 458     {
 459         /* FIXME: the biggest hack I've ever done */
 460         msg_Warn( p_input, "tark pts: %lli, granule: %i",
 461                   p_pes->i_pts, p_pes->i_dts );
 462         msleep(10000);
 463     }
 464
 465     memcpy( p_data->p_payload_start,
 466             p_oggpacket->packet + i_header_len,
 467             p_oggpacket->bytes - i_header_len );
 468
 469     p_data->p_payload_end = p_data->p_payload_start + p_pes->i_pes_size;
 470     p_data->b_discard_payload = 0;
 471
 472     input_DecodePES( p_stream->p_es->p_decoder_fifo, p_pes );
 473 }
 474
 475 /****************************************************************************
 476  * Ogg_FindLogicalStreams: Find the logical streams embedded in the physical
 477  *                         stream and fill p_ogg.
 478  *****************************************************************************
 479  * The initial page of a logical stream is marked as a 'bos' page.
 480  * Furthermore, the Ogg specification mandates that grouped bitstreams begin
 481  * together and all of the initial pages must appear before any data pages.
 482  *
 483  * On success this function returns VLC_SUCCESS.
 484  ****************************************************************************/
 485 static int Ogg_FindLogicalStreams( input_thread_t *p_input, demux_sys_t *p_ogg)
 486 {
 487     ogg_packet oggpacket;
 488     ogg_page oggpage;
 489     int i_stream;
 490
 491     while( Ogg_ReadPage( p_input, p_ogg, &oggpage ) == VLC_SUCCESS )
 492     {
 493         if( ogg_page_bos( &oggpage ) )
 494         {
 495
 496             /* All is wonderful in our fine fine little world.
 497              * We found the beginning of our first logical stream. */
 498             while( ogg_page_bos( &oggpage ) )
 499             {
 500                 p_ogg->i_streams++;
 501                 p_ogg->pp_stream =
 502                     realloc( p_ogg->pp_stream, p_ogg->i_streams *
 503                              sizeof(logical_stream_t *) );
 504
 505 #define p_stream p_ogg->pp_stream[p_ogg->i_streams - 1]
 506
 507                 p_stream = malloc( sizeof(logical_stream_t) );
 508                 memset( p_stream, 0, sizeof(logical_stream_t) );
 509
 510                 /* Setup the logical stream */
 511                 p_stream->i_serial_no = ogg_page_serialno( &oggpage );
 512                 ogg_stream_init( &p_stream->os, p_stream->i_serial_no );
 513
 514                 /* Extract the initial header from the first page and verify
 515                  * the codec type of tis Ogg bitstream */
 516                 if( ogg_stream_pagein( &p_stream->os, &oggpage ) < 0 )
 517                 {
 518                     /* error. stream version mismatch perhaps */
 519                     msg_Err( p_input, "Error reading first page of "
 520                              "Ogg bitstream data" );
 521                     return VLC_EGENERIC;
 522                 }
 523
 524                 /* FIXME: check return value */
 525                 ogg_stream_packetpeek( &p_stream->os, &oggpacket );
 526
 527                 /* Check for Vorbis header */
 528                 if( oggpacket.bytes >= 7 &&
 529                     ! strncmp( &oggpacket.packet[1], "vorbis", 6 ) )
 530                 {
 531                     oggpack_buffer opb;
 532
 533                     msg_Dbg( p_input, "found vorbis header" );
 534                     p_stream->i_cat = AUDIO_ES;
 535                     p_stream->i_fourcc = VLC_FOURCC( 'v','o','r','b' );
 536
 537                     /* Signal that we want to keep a backup of the vorbis
 538                      * stream headers. They will be used when switching between
 539                      * audio streams. */
 540                     p_stream->b_force_backup = 1;
 541
 542                     /* Cheat and get additionnal info ;) */
 543                     oggpack_readinit( &opb, oggpacket.packet, oggpacket.bytes);
 544                     oggpack_adv( &opb, 96 );
 545                     p_stream->f_rate = oggpack_read( &opb, 32 );
 546                     oggpack_adv( &opb, 32 );
 547                     p_stream->i_bitrate = oggpack_read( &opb, 32 );
 548                 }
 549                 /* Check for Theora header */
 550                 else if( oggpacket.bytes >= 7 &&
 551                          ! strncmp( &oggpacket.packet[1], "theora", 6 ) )
 552                 {
 553 #ifdef HAVE_OGGPACKB
 554                     oggpack_buffer opb;
 555                     int i_fps_numerator;
 556                     int i_fps_denominator;
 557                     int i_keyframe_frequency_force;
 558 #endif
 559
 560                     msg_Dbg( p_input, "found theora header" );
 561 #ifdef HAVE_OGGPACKB
 562                     p_stream->i_cat = VIDEO_ES;
 563                     p_stream->i_fourcc = VLC_FOURCC( 't','h','e','o' );
 564
 565                     /* Cheat and get additionnal info ;) */
 566                     oggpackB_readinit(&opb, oggpacket.packet, oggpacket.bytes);
 567                     oggpackB_adv( &opb, 56 );
 568                     oggpackB_read( &opb, 8 ); /* major version num */
 569                     oggpackB_read( &opb, 8 ); /* minor version num */
 570                     oggpackB_read( &opb, 8 ); /* subminor version num */
 571                     oggpackB_read( &opb, 16 ) /*<< 4*/; /* width */
 572                     oggpackB_read( &opb, 16 ) /*<< 4*/; /* height */
 573                     i_fps_numerator = oggpackB_read( &opb, 32 );
 574                     i_fps_denominator = oggpackB_read( &opb, 32 );
 575                     oggpackB_read( &opb, 24 ); /* aspect_numerator */
 576                     oggpackB_read( &opb, 24 ); /* aspect_denominator */
 577                     i_keyframe_frequency_force = 1 << oggpackB_read( &opb, 5 );
 578                     p_stream->i_bitrate = oggpackB_read( &opb, 24 );
 579                     oggpackB_read(&opb,6); /* quality */
 580
 581                     /* granule_shift = i_log( frequency_force -1 ) */
 582                     p_stream->i_theora_keyframe_granule_shift = 0;
 583                     i_keyframe_frequency_force--;
 584                     while( i_keyframe_frequency_force )
 585                     {
 586                         p_stream->i_theora_keyframe_granule_shift++;
 587                         i_keyframe_frequency_force >>= 1;
 588                     }
 589
 590                     p_stream->f_rate = (float)i_fps_numerator /
 591                                                 i_fps_denominator;
 592                     msg_Dbg( p_input,
 593                              "found theora header, bitrate: %i, rate: %f",
 594                              p_stream->i_bitrate, p_stream->f_rate );
 595 #else /* HAVE_OGGPACKB */
 596                     msg_Dbg( p_input, "the ogg demuxer has been compiled "
 597                              "without support for the oggpackB extension."
 598                              "The theora stream won't be decoded." );
 599                     free( p_stream );
 600                     p_ogg->i_streams--;
 601                     continue;
 602 #endif /* HAVE_OGGPACKB */
 603                 }
 604                 /* Check for Tarkin header */
 605                 else if( oggpacket.bytes >= 7 &&
 606                          ! strncmp( &oggpacket.packet[1], "tarkin", 6 ) )
 607                 {
 608                     oggpack_buffer opb;
 609
 610                     msg_Dbg( p_input, "found tarkin header" );
 611                     p_stream->i_cat = VIDEO_ES;
 612                     p_stream->i_fourcc = VLC_FOURCC( 't','a','r','k' );
 613
 614                     /* Cheat and get additionnal info ;) */
 615                     oggpack_readinit( &opb, oggpacket.packet, oggpacket.bytes);
 616                     oggpack_adv( &opb, 88 );
 617                     oggpack_adv( &opb, 104 );
 618                     p_stream->i_bitrate = oggpack_read( &opb, 32 );
 619                     p_stream->f_rate = 2; /* FIXME */
 620                     msg_Dbg( p_input,
 621                              "found tarkin header, bitrate: %i, rate: %f",
 622                              p_stream->i_bitrate, p_stream->f_rate );
 623                 }
 624                 else if( (*oggpacket.packet & PACKET_TYPE_BITS )
 625                          == PACKET_TYPE_HEADER &&
 626                          oggpacket.bytes >= (int)sizeof(stream_header)+1 )
 627                 {
 628                     stream_header *st = (stream_header *)(oggpacket.packet+1);
 629
 630                     /* Check for video header (new format) */
 631                     if( !strncmp( st->streamtype, "video", 5 ) )
 632                     {
 633                         p_stream->i_cat = VIDEO_ES;
 634
 635                         /* We need to get rid of the header packet */
 636                         ogg_stream_packetout( &p_stream->os, &oggpacket );
 637
 638                         p_stream->p_bih = (BITMAPINFOHEADER *)
 639                             malloc( sizeof(BITMAPINFOHEADER) );
 640                         if( !p_stream->p_bih )
 641                         {
 642                             /* Mem allocation error, just ignore the stream */
 643                             free( p_stream );
 644                             p_ogg->i_streams--;
 645                             continue;
 646                         }
 647                         p_stream->p_bih->biSize = sizeof(BITMAPINFOHEADER);
 648                         p_stream->p_bih->biCompression=
 649                             p_stream->i_fourcc = VLC_FOURCC( st->subtype[0],
 650                                                              st->subtype[1],
 651                                                              st->subtype[2],
 652                                                              st->subtype[3] );
 653                         msg_Dbg( p_input, "found video header of type: %.4s",
 654                                  (char *)&p_stream->i_fourcc );
 655
 656                         p_stream->f_rate = 10000000.0 /
 657                             GetQWLE((uint8_t *)&st->time_unit);
 658                         p_stream->p_bih->biBitCount =
 659                             GetWLE((uint8_t *)&st->bits_per_sample);
 660                         p_stream->p_bih->biWidth =
 661                             GetDWLE((uint8_t *)&st->sh.video.width);
 662                         p_stream->p_bih->biHeight =
 663                             GetDWLE((uint8_t *)&st->sh.video.height);
 664                         p_stream->p_bih->biPlanes= 1 ;
 665                         p_stream->p_bih->biSizeImage =
 666                             (p_stream->p_bih->biBitCount >> 3) *
 667                             p_stream->p_bih->biWidth *
 668                             p_stream->p_bih->biHeight;
 669
 670                         msg_Dbg( p_input,
 671                              "fps: %f, width:%i; height:%i, bitcount:%i",
 672                             p_stream->f_rate, p_stream->p_bih->biWidth,
 673                             p_stream->p_bih->biHeight,
 674                             p_stream->p_bih->biBitCount);
 675
 676                         p_stream->i_bitrate = 0;
 677                     }
 678                     /* Check for audio header (new format) */
 679                     else if( !strncmp( st->streamtype, "audio", 5 ) )
 680                     {
 681                         char p_buffer[5];
 682
 683                         p_stream->i_cat = AUDIO_ES;
 684
 685                         /* We need to get rid of the header packet */
 686                         ogg_stream_packetout( &p_stream->os, &oggpacket );
 687
 688                         p_stream->p_wf = (WAVEFORMATEX *)
 689                             malloc( sizeof(WAVEFORMATEX) );
 690                         if( !p_stream->p_wf )
 691                         {
 692                             /* Mem allocation error, just ignore the stream */
 693                             free( p_stream );
 694                             p_ogg->i_streams--;
 695                             continue;
 696                         }
 697
 698                         memcpy( p_buffer, st->subtype, 4 );
 699                         p_buffer[4] = '\0';
 700                         p_stream->p_wf->wFormatTag = strtol(p_buffer,NULL,16);
 701                         p_stream->p_wf->nChannels =
 702                             GetWLE((uint8_t *)&st->sh.audio.channels);
 703                         p_stream->f_rate = p_stream->p_wf->nSamplesPerSec =
 704                             GetQWLE((uint8_t *)&st->samples_per_unit);
 705                         p_stream->i_bitrate = p_stream->p_wf->nAvgBytesPerSec =
 706                             GetDWLE((uint8_t *)&st->sh.audio.avgbytespersec);
 707                         p_stream->i_bitrate *= 8;
 708                         p_stream->p_wf->nBlockAlign =
 709                             GetWLE((uint8_t *)&st->sh.audio.blockalign);
 710                         p_stream->p_wf->wBitsPerSample =
 711                             GetWLE((uint8_t *)&st->bits_per_sample);
 712                         p_stream->p_wf->cbSize = 0;
 713
 714                         switch( p_stream->p_wf->wFormatTag )
 715                         {
 716                         case WAVE_FORMAT_PCM:
 717                             p_stream->i_fourcc =
 718                                 VLC_FOURCC( 'a', 'r', 'a', 'w' );
 719                             break;
 720                         case WAVE_FORMAT_MPEG:
 721                         case WAVE_FORMAT_MPEGLAYER3:
 722                             p_stream->i_fourcc =
 723                                 VLC_FOURCC( 'm', 'p', 'g', 'a' );
 724                             break;
 725                         case WAVE_FORMAT_A52:
 726                             p_stream->i_fourcc =
 727                                 VLC_FOURCC( 'a', '5', '2', ' ' );
 728                             break;
 729                         case WAVE_FORMAT_WMA1:
 730                             p_stream->i_fourcc =
 731                                 VLC_FOURCC( 'w', 'm', 'a', '1' );
 732                             break;
 733                         case WAVE_FORMAT_WMA2:
 734                             p_stream->i_fourcc =
 735                                 VLC_FOURCC( 'w', 'm', 'a', '2' );
 736                             break;
 737                         default:
 738                             p_stream->i_fourcc = VLC_FOURCC( 'm', 's',
 739                                 ( p_stream->p_wf->wFormatTag >> 8 ) & 0xff,
 740                                 p_stream->p_wf->wFormatTag & 0xff );
 741                         }
 742
 743                         msg_Dbg( p_input, "found audio header of type: %.4s",
 744                                  (char *)&p_stream->i_fourcc );
 745                         msg_Dbg( p_input, "audio:0x%4.4x channels:%d %dHz "
 746                                  "%dbits/sample %dkb/s",
 747                                  p_stream->p_wf->wFormatTag,
 748                                  p_stream->p_wf->nChannels,
 749                                  p_stream->p_wf->nSamplesPerSec,
 750                                  p_stream->p_wf->wBitsPerSample,
 751                                  p_stream->p_wf->nAvgBytesPerSec * 8 / 1024 );
 752                     }
 753                     /* Check for text (subtitles) header */
 754                     else if( !strncmp(st->streamtype, "text", 4) )
 755                     {
 756                         /* We need to get rid of the header packet */
 757                         ogg_stream_packetout( &p_stream->os, &oggpacket );
 758
 759                         msg_Dbg( p_input, "found text subtitles header" );
 760                         p_stream->i_cat = SPU_ES;
 761                         p_stream->i_fourcc =
 762                             VLC_FOURCC( 's', 'u', 'b', 't' );
 763                         p_stream->f_rate = 1000; /* granulepos is in milisec */
 764                     }
 765                     else
 766                     {
 767                         msg_Dbg( p_input, "stream %d has a header marker "
 768                             "but is of an unknown type", p_ogg->i_streams-1 );
 769                         free( p_stream );
 770                         p_ogg->i_streams--;
 771                     }
 772                 }
 773                 else
 774                 {
 775                     msg_Dbg( p_input, "stream %d is of unknown type",
 776                              p_ogg->i_streams-1 );
 777                     free( p_stream );
 778                     p_ogg->i_streams--;
 779                 }
 780
 781 #undef p_stream
 782
 783                 if( Ogg_ReadPage( p_input, p_ogg, &oggpage ) != VLC_SUCCESS )
 784                     return VLC_EGENERIC;
 785             }
 786
 787             /* This is the first data page, which means we are now finished
 788              * with the initial pages. We just need to store it in the relevant
 789              * bitstream. */
 790             for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
 791             {
 792                 if( ogg_stream_pagein( &p_ogg->pp_stream[i_stream]->os,
 793                                        &oggpage ) == 0 )
 794                 {
 795                     break;
 796                 }
 797             }
 798             return VLC_SUCCESS;
 799         }
 800     }
 801     return VLC_EGENERIC;
 802 }
 803
 804 /*****************************************************************************
 805  * Activate: initializes ogg demux structures
 806  *****************************************************************************/
 807 static int Activate( vlc_object_t * p_this )
 808 {
 809     int i_stream, b_forced;
 810     demux_sys_t    *p_ogg;
 811     input_thread_t *p_input = (input_thread_t *)p_this;
 812
 813     p_input->p_demux_data = NULL;
 814     b_forced = ( ( *p_input->psz_demux )&&
 815                  ( !strncmp( p_input->psz_demux, "ogg", 10 ) ) ) ? 1 : 0;
 816
 817     /* Check if we are dealing with an ogg stream */
 818     if( !b_forced && ( Ogg_Check( p_input ) != VLC_SUCCESS ) )
 819         return -1;
 820
 821     /* Allocate p_ogg */
 822     if( !( p_ogg = malloc( sizeof( demux_sys_t ) ) ) )
 823     {
 824         msg_Err( p_input, "out of memory" );
 825         goto error;
 826     }
 827     memset( p_ogg, 0, sizeof( demux_sys_t ) );
 828     p_input->p_demux_data = p_ogg;
 829
 830     p_ogg->i_pcr  = 0;
 831     p_ogg->b_seekable = ( ( p_input->stream.b_seekable )
 832                         &&( p_input->stream.i_method == INPUT_METHOD_FILE ) );
 833
 834     /* Initialize the Ogg physical bitstream parser */
 835     ogg_sync_init( &p_ogg->oy );
 836
 837     /* Find the logical streams embedded in the physical stream and
 838      * initialize our p_ogg structure. */
 839     if( Ogg_FindLogicalStreams( p_input, p_ogg ) != VLC_SUCCESS )
 840     {
 841         msg_Err( p_input, "couldn't find an ogg logical stream" );
 842         goto error;
 843     }
 844
 845     /* Set the demux function */
 846     p_input->pf_demux = Demux;
 847
 848     /* Initialize access plug-in structures. */
 849     if( p_input->i_mtu == 0 )
 850     {
 851         /* Improve speed. */
 852         p_input->i_bufsize = INPUT_DEFAULT_BUFSIZE;
 853     }
 854
 855     /* Create one program */
 856     vlc_mutex_lock( &p_input->stream.stream_lock );
 857     if( input_InitStream( p_input, 0 ) == -1)
 858     {
 859         vlc_mutex_unlock( &p_input->stream.stream_lock );
 860         msg_Err( p_input, "cannot init stream" );
 861         goto error;
 862     }
 863     if( input_AddProgram( p_input, 0, 0) == NULL )
 864     {
 865         vlc_mutex_unlock( &p_input->stream.stream_lock );
 866         msg_Err( p_input, "cannot add program" );
 867         goto error;
 868     }
 869     p_input->stream.p_selected_program = p_input->stream.pp_programs[0];
 870     p_input->stream.i_mux_rate = 0;
 871     vlc_mutex_unlock( &p_input->stream.stream_lock );
 872
 873     for( i_stream = 0 ; i_stream < p_ogg->i_streams; i_stream++ )
 874     {
 875 #define p_stream p_ogg->pp_stream[i_stream]
 876         vlc_mutex_lock( &p_input->stream.stream_lock );
 877         p_stream->p_es = input_AddES( p_input,
 878                                       p_input->stream.p_selected_program,
 879                                       p_ogg->i_streams + 1, 0 );
 880         p_input->stream.i_mux_rate += (p_stream->i_bitrate / ( 8 * 50 ));
 881         vlc_mutex_unlock( &p_input->stream.stream_lock );
 882         p_stream->p_es->i_stream_id = p_stream->p_es->i_id = i_stream;
 883         p_stream->p_es->i_fourcc = p_stream->i_fourcc;
 884         p_stream->p_es->i_cat = p_stream->i_cat;
 885         p_stream->p_es->p_demux_data = p_stream->p_bih ?
 886             (void *)p_stream->p_bih : (void *)p_stream->p_wf;
 887 #undef p_stream
 888     }
 889
 890     for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
 891     {
 892 #define p_stream  p_ogg->pp_stream[i_stream]
 893         switch( p_stream->p_es->i_cat )
 894         {
 895             case( VIDEO_ES ):
 896                 if( (p_ogg->p_stream_video == NULL) )
 897                 {
 898                     p_ogg->p_stream_video = p_stream;
 899                     /* TODO add test to see if a decoder has been found */
 900                     Ogg_StreamStart( p_input, p_ogg, i_stream );
 901                 }
 902                 break;
 903
 904             case( AUDIO_ES ):
 905                 if( (p_ogg->p_stream_audio == NULL) )
 906                 {
 907                     int i_audio = config_GetInt( p_input, "audio-channel" );
 908                     if( i_audio == i_stream || i_audio <= 0 ||
 909                         i_audio >= p_ogg->i_streams ||
 910                         p_ogg->pp_stream[i_audio]->p_es->i_cat != AUDIO_ES )
 911                     {
 912                         p_ogg->p_stream_audio = p_stream;
 913                         Ogg_StreamStart( p_input, p_ogg, i_stream );
 914                     }
 915                 }
 916                 break;
 917
 918             case( SPU_ES ):
 919                 if( (p_ogg->p_stream_spu == NULL) )
 920                 {
 921                     /* for spu, default is none */
 922                     int i_spu = config_GetInt( p_input, "spu-channel" );
 923                     if( i_spu < 0 || i_spu >= p_ogg->i_streams ||
 924                         p_ogg->pp_stream[i_spu]->p_es->i_cat != SPU_ES )
 925                     {
 926                         break;
 927                     }
 928                     else if( i_spu == i_stream )
 929                     {
 930                         p_ogg->p_stream_spu = p_stream;
 931                         Ogg_StreamStart( p_input, p_ogg, i_stream );
 932                     }
 933                 }
 934                 break;
 935
 936             default:
 937                 break;
 938         }
 939 #undef p_stream
 940     }
 941
 942     /* we select the first audio and video ES */
 943     vlc_mutex_lock( &p_input->stream.stream_lock );
 944     if( !p_ogg->p_stream_video )
 945     {
 946         msg_Warn( p_input, "no video stream found" );
 947     }
 948     if( !p_ogg->p_stream_audio )
 949     {
 950         msg_Warn( p_input, "no audio stream found!" );
 951     }
 952     p_input->stream.p_selected_program->b_is_ok = 1;
 953     vlc_mutex_unlock( &p_input->stream.stream_lock );
 954
 955     /* Call the pace control */
 956     input_ClockManageRef( p_input,
 957                           p_input->stream.p_selected_program,
 958                           p_ogg->i_pcr );
 959
 960     return 0;
 961
 962  error:
 963     Deactivate( (vlc_object_t *)p_input );
 964     return -1;
 965
 966 }
 967
 968 /*****************************************************************************
 969  * Deactivate: frees unused data
 970  *****************************************************************************/
 971 static void Deactivate( vlc_object_t *p_this )
 972 {
 973     input_thread_t *p_input = (input_thread_t *)p_this;
 974     demux_sys_t *p_ogg = (demux_sys_t *)p_input->p_demux_data  ;
 975     int i, j;
 976
 977     if( p_ogg )
 978     {
 979         /* Cleanup the bitstream parser */
 980         ogg_sync_clear( &p_ogg->oy );
 981
 982         for( i = 0; i < p_ogg->i_streams; i++ )
 983         {
 984             ogg_stream_clear( &p_ogg->pp_stream[i]->os );
 985             for( j = 0; j < p_ogg->pp_stream[i]->i_packets_backup; j++ )
 986             {
 987                 free( p_ogg->pp_stream[i]->p_packets_backup[j].packet );
 988             }
 989             if( p_ogg->pp_stream[i]->p_packets_backup)
 990                 free( p_ogg->pp_stream[i]->p_packets_backup );
 991 #if 0 /* hmmm, it's already freed in input_DelES() */
 992             if( p_ogg->pp_stream[i]->p_bih )
 993                 free( p_ogg->pp_stream[i]->p_bih );
 994             if( p_ogg->pp_stream[i]->p_wf )
 995                 free( p_ogg->pp_stream[i]->p_wf );
 996 #endif
 997             free( p_ogg->pp_stream[i] );
 998         }
 999         if( p_ogg->pp_stream ) free( p_ogg->pp_stream );
1000
1001         free( p_ogg );
1002     }
1003 }
1004
1005 /*****************************************************************************
1006  * Demux: reads and demuxes data packets
1007  *****************************************************************************
1008  * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
1009  *****************************************************************************/
1010 static int Demux( input_thread_t * p_input )
1011 {
1012     int i, i_stream, b_eos = 0;
1013     ogg_page    oggpage;
1014     ogg_packet  oggpacket;
1015     demux_sys_t *p_ogg  = (demux_sys_t *)p_input->p_demux_data;
1016
1017 #define p_stream p_ogg->pp_stream[i_stream]
1018     /* detect new selected/unselected streams */
1019     for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
1020     {
1021         if( p_stream->p_es )
1022         {
1023             if( p_stream->p_es->p_decoder_fifo &&
1024                 !p_stream->i_activated )
1025             {
1026                 Ogg_StreamStart( p_input, p_ogg, i_stream );
1027             }
1028             else
1029             if( !p_stream->p_es->p_decoder_fifo &&
1030                 p_stream->i_activated )
1031             {
1032                 Ogg_StreamStop( p_input, p_ogg, i_stream );
1033             }
1034         }
1035     }
1036
1037     /* search for new video and audio stream to select
1038      * if current have been unselected */
1039     if( ( !p_ogg->p_stream_video )
1040             || ( !p_ogg->p_stream_video->p_es->p_decoder_fifo ) )
1041     {
1042         p_ogg->p_stream_video = NULL;
1043         for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
1044         {
1045             if( ( p_stream->i_cat == VIDEO_ES )
1046                   &&( p_stream->p_es->p_decoder_fifo ) )
1047             {
1048                 p_ogg->p_stream_video = p_stream;
1049                 break;
1050             }
1051         }
1052     }
1053     if( ( !p_ogg->p_stream_audio )
1054             ||( !p_ogg->p_stream_audio->p_es->p_decoder_fifo ) )
1055     {
1056         p_ogg->p_stream_audio = NULL;
1057         for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
1058         {
1059             if( ( p_stream->i_cat == AUDIO_ES )
1060                   &&( p_stream->p_es->p_decoder_fifo ) )
1061             {
1062                 p_ogg->p_stream_audio = p_stream;
1063                 break;
1064             }
1065         }
1066     }
1067
1068     if( p_input->stream.p_selected_program->i_synchro_state == SYNCHRO_REINIT )
1069     {
1070         msg_Warn( p_input, "synchro reinit" );
1071
1072         /* An ogg packet does only contain the starting date of the next
1073          * packet, not its own starting date.
1074          * As a quick work around, we just skip an oggpage */
1075
1076         for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
1077         {
1078             /* we'll trash all the data until we find the next pcr */
1079             p_stream->b_reinit = 1;
1080             p_stream->i_pcr = -1;
1081             p_stream->i_interpolated_pcr = -1;
1082         }
1083         p_ogg->b_reinit = 1;
1084     }
1085
1086
1087     /*
1088      * Demux ogg pages from the stream
1089      */
1090     for( i = 0; i < PAGES_READ_ONCE || p_ogg->b_reinit;  i++ )
1091     {
1092         if( Ogg_ReadPage( p_input, p_ogg, &oggpage ) != VLC_SUCCESS )
1093         {
1094             b_eos = 1;
1095             break;
1096         }
1097
1098         for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
1099         {
1100             if( ogg_stream_pagein( &p_stream->os, &oggpage ) != 0 )
1101                 continue;
1102
1103             while( ogg_stream_packetout( &p_stream->os, &oggpacket ) > 0 )
1104             {
1105                 if( !p_stream->p_es )
1106                 {
1107                     break;
1108                 }
1109
1110                 if( p_stream->b_reinit )
1111                 {
1112                     if( oggpacket.granulepos >= 0 )
1113                     {
1114                         p_stream->b_reinit = 0;
1115
1116                         /* Convert the next granulepos into a pcr */
1117                         Ogg_UpdatePCR( p_stream, &oggpacket );
1118
1119                         /* Call the pace control to reinitialize
1120                          * the system clock */
1121                          input_ClockManageRef( p_input,
1122                              p_input->stream.p_selected_program,
1123                              p_stream->i_pcr );
1124
1125                          if( (!p_ogg->p_stream_video ||
1126                               !p_ogg->p_stream_video->b_reinit) &&
1127                              (!p_ogg->p_stream_audio ||
1128                               !p_ogg->p_stream_audio->b_reinit) )
1129                          {
1130                              p_ogg->b_reinit = 0;
1131                          }
1132                     }
1133                     continue;
1134                 }
1135
1136                 Ogg_DecodePacket( p_input, p_stream, &oggpacket );
1137
1138             }
1139         }
1140     }
1141
1142     i_stream = 0;
1143     p_ogg->i_pcr = p_stream->i_interpolated_pcr;
1144     for( ; i_stream < p_ogg->i_streams; i_stream++ )
1145     {
1146         if( p_stream->i_cat == SPU_ES )
1147             continue;
1148
1149         if( p_stream->i_interpolated_pcr < p_ogg->i_pcr )
1150             p_ogg->i_pcr = p_stream->i_interpolated_pcr;
1151     }
1152 #undef p_stream
1153
1154
1155     /* Call the pace control */
1156     input_ClockManageRef( p_input, p_input->stream.p_selected_program,
1157                           p_ogg->i_pcr );
1158
1159     /* Did we reach the end of stream ? */
1160     return( b_eos ? 0 : 1 );
1161 }