1 /*****************************************************************************
2 * es_out.c: Es Out handler for input.
3 *****************************************************************************
4 * Copyright (C) 2003-2004 VideoLAN
5 * $Id: es_out.c,v 1.20 2004/01/22 00:00:34 fenrir Exp $
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
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.
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.
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 *****************************************************************************/
24 /*****************************************************************************
26 *****************************************************************************/
30 #include <vlc/input.h>
31 #include <vlc/decoder.h>
33 #include "vlc_playlist.h"
36 /*****************************************************************************
38 *****************************************************************************/
42 es_descriptor_t *p_es;
47 input_thread_t *p_input;
70 es_out_id_t *p_es_audio;
71 es_out_id_t *p_es_video;
72 es_out_id_t *p_es_sub;
75 static es_out_id_t *EsOutAdd ( es_out_t *, es_format_t * );
76 static int EsOutSend ( es_out_t *, es_out_id_t *, block_t * );
77 static void EsOutDel ( es_out_t *, es_out_id_t * );
78 static int EsOutControl( es_out_t *, int i_query, va_list );
81 /*****************************************************************************
83 *****************************************************************************/
84 es_out_t *input_EsOutNew( input_thread_t *p_input )
86 es_out_t *out = malloc( sizeof( es_out_t ) );
87 es_out_sys_t *p_sys = malloc( sizeof( es_out_sys_t ) );
90 out->pf_add = EsOutAdd;
91 out->pf_send = EsOutSend;
92 out->pf_del = EsOutDel;
93 out->pf_control = EsOutControl;
96 p_sys->p_input = p_input;
97 p_sys->b_pcr_set = VLC_FALSE;
99 p_sys->b_active = VLC_FALSE;
100 p_sys->i_mode = ES_OUT_MODE_AUTO;
111 var_Get( p_input, "audio-channel", &val );
112 p_sys->i_audio_last = val.i_int;
114 var_Get( p_input, "spu-channel", &val );
115 p_sys->i_sub_last = val.i_int;
117 p_sys->p_es_audio = NULL;
118 p_sys->p_es_video = NULL;
119 p_sys->p_es_sub = NULL;
124 /*****************************************************************************
126 *****************************************************************************/
127 void input_EsOutDelete( es_out_t *out )
129 es_out_sys_t *p_sys = out->p_sys;
132 for( i = 0; i < p_sys->i_es; i++ )
134 free( p_sys->es[i] );
143 /*****************************************************************************
145 *****************************************************************************/
146 static pgrm_descriptor_t *EsOutAddProgram( es_out_t *out, int i_group )
148 input_thread_t *p_input = out->p_sys->p_input;
149 pgrm_descriptor_t *p_prgm;
150 es_descriptor_t *p_pmt;
152 /* FIXME we should use a object variable but a lot of place in src/input
153 * have to be changed */
154 int i_select = config_GetInt( p_input, "program" );
157 p_prgm = input_AddProgram( p_input, i_group, 0 );
159 /* XXX welcome to kludge, add a dummy es, if you want to understand
160 * why have a look at input_SetProgram. Basicaly, it assume the first
161 * es to be the PMT, how that is stupide, nevertheless it is needed for
162 * the old ts demuxer */
163 p_pmt = input_AddES( p_input, p_prgm, 0, UNKNOWN_ES, NULL, 0 );
164 p_pmt->i_fourcc = VLC_FOURCC( 'n', 'u', 'l', 'l' );
166 /* Select i_select or the first by default */
167 if( p_input->stream.p_selected_program == NULL &&
168 ( i_select <= 0 || i_select == i_group ) )
170 p_input->stream.p_selected_program = p_prgm;
176 /*****************************************************************************
177 * EsOutSelect: Select an ES given the current mode
178 * XXX: you need to take a the lock before (stream.stream_lock)
179 *****************************************************************************/
180 static void EsOutSelect( es_out_t *out, es_out_id_t *es, vlc_bool_t b_force )
182 es_out_sys_t *p_sys = out->p_sys;
183 input_thread_t *p_input = p_sys->p_input;
185 int i_cat = es->p_es->i_cat;
187 if( !p_sys->b_active ||
188 ( !b_force && es->p_es->fmt.i_priority < 0 ) )
193 if( p_sys->i_mode == ES_OUT_MODE_ALL || b_force )
195 if( !es->p_es->p_dec )
197 input_SelectES( p_input, es->p_es );
200 else if( p_sys->i_mode == ES_OUT_MODE_AUTO )
204 if( es->p_es->p_pgrm != NULL &&
205 es->p_es->p_pgrm != p_input->stream.p_selected_program )
210 if( i_cat == AUDIO_ES )
212 if( p_sys->p_es_audio &&
213 p_sys->p_es_audio->p_es->fmt.i_priority >=
214 es->p_es->fmt.i_priority )
218 i_wanted = p_sys->i_audio_last >= 0 ?
219 p_sys->i_audio_last : es->i_channel;
221 else if( i_cat == SPU_ES )
223 if( p_sys->p_es_sub &&
224 p_sys->p_es_sub->p_es->fmt.i_priority >=
225 es->p_es->fmt.i_priority )
229 i_wanted = p_sys->i_sub_last;
231 else if( i_cat == VIDEO_ES )
233 i_wanted = es->i_channel;
236 if( i_wanted == es->i_channel && es->p_es->p_dec == NULL )
238 input_SelectES( p_input, es->p_es );
242 /* FIXME TODO handle priority here */
243 if( es->p_es->p_dec )
245 if( i_cat == AUDIO_ES )
247 if( p_sys->i_mode == ES_OUT_MODE_AUTO &&
248 p_sys->p_es_audio && p_sys->p_es_audio->p_es->p_dec )
250 input_UnselectES( p_input, p_sys->p_es_audio->p_es );
252 p_sys->p_es_audio = es;
254 else if( i_cat == SPU_ES )
256 if( p_sys->i_mode == ES_OUT_MODE_AUTO &&
257 p_sys->p_es_sub && p_sys->p_es_sub->p_es->p_dec )
259 input_UnselectES( p_input, p_sys->p_es_sub->p_es );
261 p_sys->p_es_sub = es;
263 else if( i_cat == VIDEO_ES )
265 p_sys->p_es_video = es;
270 /*****************************************************************************
272 *****************************************************************************/
273 static es_out_id_t *EsOutAdd( es_out_t *out, es_format_t *fmt )
275 es_out_sys_t *p_sys = out->p_sys;
276 playlist_t *p_playlist = NULL;
277 input_thread_t *p_input = p_sys->p_input;
278 es_out_id_t *es = malloc( sizeof( es_out_id_t ) );
279 pgrm_descriptor_t *p_prgm = NULL;
280 char psz_cat[sizeof( "Stream " ) + 10];
281 input_info_category_t *p_cat;
283 vlc_mutex_lock( &p_input->stream.stream_lock );
284 if( fmt->i_group >= 0 )
287 p_prgm = input_FindProgram( p_input, fmt->i_group );
292 p_prgm = EsOutAddProgram( out, fmt->i_group );
297 fmt->i_id = out->p_sys->i_id - 1;
300 es->p_es = input_AddES( p_input,
304 fmt->psz_language, 0 );
305 es->p_es->i_stream_id = fmt->i_id;
306 es->p_es->i_fourcc = fmt->i_codec;
313 malloc( sizeof( WAVEFORMATEX ) + fmt->i_extra);
315 p_wf->wFormatTag = WAVE_FORMAT_UNKNOWN;
316 p_wf->nChannels = fmt->audio.i_channels;
317 p_wf->nSamplesPerSec = fmt->audio.i_rate;
318 p_wf->nAvgBytesPerSec = fmt->i_bitrate / 8;
319 p_wf->nBlockAlign = fmt->audio.i_blockalign;
320 p_wf->wBitsPerSample = fmt->audio.i_bitspersample;
321 p_wf->cbSize = fmt->i_extra;
322 if( fmt->i_extra > 0 )
324 memcpy( &p_wf[1], fmt->p_extra, fmt->i_extra );
326 es->p_es->p_waveformatex = p_wf;
328 es->i_channel = p_sys->i_audio;
333 BITMAPINFOHEADER *p_bih = malloc( sizeof( BITMAPINFOHEADER ) +
335 p_bih->biSize = sizeof(BITMAPINFOHEADER) + fmt->i_extra;
336 p_bih->biWidth = fmt->video.i_width;
337 p_bih->biHeight = fmt->video.i_height;
339 p_bih->biBitCount = 24;
340 p_bih->biCompression = fmt->i_codec;
341 p_bih->biSizeImage = fmt->video.i_width *
343 p_bih->biXPelsPerMeter = 0;
344 p_bih->biYPelsPerMeter = 0;
345 p_bih->biClrUsed = 0;
346 p_bih->biClrImportant = 0;
348 if( fmt->i_extra > 0 )
350 memcpy( &p_bih[1], fmt->p_extra, fmt->i_extra );
352 es->p_es->p_bitmapinfoheader = p_bih;
354 es->i_channel = p_sys->i_video;
359 subtitle_data_t *p_sub = malloc( sizeof( subtitle_data_t ) );
360 memset( p_sub, 0, sizeof( subtitle_data_t ) );
361 if( fmt->i_extra > 0 )
363 p_sub->psz_header = malloc( fmt->i_extra + 1 );
364 memcpy( p_sub->psz_header, fmt->p_extra , fmt->i_extra );
365 /* just to be sure */
366 ((uint8_t*)fmt->p_extra)[fmt->i_extra] = '\0';
368 /* FIXME beuuuuuurk */
369 es->p_es->p_demux_data = p_sub;
371 es->i_channel = p_sys->i_sub;
380 sprintf( psz_cat, _("Stream %d"), out->p_sys->i_id - 1 );
381 /* Get a category and the playlist */
382 if( ( p_cat = input_InfoCategory( p_input, psz_cat ) ) &&
383 ( p_playlist = (playlist_t *)vlc_object_find( p_input,
384 VLC_OBJECT_PLAYLIST, FIND_ANYWHERE ) ) )
386 /* Add information */
390 if( fmt->psz_description )
392 input_AddInfo( p_cat, _("Description"), "%s",
393 fmt->psz_description );
394 playlist_AddInfo( p_playlist, -1, psz_cat,
395 _("Description"), "%s",
396 fmt->psz_description );
398 input_AddInfo( p_cat, _("Codec"), "%.4s",
399 (char*)&fmt->i_codec );
400 playlist_AddInfo( p_playlist, -1, psz_cat,
401 _("Codec"),"%.4s",(char*)&fmt->i_codec );
403 input_AddInfo( p_cat, _("Type"), _("Audio") );
404 playlist_AddInfo( p_playlist, -1, psz_cat,
405 _("Type"), _("Audio") );
407 if( fmt->audio.i_channels > 0 )
409 input_AddInfo( p_cat, _("Channels"), "%d",
410 fmt->audio.i_channels );
411 playlist_AddInfo( p_playlist, -1, psz_cat,
412 _("Channels"), "%d", fmt->audio.i_channels );
414 if( fmt->psz_language )
416 input_AddInfo( p_cat, _("Language"), "%s",
418 playlist_AddInfo( p_playlist, -1, psz_cat,
422 if( fmt->audio.i_rate > 0 )
424 input_AddInfo( p_cat, _("Sample rate"), _("%d Hz"),
426 playlist_AddInfo( p_playlist, -1, psz_cat,
427 _("Sample rate"), _("%d Hz"),
430 if( fmt->i_bitrate > 0 )
432 input_AddInfo( p_cat, _("Bitrate"), _("%d bps"),
434 playlist_AddInfo( p_playlist, -1, psz_cat,
435 _("Bitrate"), _("%d bps"),
438 if( fmt->audio.i_bitspersample )
440 input_AddInfo( p_cat, _("Bits per sample"), "%d",
441 fmt->audio.i_bitspersample );
442 playlist_AddInfo( p_playlist, -1, psz_cat,
443 _("Bits per sample"), "%d",
444 fmt->audio.i_bitspersample );
448 if( fmt->psz_description )
450 input_AddInfo( p_cat, _("Description"), "%s",
451 fmt->psz_description );
452 playlist_AddInfo( p_playlist, -1, psz_cat,
453 _("Description"), "%s",
454 fmt->psz_description );
456 input_AddInfo( p_cat, _("Type"), _("Video") );
457 playlist_AddInfo( p_playlist, -1, psz_cat,
458 _("Type"), _("Video") );
460 input_AddInfo( p_cat, _("Codec"), "%.4s",
461 (char*)&fmt->i_codec );
462 playlist_AddInfo( p_playlist, -1, psz_cat,
464 (char*)&fmt->i_codec );
466 if( fmt->video.i_width > 0 && fmt->video.i_height > 0 )
468 input_AddInfo( p_cat, _("Resolution"), "%dx%d",
469 fmt->video.i_width, fmt->video.i_height );
470 playlist_AddInfo( p_playlist, -1, psz_cat,
471 _("Resolution"), "%dx%d",
472 fmt->video.i_width, fmt->video.i_height );
474 if( fmt->video.i_visible_width > 0 &&
475 fmt->video.i_visible_height > 0 )
477 input_AddInfo( p_cat, _("Display resolution"), "%dx%d",
478 fmt->video.i_visible_width,
479 fmt->video.i_visible_height);
480 playlist_AddInfo( p_playlist, -1, psz_cat,
481 _("Display resolution"), "%dx%d",
482 fmt->video.i_visible_width,
483 fmt->video.i_visible_height);
487 input_AddInfo( p_cat, _("Type"), _("Subtitle") );
488 playlist_AddInfo( p_playlist, -1, psz_cat,
489 _("Type"), _("Subtitle") );
490 input_AddInfo( p_cat, _("Codec"), "%.4s",
491 (char*)&fmt->i_codec );
492 playlist_AddInfo( p_playlist, -1, psz_cat,
494 (char*)&fmt->i_codec );
500 if( p_playlist ) vlc_object_release( p_playlist );
504 * XXX change that when we do group too */
507 EsOutSelect( out, es, VLC_FALSE );
510 vlc_mutex_unlock( &p_input->stream.stream_lock );
512 es->p_es->fmt = *fmt;
514 TAB_APPEND( out->p_sys->i_es, out->p_sys->es, es );
515 p_sys->i_id++; /* always incremented */
532 /*****************************************************************************
534 *****************************************************************************/
535 static int EsOutSend( es_out_t *out, es_out_id_t *es, block_t *p_block )
537 es_out_sys_t *p_sys = out->p_sys;
539 if( p_sys->b_pcr_set )
541 pgrm_descriptor_t *p_pgrm = es->p_es->p_pgrm;
542 input_thread_t *p_input = p_sys->p_input;
546 p_pgrm = p_sys->p_input->stream.p_selected_program;
549 if( p_block->i_dts > 0 && p_pgrm )
552 input_ClockGetTS( p_input, p_pgrm, p_block->i_dts * 9 / 100 );
554 if( p_block->i_pts > 0 && p_pgrm )
557 input_ClockGetTS( p_input, p_pgrm, p_block->i_pts * 9 / 100 );
561 vlc_mutex_lock( &out->p_sys->p_input->stream.stream_lock );
562 p_block->i_rate = out->p_sys->p_input->stream.control.i_rate;
563 if( es->p_es->p_dec &&
564 (es->p_es->i_cat!=AUDIO_ES || !p_sys->p_input->stream.control.b_mute) )
566 input_DecodeBlock( es->p_es->p_dec, p_block );
570 block_Release( p_block );
572 vlc_mutex_unlock( &out->p_sys->p_input->stream.stream_lock );
577 /*****************************************************************************
579 *****************************************************************************/
580 static void EsOutDel( es_out_t *out, es_out_id_t *es )
582 es_out_sys_t *p_sys = out->p_sys;
584 TAB_REMOVE( p_sys->i_es, p_sys->es, es );
586 switch( es->p_es->i_cat )
599 /* We don't try to reselect */
600 vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
601 if( es->p_es->p_dec )
603 input_UnselectES( p_sys->p_input, es->p_es );
606 if( es->p_es->p_waveformatex )
608 free( es->p_es->p_waveformatex );
609 es->p_es->p_waveformatex = NULL;
611 if( es->p_es->p_bitmapinfoheader )
613 free( es->p_es->p_bitmapinfoheader );
614 es->p_es->p_bitmapinfoheader = NULL;
616 input_DelES( p_sys->p_input, es->p_es );
618 if( p_sys->p_es_audio == es ) p_sys->p_es_audio = NULL;
619 if( p_sys->p_es_video == es ) p_sys->p_es_video = NULL;
620 if( p_sys->p_es_sub == es ) p_sys->p_es_sub = NULL;
622 vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
627 /*****************************************************************************
629 *****************************************************************************/
630 static int EsOutControl( es_out_t *out, int i_query, va_list args )
632 es_out_sys_t *p_sys = out->p_sys;
640 case ES_OUT_SET_ES_STATE:
641 vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
642 es = (es_out_id_t*) va_arg( args, es_out_id_t * );
643 b = (vlc_bool_t) va_arg( args, vlc_bool_t );
644 if( b && es->p_es->p_dec == NULL )
646 input_SelectES( p_sys->p_input, es->p_es );
647 vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
648 return es->p_es->p_dec ? VLC_SUCCESS : VLC_EGENERIC;
650 else if( !b && es->p_es->p_dec )
652 input_UnselectES( p_sys->p_input, es->p_es );
653 vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
656 vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
659 case ES_OUT_GET_ES_STATE:
660 es = (es_out_id_t*) va_arg( args, es_out_id_t * );
661 pb = (vlc_bool_t*) va_arg( args, vlc_bool_t * );
663 *pb = es->p_es->p_dec ? VLC_TRUE : VLC_FALSE;
666 case ES_OUT_SET_ACTIVE:
668 b = (vlc_bool_t) va_arg( args, vlc_bool_t );
674 val.b_bool = VLC_TRUE;
675 var_Set( p_sys->p_input, "intf-change", val );
680 case ES_OUT_GET_ACTIVE:
681 pb = (vlc_bool_t*) va_arg( args, vlc_bool_t * );
682 *pb = p_sys->b_active;
685 case ES_OUT_SET_MODE:
686 i = (int) va_arg( args, int );
687 if( i == ES_OUT_MODE_NONE || i == ES_OUT_MODE_ALL ||
688 i == ES_OUT_MODE_AUTO )
694 /* Reapply policy mode */
695 vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
696 for( i = 0; i < p_sys->i_es; i++ )
698 if( p_sys->es[i]->p_es->p_dec )
700 input_UnselectES( p_sys->p_input, p_sys->es[i]->p_es );
703 for( i = 0; i < p_sys->i_es; i++ )
705 EsOutSelect( out, p_sys->es[i], VLC_FALSE );
707 vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
709 val.b_bool = VLC_TRUE;
710 var_Set( p_sys->p_input, "intf-change", val );
716 case ES_OUT_GET_MODE:
717 pi = (int*) va_arg( args, int* );
722 es = (es_out_id_t*) va_arg( args, es_out_id_t * );
725 for( i = 0; i < p_sys->i_es; i++ )
727 vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
728 if( p_sys->es[i]->p_es->p_dec )
730 input_UnselectES( p_sys->p_input, p_sys->es[i]->p_es );
732 vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
737 vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
738 EsOutSelect( out, es, VLC_TRUE );
739 vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
744 case ES_OUT_SET_GROUP_PCR:
746 pgrm_descriptor_t *p_prgm = NULL;
749 if( i_query == ES_OUT_SET_PCR )
751 p_prgm = p_sys->p_input->stream.p_selected_program;
755 int i_group = (int)va_arg( args, int );
756 p_prgm = input_FindProgram( p_sys->p_input, i_group );
759 /* we create the requested program */
760 p_prgm = EsOutAddProgram( out, i_group );
763 i_pcr = (int64_t)va_arg( args, int64_t );
768 input_ClockManageRef( p_sys->p_input, p_prgm, i_pcr * 9 / 100);
770 p_sys->b_pcr_set = VLC_TRUE;
774 case ES_OUT_RESET_PCR:
775 for( i = 0; i < p_sys->p_input->stream.i_pgrm_number; i++ )
777 p_sys->p_input->stream.pp_programs[i]->i_synchro_state = SYNCHRO_REINIT;
779 p_sys->b_pcr_set = VLC_TRUE;
783 msg_Err( p_sys->p_input, "unknown query in es_out_Control" );