1 /*****************************************************************************
2 * ogg.c : ogg stream demux module for vlc
3 *****************************************************************************
4 * Copyright (C) 2001-2003 VideoLAN
5 * $Id: ogg.c,v 1.58 2004/02/15 13:16:43 gbazin Exp $
7 * Author: Gildas Bazin <gbazin@netcourrier.com>
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.
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.
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 *****************************************************************************/
24 /*****************************************************************************
26 *****************************************************************************/
28 #include <vlc/input.h>
35 #define OGG_BLOCK_SIZE 4096
37 /*****************************************************************************
38 * Definitions of structures and functions used by this plugins
39 *****************************************************************************/
40 typedef struct logical_stream_s
42 ogg_stream_state os; /* logical stream of packets */
51 /* the header of some logical streams (eg vorbis) contain essential
52 * data for the decoder. We back them up here in case we need to re-feed
53 * them to the decoder. */
56 ogg_packet *p_packets_backup;
58 /* program clock reference (in units of 90kHz) derived from the previous
61 mtime_t i_interpolated_pcr;
62 mtime_t i_previous_pcr;
66 int i_theora_keyframe_granule_shift;
72 ogg_sync_state oy; /* sync and verify incoming physical bitstream */
74 int i_streams; /* number of logical bitstreams */
75 logical_stream_t **pp_stream; /* pointer to an array of logical streams */
77 /* program clock reference (in units of 90kHz) derived from the pcr of
80 int i_old_synchro_state;
86 /* OggDS headers for the new header format (used in ogm files) */
87 typedef struct stream_header_video
91 } stream_header_video;
93 typedef struct stream_header_audio
96 ogg_int16_t blockalign;
97 ogg_int32_t avgbytespersec;
98 } stream_header_audio;
100 typedef struct stream_header
105 ogg_int32_t size; /* size of the structure */
107 ogg_int64_t time_unit; /* in reference time */
108 ogg_int64_t samples_per_unit;
109 ogg_int32_t default_len; /* in media time */
111 ogg_int32_t buffersize;
112 ogg_int16_t bits_per_sample;
117 stream_header_video video;
119 stream_header_audio audio;
123 /* Some defines from OggDS */
124 #define PACKET_TYPE_HEADER 0x01
125 #define PACKET_TYPE_BITS 0x07
126 #define PACKET_LEN_BITS01 0xc0
127 #define PACKET_LEN_BITS2 0x02
128 #define PACKET_IS_SYNCPOINT 0x08
130 /*****************************************************************************
132 *****************************************************************************/
133 static int Activate ( vlc_object_t * );
134 static void Deactivate( vlc_object_t * );
135 static int Demux ( input_thread_t * );
136 static int Control ( input_thread_t *, int, va_list );
138 /* Bitstream manipulation */
139 static int Ogg_Check ( input_thread_t *p_input );
140 static int Ogg_ReadPage ( input_thread_t *, demux_sys_t *, ogg_page * );
141 static void Ogg_UpdatePCR ( logical_stream_t *, ogg_packet * );
142 static void Ogg_DecodePacket ( input_thread_t *p_input,
143 logical_stream_t *p_stream, ogg_packet * );
145 static int Ogg_BeginningOfStream( input_thread_t *p_input, demux_sys_t *p_ogg);
146 static int Ogg_FindLogicalStreams( input_thread_t *p_input,demux_sys_t *p_ogg);
147 static void Ogg_EndOfStream( input_thread_t *p_input, demux_sys_t *p_ogg );
149 /*****************************************************************************
151 *****************************************************************************/
153 set_description( _("Ogg stream demuxer" ) );
154 set_capability( "demux", 50 );
155 set_callbacks( Activate, Deactivate );
156 add_shortcut( "ogg" );
159 /****************************************************************************
160 * Ogg_Check: Check we are dealing with an ogg stream.
161 ****************************************************************************/
162 static int Ogg_Check( input_thread_t *p_input )
165 int i_size = input_Peek( p_input, &p_peek, 4 );
167 /* Check for the Ogg capture pattern */
168 if( !(i_size>3) || !(p_peek[0] == 'O') || !(p_peek[1] == 'g') ||
169 !(p_peek[2] == 'g') || !(p_peek[3] == 'S') )
172 /* FIXME: Capture pattern might not be enough so we can also check for the
173 * the first complete page */
178 /****************************************************************************
179 * Ogg_ReadPage: Read a full Ogg page from the physical bitstream.
180 ****************************************************************************
181 * Returns VLC_SUCCESS if a page has been read. An error might happen if we
182 * are at the end of stream.
183 ****************************************************************************/
184 static int Ogg_ReadPage( input_thread_t *p_input, demux_sys_t *p_ogg,
185 ogg_page *p_oggpage )
188 data_packet_t *p_data;
191 while( ogg_sync_pageout( &p_ogg->oy, p_oggpage ) != 1 )
193 i_read = input_SplitBuffer( p_input, &p_data, OGG_BLOCK_SIZE );
197 p_buffer = ogg_sync_buffer( &p_ogg->oy, i_read );
198 p_input->p_vlc->pf_memcpy( p_buffer, p_data->p_payload_start, i_read );
199 ogg_sync_wrote( &p_ogg->oy, i_read );
200 input_DeletePacket( p_input->p_method_data, p_data );
206 /****************************************************************************
207 * Ogg_UpdatePCR: update the PCR (90kHz program clock reference) for the
209 ****************************************************************************/
210 static void Ogg_UpdatePCR( logical_stream_t *p_stream,
211 ogg_packet *p_oggpacket )
213 /* Convert the granulepos into a pcr */
214 if( p_oggpacket->granulepos >= 0 )
216 if( p_stream->fmt.i_codec != VLC_FOURCC( 't','h','e','o' ) )
218 p_stream->i_pcr = p_oggpacket->granulepos * 90000
223 ogg_int64_t iframe = p_oggpacket->granulepos >>
224 p_stream->i_theora_keyframe_granule_shift;
225 ogg_int64_t pframe = p_oggpacket->granulepos -
226 ( iframe << p_stream->i_theora_keyframe_granule_shift );
228 p_stream->i_pcr = ( iframe + pframe ) * 90000
232 p_stream->i_interpolated_pcr = p_stream->i_pcr;
236 p_stream->i_pcr = -1;
238 /* no granulepos available, try to interpolate the pcr.
239 * If we can't then don't touch the old value. */
240 if( p_stream->fmt.i_cat == VIDEO_ES )
241 /* 1 frame per packet */
242 p_stream->i_interpolated_pcr += (90000 / p_stream->f_rate);
243 else if( p_stream->fmt.i_bitrate )
244 p_stream->i_interpolated_pcr += ( p_oggpacket->bytes * 90000
245 / p_stream->fmt.i_bitrate / 8 );
249 /****************************************************************************
250 * Ogg_DecodePacket: Decode an Ogg packet.
251 ****************************************************************************/
252 static void Ogg_DecodePacket( input_thread_t *p_input,
253 logical_stream_t *p_stream,
254 ogg_packet *p_oggpacket )
257 vlc_bool_t b_selected;
258 int i_header_len = 0;
262 if( !p_oggpacket->bytes )
264 msg_Dbg( p_input, "discarding 0 sized packet" );
268 if( p_stream->b_force_backup )
270 ogg_packet *p_packet_backup;
271 p_stream->i_packets_backup++;
272 switch( p_stream->fmt.i_codec )
274 case VLC_FOURCC( 'v','o','r','b' ):
275 case VLC_FOURCC( 's','p','x',' ' ):
276 case VLC_FOURCC( 't','h','e','o' ):
277 if( p_stream->i_packets_backup == 3 ) p_stream->b_force_backup = 0;
280 case VLC_FOURCC( 'f','l','a','c' ):
281 if( p_stream->i_packets_backup == 1 ) return;
282 else if( p_stream->i_packets_backup == 2 )
284 /* Parse the STREAMINFO metadata */
286 bs_init( &s, p_oggpacket->packet, p_oggpacket->bytes );
288 if( bs_read( &s, 7 ) == 0 )
290 if( bs_read( &s, 24 ) >= 34 /*size STREAMINFO*/ )
293 p_stream->f_rate = p_stream->fmt.audio.i_rate =
295 p_stream->fmt.audio.i_channels =
296 bs_read( &s, 3 ) + 1;
298 msg_Dbg( p_input, "FLAC header, channels: %i, rate: %i",
299 p_stream->fmt.audio.i_channels,
300 (int)p_stream->f_rate );
304 msg_Dbg( p_input, "FLAC STREAMINFO metadata too short" );
307 /* Store STREAMINFO for the decoder and packetizer */
308 p_stream->fmt.i_extra = p_oggpacket->bytes + 4;
309 p_stream->fmt.p_extra = malloc( p_stream->fmt.i_extra );
310 memcpy( p_stream->fmt.p_extra, "fLaC", 4);
311 memcpy( ((uint8_t *)p_stream->fmt.p_extra) + 4,
312 p_oggpacket->packet, p_oggpacket->bytes );
314 /* Fake this as the last metadata block */
315 ((uint8_t*)p_stream->fmt.p_extra)[4] |= 0x80;
317 p_stream->p_es = es_out_Add( p_input->p_es_out,
322 /* This ain't a STREAMINFO metadata */
323 msg_Dbg( p_input, "Invalid FLAC STREAMINFO metadata" );
325 p_stream->b_force_backup = 0;
326 p_stream->i_packets_backup = 0;
328 if( p_oggpacket->granulepos >= 0 )
329 Ogg_UpdatePCR( p_stream, p_oggpacket );
331 p_stream->i_previous_pcr = 0;
337 p_stream->b_force_backup = 0;
341 /* Backup the ogg packet (likely an header packet) */
342 p_stream->p_packets_backup =
343 realloc( p_stream->p_packets_backup, p_stream->i_packets_backup *
344 sizeof(ogg_packet) );
347 &p_stream->p_packets_backup[p_stream->i_packets_backup - 1];
349 p_packet_backup->bytes = p_oggpacket->bytes;
350 p_packet_backup->granulepos = p_oggpacket->granulepos;
352 if( p_oggpacket->granulepos >= 0 )
354 /* Because of vorbis granulepos scheme we must set the pcr for the
355 * 1st header packet so it doesn't get discarded in the
357 Ogg_UpdatePCR( p_stream, p_oggpacket );
360 p_packet_backup->packet = malloc( p_oggpacket->bytes );
361 if( !p_packet_backup->packet ) return;
362 memcpy( p_packet_backup->packet, p_oggpacket->packet,
363 p_oggpacket->bytes );
366 /* Check the ES is selected */
367 es_out_Control( p_input->p_es_out, ES_OUT_GET_ES_STATE,
368 p_stream->p_es, &b_selected );
370 if( b_selected && !p_stream->b_activated )
372 p_stream->b_activated = VLC_TRUE;
374 /* Newly activated stream, feed the backup headers to the decoder */
375 if( !p_stream->b_force_backup )
378 for( i = 0; i < p_stream->i_packets_backup; i++ )
380 /* Set correct starting date in header packets */
381 p_stream->p_packets_backup[i].granulepos =
382 p_stream->i_interpolated_pcr * p_stream->f_rate / 90000;
384 Ogg_DecodePacket( p_input, p_stream,
385 &p_stream->p_packets_backup[i] );
390 /* Convert the pcr into a pts */
391 if( p_stream->fmt.i_codec == VLC_FOURCC( 'v','o','r','b' ) ||
392 p_stream->fmt.i_codec == VLC_FOURCC( 's','p','x',' ' ) ||
393 p_stream->fmt.i_codec == VLC_FOURCC( 'f','l','a','c' ) )
395 if( p_stream->i_pcr >= 0 )
397 /* This is for streams where the granulepos of the header packets
398 * doesn't match these of the data packets (eg. ogg web radios). */
399 if( p_stream->i_previous_pcr == 0 &&
400 p_stream->i_pcr > 3 * DEFAULT_PTS_DELAY * 9/100 )
401 p_input->stream.p_selected_program->i_synchro_state =
404 p_stream->i_previous_pcr = p_stream->i_pcr;
406 /* Call the pace control */
407 if( p_input->stream.p_selected_program->i_synchro_state ==
409 input_ClockManageRef( p_input,
410 p_input->stream.p_selected_program,
413 /* The granulepos is the end date of the sample */
414 i_pts = input_ClockGetTS( p_input,
415 p_input->stream.p_selected_program,
420 /* Convert the granulepos into the next pcr */
421 Ogg_UpdatePCR( p_stream, p_oggpacket );
423 if( p_stream->i_pcr >= 0 )
425 /* This is for streams where the granulepos of the header packets
426 * doesn't match these of the data packets (eg. ogg web radios). */
427 if( p_stream->i_previous_pcr == 0 &&
428 p_stream->i_pcr > 3 * DEFAULT_PTS_DELAY * 9/100 )
429 p_input->stream.p_selected_program->i_synchro_state =
432 /* Call the pace control */
433 if( p_input->stream.p_selected_program->i_synchro_state ==
435 input_ClockManageRef( p_input, p_input->stream.p_selected_program,
439 if( p_stream->fmt.i_codec != VLC_FOURCC( 'v','o','r','b' ) &&
440 p_stream->fmt.i_codec != VLC_FOURCC( 's','p','x',' ' ) &&
441 p_stream->fmt.i_codec != VLC_FOURCC( 'f','l','a','c' ) &&
442 p_stream->i_pcr >= 0 )
444 p_stream->i_previous_pcr = p_stream->i_pcr;
446 /* The granulepos is the start date of the sample */
447 i_pts = input_ClockGetTS( p_input, p_input->stream.p_selected_program,
453 /* This stream isn't currently selected so we don't need to decode it,
454 * but we did need to store its pcr as it might be selected later on */
455 p_stream->b_activated = VLC_FALSE;
459 if( !( p_block = block_New( p_input, p_oggpacket->bytes ) ) ) return;
461 if( p_stream->fmt.i_cat == AUDIO_ES )
462 p_block->i_dts = p_block->i_pts = i_pts;
463 else if( p_stream->fmt.i_cat == SPU_ES )
465 p_block->i_pts = i_pts;
468 else if( p_stream->fmt.i_codec == VLC_FOURCC( 't','h','e','o' ) )
469 p_block->i_dts = p_block->i_pts = i_pts;
472 p_block->i_dts = i_pts;
476 if( p_stream->fmt.i_codec != VLC_FOURCC( 'v','o','r','b' ) &&
477 p_stream->fmt.i_codec != VLC_FOURCC( 's','p','x',' ' ) &&
478 p_stream->fmt.i_codec != VLC_FOURCC( 'f','l','a','c' ) &&
479 p_stream->fmt.i_codec != VLC_FOURCC( 't','a','r','k' ) &&
480 p_stream->fmt.i_codec != VLC_FOURCC( 't','h','e','o' ) )
482 /* We remove the header from the packet */
483 i_header_len = (*p_oggpacket->packet & PACKET_LEN_BITS01) >> 6;
484 i_header_len |= (*p_oggpacket->packet & PACKET_LEN_BITS2) << 1;
486 if( p_stream->fmt.i_codec == VLC_FOURCC( 's','u','b','t' ))
488 /* But with subtitles we need to retrieve the duration first */
491 if( i_header_len > 0 && p_oggpacket->bytes >= i_header_len + 1 )
493 for( i = 0, lenbytes = 0; i < i_header_len; i++ )
495 lenbytes = lenbytes << 8;
496 lenbytes += *(p_oggpacket->packet + i_header_len - i);
499 if( p_oggpacket->bytes - 1 - i_header_len > 2 ||
500 ( p_oggpacket->packet[i_header_len + 1] != ' ' &&
501 p_oggpacket->packet[i_header_len + 1] != 0 &&
502 p_oggpacket->packet[i_header_len + 1] != '\n' &&
503 p_oggpacket->packet[i_header_len + 1] != '\r' ) )
505 p_block->i_dts = p_block->i_pts + (mtime_t)lenbytes * 1000;
510 p_block->i_buffer -= i_header_len;
513 if( p_stream->fmt.i_codec == VLC_FOURCC( 't','a','r','k' ) )
515 /* FIXME: the biggest hack I've ever done */
516 msg_Warn( p_input, "tarkin pts: "I64Fd", granule: "I64Fd,
517 p_block->i_pts, p_block->i_dts );
521 memcpy( p_block->p_buffer, p_oggpacket->packet + i_header_len,
522 p_oggpacket->bytes - i_header_len );
524 es_out_Send( p_input->p_es_out, p_stream->p_es, p_block );
527 /****************************************************************************
528 * Ogg_FindLogicalStreams: Find the logical streams embedded in the physical
529 * stream and fill p_ogg.
530 *****************************************************************************
531 * The initial page of a logical stream is marked as a 'bos' page.
532 * Furthermore, the Ogg specification mandates that grouped bitstreams begin
533 * together and all of the initial pages must appear before any data pages.
535 * On success this function returns VLC_SUCCESS.
536 ****************************************************************************/
537 static int Ogg_FindLogicalStreams( input_thread_t *p_input, demux_sys_t *p_ogg)
539 ogg_packet oggpacket;
543 #define p_stream p_ogg->pp_stream[p_ogg->i_streams - 1]
545 while( Ogg_ReadPage( p_input, p_ogg, &oggpage ) == VLC_SUCCESS )
547 if( ogg_page_bos( &oggpage ) )
550 /* All is wonderful in our fine fine little world.
551 * We found the beginning of our first logical stream. */
552 while( ogg_page_bos( &oggpage ) )
556 realloc( p_ogg->pp_stream, p_ogg->i_streams *
557 sizeof(logical_stream_t *) );
559 p_stream = malloc( sizeof(logical_stream_t) );
560 memset( p_stream, 0, sizeof(logical_stream_t) );
562 es_format_Init( &p_stream->fmt, 0, 0 );
563 p_stream->b_activated = VLC_TRUE;
565 /* Setup the logical stream */
566 p_stream->i_serial_no = ogg_page_serialno( &oggpage );
567 ogg_stream_init( &p_stream->os, p_stream->i_serial_no );
569 /* Extract the initial header from the first page and verify
570 * the codec type of tis Ogg bitstream */
571 if( ogg_stream_pagein( &p_stream->os, &oggpage ) < 0 )
573 /* error. stream version mismatch perhaps */
574 msg_Err( p_input, "error reading first page of "
575 "Ogg bitstream data" );
579 /* FIXME: check return value */
580 ogg_stream_packetpeek( &p_stream->os, &oggpacket );
582 /* Check for Vorbis header */
583 if( oggpacket.bytes >= 7 &&
584 ! strncmp( &oggpacket.packet[1], "vorbis", 6 ) )
588 msg_Dbg( p_input, "found vorbis header" );
589 p_stream->fmt.i_cat = AUDIO_ES;
590 p_stream->fmt.i_codec = VLC_FOURCC( 'v','o','r','b' );
592 /* Signal that we want to keep a backup of the vorbis
593 * stream headers. They will be used when switching between
595 p_stream->b_force_backup = 1;
597 /* Cheat and get additionnal info ;) */
598 oggpack_readinit( &opb, oggpacket.packet, oggpacket.bytes);
599 oggpack_adv( &opb, 88 );
600 p_stream->fmt.audio.i_channels = oggpack_read( &opb, 8 );
601 p_stream->f_rate = p_stream->fmt.audio.i_rate =
602 oggpack_read( &opb, 32 );
603 oggpack_adv( &opb, 32 );
604 p_stream->fmt.i_bitrate = oggpack_read( &opb, 32 );
606 /* Check for Speex header */
607 else if( oggpacket.bytes >= 7 &&
608 ! strncmp( &oggpacket.packet[0], "Speex", 5 ) )
612 p_stream->fmt.i_cat = AUDIO_ES;
613 p_stream->fmt.i_codec = VLC_FOURCC( 's','p','x',' ' );
615 /* Signal that we want to keep a backup of the vorbis
616 * stream headers. They will be used when switching between
618 p_stream->b_force_backup = 1;
620 /* Cheat and get additionnal info ;) */
621 oggpack_readinit( &opb, oggpacket.packet, oggpacket.bytes);
622 oggpack_adv( &opb, 224 );
623 oggpack_adv( &opb, 32 ); /* speex_version_id */
624 oggpack_adv( &opb, 32 ); /* header_size */
625 p_stream->f_rate = p_stream->fmt.audio.i_rate =
626 oggpack_read( &opb, 32 );
627 oggpack_adv( &opb, 32 ); /* mode */
628 oggpack_adv( &opb, 32 ); /* mode_bitstream_version */
629 p_stream->fmt.audio.i_channels = oggpack_read( &opb, 32 );
630 p_stream->fmt.i_bitrate = oggpack_read( &opb, 32 );
632 msg_Dbg( p_input, "found speex header, channels: %i, "
633 "rate: %i, bitrate: %i",
634 p_stream->fmt.audio.i_channels,
635 (int)p_stream->f_rate, p_stream->fmt.i_bitrate );
637 /* Check for Flac header */
638 else if( oggpacket.bytes >= 4 &&
639 ! strncmp( &oggpacket.packet[0], "fLaC", 4 ) )
641 msg_Dbg( p_input, "found FLAC header" );
643 /* Grrrr!!!! Did they really have to put all the
644 * important info in the second header packet!!!
645 * (STREAMINFO metadata is in the following packet) */
646 p_stream->b_force_backup = 1;
648 p_stream->fmt.i_cat = AUDIO_ES;
649 p_stream->fmt.i_codec = VLC_FOURCC( 'f','l','a','c' );
651 /* Check for Theora header */
652 else if( oggpacket.bytes >= 7 &&
653 ! strncmp( &oggpacket.packet[1], "theora", 6 ) )
657 int i_fps_denominator;
658 int i_keyframe_frequency_force;
660 msg_Dbg( p_input, "found theora header" );
661 p_stream->fmt.i_cat = VIDEO_ES;
662 p_stream->fmt.i_codec = VLC_FOURCC( 't','h','e','o' );
664 /* Signal that we want to keep a backup of the vorbis
665 * stream headers. They will be used when switching between
667 p_stream->b_force_backup = 1;
669 /* Cheat and get additionnal info ;) */
670 bs_init( &bitstream, oggpacket.packet, oggpacket.bytes );
671 bs_skip( &bitstream, 56 );
672 bs_read( &bitstream, 8 ); /* major version num */
673 bs_read( &bitstream, 8 ); /* minor version num */
674 bs_read( &bitstream, 8 ); /* subminor version num */
675 bs_read( &bitstream, 16 ) /*<< 4*/; /* width */
676 bs_read( &bitstream, 16 ) /*<< 4*/; /* height */
677 bs_read( &bitstream, 24 ); /* frame width */
678 bs_read( &bitstream, 24 ); /* frame height */
679 bs_read( &bitstream, 8 ); /* x offset */
680 bs_read( &bitstream, 8 ); /* y offset */
682 i_fps_numerator = bs_read( &bitstream, 32 );
683 i_fps_denominator = bs_read( &bitstream, 32 );
684 bs_read( &bitstream, 24 ); /* aspect_numerator */
685 bs_read( &bitstream, 24 ); /* aspect_denominator */
686 i_keyframe_frequency_force = 1 << bs_read( &bitstream, 5 );
687 bs_read( &bitstream, 8 ); /* colorspace */
688 p_stream->fmt.i_bitrate = bs_read( &bitstream, 24 );
689 bs_read( &bitstream, 6 ); /* quality */
691 /* granule_shift = i_log( frequency_force -1 ) */
692 p_stream->i_theora_keyframe_granule_shift = 0;
693 i_keyframe_frequency_force--;
694 while( i_keyframe_frequency_force )
696 p_stream->i_theora_keyframe_granule_shift++;
697 i_keyframe_frequency_force >>= 1;
700 p_stream->f_rate = ((float)i_fps_numerator) /
703 "found theora header, bitrate: %i, rate: %f",
704 p_stream->fmt.i_bitrate, p_stream->f_rate );
706 /* Save this data in p_extra for ffmpeg */
707 p_stream->fmt.i_extra = oggpacket.bytes;
708 p_stream->fmt.p_extra = malloc( oggpacket.bytes );
709 memcpy( p_stream->fmt.p_extra,
710 oggpacket.packet, oggpacket.bytes );
712 /* Check for Tarkin header */
713 else if( oggpacket.bytes >= 7 &&
714 ! strncmp( &oggpacket.packet[1], "tarkin", 6 ) )
718 msg_Dbg( p_input, "found tarkin header" );
719 p_stream->fmt.i_cat = VIDEO_ES;
720 p_stream->fmt.i_codec = VLC_FOURCC( 't','a','r','k' );
722 /* Cheat and get additionnal info ;) */
723 oggpack_readinit( &opb, oggpacket.packet, oggpacket.bytes);
724 oggpack_adv( &opb, 88 );
725 oggpack_adv( &opb, 104 );
726 p_stream->fmt.i_bitrate = oggpack_read( &opb, 32 );
727 p_stream->f_rate = 2; /* FIXME */
729 "found tarkin header, bitrate: %i, rate: %f",
730 p_stream->fmt.i_bitrate, p_stream->f_rate );
732 else if( oggpacket.bytes >= 142 &&
733 !strncmp( &oggpacket.packet[1],
734 "Direct Show Samples embedded in Ogg", 35 ))
736 /* Old header type */
738 /* Check for video header (old format) */
739 if( GetDWLE((oggpacket.packet+96)) == 0x05589f80 &&
740 oggpacket.bytes >= 184 )
742 p_stream->fmt.i_cat = VIDEO_ES;
743 p_stream->fmt.i_codec =
744 VLC_FOURCC( oggpacket.packet[68],
745 oggpacket.packet[69],
746 oggpacket.packet[70],
747 oggpacket.packet[71] );
748 msg_Dbg( p_input, "found video header of type: %.4s",
749 (char *)&p_stream->fmt.i_codec );
751 p_stream->f_rate = 10000000.0 /
752 GetQWLE((oggpacket.packet+164));
753 p_stream->fmt.video.i_bits_per_pixel =
754 GetWLE((oggpacket.packet+182));
755 if( !p_stream->fmt.video.i_bits_per_pixel )
757 p_stream->fmt.video.i_bits_per_pixel = 24;
758 p_stream->fmt.video.i_width =
759 GetDWLE((oggpacket.packet+176));
760 p_stream->fmt.video.i_height =
761 GetDWLE((oggpacket.packet+180));
764 "fps: %f, width:%i; height:%i, bitcount:%i",
766 p_stream->fmt.video.i_width,
767 p_stream->fmt.video.i_height,
768 p_stream->fmt.video.i_bits_per_pixel);
771 /* Check for audio header (old format) */
772 else if( GetDWLE((oggpacket.packet+96)) == 0x05589F81 )
774 unsigned int i_extra_size;
775 unsigned int i_format_tag;
777 p_stream->fmt.i_cat = AUDIO_ES;
779 i_extra_size = GetWLE((oggpacket.packet+140));
782 p_stream->fmt.i_extra = i_extra_size;
783 p_stream->fmt.p_extra = malloc( i_extra_size );
784 memcpy( p_stream->fmt.p_extra,
785 oggpacket.packet + 142, i_extra_size );
788 i_format_tag = GetWLE((oggpacket.packet+124));
789 p_stream->fmt.audio.i_channels =
790 GetWLE((oggpacket.packet+126));
791 p_stream->f_rate = p_stream->fmt.audio.i_rate =
792 GetDWLE((oggpacket.packet+128));
793 p_stream->fmt.i_bitrate =
794 GetDWLE((oggpacket.packet+132)) * 8;
795 p_stream->fmt.audio.i_blockalign =
796 GetWLE((oggpacket.packet+136));
797 p_stream->fmt.audio.i_bitspersample =
798 GetWLE((oggpacket.packet+138));
800 switch( i_format_tag )
802 case WAVE_FORMAT_PCM:
803 p_stream->fmt.i_codec =
804 VLC_FOURCC( 'a', 'r', 'a', 'w' );
806 case WAVE_FORMAT_MPEG:
807 case WAVE_FORMAT_MPEGLAYER3:
808 p_stream->fmt.i_codec =
809 VLC_FOURCC( 'm', 'p', 'g', 'a' );
811 case WAVE_FORMAT_A52:
812 p_stream->fmt.i_codec =
813 VLC_FOURCC( 'a', '5', '2', ' ' );
815 case WAVE_FORMAT_WMA1:
816 p_stream->fmt.i_codec =
817 VLC_FOURCC( 'w', 'm', 'a', '1' );
819 case WAVE_FORMAT_WMA2:
820 p_stream->fmt.i_codec =
821 VLC_FOURCC( 'w', 'm', 'a', '2' );
824 p_stream->fmt.i_codec = VLC_FOURCC( 'm', 's',
825 ( i_format_tag >> 8 ) & 0xff,
826 i_format_tag & 0xff );
829 msg_Dbg( p_input, "found audio header of type: %.4s",
830 (char *)&p_stream->fmt.i_codec );
831 msg_Dbg( p_input, "audio:0x%4.4x channels:%d %dHz "
832 "%dbits/sample %dkb/s",
834 p_stream->fmt.audio.i_channels,
835 p_stream->fmt.audio.i_rate,
836 p_stream->fmt.audio.i_bitspersample,
837 p_stream->fmt.i_bitrate / 1024 );
842 msg_Dbg( p_input, "stream %d has an old header "
843 "but is of an unknown type", p_ogg->i_streams-1 );
848 else if( (*oggpacket.packet & PACKET_TYPE_BITS )
849 == PACKET_TYPE_HEADER &&
850 oggpacket.bytes >= (int)sizeof(stream_header)+1 )
852 stream_header *st = (stream_header *)(oggpacket.packet+1);
854 /* Check for video header (new format) */
855 if( !strncmp( st->streamtype, "video", 5 ) )
857 p_stream->fmt.i_cat = VIDEO_ES;
859 /* We need to get rid of the header packet */
860 ogg_stream_packetout( &p_stream->os, &oggpacket );
862 p_stream->fmt.i_codec =
863 VLC_FOURCC( st->subtype[0], st->subtype[1],
864 st->subtype[2], st->subtype[3] );
865 msg_Dbg( p_input, "found video header of type: %.4s",
866 (char *)&p_stream->fmt.i_codec );
868 p_stream->f_rate = 10000000.0 /
869 GetQWLE(&st->time_unit);
870 p_stream->fmt.video.i_bits_per_pixel =
871 GetWLE(&st->bits_per_sample);
872 p_stream->fmt.video.i_width =
873 GetDWLE(&st->sh.video.width);
874 p_stream->fmt.video.i_height =
875 GetDWLE(&st->sh.video.height);
878 "fps: %f, width:%i; height:%i, bitcount:%i",
880 p_stream->fmt.video.i_width,
881 p_stream->fmt.video.i_height,
882 p_stream->fmt.video.i_bits_per_pixel );
884 /* Check for audio header (new format) */
885 else if( !strncmp( st->streamtype, "audio", 5 ) )
890 p_stream->fmt.i_cat = AUDIO_ES;
892 /* We need to get rid of the header packet */
893 ogg_stream_packetout( &p_stream->os, &oggpacket );
895 memcpy( p_buffer, st->subtype, 4 );
897 i_format_tag = strtol(p_buffer,NULL,16);
898 p_stream->fmt.audio.i_channels =
899 GetWLE(&st->sh.audio.channels);
900 p_stream->f_rate = p_stream->fmt.audio.i_rate =
901 GetQWLE(&st->samples_per_unit);
902 p_stream->fmt.i_bitrate =
903 GetDWLE(&st->sh.audio.avgbytespersec) * 8;
904 p_stream->fmt.audio.i_blockalign =
905 GetWLE(&st->sh.audio.blockalign);
906 p_stream->fmt.audio.i_bitspersample =
907 GetWLE(&st->bits_per_sample);
909 switch( i_format_tag )
911 case WAVE_FORMAT_PCM:
912 p_stream->fmt.i_codec =
913 VLC_FOURCC( 'a', 'r', 'a', 'w' );
915 case WAVE_FORMAT_MPEG:
916 case WAVE_FORMAT_MPEGLAYER3:
917 p_stream->fmt.i_codec =
918 VLC_FOURCC( 'm', 'p', 'g', 'a' );
920 case WAVE_FORMAT_A52:
921 p_stream->fmt.i_codec =
922 VLC_FOURCC( 'a', '5', '2', ' ' );
924 case WAVE_FORMAT_WMA1:
925 p_stream->fmt.i_codec =
926 VLC_FOURCC( 'w', 'm', 'a', '1' );
928 case WAVE_FORMAT_WMA2:
929 p_stream->fmt.i_codec =
930 VLC_FOURCC( 'w', 'm', 'a', '2' );
933 p_stream->fmt.i_codec = VLC_FOURCC( 'm', 's',
934 ( i_format_tag >> 8 ) & 0xff,
935 i_format_tag & 0xff );
938 msg_Dbg( p_input, "found audio header of type: %.4s",
939 (char *)&p_stream->fmt.i_codec );
940 msg_Dbg( p_input, "audio:0x%4.4x channels:%d %dHz "
941 "%dbits/sample %dkb/s",
943 p_stream->fmt.audio.i_channels,
944 p_stream->fmt.audio.i_rate,
945 p_stream->fmt.audio.i_bitspersample,
946 p_stream->fmt.i_bitrate / 1024 );
948 /* Check for text (subtitles) header */
949 else if( !strncmp(st->streamtype, "text", 4) )
951 /* We need to get rid of the header packet */
952 ogg_stream_packetout( &p_stream->os, &oggpacket );
954 msg_Dbg( p_input, "found text subtitles header" );
955 p_stream->fmt.i_cat = SPU_ES;
956 p_stream->fmt.i_codec = VLC_FOURCC('s','u','b','t');
957 p_stream->f_rate = 1000; /* granulepos is in milisec */
961 msg_Dbg( p_input, "stream %d has a header marker "
962 "but is of an unknown type", p_ogg->i_streams-1 );
969 msg_Dbg( p_input, "stream %d is of unknown type",
970 p_ogg->i_streams-1 );
975 if( Ogg_ReadPage( p_input, p_ogg, &oggpage ) != VLC_SUCCESS )
979 /* This is the first data page, which means we are now finished
980 * with the initial pages. We just need to store it in the relevant
982 for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
984 if( ogg_stream_pagein( &p_ogg->pp_stream[i_stream]->os,
999 /*****************************************************************************
1000 * Activate: initializes ogg demux structures
1001 *****************************************************************************/
1002 static int Activate( vlc_object_t * p_this )
1004 input_thread_t *p_input = (input_thread_t *)p_this;
1008 p_input->p_demux_data = NULL;
1009 b_forced = ( ( *p_input->psz_demux )&&
1010 ( !strncmp( p_input->psz_demux, "ogg", 10 ) ) ) ? 1 : 0;
1012 /* Check if we are dealing with an ogg stream */
1013 if( !b_forced && ( Ogg_Check( p_input ) != VLC_SUCCESS ) )
1016 /* Allocate p_ogg */
1017 if( !( p_ogg = malloc( sizeof( demux_sys_t ) ) ) )
1019 msg_Err( p_input, "out of memory" );
1022 memset( p_ogg, 0, sizeof( demux_sys_t ) );
1023 p_input->p_demux_data = p_ogg;
1024 p_ogg->pp_stream = NULL;
1026 /* Initialize the Ogg physical bitstream parser */
1027 ogg_sync_init( &p_ogg->oy );
1029 /* Set exported functions */
1030 p_input->pf_demux = Demux;
1031 p_input->pf_demux_control = Control;
1033 /* Initialize access plug-in structures. */
1034 if( p_input->i_mtu == 0 )
1036 /* Improve speed. */
1037 p_input->i_bufsize = INPUT_DEFAULT_BUFSIZE;
1040 /* Create one program */
1041 vlc_mutex_lock( &p_input->stream.stream_lock );
1042 if( input_InitStream( p_input, 0 ) == -1)
1044 vlc_mutex_unlock( &p_input->stream.stream_lock );
1045 msg_Err( p_input, "cannot init stream" );
1048 if( input_AddProgram( p_input, 0, 0) == NULL )
1050 vlc_mutex_unlock( &p_input->stream.stream_lock );
1051 msg_Err( p_input, "cannot add program" );
1054 p_input->stream.p_selected_program = p_input->stream.pp_programs[0];
1055 vlc_mutex_unlock( &p_input->stream.stream_lock );
1057 /* Begnning of stream, tell the demux to look for elementary streams. */
1060 p_ogg->i_old_synchro_state = !SYNCHRO_REINIT;
1065 Deactivate( (vlc_object_t *)p_input );
1070 /****************************************************************************
1071 * Ogg_BeginningOfStream: Look for Beginning of Stream ogg pages and add
1072 * Elementary streams.
1073 ****************************************************************************/
1074 static int Ogg_BeginningOfStream( input_thread_t *p_input, demux_sys_t *p_ogg)
1078 /* Find the logical streams embedded in the physical stream and
1079 * initialize our p_ogg structure. */
1080 if( Ogg_FindLogicalStreams( p_input, p_ogg ) != VLC_SUCCESS )
1082 msg_Warn( p_input, "couldn't find any ogg logical stream" );
1083 return VLC_EGENERIC;
1086 vlc_mutex_lock( &p_input->stream.stream_lock );
1087 p_input->stream.i_mux_rate = 0;
1088 vlc_mutex_unlock( &p_input->stream.stream_lock );
1090 for( i_stream = 0 ; i_stream < p_ogg->i_streams; i_stream++ )
1092 #define p_stream p_ogg->pp_stream[i_stream]
1093 if( p_stream->fmt.i_codec != VLC_FOURCC('f','l','a','c') )
1094 p_stream->p_es = es_out_Add( p_input->p_es_out, &p_stream->fmt );
1096 vlc_mutex_lock( &p_input->stream.stream_lock );
1097 p_input->stream.i_mux_rate += (p_stream->fmt.i_bitrate / ( 8 * 50 ));
1098 vlc_mutex_unlock( &p_input->stream.stream_lock );
1100 p_stream->i_pcr = p_stream->i_previous_pcr =
1101 p_stream->i_interpolated_pcr = -1;
1102 p_stream->b_reinit = 0;
1109 /****************************************************************************
1110 * Ogg_EndOfStream: clean up the ES when an End of Stream is detected.
1111 ****************************************************************************/
1112 static void Ogg_EndOfStream( input_thread_t *p_input, demux_sys_t *p_ogg )
1116 #define p_stream p_ogg->pp_stream[i_stream]
1117 for( i_stream = 0 ; i_stream < p_ogg->i_streams; i_stream++ )
1119 if( p_stream->p_es )
1120 es_out_Del( p_input->p_es_out, p_stream->p_es );
1122 vlc_mutex_lock( &p_input->stream.stream_lock );
1123 p_input->stream.i_mux_rate -= (p_stream->fmt.i_bitrate / ( 8 * 50 ));
1124 vlc_mutex_unlock( &p_input->stream.stream_lock );
1126 ogg_stream_clear( &p_ogg->pp_stream[i_stream]->os );
1127 for( j = 0; j < p_ogg->pp_stream[i_stream]->i_packets_backup; j++ )
1129 free( p_ogg->pp_stream[i_stream]->p_packets_backup[j].packet );
1131 if( p_ogg->pp_stream[i_stream]->p_packets_backup)
1132 free( p_ogg->pp_stream[i_stream]->p_packets_backup );
1134 es_format_Clean( &p_stream->fmt );
1136 free( p_ogg->pp_stream[i_stream] );
1141 if( p_ogg->pp_stream ) free( p_ogg->pp_stream );
1142 p_ogg->pp_stream = NULL;
1143 p_ogg->i_streams = 0;
1146 /*****************************************************************************
1147 * Deactivate: frees unused data
1148 *****************************************************************************/
1149 static void Deactivate( vlc_object_t *p_this )
1151 input_thread_t *p_input = (input_thread_t *)p_this;
1152 demux_sys_t *p_ogg = (demux_sys_t *)p_input->p_demux_data ;
1156 /* Cleanup the bitstream parser */
1157 ogg_sync_clear( &p_ogg->oy );
1159 Ogg_EndOfStream( p_input, p_ogg );
1165 /*****************************************************************************
1166 * Demux: reads and demuxes data packets
1167 *****************************************************************************
1168 * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
1169 *****************************************************************************/
1170 static int Demux( input_thread_t * p_input )
1172 demux_sys_t *p_ogg = (demux_sys_t *)p_input->p_demux_data;
1174 ogg_packet oggpacket;
1177 #define p_stream p_ogg->pp_stream[i_stream]
1179 if( p_ogg->i_eos == p_ogg->i_streams )
1183 msg_Dbg( p_input, "end of a group of logical streams" );
1184 Ogg_EndOfStream( p_input, p_ogg );
1187 if( Ogg_BeginningOfStream( p_input, p_ogg ) != VLC_SUCCESS ) return 0;
1190 msg_Dbg( p_input, "beginning of a group of logical streams" );
1192 p_input->stream.p_selected_program->i_synchro_state = SYNCHRO_REINIT;
1193 input_ClockManageRef( p_input, p_input->stream.p_selected_program, 0 );
1196 if( p_input->stream.p_selected_program->i_synchro_state == SYNCHRO_REINIT )
1198 msg_Warn( p_input, "synchro reinit" );
1200 if( p_ogg->i_old_synchro_state != SYNCHRO_REINIT )
1201 for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
1203 /* we'll trash all the data until we find the next pcr */
1204 p_stream->b_reinit = 1;
1205 p_stream->i_pcr = -1;
1206 p_stream->i_interpolated_pcr = -1;
1207 ogg_stream_reset( &p_stream->os );
1209 if( p_ogg->i_old_synchro_state != SYNCHRO_REINIT )
1210 ogg_sync_reset( &p_ogg->oy );
1214 * Demux an ogg page from the stream
1216 if( Ogg_ReadPage( p_input, p_ogg, &oggpage ) != VLC_SUCCESS )
1221 /* Test for End of Stream */
1222 if( ogg_page_eos( &oggpage ) ) p_ogg->i_eos++;
1225 for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
1227 if( ogg_stream_pagein( &p_stream->os, &oggpage ) != 0 )
1230 while( ogg_stream_packetout( &p_stream->os, &oggpacket ) > 0 )
1232 if( p_stream->b_reinit )
1234 /* If synchro is re-initialized we need to drop all the packets
1235 * until we find a new dated one. */
1236 Ogg_UpdatePCR( p_stream, &oggpacket );
1238 if( p_stream->i_pcr >= 0 )
1240 p_stream->b_reinit = 0;
1244 p_stream->i_interpolated_pcr = -1;
1248 /* An Ogg/vorbis packet contains an end date granulepos */
1249 if( p_stream->fmt.i_codec == VLC_FOURCC( 'v','o','r','b' ) ||
1250 p_stream->fmt.i_codec == VLC_FOURCC( 's','p','x',' ' ) ||
1251 p_stream->fmt.i_codec == VLC_FOURCC( 'f','l','a','c' ) )
1253 if( ogg_stream_packetout( &p_stream->os, &oggpacket ) > 0 )
1255 Ogg_DecodePacket( p_input, p_stream, &oggpacket );
1259 input_ClockManageRef( p_input,
1260 p_input->stream.p_selected_program,
1267 Ogg_DecodePacket( p_input, p_stream, &oggpacket );
1272 i_stream = 0; p_ogg->i_pcr = -1;
1273 for( ; i_stream < p_ogg->i_streams; i_stream++ )
1275 if( p_stream->fmt.i_cat == SPU_ES )
1277 if( p_stream->i_interpolated_pcr < 0 )
1280 if( p_ogg->i_pcr < 0 || p_stream->i_interpolated_pcr < p_ogg->i_pcr )
1281 p_ogg->i_pcr = p_stream->i_interpolated_pcr;
1284 if( p_ogg->i_pcr >= 0 )
1286 input_ClockManageRef( p_input, p_input->stream.p_selected_program,
1292 p_ogg->i_old_synchro_state =
1293 p_input->stream.p_selected_program->i_synchro_state;
1298 /*****************************************************************************
1300 *****************************************************************************/
1301 static int Control( input_thread_t *p_input, int i_query, va_list args )
1303 demux_sys_t *p_ogg = (demux_sys_t *)p_input->p_demux_data;
1308 case DEMUX_GET_TIME:
1309 pi64 = (int64_t*)va_arg( args, int64_t * );
1310 *pi64 = p_ogg->i_pcr * 100 / 9;
1314 return demux_vaControlDefault( p_input, i_query, args );