es_out.c 25.4 KB
Newer Older
1 2 3
/*****************************************************************************
 * es_out.c: Es Out handler for input.
 *****************************************************************************
4
 * Copyright (C) 2003-2004 VideoLAN
Clément Stenac's avatar
Clément Stenac committed
5
 * $Id: es_out.c,v 1.21 2004/01/25 17:16:05 zorglub Exp $
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
 *
 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
 *
 * 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
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
#include <stdlib.h>

#include <vlc/vlc.h>
#include <vlc/input.h>
#include <vlc/decoder.h>

33
#include "vlc_playlist.h"
34 35 36 37 38
#include "codecs.h"

/*****************************************************************************
 * Local prototypes
 *****************************************************************************/
39 40 41 42 43 44
struct es_out_id_t
{
    int             i_channel;
    es_descriptor_t *p_es;
};

45 46 47
struct es_out_sys_t
{
    input_thread_t *p_input;
48
    vlc_bool_t      b_pcr_set;
49

50
    /* all es */
51 52
    int         i_id;

53 54
    int         i_es;
    es_out_id_t **es;
55

56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72
    /* mode gestion */
    vlc_bool_t  b_active;
    int         i_mode;

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

    /* es to select */
    int         i_audio_last;
    int         i_sub_last;

    /* current main es */
    es_out_id_t *p_es_audio;
    es_out_id_t *p_es_video;
    es_out_id_t *p_es_sub;
73 74 75 76 77 78 79 80
};

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 * );
static int          EsOutControl( es_out_t *, int i_query, va_list );


Clément Stenac's avatar
Clément Stenac committed
81 82 83 84 85 86
/**
 * Create a new es_out structure
 *
 * \param p_input The related input thread
 * \return the new es_out_t
 */
87 88
es_out_t *input_EsOutNew( input_thread_t *p_input )
{
89 90 91
    es_out_t     *out = malloc( sizeof( es_out_t ) );
    es_out_sys_t *p_sys = malloc( sizeof( es_out_sys_t ) );
    vlc_value_t  val;
92 93 94 95 96

    out->pf_add     = EsOutAdd;
    out->pf_send    = EsOutSend;
    out->pf_del     = EsOutDel;
    out->pf_control = EsOutControl;
97 98 99
    out->p_sys      = p_sys;

    p_sys->p_input = p_input;
100
    p_sys->b_pcr_set = VLC_FALSE;
101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122

    p_sys->b_active = VLC_FALSE;
    p_sys->i_mode   = ES_OUT_MODE_AUTO;

    p_sys->i_id    = 1;

    p_sys->i_es    = 0;
    p_sys->es      = NULL;

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

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

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

    p_sys->p_es_audio = NULL;
    p_sys->p_es_video = NULL;
    p_sys->p_es_sub   = NULL;
123 124 125 126

    return out;
}

Clément Stenac's avatar
Clément Stenac committed
127 128 129 130 131 132
/**
 * Deletes an es_out structure
 *
 * \param out  the es_out structure to destroy
 * \return nothing
 */
133 134 135 136 137
void input_EsOutDelete( es_out_t *out )
{
    es_out_sys_t *p_sys = out->p_sys;
    int i;

138
    for( i = 0; i < p_sys->i_es; i++ )
139
    {
140
        free( p_sys->es[i] );
141
    }
142
    if( p_sys->es )
143
    {
144
        free( p_sys->es );
145 146 147 148
    }
    free( p_sys );
    free( out );
}
Clément Stenac's avatar
Clément Stenac committed
149 150 151 152 153 154 155 156

/**
 * Add a program
 *
 * \param out the es_out
 * \param i_group ...
 * \return a program descriptor for the new program
 */
157 158 159 160 161 162 163 164 165 166 167 168 169 170 171
static pgrm_descriptor_t *EsOutAddProgram( es_out_t *out, int i_group )
{
    input_thread_t    *p_input = out->p_sys->p_input;
    pgrm_descriptor_t *p_prgm;
    es_descriptor_t   *p_pmt;

    /* FIXME we should use a object variable but a lot of place in src/input
     * have to be changed */
    int               i_select = config_GetInt( p_input, "program" );

    /* create it */
    p_prgm = input_AddProgram( p_input, i_group, 0 );

    /* XXX welcome to kludge, add a dummy es, if you want to understand
     * why have a look at input_SetProgram. Basicaly, it assume the first
Clément Stenac's avatar
Clément Stenac committed
172
     * es to be the PMT, how that is stupid, nevertheless it is needed for
173 174 175 176 177 178 179 180 181 182 183 184 185
     * the old ts demuxer */
    p_pmt = input_AddES( p_input, p_prgm, 0, UNKNOWN_ES, NULL, 0 );
    p_pmt->i_fourcc = VLC_FOURCC( 'n', 'u', 'l', 'l' );

    /* Select i_select or the first by default */
    if( p_input->stream.p_selected_program == NULL &&
        ( i_select <= 0 || i_select == i_group ) )
    {
        p_input->stream.p_selected_program = p_prgm;
    }

    return p_prgm;
}
186

Clément Stenac's avatar
Clément Stenac committed
187 188
/**
 * Select an ES given the current mode
189
 * XXX: you need to take a the lock before (stream.stream_lock)
Clément Stenac's avatar
Clément Stenac committed
190 191 192 193 194 195
 *
 * \param out The es_out structure
 * \param es es_out_id structure
 * \param b_force ...
 * \return nothing
 */
196 197 198 199 200 201 202
static void EsOutSelect( es_out_t *out, es_out_id_t *es, vlc_bool_t b_force )
{
    es_out_sys_t      *p_sys = out->p_sys;
    input_thread_t    *p_input = p_sys->p_input;

    int i_cat = es->p_es->i_cat;

203 204
    if( !p_sys->b_active ||
        ( !b_force && es->p_es->fmt.i_priority < 0 ) )
205 206 207 208 209 210
    {
        return;
    }

    if( p_sys->i_mode == ES_OUT_MODE_ALL || b_force )
    {
211 212 213 214
        if( !es->p_es->p_dec )
        {
            input_SelectES( p_input, es->p_es );
        }
215 216 217 218 219
    }
    else if( p_sys->i_mode == ES_OUT_MODE_AUTO )
    {
        int i_wanted  = -1;

220 221 222 223 224 225
        if( es->p_es->p_pgrm != NULL &&
            es->p_es->p_pgrm != p_input->stream.p_selected_program )
        {
            return;
        }

226 227
        if( i_cat == AUDIO_ES )
        {
Gildas Bazin's avatar
 
Gildas Bazin committed
228 229 230
            if( p_sys->p_es_audio &&
                p_sys->p_es_audio->p_es->fmt.i_priority >=
                    es->p_es->fmt.i_priority )
231 232 233
            {
                return;
            }
Gildas Bazin's avatar
 
Gildas Bazin committed
234 235
            i_wanted  = p_sys->i_audio_last >= 0 ?
                            p_sys->i_audio_last : es->i_channel;
236 237 238
        }
        else if( i_cat == SPU_ES )
        {
Gildas Bazin's avatar
 
Gildas Bazin committed
239 240 241
            if( p_sys->p_es_sub &&
                p_sys->p_es_sub->p_es->fmt.i_priority >=
                    es->p_es->fmt.i_priority )
242 243 244 245 246 247 248 249 250 251
            {
                return;
            }
            i_wanted  = p_sys->i_sub_last;
        }
        else if( i_cat == VIDEO_ES )
        {
            i_wanted  = es->i_channel;
        }

Laurent Aimar's avatar
Laurent Aimar committed
252
        if( i_wanted == es->i_channel && es->p_es->p_dec == NULL )
253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285
        {
            input_SelectES( p_input, es->p_es );
        }
    }

    /* FIXME TODO handle priority here */
    if( es->p_es->p_dec )
    {
        if( i_cat == AUDIO_ES )
        {
            if( p_sys->i_mode == ES_OUT_MODE_AUTO &&
                p_sys->p_es_audio && p_sys->p_es_audio->p_es->p_dec )
            {
                input_UnselectES( p_input, p_sys->p_es_audio->p_es );
            }
            p_sys->p_es_audio = es;
        }
        else if( i_cat == SPU_ES )
        {
            if( p_sys->i_mode == ES_OUT_MODE_AUTO &&
                p_sys->p_es_sub && p_sys->p_es_sub->p_es->p_dec )
            {
                input_UnselectES( p_input, p_sys->p_es_sub->p_es );
            }
            p_sys->p_es_sub = es;
        }
        else if( i_cat == VIDEO_ES )
        {
            p_sys->p_es_video = es;
        }
    }
}

Clément Stenac's avatar
Clément Stenac committed
286 287 288 289 290 291 292
/**
 * Add an es_out
 *
 * \param out the es_out to add
 * \param fmt the es_format of the es_out
 * \return an es_out id
 */
293 294 295
static es_out_id_t *EsOutAdd( es_out_t *out, es_format_t *fmt )
{
    es_out_sys_t      *p_sys = out->p_sys;
296
    playlist_t        *p_playlist = NULL;
297
    input_thread_t    *p_input = p_sys->p_input;
298
    es_out_id_t       *es = malloc( sizeof( es_out_id_t ) );
299
    pgrm_descriptor_t *p_prgm = NULL;
Clément Stenac's avatar
Clément Stenac committed
300
    char              psz_cat[sizeof( _("Stream ") ) + 10];
301 302 303 304 305 306 307 308 309 310
    input_info_category_t *p_cat;

    vlc_mutex_lock( &p_input->stream.stream_lock );
    if( fmt->i_group >= 0 )
    {
        /* search program */
        p_prgm = input_FindProgram( p_input, fmt->i_group );

        if( p_prgm == NULL )
        {
311 312
            /* Create it */
            p_prgm = EsOutAddProgram( out, fmt->i_group );
313 314
        }
    }
315 316 317 318
    if( fmt->i_id < 0 )
    {
        fmt->i_id = out->p_sys->i_id - 1;
    }
319

320
    es->p_es = input_AddES( p_input,
321
                            p_prgm,
322
                            fmt->i_id + 1,
323
                            fmt->i_cat,
Laurent Aimar's avatar
Laurent Aimar committed
324
                            fmt->psz_language, 0 );
325
    es->p_es->i_stream_id = fmt->i_id;
326
    es->p_es->i_fourcc = fmt->i_codec;
327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345

    switch( fmt->i_cat )
    {
        case AUDIO_ES:
        {
            WAVEFORMATEX *p_wf =
                malloc( sizeof( WAVEFORMATEX ) + fmt->i_extra);

            p_wf->wFormatTag        = WAVE_FORMAT_UNKNOWN;
            p_wf->nChannels         = fmt->audio.i_channels;
            p_wf->nSamplesPerSec    = fmt->audio.i_rate;
            p_wf->nAvgBytesPerSec   = fmt->i_bitrate / 8;
            p_wf->nBlockAlign       = fmt->audio.i_blockalign;
            p_wf->wBitsPerSample    = fmt->audio.i_bitspersample;
            p_wf->cbSize            = fmt->i_extra;
            if( fmt->i_extra > 0 )
            {
                memcpy( &p_wf[1], fmt->p_extra, fmt->i_extra );
            }
346 347 348
            es->p_es->p_waveformatex = p_wf;

            es->i_channel = p_sys->i_audio;
349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371
            break;
        }
        case VIDEO_ES:
        {
            BITMAPINFOHEADER *p_bih = malloc( sizeof( BITMAPINFOHEADER ) +
                                              fmt->i_extra );
            p_bih->biSize           = sizeof(BITMAPINFOHEADER) + fmt->i_extra;
            p_bih->biWidth          = fmt->video.i_width;
            p_bih->biHeight         = fmt->video.i_height;
            p_bih->biPlanes         = 1;
            p_bih->biBitCount       = 24;
            p_bih->biCompression    = fmt->i_codec;
            p_bih->biSizeImage      = fmt->video.i_width *
                                          fmt->video.i_height;
            p_bih->biXPelsPerMeter  = 0;
            p_bih->biYPelsPerMeter  = 0;
            p_bih->biClrUsed        = 0;
            p_bih->biClrImportant   = 0;

            if( fmt->i_extra > 0 )
            {
                memcpy( &p_bih[1], fmt->p_extra, fmt->i_extra );
            }
372 373 374
            es->p_es->p_bitmapinfoheader = p_bih;

            es->i_channel = p_sys->i_video;
375 376 377 378 379 380 381 382
            break;
        }
        case SPU_ES:
        {
            subtitle_data_t *p_sub = malloc( sizeof( subtitle_data_t ) );
            memset( p_sub, 0, sizeof( subtitle_data_t ) );
            if( fmt->i_extra > 0 )
            {
383
                p_sub->psz_header = malloc( fmt->i_extra  + 1 );
384
                memcpy( p_sub->psz_header, fmt->p_extra , fmt->i_extra );
385 386
                /* just to be sure */
                ((uint8_t*)fmt->p_extra)[fmt->i_extra] = '\0';
387 388
            }
            /* FIXME beuuuuuurk */
389 390 391
            es->p_es->p_demux_data = p_sub;

            es->i_channel = p_sys->i_sub;
392 393
            break;
        }
394

395
        default:
396
            es->i_channel = 0;
397 398 399
            break;
    }

400
    sprintf( psz_cat, _("Stream %d"), out->p_sys->i_id - 1 );
401 402 403 404
    /* Get a category and the playlist */
    if( ( p_cat = input_InfoCategory( p_input, psz_cat ) ) &&
        ( p_playlist = (playlist_t *)vlc_object_find( p_input,
                       VLC_OBJECT_PLAYLIST, FIND_ANYWHERE ) ) )
405 406 407 408 409
    {
        /* Add information */
        switch( fmt->i_cat )
        {
            case AUDIO_ES:
Gildas Bazin's avatar
 
Gildas Bazin committed
410 411
                if( fmt->psz_description )
                {
Sam Hocevar's avatar
Sam Hocevar committed
412
                    input_AddInfo( p_cat, _("Description"), "%s",
Gildas Bazin's avatar
 
Gildas Bazin committed
413
                                   fmt->psz_description );
414 415 416
                    playlist_AddInfo( p_playlist, -1, psz_cat,
                                     _("Description"), "%s",
                                     fmt->psz_description );
Gildas Bazin's avatar
 
Gildas Bazin committed
417
                }
418 419
                input_AddInfo( p_cat, _("Codec"), "%.4s",
                               (char*)&fmt->i_codec );
420 421 422
                playlist_AddInfo( p_playlist, -1, psz_cat,
                                  _("Codec"),"%.4s",(char*)&fmt->i_codec );

423
                input_AddInfo( p_cat, _("Type"), _("Audio") );
424 425 426
                playlist_AddInfo( p_playlist, -1, psz_cat,
                                 _("Type"), _("Audio") );

427 428 429 430
                if( fmt->audio.i_channels > 0 )
                {
                    input_AddInfo( p_cat, _("Channels"), "%d",
                                   fmt->audio.i_channels );
431 432
                    playlist_AddInfo( p_playlist, -1, psz_cat,
                                      _("Channels"), "%d",                                                            fmt->audio.i_channels );
433
                }
Gildas Bazin's avatar
 
Gildas Bazin committed
434 435
                if( fmt->psz_language )
                {
Sam Hocevar's avatar
Sam Hocevar committed
436
                    input_AddInfo( p_cat, _("Language"), "%s",
Gildas Bazin's avatar
 
Gildas Bazin committed
437
                                   fmt->psz_language );
438 439 440
                    playlist_AddInfo( p_playlist, -1, psz_cat,
                                     _("Language"), "%s",
                                     fmt->psz_language );
Gildas Bazin's avatar
 
Gildas Bazin committed
441
                }
442 443
                if( fmt->audio.i_rate > 0 )
                {
444
                    input_AddInfo( p_cat, _("Sample rate"), _("%d Hz"),
445
                                   fmt->audio.i_rate );
446 447 448
                    playlist_AddInfo( p_playlist, -1, psz_cat,
                                     _("Sample rate"), _("%d Hz"),
                                      fmt->audio.i_rate );
449 450 451
                }
                if( fmt->i_bitrate > 0 )
                {
452
                    input_AddInfo( p_cat, _("Bitrate"), _("%d bps"),
453
                                   fmt->i_bitrate );
454 455 456
                    playlist_AddInfo( p_playlist, -1, psz_cat,
                                    _("Bitrate"), _("%d bps"),
                                     fmt->i_bitrate );
457 458 459
                }
                if( fmt->audio.i_bitspersample )
                {
Sam Hocevar's avatar
Sam Hocevar committed
460
                    input_AddInfo( p_cat, _("Bits per sample"), "%d",
461
                                   fmt->audio.i_bitspersample );
462 463 464
                    playlist_AddInfo( p_playlist, -1, psz_cat,
                                     _("Bits per sample"), "%d",
                                     fmt->audio.i_bitspersample );
465 466 467
                }
                break;
            case VIDEO_ES:
Gildas Bazin's avatar
 
Gildas Bazin committed
468 469
                if( fmt->psz_description )
                {
Sam Hocevar's avatar
Sam Hocevar committed
470
                    input_AddInfo( p_cat, _("Description"), "%s",
Gildas Bazin's avatar
 
Gildas Bazin committed
471
                                   fmt->psz_description );
472 473 474
                    playlist_AddInfo( p_playlist, -1, psz_cat,
                                     _("Description"), "%s",
                                     fmt->psz_description );
Gildas Bazin's avatar
 
Gildas Bazin committed
475
                }
476
                input_AddInfo( p_cat, _("Type"), _("Video") );
477 478 479
                playlist_AddInfo( p_playlist, -1, psz_cat,
                                _("Type"), _("Video") );

480 481
                input_AddInfo( p_cat, _("Codec"), "%.4s",
                               (char*)&fmt->i_codec );
482 483 484 485
                playlist_AddInfo( p_playlist, -1, psz_cat,
                                 _("Codec"), "%.4s",
                                 (char*)&fmt->i_codec );

486 487 488 489
                if( fmt->video.i_width > 0 && fmt->video.i_height > 0 )
                {
                    input_AddInfo( p_cat, _("Resolution"), "%dx%d",
                                   fmt->video.i_width, fmt->video.i_height );
490 491 492
                    playlist_AddInfo( p_playlist, -1, psz_cat,
                                    _("Resolution"), "%dx%d",
                                    fmt->video.i_width, fmt->video.i_height );
493 494 495 496
                }
                if( fmt->video.i_visible_width > 0 &&
                    fmt->video.i_visible_height > 0 )
                {
Sam Hocevar's avatar
Sam Hocevar committed
497
                    input_AddInfo( p_cat, _("Display resolution"), "%dx%d",
498 499
                                   fmt->video.i_visible_width,
                                   fmt->video.i_visible_height);
500 501 502 503
                     playlist_AddInfo( p_playlist, -1, psz_cat,
                                       _("Display resolution"), "%dx%d",
                                       fmt->video.i_visible_width,
                                       fmt->video.i_visible_height);
504 505 506 507
                }
                break;
            case SPU_ES:
                input_AddInfo( p_cat, _("Type"), _("Subtitle") );
508 509
                playlist_AddInfo( p_playlist, -1, psz_cat,
                                   _("Type"), _("Subtitle") );
510 511
                input_AddInfo( p_cat, _("Codec"), "%.4s",
                               (char*)&fmt->i_codec );
512 513 514
                playlist_AddInfo( p_playlist, -1, psz_cat,
                                 _("Codec"), "%.4s",
                                 (char*)&fmt->i_codec );
515 516 517 518 519
                break;
            default:

                break;
        }
520
        if( p_playlist ) vlc_object_release( p_playlist );
521 522
    }

523 524 525
    /* Apply mode
     * XXX change that when we do group too */
    if( 1 )
526
    {
527
        EsOutSelect( out, es, VLC_FALSE );
528
    }
529 530 531 532 533 534 535 536

    vlc_mutex_unlock( &p_input->stream.stream_lock );

    es->p_es->fmt = *fmt;

    TAB_APPEND( out->p_sys->i_es, out->p_sys->es, es );
    p_sys->i_id++;  /* always incremented */
    switch( fmt->i_cat )
537
    {
538 539 540 541 542 543 544 545 546
        case AUDIO_ES:
            p_sys->i_audio++;
            break;
        case SPU_ES:
            p_sys->i_sub++;
            break;
        case VIDEO_ES:
            p_sys->i_video++;
            break;
547
    }
548 549

    return es;
550 551
}

Clément Stenac's avatar
Clément Stenac committed
552 553 554 555 556 557 558
/**
 * Send a block for the given es_out
 *
 * \param out the es_out to send from
 * \param es the es_out_id
 * \param p_block the data block to send
 */
559
static int EsOutSend( es_out_t *out, es_out_id_t *es, block_t *p_block )
560
{
561 562
    es_out_sys_t *p_sys = out->p_sys;

563
    if( p_sys->b_pcr_set )
564
    {
565 566 567 568 569 570 571
        pgrm_descriptor_t *p_pgrm = es->p_es->p_pgrm;
        input_thread_t    *p_input = p_sys->p_input;

        if( p_pgrm == NULL )
        {
            p_pgrm = p_sys->p_input->stream.p_selected_program;
        }
572

573
        if( p_block->i_dts > 0 && p_pgrm )
574
        {
Gildas Bazin's avatar
 
Gildas Bazin committed
575
            p_block->i_dts =
576
                input_ClockGetTS( p_input, p_pgrm, p_block->i_dts * 9 / 100 );
577
        }
578
        if( p_block->i_pts > 0 && p_pgrm )
579
        {
Gildas Bazin's avatar
 
Gildas Bazin committed
580
            p_block->i_pts =
581
                input_ClockGetTS( p_input, p_pgrm, p_block->i_pts * 9 / 100 );
582 583
        }
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
584

585
    vlc_mutex_lock( &out->p_sys->p_input->stream.stream_lock );
Gildas Bazin's avatar
 
Gildas Bazin committed
586 587 588
    p_block->i_rate = out->p_sys->p_input->stream.control.i_rate;
    if( es->p_es->p_dec &&
        (es->p_es->i_cat!=AUDIO_ES || !p_sys->p_input->stream.control.b_mute) )
589
    {
590
        input_DecodeBlock( es->p_es->p_dec, p_block );
591 592 593
    }
    else
    {
594
        block_Release( p_block );
595
    }
596
    vlc_mutex_unlock( &out->p_sys->p_input->stream.stream_lock );
Gildas Bazin's avatar
 
Gildas Bazin committed
597

598 599 600 601 602 603
    return VLC_SUCCESS;
}

/*****************************************************************************
 * EsOutDel:
 *****************************************************************************/
604
static void EsOutDel( es_out_t *out, es_out_id_t *es )
605 606 607
{
    es_out_sys_t *p_sys = out->p_sys;

608 609 610 611 612 613 614 615 616 617 618 619 620 621
    TAB_REMOVE( p_sys->i_es, p_sys->es, es );

    switch( es->p_es->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;
    }
622

623
    /* We don't try to reselect */
624
    vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
625
    if( es->p_es->p_dec )
626
    {
627
        input_UnselectES( p_sys->p_input, es->p_es );
628
    }
629 630

    if( es->p_es->p_waveformatex )
631
    {
632 633
        free( es->p_es->p_waveformatex );
        es->p_es->p_waveformatex = NULL;
634
    }
635
    if( es->p_es->p_bitmapinfoheader )
636
    {
637 638
        free( es->p_es->p_bitmapinfoheader );
        es->p_es->p_bitmapinfoheader = NULL;
639
    }
640 641
    input_DelES( p_sys->p_input, es->p_es );

Gildas Bazin's avatar
 
Gildas Bazin committed
642 643 644 645
    if( p_sys->p_es_audio == es ) p_sys->p_es_audio = NULL;
    if( p_sys->p_es_video == es ) p_sys->p_es_video = NULL;
    if( p_sys->p_es_sub   == es ) p_sys->p_es_sub   = NULL;

646 647
    vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );

648
    free( es );
649 650
}

Clément Stenac's avatar
Clément Stenac committed
651 652 653 654 655 656 657 658
/**
 * Control query handler
 *
 * \param out the es_out to control
 * \param i_query A es_out query as defined in include/ninput.h
 * \param args a variable list of arguments for the query
 * \return VLC_SUCCESS or an error code
 */
659 660 661 662
static int EsOutControl( es_out_t *out, int i_query, va_list args )
{
    es_out_sys_t *p_sys = out->p_sys;
    vlc_bool_t  b, *pb;
663 664 665 666
    int         i, *pi;

    es_out_id_t *es;

667 668
    switch( i_query )
    {
669
        case ES_OUT_SET_ES_STATE:
670
            vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
671
            es = (es_out_id_t*) va_arg( args, es_out_id_t * );
672
            b = (vlc_bool_t) va_arg( args, vlc_bool_t );
673
            if( b && es->p_es->p_dec == NULL )
674
            {
675
                input_SelectES( p_sys->p_input, es->p_es );
676
                vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
677
                return es->p_es->p_dec ? VLC_SUCCESS : VLC_EGENERIC;
678
            }
679
            else if( !b && es->p_es->p_dec )
680
            {
681
                input_UnselectES( p_sys->p_input, es->p_es );
682 683 684
                vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
                return VLC_SUCCESS;
            }
Laurent Aimar's avatar
Laurent Aimar committed
685 686 687
            vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
            return VLC_SUCCESS;

688 689
        case ES_OUT_GET_ES_STATE:
            es = (es_out_id_t*) va_arg( args, es_out_id_t * );
690 691
            pb = (vlc_bool_t*) va_arg( args, vlc_bool_t * );

692 693 694 695
            *pb = es->p_es->p_dec ? VLC_TRUE : VLC_FALSE;
            return VLC_SUCCESS;

        case ES_OUT_SET_ACTIVE:
696
        {
697 698
            b = (vlc_bool_t) va_arg( args, vlc_bool_t );
            p_sys->b_active = b;
699 700 701 702 703 704 705

            if( b )
            {
                vlc_value_t val;
                val.b_bool = VLC_TRUE;
                var_Set( p_sys->p_input, "intf-change", val );
            }
706
            return VLC_SUCCESS;
707
        }
708 709 710 711 712 713 714 715

        case ES_OUT_GET_ACTIVE:
            pb = (vlc_bool_t*) va_arg( args, vlc_bool_t * );
            *pb = p_sys->b_active;
            return VLC_SUCCESS;

        case ES_OUT_SET_MODE:
            i = (int) va_arg( args, int );
Gildas Bazin's avatar
 
Gildas Bazin committed
716 717
            if( i == ES_OUT_MODE_NONE || i == ES_OUT_MODE_ALL ||
                i == ES_OUT_MODE_AUTO )
718
            {
719 720
                vlc_value_t val;

721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736
                p_sys->i_mode = i;

                /* Reapply policy mode */
                vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
                for( i = 0; i < p_sys->i_es; i++ )
                {
                    if( p_sys->es[i]->p_es->p_dec )
                    {
                        input_UnselectES( p_sys->p_input, p_sys->es[i]->p_es );
                    }
                }
                for( i = 0; i < p_sys->i_es; i++ )
                {
                    EsOutSelect( out, p_sys->es[i], VLC_FALSE );
                }
                vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
737 738 739 740

                val.b_bool = VLC_TRUE;
                var_Set( p_sys->p_input, "intf-change", val );

741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769
                return VLC_SUCCESS;
            }
            return VLC_EGENERIC;

        case ES_OUT_GET_MODE:
            pi = (int*) va_arg( args, int* );
            *pi = p_sys->i_mode;
            return VLC_SUCCESS;

        case ES_OUT_SET_ES:
            es = (es_out_id_t*) va_arg( args, es_out_id_t * );
            if( es == NULL )
            {
                for( i = 0; i < p_sys->i_es; i++ )
                {
                    vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
                    if( p_sys->es[i]->p_es->p_dec )
                    {
                        input_UnselectES( p_sys->p_input, p_sys->es[i]->p_es );
                    }
                    vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
                }
            }
            else
            {
                vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
                EsOutSelect( out, es, VLC_TRUE );
                vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
            }
770 771
            return VLC_SUCCESS;

772
        case ES_OUT_SET_PCR:
773
        case ES_OUT_SET_GROUP_PCR:
774 775
        {
            pgrm_descriptor_t *p_prgm = NULL;
776
            int64_t           i_pcr;
777

778 779 780 781 782 783 784 785
            if( i_query == ES_OUT_SET_PCR )
            {
                p_prgm = p_sys->p_input->stream.p_selected_program;
            }
            else
            {
                int i_group = (int)va_arg( args, int );
                p_prgm = input_FindProgram( p_sys->p_input, i_group );
786 787 788 789 790
                if( p_prgm == NULL )
                {
                    /* we create the requested program */
                    p_prgm = EsOutAddProgram( out, i_group );
                }
791
            }
792 793 794
            i_pcr   = (int64_t)va_arg( args, int64_t );

            /* search program */
795
            if( p_prgm )
796
            {
797
                input_ClockManageRef( p_sys->p_input, p_prgm, i_pcr * 9 / 100);
798 799 800 801 802 803
            }
            p_sys->b_pcr_set = VLC_TRUE;
            return VLC_SUCCESS;
        }

        case ES_OUT_RESET_PCR:
804
            for( i = 0; i < p_sys->p_input->stream.i_pgrm_number; i++ )
805
            {
806
                p_sys->p_input->stream.pp_programs[i]->i_synchro_state = SYNCHRO_REINIT;
807 808 809 810
            }
            p_sys->b_pcr_set = VLC_TRUE;
            return VLC_SUCCESS;

811 812 813 814 815
        default:
            msg_Err( p_sys->p_input, "unknown query in es_out_Control" );
            return VLC_EGENERIC;
    }
}