]> git.sesse.net Git - vlc/blobdiff - modules/packetizer/copy.c
* src/video_output/video_output.c: new "deinterlace" object variable.
[vlc] / modules / packetizer / copy.c
index 18e95aab9936fcfabcb6918aaa144d866b269049..f238a1dbbe4f9f8199f46b98793d42ff50f3bfc4 100644 (file)
@@ -2,7 +2,7 @@
  * copy.c
  *****************************************************************************
  * Copyright (C) 2001, 2002 VideoLAN
- * $Id: copy.c,v 1.2 2003/01/08 10:26:49 fenrir Exp $
+ * $Id: copy.c,v 1.10 2003/05/16 23:02:37 gbazin Exp $
  *
  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
  *          Eric Petit <titer@videolan.org>
@@ -34,6 +34,7 @@
 #include <stdlib.h>                                      /* malloc(), free() */
 #include <string.h>                                              /* strdup() */
 
+#include "codecs.h"
 /*****************************************************************************
  * Local prototypes
  *****************************************************************************/
@@ -43,10 +44,10 @@ typedef struct packetizer_thread_s
     decoder_fifo_t          *p_fifo;
 
     /* Output properties */
-    sout_input_t            *p_sout_input;
-    sout_packet_format_t    output_format;
+    sout_packetizer_input_t *p_sout_input;
+    sout_format_t           output_format;
 
-    mtime_t i_pts_start;
+//    mtime_t i_last_pts;
 
 } packetizer_thread_t;
 
@@ -66,7 +67,7 @@ static void input_ShowPES( decoder_fifo_t *p_fifo, pes_packet_t **pp_pes );
 
 vlc_module_begin();
     set_description( _("Copy packetizer") );
-    set_capability( "packetizer", 0 );
+    set_capability( "packetizer", 1 );
     set_callbacks( Open, NULL );
 vlc_module_end();
 
@@ -84,11 +85,6 @@ static int Open( vlc_object_t *p_this )
     p_fifo->pf_run = Run;
 
     return VLC_SUCCESS;
-
-#if 0
-    if( p_fifo->i_fourcc == VLC_FOURCC( 'm', 'p', 'g', 'a') )
-        ....
-#endif
 }
 
 /*****************************************************************************
@@ -148,6 +144,11 @@ static int InitThread( packetizer_thread_t *p_pack )
 
     switch( p_pack->p_fifo->i_fourcc )
     {
+        /* video */
+        case VLC_FOURCC( 'm', '4', 's', '2'):
+        case VLC_FOURCC( 'M', '4', 'S', '2'):
+        case VLC_FOURCC( 'm', 'p', '4', 's'):
+        case VLC_FOURCC( 'M', 'P', '4', 'S'):
         case VLC_FOURCC( 'm', 'p', '4', 'v'):
         case VLC_FOURCC( 'D', 'I', 'V', 'X'):
         case VLC_FOURCC( 'd', 'i', 'v', 'x'):
@@ -155,23 +156,294 @@ static int InitThread( packetizer_thread_t *p_pack )
         case VLC_FOURCC( 'X', 'v', 'i', 'D'):
         case VLC_FOURCC( 'x', 'v', 'i', 'd'):
         case VLC_FOURCC( 'D', 'X', '5', '0'):
+        case VLC_FOURCC( 0x04, 0,   0,   0):
+        case VLC_FOURCC( '3', 'I', 'V', '2'):
             p_pack->output_format.i_fourcc = VLC_FOURCC( 'm', 'p', '4', 'v');
             p_pack->output_format.i_cat = VIDEO_ES;
             break;
         case VLC_FOURCC( 'm', 'p', 'g', 'v' ):
         case VLC_FOURCC( 'm', 'p', 'g', '1' ):
         case VLC_FOURCC( 'm', 'p', 'g', '2' ):
+        case VLC_FOURCC( 'm', 'p', '1', 'v' ):
+        case VLC_FOURCC( 'm', 'p', '2', 'v' ):
             p_pack->output_format.i_fourcc = VLC_FOURCC( 'm', 'p', 'g', 'v' );
             p_pack->output_format.i_cat = VIDEO_ES;
             break;
+
+        case VLC_FOURCC( 'd', 'i', 'v', '1' ):
+        case VLC_FOURCC( 'D', 'I', 'V', '1' ):
+        case VLC_FOURCC( 'M', 'P', 'G', '4' ):
+        case VLC_FOURCC( 'm', 'p', 'g', '4' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'D', 'I', 'V', '1' );
+            p_pack->output_format.i_cat = VIDEO_ES;
+            break;
+        case VLC_FOURCC( 'd', 'i', 'v', '2' ):
+        case VLC_FOURCC( 'D', 'I', 'V', '2' ):
+        case VLC_FOURCC( 'M', 'P', '4', '2' ):
+        case VLC_FOURCC( 'm', 'p', '4', '2' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'D', 'I', 'V', '2' );
+            p_pack->output_format.i_cat = VIDEO_ES;
+            break;
+        case VLC_FOURCC( 'd', 'i', 'v', '3' ):
+        case VLC_FOURCC( 'D', 'I', 'V', '3' ):
+        case VLC_FOURCC( 'd', 'i', 'v', '4' ):
+        case VLC_FOURCC( 'D', 'I', 'V', '4' ):
+        case VLC_FOURCC( 'd', 'i', 'v', '5' ):
+        case VLC_FOURCC( 'D', 'I', 'V', '5' ):
+        case VLC_FOURCC( 'd', 'i', 'v', '6' ):
+        case VLC_FOURCC( 'D', 'I', 'V', '6' ):
+        case VLC_FOURCC( 'M', 'P', '4', '3' ):
+        case VLC_FOURCC( 'm', 'p', '4', '3' ):
+        case VLC_FOURCC( 'm', 'p', 'g', '3' ):
+        case VLC_FOURCC( 'M', 'P', 'G', '3' ):
+        case VLC_FOURCC( 'A', 'P', '4', '1' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'D', 'I', 'V', '3' );
+            p_pack->output_format.i_cat = VIDEO_ES;
+            break;
+        case VLC_FOURCC( 'H', '2', '6', '3' ):
+        case VLC_FOURCC( 'h', '2', '6', '3' ):
+        case VLC_FOURCC( 'U', '2', '6', '3' ):
+        case VLC_FOURCC( 'u', '2', '6', '3' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'H', '2', '6', '3' );
+            p_pack->output_format.i_cat = VIDEO_ES;
+            break;
+        case VLC_FOURCC( 'I', '2', '6', '3' ):
+        case VLC_FOURCC( 'i', '2', '6', '3' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'I', '2', '6', '3' );
+            p_pack->output_format.i_cat = VIDEO_ES;
+            break;
+        case VLC_FOURCC( 'W', 'M', 'V', '1' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'W', 'M', 'V', '1' );
+            p_pack->output_format.i_cat = VIDEO_ES;
+            break;
+        case VLC_FOURCC( 'W', 'M', 'V', '2' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'W', 'M', 'V', '2' );
+            p_pack->output_format.i_cat = VIDEO_ES;
+            break;
+        case VLC_FOURCC( 'M', 'J', 'P', 'G' ):
+        case VLC_FOURCC( 'm', 'j', 'p', 'g' ):
+        case VLC_FOURCC( 'm', 'j', 'p', 'a' ):
+        case VLC_FOURCC( 'j', 'p', 'e', 'g' ):
+        case VLC_FOURCC( 'J', 'P', 'E', 'G' ):
+        case VLC_FOURCC( 'J', 'F', 'I', 'F' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'M', 'J', 'P', 'G' );
+            p_pack->output_format.i_cat = VIDEO_ES;
+            break;
+        case VLC_FOURCC( 'm', 'j', 'p', 'b' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'm', 'j', 'p', 'b' );
+            p_pack->output_format.i_cat = VIDEO_ES;
+            break;
+        case VLC_FOURCC( 'd', 'v', 's', 'l' ):
+        case VLC_FOURCC( 'd', 'v', 's', 'd' ):
+        case VLC_FOURCC( 'D', 'V', 'S', 'D' ):
+        case VLC_FOURCC( 'd', 'v', 'h', 'd' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'd', 'v', 's', 'l' );
+            p_pack->output_format.i_cat = VIDEO_ES;
+            break;
+        case VLC_FOURCC( 'S', 'V', 'Q', '1' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'S', 'V', 'Q', '1' );
+            p_pack->output_format.i_cat = VIDEO_ES;
+            break;
+        case VLC_FOURCC( 'S', 'V', 'Q', '3' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'S', 'V', 'Q', '3' );
+            p_pack->output_format.i_cat = VIDEO_ES;
+            break;
+
+        case VLC_FOURCC( 'I', '4', '2', '0' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'I', '4', '2', '0' );
+            p_pack->output_format.i_cat = VIDEO_ES;
+            break;
+        case VLC_FOURCC( 'I', '4', '2', '2' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'I', '4', '2', '2' );
+            p_pack->output_format.i_cat = VIDEO_ES;
+            break;
+
+        /* audio */
         case VLC_FOURCC( 'm', 'p', 'g', 'a' ):
             p_pack->output_format.i_fourcc = VLC_FOURCC( 'm', 'p', 'g', 'a' );
             p_pack->output_format.i_cat = AUDIO_ES;
             break;
+        case VLC_FOURCC( 'w', 'm', 'a', '1' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'w', 'm', 'a', '1' );
+            p_pack->output_format.i_cat = AUDIO_ES;
+            break;
+        case VLC_FOURCC( 'w', 'm', 'a', '2' ):
+            p_pack->output_format.i_fourcc = VLC_FOURCC( 'w', 'm', 'a', '2' );
+            p_pack->output_format.i_cat = AUDIO_ES;
+            break;
+        case VLC_FOURCC( 'a', 'r', 'a', 'w' ):
+        {
+            WAVEFORMATEX *p_wf = (WAVEFORMATEX*)p_pack->p_fifo->p_waveformatex;
+            if( p_wf )
+            {
+                switch( ( p_wf->wBitsPerSample + 7 ) / 8 )
+                {
+                    case 1:
+                        p_pack->output_format.i_fourcc = VLC_FOURCC('u','8',' ',' ');
+                        break;
+                    case 2:
+                        p_pack->output_format.i_fourcc = VLC_FOURCC('s','1','6','l');
+                        break;
+                    case 3:
+                        p_pack->output_format.i_fourcc = VLC_FOURCC('s','2','4','l');
+                        break;
+                    case 4:
+                        p_pack->output_format.i_fourcc = VLC_FOURCC('s','3','2','l');
+                        break;
+                    default:
+                        msg_Err( p_pack->p_fifo, "unknown raw audio sample size !!" );
+                        return VLC_EGENERIC;
+                }
+            }
+            else
+            {
+                msg_Err( p_pack->p_fifo, "unknown raw audio sample size !!" );
+                return VLC_EGENERIC;
+            }
+            p_pack->output_format.i_cat = AUDIO_ES;
+            break;
+        }
+        case VLC_FOURCC( 't', 'w', 'o', 's' ):
+        {
+            WAVEFORMATEX *p_wf = (WAVEFORMATEX*)p_pack->p_fifo->p_waveformatex;
+            if( p_wf )
+            {
+                switch( ( p_wf->wBitsPerSample + 7 ) / 8 )
+                {
+                    case 1:
+                        p_pack->output_format.i_fourcc = VLC_FOURCC('s','8',' ',' ');
+                        break;
+                    case 2:
+                        p_pack->output_format.i_fourcc = VLC_FOURCC('s','1','6','b');
+                        break;
+                    case 3:
+                        p_pack->output_format.i_fourcc = VLC_FOURCC('s','2','4','b');
+                        break;
+                    case 4:
+                        p_pack->output_format.i_fourcc = VLC_FOURCC('s','3','2','b');
+                        break;
+                    default:
+                        msg_Err( p_pack->p_fifo, "unknown raw audio sample size !!" );
+                        return VLC_EGENERIC;
+                }
+            }
+            else
+            {
+                msg_Err( p_pack->p_fifo, "unknown raw audio sample size !!" );
+                return VLC_EGENERIC;
+            }
+            p_pack->output_format.i_cat = AUDIO_ES;
+            break;
+        }
+        case VLC_FOURCC( 's', 'o', 'w', 't' ):
+        {
+            WAVEFORMATEX *p_wf = (WAVEFORMATEX*)p_pack->p_fifo->p_waveformatex;
+            if( p_wf )
+            {
+                switch( ( p_wf->wBitsPerSample + 7 ) / 8 )
+                {
+                    case 1:
+                        p_pack->output_format.i_fourcc = VLC_FOURCC('s','8',' ',' ');
+                        break;
+                    case 2:
+                        p_pack->output_format.i_fourcc = VLC_FOURCC('s','1','6','l');
+                        break;
+                    case 3:
+                        p_pack->output_format.i_fourcc = VLC_FOURCC('s','2','4','l');
+                        break;
+                    case 4:
+                        p_pack->output_format.i_fourcc = VLC_FOURCC('s','3','2','l');
+                        break;
+                    default:
+                        msg_Err( p_pack->p_fifo, "unknown raw audio sample size !!" );
+                        return VLC_EGENERIC;
+                }
+            }
+            else
+            {
+                msg_Err( p_pack->p_fifo, "unknown raw audio sample size !!" );
+                return VLC_EGENERIC;
+            }
+            p_pack->output_format.i_cat = AUDIO_ES;
+            break;
+        }
+
         default:
-            p_pack->output_format.i_fourcc = p_pack->p_fifo->i_fourcc;
-            p_pack->output_format.i_cat = UNKNOWN_ES;
+            msg_Err( p_pack->p_fifo, "unknown es type !!" );
+            return VLC_EGENERIC;
+            //p_pack->output_format.i_fourcc = p_pack->p_fifo->i_fourcc;
+            //p_pack->output_format.i_cat = UNKNOWN_ES;
+            //break;
+    }
+
+    switch( p_pack->output_format.i_cat )
+    {
+        case AUDIO_ES:
+            {
+                WAVEFORMATEX *p_wf = (WAVEFORMATEX*)p_pack->p_fifo->p_waveformatex;
+                if( p_wf )
+                {
+                    p_pack->output_format.i_sample_rate = p_wf->nSamplesPerSec;
+                    p_pack->output_format.i_channels    = p_wf->nChannels;
+                    p_pack->output_format.i_block_align = p_wf->nBlockAlign;
+                    p_pack->output_format.i_bitrate     = p_wf->nAvgBytesPerSec * 8;
+                    p_pack->output_format.i_extra_data  = p_wf->cbSize;
+                    if( p_wf->cbSize  > 0 )
+                    {
+                        p_pack->output_format.p_extra_data =
+                            malloc( p_pack->output_format.i_extra_data );
+                        memcpy( p_pack->output_format.p_extra_data,
+                                &p_wf[1],
+                                p_pack->output_format.i_extra_data );
+                    }
+                    else
+                    {
+                        p_pack->output_format.p_extra_data = NULL;
+                    }
+                }
+                else
+                {
+                    p_pack->output_format.i_sample_rate = 0;
+                    p_pack->output_format.i_channels    = 0;
+                    p_pack->output_format.i_block_align = 0;
+                    p_pack->output_format.i_bitrate     = 0;
+                    p_pack->output_format.i_extra_data  = 0;
+                    p_pack->output_format.p_extra_data  = NULL;
+                }
+            }
             break;
+
+        case VIDEO_ES:
+            {
+                BITMAPINFOHEADER *p_bih = (BITMAPINFOHEADER*)p_pack->p_fifo->p_bitmapinfoheader;
+
+                p_pack->output_format.i_bitrate = 0;
+                if( p_bih )
+                {
+                    p_pack->output_format.i_width  = p_bih->biWidth;
+                    p_pack->output_format.i_height = p_bih->biHeight;
+                    p_pack->output_format.i_extra_data  = p_bih->biSize - sizeof( BITMAPINFOHEADER );
+                    if( p_pack->output_format.i_extra_data > 0 )
+                    {
+                        p_pack->output_format.p_extra_data =
+                            malloc( p_pack->output_format.i_extra_data );
+                        memcpy( p_pack->output_format.p_extra_data,
+                                &p_bih[1],
+                                p_pack->output_format.i_extra_data );
+                    }
+                }
+                else
+                {
+                    p_pack->output_format.i_width  = 0;
+                    p_pack->output_format.i_height = 0;
+                    p_pack->output_format.i_extra_data  = 0;
+                    p_pack->output_format.p_extra_data  = NULL;
+                }
+            }
+            break;
+
+        default:
+            return VLC_EGENERIC;
     }
 
     p_pack->p_sout_input =
@@ -181,10 +453,10 @@ static int InitThread( packetizer_thread_t *p_pack )
     if( !p_pack->p_sout_input )
     {
         msg_Err( p_pack->p_fifo, "cannot add a new stream" );
-        return( -1 );
+        return VLC_EGENERIC;
     }
-    p_pack->i_pts_start = -1;
-    return( 0 );
+//    p_pack->i_last_pts = 0;
+    return( VLC_SUCCESS );
 }
 
 /*****************************************************************************
@@ -194,7 +466,8 @@ static void PacketizeThread( packetizer_thread_t *p_pack )
 {
     sout_buffer_t   *p_sout_buffer;
     pes_packet_t    *p_pes;
-    size_t          i_size;
+    ssize_t         i_size;
+    mtime_t         i_pts;
 
     /* **** get samples count **** */
     input_ExtractPES( p_pack->p_fifo, &p_pes );
@@ -203,23 +476,30 @@ static void PacketizeThread( packetizer_thread_t *p_pack )
         p_pack->p_fifo->b_error = 1;
         return;
     }
-    if( p_pack->i_pts_start < 0 && p_pes->i_pts > 0 )
+    i_pts = p_pes->i_pts;
+
+    if( i_pts <= 0 ) //&& p_pack->i_last_pts <= 0 )
     {
-        p_pack->i_pts_start = p_pes->i_pts;
+        msg_Dbg( p_pack->p_fifo, "need pts != 0" );
+        input_DeletePES( p_pack->p_fifo->p_packets_mgt, p_pes );
+        return;
     }
+
     i_size = p_pes->i_pes_size;
+
 //    msg_Dbg( p_pack->p_fifo, "pes size:%d", i_size );
     if( i_size > 0 )
     {
         pes_packet_t    *p_pes_next;
         data_packet_t   *p_data;
-        size_t          i_buffer;
+        ssize_t          i_buffer;
 
-        p_sout_buffer = 
+        p_sout_buffer =
             sout_BufferNew( p_pack->p_sout_input->p_sout, i_size );
         if( !p_sout_buffer )
         {
             p_pack->p_fifo->b_error = 1;
+            input_DeletePES( p_pack->p_fifo->p_packets_mgt, p_pes );
             return;
         }
         /* TODO: memcpy of the pes packet */
@@ -227,9 +507,9 @@ static void PacketizeThread( packetizer_thread_t *p_pack )
              p_data != NULL && i_buffer < i_size;
              p_data = p_data->p_next)
         {
-            size_t          i_copy;
+            ssize_t i_copy;
 
-            i_copy = __MIN( p_data->p_payload_end - p_data->p_payload_start, 
+            i_copy = __MIN( p_data->p_payload_end - p_data->p_payload_start,
                             i_size - i_buffer );
             if( i_copy > 0 )
             {
@@ -240,14 +520,14 @@ static void PacketizeThread( packetizer_thread_t *p_pack )
             i_buffer += i_copy;
         }
         p_sout_buffer->i_length = 0;
-        p_sout_buffer->i_dts = p_pes->i_pts - p_pack->i_pts_start;
-        p_sout_buffer->i_pts = p_pes->i_pts - p_pack->i_pts_start;
+        p_sout_buffer->i_dts = i_pts; //p_pes->i_pts - p_pack->i_pts_start;
+        p_sout_buffer->i_pts = i_pts; //p_pes->i_pts - p_pack->i_pts_start;
         p_sout_buffer->i_bitrate = 0;
 
         input_ShowPES( p_pack->p_fifo, &p_pes_next );
         if( p_pes_next )
         {
-            p_sout_buffer->i_length = p_pes_next->i_pts - p_pes->i_pts;
+            p_sout_buffer->i_length = p_pes_next->i_pts - i_pts;
         }
         sout_InputSendBuffer( p_pack->p_sout_input,
                                p_sout_buffer );
@@ -266,6 +546,7 @@ static void EndThread ( packetizer_thread_t *p_pack)
     {
         sout_InputDelete( p_pack->p_sout_input );
     }
+    free( p_pack );
 }
 
 static void input_ShowPES( decoder_fifo_t *p_fifo, pes_packet_t **pp_pes )