git.sesse.net Git - vlc/blob - modules/packetizer/copy.c

   1 /*****************************************************************************
   2  * copy.c
   3  *****************************************************************************
   4  * Copyright (C) 2001, 2002 VideoLAN
   5  * $Id: copy.c,v 1.7 2003/04/13 20:00:21 fenrir Exp $
   6  *
   7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   8  *          Eric Petit <titer@videolan.org>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  23  *****************************************************************************/
  24
  25 /*****************************************************************************
  26  * Preamble
  27  *****************************************************************************/
  28 #include <vlc/vlc.h>
  29 #include <vlc/aout.h>
  30 #include <vlc/decoder.h>
  31 #include <vlc/input.h>
  32 #include <vlc/sout.h>
  33
  34 #include <stdlib.h>                                      /* malloc(), free() */
  35 #include <string.h>                                              /* strdup() */
  36
  37 #include "codecs.h"
  38 /*****************************************************************************
  39  * Local prototypes
  40  *****************************************************************************/
  41 typedef struct packetizer_thread_s
  42 {
  43     /* Input properties */
  44     decoder_fifo_t          *p_fifo;
  45
  46     /* Output properties */
  47     sout_packetizer_input_t *p_sout_input;
  48     sout_format_t           output_format;
  49
  50 //    mtime_t i_last_pts;
  51
  52 } packetizer_thread_t;
  53
  54 static int  Open    ( vlc_object_t * );
  55 static int  Run     ( decoder_fifo_t * );
  56
  57 static int  InitThread     ( packetizer_thread_t * );
  58 static void PacketizeThread   ( packetizer_thread_t * );
  59 static void EndThread      ( packetizer_thread_t * );
  60
  61
  62 static void input_ShowPES( decoder_fifo_t *p_fifo, pes_packet_t **pp_pes );
  63
  64 /*****************************************************************************
  65  * Module descriptor
  66  *****************************************************************************/
  67
  68 vlc_module_begin();
  69     set_description( _("Copy packetizer") );
  70     set_capability( "packetizer", 1 );
  71     set_callbacks( Open, NULL );
  72 vlc_module_end();
  73
  74
  75 /*****************************************************************************
  76  * OpenDecoder: probe the packetizer and return score
  77  *****************************************************************************
  78  * Tries to launch a decoder and return score so that the interface is able
  79  * to choose.
  80  *****************************************************************************/
  81 static int Open( vlc_object_t *p_this )
  82 {
  83     decoder_fifo_t *p_fifo = (decoder_fifo_t*) p_this;
  84
  85     p_fifo->pf_run = Run;
  86
  87     return VLC_SUCCESS;
  88 }
  89
  90 /*****************************************************************************
  91  * RunDecoder: this function is called just after the thread is created
  92  *****************************************************************************/
  93 static int Run( decoder_fifo_t *p_fifo )
  94 {
  95     packetizer_thread_t *p_pack;
  96     int b_error;
  97
  98     msg_Info( p_fifo, "Running copy packetizer" );
  99     if( !( p_pack = malloc( sizeof( packetizer_thread_t ) ) ) )
 100     {
 101         msg_Err( p_fifo, "out of memory" );
 102         DecoderError( p_fifo );
 103         return( -1 );
 104     }
 105     memset( p_pack, 0, sizeof( packetizer_thread_t ) );
 106
 107     p_pack->p_fifo = p_fifo;
 108
 109     if( InitThread( p_pack ) != 0 )
 110     {
 111         DecoderError( p_fifo );
 112         return( -1 );
 113     }
 114
 115     while( ( !p_pack->p_fifo->b_die )&&( !p_pack->p_fifo->b_error ) )
 116     {
 117         PacketizeThread( p_pack );
 118     }
 119
 120
 121     if( ( b_error = p_pack->p_fifo->b_error ) )
 122     {
 123         DecoderError( p_pack->p_fifo );
 124     }
 125
 126     EndThread( p_pack );
 127     if( b_error )
 128     {
 129         return( -1 );
 130     }
 131
 132     return( 0 );
 133 }
 134
 135
 136 #define FREE( p ) if( p ) free( p ); p = NULL
 137
 138 /*****************************************************************************
 139  * InitThread: initialize data before entering main loop
 140  *****************************************************************************/
 141
 142 static int InitThread( packetizer_thread_t *p_pack )
 143 {
 144
 145     switch( p_pack->p_fifo->i_fourcc )
 146     {
 147         case VLC_FOURCC( 'm', '4', 's', '2'):
 148         case VLC_FOURCC( 'M', '4', 'S', '2'):
 149         case VLC_FOURCC( 'm', 'p', '4', 's'):
 150         case VLC_FOURCC( 'M', 'P', '4', 'S'):
 151         case VLC_FOURCC( 'm', 'p', '4', 'v'):
 152         case VLC_FOURCC( 'D', 'I', 'V', 'X'):
 153         case VLC_FOURCC( 'd', 'i', 'v', 'x'):
 154         case VLC_FOURCC( 'X', 'V', 'I', 'D'):
 155         case VLC_FOURCC( 'X', 'v', 'i', 'D'):
 156         case VLC_FOURCC( 'x', 'v', 'i', 'd'):
 157         case VLC_FOURCC( 'D', 'X', '5', '0'):
 158         case VLC_FOURCC( 0x04, 0,   0,   0):
 159         case VLC_FOURCC( '3', 'I', 'V', '2'):
 160             p_pack->output_format.i_fourcc = VLC_FOURCC( 'm', 'p', '4', 'v');
 161             p_pack->output_format.i_cat = VIDEO_ES;
 162             break;
 163         case VLC_FOURCC( 'm', 'p', 'g', 'v' ):
 164         case VLC_FOURCC( 'm', 'p', 'g', '1' ):
 165         case VLC_FOURCC( 'm', 'p', 'g', '2' ):
 166         case VLC_FOURCC( 'm', 'p', '1', 'v' ):
 167         case VLC_FOURCC( 'm', 'p', '2', 'v' ):
 168             p_pack->output_format.i_fourcc = VLC_FOURCC( 'm', 'p', 'g', 'v' );
 169             p_pack->output_format.i_cat = VIDEO_ES;
 170             break;
 171         case VLC_FOURCC( 'm', 'p', 'g', 'a' ):
 172             p_pack->output_format.i_fourcc = VLC_FOURCC( 'm', 'p', 'g', 'a' );
 173             p_pack->output_format.i_cat = AUDIO_ES;
 174             break;
 175
 176         case VLC_FOURCC( 'd', 'i', 'v', '1' ):
 177         case VLC_FOURCC( 'D', 'I', 'V', '1' ):
 178         case VLC_FOURCC( 'M', 'P', 'G', '4' ):
 179         case VLC_FOURCC( 'm', 'p', 'g', '4' ):
 180             p_pack->output_format.i_fourcc = VLC_FOURCC( 'D', 'I', 'V', '1' );
 181             p_pack->output_format.i_cat = VIDEO_ES;
 182             break;
 183         case VLC_FOURCC( 'd', 'i', 'v', '2' ):
 184         case VLC_FOURCC( 'D', 'I', 'V', '2' ):
 185         case VLC_FOURCC( 'M', 'P', '4', '2' ):
 186         case VLC_FOURCC( 'm', 'p', '4', '2' ):
 187             p_pack->output_format.i_fourcc = VLC_FOURCC( 'D', 'I', 'V', '2' );
 188             p_pack->output_format.i_cat = VIDEO_ES;
 189             break;
 190         case VLC_FOURCC( 'd', 'i', 'v', '3' ):
 191         case VLC_FOURCC( 'D', 'I', 'V', '3' ):
 192         case VLC_FOURCC( 'd', 'i', 'v', '4' ):
 193         case VLC_FOURCC( 'D', 'I', 'V', '4' ):
 194         case VLC_FOURCC( 'd', 'i', 'v', '5' ):
 195         case VLC_FOURCC( 'D', 'I', 'V', '5' ):
 196         case VLC_FOURCC( 'd', 'i', 'v', '6' ):
 197         case VLC_FOURCC( 'D', 'I', 'V', '6' ):
 198         case VLC_FOURCC( 'M', 'P', '4', '3' ):
 199         case VLC_FOURCC( 'm', 'p', '4', '3' ):
 200         case VLC_FOURCC( 'm', 'p', 'g', '3' ):
 201         case VLC_FOURCC( 'M', 'P', 'G', '3' ):
 202         case VLC_FOURCC( 'A', 'P', '4', '1' ):
 203             p_pack->output_format.i_fourcc = VLC_FOURCC( 'D', 'I', 'V', '3' );
 204             p_pack->output_format.i_cat = VIDEO_ES;
 205             break;
 206         case VLC_FOURCC( 'H', '2', '6', '3' ):
 207         case VLC_FOURCC( 'h', '2', '6', '3' ):
 208         case VLC_FOURCC( 'U', '2', '6', '3' ):
 209         case VLC_FOURCC( 'u', '2', '6', '3' ):
 210             p_pack->output_format.i_fourcc = VLC_FOURCC( 'H', '2', '6', '3' );
 211             p_pack->output_format.i_cat = VIDEO_ES;
 212             break;
 213         case VLC_FOURCC( 'I', '2', '6', '3' ):
 214         case VLC_FOURCC( 'i', '2', '6', '3' ):
 215             p_pack->output_format.i_fourcc = VLC_FOURCC( 'I', '2', '6', '3' );
 216             p_pack->output_format.i_cat = VIDEO_ES;
 217             break;
 218         case VLC_FOURCC( 'W', 'M', 'V', '1' ):
 219             p_pack->output_format.i_fourcc = VLC_FOURCC( 'W', 'M', 'V', '1' );
 220             p_pack->output_format.i_cat = VIDEO_ES;
 221             break;
 222         case VLC_FOURCC( 'W', 'M', 'V', '2' ):
 223             p_pack->output_format.i_fourcc = VLC_FOURCC( 'W', 'M', 'V', '2' );
 224             p_pack->output_format.i_cat = VIDEO_ES;
 225             break;
 226         case VLC_FOURCC( 'M', 'J', 'P', 'G' ):
 227         case VLC_FOURCC( 'm', 'j', 'p', 'g' ):
 228         case VLC_FOURCC( 'm', 'j', 'p', 'a' ):
 229         case VLC_FOURCC( 'j', 'p', 'e', 'g' ):
 230         case VLC_FOURCC( 'J', 'P', 'E', 'G' ):
 231         case VLC_FOURCC( 'J', 'F', 'I', 'F' ):
 232             p_pack->output_format.i_fourcc = VLC_FOURCC( 'M', 'J', 'P', 'G' );
 233             p_pack->output_format.i_cat = VIDEO_ES;
 234             break;
 235         case VLC_FOURCC( 'm', 'j', 'p', 'b' ):
 236             p_pack->output_format.i_fourcc = VLC_FOURCC( 'm', 'j', 'p', 'b' );
 237             p_pack->output_format.i_cat = VIDEO_ES;
 238             break;
 239         case VLC_FOURCC( 'd', 'v', 's', 'l' ):
 240         case VLC_FOURCC( 'd', 'v', 's', 'd' ):
 241         case VLC_FOURCC( 'D', 'V', 'S', 'D' ):
 242         case VLC_FOURCC( 'd', 'v', 'h', 'd' ):
 243             p_pack->output_format.i_fourcc = VLC_FOURCC( 'd', 'v', 's', 'l' );
 244             p_pack->output_format.i_cat = VIDEO_ES;
 245             break;
 246         case VLC_FOURCC( 'S', 'V', 'Q', '1' ):
 247             p_pack->output_format.i_fourcc = VLC_FOURCC( 'S', 'V', 'Q', '1' );
 248             p_pack->output_format.i_cat = VIDEO_ES;
 249             break;
 250
 251         case VLC_FOURCC( 'w', 'm', 'a', '1' ):
 252             p_pack->output_format.i_fourcc = VLC_FOURCC( 'w', 'm', 'a', '1' );
 253             p_pack->output_format.i_cat = AUDIO_ES;
 254             break;
 255         case VLC_FOURCC( 'w', 'm', 'a', '2' ):
 256             p_pack->output_format.i_fourcc = VLC_FOURCC( 'w', 'm', 'a', '2' );
 257             p_pack->output_format.i_cat = AUDIO_ES;
 258             break;
 259
 260         default:
 261             msg_Err( p_pack->p_fifo, "unknown es type !!" );
 262             return VLC_EGENERIC;
 263             //p_pack->output_format.i_fourcc = p_pack->p_fifo->i_fourcc;
 264             //p_pack->output_format.i_cat = UNKNOWN_ES;
 265             //break;
 266     }
 267
 268     switch( p_pack->output_format.i_cat )
 269     {
 270         case AUDIO_ES:
 271             {
 272                 WAVEFORMATEX *p_wf = (WAVEFORMATEX*)p_pack->p_fifo->p_waveformatex;
 273                 if( p_wf )
 274                 {
 275                     p_pack->output_format.i_sample_rate = p_wf->nSamplesPerSec;
 276                     p_pack->output_format.i_channels    = p_wf->nChannels;
 277                     p_pack->output_format.i_block_align = p_wf->nBlockAlign;
 278                     p_pack->output_format.i_bitrate     = p_wf->nAvgBytesPerSec * 8;
 279                     p_pack->output_format.i_extra_data  = p_wf->cbSize;
 280                     if( p_wf->cbSize  > 0 )
 281                     {
 282                         p_pack->output_format.p_extra_data =
 283                             malloc( p_pack->output_format.i_extra_data );
 284                         memcpy( p_pack->output_format.p_extra_data,
 285                                 &p_wf[1],
 286                                 p_pack->output_format.i_extra_data );
 287                     }
 288                     else
 289                     {
 290                         p_pack->output_format.p_extra_data = NULL;
 291                     }
 292                 }
 293                 else
 294                 {
 295                     p_pack->output_format.i_sample_rate = 0;
 296                     p_pack->output_format.i_channels    = 0;
 297                     p_pack->output_format.i_block_align = 0;
 298                     p_pack->output_format.i_bitrate     = 0;
 299                     p_pack->output_format.i_extra_data  = 0;
 300                     p_pack->output_format.p_extra_data  = NULL;
 301                 }
 302             }
 303             break;
 304
 305         case VIDEO_ES:
 306             {
 307                 BITMAPINFOHEADER *p_bih = (BITMAPINFOHEADER*)p_pack->p_fifo->p_bitmapinfoheader;
 308
 309                 p_pack->output_format.i_bitrate = 0;
 310                 if( p_bih )
 311                 {
 312                     p_pack->output_format.i_width  = p_bih->biWidth;
 313                     p_pack->output_format.i_height = p_bih->biHeight;
 314                     p_pack->output_format.i_extra_data  = p_bih->biSize - sizeof( BITMAPINFOHEADER );
 315                     if( p_pack->output_format.i_extra_data > 0 )
 316                     {
 317                         p_pack->output_format.p_extra_data =
 318                             malloc( p_pack->output_format.i_extra_data );
 319                         memcpy( p_pack->output_format.p_extra_data,
 320                                 &p_bih[1],
 321                                 p_pack->output_format.i_extra_data );
 322                     }
 323                 }
 324                 else
 325                 {
 326                     p_pack->output_format.i_width  = 0;
 327                     p_pack->output_format.i_height = 0;
 328                     p_pack->output_format.i_extra_data  = 0;
 329                     p_pack->output_format.p_extra_data  = NULL;
 330                 }
 331             }
 332             break;
 333
 334         default:
 335             return VLC_EGENERIC;
 336     }
 337
 338     p_pack->p_sout_input =
 339         sout_InputNew( p_pack->p_fifo,
 340                        &p_pack->output_format );
 341
 342     if( !p_pack->p_sout_input )
 343     {
 344         msg_Err( p_pack->p_fifo, "cannot add a new stream" );
 345         return VLC_EGENERIC;
 346     }
 347 //    p_pack->i_last_pts = 0;
 348     return( VLC_SUCCESS );
 349 }
 350
 351 /*****************************************************************************
 352  * PacketizeThread: packetize an unit (here copy a complete pes)
 353  *****************************************************************************/
 354 static void PacketizeThread( packetizer_thread_t *p_pack )
 355 {
 356     sout_buffer_t   *p_sout_buffer;
 357     pes_packet_t    *p_pes;
 358     ssize_t         i_size;
 359     mtime_t         i_pts;
 360
 361     /* **** get samples count **** */
 362     input_ExtractPES( p_pack->p_fifo, &p_pes );
 363     if( !p_pes )
 364     {
 365         p_pack->p_fifo->b_error = 1;
 366         return;
 367     }
 368     i_pts = p_pes->i_pts;
 369
 370     if( i_pts <= 0 ) //&& p_pack->i_last_pts <= 0 )
 371     {
 372         msg_Dbg( p_pack->p_fifo, "need pts != 0" );
 373         input_DeletePES( p_pack->p_fifo->p_packets_mgt, p_pes );
 374         return;
 375     }
 376
 377     i_size = p_pes->i_pes_size;
 378
 379 //    msg_Dbg( p_pack->p_fifo, "pes size:%d", i_size );
 380     if( i_size > 0 )
 381     {
 382         pes_packet_t    *p_pes_next;
 383         data_packet_t   *p_data;
 384         ssize_t          i_buffer;
 385
 386         p_sout_buffer =
 387             sout_BufferNew( p_pack->p_sout_input->p_sout, i_size );
 388         if( !p_sout_buffer )
 389         {
 390             p_pack->p_fifo->b_error = 1;
 391             input_DeletePES( p_pack->p_fifo->p_packets_mgt, p_pes );
 392             return;
 393         }
 394         /* TODO: memcpy of the pes packet */
 395         for( i_buffer = 0, p_data = p_pes->p_first;
 396              p_data != NULL && i_buffer < i_size;
 397              p_data = p_data->p_next)
 398         {
 399             ssize_t i_copy;
 400
 401             i_copy = __MIN( p_data->p_payload_end - p_data->p_payload_start,
 402                             i_size - i_buffer );
 403             if( i_copy > 0 )
 404             {
 405                 p_pack->p_fifo->p_vlc->pf_memcpy( p_sout_buffer->p_buffer + i_buffer,
 406                                                   p_data->p_payload_start,
 407                                                   i_copy );
 408             }
 409             i_buffer += i_copy;
 410         }
 411         p_sout_buffer->i_length = 0;
 412         p_sout_buffer->i_dts = i_pts; //p_pes->i_pts - p_pack->i_pts_start;
 413         p_sout_buffer->i_pts = i_pts; //p_pes->i_pts - p_pack->i_pts_start;
 414         p_sout_buffer->i_bitrate = 0;
 415
 416         input_ShowPES( p_pack->p_fifo, &p_pes_next );
 417         if( p_pes_next )
 418         {
 419             p_sout_buffer->i_length = p_pes_next->i_pts - i_pts;
 420         }
 421         sout_InputSendBuffer( p_pack->p_sout_input,
 422                                p_sout_buffer );
 423     }
 424
 425     input_DeletePES( p_pack->p_fifo->p_packets_mgt, p_pes );
 426 }
 427
 428
 429 /*****************************************************************************
 430  * EndThread : packetizer thread destruction
 431  *****************************************************************************/
 432 static void EndThread ( packetizer_thread_t *p_pack)
 433 {
 434     if( p_pack->p_sout_input )
 435     {
 436         sout_InputDelete( p_pack->p_sout_input );
 437     }
 438 }
 439
 440 static void input_ShowPES( decoder_fifo_t *p_fifo, pes_packet_t **pp_pes )
 441 {
 442     pes_packet_t *p_pes;
 443
 444     vlc_mutex_lock( &p_fifo->data_lock );
 445
 446     if( p_fifo->p_first == NULL )
 447     {
 448         if( p_fifo->b_die )
 449         {
 450             vlc_mutex_unlock( &p_fifo->data_lock );
 451             if( pp_pes ) *pp_pes = NULL;
 452             return;
 453         }
 454
 455         /* Signal the input thread we're waiting. This is only
 456          * needed in case of slave clock (ES plug-in) but it won't
 457          * harm. */
 458         vlc_cond_signal( &p_fifo->data_wait );
 459
 460         /* Wait for the input to tell us when we received a packet. */
 461         vlc_cond_wait( &p_fifo->data_wait, &p_fifo->data_lock );
 462     }
 463     p_pes = p_fifo->p_first;
 464     vlc_mutex_unlock( &p_fifo->data_lock );
 465
 466     if( pp_pes )
 467     {
 468         *pp_pes = p_pes;
 469     }
 470 }
 471