1 /*****************************************************************************
2 * es_out.c: Es Out handler for input.
3 *****************************************************************************
4 * Copyright (C) 2003-2004 VideoLAN
5 * $Id: es_out.c,v 1.23 2004/01/30 17:49:21 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"
37 /*****************************************************************************
39 *****************************************************************************/
43 es_descriptor_t *p_es;
48 input_thread_t *p_input;
71 es_out_id_t *p_es_audio;
72 es_out_id_t *p_es_video;
73 es_out_id_t *p_es_sub;
76 static es_out_id_t *EsOutAdd ( es_out_t *, es_format_t * );
77 static int EsOutSend ( es_out_t *, es_out_id_t *, block_t * );
78 static void EsOutDel ( es_out_t *, es_out_id_t * );
79 static int EsOutControl( es_out_t *, int i_query, va_list );
81 static char *LanguageGetName( const char *psz_code );
84 * Create a new es_out structure
86 * \param p_input The related input thread
87 * \return the new es_out_t
89 es_out_t *input_EsOutNew( input_thread_t *p_input )
91 es_out_t *out = malloc( sizeof( es_out_t ) );
92 es_out_sys_t *p_sys = malloc( sizeof( es_out_sys_t ) );
95 out->pf_add = EsOutAdd;
96 out->pf_send = EsOutSend;
97 out->pf_del = EsOutDel;
98 out->pf_control = EsOutControl;
101 p_sys->p_input = p_input;
102 p_sys->b_pcr_set = VLC_FALSE;
104 p_sys->b_active = VLC_FALSE;
105 p_sys->i_mode = ES_OUT_MODE_AUTO;
116 var_Get( p_input, "audio-channel", &val );
117 p_sys->i_audio_last = val.i_int;
119 var_Get( p_input, "spu-channel", &val );
120 p_sys->i_sub_last = val.i_int;
122 p_sys->p_es_audio = NULL;
123 p_sys->p_es_video = NULL;
124 p_sys->p_es_sub = NULL;
130 * Deletes an es_out structure
132 * \param out the es_out structure to destroy
135 void input_EsOutDelete( es_out_t *out )
137 es_out_sys_t *p_sys = out->p_sys;
140 for( i = 0; i < p_sys->i_es; i++ )
142 free( p_sys->es[i] );
155 * \param out the es_out
157 * \return a program descriptor for the new program
159 static pgrm_descriptor_t *EsOutAddProgram( es_out_t *out, int i_group )
161 input_thread_t *p_input = out->p_sys->p_input;
162 pgrm_descriptor_t *p_prgm;
163 es_descriptor_t *p_pmt;
165 /* FIXME we should use a object variable but a lot of place in src/input
166 * have to be changed */
167 int i_select = config_GetInt( p_input, "program" );
170 p_prgm = input_AddProgram( p_input, i_group, 0 );
172 /* XXX welcome to kludge, add a dummy es, if you want to understand
173 * why have a look at input_SetProgram. Basicaly, it assume the first
174 * es to be the PMT, how that is stupid, nevertheless it is needed for
175 * the old ts demuxer */
176 p_pmt = input_AddES( p_input, p_prgm, 0, UNKNOWN_ES, NULL, 0 );
177 p_pmt->i_fourcc = VLC_FOURCC( 'n', 'u', 'l', 'l' );
179 /* Select i_select or the first by default */
180 if( p_input->stream.p_selected_program == NULL &&
181 ( i_select <= 0 || i_select == i_group ) )
183 p_input->stream.p_selected_program = p_prgm;
190 * Select an ES given the current mode
191 * XXX: you need to take a the lock before (stream.stream_lock)
193 * \param out The es_out structure
194 * \param es es_out_id structure
198 static void EsOutSelect( es_out_t *out, es_out_id_t *es, vlc_bool_t b_force )
200 es_out_sys_t *p_sys = out->p_sys;
201 input_thread_t *p_input = p_sys->p_input;
203 int i_cat = es->p_es->i_cat;
205 if( !p_sys->b_active ||
206 ( !b_force && es->p_es->fmt.i_priority < 0 ) )
211 if( p_sys->i_mode == ES_OUT_MODE_ALL || b_force )
213 if( !es->p_es->p_dec )
215 input_SelectES( p_input, es->p_es );
218 else if( p_sys->i_mode == ES_OUT_MODE_AUTO )
222 if( es->p_es->p_pgrm != NULL &&
223 es->p_es->p_pgrm != p_input->stream.p_selected_program )
228 if( i_cat == AUDIO_ES )
230 if( p_sys->p_es_audio &&
231 p_sys->p_es_audio->p_es->fmt.i_priority >=
232 es->p_es->fmt.i_priority )
236 i_wanted = p_sys->i_audio_last >= 0 ?
237 p_sys->i_audio_last : es->i_channel;
239 else if( i_cat == SPU_ES )
241 if( p_sys->p_es_sub &&
242 p_sys->p_es_sub->p_es->fmt.i_priority >=
243 es->p_es->fmt.i_priority )
247 i_wanted = p_sys->i_sub_last;
249 else if( i_cat == VIDEO_ES )
251 i_wanted = es->i_channel;
254 if( i_wanted == es->i_channel && es->p_es->p_dec == NULL )
256 input_SelectES( p_input, es->p_es );
260 /* FIXME TODO handle priority here */
261 if( es->p_es->p_dec )
263 if( i_cat == AUDIO_ES )
265 if( p_sys->i_mode == ES_OUT_MODE_AUTO &&
266 p_sys->p_es_audio && p_sys->p_es_audio->p_es->p_dec )
268 input_UnselectES( p_input, p_sys->p_es_audio->p_es );
270 p_sys->p_es_audio = es;
272 else if( i_cat == SPU_ES )
274 if( p_sys->i_mode == ES_OUT_MODE_AUTO &&
275 p_sys->p_es_sub && p_sys->p_es_sub->p_es->p_dec )
277 input_UnselectES( p_input, p_sys->p_es_sub->p_es );
279 p_sys->p_es_sub = es;
281 else if( i_cat == VIDEO_ES )
283 p_sys->p_es_video = es;
291 * \param out the es_out to add
292 * \param fmt the es_format of the es_out
293 * \return an es_out id
295 static es_out_id_t *EsOutAdd( es_out_t *out, es_format_t *fmt )
297 es_out_sys_t *p_sys = out->p_sys;
298 playlist_t *p_playlist = NULL;
299 input_thread_t *p_input = p_sys->p_input;
300 es_out_id_t *es = malloc( sizeof( es_out_id_t ) );
301 pgrm_descriptor_t *p_prgm = NULL;
302 char psz_cat[sizeof( _("Stream ") ) + 10];
303 input_info_category_t *p_cat;
304 char *psz_description;
306 vlc_mutex_lock( &p_input->stream.stream_lock );
307 if( fmt->i_group >= 0 )
310 p_prgm = input_FindProgram( p_input, fmt->i_group );
315 p_prgm = EsOutAddProgram( out, fmt->i_group );
320 fmt->i_id = out->p_sys->i_id - 1;
323 psz_description = LanguageGetName( fmt->psz_language );
324 es->p_es = input_AddES( p_input,
328 psz_description, 0 );
329 es->p_es->i_stream_id = fmt->i_id;
330 es->p_es->i_fourcc = fmt->i_codec;
331 free( psz_description );
338 malloc( sizeof( WAVEFORMATEX ) + fmt->i_extra);
340 p_wf->wFormatTag = WAVE_FORMAT_UNKNOWN;
341 p_wf->nChannels = fmt->audio.i_channels;
342 p_wf->nSamplesPerSec = fmt->audio.i_rate;
343 p_wf->nAvgBytesPerSec = fmt->i_bitrate / 8;
344 p_wf->nBlockAlign = fmt->audio.i_blockalign;
345 p_wf->wBitsPerSample = fmt->audio.i_bitspersample;
346 p_wf->cbSize = fmt->i_extra;
347 if( fmt->i_extra > 0 )
349 memcpy( &p_wf[1], fmt->p_extra, fmt->i_extra );
351 es->p_es->p_waveformatex = p_wf;
353 es->i_channel = p_sys->i_audio;
358 BITMAPINFOHEADER *p_bih = malloc( sizeof( BITMAPINFOHEADER ) +
360 p_bih->biSize = sizeof(BITMAPINFOHEADER) + fmt->i_extra;
361 p_bih->biWidth = fmt->video.i_width;
362 p_bih->biHeight = fmt->video.i_height;
364 p_bih->biBitCount = 24;
365 p_bih->biCompression = fmt->i_codec;
366 p_bih->biSizeImage = fmt->video.i_width *
368 p_bih->biXPelsPerMeter = 0;
369 p_bih->biYPelsPerMeter = 0;
370 p_bih->biClrUsed = 0;
371 p_bih->biClrImportant = 0;
373 if( fmt->i_extra > 0 )
375 memcpy( &p_bih[1], fmt->p_extra, fmt->i_extra );
377 es->p_es->p_bitmapinfoheader = p_bih;
379 es->i_channel = p_sys->i_video;
384 subtitle_data_t *p_sub = malloc( sizeof( subtitle_data_t ) );
385 memset( p_sub, 0, sizeof( subtitle_data_t ) );
386 if( fmt->i_extra > 0 )
388 p_sub->psz_header = malloc( fmt->i_extra + 1 );
389 memcpy( p_sub->psz_header, fmt->p_extra , fmt->i_extra );
390 /* just to be sure */
391 ((uint8_t*)fmt->p_extra)[fmt->i_extra] = '\0';
393 /* FIXME beuuuuuurk */
394 es->p_es->p_demux_data = p_sub;
396 es->i_channel = p_sys->i_sub;
405 sprintf( psz_cat, _("Stream %d"), out->p_sys->i_id - 1 );
406 /* Get a category and the playlist */
407 if( ( p_cat = input_InfoCategory( p_input, psz_cat ) ) &&
408 ( p_playlist = (playlist_t *)vlc_object_find( p_input,
409 VLC_OBJECT_PLAYLIST, FIND_ANYWHERE ) ) )
411 /* Add information */
415 if( fmt->psz_description )
417 input_AddInfo( p_cat, _("Description"), "%s",
418 fmt->psz_description );
419 playlist_AddInfo( p_playlist, -1, psz_cat,
420 _("Description"), "%s",
421 fmt->psz_description );
423 input_AddInfo( p_cat, _("Codec"), "%.4s",
424 (char*)&fmt->i_codec );
425 playlist_AddInfo( p_playlist, -1, psz_cat,
426 _("Codec"),"%.4s",(char*)&fmt->i_codec );
428 input_AddInfo( p_cat, _("Type"), _("Audio") );
429 playlist_AddInfo( p_playlist, -1, psz_cat,
430 _("Type"), _("Audio") );
432 if( fmt->audio.i_channels > 0 )
434 input_AddInfo( p_cat, _("Channels"), "%d",
435 fmt->audio.i_channels );
436 playlist_AddInfo( p_playlist, -1, psz_cat,
437 _("Channels"), "%d", fmt->audio.i_channels );
439 if( fmt->psz_language )
441 input_AddInfo( p_cat, _("Language"), "%s",
443 playlist_AddInfo( p_playlist, -1, psz_cat,
447 if( fmt->audio.i_rate > 0 )
449 input_AddInfo( p_cat, _("Sample rate"), _("%d Hz"),
451 playlist_AddInfo( p_playlist, -1, psz_cat,
452 _("Sample rate"), _("%d Hz"),
455 if( fmt->i_bitrate > 0 )
457 input_AddInfo( p_cat, _("Bitrate"), _("%d bps"),
459 playlist_AddInfo( p_playlist, -1, psz_cat,
460 _("Bitrate"), _("%d bps"),
463 if( fmt->audio.i_bitspersample )
465 input_AddInfo( p_cat, _("Bits per sample"), "%d",
466 fmt->audio.i_bitspersample );
467 playlist_AddInfo( p_playlist, -1, psz_cat,
468 _("Bits per sample"), "%d",
469 fmt->audio.i_bitspersample );
473 if( fmt->psz_description )
475 input_AddInfo( p_cat, _("Description"), "%s",
476 fmt->psz_description );
477 playlist_AddInfo( p_playlist, -1, psz_cat,
478 _("Description"), "%s",
479 fmt->psz_description );
481 input_AddInfo( p_cat, _("Type"), _("Video") );
482 playlist_AddInfo( p_playlist, -1, psz_cat,
483 _("Type"), _("Video") );
485 input_AddInfo( p_cat, _("Codec"), "%.4s",
486 (char*)&fmt->i_codec );
487 playlist_AddInfo( p_playlist, -1, psz_cat,
489 (char*)&fmt->i_codec );
491 if( fmt->video.i_width > 0 && fmt->video.i_height > 0 )
493 input_AddInfo( p_cat, _("Resolution"), "%dx%d",
494 fmt->video.i_width, fmt->video.i_height );
495 playlist_AddInfo( p_playlist, -1, psz_cat,
496 _("Resolution"), "%dx%d",
497 fmt->video.i_width, fmt->video.i_height );
499 if( fmt->video.i_visible_width > 0 &&
500 fmt->video.i_visible_height > 0 )
502 input_AddInfo( p_cat, _("Display resolution"), "%dx%d",
503 fmt->video.i_visible_width,
504 fmt->video.i_visible_height);
505 playlist_AddInfo( p_playlist, -1, psz_cat,
506 _("Display resolution"), "%dx%d",
507 fmt->video.i_visible_width,
508 fmt->video.i_visible_height);
512 input_AddInfo( p_cat, _("Type"), _("Subtitle") );
513 playlist_AddInfo( p_playlist, -1, psz_cat,
514 _("Type"), _("Subtitle") );
515 input_AddInfo( p_cat, _("Codec"), "%.4s",
516 (char*)&fmt->i_codec );
517 playlist_AddInfo( p_playlist, -1, psz_cat,
519 (char*)&fmt->i_codec );
525 if( p_playlist ) vlc_object_release( p_playlist );
528 es_format_Copy( &es->p_es->fmt, fmt );
531 * XXX change that when we do group too */
534 EsOutSelect( out, es, VLC_FALSE );
536 vlc_mutex_unlock( &p_input->stream.stream_lock );
538 TAB_APPEND( out->p_sys->i_es, out->p_sys->es, es );
539 p_sys->i_id++; /* always incremented */
557 * Send a block for the given es_out
559 * \param out the es_out to send from
560 * \param es the es_out_id
561 * \param p_block the data block to send
563 static int EsOutSend( es_out_t *out, es_out_id_t *es, block_t *p_block )
565 es_out_sys_t *p_sys = out->p_sys;
567 if( p_sys->b_pcr_set )
569 pgrm_descriptor_t *p_pgrm = es->p_es->p_pgrm;
570 input_thread_t *p_input = p_sys->p_input;
574 p_pgrm = p_sys->p_input->stream.p_selected_program;
577 if( p_block->i_dts > 0 && p_pgrm )
580 input_ClockGetTS( p_input, p_pgrm, p_block->i_dts * 9 / 100 );
582 if( p_block->i_pts > 0 && p_pgrm )
585 input_ClockGetTS( p_input, p_pgrm, p_block->i_pts * 9 / 100 );
589 vlc_mutex_lock( &out->p_sys->p_input->stream.stream_lock );
590 p_block->i_rate = out->p_sys->p_input->stream.control.i_rate;
591 if( es->p_es->p_dec &&
592 (es->p_es->i_cat!=AUDIO_ES || !p_sys->p_input->stream.control.b_mute) )
594 input_DecodeBlock( es->p_es->p_dec, p_block );
598 block_Release( p_block );
600 vlc_mutex_unlock( &out->p_sys->p_input->stream.stream_lock );
605 /*****************************************************************************
607 *****************************************************************************/
608 static void EsOutDel( es_out_t *out, es_out_id_t *es )
610 es_out_sys_t *p_sys = out->p_sys;
612 TAB_REMOVE( p_sys->i_es, p_sys->es, es );
614 switch( es->p_es->i_cat )
627 /* We don't try to reselect */
628 vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
629 if( es->p_es->p_dec )
631 input_UnselectES( p_sys->p_input, es->p_es );
634 if( es->p_es->p_waveformatex )
636 free( es->p_es->p_waveformatex );
637 es->p_es->p_waveformatex = NULL;
639 if( es->p_es->p_bitmapinfoheader )
641 free( es->p_es->p_bitmapinfoheader );
642 es->p_es->p_bitmapinfoheader = NULL;
644 input_DelES( p_sys->p_input, es->p_es );
646 if( p_sys->p_es_audio == es ) p_sys->p_es_audio = NULL;
647 if( p_sys->p_es_video == es ) p_sys->p_es_video = NULL;
648 if( p_sys->p_es_sub == es ) p_sys->p_es_sub = NULL;
650 vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
656 * Control query handler
658 * \param out the es_out to control
659 * \param i_query A es_out query as defined in include/ninput.h
660 * \param args a variable list of arguments for the query
661 * \return VLC_SUCCESS or an error code
663 static int EsOutControl( es_out_t *out, int i_query, va_list args )
665 es_out_sys_t *p_sys = out->p_sys;
673 case ES_OUT_SET_ES_STATE:
674 vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
675 es = (es_out_id_t*) va_arg( args, es_out_id_t * );
676 b = (vlc_bool_t) va_arg( args, vlc_bool_t );
677 if( b && es->p_es->p_dec == NULL )
679 input_SelectES( p_sys->p_input, es->p_es );
680 vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
681 return es->p_es->p_dec ? VLC_SUCCESS : VLC_EGENERIC;
683 else if( !b && es->p_es->p_dec )
685 input_UnselectES( p_sys->p_input, es->p_es );
686 vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
689 vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
692 case ES_OUT_GET_ES_STATE:
693 es = (es_out_id_t*) va_arg( args, es_out_id_t * );
694 pb = (vlc_bool_t*) va_arg( args, vlc_bool_t * );
696 *pb = es->p_es->p_dec ? VLC_TRUE : VLC_FALSE;
699 case ES_OUT_SET_ACTIVE:
701 b = (vlc_bool_t) va_arg( args, vlc_bool_t );
707 val.b_bool = VLC_TRUE;
708 var_Set( p_sys->p_input, "intf-change", val );
713 case ES_OUT_GET_ACTIVE:
714 pb = (vlc_bool_t*) va_arg( args, vlc_bool_t * );
715 *pb = p_sys->b_active;
718 case ES_OUT_SET_MODE:
719 i = (int) va_arg( args, int );
720 if( i == ES_OUT_MODE_NONE || i == ES_OUT_MODE_ALL ||
721 i == ES_OUT_MODE_AUTO )
727 /* Reapply policy mode */
728 vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
729 for( i = 0; i < p_sys->i_es; i++ )
731 if( p_sys->es[i]->p_es->p_dec )
733 input_UnselectES( p_sys->p_input, p_sys->es[i]->p_es );
736 for( i = 0; i < p_sys->i_es; i++ )
738 EsOutSelect( out, p_sys->es[i], VLC_FALSE );
740 vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
742 val.b_bool = VLC_TRUE;
743 var_Set( p_sys->p_input, "intf-change", val );
749 case ES_OUT_GET_MODE:
750 pi = (int*) va_arg( args, int* );
755 es = (es_out_id_t*) va_arg( args, es_out_id_t * );
758 for( i = 0; i < p_sys->i_es; i++ )
760 vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
761 if( p_sys->es[i]->p_es->p_dec )
763 input_UnselectES( p_sys->p_input, p_sys->es[i]->p_es );
765 vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
770 vlc_mutex_lock( &p_sys->p_input->stream.stream_lock );
771 EsOutSelect( out, es, VLC_TRUE );
772 vlc_mutex_unlock( &p_sys->p_input->stream.stream_lock );
777 case ES_OUT_SET_GROUP_PCR:
779 pgrm_descriptor_t *p_prgm = NULL;
782 if( i_query == ES_OUT_SET_PCR )
784 p_prgm = p_sys->p_input->stream.p_selected_program;
788 int i_group = (int)va_arg( args, int );
789 p_prgm = input_FindProgram( p_sys->p_input, i_group );
792 /* we create the requested program */
793 p_prgm = EsOutAddProgram( out, i_group );
796 i_pcr = (int64_t)va_arg( args, int64_t );
801 input_ClockManageRef( p_sys->p_input, p_prgm, i_pcr * 9 / 100);
803 p_sys->b_pcr_set = VLC_TRUE;
807 case ES_OUT_RESET_PCR:
808 for( i = 0; i < p_sys->p_input->stream.i_pgrm_number; i++ )
810 p_sys->p_input->stream.pp_programs[i]->i_synchro_state = SYNCHRO_REINIT;
812 p_sys->b_pcr_set = VLC_TRUE;
816 msg_Err( p_sys->p_input, "unknown query in es_out_Control" );
821 /****************************************************************************
822 * LanguageGetName: try to expend iso639 into plain name
823 ****************************************************************************/
824 static char *LanguageGetName( const char *psz_code )
826 const iso639_lang_t *pl;
828 if( psz_code == NULL )
833 if( strlen( psz_code ) == 2 )
835 pl = GetLang_1( psz_code );
837 else if( strlen( psz_code ) == 3 )
839 pl = GetLang_2B( psz_code );
840 if( !strcmp( pl->psz_iso639_1, "??" ) )
842 pl = GetLang_2T( psz_code );
847 return strdup( psz_code );
850 if( !strcmp( pl->psz_iso639_1, "??" ) )
852 return strdup( psz_code );
856 if( *pl->psz_native_name )
858 return strdup( pl->psz_native_name );
860 return strdup( pl->psz_eng_name );