]> git.sesse.net Git - vlc/blobdiff - modules/mux/ogg.c
* all: remove sout_format_t and use es_format_t instead.
[vlc] / modules / mux / ogg.c
index b398397286f59379cf98352cc8766e932fa90cc4..3e84a6f615a285a6b8a1b6a9548ee1ada1948d89 100644 (file)
@@ -1,8 +1,8 @@
 /*****************************************************************************
- * ogg.c
+ * ogg.c: ogg muxer module for vlc
  *****************************************************************************
  * Copyright (C) 2001, 2002 VideoLAN
- * $Id: ogg.c,v 1.6 2003/06/23 23:51:31 gbazin Exp $
+ * $Id: ogg.c,v 1.21 2003/11/21 15:32:08 fenrir Exp $
  *
  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
  *          Gildas Bazin <gbazin@netcourrier.com>
  *****************************************************************************/
 #include <stdlib.h>
 #include <string.h>
-#include <sys/types.h>
+
+#ifdef HAVE_TIME_H
+#   include <time.h>
+#endif
 
 #include <vlc/vlc.h>
 #include <vlc/input.h>
 /*****************************************************************************
  * Exported prototypes
  *****************************************************************************/
-static int     Open   ( vlc_object_t * );
-static void    Close  ( vlc_object_t * );
+static int  Open   ( vlc_object_t * );
+static void Close  ( vlc_object_t * );
 
 static int Capability(sout_mux_t *, int, void *, void * );
 static int AddStream( sout_mux_t *, sout_input_t * );
 static int DelStream( sout_mux_t *, sout_input_t * );
 static int Mux      ( sout_mux_t * );
 
+static sout_buffer_t *OggCreateHeader( sout_mux_t *, mtime_t );
+static sout_buffer_t *OggCreateFooter( sout_mux_t *, mtime_t );
+
 /*****************************************************************************
  * Module descriptor
  *****************************************************************************/
@@ -60,29 +66,39 @@ vlc_module_begin();
 vlc_module_end();
 
 /*****************************************************************************
- *
+ * Misc declarations
  *****************************************************************************/
 #define FREE( p ) if( p ) { free( p ); (p) = NULL; }
 
+/* Structures used for OggDS headers used in ogm files */
+
 #define PACKET_TYPE_HEADER   0x01
 #define PACKET_TYPE_COMMENT  0x03
+#define PACKET_IS_SYNCPOINT  0x08
 
-#define PACKET_IS_SYNCPOINT      0x08
-
-typedef struct __attribute__((__packed__))
+typedef struct
+#ifdef HAVE_ATTRIBUTE_PACKED
+    __attribute__((__packed__))
+#endif
 {
     int32_t i_width;
     int32_t i_height;
-} ogg_stream_header_video_t;
+} oggds_header_video_t;
 
-typedef struct __attribute__((__packed__))
+typedef struct
+#ifdef HAVE_ATTRIBUTE_PACKED
+    __attribute__((__packed__))
+#endif
 {
     int16_t i_channels;
     int16_t i_block_align;
     int32_t i_avgbytespersec;
-} ogg_stream_header_audio_t;
+} oggds_header_audio_t;
 
-typedef struct __attribute__((__packed__))
+typedef struct
+#ifdef HAVE_ATTRIBUTE_PACKED
+    __attribute__((__packed__))
+#endif
 {
     uint8_t i_packet_type;
 
@@ -97,38 +113,20 @@ typedef struct __attribute__((__packed__))
 
     int32_t i_buffer_size;
     int16_t i_bits_per_sample;
-    int16_t i_padding_0;            // hum hum
+
+    int16_t i_padding_0; /* Because the original is using MSVC packing style */
+
     union
     {
-        ogg_stream_header_video_t video;
-        ogg_stream_header_audio_t audio;
+        oggds_header_video_t video;
+        oggds_header_audio_t audio;
     } header;
 
-} ogg_stream_header_t;
-
-
-typedef struct
-{
-    int i_cat;
-    int i_fourcc;
-
-    ogg_stream_header_t header;
+    int32_t i_padding_1; /* Because the original is using MSVC packing style */
 
-    int i_packet_no;
+} oggds_header_t;
 
-    mtime_t             i_dts;
-    mtime_t             i_length;
-    ogg_stream_state    os;
-
-} ogg_stream_t;
-
-struct sout_mux_sys_t
-{
-    int     b_write_header;
-    int     i_streams;
-
-    mtime_t i_start_dts;
-};
+/* Helper writer functions */
 
 #define SetWLE( p, v ) _SetWLE( (uint8_t*)p, v)
 static void _SetWLE( uint8_t *p, uint16_t i_dw )
@@ -152,12 +150,101 @@ static void _SetQWLE( uint8_t *p, uint64_t i_qw )
     SetDWLE( p+4, ( i_qw >> 32)&0xffffffff );
 }
 
+/*
+ * TODO  move this function to src/stream_output.c (used by nearly all muxers)
+ */
+static int MuxGetStream( sout_mux_t *p_mux, int *pi_stream, mtime_t *pi_dts )
+{
+    mtime_t i_dts;
+    int     i_stream;
+    int     i;
+
+    for( i = 0, i_dts = 0, i_stream = -1; i < p_mux->i_nb_inputs; i++ )
+    {
+        sout_fifo_t  *p_fifo;
+
+        p_fifo = p_mux->pp_inputs[i]->p_fifo;
+
+        /* We don't really need to have anything in the SPU fifo */
+        if( p_mux->pp_inputs[i]->p_fmt->i_cat == SPU_ES &&
+            p_fifo->i_depth == 0 ) continue;
+
+        if( p_fifo->i_depth > 2 ||
+            /* Special case for SPUs */
+            ( p_mux->pp_inputs[i]->p_fmt->i_cat == SPU_ES &&
+              p_fifo->i_depth > 0 ) )
+        {
+            sout_buffer_t *p_buf;
+
+            p_buf = sout_FifoShow( p_fifo );
+            if( i_stream < 0 || p_buf->i_dts < i_dts )
+            {
+                i_dts = p_buf->i_dts;
+                i_stream = i;
+            }
+        }
+        else
+        {
+            // wait that all fifo have at least 3 packets (3 vorbis headers)
+            return( -1 );
+        }
+    }
+    if( pi_stream )
+    {
+        *pi_stream = i_stream;
+    }
+    if( pi_dts )
+    {
+        *pi_dts = i_dts;
+    }
+    return( i_stream );
+}
+
+/*****************************************************************************
+ * Definitions of structures and functions used by this plugins 
+ *****************************************************************************/
+typedef struct
+{
+    int i_cat;
+    int i_fourcc;
+
+    int b_new;
+
+    mtime_t i_dts;
+    mtime_t i_length;
+    int     i_packet_no;
+    int     i_serial_no;
+    int     i_keyframe_granule_shift; /* Theora only */
+    ogg_stream_state os;
+
+    oggds_header_t oggds_header;
+
+    sout_buffer_t *pp_sout_headers[3];
+    int           i_sout_headers;
+
+} ogg_stream_t;
+
+struct sout_mux_sys_t
+{
+    int     i_streams;
+
+    mtime_t i_start_dts;
+    int     i_next_serial_no;
+
+    /* number of logical streams pending to be added */
+    int i_add_streams;
+
+    /* logical streams pending to be deleted */
+    int i_del_streams;
+    ogg_stream_t **pp_del_streams;
+};
+
 static void OggSetDate( sout_buffer_t *, mtime_t , mtime_t  );
 static sout_buffer_t *OggStreamFlush( sout_mux_t *, ogg_stream_state *,
                                       mtime_t );
 
 /*****************************************************************************
- * Open:
+ * Open: Open muxer
  *****************************************************************************/
 static int Open( vlc_object_t *p_this )
 {
@@ -168,7 +255,9 @@ static int Open( vlc_object_t *p_this )
 
     p_sys                 = malloc( sizeof( sout_mux_sys_t ) );
     p_sys->i_streams      = 0;
-    p_sys->b_write_header = VLC_TRUE;
+    p_sys->i_add_streams  = 0;
+    p_sys->i_del_streams  = 0;
+    p_sys->pp_del_streams = 0;
 
     p_mux->p_sys        = p_sys;
     p_mux->pf_capacity  = Capability;
@@ -177,20 +266,50 @@ static int Open( vlc_object_t *p_this )
     p_mux->pf_mux       = Mux;
     p_mux->i_preheader  = 1;
 
+    /* First serial number is random.
+     * (Done like this because on win32 you need to seed the random number
+     *  generator once per thread). */
+    srand( (unsigned int)time( NULL ) );
+    p_sys->i_next_serial_no = rand();
+
     return VLC_SUCCESS;
 }
 
 /*****************************************************************************
- * Close:
+ * Close: Finalize ogg bitstream and close muxer
  *****************************************************************************/
-
 static void Close( vlc_object_t * p_this )
 {
-    sout_mux_t      *p_mux = (sout_mux_t*)p_this;
-    sout_mux_sys_t  *p_sys = p_mux->p_sys;
+    sout_mux_t     *p_mux = (sout_mux_t*)p_this;
+    sout_mux_sys_t *p_sys = p_mux->p_sys;
 
     msg_Info( p_mux, "Close" );
 
+    if( p_sys->i_del_streams )
+    {
+        sout_buffer_t *p_og = NULL;
+        mtime_t i_dts = -1;
+        int i;
+
+        /* Close the current ogg stream */
+        msg_Dbg( p_mux, "writing footer" );
+        sout_BufferChain( &p_og, OggCreateFooter( p_mux, 0 ) );
+
+        /* Remove deleted logical streams */
+        for( i = 0; i < p_sys->i_del_streams; i++ )
+        {
+            i_dts = p_sys->pp_del_streams[i]->i_dts;
+            ogg_stream_clear( &p_sys->pp_del_streams[i]->os );
+            FREE( p_sys->pp_del_streams[i] );
+        }
+        FREE( p_sys->pp_del_streams );
+        p_sys->i_streams -= p_sys->i_del_streams;
+
+        /* Write footer */
+        OggSetDate( p_og, i_dts, 0 );
+        sout_AccessOutWrite( p_mux->p_access, p_og );
+    }
+
     free( p_sys );
 }
 
@@ -200,57 +319,67 @@ static int Capability( sout_mux_t *p_mux, int i_query, void *p_args,
    switch( i_query )
    {
         case SOUT_MUX_CAP_GET_ADD_STREAM_ANY_TIME:
-            *(vlc_bool_t*)p_answer = VLC_FALSE;
+            *(vlc_bool_t*)p_answer = VLC_TRUE;
             return( SOUT_MUX_CAP_ERR_OK );
         default:
             return( SOUT_MUX_CAP_ERR_UNIMPLEMENTED );
    }
 }
 
+/*****************************************************************************
+ * AddStream: Add an elementary stream to the muxed stream
+ *****************************************************************************/
 static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input )
 {
-    sout_mux_sys_t  *p_sys = p_mux->p_sys;
-    ogg_stream_t        *p_stream;
+    sout_mux_sys_t *p_sys = p_mux->p_sys;
+    ogg_stream_t   *p_stream;
 
     msg_Dbg( p_mux, "adding input" );
-    p_input->p_sys = (void*)p_stream = malloc( sizeof( ogg_stream_t ) );
+
+    p_input->p_sys = (void *)p_stream = malloc( sizeof( ogg_stream_t ) );
 
     p_stream->i_cat       = p_input->p_fmt->i_cat;
-    p_stream->i_fourcc    = p_input->p_fmt->i_fourcc;
+    p_stream->i_fourcc    = p_input->p_fmt->i_codec;
+    p_stream->i_serial_no = p_sys->i_next_serial_no++;
     p_stream->i_packet_no = 0;
 
-    p_stream->header.i_packet_type = PACKET_TYPE_HEADER;
+    p_stream->i_sout_headers = 0;
+
+    memset( &p_stream->oggds_header, 0, sizeof(p_stream->oggds_header) );
+    p_stream->oggds_header.i_packet_type = PACKET_TYPE_HEADER;
     switch( p_input->p_fmt->i_cat )
     {
     case VIDEO_ES:
-        switch( p_input->p_fmt->i_fourcc )
+        switch( p_stream->i_fourcc )
         {
         case VLC_FOURCC( 'm', 'p','4', 'v' ):
         case VLC_FOURCC( 'D', 'I','V', '3' ):
-            memcpy( p_stream->header.stream_type, "video    ", 8 );
-            if( p_input->p_fmt->i_fourcc == VLC_FOURCC( 'm', 'p','4', 'v' ) )
+            memcpy( p_stream->oggds_header.stream_type, "video", 5 );
+            if( p_stream->i_fourcc == VLC_FOURCC( 'm', 'p','4', 'v' ) )
             {
-                memcpy( p_stream->header.sub_type, "XVID", 4 );
+                memcpy( p_stream->oggds_header.sub_type, "XVID", 4 );
             }
-            else if( p_input->p_fmt->i_fourcc ==
-                     VLC_FOURCC( 'D', 'I','V', '3' ) )
+            else if( p_stream->i_fourcc == VLC_FOURCC( 'D', 'I','V', '3' ) )
             {
-                memcpy( p_stream->header.sub_type, "DIV3", 4 );
+                memcpy( p_stream->oggds_header.sub_type, "DIV3", 4 );
             }
-            SetDWLE( &p_stream->header.i_size,
-                     sizeof( ogg_stream_header_t ) - 1);
-            /* XXX this won't make mplayer happy,
-             * but vlc can read that without any problem so...*/
-            SetQWLE( &p_stream->header.i_time_unit, 10*1000 );
-            //(int64_t)10*1000*1000/(int64_t)25 );  // FIXME (25fps)
-            SetQWLE( &p_stream->header.i_samples_per_unit, 1 );
-            SetDWLE( &p_stream->header.i_default_len, 0 );      /* ??? */
-            SetDWLE( &p_stream->header.i_buffer_size, 1024*1024 );
-            SetWLE( &p_stream->header.i_bits_per_sample, 0 );
-            SetDWLE( &p_stream->header.header.video.i_width,
-                     p_input->p_fmt->i_width );
-            SetDWLE( &p_stream->header.header.video.i_height,
-                     p_input->p_fmt->i_height );
+            SetDWLE( &p_stream->oggds_header.i_size,
+                     sizeof( oggds_header_t ) - 1);
+            SetQWLE( &p_stream->oggds_header.i_time_unit,
+                     I64C(10000000)/(int64_t)25 );  // FIXME (25fps)
+            SetQWLE( &p_stream->oggds_header.i_samples_per_unit, 1 );
+            SetDWLE( &p_stream->oggds_header.i_default_len, 1 ); /* ??? */
+            SetDWLE( &p_stream->oggds_header.i_buffer_size, 1024*1024 );
+            SetWLE( &p_stream->oggds_header.i_bits_per_sample, 0 );
+            SetDWLE( &p_stream->oggds_header.header.video.i_width,
+                     p_input->p_fmt->video.i_width );
+            SetDWLE( &p_stream->oggds_header.header.video.i_height,
+                     p_input->p_fmt->video.i_height );
+            msg_Dbg( p_mux, "mp4v/div3 stream" );
+            break;
+
+        case VLC_FOURCC( 't', 'h', 'e', 'o' ):
+            msg_Dbg( p_mux, "theora stream" );
             break;
 
         default:
@@ -258,124 +387,129 @@ static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input )
             return( VLC_EGENERIC );
         }
         break;
+
     case AUDIO_ES:
-        switch( p_input->p_fmt->i_fourcc )
+        switch( p_stream->i_fourcc )
         {
         case VLC_FOURCC( 'm', 'p','g', 'a' ):
         case VLC_FOURCC( 'a', '5','2', ' ' ):
-            memcpy( p_stream->header.stream_type, "audio    ", 8 );
-            if( p_input->p_fmt->i_fourcc == VLC_FOURCC( 'm', 'p','g', 'a' ) )
+            memcpy( p_stream->oggds_header.stream_type, "audio", 5 );
+            if( p_stream->i_fourcc == VLC_FOURCC( 'm', 'p','g', 'a' ) )
             {
-                memcpy( p_stream->header.sub_type, "55  ", 4 );
+                memcpy( p_stream->oggds_header.sub_type, "55  ", 4 );
             }
-            else if( p_input->p_fmt->i_fourcc ==
-                     VLC_FOURCC( 'a', '5','2', ' ' ) )
+            else if( p_stream->i_fourcc == VLC_FOURCC( 'a', '5','2', ' ' ) )
             {
-                memcpy( p_stream->header.sub_type, "2000", 4 );
+                memcpy( p_stream->oggds_header.sub_type, "2000", 4 );
             }
-            SetDWLE( &p_stream->header.i_size,
-                     sizeof( ogg_stream_header_t ) - 1);
-            SetQWLE( &p_stream->header.i_time_unit, 1000000 );  /* is it used ? */
-            SetDWLE( &p_stream->header.i_default_len, 0 );      /* ??? */
-            SetDWLE( &p_stream->header.i_buffer_size, 30*1024 );
-            SetQWLE( &p_stream->header.i_samples_per_unit,
-                     p_input->p_fmt->i_sample_rate );
-            SetWLE( &p_stream->header.i_bits_per_sample, 0 );
-            SetDWLE( &p_stream->header.header.audio.i_channels,
-                     p_input->p_fmt->i_channels );
-            SetDWLE( &p_stream->header.header.audio.i_block_align,
-                     p_input->p_fmt->i_block_align );
-            SetDWLE( &p_stream->header.header.audio.i_avgbytespersec, 0 );
+            SetDWLE( &p_stream->oggds_header.i_size,
+                     sizeof( oggds_header_t ) - 1);
+            SetQWLE( &p_stream->oggds_header.i_time_unit, 0 /* not used */ );
+            SetDWLE( &p_stream->oggds_header.i_default_len, 1 );
+            SetDWLE( &p_stream->oggds_header.i_buffer_size, 30*1024 );
+            SetQWLE( &p_stream->oggds_header.i_samples_per_unit,
+                     p_input->p_fmt->audio.i_rate );
+            SetWLE( &p_stream->oggds_header.i_bits_per_sample, 0 );
+            SetDWLE( &p_stream->oggds_header.header.audio.i_channels,
+                     p_input->p_fmt->audio.i_channels );
+            SetDWLE( &p_stream->oggds_header.header.audio.i_block_align,
+                     p_input->p_fmt->audio.i_blockalign );
+            SetDWLE( &p_stream->oggds_header.header.audio.i_avgbytespersec, 0);
+            msg_Dbg( p_mux, "mpga/a52 stream" );
             break;
 
         case VLC_FOURCC( 'v', 'o', 'r', 'b' ):
-          msg_Dbg( p_mux, "vorbis stream" );
-          break;
+            msg_Dbg( p_mux, "vorbis stream" );
+            break;
+
+        case VLC_FOURCC( 's', 'p', 'x', ' ' ):
+            msg_Dbg( p_mux, "speex stream" );
+            break;
+
         default:
             FREE( p_input->p_sys );
             return( VLC_EGENERIC );
         }
         break;
 
+    case SPU_ES:
+        switch( p_stream->i_fourcc )
+        {
+        case VLC_FOURCC( 's', 'u','b', 't' ):
+            memcpy( p_stream->oggds_header.stream_type, "text", 4 );
+            msg_Dbg( p_mux, "subtitles stream" );
+            break;
+
+        default:
+            FREE( p_input->p_sys );
+            return( VLC_EGENERIC );
+        }
+        break;
     default:
         FREE( p_input->p_sys );
         return( VLC_EGENERIC );
     }
 
-    ogg_stream_init( &p_stream->os, rand () );
+    p_stream->b_new = VLC_TRUE;
+
+    p_sys->i_add_streams++;
 
-    p_sys->i_streams++;
     return( VLC_SUCCESS );
 }
 
+/*****************************************************************************
+ * DelStream: Delete an elementary stream from the muxed stream
+ *****************************************************************************/
 static int DelStream( sout_mux_t *p_mux, sout_input_t *p_input )
 {
-    ogg_stream_t        *p_stream = (ogg_stream_t*)p_input->p_sys;
-    sout_buffer_t       *p_og;
+    sout_mux_sys_t *p_sys  = p_mux->p_sys;
+    ogg_stream_t   *p_stream = (ogg_stream_t*)p_input->p_sys;
+    sout_buffer_t  *p_og;
 
     msg_Dbg( p_mux, "removing input" );
 
     /* flush all remaining data */
-
     if( p_input->p_sys )
     {
-        p_og = OggStreamFlush( p_mux, &p_stream->os, 0 );
-        if( p_og )
+        int i;
+
+        if( ( p_og = OggStreamFlush( p_mux, &p_stream->os, 0 ) ) )
         {
             OggSetDate( p_og, p_stream->i_dts, p_stream->i_length );
-
             sout_AccessOutWrite( p_mux->p_access, p_og );
         }
 
-        ogg_stream_clear( &p_stream->os );
-
-        FREE( p_input->p_sys );
-    }
-    return( 0 );
-}
-
-/*
- * TODO  move this function to src/stream_output.c (used by nearly all muxers)
- */
-static int MuxGetStream( sout_mux_t *p_mux, int *pi_stream, mtime_t *pi_dts )
-{
-    mtime_t i_dts;
-    int     i_stream;
-    int     i;
-
-    for( i = 0, i_dts = 0, i_stream = -1; i < p_mux->i_nb_inputs; i++ )
-    {
-        sout_fifo_t  *p_fifo;
-
-        p_fifo = p_mux->pp_inputs[i]->p_fifo;
-
-        if( p_fifo->i_depth > 1 )
+        for( i = 0; i < p_stream->i_sout_headers; i++ )
         {
-            sout_buffer_t *p_buf;
+            sout_BufferDelete( p_mux->p_sout, p_stream->pp_sout_headers[i] );
+            p_stream->i_sout_headers = 0;
+        }
 
-            p_buf = sout_FifoShow( p_fifo );
-            if( i_stream < 0 || p_buf->i_dts < i_dts )
-            {
-                i_dts = p_buf->i_dts;
-                i_stream = i;
-            }
+        /* move input in delete queue */
+        if( !p_stream->b_new )
+        {
+            p_sys->pp_del_streams = realloc( p_sys->pp_del_streams,
+                                             (p_sys->i_del_streams + 1) *
+                                             sizeof(ogg_stream_t *) );
+            p_sys->pp_del_streams[p_sys->i_del_streams++] = p_stream;
         }
         else
         {
-            return( -1 ); // wait that all fifo have at least 2 packets
+            /* Wasn't already added so get rid of it */
+            ogg_stream_clear( &p_stream->os );
+            FREE( p_stream );
+            p_sys->i_add_streams--;
         }
     }
-    if( pi_stream )
-    {
-        *pi_stream = i_stream;
-    }
-    if( pi_dts )
-    {
-        *pi_dts = i_dts;
-    }
-    return( i_stream );
+
+    p_input->p_sys = NULL;
+
+    return( 0 );
 }
 
+/*****************************************************************************
+ * Ogg bitstream manipulation routines
+ *****************************************************************************/
 static sout_buffer_t *OggStreamFlush( sout_mux_t *p_mux,
                                       ogg_stream_state *p_os, mtime_t i_pts )
 {
@@ -391,6 +525,7 @@ static sout_buffer_t *OggStreamFlush( sout_mux_t *p_mux,
         {
             break;
         }
+
         i_size = og.header_len + og.body_len;
         p_og = sout_BufferNew( p_mux->p_sout, i_size);
 
@@ -424,6 +559,7 @@ static sout_buffer_t *OggStreamPageOut( sout_mux_t *p_mux,
         {
             break;
         }
+
         i_size = og.header_len + og.body_len;
         p_og = sout_BufferNew( p_mux->p_sout, i_size);
 
@@ -453,64 +589,104 @@ static sout_buffer_t *OggCreateHeader( sout_mux_t *p_mux, mtime_t i_dts )
      * must appear first in the ogg stream so we take care of them first. */
     for( i = 0; i < p_mux->i_nb_inputs; i++ )
     {
-        ogg_stream_t *p_stream;
+        ogg_stream_t *p_stream = (ogg_stream_t*)p_mux->pp_inputs[i]->p_sys;
+        p_stream->b_new = VLC_FALSE;
 
-        p_stream = (ogg_stream_t*)p_mux->pp_inputs[i]->p_sys;
+        msg_Dbg( p_mux, "creating header for %4.4s",
+                 (char *)&p_stream->i_fourcc );
 
-        if( p_stream->i_fourcc == VLC_FOURCC( 'v', 'o', 'r', 'b' ) )
+        ogg_stream_init( &p_stream->os, p_stream->i_serial_no );
+        p_stream->i_packet_no = 0;
+
+        if( p_stream->i_fourcc == VLC_FOURCC( 'v', 'o', 'r', 'b' ) ||
+            p_stream->i_fourcc == VLC_FOURCC( 's', 'p', 'x', ' ' ) ||
+            p_stream->i_fourcc == VLC_FOURCC( 't', 'h', 'e', 'o' ) )
         {
             /* Special case, headers are already there in the
-             * incoming stream */
+             * incoming stream or we backed them up earlier */
 
-            /* first packet in order: vorbis info */
-            p_og = sout_FifoGet( p_mux->pp_inputs[i]->p_fifo );
-            op.packet = p_og->p_buffer;
-            op.bytes  = p_og->i_size;
-            op.b_o_s  = 1;
-            op.e_o_s  = 0;
-            op.granulepos = 0;
-            op.packetno = p_stream->i_packet_no++;
-            ogg_stream_packetin( &p_stream->os, &op );
+            /* first packet in order: vorbis/speex/theora info */
+            if( !p_stream->i_sout_headers )
+            {
+                p_og = sout_FifoGet( p_mux->pp_inputs[i]->p_fifo );
+                op.packet = p_og->p_buffer;
+                op.bytes  = p_og->i_size;
+                op.b_o_s  = 1;
+                op.e_o_s  = 0;
+                op.granulepos = 0;
+                op.packetno = p_stream->i_packet_no++;
+                ogg_stream_packetin( &p_stream->os, &op );
+                p_stream->pp_sout_headers[0] =
+                    OggStreamFlush( p_mux, &p_stream->os, 0 );
+                p_stream->i_sout_headers++;
+            }
+            p_og = sout_BufferDuplicate( p_mux->p_sout,
+                                         p_stream->pp_sout_headers[0] );
+
+            /* Get keyframe_granule_shift for theora granulepos calculation */
+            if( p_stream->i_fourcc == VLC_FOURCC( 't', 'h', 'e', 'o' ) )
+            {
+                int i_keyframe_frequency_force = 1 << (op.packet[36] >> 3);
+
+                /* granule_shift = i_log( frequency_force -1 ) */
+                p_stream->i_keyframe_granule_shift = 0;
+                i_keyframe_frequency_force--;
+                while( i_keyframe_frequency_force )
+                {
+                    p_stream->i_keyframe_granule_shift++;
+                    i_keyframe_frequency_force >>= 1;
+                }
+            }
         }
         else
         {
             /* ds header */
-            op.packet = (uint8_t*)&p_stream->header;
-            op.bytes  = sizeof( ogg_stream_t );
+            op.packet = (uint8_t*)&p_stream->oggds_header;
+            op.bytes  = sizeof( oggds_header_t );
             op.b_o_s  = 1;
             op.e_o_s  = 0;
             op.granulepos = 0;
             op.packetno = p_stream->i_packet_no++;
             ogg_stream_packetin( &p_stream->os, &op );
+            p_og = OggStreamFlush( p_mux, &p_stream->os, 0 );
         }
 
-        p_og = OggStreamFlush( p_mux, &p_stream->os, 0 );
         sout_BufferChain( &p_hdr, p_og );
     }
 
     /* Take care of the non b_o_s headers */
     for( i = 0; i < p_mux->i_nb_inputs; i++ )
     {
-        ogg_stream_t *p_stream;
+        ogg_stream_t *p_stream = (ogg_stream_t*)p_mux->pp_inputs[i]->p_sys;
 
-        p_stream = (ogg_stream_t*)p_mux->pp_inputs[i]->p_sys;
-
-        if( p_stream->i_fourcc == VLC_FOURCC( 'v', 'o', 'r', 'b' ) )
+        if( p_stream->i_fourcc == VLC_FOURCC( 'v', 'o', 'r', 'b' ) ||
+            p_stream->i_fourcc == VLC_FOURCC( 's', 'p', 'x', ' ' ) ||
+            p_stream->i_fourcc == VLC_FOURCC( 't', 'h', 'e', 'o' ) )
         {
             /* Special case, headers are already there in the incoming stream.
              * We need to gather them an mark them as headers. */
             int j;
             for( j = 0; j < 2; j++ )
             {
-                /* next packets in order: comments and codebooks */
-                p_og = sout_FifoGet( p_mux->pp_inputs[i]->p_fifo );
-                op.packet = p_og->p_buffer;
-                op.bytes  = p_og->i_size;
-                op.b_o_s  = 0;
-                op.e_o_s  = 0;
-                op.granulepos = 0;
-                op.packetno = p_stream->i_packet_no++;
-                ogg_stream_packetin( &p_stream->os, &op );
+                if( p_stream->i_sout_headers < j + 2 )
+                {
+                    /* next packets in order: comments and codebooks */
+                    p_og = sout_FifoGet( p_mux->pp_inputs[i]->p_fifo );
+                    op.packet = p_og->p_buffer;
+                    op.bytes  = p_og->i_size;
+                    op.b_o_s  = 0;
+                    op.e_o_s  = 0;
+                    op.granulepos = 0;
+                    op.packetno = p_stream->i_packet_no++;
+                    ogg_stream_packetin( &p_stream->os, &op );
+                    p_stream->pp_sout_headers[j+1] =
+                        OggStreamFlush( p_mux, &p_stream->os, 0 );
+                    p_stream->i_sout_headers++;
+                }
+
+                p_og = sout_BufferDuplicate( p_mux->p_sout,
+                                             p_stream->pp_sout_headers[j+1] );
+                sout_BufferChain( &p_hdr, p_og );
             }
         }
         else
@@ -520,7 +696,7 @@ static sout_buffer_t *OggCreateHeader( sout_mux_t *p_mux, mtime_t i_dts )
 
             /* comment */
             com[0] = PACKET_TYPE_COMMENT;
-            i_com = snprintf( &com[1], 128, "VLC 0.5.x stream output" ) + 1;
+            i_com = snprintf( &com[1], 128, VERSION" stream output" ) + 1;
             op.packet = com;
             op.bytes  = i_com;
             op.b_o_s  = 0;
@@ -528,10 +704,25 @@ static sout_buffer_t *OggCreateHeader( sout_mux_t *p_mux, mtime_t i_dts )
             op.granulepos = 0;
             op.packetno = p_stream->i_packet_no++;
             ogg_stream_packetin( &p_stream->os, &op );
+            p_og = OggStreamFlush( p_mux, &p_stream->os, 0 );
+            sout_BufferChain( &p_hdr, p_og );
         }
 
-        p_og = OggStreamFlush( p_mux, &p_stream->os, 0 );
-        sout_BufferChain( &p_hdr, p_og );
+        /* Special case for mp4v */
+        if( p_stream->i_fourcc == VLC_FOURCC( 'm', 'p', '4', 'v' ) &&
+            p_mux->pp_inputs[i]->p_fmt->i_extra )
+        {
+            /* Send a packet with the VOL data */
+            op.bytes  = p_mux->pp_inputs[i]->p_fmt->i_extra;
+            op.packet = p_mux->pp_inputs[i]->p_fmt->p_extra;
+            op.b_o_s  = 0;
+            op.e_o_s  = 0;
+            op.granulepos = 0;
+            op.packetno = p_stream->i_packet_no++;
+            ogg_stream_packetin( &p_stream->os, &op );
+            p_og = OggStreamFlush( p_mux, &p_stream->os, 0 );
+            sout_BufferChain( &p_hdr, p_og );
+        }
     }
 
     /* set HEADER flag */
@@ -542,6 +733,68 @@ static sout_buffer_t *OggCreateHeader( sout_mux_t *p_mux, mtime_t i_dts )
     return( p_hdr );
 }
 
+static sout_buffer_t *OggCreateFooter( sout_mux_t *p_mux, mtime_t i_dts )
+{
+    sout_mux_sys_t *p_sys = p_mux->p_sys;
+    sout_buffer_t *p_hdr = NULL;
+    sout_buffer_t *p_og;
+    ogg_packet    op;
+    int i;
+
+    /* flush all remaining data */
+    for( i = 0; i < p_mux->i_nb_inputs; i++ )
+    {
+        ogg_stream_t *p_stream = p_mux->pp_inputs[i]->p_sys;
+
+        /* skip newly added streams */
+        if( p_stream->b_new ) continue;
+
+        if( ( p_og = OggStreamFlush( p_mux, &p_stream->os, 0 ) ) )
+        {
+            OggSetDate( p_og, p_stream->i_dts, p_stream->i_length );
+            sout_AccessOutWrite( p_mux->p_access, p_og );
+        }
+    }
+
+    /* Write eos packets for each stream. */
+    for( i = 0; i < p_mux->i_nb_inputs; i++ )
+    {
+        ogg_stream_t *p_stream = p_mux->pp_inputs[i]->p_sys;
+
+        /* skip newly added streams */
+        if( p_stream->b_new ) continue;
+
+        op.packet = NULL;
+        op.bytes  = 0;
+        op.b_o_s  = 0;
+        op.e_o_s  = 1;
+        op.granulepos = -1;
+        op.packetno = p_stream->i_packet_no++;
+        ogg_stream_packetin( &p_stream->os, &op );
+
+        p_og = OggStreamFlush( p_mux, &p_stream->os, 0 );
+        sout_BufferChain( &p_hdr, p_og );
+        ogg_stream_clear( &p_stream->os );
+    }
+
+    for( i = 0; i < p_sys->i_del_streams; i++ )
+    {
+        op.packet = NULL;
+        op.bytes  = 0;
+        op.b_o_s  = 0;
+        op.e_o_s  = 1;
+        op.granulepos = -1;
+        op.packetno = p_sys->pp_del_streams[i]->i_packet_no++;
+        ogg_stream_packetin( &p_sys->pp_del_streams[i]->os, &op );
+
+        p_og = OggStreamFlush( p_mux, &p_sys->pp_del_streams[i]->os, 0 );
+        sout_BufferChain( &p_hdr, p_og );
+        ogg_stream_clear( &p_sys->pp_del_streams[i]->os );
+    }
+
+    return( p_hdr );
+}
+
 static void OggSetDate( sout_buffer_t *p_og, mtime_t i_dts, mtime_t i_length )
 {
     int i_count;
@@ -563,49 +816,74 @@ static void OggSetDate( sout_buffer_t *p_og, mtime_t i_dts, mtime_t i_length )
     }
 }
 
+/*****************************************************************************
+ * Mux: multiplex available data in input fifos into the Ogg bitstream
+ *****************************************************************************/
 static int Mux( sout_mux_t *p_mux )
 {
-    sout_mux_sys_t      *p_sys  = p_mux->p_sys;
-    sout_buffer_t       *p_og = NULL;
-    int                 i_stream;
-    mtime_t             i_dts;
+    sout_mux_sys_t *p_sys = p_mux->p_sys;
+    sout_buffer_t  *p_og = NULL;
+    int            i_stream;
+    mtime_t        i_dts;
 
-    if( p_sys->b_write_header )
+    if( p_sys->i_add_streams || p_sys->i_del_streams )
     {
+        /* Open new ogg stream */
         if( MuxGetStream( p_mux, &i_stream, &i_dts) < 0 )
         {
-            msg_Dbg( p_mux, "waiting data..." );
+            msg_Dbg( p_mux, "waiting for data..." );
             return( VLC_SUCCESS );
         }
-        p_sys->i_start_dts = i_dts;
+
+        if( p_sys->i_streams )
+        {
+            /* Close current ogg stream */
+            int i;
+
+            msg_Dbg( p_mux, "writing footer" );
+            sout_BufferChain( &p_og, OggCreateFooter( p_mux, 0 ) );
+
+            /* Remove deleted logical streams */
+            for( i = 0; i < p_sys->i_del_streams; i++ )
+            {
+                FREE( p_sys->pp_del_streams[i] );
+            }
+            FREE( p_sys->pp_del_streams );
+            p_sys->i_streams = 0;
+        }
 
         msg_Dbg( p_mux, "writing header" );
+        p_sys->i_start_dts = i_dts;
+        p_sys->i_streams = p_mux->i_nb_inputs;
+        p_sys->i_del_streams = 0;
+        p_sys->i_add_streams = 0;
         sout_BufferChain( &p_og, OggCreateHeader( p_mux, i_dts ) );
-        p_sys->b_write_header = VLC_FALSE;
+
+        /* Write header and/or footer */
+        OggSetDate( p_og, i_dts, 0 );
+        sout_AccessOutWrite( p_mux->p_access, p_og );
+        p_og = NULL;
     }
 
     for( ;; )
     {
-        sout_input_t    *p_input;
-        ogg_stream_t    *p_stream;
-        sout_buffer_t   *p_data;
-        ogg_packet          op;
+        sout_input_t  *p_input;
+        ogg_stream_t  *p_stream;
+        sout_buffer_t *p_data;
+        ogg_packet    op;
 
         if( MuxGetStream( p_mux, &i_stream, &i_dts) < 0 )
         {
-            //msg_Dbg( p_mux, "waiting data..." );
             return( VLC_SUCCESS );
         }
-        //msg_Dbg( p_mux, "doing job" );
-
-        if( p_sys->i_start_dts <= 0 ) p_sys->i_start_dts = i_dts;
 
         p_input  = p_mux->pp_inputs[i_stream];
         p_stream = (ogg_stream_t*)p_input->p_sys;
+        p_data   = sout_FifoGet( p_input->p_fifo );
 
-        p_data  = sout_FifoGet( p_input->p_fifo );
-
-        if( p_stream->i_fourcc != VLC_FOURCC( 'v', 'o', 'r', 'b' ) )
+        if( p_stream->i_fourcc != VLC_FOURCC( 'v', 'o', 'r', 'b' ) &&
+            p_stream->i_fourcc != VLC_FOURCC( 's', 'p', 'x', ' ' ) &&
+            p_stream->i_fourcc != VLC_FOURCC( 't', 'h', 'e', 'o' ) )
         {
             sout_BufferReallocFromPreHeader( p_mux->p_sout, p_data, 1 );
             p_data->p_buffer[0] = PACKET_IS_SYNCPOINT;      // FIXME
@@ -619,29 +897,54 @@ static int Mux( sout_mux_t *p_mux )
 
         if( p_stream->i_cat == AUDIO_ES )
         {
-            if( p_stream->i_fourcc == VLC_FOURCC( 'v', 'o', 'r', 'b' ) )
+            if( p_stream->i_fourcc == VLC_FOURCC( 'v', 'o', 'r', 'b' ) ||
+                p_stream->i_fourcc == VLC_FOURCC( 's', 'p', 'x', ' ' ) )
             {
                 /* number of sample from begining + current packet */
                 op.granulepos =
                     ( i_dts + p_data->i_length - p_sys->i_start_dts ) *
-                    p_input->p_fmt->i_sample_rate / (int64_t)1000000;
+                    p_input->p_fmt->audio.i_rate / I64C(1000000);
             }
             else
             {
                 /* number of sample from begining */
                 op.granulepos = ( i_dts - p_sys->i_start_dts ) *
-                    p_stream->header.i_samples_per_unit / (int64_t)1000000;
+                    p_stream->oggds_header.i_samples_per_unit / I64C(1000000);
             }
         }
         else if( p_stream->i_cat == VIDEO_ES )
         {
+            if( p_stream->i_fourcc == VLC_FOURCC( 't', 'h', 'e', 'o' ) )
+            {
+                /* FIXME, we assume only keyframes and 25fps */
+                op.granulepos = ( ( i_dts - p_sys->i_start_dts ) * I64C(25)
+                    / I64C(1000000) ) << p_stream->i_keyframe_granule_shift;
+            }
+            else
+                op.granulepos = ( i_dts - p_sys->i_start_dts ) * I64C(10) /
+                    p_stream->oggds_header.i_time_unit;
+        }
+        else if( p_stream->i_cat == SPU_ES )
+        {
+            /* granulepos is in milisec */
             op.granulepos = ( i_dts - p_sys->i_start_dts ) / 1000;
         }
 
         ogg_stream_packetin( &p_stream->os, &op );
 
-        sout_BufferChain( &p_og, OggStreamPageOut( p_mux, &p_stream->os,
-                                                   p_data->i_dts ) );
+        if( p_stream->i_cat == SPU_ES ||
+            p_stream->i_fourcc == VLC_FOURCC( 's', 'p', 'x', ' ' ) )
+        {
+            /* Subtitles or Speex packets are quite small so they 
+             * need to be flushed to be sent on time */
+            sout_BufferChain( &p_og, OggStreamFlush( p_mux, &p_stream->os,
+                                                     p_data->i_dts ) );
+        }
+        else
+        {
+            sout_BufferChain( &p_og, OggStreamPageOut( p_mux, &p_stream->os,
+                                                       p_data->i_dts ) );
+        }
 
         if( p_og )
         {
@@ -649,7 +952,6 @@ static int Mux( sout_mux_t *p_mux )
             p_stream->i_dts = -1;
             p_stream->i_length = 0;
 
-            msg_Dbg( p_mux, "writing data..." );
             sout_AccessOutWrite( p_mux->p_access, p_og );
 
             p_og = NULL;