]> git.sesse.net Git - vlc/blobdiff - modules/demux/ps.c
Improvements to preferences
[vlc] / modules / demux / ps.c
index be2ec269010a39672da13402cdb2393bc1dba2f0..c069c0f7042e03fa3c5a4226ca9561c3d6103a18 100644 (file)
@@ -1,8 +1,8 @@
 /*****************************************************************************
- * ps.c
+ * ps.c: Program Stream demux module for VLC.
  *****************************************************************************
  * Copyright (C) 2004 VideoLAN
- * $Id: ps.c,v 1.2 2004/01/25 20:05:28 hartman Exp $
+ * $Id$
  *
  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
  *
 /*****************************************************************************
  * Module descriptor
  *****************************************************************************/
-static int  Open    ( vlc_object_t * );
+static int  Open   ( vlc_object_t * );
+static int  OpenAlt( vlc_object_t * );
 static void Close  ( vlc_object_t * );
 
 vlc_module_begin();
     set_description( _("PS demuxer") );
-    set_capability( "demux2", 0 );
+    set_category( CAT_INPUT );
+    set_subcategory( SUBCAT_INPUT_DEMUX );
+    set_capability( "demux2", 1 );
     set_callbacks( Open, Close );
-    add_shortcut( "ps2" );
+    add_shortcut( "ps" );
+
+    add_submodule();
+    set_description( _("PS demuxer") );
+    set_capability( "demux2", 9 );
+    set_callbacks( OpenAlt, Close );
 vlc_module_end();
 
 /*****************************************************************************
@@ -55,7 +63,8 @@ vlc_module_end();
 
 struct demux_sys_t
 {
-    ps_track_t   tk[PS_TK_COUNT];
+    ps_psm_t    psm;
+    ps_track_t  tk[PS_TK_COUNT];
 
     int64_t     i_scr;
     int         i_mux_rate;
@@ -83,9 +92,11 @@ static int Open( vlc_object_t *p_this )
         return VLC_EGENERIC;
     }
 
-    if( p_peek[0] != 0 || p_peek[1] != 0 || p_peek[2] != 1 || p_peek[3] < 0xb9 )
+    if( p_peek[0] != 0 || p_peek[1] != 0 ||
+        p_peek[2] != 1 || p_peek[3] < 0xb9 )
     {
-        msg_Warn( p_demux, "this does not look like an MPEG PS stream, continuing anyway" );
+        msg_Warn( p_demux, "this does not look like an MPEG PS stream, "
+                  "continuing anyway" );
     }
 
     /* Fill p_demux field */
@@ -97,6 +108,7 @@ static int Open( vlc_object_t *p_this )
     p_sys->i_mux_rate = 0;
     p_sys->i_scr      = -1;
 
+    ps_psm_init( &p_sys->psm );
     ps_track_init( p_sys->tk );
 
     /* TODO prescanning of ES */
@@ -104,6 +116,26 @@ static int Open( vlc_object_t *p_this )
     return VLC_SUCCESS;
 }
 
+static int OpenAlt( vlc_object_t *p_this )
+{
+    demux_t *p_demux = (demux_t*)p_this;
+    uint8_t *p_peek;
+
+    if( stream_Peek( p_demux->s, &p_peek, 4 ) < 4 )
+    {
+        msg_Err( p_demux, "cannot peek" );
+        return VLC_EGENERIC;
+    }
+
+    if( p_peek[0] != 0 || p_peek[1] != 0 ||
+        p_peek[2] != 1 || p_peek[3] < 0xb9 )
+    {
+        if( !p_demux->b_force ) return VLC_EGENERIC;
+    }
+
+    return Open( p_this );
+}
+
 /*****************************************************************************
  * Close
  *****************************************************************************/
@@ -111,6 +143,19 @@ static void Close( vlc_object_t *p_this )
 {
     demux_t     *p_demux = (demux_t*)p_this;
     demux_sys_t *p_sys = p_demux->p_sys;
+    int i;
+
+    for( i = 0; i < PS_TK_COUNT; i++ )
+    {
+        ps_track_t *tk = &p_sys->tk[i];
+        if( tk->b_seen )
+        {
+            es_format_Clean( &tk->fmt );
+            if( tk->es ) es_out_Del( p_demux->out, tk->es );
+        }
+    }
+
+    ps_psm_destroy( &p_sys->psm );
 
     free( p_sys );
 }
@@ -121,9 +166,9 @@ static void Close( vlc_object_t *p_this )
 static int Demux( demux_t *p_demux )
 {
     demux_sys_t *p_sys = p_demux->p_sys;
-    int i_ret;
+    int i_ret, i_id, i_mux_rate;
     uint32_t i_code;
-    block_t  *p_pkt;
+    block_t *p_pkt;
 
     i_ret = ps_pkt_resynch( p_demux->s, &i_code );
     if( i_ret < 0 )
@@ -143,76 +188,88 @@ static int Demux( demux_t *p_demux )
 
     switch( i_code )
     {
-        case 0x1b9:
-            block_Release( p_pkt );
-            break;
-        case 0x1ba:
+    case 0x1b9:
+        block_Release( p_pkt );
+        break;
+
+    case 0x1ba:
+        if( !ps_pkt_parse_pack( p_pkt, &p_sys->i_scr, &i_mux_rate ) )
         {
-            int i_mux_rate;
+            /* done later on to work around bad vcd/svcd streams */
+            /* es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_scr ); */
+            if( i_mux_rate > 0 ) p_sys->i_mux_rate = i_mux_rate;
+        }
+        block_Release( p_pkt );
+        break;
 
-            if( !ps_pkt_parse_pack( p_pkt, &p_sys->i_scr, &i_mux_rate ) )
+    case 0x1bb:
+        if( !ps_pkt_parse_system( p_pkt, &p_sys->psm, p_sys->tk ) )
+        {
+            int i;
+            for( i = 0; i < PS_TK_COUNT; i++ )
             {
-                es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_scr );
-                if( i_mux_rate > 0 )
+                ps_track_t *tk = &p_sys->tk[i];
+
+                if( tk->b_seen && !tk->es && tk->fmt.i_cat != UNKNOWN_ES )
                 {
-                    p_sys->i_mux_rate = i_mux_rate;
+                    tk->es = es_out_Add( p_demux->out, &tk->fmt );
                 }
             }
-            block_Release( p_pkt );
-            break;
         }
-        case 0x1bb:
-            if( !ps_pkt_parse_system( p_pkt, p_sys->tk ) )
+        block_Release( p_pkt );
+        break;
+
+    case 0x1bc:
+        /* msg_Dbg( p_demux, "received PSM"); */
+        ps_psm_fill( &p_sys->psm, p_pkt, p_sys->tk, p_demux->out );
+        block_Release( p_pkt );
+        break;
+
+    default:
+        if( (i_id = ps_pkt_id( p_pkt )) >= 0xc0 )
+        {
+            ps_track_t *tk = &p_sys->tk[PS_ID_TO_TK(i_id)];
+
+            if( !tk->b_seen )
             {
-                int i;
-                for( i = 0; i < PS_TK_COUNT; i++ )
+                if( !ps_track_fill( tk, &p_sys->psm, i_id ) )
                 {
-                    ps_track_t *tk = &p_sys->tk[i];
-
-                    if( tk->b_seen && !tk->es && tk->fmt.i_cat != UNKNOWN_ES )
-                    {
-                        tk->es = es_out_Add( p_demux->out, &tk->fmt );
-                    }
+                    tk->es = es_out_Add( p_demux->out, &tk->fmt );
                 }
+                tk->b_seen = VLC_TRUE;
             }
-            block_Release( p_pkt );
-            break;
-        case 0x1bc:
-            /* TODO PSM */
-            block_Release( p_pkt );
-            break;
 
-        default:
-        {
-            int        i_id = ps_pkt_id( p_pkt );
-            if( i_id >= 0xc0 )
+            /* The popular VCD/SVCD subtitling WinSubMux does not
+             * renumber the SCRs when merging subtitles into the PES */
+            if( tk->b_seen &&
+                ( tk->fmt.i_codec == VLC_FOURCC('o','g','t',' ') ||
+                  tk->fmt.i_codec == VLC_FOURCC('c','v','d',' ') ) )
             {
-                ps_track_t *tk = &p_sys->tk[PS_ID_TO_TK(i_id)];
+                p_sys->i_scr = -1;
+            }
 
-                if( !tk->b_seen )
-                {
-                    if( !ps_track_fill( tk, i_id ) )
-                    {
-                        tk->es = es_out_Add( p_demux->out, &tk->fmt );
-                    }
-                    tk->b_seen = VLC_TRUE;
-                }
-                if( tk->b_seen && tk->es && !ps_pkt_parse_pes( p_pkt, tk->i_skip ) )
-                {
-                    es_out_Send( p_demux->out, tk->es, p_pkt );
-                }
-                else
-                {
-                    block_Release( p_pkt );
-                }
+            if( p_sys->i_scr > 0 )
+                es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_scr );
+
+            p_sys->i_scr = -1;
+
+            if( tk->b_seen && tk->es &&
+                !ps_pkt_parse_pes( p_pkt, tk->i_skip ) )
+            {
+                es_out_Send( p_demux->out, tk->es, p_pkt );
             }
             else
             {
                 block_Release( p_pkt );
             }
-            break;
         }
+        else
+        {
+            block_Release( p_pkt );
+        }
+        break;
     }
+
     return 1;
 }
 
@@ -239,6 +296,7 @@ static int Control( demux_t *p_demux, int i_query, va_list args )
                 *pf = 0.0;
             }
             return VLC_SUCCESS;
+
         case DEMUX_SET_POSITION:
             f = (double) va_arg( args, double );
             i64 = stream_Size( p_demux->s );
@@ -251,7 +309,8 @@ static int Control( demux_t *p_demux, int i_query, va_list args )
             pi64 = (int64_t*)va_arg( args, int64_t * );
             if( p_sys->i_mux_rate > 0 )
             {
-                *pi64 = (int64_t)1000000 * ( stream_Tell( p_demux->s ) / 50 ) / p_sys->i_mux_rate;
+                *pi64 = (int64_t)1000000 * ( stream_Tell( p_demux->s ) / 50 ) /
+                    p_sys->i_mux_rate;
                 return VLC_SUCCESS;
             }
             *pi64 = 0;
@@ -261,7 +320,8 @@ static int Control( demux_t *p_demux, int i_query, va_list args )
             pi64 = (int64_t*)va_arg( args, int64_t * );
             if( p_sys->i_mux_rate > 0 )
             {
-                *pi64 = (int64_t)1000000 * ( stream_Size( p_demux->s ) / 50 ) / p_sys->i_mux_rate;
+                *pi64 = (int64_t)1000000 * ( stream_Size( p_demux->s ) / 50 ) /
+                    p_sys->i_mux_rate;
                 return VLC_SUCCESS;
             }
             *pi64 = 0;
@@ -278,7 +338,7 @@ static int Control( demux_t *p_demux, int i_query, va_list args )
  * Divers:
  *****************************************************************************/
 
-/* PSResynch: resynch on a systeme starcode
+/* PSResynch: resynch on a system starcode
  *  It doesn't skip more than 512 bytes
  *  -1 -> error, 0 -> not synch, 1 -> ok
  */
@@ -292,7 +352,8 @@ static int ps_pkt_resynch( stream_t *s, uint32_t *pi_code )
     {
         return -1;
     }
-    if( p_peek[0] == 0 && p_peek[1] == 0 && p_peek[2] == 1 && p_peek[3] >= 0xb9 )
+    if( p_peek[0] == 0 && p_peek[1] == 0 && p_peek[2] == 1 &&
+        p_peek[3] >= 0xb9 )
     {
         *pi_code = 0x100 | p_peek[3];
         return 1;
@@ -310,7 +371,8 @@ static int ps_pkt_resynch( stream_t *s, uint32_t *pi_code )
         {
             break;
         }
-        if( p_peek[0] == 0 && p_peek[1] == 0 && p_peek[2] == 1 && p_peek[3] >= 0xb9 )
+        if( p_peek[0] == 0 && p_peek[1] == 0 && p_peek[2] == 1 &&
+            p_peek[3] >= 0xb9 )
         {
             *pi_code = 0x100 | p_peek[3];
             return stream_Read( s, NULL, i_skip ) == i_skip ? 1 : -1;
@@ -329,9 +391,33 @@ static block_t *ps_pkt_read( stream_t *s, uint32_t i_code )
     int      i_peek = stream_Peek( s, &p_peek, 14 );
     int      i_size = ps_pkt_size( p_peek, i_peek );
 
-    if( i_size > 0 )
+    if( i_size <= 6 && p_peek[3] > 0xba )
     {
+        /* Special case, search the next start code */
+        i_size = 6;
+        for( ;; )
+        {
+            i_peek = stream_Peek( s, &p_peek, i_size + 1024 );
+            if( i_peek <= i_size + 4 )
+            {
+                return NULL;
+            }
+            while( i_size <= i_peek - 4 )
+            {
+                if( p_peek[i_size] == 0x00 && p_peek[i_size+1] == 0x00 &&
+                    p_peek[i_size+2] == 0x01 && p_peek[i_size+3] >= 0xb9 )
+                {
+                    return stream_Block( s, i_size );
+                }
+                i_size++;
+            }
+        }
+    }
+    else
+    {
+        /* Normal case */
         return  stream_Block( s, i_size );
     }
+
     return NULL;
 }