es_out.c 64.8 KB
Newer Older
1 2 3
/*****************************************************************************
 * es_out.c: Es Out handler for input.
 *****************************************************************************
4
 * Copyright (C) 2003-2004 the VideoLAN team
5 6 7
 * $Id$
 *
 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8
 *          Jean-Paul Saman <jpsaman #_at_# m2x dot nl>
9 10 11 12 13 14 15 16 17 18 19 20 21
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
Antoine Cellerier's avatar
Antoine Cellerier committed
22
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23 24 25 26 27
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
28 29 30 31
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif

32
#include <stdio.h>
33 34
#include <assert.h>
#include <vlc_common.h>
35

Clément Stenac's avatar
Clément Stenac committed
36 37 38
#include <vlc_input.h>
#include <vlc_es_out.h>
#include <vlc_block.h>
39
#include <vlc_aout.h>
40 41 42

#include "input_internal.h"

43 44
#include "../stream_output/stream_output.h"

45
#include <vlc_iso_lang.h>
46
/* FIXME we should find a better way than including that */
47
#include "../text/iso-639_def.h"
48 49 50 51 52 53 54 55 56 57 58 59

/*****************************************************************************
 * Local prototypes
 *****************************************************************************/
typedef struct
{
    /* Program ID */
    int i_id;

    /* Number of es for this pgrm */
    int i_es;

60
    bool b_selected;
61 62 63 64

    /* Clock for this program */
    input_clock_t clock;

65
    char    *psz_name;
66
    char    *psz_now_playing;
67
    char    *psz_publisher;
68

69
    vlc_epg_t *p_epg;
70 71 72 73 74 75 76 77
} es_out_pgrm_t;

struct es_out_id_t
{
    /* ES ID */
    int       i_id;
    es_out_pgrm_t *p_pgrm;

78
    /* Misc. */
79 80 81 82 83 84 85
    int64_t i_preroll_end;

    /* Channel in the track type */
    int         i_channel;
    es_format_t fmt;
    char        *psz_language;
    char        *psz_language_code;
86

87
    decoder_t   *p_dec;
88
    decoder_t   *p_dec_record;
89 90

    /* Fields for Video with CC */
91
    bool  pb_cc_present[4];
92 93 94 95
    es_out_id_t  *pp_cc_es[4];

    /* Field for CC track from a master video */
    es_out_id_t *p_master;
96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113
};

struct es_out_sys_t
{
    input_thread_t *p_input;

    /* all programs */
    int           i_pgrm;
    es_out_pgrm_t **pgrm;
    es_out_pgrm_t **pp_selected_pgrm; /* --programs */
    es_out_pgrm_t *p_pgrm;  /* Master program */

    /* all es */
    int         i_id;
    int         i_es;
    es_out_id_t **es;

    /* mode gestion */
114
    bool  b_active;
115 116 117 118 119 120 121 122
    int         i_mode;

    /* es count */
    int         i_audio;
    int         i_video;
    int         i_sub;

    /* es to select */
123 124
    int         i_audio_last, i_audio_id;
    int         i_sub_last, i_sub_id;
125
    int         i_default_sub_id;   /* As specified in container; if applicable */
126 127 128 129 130 131 132 133 134 135 136
    char        **ppsz_audio_language;
    char        **ppsz_sub_language;

    /* current main es */
    es_out_id_t *p_es_audio;
    es_out_id_t *p_es_video;
    es_out_id_t *p_es_sub;

    /* delay */
    int64_t i_audio_delay;
    int64_t i_spu_delay;
137 138 139

    /* Rate used to rescale ES ts */
    int         i_rate;
140 141 142

    /* Record */
    sout_instance_t *p_sout_record;
143 144 145 146 147
};

static es_out_id_t *EsOutAdd    ( es_out_t *, es_format_t * );
static int          EsOutSend   ( es_out_t *, es_out_id_t *, block_t * );
static void         EsOutDel    ( es_out_t *, es_out_id_t * );
148
static void         EsOutSelect( es_out_t *out, es_out_id_t *es, bool b_force );
149 150 151 152
static int          EsOutControl( es_out_t *, int i_query, va_list );

static void         EsOutAddInfo( es_out_t *, es_out_id_t *es );

153
static bool EsIsSelected( es_out_id_t *es );
154
static void EsSelect( es_out_t *out, es_out_id_t *es );
155
static void EsUnselect( es_out_t *out, es_out_id_t *es, bool b_update );
156 157 158 159 160
static char *LanguageGetName( const char *psz_code );
static char *LanguageGetCode( const char *psz_lang );
static char **LanguageSplit( const char *psz_langs );
static int LanguageArrayIndex( char **ppsz_langs, char *psz_lang );

161 162
static char *EsOutProgramGetMetaName( es_out_pgrm_t *p_pgrm );

163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180
static const vlc_fourcc_t EsOutFourccClosedCaptions[4] = {
    VLC_FOURCC('c', 'c', '1', ' '),
    VLC_FOURCC('c', 'c', '2', ' '),
    VLC_FOURCC('c', 'c', '3', ' '),
    VLC_FOURCC('c', 'c', '4', ' '),
};
static inline int EsOutGetClosedCaptionsChannel( vlc_fourcc_t fcc )
{
    int i;
    for( i = 0; i < 4; i++ )
    {
        if( fcc == EsOutFourccClosedCaptions[i] )
            return i;
    }
    return -1;
}


181 182 183
/*****************************************************************************
 * input_EsOutNew:
 *****************************************************************************/
184
es_out_t *input_EsOutNew( input_thread_t *p_input, int i_rate )
185 186 187 188
{
    vlc_value_t  val;
    int i;

189
    es_out_t     *out = malloc( sizeof( es_out_t ) );
190
    if( !out ) return NULL;
191 192

    es_out_sys_t *p_sys = malloc( sizeof( es_out_sys_t ) );
193 194 195 196 197 198
    if( !p_sys )
    {
        free( out );
        return NULL;
    }

199 200 201 202 203
    out->pf_add     = EsOutAdd;
    out->pf_send    = EsOutSend;
    out->pf_del     = EsOutDel;
    out->pf_control = EsOutControl;
    out->p_sys      = p_sys;
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
204
    out->b_sout     = p_input->p->p_sout != NULL;
205 206 207

    p_sys->p_input = p_input;

208
    p_sys->b_active = false;
209 210 211
    p_sys->i_mode   = ES_OUT_MODE_AUTO;


212
    TAB_INIT( p_sys->i_pgrm, p_sys->pgrm );
213 214 215
    p_sys->p_pgrm   = NULL;

    p_sys->i_id    = 0;
216 217

    TAB_INIT( p_sys->i_es, p_sys->es );
218 219 220 221 222 223 224 225 226 227 228 229

    p_sys->i_audio = 0;
    p_sys->i_video = 0;
    p_sys->i_sub   = 0;

    /* */
    var_Get( p_input, "audio-track", &val );
    p_sys->i_audio_last = val.i_int;

    var_Get( p_input, "sub-track", &val );
    p_sys->i_sub_last = val.i_int;

230 231
    p_sys->i_default_sub_id   = -1;

232
    if( !p_input->b_preparsing )
233
    {
234 235 236 237 238 239 240 241
        var_Get( p_input, "audio-language", &val );
        p_sys->ppsz_audio_language = LanguageSplit(val.psz_string);
        if( p_sys->ppsz_audio_language )
        {
            for( i = 0; p_sys->ppsz_audio_language[i]; i++ )
                msg_Dbg( p_input, "selected audio language[%d] %s",
                         i, p_sys->ppsz_audio_language[i] );
        }
242
        free( val.psz_string );
243 244 245 246 247 248 249 250 251

        var_Get( p_input, "sub-language", &val );
        p_sys->ppsz_sub_language = LanguageSplit(val.psz_string);
        if( p_sys->ppsz_sub_language )
        {
            for( i = 0; p_sys->ppsz_sub_language[i]; i++ )
                msg_Dbg( p_input, "selected subtitle language[%d] %s",
                         i, p_sys->ppsz_sub_language[i] );
        }
252
        free( val.psz_string );
253
    }
254
    else
255
    {
256 257
        p_sys->ppsz_sub_language = NULL;
        p_sys->ppsz_audio_language = NULL;
258 259
    }

260 261 262 263 264 265
    var_Get( p_input, "audio-track-id", &val );
    p_sys->i_audio_id = val.i_int;

    var_Get( p_input, "sub-track-id", &val );
    p_sys->i_sub_id = val.i_int;

266 267 268 269 270 271 272
    p_sys->p_es_audio = NULL;
    p_sys->p_es_video = NULL;
    p_sys->p_es_sub   = NULL;

    p_sys->i_audio_delay= 0;
    p_sys->i_spu_delay  = 0;

273 274
    p_sys->i_rate = i_rate;

275 276
    p_sys->p_sout_record = NULL;

277 278 279 280 281 282 283 284 285 286 287
    return out;
}

/*****************************************************************************
 * input_EsOutDelete:
 *****************************************************************************/
void input_EsOutDelete( es_out_t *out )
{
    es_out_sys_t *p_sys = out->p_sys;
    int i;

288 289 290
    if( p_sys->p_sout_record )
        input_EsOutSetRecord( out, false );

291 292 293 294
    for( i = 0; i < p_sys->i_es; i++ )
    {
        if( p_sys->es[i]->p_dec )
            input_DecoderDelete( p_sys->es[i]->p_dec );
295

296 297
        free( p_sys->es[i]->psz_language );
        free( p_sys->es[i]->psz_language_code );
298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313
        es_format_Clean( &p_sys->es[i]->fmt );

        free( p_sys->es[i] );
    }
    if( p_sys->ppsz_audio_language )
    {
        for( i = 0; p_sys->ppsz_audio_language[i]; i++ )
            free( p_sys->ppsz_audio_language[i] );
        free( p_sys->ppsz_audio_language );
    }
    if( p_sys->ppsz_sub_language )
    {
        for( i = 0; p_sys->ppsz_sub_language[i]; i++ )
            free( p_sys->ppsz_sub_language[i] );
        free( p_sys->ppsz_sub_language );
    }
314
    free( p_sys->es );
315

316
    /* FIXME duplicate work EsOutProgramDel (but we cannot use it) add a EsOutProgramClean ? */
317 318
    for( i = 0; i < p_sys->i_pgrm; i++ )
    {
319
        es_out_pgrm_t *p_pgrm = p_sys->pgrm[i];
320 321 322
        free( p_pgrm->psz_now_playing );
        free( p_pgrm->psz_publisher );
        free( p_pgrm->psz_name );
323 324 325 326
        if( p_pgrm->p_epg )
            vlc_epg_Delete( p_pgrm->p_epg );

        free( p_pgrm );
327
    }
328
    TAB_CLEAN( p_sys->i_pgrm, p_sys->pgrm );
329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350

    free( p_sys );
    free( out );
}

es_out_id_t *input_EsOutGetFromID( es_out_t *out, int i_id )
{
    int i;
    if( i_id < 0 )
    {
        /* Special HACK, -i_id is tha cat of the stream */
        return (es_out_id_t*)((uint8_t*)NULL-i_id);
    }

    for( i = 0; i < out->p_sys->i_es; i++ )
    {
        if( out->p_sys->es[i]->i_id == i_id )
            return out->p_sys->es[i];
    }
    return NULL;
}

351
static void EsOutDiscontinuity( es_out_t *out, bool b_flush, bool b_audio )
352 353 354 355 356 357 358
{
    es_out_sys_t      *p_sys = out->p_sys;
    int i;

    for( i = 0; i < p_sys->i_es; i++ )
    {
        es_out_id_t *es = p_sys->es[i];
359

360 361 362
        /* Send a dummy block to let decoder know that
         * there is a discontinuity */
        if( es->p_dec && ( !b_audio || es->fmt.i_cat == AUDIO_ES ) )
363
        {
364
            input_DecoderDiscontinuity( es->p_dec, b_flush );
365 366 367
            if( es->p_dec_record )
                input_DecoderDiscontinuity( es->p_dec_record, b_flush );
        }
368 369
    }
}
370
void input_EsOutChangeRate( es_out_t *out, int i_rate )
371 372 373 374
{
    es_out_sys_t      *p_sys = out->p_sys;
    int i;

375
    p_sys->i_rate = i_rate;
376
    EsOutDiscontinuity( out, false, false );
377

378
    for( i = 0; i < p_sys->i_pgrm; i++ )
379
        input_ClockSetRate( &p_sys->pgrm[i]->clock, i_rate );
380
}
381

382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414
int input_EsOutSetRecord(  es_out_t *out, bool b_record )
{
    es_out_sys_t   *p_sys = out->p_sys;
    input_thread_t *p_input = p_sys->p_input;

    assert( ( b_record && !p_sys->p_sout_record ) || ( !b_record && p_sys->p_sout_record ) );

    if( b_record )
    {
        char *psz_path = var_CreateGetString( p_input, "input-record-path" );
        if( !psz_path || *psz_path == '\0' )
        {
            free( psz_path );
            psz_path = strdup( config_GetHomeDir() );
        }

        char *psz_sout = NULL;  // TODO conf

        if( !psz_sout && psz_path )
        {
            char *psz_file = input_CreateFilename( VLC_OBJECT(p_input), psz_path, INPUT_RECORD_PREFIX, NULL );
            if( psz_file )
            {
                if( asprintf( &psz_sout, "#record{dst-prefix='%s'}", psz_file ) < 0 )
                    psz_sout = NULL;
                free( psz_file );
            }
        }
        free( psz_path );

        if( !psz_sout )
            return VLC_EGENERIC;

415
#ifdef ENABLE_SOUT
416
        p_sys->p_sout_record = sout_NewInstance( p_input, psz_sout );
417
#endif
418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444
        free( psz_sout );

        if( !p_sys->p_sout_record )
            return VLC_EGENERIC;

        for( int i = 0; i < p_sys->i_es; i++ )
        {
            es_out_id_t *p_es = p_sys->es[i];

            if( !p_es->p_dec || p_es->p_master )
                continue;

            p_es->p_dec_record = input_DecoderNew( p_input, &p_es->fmt, p_sys->p_sout_record );
        }
    }
    else
    {
        for( int i = 0; i < p_sys->i_es; i++ )
        {
            es_out_id_t *p_es = p_sys->es[i];

            if( !p_es->p_dec_record )
                continue;

            input_DecoderDelete( p_es->p_dec_record );
            p_es->p_dec_record = NULL;
        }
445
#ifdef ENABLE_SOUT
446
        sout_DeleteInstance( p_sys->p_sout_record );
447
#endif
448 449 450 451 452
        p_sys->p_sout_record = NULL;
    }

    return VLC_SUCCESS;
}
453 454 455 456 457 458 459 460 461
void input_EsOutSetDelay( es_out_t *out, int i_cat, int64_t i_delay )
{
    es_out_sys_t *p_sys = out->p_sys;

    if( i_cat == AUDIO_ES )
        p_sys->i_audio_delay = i_delay;
    else if( i_cat == SPU_ES )
        p_sys->i_spu_delay = i_delay;
}
462 463 464 465 466 467 468 469 470
void input_EsOutChangeState( es_out_t *out )
{
    es_out_sys_t *p_sys = out->p_sys;
    input_thread_t *p_input = p_sys->p_input;

    if( p_input->i_state  == PAUSE_S )
    {
        /* Send discontinuity to decoders (it will allow them to flush
         *                  * if implemented */
471
        EsOutDiscontinuity( out, false, false );
472 473 474 475 476 477 478
    }
    else
    {
        /* Out of pause, reset pcr */
        es_out_Control( out, ES_OUT_RESET_PCR );
    }
}
Laurent Aimar's avatar
Laurent Aimar committed
479 480 481 482 483
void input_EsOutChangePosition( es_out_t *out )
{
    //es_out_sys_t *p_sys = out->p_sys;

    es_out_Control( out, ES_OUT_RESET_PCR );
484
    EsOutDiscontinuity( out, true, false );
Laurent Aimar's avatar
Laurent Aimar committed
485
}
486

487
bool input_EsOutDecodersEmpty( es_out_t *out )
488 489 490 491 492 493 494 495 496
{
    es_out_sys_t      *p_sys = out->p_sys;
    int i;

    for( i = 0; i < p_sys->i_es; i++ )
    {
        es_out_id_t *es = p_sys->es[i];

        if( es->p_dec && !input_DecoderEmpty( es->p_dec ) )
497
            return false;
498 499
        if( es->p_dec_record && !input_DecoderEmpty( es->p_dec_record ) )
            return false;
500
    }
501
    return true;
502 503 504 505 506
}

/*****************************************************************************
 *
 *****************************************************************************/
507
static void EsOutESVarUpdateGeneric( es_out_t *out, int i_id, es_format_t *fmt, const char *psz_language,
508
                                     bool b_delete )
509 510 511
{
    es_out_sys_t      *p_sys = out->p_sys;
    input_thread_t    *p_input = p_sys->p_input;
512
    const  bool b_teletext = fmt->i_cat == SPU_ES && fmt->i_codec == VLC_FOURCC( 't', 'e', 'l', 'x' );
513 514
    vlc_value_t       val, text;

Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
515
    const char *psz_var;
516

517
    if( fmt->i_cat == AUDIO_ES )
518
        psz_var = "audio-es";
519
    else if( fmt->i_cat == VIDEO_ES )
520
        psz_var = "video-es";
521
    else if( fmt->i_cat == SPU_ES )
522 523 524 525 526 527
        psz_var = "spu-es";
    else
        return;

    if( b_delete )
    {
528 529 530
        if( b_teletext )
            var_SetInteger( p_sys->p_input, "teletext-es", -1 );

531
        val.i_int = i_id;
532
        var_Change( p_input, psz_var, VLC_VAR_DELCHOICE, &val, NULL );
533

534
        var_SetBool( p_sys->p_input, "intf-change", true );
535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550
        return;
    }

    /* Get the number of ES already added */
    var_Change( p_input, psz_var, VLC_VAR_CHOICESCOUNT, &val, NULL );
    if( val.i_int == 0 )
    {
        vlc_value_t val2;

        /* First one, we need to add the "Disable" choice */
        val2.i_int = -1; text.psz_string = _("Disable");
        var_Change( p_input, psz_var, VLC_VAR_ADDCHOICE, &val2, &text );
        val.i_int++;
    }

    /* Take care of the ES description */
551
    if( fmt->psz_description && *fmt->psz_description )
552
    {
553
        if( psz_language && *psz_language )
554
        {
555 556 557 558
            text.psz_string = malloc( strlen( fmt->psz_description) +
                                      strlen( psz_language ) + 10 );
            sprintf( text.psz_string, "%s - [%s]", fmt->psz_description,
                                                   psz_language );
559
        }
560
        else text.psz_string = strdup( fmt->psz_description );
561 562 563
    }
    else
    {
564
        if( psz_language && *psz_language )
565
        {
566 567
            if( asprintf( &text.psz_string, "%s %i - [%s]", _( "Track" ), val.i_int, psz_language ) == -1 )
                text.psz_string = NULL;
568 569 570
        }
        else
        {
571 572
            if( asprintf( &text.psz_string, "%s %i", _( "Track" ), val.i_int ) == -1 )
                text.psz_string = NULL;
573 574 575
        }
    }

576
    val.i_int = i_id;
577 578 579 580
    var_Change( p_input, psz_var, VLC_VAR_ADDCHOICE, &val, &text );

    free( text.psz_string );

581 582 583
    if( b_teletext )
        var_SetInteger( p_sys->p_input, "teletext-es", i_id );

584
    var_SetBool( p_sys->p_input, "intf-change", true );
585 586
}

587
static void EsOutESVarUpdate( es_out_t *out, es_out_id_t *es,
588
                              bool b_delete )
589 590 591 592
{
    EsOutESVarUpdateGeneric( out, es->i_id, &es->fmt, es->psz_language, b_delete );
}

593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608
/* EsOutProgramSelect:
 *  Select a program and update the object variable
 */
static void EsOutProgramSelect( es_out_t *out, es_out_pgrm_t *p_pgrm )
{
    es_out_sys_t      *p_sys = out->p_sys;
    input_thread_t    *p_input = p_sys->p_input;
    vlc_value_t       val;
    int               i;

    if( p_sys->p_pgrm == p_pgrm )
        return; /* Nothing to do */

    if( p_sys->p_pgrm )
    {
        es_out_pgrm_t *old = p_sys->p_pgrm;
Clément Stenac's avatar
Clément Stenac committed
609
        msg_Dbg( p_input, "unselecting program id=%d", old->i_id );
610 611 612

        for( i = 0; i < p_sys->i_es; i++ )
        {
613
            if( p_sys->es[i]->p_pgrm == old && EsIsSelected( p_sys->es[i] ) &&
614
                p_sys->i_mode != ES_OUT_MODE_ALL )
615
                EsUnselect( out, p_sys->es[i], true );
616 617 618 619 620 621 622
        }

        p_sys->p_es_audio = NULL;
        p_sys->p_es_sub = NULL;
        p_sys->p_es_video = NULL;
    }

Clément Stenac's avatar
Clément Stenac committed
623
    msg_Dbg( p_input, "selecting program id=%d", p_pgrm->i_id );
624 625

    /* Mark it selected */
626
    p_pgrm->b_selected = true;
627 628 629 630

    /* Switch master stream */
    if( p_sys->p_pgrm && p_sys->p_pgrm->clock.b_master )
    {
631
        p_sys->p_pgrm->clock.b_master = false;
632
    }
633
    p_pgrm->clock.b_master = true;
634 635 636 637 638 639 640 641 642 643
    p_sys->p_pgrm = p_pgrm;

    /* Update "program" */
    val.i_int = p_pgrm->i_id;
    var_Change( p_input, "program", VLC_VAR_SETVALUE, &val, NULL );

    /* Update "es-*" */
    var_Change( p_input, "audio-es", VLC_VAR_CLEARCHOICES, NULL, NULL );
    var_Change( p_input, "video-es", VLC_VAR_CLEARCHOICES, NULL, NULL );
    var_Change( p_input, "spu-es",   VLC_VAR_CLEARCHOICES, NULL, NULL );
644
    var_SetInteger( p_input, "teletext-es", -1 );
645 646 647
    for( i = 0; i < p_sys->i_es; i++ )
    {
        if( p_sys->es[i]->p_pgrm == p_sys->p_pgrm )
648 649
            EsOutESVarUpdate( out, p_sys->es[i], false );
        EsOutSelect( out, p_sys->es[i], false );
650 651
    }

652
    /* Update now playing */
653 654 655 656
    input_item_SetNowPlaying( p_input->p->input.p_item,
                              p_pgrm->psz_now_playing );
    input_item_SetPublisher( p_input->p->input.p_item,
                             p_pgrm->psz_publisher );
657

658
    var_SetBool( p_sys->p_input, "intf-change", true );
659 660 661 662 663 664 665 666 667 668 669 670
}

/* EsOutAddProgram:
 *  Add a program
 */
static es_out_pgrm_t *EsOutProgramAdd( es_out_t *out, int i_group )
{
    es_out_sys_t      *p_sys = out->p_sys;
    input_thread_t    *p_input = p_sys->p_input;
    vlc_value_t       val;

    es_out_pgrm_t *p_pgrm = malloc( sizeof( es_out_pgrm_t ) );
671
    if( !p_pgrm ) return NULL;
672 673 674 675

    /* Init */
    p_pgrm->i_id = i_group;
    p_pgrm->i_es = 0;
676
    p_pgrm->b_selected = false;
677
    p_pgrm->psz_name = NULL;
678
    p_pgrm->psz_now_playing = NULL;
679 680
    p_pgrm->psz_publisher = NULL;
    p_pgrm->p_epg = NULL;
681
    input_ClockInit( &p_pgrm->clock, false, p_input->p->input.i_cr_average, p_sys->i_rate );
682 683 684 685 686 687 688 689 690 691 692 693 694 695

    /* Append it */
    TAB_APPEND( p_sys->i_pgrm, p_sys->pgrm, p_pgrm );

    /* Update "program" variable */
    val.i_int = i_group;
    var_Change( p_input, "program", VLC_VAR_ADDCHOICE, &val, NULL );

    if( i_group == var_GetInteger( p_input, "program" ) )
    {
        EsOutProgramSelect( out, p_pgrm );
    }
    else
    {
696
        var_SetBool( p_sys->p_input, "intf-change", true );
697 698 699 700
    }
    return p_pgrm;
}

701 702 703
/* EsOutDelProgram:
 *  Delete a program
 */
704
static int EsOutProgramDel( es_out_t *out, int i_group )
705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720
{
    es_out_sys_t      *p_sys = out->p_sys;
    input_thread_t    *p_input = p_sys->p_input;
    es_out_pgrm_t     *p_pgrm = NULL;
    vlc_value_t       val;
    int               i;

    for( i = 0; i < p_sys->i_pgrm; i++ )
    {
        if( p_sys->pgrm[i]->i_id == i_group )
        {
            p_pgrm = p_sys->pgrm[i];
            break;
        }
    }

721 722
    if( p_pgrm == NULL )
        return VLC_EGENERIC;
723 724 725 726 727

    if( p_pgrm->i_es )
    {
        msg_Dbg( p_input, "can't delete program %d which still has %i ES",
                 i_group, p_pgrm->i_es );
728
        return VLC_EGENERIC;
729 730 731 732 733
    }

    TAB_REMOVE( p_sys->i_pgrm, p_sys->pgrm, p_pgrm );

    /* If program is selected we need to unselect it */
734
    if( p_sys->p_pgrm == p_pgrm ) p_sys->p_pgrm = NULL;
735

736 737 738
    free( p_pgrm->psz_name );
    free( p_pgrm->psz_now_playing );
    free( p_pgrm->psz_publisher );
739 740
    if( p_pgrm->p_epg )
        vlc_epg_Delete( p_pgrm->p_epg );
741 742 743 744 745 746
    free( p_pgrm );

    /* Update "program" variable */
    val.i_int = i_group;
    var_Change( p_input, "program", VLC_VAR_DELCHOICE, &val, NULL );

747
    var_SetBool( p_sys->p_input, "intf-change", true );
748 749

    return VLC_SUCCESS;
750 751
}

752 753
/* EsOutProgramMeta:
 */
754 755 756 757
static char *EsOutProgramGetMetaName( es_out_pgrm_t *p_pgrm )
{
    char *psz = NULL;
    if( p_pgrm->psz_name )
758 759
    {
        if( asprintf( &psz, _("%s [%s %d]"), p_pgrm->psz_name, _("Program"), p_pgrm->i_id ) == -1 )
Rémi Duraffort's avatar
Rémi Duraffort committed
760
            return NULL;
761
    }
762
    else
763 764
    {
        if( asprintf( &psz, "%s %d", _("Program"), p_pgrm->i_id ) == -1 )
Rémi Duraffort's avatar
Rémi Duraffort committed
765
            return NULL;
766
    }
767 768 769
    return psz;
}

770 771 772
static void EsOutProgramMeta( es_out_t *out, int i_group, vlc_meta_t *p_meta )
{
    es_out_sys_t      *p_sys = out->p_sys;
773
    es_out_pgrm_t     *p_pgrm = NULL;
774
    input_thread_t    *p_input = p_sys->p_input;
775
    char              *psz_cat;
776 777
    const char        *psz_title = NULL;
    const char        *psz_provider = NULL;
778 779 780
    int i;

    msg_Dbg( p_input, "EsOutProgramMeta: number=%d", i_group );
781

782
    /* Check against empty meta data (empty for what we handle) */
783 784 785 786
    if( !vlc_meta_Get( p_meta, vlc_meta_Title) &&
        !vlc_meta_Get( p_meta, vlc_meta_NowPlaying) &&
        !vlc_meta_Get( p_meta, vlc_meta_Publisher) &&
        vlc_dictionary_keys_count( &p_meta->extra_tags ) <= 0 )
787 788 789
    {
        return;
    }
790
    /* Find program */
791 792 793 794 795 796 797 798
    for( i = 0; i < p_sys->i_pgrm; i++ )
    {
        if( p_sys->pgrm[i]->i_id == i_group )
        {
            p_pgrm = p_sys->pgrm[i];
            break;
        }
    }
799 800
    if( p_pgrm == NULL )
        p_pgrm = EsOutProgramAdd( out, i_group );   /* Create it */
801

802
    /* */
803 804
    psz_title = vlc_meta_Get( p_meta, vlc_meta_Title);
    psz_provider = vlc_meta_Get( p_meta, vlc_meta_Publisher);
805 806

    /* Update the description text of the program */
Clément Stenac's avatar
Clément Stenac committed
807
    if( psz_title && *psz_title )
808 809 810 811
    {
        vlc_value_t val;
        vlc_value_t text;

812 813 814 815 816 817 818 819 820
        if( !p_pgrm->psz_name || strcmp( p_pgrm->psz_name, psz_title ) )
        {
            char *psz_cat = EsOutProgramGetMetaName( p_pgrm );

            /* Remove old entries */
            input_Control( p_input, INPUT_DEL_INFO, psz_cat, NULL );
            /* TODO update epg name */
            free( psz_cat );
        }
821
        free( p_pgrm->psz_name );
822 823
        p_pgrm->psz_name = strdup( psz_title );

824 825 826 827 828 829
        /* ugly but it works */
        val.i_int = i_group;
        var_Change( p_input, "program", VLC_VAR_DELCHOICE, &val, NULL );

        if( psz_provider && *psz_provider )
        {
830 831 832 833 834
            if( asprintf( &text.psz_string, "%s [%s]", psz_title, psz_provider ) != -1 )
            {
                var_Change( p_input, "program", VLC_VAR_ADDCHOICE, &val, &text );
                free( text.psz_string );
            }
835 836 837
        }
        else
        {
838
            text.psz_string = (char *)psz_title;
839 840 841 842
            var_Change( p_input, "program", VLC_VAR_ADDCHOICE, &val, &text );
        }
    }

843 844 845
    psz_cat = EsOutProgramGetMetaName( p_pgrm );
    if( psz_provider )
    {
846
        if( p_sys->p_pgrm == p_pgrm )
847 848
            input_item_SetPublisher( p_input->p->input.p_item, psz_provider );
        input_Control( p_input, INPUT_ADD_INFO, psz_cat, input_MetaTypeToLocalizedString(vlc_meta_Publisher), psz_provider );
849
    }
850 851 852 853 854 855 856 857
    char ** ppsz_all_keys = vlc_dictionary_all_keys( &p_meta->extra_tags );
    for( i = 0; ppsz_all_keys[i]; i++ )
    {
        input_Control( p_input, INPUT_ADD_INFO, psz_cat, _(ppsz_all_keys[i]),
                       vlc_dictionary_value_for_key( &p_meta->extra_tags, ppsz_all_keys[i] ) );
        free( ppsz_all_keys[i] );
    }
    free( ppsz_all_keys );
858 859 860 861 862 863 864 865 866 867 868 869

    free( psz_cat );
}

static void vlc_epg_Merge( vlc_epg_t *p_dst, const vlc_epg_t *p_src )
{
    int i;

    /* Add new event */
    for( i = 0; i < p_src->i_event; i++ )
    {
        vlc_epg_event_t *p_evt = p_src->pp_event[i];
870
        bool b_add = true;
871 872 873
        int j;

        for( j = 0; j < p_dst->i_event; j++ )
874
        {
875 876
            if( p_dst->pp_event[j]->i_start == p_evt->i_start && p_dst->pp_event[j]->i_duration == p_evt->i_duration )
            {
877
                b_add = false;
878 879 880 881
                break;
            }
            if( p_dst->pp_event[j]->i_start > p_evt->i_start )
                break;
882
        }
883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935
        if( b_add )
        {
            vlc_epg_event_t *p_copy = malloc( sizeof(vlc_epg_event_t) );
            if( !p_copy )
                break;
            memset( p_copy, 0, sizeof(vlc_epg_event_t) );
            p_copy->i_start = p_evt->i_start;
            p_copy->i_duration = p_evt->i_duration;
            p_copy->psz_name = p_evt->psz_name ? strdup( p_evt->psz_name ) : NULL;
            p_copy->psz_short_description = p_evt->psz_short_description ? strdup( p_evt->psz_short_description ) : NULL;
            p_copy->psz_description = p_evt->psz_description ? strdup( p_evt->psz_description ) : NULL;
            TAB_INSERT( p_dst->i_event, p_dst->pp_event, p_copy, j );
        }
    }
    /* Update current */
    vlc_epg_SetCurrent( p_dst, p_src->p_current ? p_src->p_current->i_start : -1 );

    /* Keep only 1 old event  */
    if( p_dst->p_current )
    {
        while( p_dst->i_event > 1 && p_dst->pp_event[0] != p_dst->p_current && p_dst->pp_event[1] != p_dst->p_current )
            TAB_REMOVE( p_dst->i_event, p_dst->pp_event, p_dst->pp_event[0] );
    }
}

static void EsOutProgramEpg( es_out_t *out, int i_group, vlc_epg_t *p_epg )
{
    es_out_sys_t      *p_sys = out->p_sys;
    input_thread_t    *p_input = p_sys->p_input;
    es_out_pgrm_t     *p_pgrm = NULL;
    char *psz_cat;
    int i;

    /* Find program */
    for( i = 0; i < p_sys->i_pgrm; i++ )
    {
        if( p_sys->pgrm[i]->i_id == i_group )
        {
            p_pgrm = p_sys->pgrm[i];
            break;
        }
    }
    if( p_pgrm == NULL )
        p_pgrm = EsOutProgramAdd( out, i_group );   /* Create it */

    /* Merge EPG */
    if( !p_pgrm->p_epg )
        p_pgrm->p_epg = vlc_epg_New( p_pgrm->psz_name );
    vlc_epg_Merge( p_pgrm->p_epg, p_epg );

    /* Update info */
    psz_cat = EsOutProgramGetMetaName( p_pgrm );
#ifdef HAVE_LOCALTIME_R
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
936 937 938
    char *psz_epg;
    if( asprintf( &psz_epg, "EPG %s", psz_cat ) == -1 )
        psz_epg = NULL;
939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963
    input_Control( p_input, INPUT_DEL_INFO, psz_epg, NULL );
    msg_Dbg( p_input, "EsOutProgramEpg: number=%d name=%s", i_group, p_pgrm->p_epg->psz_name );
    for( i = 0; i < p_pgrm->p_epg->i_event; i++ )
    {
        const vlc_epg_event_t *p_evt = p_pgrm->p_epg->pp_event[i];
        time_t t_start = (time_t)p_evt->i_start;
        struct tm tm_start;
        char psz_start[128];

        localtime_r( &t_start, &tm_start );

        snprintf( psz_start, sizeof(psz_start), "%2.2d:%2.2d:%2.2d", tm_start.tm_hour, tm_start.tm_min, tm_start.tm_sec );
        if( p_evt->psz_short_description || p_evt->psz_description )
            input_Control( p_input, INPUT_ADD_INFO, psz_epg, psz_start, "%s (%2.2d:%2.2d) - %s",
                           p_evt->psz_name,
                           p_evt->i_duration/60/60, (p_evt->i_duration/60)%60,
                           p_evt->psz_short_description ? p_evt->psz_short_description : p_evt->psz_description );
        else
            input_Control( p_input, INPUT_ADD_INFO, psz_epg, psz_start, "%s (%2.2d:%2.2d)",
                           p_evt->psz_name,
                           p_evt->i_duration/60/60, (p_evt->i_duration/60)%60 );
    }
    free( psz_epg );
#endif
    /* Update now playing */
964
    free( p_pgrm->psz_now_playing );
965 966 967
    p_pgrm->psz_now_playing = NULL;
    if( p_epg->p_current && p_epg->p_current->psz_name && *p_epg->p_current->psz_name )
        p_pgrm->psz_now_playing = strdup( p_epg->p_current->psz_name );
968 969

    if( p_pgrm == p_sys->p_pgrm )
970
        input_item_SetNowPlaying( p_input->p->input.p_item, p_pgrm->psz_now_playing );
971 972

    if( p_pgrm->psz_now_playing )
973 974 975 976 977
    {
        input_Control( p_input, INPUT_ADD_INFO, psz_cat,
            input_MetaTypeToLocalizedString(vlc_meta_NowPlaying),
            p_pgrm->psz_now_playing );
    }
978
    else
979 980 981 982
    {
        input_Control( p_input, INPUT_DEL_INFO, psz_cat,
            input_MetaTypeToLocalizedString(vlc_meta_NowPlaying) );
    }
983

984 985 986
    free( psz_cat );
}

987 988 989 990 991 992 993 994 995 996 997 998
/* EsOutAdd:
 *  Add an es_out
 */
static es_out_id_t *EsOutAdd( es_out_t *out, es_format_t *fmt )
{
    es_out_sys_t      *p_sys = out->p_sys;
    input_thread_t    *p_input = p_sys->p_input;

    es_out_id_t       *es = malloc( sizeof( es_out_id_t ) );
    es_out_pgrm_t     *p_pgrm = NULL;
    int i;

999 1000
    if( !es ) return NULL;

1001 1002 1003
    if( fmt->i_group < 0 )
    {
        msg_Err( p_input, "invalid group number" );
1004
        free( es );
1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036
        return NULL;
    }

    /* Search the program */
    for( i = 0; i < p_sys->i_pgrm; i++ )
    {
        if( fmt->i_group == p_sys->pgrm[i]->i_id )
        {
            p_pgrm = p_sys->pgrm[i];
            break;
        }
    }
    if( p_pgrm == NULL )
    {
        /* Create a new one */
        p_pgrm = EsOutProgramAdd( out, fmt->i_group );
    }

    /* Increase ref count for program */
    p_pgrm->i_es++;

    /* Set up ES */
    if( fmt->i_id < 0 )
        fmt->i_id = out->p_sys->i_id;
    es->i_id = fmt->i_id;
    es->p_pgrm = p_pgrm;
    es_format_Copy( &es->fmt, fmt );
    es->i_preroll_end = -1;

    switch( fmt->i_cat )
    {
    case AUDIO_ES:
1037 1038 1039
    {
        audio_replay_gain_t rg;

1040
        es->i_channel = p_sys->i_audio;
1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059

        vlc_mutex_lock( &p_input->p->input.p_item->lock );
        memset( &rg, 0, sizeof(rg) );
        vlc_audio_replay_gain_MergeFromMeta( &rg, p_input->p->input.p_item->p_meta );
        vlc_mutex_unlock( &p_input->p->input.p_item->lock );

        for( i = 0; i < AUDIO_REPLAY_GAIN_MAX; i++ )
        {
            if( !es->fmt.audio_replay_gain.pb_peak[i] )
            {
                es->fmt.audio_replay_gain.pb_peak[i] = rg.pb_peak[i];
                es->fmt.audio_replay_gain.pf_peak[i] = rg.pf_peak[i];
            }
            if( !es->fmt.audio_replay_gain.pb_gain[i] )
            {
                es->fmt.audio_replay_gain.pb_gain[i] = rg.pb_gain[i];
                es->fmt.audio_replay_gain.pf_gain[i] = rg.pf_gain[i];
            }
        }
1060
        break;
1061
    }
1062 1063 1064

    case VIDEO_ES:
        es->i_channel = p_sys->i_video;
1065
        if( fmt->video.i_frame_rate && fmt->video.i_frame_rate_base )
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
1066 1067 1068 1069
            vlc_ureduce( &es->fmt.video.i_frame_rate,
                         &es->fmt.video.i_frame_rate_base,
                         fmt->video.i_frame_rate,
                         fmt->video.i_frame_rate_base, 0 );
1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082
        break;

    case SPU_ES:
        es->i_channel = p_sys->i_sub;
        break;

    default:
        es->i_channel = 0;
        break;
    }
    es->psz_language = LanguageGetName( fmt->psz_language ); /* remember so we only need to do it once */
    es->psz_language_code = LanguageGetCode( fmt->psz_language );
    es->p_dec = NULL;
1083
    es->p_dec_record = NULL;
1084
    for( i = 0; i < 4; i++ )
1085
        es->pb_cc_present[i] = false;
1086
    es->p_master = NULL;
1087 1088

    if( es->p_pgrm == p_sys->p_pgrm )
1089
        EsOutESVarUpdate( out, es, false );
1090 1091

    /* Select it if needed */
1092
    EsOutSelect( out, es, false );
1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114


    TAB_APPEND( out->p_sys->i_es, out->p_sys->es, es );
    p_sys->i_id++;  /* always incremented */
    switch( fmt->i_cat )
    {
        case AUDIO_ES:
            p_sys->i_audio++;
            break;
        case SPU_ES:
            p_sys->i_sub++;
            break;
        case VIDEO_ES:
            p_sys->i_video++;
            break;
    }

    EsOutAddInfo( out, es );

    return es;
}

1115
static bool EsIsSelected( es_out_id_t *es )
1116 1117 1118
{
    if( es->p_master )
    {
1119
        bool b_decode = false;
1120
        if( es->p_master->p_dec )
1121 1122 1123 1124 1125
        {
            int i_channel = EsOutGetClosedCaptionsChannel( es->fmt.i_codec );
            if( i_channel != -1 )
                input_DecoderGetCcState( es->p_master->p_dec, &b_decode, i_channel );
        }
1126 1127 1128 1129 1130 1131 1132
        return b_decode;
    }
    else
    {
        return es->p_dec != NULL;
    }
}
1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158
static void EsCreateDecoder( es_out_t *out, es_out_id_t *p_es )
{
    es_out_sys_t   *p_sys = out->p_sys;
    input_thread_t *p_input = p_sys->p_input;

    p_es->p_dec = input_DecoderNew( p_input, &p_es->fmt, p_input->p->p_sout );
    if( p_es->p_dec && !p_es->p_master && p_sys->p_sout_record )
        p_es->p_dec_record = input_DecoderNew( p_input, &p_es->fmt, p_sys->p_sout_record );
}
static void EsDestroyDecoder( es_out_t *out, es_out_id_t *p_es )
{
    es_out_sys_t   *p_sys = out->p_sys;

    if( !p_es->p_dec )
        return;

    input_DecoderDelete( p_es->p_dec );
    p_es->p_dec = NULL;

    if( p_es->p_dec_record )
    {
        input_DecoderDelete( p_es->p_dec_record );
        p_es->p_dec_record = NULL;
    }
}

1159 1160 1161 1162 1163
static void EsSelect( es_out_t *out, es_out_id_t *es )
{
    es_out_sys_t   *p_sys = out->p_sys;
    input_thread_t *p_input = p_sys->p_input;
    vlc_value_t    val;
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
1164
    const char     *psz_var;
1165

1166
    if( EsIsSelected( es ) )
1167 1168 1169 1170 1171
    {
        msg_Warn( p_input, "ES 0x%x is already selected", es->i_id );
        return;
    }

1172
    if( es->p_master )
1173
    {
1174
        int i_channel;
1175 1176 1177
        if( !es->p_master->p_dec )
            return;

1178
        i_channel = EsOutGetClosedCaptionsChannel( es->fmt.i_codec );
1179
        if( i_channel == -1 ||<