git.sesse.net Git - vlc/blob - src/video_parser/video_parser.c

   1 /*******************************************************************************
   2  * video_parser.c : video parser thread
   3  * (c)1999 VideoLAN
   4  *******************************************************************************/
   5
   6 /* ?? passer en terminate/destroy avec les signaux supplémentaires */
   7
   8 /*******************************************************************************
   9  * Preamble
  10  *******************************************************************************/
  11 #include <errno.h>
  12 #include <stdlib.h>
  13 #include <stdio.h>
  14 #include <unistd.h>
  15 #include <string.h>
  16 #include <sys/uio.h>
  17 #include <X11/Xlib.h>
  18 #include <X11/extensions/XShm.h>
  19
  20 #include "config.h"
  21 #include "common.h"
  22 #include "mtime.h"
  23 #include "vlc_thread.h"
  24
  25 #include "intf_msg.h"
  26 #include "debug.h"                     /* ?? temporaire, requis par netlist.h */
  27
  28 #include "input.h"
  29 #include "input_netlist.h"
  30 #include "decoder_fifo.h"
  31 #include "video.h"
  32 #include "video_output.h"
  33
  34 #include "vdec_idct.h"
  35 #include "video_decoder.h"
  36 #include "vdec_motion.h"
  37
  38 #include "vpar_blocks.h"
  39 #include "vpar_headers.h"
  40 #include "video_fifo.h"
  41 #include "vpar_synchro.h"
  42 #include "video_parser.h"
  43
  44 /*
  45  * Local prototypes
  46  */
  47 //static int      CheckConfiguration  ( video_cfg_t *p_cfg );
  48 static int      InitThread          ( vpar_thread_t *p_vpar );
  49 static void     RunThread           ( vpar_thread_t *p_vpar );
  50 static void     ErrorThread         ( vpar_thread_t *p_vpar );
  51 static void     EndThread           ( vpar_thread_t *p_vpar );
  52
  53 /*******************************************************************************
  54  * vpar_CreateThread: create a generic parser thread
  55  *******************************************************************************
  56  * This function creates a new video parser thread, and returns a pointer
  57  * to its description. On error, it returns NULL.
  58  * Following configuration properties are used:
  59  * ??
  60  *******************************************************************************/
  61 #include "main.h"
  62 #include "interface.h"
  63 extern main_t* p_main;
  64 vpar_thread_t * vpar_CreateThread( /* video_cfg_t *p_cfg, */ input_thread_t *p_input /*,
  65                                    vout_thread_t *p_vout, int *pi_status */ )
  66 {
  67     vpar_thread_t *     p_vpar;
  68
  69     intf_DbgMsg("vpar debug: creating video parser thread\n");
  70
  71     /* Allocate the memory needed to store the thread's structure */
  72     if ( (p_vpar = (vpar_thread_t *)malloc( sizeof(vpar_thread_t) )) == NULL )
  73     {
  74         intf_ErrMsg("vpar error: not enough memory for vpar_CreateThread() to create the new thread\n");
  75         return( NULL );
  76     }
  77
  78     /*
  79      * Initialize the thread properties
  80      */
  81     p_vpar->b_die = 0;
  82     p_vpar->b_error = 0;
  83
  84     /*
  85      * Initialize the input properties
  86      */
  87     /* Initialize the decoder fifo's data lock and conditional variable and set
  88      * its buffer as empty */
  89     vlc_mutex_init( &p_vpar->fifo.data_lock );
  90     vlc_cond_init( &p_vpar->fifo.data_wait );
  91     p_vpar->fifo.i_start = 0;
  92     p_vpar->fifo.i_end = 0;
  93     /* Initialize the bit stream structure */
  94     p_vpar->bit_stream.p_input = p_input;
  95     p_vpar->bit_stream.p_decoder_fifo = &p_vpar->fifo;
  96     p_vpar->bit_stream.fifo.buffer = 0;
  97     p_vpar->bit_stream.fifo.i_available = 0;
  98
  99     /*
 100      * Initialize the synchro properties
 101      */
 102     p_vpar->synchro.modulo = 0;
 103     /* assume there were about 3 P and 6 B images between I's */
 104     p_vpar->synchro.current_p_count = 1;
 105     p_vpar->synchro.p_count_predict = 3;
 106     p_vpar->synchro.current_b_count = 1;
 107     p_vpar->synchro.b_count_predict = 6;
 108     {
 109         int i;
 110         for( i=0; i<6; i++)
 111         {
 112             p_vpar->synchro.tab_p[i].mean = 3;
 113             p_vpar->synchro.tab_p[i].deviation = .5;
 114
 115             p_vpar->synchro.tab_b[i].mean = 6;
 116             p_vpar->synchro.tab_b[i].deviation = .5;
 117         }
 118     }
 119
 120 /* FIXME !!!! */
 121 p_vpar->p_vout = p_main->p_intf->p_vout;
 122
 123     /* Spawn the video parser thread */
 124     if ( vlc_thread_create(&p_vpar->thread_id, "video parser", (vlc_thread_func_t)RunThread, (void *)p_vpar) )
 125     {
 126         intf_ErrMsg("vpar error: can't spawn video parser thread\n");
 127         free( p_vpar );
 128         return( NULL );
 129     }
 130
 131     intf_DbgMsg("vpar debug: video parser thread (%p) created\n", p_vpar);
 132     return( p_vpar );
 133 }
 134
 135 /*******************************************************************************
 136  * vpar_DestroyThread: destroy a generic parser thread
 137  *******************************************************************************
 138  * Destroy a terminated thread. This function will return 0 if the thread could
 139  * be destroyed, and non 0 else. The last case probably means that the thread
 140  * was still active, and another try may succeed.
 141  *******************************************************************************/
 142 void vpar_DestroyThread( vpar_thread_t *p_vpar /*, int *pi_status */ )
 143 {
 144     intf_DbgMsg("vpar debug: requesting termination of video parser thread %p\n", p_vpar);
 145
 146     /* Ask thread to kill itself */
 147     p_vpar->b_die = 1;
 148     /* Make sure the parser thread leaves the GetByte() function */
 149     vlc_mutex_lock( &(p_vpar->fifo.data_lock) );
 150     vlc_cond_signal( &(p_vpar->fifo.data_wait) );
 151     vlc_mutex_unlock( &(p_vpar->fifo.data_lock) );
 152
 153     /* Waiting for the parser thread to exit */
 154     /* Remove this as soon as the "status" flag is implemented */
 155     vlc_thread_join( p_vpar->thread_id );
 156 }
 157
 158 /* following functions are local */
 159
 160 /*******************************************************************************
 161  * CheckConfiguration: check vpar_CreateThread() configuration
 162  *******************************************************************************
 163  * Set default parameters where required. In DEBUG mode, check if configuration
 164  * is valid.
 165  *******************************************************************************/
 166 #if 0
 167 static int CheckConfiguration( video_cfg_t *p_cfg )
 168 {
 169     /* ?? */
 170
 171     return( 0 );
 172 }
 173 #endif
 174
 175 /*******************************************************************************
 176  * InitThread: initialize vpar output thread
 177  *******************************************************************************
 178  * This function is called from RunThread and performs the second step of the
 179  * initialization. It returns 0 on success. Note that the thread's flag are not
 180  * modified inside this function.
 181  *******************************************************************************/
 182 static int InitThread( vpar_thread_t *p_vpar )
 183 {
 184     int     i_dummy;
 185
 186     intf_DbgMsg("vpar debug: initializing video parser thread %p\n", p_vpar);
 187
 188     /* Our first job is to initialize the bit stream structure with the
 189      * beginning of the input stream */
 190     vlc_mutex_lock( &p_vpar->fifo.data_lock );
 191     while ( DECODER_FIFO_ISEMPTY(p_vpar->fifo) )
 192     {
 193         vlc_cond_wait( &p_vpar->fifo.data_wait, &p_vpar->fifo.data_lock );
 194     }
 195     p_vpar->bit_stream.p_ts = DECODER_FIFO_START( p_vpar->fifo )->p_first_ts;
 196     p_vpar->bit_stream.i_byte = p_vpar->bit_stream.p_ts->i_payload_start;
 197     vlc_mutex_unlock( &p_vpar->fifo.data_lock );
 198
 199     /* Initialize parsing data */
 200     p_vpar->sequence.p_forward = NULL;
 201     p_vpar->sequence.p_backward = NULL;
 202     p_vpar->sequence.intra_quant.b_allocated = 0;
 203     p_vpar->sequence.nonintra_quant.b_allocated = 0;
 204     p_vpar->sequence.chroma_intra_quant.b_allocated = 0;
 205     p_vpar->sequence.chroma_nonintra_quant.b_allocated = 0;
 206     /* Initialize copyright information */
 207     p_vpar->sequence.b_copyright_flag = 0;
 208     p_vpar->sequence.b_original = 0;
 209     p_vpar->sequence.i_copyright_id = 0;
 210     p_vpar->sequence.i_copyright_nb = 0;
 211
 212     p_vpar->picture.p_picture = NULL;
 213     p_vpar->picture.i_current_structure = 0;
 214
 215     /* Initialize other properties */
 216 #ifdef STATS
 217     p_vpar->c_loops = 0;
 218     p_vpar->c_idle_loops = 0;
 219     p_vpar->c_pictures = 0;
 220     p_vpar->c_i_pictures = 0;
 221     p_vpar->c_p_pictures = 0;
 222     p_vpar->c_b_pictures = 0;
 223     p_vpar->c_decoded_pictures = 0;
 224     p_vpar->c_decoded_i_pictures = 0;
 225     p_vpar->c_decoded_p_pictures = 0;
 226     p_vpar->c_decoded_b_pictures = 0;
 227 #endif
 228
 229     /* Initialize video FIFO */
 230     vpar_InitFIFO( p_vpar );
 231
 232     bzero( p_vpar->p_vdec, NB_VDEC*sizeof(vdec_thread_t *) );
 233
 234     /* Spawn video_decoder threads */
 235     /* ??? modify the number of vdecs at runtime ? */
 236     for( i_dummy = 0; i_dummy < NB_VDEC; i_dummy++ )
 237     {
 238         if( (p_vpar->p_vdec[i_dummy] = vdec_CreateThread( p_vpar )) == NULL )
 239         {
 240             return( 1 );
 241         }
 242     }
 243
 244     /* Initialize lookup tables */
 245 #if defined(MPEG2_COMPLIANT) && !defined(VDEC_DFT)
 246     vpar_InitCrop( p_vpar );
 247 #endif
 248     vpar_InitMbAddrInc( p_vpar );
 249     vpar_InitDCTTables( p_vpar );
 250     vpar_InitPMBType( p_vpar );
 251     vpar_InitBMBType( p_vpar );
 252     vpar_InitCodedPattern( p_vpar );
 253     vpar_InitDCTTables( p_vpar );
 254
 255     /* Mark thread as running and return */
 256     intf_DbgMsg("vpar debug: InitThread(%p) succeeded\n", p_vpar);
 257     return( 0 );
 258 }
 259
 260 /*******************************************************************************
 261  * RunThread: generic parser thread
 262  *******************************************************************************
 263  * Video parser thread. This function does only returns when the thread is
 264  * terminated.
 265  *******************************************************************************/
 266 static void RunThread( vpar_thread_t *p_vpar )
 267 {
 268     intf_DbgMsg("vpar debug: running video parser thread (%p) (pid == %i)\n", p_vpar, getpid());
 269
 270     /*
 271      * Initialize thread
 272      */
 273     p_vpar->b_error = InitThread( p_vpar );
 274     if( p_vpar->b_error )
 275     {
 276         return;
 277     }
 278     p_vpar->b_run = 1;
 279
 280     /*
 281      * Main loop - it is not executed if an error occured during
 282      * initialization
 283      */
 284     while( (!p_vpar->b_die) && (!p_vpar->b_error) )
 285     {
 286         /* Find the next sequence header in the stream */
 287         p_vpar->b_error = vpar_NextSequenceHeader( p_vpar );
 288
 289 #ifdef STATS
 290         p_vpar->c_sequences++;
 291 #endif
 292
 293         while( (!p_vpar->b_die) && (!p_vpar->b_error) )
 294         {
 295             /* Parse the next sequence, group or picture header */
 296             if( vpar_ParseHeader( p_vpar ) )
 297             {
 298                 /* End of sequence */
 299                 break;
 300             };
 301         }
 302     }
 303
 304     /*
 305      * Error loop
 306      */
 307     if( p_vpar->b_error )
 308     {
 309         ErrorThread( p_vpar );
 310     }
 311
 312     /* End of thread */
 313     EndThread( p_vpar );
 314     p_vpar->b_run = 0;
 315 }
 316
 317 /*******************************************************************************
 318  * ErrorThread: RunThread() error loop
 319  *******************************************************************************
 320  * This function is called when an error occured during thread main's loop. The
 321  * thread can still receive feed, but must be ready to terminate as soon as
 322  * possible.
 323  *******************************************************************************/
 324 static void ErrorThread( vpar_thread_t *p_vpar )
 325 {
 326     /* Wait until a `die' order */
 327     while( !p_vpar->b_die )
 328     {
 329         /* We take the lock, because we are going to read/write the start/end
 330          * indexes of the parser fifo */
 331         vlc_mutex_lock( &p_vpar->fifo.data_lock );
 332
 333         /* ?? trash all trashable PES packets */
 334         while( !DECODER_FIFO_ISEMPTY(p_vpar->fifo) )
 335         {
 336             input_NetlistFreePES( p_vpar->bit_stream.p_input,
 337                                   DECODER_FIFO_START(p_vpar->fifo) );
 338             DECODER_FIFO_INCSTART( p_vpar->fifo );
 339         }
 340
 341         vlc_mutex_unlock( &p_vpar->fifo.data_lock );
 342         /* Sleep a while */
 343         msleep( VPAR_IDLE_SLEEP );
 344     }
 345 }
 346
 347 /*******************************************************************************
 348  * EndThread: thread destruction
 349  *******************************************************************************
 350  * This function is called when the thread ends after a sucessfull
 351  * initialization.
 352  *******************************************************************************/
 353 static void EndThread( vpar_thread_t *p_vpar )
 354 {
 355     int i_dummy;
 356
 357     intf_DbgMsg("vpar debug: destroying video parser thread %p\n", p_vpar);
 358
 359 #ifdef DEBUG
 360     /* Check for remaining PES packets */
 361     /* ?? */
 362 #endif
 363
 364     /* Destroy thread structures allocated by InitThread */
 365 //    vout_DestroyStream( p_vpar->p_vout, p_vpar->i_stream );
 366     /* ?? */
 367
 368     /* Destroy vdec threads */
 369     for( i_dummy = 0; i_dummy < NB_VDEC; i_dummy++ )
 370     {
 371         if( p_vpar->p_vdec[i_dummy] != NULL )
 372             vdec_DestroyThread( p_vpar->p_vdec[i_dummy] );
 373         else
 374             break;
 375     }
 376
 377     intf_DbgMsg("vpar debug: EndThread(%p)\n", p_vpar);
 378 }