X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=modules%2Fcodec%2Flibmpeg2.c;h=9863018d8b6f0a0c70624f1f7a4f9a07598bfb2f;hb=78d87996ccb92d1dc91c9987685f976ed3be08a6;hp=dfe6551e8fa8a8b940f082a5ce2fd8c282244182;hpb=27d483e9ef7a451397d7857251c8d67097661f1d;p=vlc diff --git a/modules/codec/libmpeg2.c b/modules/codec/libmpeg2.c index dfe6551e8f..9863018d8b 100644 --- a/modules/codec/libmpeg2.c +++ b/modules/codec/libmpeg2.c @@ -29,21 +29,16 @@ # include "config.h" #endif -#include +#include #include -#include #include +#include +#include "../codec/cc.h" -#include +#include #include -/* Aspect ratio (ISO/IEC 13818-2 section 6.3.3, table 6-3) */ -#define AR_SQUARE_PICTURE 1 /* square pixels */ -#define AR_4_3_PICTURE 2 /* 4:3 picture (TV) */ -#define AR_16_9_PICTURE 3 /* 16:9 picture (wide screen) */ -#define AR_221_1_PICTURE 4 /* 2.21:1 picture (movie) */ - /***************************************************************************** * decoder_sys_t : libmpeg2 decoder descriptor *****************************************************************************/ @@ -63,7 +58,6 @@ struct decoder_sys_t mtime_t i_current_pts; mtime_t i_previous_dts; mtime_t i_current_dts; - int i_current_rate; picture_t * p_picture_to_destroy; bool b_garbage_pic; bool b_after_sequence_header; /* is it the next frame after @@ -77,11 +71,18 @@ struct decoder_sys_t * Output properties */ decoder_synchro_t *p_synchro; - int i_aspect; - int i_sar_num; - int i_sar_den; - mtime_t i_last_frame_pts; - + int i_aspect; + int i_sar_num; + int i_sar_den; + mtime_t i_last_frame_pts; + + /* Closed captioning support */ + uint32_t i_cc_flags; + mtime_t i_cc_pts; + mtime_t i_cc_dts; + cc_data_t cc; + uint8_t *p_gop_user_data; + uint32_t i_gop_user_data; }; /***************************************************************************** @@ -91,6 +92,7 @@ static int OpenDecoder( vlc_object_t * ); static void CloseDecoder( vlc_object_t * ); static picture_t *DecodeBlock( decoder_t *, block_t ** ); +static block_t *GetCc( decoder_t *p_dec, bool pb_present[4] ); static picture_t *GetNewPicture( decoder_t *, uint8_t ** ); static void GetAR( decoder_t *p_dec ); @@ -98,14 +100,14 @@ static void GetAR( decoder_t *p_dec ); /***************************************************************************** * Module descriptor *****************************************************************************/ -vlc_module_begin(); - set_description( _("MPEG I/II video decoder (using libmpeg2)") ); - set_capability( "decoder", 150 ); - set_category( CAT_INPUT ); - set_subcategory( SUBCAT_INPUT_VCODEC ); - set_callbacks( OpenDecoder, CloseDecoder ); - add_shortcut( "libmpeg2" ); -vlc_module_end(); +vlc_module_begin () + set_description( N_("MPEG I/II video decoder (using libmpeg2)") ) + set_capability( "decoder", 150 ) + set_category( CAT_INPUT ) + set_subcategory( SUBCAT_INPUT_VCODEC ) + set_callbacks( OpenDecoder, CloseDecoder ) + add_shortcut( "libmpeg2" ) +vlc_module_end () /***************************************************************************** * OpenDecoder: probe the decoder and return score @@ -116,24 +118,29 @@ static int OpenDecoder( vlc_object_t *p_this ) decoder_sys_t *p_sys; uint32_t i_accel = 0; - if( p_dec->fmt_in.i_codec != VLC_FOURCC('m','p','g','v') && - p_dec->fmt_in.i_codec != VLC_FOURCC('m','p','g','1') && - /* Pinnacle hardware-mpeg1 */ - p_dec->fmt_in.i_codec != VLC_FOURCC('P','I','M','1') && - p_dec->fmt_in.i_codec != VLC_FOURCC('m','p','2','v') && - p_dec->fmt_in.i_codec != VLC_FOURCC('m','p','g','2') && - p_dec->fmt_in.i_codec != VLC_FOURCC('h','d','v','2') ) - { + if( p_dec->fmt_in.i_codec != VLC_CODEC_MPGV ) return VLC_EGENERIC; + + /* Select onl recognized original format (standard mpeg video) */ + switch( p_dec->fmt_in.i_original_fourcc ) + { + case VLC_FOURCC('m','p','g','1'): + case VLC_FOURCC('m','p','g','2'): + case VLC_FOURCC('m','p','g','v'): + case VLC_FOURCC('P','I','M','1'): + case VLC_FOURCC('h','d','v','2'): + break; + default: + if( p_dec->fmt_in.i_original_fourcc ) + return VLC_EGENERIC; + break; } /* Allocate the memory needed to store the decoder's structure */ - if( ( p_dec->p_sys = p_sys = - (decoder_sys_t *)malloc(sizeof(decoder_sys_t)) ) == NULL ) + if( ( p_dec->p_sys = p_sys = calloc( 1, sizeof(*p_sys)) ) == NULL ) return VLC_ENOMEM; /* Initialize the thread properties */ - memset( p_sys, 0, sizeof(decoder_sys_t) ); p_sys->p_mpeg2dec = NULL; p_sys->p_synchro = NULL; p_sys->p_info = NULL; @@ -148,6 +155,16 @@ static int OpenDecoder( vlc_object_t *p_this ) p_sys->b_skip = 0; p_sys->b_preroll = false; + p_sys->i_cc_pts = 0; + p_sys->i_cc_dts = 0; + p_sys->i_cc_flags = 0; +#if MPEG2_RELEASE >= MPEG2_VERSION (0, 5, 0) + p_dec->pf_get_cc = GetCc; + cc_Init( &p_sys->cc ); +#endif + p_sys->p_gop_user_data = NULL; + p_sys->i_gop_user_data = 0; + #if defined( __i386__ ) || defined( __x86_64__ ) if( vlc_CPU() & CPU_CAPABILITY_MMX ) { @@ -191,6 +208,8 @@ static int OpenDecoder( vlc_object_t *p_this ) p_sys->p_info = mpeg2_info( p_sys->p_mpeg2dec ); p_dec->pf_decode_video = DecodeBlock; + p_dec->fmt_out.i_cat = VIDEO_ES; + p_dec->fmt_out.i_codec = 0; return VLC_SUCCESS; } @@ -222,6 +241,9 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block ) block_Release( p_block ); return NULL; } + if( p_block->i_flags & (BLOCK_FLAG_DISCONTINUITY + | BLOCK_FLAG_CORRUPTED)) + cc_Flush( &p_sys->cc ); if( (p_block->i_flags & (BLOCK_FLAG_DISCONTINUITY | BLOCK_FLAG_CORRUPTED)) && @@ -241,7 +263,10 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block ) uint8_t *buf[3]; buf[0] = buf[1] = buf[2] = NULL; if( (p_pic = GetNewPicture( p_dec, buf )) == NULL ) + { + p_block->i_buffer = 0; break; + } mpeg2_set_buf( p_sys->p_mpeg2dec, buf, p_pic ); mpeg2_stride( p_sys->p_mpeg2dec, p_pic->p[Y_PLANE].i_pitch ); } @@ -250,7 +275,7 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block ) if ( p_sys->b_slice_i ) { decoder_SynchroNewPicture( p_sys->p_synchro, - I_CODING_TYPE, 2, 0, 0, p_sys->i_current_rate, + I_CODING_TYPE, 2, 0, 0, p_sys->p_info->sequence->flags & SEQ_FLAG_LOW_DELAY ); decoder_SynchroDecode( p_sys->p_synchro ); decoder_SynchroEnd( p_sys->p_synchro, I_CODING_TYPE, 0 ); @@ -286,17 +311,13 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block ) p_sys->i_current_dts = p_block->i_dts; } - p_sys->i_current_rate = p_block->i_rate; - mpeg2_buffer( p_sys->p_mpeg2dec, p_block->p_buffer, p_block->p_buffer + p_block->i_buffer ); p_block->i_buffer = 0; break; -#ifdef STATE_SEQUENCE_MODIFIED -/* FIXME - the above ifdef is always FALSE, even with libmpeg2-0.5.0 - * Need to improve the implementation to avoid invalid/unref pictures */ +#if MPEG2_RELEASE >= MPEG2_VERSION (0, 5, 0) case STATE_SEQUENCE_MODIFIED: GetAR( p_dec ); @@ -330,7 +351,7 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block ) /* For some reason, libmpeg2 will put this pic twice in * discard_picture. This can be considered a bug in libmpeg2. */ - p_dec->pf_picture_link( p_dec, p_pic ); + decoder_LinkPicture( p_dec, p_pic ); if( p_sys->p_synchro ) { @@ -360,7 +381,7 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block ) /* Intra-slice refresh. Simulate a blank I picture. */ msg_Dbg( p_dec, "intra-slice refresh stream" ); decoder_SynchroNewPicture( p_sys->p_synchro, - I_CODING_TYPE, 2, 0, 0, p_sys->i_current_rate, + I_CODING_TYPE, 2, 0, 0, p_sys->p_info->sequence->flags & SEQ_FLAG_LOW_DELAY ); decoder_SynchroDecode( p_sys->p_synchro ); decoder_SynchroEnd( p_sys->p_synchro, I_CODING_TYPE, 0 ); @@ -410,23 +431,39 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block ) p_sys->p_info->current_picture->flags & PIC_MASK_CODING_TYPE, p_sys->p_info->current_picture->nb_fields == 1 ? 2 : p_sys->p_info->current_picture->nb_fields, i_pts, i_dts, - p_sys->i_current_rate, p_sys->p_info->sequence->flags & SEQ_FLAG_LOW_DELAY ); + + bool b_skip = false; if( !p_dec->b_pace_control && !p_sys->b_preroll && !(p_sys->b_slice_i && ((p_sys->p_info->current_picture->flags - & PIC_MASK_CODING_TYPE) == P_CODING_TYPE)) + & PIC_MASK_CODING_TYPE) == PIC_FLAG_CODING_TYPE_P)) && !decoder_SynchroChoose( p_sys->p_synchro, p_sys->p_info->current_picture->flags & PIC_MASK_CODING_TYPE, /*p_sys->p_vout->render_time*/ 0 /*FIXME*/, p_sys->p_info->sequence->flags & SEQ_FLAG_LOW_DELAY ) ) + { + b_skip = true; + } + + p_pic = NULL; + if( !b_skip ) + p_pic = GetNewPicture( p_dec, buf ); + + if( b_skip || !p_pic ) { mpeg2_skip( p_sys->p_mpeg2dec, 1 ); p_sys->b_skip = 1; decoder_SynchroTrash( p_sys->p_synchro ); mpeg2_set_buf( p_sys->p_mpeg2dec, buf, NULL ); + + if( !b_skip ) + { + block_Release( p_block ); + return NULL; + } } else { @@ -434,18 +471,49 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block ) p_sys->b_skip = 0; decoder_SynchroDecode( p_sys->p_synchro ); - if( (p_pic = GetNewPicture( p_dec, buf )) == NULL ) + mpeg2_set_buf( p_sys->p_mpeg2dec, buf, p_pic ); + mpeg2_stride( p_sys->p_mpeg2dec, p_pic->p[Y_PLANE].i_pitch ); + } + if( p_sys->p_info->user_data_len > 2 || p_sys->i_gop_user_data > 2 ) + { + p_sys->i_cc_pts = i_pts; + p_sys->i_cc_dts = i_dts; + if( (p_sys->p_info->current_picture->flags + & PIC_MASK_CODING_TYPE) == PIC_FLAG_CODING_TYPE_P ) + p_sys->i_cc_flags = BLOCK_FLAG_TYPE_P; + else if( (p_sys->p_info->current_picture->flags + & PIC_MASK_CODING_TYPE) == PIC_FLAG_CODING_TYPE_B ) + p_sys->i_cc_flags = BLOCK_FLAG_TYPE_B; + else p_sys->i_cc_flags = BLOCK_FLAG_TYPE_I; + + if( p_sys->i_gop_user_data > 2 ) { - block_Release( p_block ); - return NULL; + /* We now have picture info for any cached user_data out of the gop */ + cc_Extract( &p_sys->cc, &p_sys->p_gop_user_data[0], p_sys->i_gop_user_data ); + p_sys->i_gop_user_data = 0; } - mpeg2_set_buf( p_sys->p_mpeg2dec, buf, p_pic ); - mpeg2_stride( p_sys->p_mpeg2dec, p_pic->p[Y_PLANE].i_pitch ); + /* Extract the CC from the user_data of the picture */ + if( p_sys->p_info->user_data_len > 2 ) + cc_Extract( &p_sys->cc, &p_sys->p_info->user_data[0], p_sys->p_info->user_data_len ); } } break; + case STATE_GOP: + /* There can be userdata in a GOP. It needs to be remembered for the next picture. */ + if( p_sys->p_info->user_data_len > 2 ) + { + free( p_sys->p_gop_user_data ); + p_sys->p_gop_user_data = calloc( p_sys->p_info->user_data_len, sizeof(uint8_t) ); + if( p_sys->p_gop_user_data ) + { + p_sys->i_gop_user_data = p_sys->p_info->user_data_len; + memcpy( p_sys->p_gop_user_data, p_sys->p_info->user_data, p_sys->p_info->user_data_len ); + } + } + break; + case STATE_END: case STATE_SLICE: p_pic = NULL; @@ -460,7 +528,7 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block ) p_sys->b_garbage_pic ); p_sys->b_garbage_pic = 0; - if ( p_sys->p_picture_to_destroy != p_pic ) + if( p_sys->p_picture_to_destroy != p_pic ) { p_pic->date = decoder_SynchroDate( p_sys->p_synchro ); } @@ -474,8 +542,8 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block ) if( p_sys->p_info->discard_fbuf && p_sys->p_info->discard_fbuf->id ) { - p_dec->pf_picture_unlink( p_dec, - p_sys->p_info->discard_fbuf->id ); + decoder_UnlinkPicture( p_dec, + p_sys->p_info->discard_fbuf->id ); } /* For still frames */ @@ -490,7 +558,6 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block ) return p_pic; } - break; case STATE_INVALID: @@ -501,12 +568,13 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block ) msg_Warn( p_dec, "invalid picture encountered" ); if ( ( p_sys->p_info->current_picture == NULL ) || ( ( p_sys->p_info->current_picture->flags & - PIC_MASK_CODING_TYPE) != B_CODING_TYPE ) ) + PIC_MASK_CODING_TYPE) != PIC_FLAG_CODING_TYPE_B ) ) { if( p_sys->p_synchro ) decoder_SynchroReset( p_sys->p_synchro ); } mpeg2_skip( p_sys->p_mpeg2dec, 1 ); p_sys->b_skip = 1; + cc_Flush( &p_sys->cc ); if( p_sys->p_info->current_fbuf && p_sys->p_info->current_fbuf->id ) @@ -540,7 +608,7 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block ) if( p_sys->b_slice_i ) { decoder_SynchroNewPicture( p_sys->p_synchro, - I_CODING_TYPE, 2, 0, 0, p_sys->i_current_rate, + I_CODING_TYPE, 2, 0, 0, p_sys->p_info->sequence->flags & SEQ_FLAG_LOW_DELAY ); decoder_SynchroDecode( p_sys->p_synchro ); decoder_SynchroEnd( p_sys->p_synchro, I_CODING_TYPE, 0 ); @@ -565,6 +633,8 @@ static void CloseDecoder( vlc_object_t *p_this ) decoder_t *p_dec = (decoder_t *)p_this; decoder_sys_t *p_sys = p_dec->p_sys; + free( p_sys->p_gop_user_data ); + if( p_sys->p_synchro ) decoder_SynchroRelease( p_sys->p_synchro ); if( p_sys->p_mpeg2dec ) mpeg2_close( p_sys->p_mpeg2dec ); @@ -601,10 +671,10 @@ static picture_t *GetNewPicture( decoder_t *p_dec, uint8_t **pp_buf ) p_dec->fmt_out.i_codec = ( p_sys->p_info->sequence->chroma_height < p_sys->p_info->sequence->height ) ? - VLC_FOURCC('I','4','2','0') : VLC_FOURCC('I','4','2','2'); + VLC_CODEC_I420 : VLC_CODEC_I422; /* Get a new picture */ - p_pic = p_dec->pf_vout_buffer_new( p_dec ); + p_pic = decoder_NewPicture( p_dec ); if( p_pic == NULL ) return NULL; @@ -615,7 +685,7 @@ static picture_t *GetNewPicture( decoder_t *p_dec, uint8_t **pp_buf ) p_pic->i_nb_fields = p_sys->p_info->current_picture != NULL ? p_sys->p_info->current_picture->nb_fields : 2; - p_dec->pf_picture_link( p_dec, p_pic ); + decoder_LinkPicture( p_dec, p_pic ); pp_buf[0] = p_pic->p[0].p_pixels; pp_buf[1] = p_pic->p[1].p_pixels; @@ -624,6 +694,33 @@ static picture_t *GetNewPicture( decoder_t *p_dec, uint8_t **pp_buf ) return p_pic; } +/***************************************************************************** + * GetCc: Retrieves the Closed Captions for the CC decoder. + *****************************************************************************/ +static block_t *GetCc( decoder_t *p_dec, bool pb_present[4] ) +{ + decoder_sys_t *p_sys = p_dec->p_sys; + block_t *p_cc = NULL; + int i; + + for( i = 0; i < 4; i++ ) + pb_present[i] = p_sys->cc.pb_present[i]; + + if( p_sys->cc.i_data <= 0 ) + return NULL; + + p_cc = block_New( p_dec, p_sys->cc.i_data); + if( p_cc ) + { + memcpy( p_cc->p_buffer, p_sys->cc.p_data, p_sys->cc.i_data ); + p_cc->i_dts = + p_cc->i_pts = p_sys->cc.b_reorder ? p_sys->i_cc_pts : p_sys->i_cc_dts; + p_cc->i_flags = ( p_sys->cc.b_reorder ? p_sys->i_cc_flags : BLOCK_FLAG_TYPE_P ) & ( BLOCK_FLAG_TYPE_I|BLOCK_FLAG_TYPE_P|BLOCK_FLAG_TYPE_B); + } + cc_Flush( &p_sys->cc ); + return p_cc; +} + /***************************************************************************** * GetAR: Get aspect ratio *****************************************************************************/ @@ -635,31 +732,6 @@ static void GetAR( decoder_t *p_dec ) if( p_dec->fmt_in.video.i_aspect ) { p_sys->i_aspect = p_dec->fmt_in.video.i_aspect; - if( p_sys->i_aspect <= AR_221_1_PICTURE ) - switch( p_sys->i_aspect ) - { - case AR_4_3_PICTURE: - p_sys->i_aspect = VOUT_ASPECT_FACTOR * 4 / 3; - p_sys->i_sar_num = p_sys->p_info->sequence->picture_height * 4; - p_sys->i_sar_den = p_sys->p_info->sequence->picture_width * 3; - break; - case AR_16_9_PICTURE: - p_sys->i_aspect = VOUT_ASPECT_FACTOR * 16 / 9; - p_sys->i_sar_num = p_sys->p_info->sequence->picture_height * 16; - p_sys->i_sar_den = p_sys->p_info->sequence->picture_width * 9; - break; - case AR_221_1_PICTURE: - p_sys->i_aspect = VOUT_ASPECT_FACTOR * 221 / 100; - p_sys->i_sar_num = p_sys->p_info->sequence->picture_height * 221; - p_sys->i_sar_den = p_sys->p_info->sequence->picture_width * 100; - break; - case AR_SQUARE_PICTURE: - p_sys->i_aspect = VOUT_ASPECT_FACTOR * - p_sys->p_info->sequence->picture_width / - p_sys->p_info->sequence->picture_height; - p_sys->i_sar_num = p_sys->i_sar_den = 1; - break; - } } else {