git.sesse.net Git - vlc/blob - src/input/input_programs.c

   1 /*****************************************************************************
   2  * input_programs.c: es_descriptor_t, pgrm_descriptor_t management
   3  *****************************************************************************
   4  * Copyright (C) 1999, 2000 VideoLAN
   5  * $Id: input_programs.c,v 1.28 2001/02/07 15:32:26 massiot Exp $
   6  *
   7  * Authors:
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  22  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27 #include "defs.h"
  28
  29 #include <stdlib.h>
  30
  31 #include "config.h"
  32 #include "common.h"
  33 #include "threads.h"
  34 #include "mtime.h"
  35 #include "debug.h"
  36
  37 #include "intf_msg.h"
  38
  39 #include "stream_control.h"
  40 #include "input_ext-intf.h"
  41 #include "input_ext-dec.h"
  42 #include "input.h"
  43
  44 #include "main.h"                                     /* --noaudio --novideo */
  45
  46 /*
  47  * NOTICE : all of these functions expect you to have taken the lock on
  48  * p_input->stream.lock
  49  */
  50
  51 /*****************************************************************************
  52  * input_InitStream: init the stream descriptor of the given input
  53  *****************************************************************************/
  54 int input_InitStream( input_thread_t * p_input, size_t i_data_len )
  55 {
  56     p_input->stream.i_stream_id = 0;
  57     p_input->stream.pp_es = NULL;
  58     p_input->stream.pp_selected_es = NULL;
  59     p_input->stream.pp_programs = NULL;
  60
  61     if( i_data_len )
  62     {
  63         if ( (p_input->stream.p_demux_data = malloc( i_data_len )) == NULL )
  64         {
  65             intf_ErrMsg( "Unable to allocate memory in input_InitStream");
  66             return 1;
  67         }
  68         memset( p_input->stream.p_demux_data, 0, i_data_len );
  69     }
  70
  71     return 0;
  72 }
  73
  74 /*****************************************************************************
  75  * input_EndStream: free all stream descriptors
  76  *****************************************************************************/
  77 void input_EndStream( input_thread_t * p_input )
  78 {
  79     /* Free all programs and associated ES, and associated decoders. */
  80     while( p_input->stream.i_pgrm_number )
  81     {
  82         input_DelProgram( p_input, p_input->stream.pp_programs[0] );
  83     }
  84
  85     /* Free standalone ES */
  86     while( p_input->stream.i_es_number )
  87     {
  88         input_DelES( p_input, p_input->stream.pp_es[0] );
  89     }
  90 }
  91
  92 /*****************************************************************************
  93  * input_FindProgram: returns a pointer to a program described by its ID
  94  *****************************************************************************/
  95 pgrm_descriptor_t * input_FindProgram( input_thread_t * p_input, u16 i_pgrm_id )
  96 {
  97     int     i;
  98
  99     for( i = 0; i < p_input->stream.i_pgrm_number; i++ )
 100     {
 101         if( p_input->stream.pp_programs[i]->i_number == i_pgrm_id )
 102         {
 103             return p_input->stream.pp_programs[i];
 104         }
 105     }
 106
 107     return( NULL );
 108 }
 109
 110 /*****************************************************************************
 111  * input_AddProgram: add and init a program descriptor
 112  *****************************************************************************
 113  * This program descriptor will be referenced in the given stream descriptor
 114  *****************************************************************************/
 115 pgrm_descriptor_t * input_AddProgram( input_thread_t * p_input,
 116                                       u16 i_pgrm_id, size_t i_data_len )
 117 {
 118     /* Where to add the pgrm */
 119     int i_pgrm_index = p_input->stream.i_pgrm_number;
 120
 121     intf_DbgMsg("Adding description for pgrm %d", i_pgrm_id);
 122
 123     /* Add an entry to the list of program associated with the stream */
 124     p_input->stream.i_pgrm_number++;
 125     p_input->stream.pp_programs = realloc( p_input->stream.pp_programs,
 126                                            p_input->stream.i_pgrm_number
 127                                             * sizeof(pgrm_descriptor_t *) );
 128     if( p_input->stream.pp_programs == NULL )
 129     {
 130         intf_ErrMsg( "Unable to realloc memory in input_AddProgram" );
 131         return( NULL );
 132     }
 133
 134     /* Allocate the structure to store this description */
 135     p_input->stream.pp_programs[i_pgrm_index] =
 136                                         malloc( sizeof(pgrm_descriptor_t) );
 137     if( p_input->stream.pp_programs[i_pgrm_index] == NULL )
 138     {
 139         intf_ErrMsg( "Unable to allocate memory in input_AddProgram" );
 140         return( NULL );
 141     }
 142
 143     /* Init this entry */
 144     p_input->stream.pp_programs[i_pgrm_index]->i_number = i_pgrm_id;
 145     p_input->stream.pp_programs[i_pgrm_index]->b_is_ok = 0;
 146     p_input->stream.pp_programs[i_pgrm_index]->i_version = 0;
 147
 148     p_input->stream.pp_programs[i_pgrm_index]->i_es_number = 0;
 149     p_input->stream.pp_programs[i_pgrm_index]->pp_es = NULL;
 150
 151     input_ClockInit( p_input->stream.pp_programs[i_pgrm_index] );
 152
 153     p_input->stream.pp_programs[i_pgrm_index]->i_synchro_state
 154                                                 = SYNCHRO_START;
 155     p_input->stream.pp_programs[i_pgrm_index]->b_discontinuity = 0;
 156
 157     p_input->stream.pp_programs[i_pgrm_index]->p_vout
 158                                             = p_input->p_default_vout;
 159     p_input->stream.pp_programs[i_pgrm_index]->p_aout
 160                                             = p_input->p_default_aout;
 161
 162     if( i_data_len )
 163     {
 164         p_input->stream.pp_programs[i_pgrm_index]->p_demux_data =
 165             malloc( i_data_len );
 166         if( p_input->stream.pp_programs[i_pgrm_index]->p_demux_data == NULL )
 167         {
 168             intf_ErrMsg( "Unable to allocate memory in input_AddProgram" );
 169             return( NULL );
 170         }
 171         memset( p_input->stream.pp_programs[i_pgrm_index]->p_demux_data, 0,
 172                 i_data_len );
 173     }
 174
 175     return p_input->stream.pp_programs[i_pgrm_index];
 176 }
 177
 178 /*****************************************************************************
 179  * input_DelProgram: destroy a program descriptor
 180  *****************************************************************************
 181  * All ES descriptions referenced in the descriptor will be deleted.
 182  *****************************************************************************/
 183 void input_DelProgram( input_thread_t * p_input, pgrm_descriptor_t * p_pgrm )
 184 {
 185     int i_pgrm_index;
 186
 187     ASSERT( p_pgrm );
 188
 189     intf_DbgMsg("Deleting description for pgrm %d", p_pgrm->i_number);
 190
 191     /* Free the structures that describe the es that belongs to that program */
 192     while( p_pgrm->i_es_number )
 193     {
 194         input_DelES( p_input, p_pgrm->pp_es[0] );
 195     }
 196
 197     /* Free the demux data */
 198     if( p_pgrm->p_demux_data != NULL )
 199     {
 200         free( p_pgrm->p_demux_data );
 201     }
 202
 203     /* Find the program in the programs table */
 204     for( i_pgrm_index = 0; i_pgrm_index < p_input->stream.i_pgrm_number;
 205          i_pgrm_index++ )
 206     {
 207         if( p_input->stream.pp_programs[i_pgrm_index] == p_pgrm )
 208             break;
 209     }
 210
 211     /* Remove this program from the stream's list of programs */
 212     p_input->stream.i_pgrm_number--;
 213
 214     p_input->stream.pp_programs[i_pgrm_index] =
 215         p_input->stream.pp_programs[p_input->stream.i_pgrm_number];
 216     p_input->stream.pp_programs = realloc( p_input->stream.pp_programs,
 217                                            p_input->stream.i_pgrm_number
 218                                             * sizeof(pgrm_descriptor_t *) );
 219
 220     if( p_input->stream.i_pgrm_number && p_input->stream.pp_programs == NULL)
 221     {
 222         intf_ErrMsg( "input error: unable to realloc program list"
 223                      " in input_DelProgram" );
 224     }
 225
 226     /* Free the description of this program */
 227     free( p_pgrm );
 228 }
 229
 230 /*****************************************************************************
 231  * input_FindES: returns a pointer to an ES described by its ID
 232  *****************************************************************************/
 233 es_descriptor_t * input_FindES( input_thread_t * p_input, u16 i_es_id )
 234 {
 235     int     i;
 236
 237     for( i = 0; i < p_input->stream.i_es_number; i++ )
 238     {
 239         if( p_input->stream.pp_es[i]->i_id == i_es_id )
 240         {
 241             return p_input->stream.pp_es[i];
 242         }
 243     }
 244
 245     return( NULL );
 246 }
 247
 248 /*****************************************************************************
 249  * input_AddES:
 250  *****************************************************************************
 251  * Reserve a slot in the table of ES descriptors for the ES and add it to the
 252  * list of ES of p_pgrm. If p_pgrm if NULL, then the ES is considered as stand
 253  * alone (PSI ?)
 254  *****************************************************************************/
 255 es_descriptor_t * input_AddES( input_thread_t * p_input,
 256                                pgrm_descriptor_t * p_pgrm, u16 i_es_id,
 257                                size_t i_data_len )
 258 {
 259     es_descriptor_t * p_es;
 260
 261     intf_DbgMsg("Adding description for ES 0x%x", i_es_id);
 262
 263     p_es = (es_descriptor_t *)malloc( sizeof(es_descriptor_t) );
 264     if( p_es == NULL )
 265     {
 266         intf_ErrMsg( "Unable to allocate memory in input_AddES" );
 267         return( NULL);
 268     }
 269     p_input->stream.i_es_number++;
 270     p_input->stream.pp_es = realloc( p_input->stream.pp_es,
 271                                      p_input->stream.i_es_number
 272                                       * sizeof(es_descriptor_t *) );
 273     if( p_input->stream.pp_es == NULL )
 274     {
 275         intf_ErrMsg( "Unable to realloc memory in input_AddES" );
 276         return( NULL );
 277     }
 278     p_input->stream.pp_es[p_input->stream.i_es_number - 1] = p_es;
 279
 280     /* Init its values */
 281     p_es->i_id = i_es_id;
 282     p_es->b_discontinuity = 0;
 283     p_es->p_pes = NULL;
 284     p_es->p_decoder_fifo = NULL;
 285     p_es->b_audio = 0;
 286
 287     if( i_data_len )
 288     {
 289         p_es->p_demux_data = malloc( i_data_len );
 290         if( p_es->p_demux_data == NULL )
 291         {
 292             intf_ErrMsg( "Unable to allocate memory in input_AddES" );
 293             return( NULL );
 294         }
 295         memset( p_es->p_demux_data, 0, i_data_len );
 296     }
 297     else
 298     {
 299         p_es->p_demux_data = NULL;
 300     }
 301
 302     /* Add this ES to the program definition if one is given */
 303     if( p_pgrm )
 304     {
 305         p_pgrm->i_es_number++;
 306         p_pgrm->pp_es = realloc( p_pgrm->pp_es,
 307                                  p_pgrm->i_es_number
 308                                   * sizeof(es_descriptor_t *) );
 309         if( p_pgrm->pp_es == NULL )
 310         {
 311             intf_ErrMsg( "Unable to realloc memory in input_AddES" );
 312             return( NULL );
 313         }
 314         p_pgrm->pp_es[p_pgrm->i_es_number - 1] = p_es;
 315         p_es->p_pgrm = p_pgrm;
 316     }
 317     else
 318     {
 319         p_es->p_pgrm = NULL;
 320     }
 321
 322     return p_es;
 323 }
 324
 325 /*****************************************************************************
 326  * input_DelES:
 327  *****************************************************************************/
 328 void input_DelES( input_thread_t * p_input, es_descriptor_t * p_es )
 329 {
 330     int                     i_index, i_es_index;
 331     pgrm_descriptor_t *     p_pgrm;
 332
 333     ASSERT( p_es );
 334     p_pgrm = p_es->p_pgrm;
 335
 336     /* Kill associated decoder, if any. */
 337     if( p_es->p_decoder_fifo != NULL )
 338     {
 339         input_EndDecoder( p_input, p_es );
 340     }
 341
 342     /* Remove this ES from the description of the program if it is associated to
 343      * one */
 344     if( p_pgrm )
 345     {
 346         for( i_index = 0; i_index < p_pgrm->i_es_number; i_index++ )
 347         {
 348             if( p_pgrm->pp_es[i_index] == p_es )
 349             {
 350                 p_pgrm->i_es_number--;
 351                 p_pgrm->pp_es[i_index] = p_pgrm->pp_es[p_pgrm->i_es_number];
 352                 p_pgrm->pp_es = realloc( p_pgrm->pp_es,
 353                                          p_pgrm->i_es_number
 354                                           * sizeof(es_descriptor_t *));
 355                 if( p_pgrm->i_es_number && p_pgrm->pp_es == NULL )
 356                 {
 357                     intf_ErrMsg( "Unable to realloc memory in input_DelES" );
 358                 }
 359                 break;
 360             }
 361         }
 362     }
 363
 364     /* Free the demux data */
 365     if( p_es->p_demux_data != NULL )
 366     {
 367         free( p_es->p_demux_data );
 368     }
 369
 370     /* Find the ES in the ES table */
 371     for( i_es_index = 0; i_es_index < p_input->stream.i_es_number;
 372          i_es_index++ )
 373     {
 374         if( p_input->stream.pp_es[i_es_index] == p_es )
 375             break;
 376     }
 377
 378     /* Free the ES */
 379     free( p_es );
 380     p_input->stream.i_es_number--;
 381     p_input->stream.pp_es[i_es_index] =
 382                     p_input->stream.pp_es[p_input->stream.i_es_number];
 383     p_input->stream.pp_es = realloc( p_input->stream.pp_es,
 384                                      p_input->stream.i_es_number
 385                                       * sizeof(es_descriptor_t *));
 386     if( p_input->stream.i_es_number && p_input->stream.pp_es == NULL )
 387     {
 388         intf_ErrMsg( "Unable to realloc memory in input_DelES" );
 389     }
 390
 391 }
 392
 393 #ifdef STATS
 394 /*****************************************************************************
 395  * input_DumpStream: dumps the contents of a stream descriptor
 396  *****************************************************************************/
 397 void input_DumpStream( input_thread_t * p_input )
 398 {
 399     int i, j;
 400 #define S   p_input->stream
 401     intf_Msg( "input info: Dumping stream ID 0x%x", S.i_stream_id );
 402     if( S.b_seekable )
 403         intf_Msg( "input info: seekable stream, position: %d/%d",
 404                   S.i_tell, S.i_size );
 405     else
 406         intf_Msg( "input info: %s", S.b_pace_control ? "pace controlled" :
 407                   "pace un-controlled" );
 408 #undef S
 409     for( i = 0; i < p_input->stream.i_pgrm_number; i++ )
 410     {
 411 #define P   p_input->stream.pp_programs[i]
 412         intf_Msg( "input info: Dumping program 0x%x, version %d (%s)",
 413                   P->i_number, P->i_version,
 414                   P->b_is_ok ? "complete" : "partial" );
 415 #undef P
 416         for( j = 0; j < p_input->stream.pp_programs[i]->i_es_number; j++ )
 417         {
 418 #define ES  p_input->stream.pp_programs[i]->pp_es[j]
 419             intf_Msg( "input info: ES 0x%x, stream 0x%x, type 0x%x, %s",
 420                       ES->i_id, ES->i_stream_id, ES->i_type,
 421                       ES->p_decoder_fifo != NULL ? "selected" : "not selected");
 422 #undef ES
 423         }
 424     }
 425 }
 426 #endif
 427
 428 /*****************************************************************************
 429  * InitDecConfig: initializes a decoder_config_t
 430  *****************************************************************************/
 431 static int InitDecConfig( input_thread_t * p_input, es_descriptor_t * p_es,
 432                           decoder_config_t * p_config )
 433 {
 434     p_config->i_id = p_es->i_id;
 435     p_config->i_type = p_es->i_type;
 436     p_config->p_stream_ctrl =
 437         &p_input->stream.control;
 438
 439     /* Decoder FIFO */
 440     if( (p_config->p_decoder_fifo =
 441             (decoder_fifo_t *)malloc( sizeof(decoder_fifo_t) )) == NULL )
 442     {
 443         intf_ErrMsg( "Out of memory" );
 444         return( -1 );
 445     }
 446
 447     vlc_mutex_init(&p_config->p_decoder_fifo->data_lock);
 448     vlc_cond_init(&p_config->p_decoder_fifo->data_wait);
 449     p_config->p_decoder_fifo->i_start = p_config->p_decoder_fifo->i_end = 0;
 450     p_config->p_decoder_fifo->b_die = p_config->p_decoder_fifo->b_error = 0;
 451     p_config->p_decoder_fifo->p_packets_mgt = p_input->p_method_data;
 452     p_config->p_decoder_fifo->pf_delete_pes =
 453         p_input->p_plugin->pf_delete_pes;
 454     p_es->p_decoder_fifo = p_config->p_decoder_fifo;
 455
 456     p_config->pf_init_bit_stream = InitBitstream;
 457
 458     return( 0 );
 459 }
 460
 461 /*****************************************************************************
 462  * GetVdecConfig: returns a valid vdec_config_t
 463  *****************************************************************************/
 464 static vdec_config_t * GetVdecConfig( input_thread_t * p_input,
 465                                       es_descriptor_t * p_es )
 466 {
 467     vdec_config_t *     p_config;
 468
 469     p_config = (vdec_config_t *)malloc( sizeof(vdec_config_t) );
 470     if( p_config == NULL )
 471     {
 472         intf_ErrMsg( "Unable to allocate memory in GetVdecConfig" );
 473         return( NULL );
 474     }
 475     p_config->p_vout = p_input->p_default_vout;
 476     if( InitDecConfig( p_input, p_es, &p_config->decoder_config ) == -1 )
 477     {
 478         free( p_config );
 479         return( NULL );
 480     }
 481
 482     return( p_config );
 483 }
 484
 485 /*****************************************************************************
 486  * GetAdecConfig: returns a valid adec_config_t
 487  *****************************************************************************/
 488 static adec_config_t * GetAdecConfig( input_thread_t * p_input,
 489                                       es_descriptor_t * p_es )
 490 {
 491     adec_config_t *     p_config;
 492
 493     p_config = (adec_config_t *)malloc( sizeof(adec_config_t));
 494     if( p_config == NULL )
 495     {
 496         intf_ErrMsg( "Unable to allocate memory in GetAdecConfig" );
 497         return( NULL );
 498     }
 499     p_config->p_aout = p_input->p_default_aout;
 500     if( InitDecConfig( p_input, p_es, &p_config->decoder_config ) == -1 )
 501     {
 502         free( p_config );
 503         return( NULL );
 504     }
 505
 506     return( p_config );
 507 }
 508
 509 /*****************************************************************************
 510  * input_SelectES: selects an ES and spawns the associated decoder
 511  *****************************************************************************/
 512 /* FIXME */
 513 vlc_thread_t adec_CreateThread( void * );
 514 vlc_thread_t ac3dec_CreateThread( void * );
 515 vlc_thread_t vpar_CreateThread( void * );
 516 vlc_thread_t spudec_CreateThread( void * );
 517
 518 int input_SelectES( input_thread_t * p_input, es_descriptor_t * p_es )
 519 {
 520     /* FIXME ! */
 521     decoder_capabilities_t  decoder;
 522
 523 #ifdef DEBUG_INPUT
 524     intf_DbgMsg( "Selecting ES 0x%x", p_es->i_id );
 525 #endif
 526
 527     if( p_es->p_decoder_fifo != NULL )
 528     {
 529         intf_ErrMsg( "ES %d is already selected", p_es->i_id );
 530         return( -1 );
 531     }
 532
 533     switch( p_es->i_type )
 534     {
 535     case MPEG1_AUDIO_ES:
 536     case MPEG2_AUDIO_ES:
 537         if( p_main->b_audio )
 538         {
 539             decoder.pf_create_thread = adec_CreateThread;
 540             p_es->thread_id = input_RunDecoder( &decoder,
 541                                     (void *)GetAdecConfig( p_input, p_es ) );
 542         }
 543         break;
 544
 545     case MPEG1_VIDEO_ES:
 546     case MPEG2_VIDEO_ES:
 547         if( p_main->b_video )
 548         {
 549             decoder.pf_create_thread = vpar_CreateThread;
 550             p_es->thread_id = input_RunDecoder( &decoder,
 551                                     (void *)GetVdecConfig( p_input, p_es ) );
 552         }
 553         break;
 554
 555     case AC3_AUDIO_ES:
 556         if( p_main->b_audio )
 557         {
 558             decoder.pf_create_thread = ac3dec_CreateThread;
 559             p_es->thread_id = input_RunDecoder( &decoder,
 560                                     (void *)GetAdecConfig( p_input, p_es ) );
 561         }
 562         break;
 563
 564     case DVD_SPU_ES:
 565         if( p_main->b_video )
 566         {
 567             decoder.pf_create_thread = spudec_CreateThread;
 568             p_es->thread_id = input_RunDecoder( &decoder,
 569                                     (void *)GetVdecConfig( p_input, p_es ) );
 570         }
 571         break;
 572
 573     default:
 574         intf_ErrMsg( "Unknown stream type %d", p_es->i_type );
 575         return( -1 );
 576         break;
 577     }
 578
 579     if( p_es->thread_id == 0 )
 580     {
 581         return( -1 );
 582     }
 583
 584     if( p_es->p_decoder_fifo != NULL )
 585     {
 586         p_input->stream.i_selected_es_number++;
 587         p_input->stream.pp_selected_es = realloc(
 588                                            p_input->stream.pp_selected_es,
 589                                            p_input->stream.i_selected_es_number
 590                                             * sizeof(es_descriptor_t *) );
 591         if( p_input->stream.pp_selected_es == NULL )
 592         {
 593             intf_ErrMsg( "Unable to realloc memory in input_SelectES" );
 594             return(-1);
 595         }
 596         p_input->stream.pp_selected_es[p_input->stream.i_selected_es_number - 1]
 597                 = p_es;
 598     }
 599     return( 0 );
 600 }