6456be74ea7844ed16294547ab2d63f1b41ab83d
[vlc.git] / 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.11 2002/11/21 09:39:39 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( p_stream->i_bitrate )
344             p_stream->i_interpolated_pcr += ( p_oggpacket->bytes * 90000
345                                               / p_stream->i_bitrate / 8 );
346         else
347             p_stream->i_interpolated_pcr = -1;
348     }
349 }
350
351 /****************************************************************************
352  * Ogg_DecodePacket: Decode an Ogg packet.
353  ****************************************************************************/
354 static void Ogg_DecodePacket( input_thread_t *p_input,
355                               logical_stream_t *p_stream,
356                               ogg_packet *p_oggpacket )
357 {
358     pes_packet_t  *p_pes;
359     data_packet_t *p_data;
360     //demux_sys_t *p_ogg = p_input->p_demux_data;
361     int i_header_len = 0;
362
363     if( p_stream->b_force_backup )
364     {
365         /* Backup the ogg packet (likely an header packet) */
366         ogg_packet *p_packet_backup;
367         p_stream->i_packets_backup++;
368         p_stream->p_packets_backup =
369             realloc( p_stream->p_packets_backup, p_stream->i_packets_backup *
370                      sizeof(ogg_packet) );
371
372         p_packet_backup =
373             &p_stream->p_packets_backup[p_stream->i_packets_backup - 1];
374
375         p_packet_backup->bytes = p_oggpacket->bytes;
376         p_packet_backup->granulepos = p_oggpacket->granulepos;
377         p_packet_backup->packet = malloc( p_oggpacket->bytes );
378         if( !p_packet_backup->packet ) return;
379         memcpy( p_packet_backup->packet, p_oggpacket->packet,
380                 p_oggpacket->bytes );
381
382         switch( p_stream->i_fourcc )
383         {
384         case VLC_FOURCC( 'v','o','r','b' ):
385           if( p_stream->i_packets_backup == 3 ) p_stream->b_force_backup = 0;
386           break;
387
388         default:
389           p_stream->b_force_backup = 0;
390           break;
391         }
392     }
393
394     if( !p_stream->p_es->p_decoder_fifo )
395     {
396         /* This stream isn't currently selected so we don't need to decode it,
397          * but we do need to store its pcr as it might be selected later on. */
398         Ogg_UpdatePCR( p_stream, p_oggpacket );
399
400         return;
401     }
402
403     if( !( p_pes = input_NewPES( p_input->p_method_data ) ) )
404     {
405         return;
406     }
407     if( !( p_data = input_NewPacket( p_input->p_method_data,
408                                      p_oggpacket->bytes ) ) )
409     {
410         input_DeletePES( p_input->p_method_data, p_pes );
411         return;
412     }
413
414     /* Convert the pcr into a pts */
415     if( p_stream->i_cat != SPU_ES )
416     {
417         p_pes->i_pts = ( p_stream->i_pcr < 0 ) ? 0 :
418             input_ClockGetTS( p_input, p_input->stream.p_selected_program,
419                               p_stream->i_pcr );
420     }
421     else
422     {
423         /* Of course subtitles had to be different! */
424         p_pes->i_pts = ( p_oggpacket->granulepos < 0 ) ? 0 :
425             input_ClockGetTS( p_input, p_input->stream.p_selected_program,
426                               p_oggpacket->granulepos * 90000 /
427                               p_stream->f_rate );
428     }
429
430     /* Convert the next granulepos into a pcr */
431     Ogg_UpdatePCR( p_stream, p_oggpacket );
432
433     p_pes->i_nb_data = 1;
434     p_pes->i_dts = p_oggpacket->granulepos;
435     p_pes->p_first = p_pes->p_last = p_data;
436     p_pes->i_pes_size = p_oggpacket->bytes;
437
438     if( p_stream->i_fourcc != VLC_FOURCC( 'v','o','r','b' ) &&
439         p_stream->i_fourcc != VLC_FOURCC( 't','a','r','k' ) &&
440         p_stream->i_fourcc != VLC_FOURCC( 't','h','e','o' ) )
441     {
442         /* Remove the header from the packet */
443         i_header_len = (*p_oggpacket->packet & PACKET_LEN_BITS01) >> 6;
444         i_header_len |= (*p_oggpacket->packet & PACKET_LEN_BITS2) << 1;
445         i_header_len++;
446
447         p_pes->i_pes_size -= i_header_len;
448         p_pes->i_dts = 0;
449     }
450
451     if( p_stream->i_fourcc == VLC_FOURCC( 't','a','r','k' ) )
452     {
453         /* FIXME: the biggest hack I've ever done */
454         msg_Warn( p_input, "tark pts: %lli, granule: %i",
455                   p_pes->i_pts, p_pes->i_dts );
456         msleep(10000);
457     }
458
459     memcpy( p_data->p_payload_start,
460             p_oggpacket->packet + i_header_len,
461             p_oggpacket->bytes - i_header_len );
462
463     p_data->p_payload_end = p_data->p_payload_start + p_pes->i_pes_size;
464     p_data->b_discard_payload = 0;
465
466     input_DecodePES( p_stream->p_es->p_decoder_fifo, p_pes );
467 }
468
469 /****************************************************************************
470  * Ogg_FindLogicalStreams: Find the logical streams embedded in the physical
471  *                         stream and fill p_ogg.
472  *****************************************************************************
473  * The initial page of a logical stream is marked as a 'bos' page.
474  * Furthermore, the Ogg specification mandates that grouped bitstreams begin
475  * together and all of the initial pages must appear before any data pages.
476  *
477  * On success this function returns VLC_SUCCESS.
478  ****************************************************************************/
479 static int Ogg_FindLogicalStreams( input_thread_t *p_input, demux_sys_t *p_ogg)
480 {
481     ogg_packet oggpacket;
482     ogg_page oggpage;
483     int i_stream;
484
485     while( Ogg_ReadPage( p_input, p_ogg, &oggpage ) == VLC_SUCCESS )
486     {
487         if( ogg_page_bos( &oggpage ) )
488         {
489
490             /* All is wonderful in our fine fine little world.
491              * We found the beginning of our first logical stream. */
492             while( ogg_page_bos( &oggpage ) )
493             {
494                 p_ogg->i_streams++;
495                 p_ogg->pp_stream =
496                     realloc( p_ogg->pp_stream, p_ogg->i_streams *
497                              sizeof(logical_stream_t *) );
498
499 #define p_stream p_ogg->pp_stream[p_ogg->i_streams - 1]
500
501                 p_stream = malloc( sizeof(logical_stream_t) );
502                 memset( p_stream, 0, sizeof(logical_stream_t) );
503
504                 /* Setup the logical stream */
505                 p_stream->i_serial_no = ogg_page_serialno( &oggpage );
506                 ogg_stream_init( &p_stream->os, p_stream->i_serial_no );
507
508                 /* Extract the initial header from the first page and verify
509                  * the codec type of tis Ogg bitstream */
510                 if( ogg_stream_pagein( &p_stream->os, &oggpage ) < 0 )
511                 {
512                     /* error. stream version mismatch perhaps */
513                     msg_Err( p_input, "Error reading first page of "
514                              "Ogg bitstream data" );
515                     return VLC_EGENERIC;
516                 }
517
518                 /* FIXME: check return value */
519                 ogg_stream_packetpeek( &p_stream->os, &oggpacket );
520
521                 /* Check for Vorbis header */
522                 if( oggpacket.bytes >= 7 &&
523                     ! strncmp( &oggpacket.packet[1], "vorbis", 6 ) )
524                 {
525                     oggpack_buffer opb;
526
527                     msg_Dbg( p_input, "found vorbis header" );
528                     p_stream->i_cat = AUDIO_ES;
529                     p_stream->i_fourcc = VLC_FOURCC( 'v','o','r','b' );
530
531                     /* Signal that we want to keep a backup of the vorbis
532                      * stream headers. They will be used when switching between
533                      * audio streams. */
534                     p_stream->b_force_backup = 1;
535
536                     /* Cheat and get additionnal info ;) */
537                     oggpack_readinit( &opb, oggpacket.packet, oggpacket.bytes);
538                     oggpack_adv( &opb, 96 );
539                     p_stream->f_rate = oggpack_read( &opb, 32 );
540                     oggpack_adv( &opb, 32 );
541                     p_stream->i_bitrate = oggpack_read( &opb, 32 );
542                 }
543                 /* Check for Theora header */
544                 else if( oggpacket.bytes >= 7 &&
545                          ! strncmp( &oggpacket.packet[1], "theora", 6 ) )
546                 {
547 #ifdef HAVE_OGGPACKB
548                     oggpack_buffer opb;
549                     int i_fps_numerator;
550                     int i_fps_denominator;
551                     int i_keyframe_frequency_force;
552 #endif
553
554                     msg_Dbg( p_input, "found theora header" );
555 #ifdef HAVE_OGGPACKB
556                     p_stream->i_cat = VIDEO_ES;
557                     p_stream->i_fourcc = VLC_FOURCC( 't','h','e','o' );
558
559                     /* Cheat and get additionnal info ;) */
560                     oggpackB_readinit(&opb, oggpacket.packet, oggpacket.bytes);
561                     oggpackB_adv( &opb, 56 );
562                     oggpackB_read( &opb, 8 ); /* major version num */
563                     oggpackB_read( &opb, 8 ); /* minor version num */
564                     oggpackB_read( &opb, 8 ); /* subminor version num */
565                     oggpackB_read( &opb, 16 ) /*<< 4*/; /* width */
566                     oggpackB_read( &opb, 16 ) /*<< 4*/; /* height */
567                     i_fps_numerator = oggpackB_read( &opb, 32 );
568                     i_fps_denominator = oggpackB_read( &opb, 32 );
569                     oggpackB_read( &opb, 24 ); /* aspect_numerator */
570                     oggpackB_read( &opb, 24 ); /* aspect_denominator */
571                     i_keyframe_frequency_force = 1 << oggpackB_read( &opb, 5 );
572                     p_stream->i_bitrate = oggpackB_read( &opb, 24 );
573                     oggpackB_read(&opb,6); /* quality */
574
575                     /* granule_shift = i_log( frequency_force -1 ) */
576                     p_stream->i_theora_keyframe_granule_shift = 0;
577                     i_keyframe_frequency_force--;
578                     while( i_keyframe_frequency_force )
579                     {
580                         p_stream->i_theora_keyframe_granule_shift++;
581                         i_keyframe_frequency_force >>= 1;
582                     }
583
584                     p_stream->f_rate = (float)i_fps_numerator /
585                                                 i_fps_denominator;
586                     msg_Dbg( p_input,
587                              "found theora header, bitrate: %i, rate: %f",
588                              p_stream->i_bitrate, p_stream->f_rate );
589 #else /* HAVE_OGGPACKB */
590                     msg_Dbg( p_input, "the ogg demuxer has been compiled "
591                              "without support for the oggpackB extension."
592                              "The theora stream won't be decoded." );
593                     free( p_stream );
594                     p_ogg->i_streams--;
595                     continue;
596 #endif /* HAVE_OGGPACKB */
597                 }
598                 /* Check for Tarkin header */
599                 else if( oggpacket.bytes >= 7 &&
600                          ! strncmp( &oggpacket.packet[1], "tarkin", 6 ) )
601                 {
602                     oggpack_buffer opb;
603
604                     msg_Dbg( p_input, "found tarkin header" );
605                     p_stream->i_cat = VIDEO_ES;
606                     p_stream->i_fourcc = VLC_FOURCC( 't','a','r','k' );
607
608                     /* Cheat and get additionnal info ;) */
609                     oggpack_readinit( &opb, oggpacket.packet, oggpacket.bytes);
610                     oggpack_adv( &opb, 88 );
611                     oggpack_adv( &opb, 104 );
612                     p_stream->i_bitrate = oggpack_read( &opb, 32 );
613                     p_stream->f_rate = 2; /* FIXME */
614                     msg_Dbg( p_input,
615                              "found tarkin header, bitrate: %i, rate: %f",
616                              p_stream->i_bitrate, p_stream->f_rate );
617                 }
618                 else if( (*oggpacket.packet & PACKET_TYPE_BITS )
619                          == PACKET_TYPE_HEADER && 
620                          oggpacket.bytes >= (int)sizeof(stream_header)+1 )
621                 {
622                     stream_header *st = (stream_header *)(oggpacket.packet+1);
623
624                     /* Check for video header (new format) */
625                     if( !strncmp( st->streamtype, "video", 5 ) )
626                     {
627                         p_stream->i_cat = VIDEO_ES;
628
629                         /* We need to get rid of the header packet */
630                         ogg_stream_packetout( &p_stream->os, &oggpacket );
631
632                         p_stream->p_bih = (BITMAPINFOHEADER *)
633                             malloc( sizeof(BITMAPINFOHEADER) );
634                         if( !p_stream->p_bih )
635                         {
636                             /* Mem allocation error, just ignore the stream */
637                             free( p_stream );
638                             p_ogg->i_streams--;
639                             continue;
640                         }
641                         p_stream->p_bih->biSize = sizeof(BITMAPINFOHEADER);
642                         p_stream->p_bih->biCompression=
643                             p_stream->i_fourcc = VLC_FOURCC( st->subtype[0],
644                                                              st->subtype[1],
645                                                              st->subtype[2],
646                                                              st->subtype[3] );
647                         msg_Dbg( p_input, "found video header of type: %.4s",
648                                  (char *)&p_stream->i_fourcc );
649
650                         p_stream->f_rate = 10000000.0 /
651                             GetQWLE((uint8_t *)&st->time_unit);
652                         p_stream->p_bih->biBitCount =
653                             GetWLE((uint8_t *)&st->bits_per_sample);
654                         p_stream->p_bih->biWidth =
655                             GetDWLE((uint8_t *)&st->sh.video.width);
656                         p_stream->p_bih->biHeight =
657                             GetDWLE((uint8_t *)&st->sh.video.height);
658                         p_stream->p_bih->biPlanes= 1 ;
659                         p_stream->p_bih->biSizeImage =
660                             (p_stream->p_bih->biBitCount >> 3) *
661                             p_stream->p_bih->biWidth *
662                             p_stream->p_bih->biHeight;
663
664                         msg_Dbg( p_input,
665                              "fps: %f, width:%i; height:%i, bitcount:%i",
666                             p_stream->f_rate, p_stream->p_bih->biWidth,
667                             p_stream->p_bih->biHeight,
668                             p_stream->p_bih->biBitCount);
669
670                         p_stream->i_bitrate = 0;
671                     }
672                     /* Check for audio header (new format) */
673                     else if( !strncmp( st->streamtype, "audio", 5 ) )
674                     {
675                         char p_buffer[5];
676
677                         p_stream->i_cat = AUDIO_ES;
678
679                         /* We need to get rid of the header packet */
680                         ogg_stream_packetout( &p_stream->os, &oggpacket );
681
682                         p_stream->p_wf = (WAVEFORMATEX *)
683                             malloc( sizeof(WAVEFORMATEX) );
684                         if( !p_stream->p_wf )
685                         {
686                             /* Mem allocation error, just ignore the stream */
687                             free( p_stream );
688                             p_ogg->i_streams--;
689                             continue;
690                         }
691
692                         memcpy( p_buffer, st->subtype, 4 );
693                         p_buffer[4] = '\0';
694                         p_stream->p_wf->wFormatTag = strtol(p_buffer,NULL,16);
695                         p_stream->p_wf->nChannels =
696                             GetWLE((uint8_t *)&st->sh.audio.channels);
697                         p_stream->f_rate = p_stream->p_wf->nSamplesPerSec =
698                             GetQWLE((uint8_t *)&st->samples_per_unit);
699                         p_stream->i_bitrate = p_stream->p_wf->nAvgBytesPerSec =
700                             GetDWLE((uint8_t *)&st->sh.audio.avgbytespersec);
701                         p_stream->i_bitrate *= 8;
702                         p_stream->p_wf->nBlockAlign =
703                             GetWLE((uint8_t *)&st->sh.audio.blockalign);
704                         p_stream->p_wf->wBitsPerSample =
705                             GetWLE((uint8_t *)&st->bits_per_sample);
706                         p_stream->p_wf->cbSize = 0;
707
708                         switch( p_stream->p_wf->wFormatTag )
709                         {
710                         case WAVE_FORMAT_PCM:
711                             p_stream->i_fourcc =
712                                 VLC_FOURCC( 'a', 'r', 'a', 'w' );
713                             break;
714                         case WAVE_FORMAT_MPEG:
715                         case WAVE_FORMAT_MPEGLAYER3:
716                             p_stream->i_fourcc =
717                                 VLC_FOURCC( 'm', 'p', 'g', 'a' );
718                             break;
719                         case WAVE_FORMAT_A52:
720                             p_stream->i_fourcc =
721                                 VLC_FOURCC( 'a', '5', '2', ' ' );
722                             break;
723                         case WAVE_FORMAT_WMA1:
724                             p_stream->i_fourcc =
725                                 VLC_FOURCC( 'w', 'm', 'a', '1' );
726                             break;
727                         case WAVE_FORMAT_WMA2:
728                             p_stream->i_fourcc =
729                                 VLC_FOURCC( 'w', 'm', 'a', '2' );
730                             break;
731                         default:
732                             p_stream->i_fourcc = VLC_FOURCC( 'm', 's',
733                                 ( p_stream->p_wf->wFormatTag >> 8 ) & 0xff,
734                                 p_stream->p_wf->wFormatTag & 0xff );
735                         }
736
737                         msg_Dbg( p_input, "found audio header of type: %.4s",
738                                  (char *)&p_stream->i_fourcc );
739                         msg_Dbg( p_input, "audio:0x%4.4x channels:%d %dHz "
740                                  "%dbits/sample %dkb/s",
741                                  p_stream->p_wf->wFormatTag,
742                                  p_stream->p_wf->nChannels,
743                                  p_stream->p_wf->nSamplesPerSec,
744                                  p_stream->p_wf->wBitsPerSample,
745                                  p_stream->p_wf->nAvgBytesPerSec * 8 / 1024 );
746                     }
747                     /* Check for text (subtitles) header */
748                     else if( !strncmp(st->streamtype, "text", 4) )
749                     {
750                         /* We need to get rid of the header packet */
751                         ogg_stream_packetout( &p_stream->os, &oggpacket );
752
753                         msg_Dbg( p_input, "found text subtitles header" );
754                         p_stream->i_cat = SPU_ES;
755                         p_stream->i_fourcc =
756                             VLC_FOURCC( 's', 'u', 'b', 't' );
757                         p_stream->f_rate = 1000; /* granulepos is in milisec */
758                     }
759                     else
760                     {
761                         msg_Dbg( p_input, "stream %d has a header marker "
762                             "but is of an unknown type", p_ogg->i_streams-1 );
763                         free( p_stream );
764                         p_ogg->i_streams--;
765                     }
766                 }
767                 else
768                 {
769                     msg_Dbg( p_input, "stream %d is of unknown type",
770                              p_ogg->i_streams-1 );
771                     free( p_stream );
772                     p_ogg->i_streams--;
773                 }
774
775 #undef p_stream
776
777                 if( Ogg_ReadPage( p_input, p_ogg, &oggpage ) != VLC_SUCCESS )
778                     return VLC_EGENERIC;
779             }
780
781             /* This is the first data page, which means we are now finished
782              * with the initial pages. We just need to store it in the relevant
783              * bitstream. */
784             for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
785             {
786                 if( ogg_stream_pagein( &p_ogg->pp_stream[i_stream]->os,
787                                        &oggpage ) == 0 )
788                 {
789                     break;
790                 }
791             }
792             return VLC_SUCCESS;
793         }
794     }
795     return VLC_EGENERIC;
796 }
797
798 /*****************************************************************************
799  * Activate: initializes ogg demux structures
800  *****************************************************************************/
801 static int Activate( vlc_object_t * p_this )
802 {
803     int i_stream, b_forced;
804     demux_sys_t    *p_ogg;
805     input_thread_t *p_input = (input_thread_t *)p_this;
806
807     p_input->p_demux_data = NULL;
808     b_forced = ( ( *p_input->psz_demux )&&
809                  ( !strncmp( p_input->psz_demux, "ogg", 10 ) ) ) ? 1 : 0;
810
811     /* Check if we are dealing with an ogg stream */
812     if( !b_forced && ( Ogg_Check( p_input ) != VLC_SUCCESS ) )
813         return -1;
814
815     /* Allocate p_ogg */
816     if( !( p_ogg = malloc( sizeof( demux_sys_t ) ) ) )
817     {
818         msg_Err( p_input, "out of memory" );
819         goto error;
820     }
821     memset( p_ogg, 0, sizeof( demux_sys_t ) );
822     p_input->p_demux_data = p_ogg;
823
824     p_ogg->i_pcr  = 0;
825     p_ogg->b_seekable = ( ( p_input->stream.b_seekable )
826                         &&( p_input->stream.i_method == INPUT_METHOD_FILE ) );
827
828     /* Initialize the Ogg physical bitstream parser */
829     ogg_sync_init( &p_ogg->oy );
830
831     /* Find the logical streams embedded in the physical stream and
832      * initialize our p_ogg structure. */
833     if( Ogg_FindLogicalStreams( p_input, p_ogg ) != VLC_SUCCESS )
834     {
835         msg_Err( p_input, "couldn't find an ogg logical stream" );
836         goto error;
837     }
838
839     /* Set the demux function */
840     p_input->pf_demux = Demux;
841
842     /* Initialize access plug-in structures. */
843     if( p_input->i_mtu == 0 )
844     {
845         /* Improve speed. */
846         p_input->i_bufsize = INPUT_DEFAULT_BUFSIZE;
847     }
848
849     /* Create one program */
850     vlc_mutex_lock( &p_input->stream.stream_lock );
851     if( input_InitStream( p_input, 0 ) == -1)
852     {
853         vlc_mutex_unlock( &p_input->stream.stream_lock );
854         msg_Err( p_input, "cannot init stream" );
855         goto error;
856     }
857     if( input_AddProgram( p_input, 0, 0) == NULL )
858     {
859         vlc_mutex_unlock( &p_input->stream.stream_lock );
860         msg_Err( p_input, "cannot add program" );
861         goto error;
862     }
863     p_input->stream.p_selected_program = p_input->stream.pp_programs[0];
864     p_input->stream.i_mux_rate = 0;
865     vlc_mutex_unlock( &p_input->stream.stream_lock );
866
867     for( i_stream = 0 ; i_stream < p_ogg->i_streams; i_stream++ )
868     {
869 #define p_stream p_ogg->pp_stream[i_stream]
870         vlc_mutex_lock( &p_input->stream.stream_lock );
871         p_stream->p_es = input_AddES( p_input,
872                                       p_input->stream.p_selected_program,
873                                       p_ogg->i_streams + 1, 0 );
874         p_input->stream.i_mux_rate += (p_stream->i_bitrate / ( 8 * 50 ));
875         vlc_mutex_unlock( &p_input->stream.stream_lock );
876         p_stream->p_es->i_stream_id = p_stream->p_es->i_id = i_stream;
877         p_stream->p_es->i_fourcc = p_stream->i_fourcc;
878         p_stream->p_es->i_cat = p_stream->i_cat;
879         p_stream->p_es->p_demux_data = p_stream->p_bih ?
880             (void *)p_stream->p_bih : (void *)p_stream->p_wf;
881 #undef p_stream
882     }
883
884     for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
885     {
886 #define p_stream  p_ogg->pp_stream[i_stream]
887         switch( p_stream->p_es->i_cat )
888         {
889             case( VIDEO_ES ):
890                 if( (p_ogg->p_stream_video == NULL) )
891                 {
892                     p_ogg->p_stream_video = p_stream;
893                     /* TODO add test to see if a decoder has been found */
894                     Ogg_StreamStart( p_input, p_ogg, i_stream );
895                 }
896                 break;
897
898             case( AUDIO_ES ):
899                 if( (p_ogg->p_stream_audio == NULL) )
900                 {
901                     int i_audio = config_GetInt( p_input, "audio-channel" );
902                     if( i_audio == i_stream || i_audio <= 0 ||
903                         i_audio >= p_ogg->i_streams ||
904                         p_ogg->pp_stream[i_audio]->p_es->i_cat != AUDIO_ES )
905                     {
906                         p_ogg->p_stream_audio = p_stream;
907                         Ogg_StreamStart( p_input, p_ogg, i_stream );
908                     }
909                 }
910                 break;
911
912             case( SPU_ES ):
913                 if( (p_ogg->p_stream_spu == NULL) )
914                 {
915                     /* for spu, default is none */
916                     int i_spu = config_GetInt( p_input, "spu-channel" );
917                     if( i_spu < 0 || i_spu >= p_ogg->i_streams ||
918                         p_ogg->pp_stream[i_spu]->p_es->i_cat != SPU_ES )
919                     {
920                         break;
921                     }
922                     else if( i_spu == i_stream )
923                     {
924                         p_ogg->p_stream_spu = p_stream;
925                         Ogg_StreamStart( p_input, p_ogg, i_stream );
926                     }
927                 }
928                 break;
929
930             default:
931                 break;
932         }
933 #undef p_stream
934     }
935
936     /* we select the first audio and video ES */
937     vlc_mutex_lock( &p_input->stream.stream_lock );
938     if( !p_ogg->p_stream_video )
939     {
940         msg_Warn( p_input, "no video stream found" );
941     }
942     if( !p_ogg->p_stream_audio )
943     {
944         msg_Warn( p_input, "no audio stream found!" );
945     }
946     p_input->stream.p_selected_program->b_is_ok = 1;
947     vlc_mutex_unlock( &p_input->stream.stream_lock );
948
949     /* Call the pace control */
950     input_ClockManageRef( p_input,
951                           p_input->stream.p_selected_program,
952                           p_ogg->i_pcr );
953
954     return 0;
955
956  error:
957     Deactivate( (vlc_object_t *)p_input );
958     return -1;
959
960 }
961
962 /*****************************************************************************
963  * Deactivate: frees unused data
964  *****************************************************************************/
965 static void Deactivate( vlc_object_t *p_this )
966 {
967     input_thread_t *p_input = (input_thread_t *)p_this;
968     demux_sys_t *p_ogg = (demux_sys_t *)p_input->p_demux_data  ; 
969     int i, j;
970
971     if( p_ogg )
972     {
973         /* Cleanup the bitstream parser */
974         ogg_sync_clear( &p_ogg->oy );
975
976         for( i = 0; i < p_ogg->i_streams; i++ )
977         {
978             ogg_stream_clear( &p_ogg->pp_stream[i]->os );
979             for( j = 0; j < p_ogg->pp_stream[i]->i_packets_backup; j++ )
980             {
981                 free( p_ogg->pp_stream[i]->p_packets_backup[j].packet );
982             }
983             if( p_ogg->pp_stream[i]->p_packets_backup)
984                 free( p_ogg->pp_stream[i]->p_packets_backup );
985 #if 0 /* hmmm, it's already freed in input_DelES() */
986             if( p_ogg->pp_stream[i]->p_bih )
987                 free( p_ogg->pp_stream[i]->p_bih );
988             if( p_ogg->pp_stream[i]->p_wf )
989                 free( p_ogg->pp_stream[i]->p_wf );
990 #endif
991             free( p_ogg->pp_stream[i] );
992         }
993         if( p_ogg->pp_stream ) free( p_ogg->pp_stream );
994
995         free( p_ogg );
996     }
997 }
998
999 /*****************************************************************************
1000  * Demux: reads and demuxes data packets
1001  *****************************************************************************
1002  * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
1003  *****************************************************************************/
1004 static int Demux( input_thread_t * p_input )
1005 {
1006     int i, i_stream, b_eos = 0;
1007     ogg_page    oggpage;
1008     ogg_packet  oggpacket;
1009     demux_sys_t *p_ogg  = (demux_sys_t *)p_input->p_demux_data;
1010
1011 #define p_stream p_ogg->pp_stream[i_stream]
1012     /* detect new selected/unselected streams */
1013     for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
1014     {
1015         if( p_stream->p_es )
1016         {
1017             if( p_stream->p_es->p_decoder_fifo &&
1018                 !p_stream->i_activated )
1019             {
1020                 Ogg_StreamStart( p_input, p_ogg, i_stream );
1021             }
1022             else
1023             if( !p_stream->p_es->p_decoder_fifo &&
1024                 p_stream->i_activated )
1025             {
1026                 Ogg_StreamStop( p_input, p_ogg, i_stream );
1027             }
1028         }
1029     }
1030
1031     /* search for new video and audio stream to select
1032      * if current have been unselected */
1033     if( ( !p_ogg->p_stream_video )
1034             || ( !p_ogg->p_stream_video->p_es->p_decoder_fifo ) )
1035     {
1036         p_ogg->p_stream_video = NULL;
1037         for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
1038         {
1039             if( ( p_stream->i_cat == VIDEO_ES )
1040                   &&( p_stream->p_es->p_decoder_fifo ) )
1041             {
1042                 p_ogg->p_stream_video = p_stream;
1043                 break;
1044             }
1045         }
1046     }
1047     if( ( !p_ogg->p_stream_audio )
1048             ||( !p_ogg->p_stream_audio->p_es->p_decoder_fifo ) )
1049     {
1050         p_ogg->p_stream_audio = NULL;
1051         for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
1052         {
1053             if( ( p_stream->i_cat == AUDIO_ES )
1054                   &&( p_stream->p_es->p_decoder_fifo ) )
1055             {
1056                 p_ogg->p_stream_audio = p_stream;
1057                 break;
1058             }
1059         }
1060     }
1061
1062     if( p_input->stream.p_selected_program->i_synchro_state == SYNCHRO_REINIT )
1063     {
1064         msg_Warn( p_input, "synchro reinit" );
1065
1066         /* An ogg packet does only contain the starting date of the next
1067          * packet, not its own starting date.
1068          * As a quick work around, we just skip an oggpage */
1069
1070         for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
1071         {
1072             /* we'll trash all the data until we find the next pcr */
1073             p_stream->b_reinit = 1;
1074             p_stream->i_pcr = -1;
1075             p_stream->i_interpolated_pcr = -1;
1076         }
1077         p_ogg->b_reinit = 1;
1078     }
1079
1080
1081     /*
1082      * Demux ogg pages from the stream
1083      */
1084     for( i = 0; i < PAGES_READ_ONCE || p_ogg->b_reinit;  i++ )
1085     {
1086         if( Ogg_ReadPage( p_input, p_ogg, &oggpage ) != VLC_SUCCESS )
1087         {
1088             b_eos = 1;
1089             break;
1090         }
1091
1092         for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
1093         {
1094             if( ogg_stream_pagein( &p_stream->os, &oggpage ) != 0 )
1095                 continue;
1096
1097             while( ogg_stream_packetout( &p_stream->os, &oggpacket ) > 0 )
1098             {
1099                 if( !p_stream->p_es )
1100                 {
1101                     break;
1102                 }
1103
1104                 if( p_stream->b_reinit )
1105                 {
1106                     if( oggpacket.granulepos >= 0 )
1107                     {
1108                         p_stream->b_reinit = 0;
1109
1110                         /* Convert the next granulepos into a pcr */
1111                         Ogg_UpdatePCR( p_stream, &oggpacket );
1112
1113                         /* Call the pace control to reinitialize
1114                          * the system clock */
1115                          input_ClockManageRef( p_input,
1116                              p_input->stream.p_selected_program,
1117                              p_stream->i_pcr );
1118
1119                          if( (!p_ogg->p_stream_video ||
1120                               !p_ogg->p_stream_video->b_reinit) &&
1121                              (!p_ogg->p_stream_audio ||
1122                               !p_ogg->p_stream_audio->b_reinit) )
1123                          {
1124                              p_ogg->b_reinit = 0;
1125                          }
1126                     }
1127                     continue;
1128                 }
1129
1130                 Ogg_DecodePacket( p_input, p_stream, &oggpacket );
1131
1132             }
1133         }
1134     }
1135
1136     p_ogg->i_pcr = INT_MAX;
1137     for( i_stream = 0 ; i_stream < p_ogg->i_streams; i_stream++ )
1138     {
1139         if( p_stream->i_cat == SPU_ES )
1140             continue;
1141
1142         if( p_stream->i_interpolated_pcr >= 0 &&
1143             p_stream->i_interpolated_pcr < p_ogg->i_pcr )
1144             p_ogg->i_pcr = p_stream->i_interpolated_pcr;
1145     }
1146 #undef p_stream
1147
1148
1149     /* Call the pace control */
1150     input_ClockManageRef( p_input, p_input->stream.p_selected_program,
1151                           p_ogg->i_pcr );
1152
1153     /* Did we reach the end of stream ? */
1154     return( b_eos ? 0 : 1 );
1155 }