1 /*****************************************************************************
2 * xxmc.c : HW MPEG decoder thread
3 *****************************************************************************
4 * Copyright (C) 2000-2001 VideoLAN
7 * Authors: Samuel Hocevar <sam@zoy.org>
8 * Laurent Aimar <fenrir@via.ecp.fr>
10 * This program is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
20 * You should have received a copy of the GNU General Public License
21 * along with this program; if not, write to the Free Software
22 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
23 *****************************************************************************/
25 /*****************************************************************************
27 *****************************************************************************/
32 #include <vlc_common.h>
33 #include <vlc_plugin.h>
34 #include <vlc_codec.h>
35 #include <vlc_codec_synchro.h>
44 //#include "attributes.h"
45 #include "mpeg2_internal.h"
46 //#include "xvmc_vld.h"
48 /* Aspect ratio (ISO/IEC 13818-2 section 6.3.3, table 6-3) */
49 #define AR_SQUARE_PICTURE 1 /* square pixels */
50 #define AR_3_4_PICTURE 2 /* 3:4 picture (TV) */
51 #define AR_16_9_PICTURE 3 /* 16:9 picture (wide screen) */
52 #define AR_221_1_PICTURE 4 /* 2.21:1 picture (movie) */
54 /*****************************************************************************
55 * decoder_sys_t : libmpeg2 decoder descriptor
56 *****************************************************************************/
62 mpeg2dec_t *p_mpeg2dec;
63 const mpeg2_info_t *p_info;
70 mtime_t i_previous_pts;
71 mtime_t i_current_pts;
72 mtime_t i_previous_dts;
73 mtime_t i_current_dts;
75 picture_t * p_picture_to_destroy;
77 bool b_after_sequence_header; /* is it the next frame after
78 * the sequence header ? */
79 bool b_slice_i; /* intra-slice refresh stream */
84 decoder_synchro_t *p_synchro;
86 mtime_t i_last_frame_pts;
89 /*****************************************************************************
91 *****************************************************************************/
93 static int OpenDecoder( vlc_object_t * );
94 static void CloseDecoder( vlc_object_t * );
96 static picture_t *DecodeBlock( decoder_t *, block_t ** );
98 static picture_t *GetNewPicture( decoder_t *, uint8_t ** );
100 /*****************************************************************************
102 *****************************************************************************/
104 set_description( N_("MPEG I/II hw video decoder (using libmpeg2)") )
105 set_capability( "decoder", 140 )
106 set_callbacks( OpenDecoder, CloseDecoder )
107 add_shortcut( "xxmc" )
110 /*****************************************************************************
111 * OpenDecoder: probe the decoder and return score
112 *****************************************************************************/
113 static int OpenDecoder( vlc_object_t *p_this )
116 decoder_t *p_dec = (decoder_t*)p_this;
117 decoder_sys_t *p_sys = NULL;
118 uint32_t i_accel = 0;
124 if( p_dec->fmt_in.i_codec != VLC_CODEC_MPGV )
126 /* Select onl recognized original format (standard mpeg video) */
127 switch( p_dec->fmt_in.i_original_fourcc )
129 case VLC_FOURCC('m','p','g','1'):
130 case VLC_FOURCC('m','p','g','2'):
131 case VLC_FOURCC('m','p','g','v'):
132 /* Pinnacle hardware-mpeg1 */
133 case VLC_FOURCC('P','I','M','1'):
134 /* VIA hardware-mpeg2 */
135 case VLC_FOURCC('X','x','M','C'):
137 case VLC_FOURCC('V','C','R','2'):
140 if( p_dec->fmt_in.i_original_fourcc )
145 msg_Dbg(p_dec, "OpenDecoder Entering");
147 /* Allocate the memory needed to store the decoder's structure */
148 p_dec->p_sys = p_sys = calloc( 1, sizeof(*p_sys) );
152 /* Initialize the thread properties */
153 p_sys->p_mpeg2dec = NULL;
154 p_sys->p_synchro = NULL;
155 p_sys->p_info = NULL;
156 p_sys->i_pts = mdate() + DEFAULT_PTS_DELAY;
157 p_sys->i_current_pts = 0;
158 p_sys->i_previous_pts = 0;
159 p_sys->i_current_dts = 0;
160 p_sys->i_previous_dts = 0;
161 p_sys->p_picture_to_destroy = NULL;
162 p_sys->b_garbage_pic = 0;
163 p_sys->b_slice_i = 0;
166 #if defined( __i386__ )
167 if( vlc_CPU() & CPU_CAPABILITY_MMX )
169 i_accel |= MPEG2_ACCEL_X86_MMX;
172 if( vlc_CPU() & CPU_CAPABILITY_3DNOW )
174 i_accel |= MPEG2_ACCEL_X86_3DNOW;
177 if( vlc_CPU() & CPU_CAPABILITY_MMXEXT )
179 i_accel |= MPEG2_ACCEL_X86_MMXEXT;
182 #elif defined( __powerpc__ ) || defined( SYS_DARWIN )
183 if( vlc_CPU() & CPU_CAPABILITY_ALTIVEC )
185 i_accel |= MPEG2_ACCEL_PPC_ALTIVEC;
189 /* If we do not know this CPU, trust libmpeg2's feature detection */
190 i_accel = MPEG2_ACCEL_DETECT;
194 /* Set CPU acceleration features */
195 mpeg2_accel( i_accel );
197 /* Initialize decoder */
198 p_sys->p_mpeg2dec = mpeg2_init();
199 if( p_sys->p_mpeg2dec == NULL)
201 msg_Err( p_dec, "mpeg2_init() failed" );
206 p_sys->p_info = mpeg2_info( p_sys->p_mpeg2dec );
208 p_dec->pf_decode_video = DecodeBlock;
209 p_dec->fmt_out.i_cat = VIDEO_ES;
210 p_dec->fmt_out.i_codec = 0;
212 f_wd_dec = fopen("/vlc/dec_pid", "w");
213 if (f_wd_dec != NULL)
215 fprintf(f_wd_dec, "%d\n", getpid());
220 msg_Dbg(p_dec, "OpenDecoder Leaving");
225 static void WriteDecodeFile(int value)
229 f_wd_ok = fopen("/vlc/dec_ok", "w");
232 fprintf(f_wd_ok, "%d", value);
239 /*****************************************************************************
240 * RunDecoder: the libmpeg2 decoder
241 *****************************************************************************/
242 static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
244 decoder_sys_t *p_sys = p_dec->p_sys;
250 if( !pp_block || !*pp_block )
256 state = mpeg2_parse( p_sys->p_mpeg2dec );
260 if( !p_block->i_buffer )
262 block_Release( p_block );
265 if( (p_block->i_flags&BLOCK_FLAG_DISCONTINUITY) &&
267 p_sys->p_info->sequence &&
268 p_sys->p_info->sequence->width != (unsigned int)-1 )
270 decoder_SynchroReset( p_sys->p_synchro );
271 if( p_sys->p_info->current_fbuf != NULL
272 && p_sys->p_info->current_fbuf->id != NULL )
274 p_sys->b_garbage_pic = 1;
275 p_pic = p_sys->p_info->current_fbuf->id;
280 buf[0] = buf[1] = buf[2] = NULL;
281 if( (p_pic = GetNewPicture( p_dec, buf )) == NULL )
283 mpeg2_set_buf( p_sys->p_mpeg2dec, buf, p_pic );
285 p_sys->p_picture_to_destroy = p_pic;
287 if ( p_sys->b_slice_i )
289 decoder_SynchroNewPicture( p_sys->p_synchro,
290 I_CODING_TYPE, 2, 0, 0,
291 p_sys->p_info->sequence->flags & SEQ_FLAG_LOW_DELAY );
292 decoder_SynchroDecode( p_sys->p_synchro );
293 decoder_SynchroEnd( p_sys->p_synchro, I_CODING_TYPE, 0 );
300 mpeg2_pts( p_sys->p_mpeg2dec, (uint32_t)p_block->i_pts );
302 #else /* New interface */
303 if( p_block->i_pts || p_block->i_dts )
305 mpeg2_tag_picture( p_sys->p_mpeg2dec,
306 (uint32_t)p_block->i_pts,
307 (uint32_t)p_block->i_dts );
309 p_sys->i_previous_pts = p_sys->i_current_pts;
310 p_sys->i_current_pts = p_block->i_pts;
311 p_sys->i_previous_dts = p_sys->i_current_dts;
312 p_sys->i_current_dts = p_block->i_dts;
314 p_sys->i_current_rate = p_block->i_rate;
316 mpeg2_buffer( p_sys->p_mpeg2dec, p_block->p_buffer,
317 p_block->p_buffer + p_block->i_buffer );
318 p_block->i_buffer = 0;
323 /* Initialize video output */
325 buf[0] = buf[1] = buf[2] = NULL;
327 /* Check whether the input gave a particular aspect ratio */
328 if( p_dec->fmt_in.video.i_aspect )
330 p_sys->i_aspect = p_dec->fmt_in.video.i_aspect;
331 if( p_sys->i_aspect <= AR_221_1_PICTURE )
332 switch( p_sys->i_aspect )
335 p_sys->i_aspect = VOUT_ASPECT_FACTOR * 4 / 3;
337 case AR_16_9_PICTURE:
338 p_sys->i_aspect = VOUT_ASPECT_FACTOR * 16 / 9;
340 case AR_221_1_PICTURE:
341 p_sys->i_aspect = VOUT_ASPECT_FACTOR * 221 / 100;
343 case AR_SQUARE_PICTURE:
344 p_sys->i_aspect = VOUT_ASPECT_FACTOR *
345 p_sys->p_info->sequence->width /
346 p_sys->p_info->sequence->height;
352 /* Use the value provided in the MPEG sequence header */
353 if( p_sys->p_info->sequence->pixel_height > 0 )
356 ((uint64_t)p_sys->p_info->sequence->display_width) *
357 p_sys->p_info->sequence->pixel_width *
359 p_sys->p_info->sequence->display_height /
360 p_sys->p_info->sequence->pixel_height;
364 /* Invalid aspect, assume 4:3.
365 * This shouldn't happen and if it does it is a bug
366 * in libmpeg2 (likely triggered by an invalid stream) */
367 p_sys->i_aspect = VOUT_ASPECT_FACTOR * 4 / 3;
371 msg_Dbg( p_dec, "%dx%d, aspect %d, %u.%03u fps",
372 p_sys->p_info->sequence->width,
373 p_sys->p_info->sequence->height, p_sys->i_aspect,
374 (uint32_t)((uint64_t)1001000000 * 27 /
375 p_sys->p_info->sequence->frame_period / 1001),
376 (uint32_t)((uint64_t)1001000000 * 27 /
377 p_sys->p_info->sequence->frame_period % 1001) );
379 mpeg2_custom_fbuf( p_sys->p_mpeg2dec, 1 );
381 /* Set the first 2 reference frames */
382 mpeg2_set_buf( p_sys->p_mpeg2dec, buf, NULL );
384 if( (p_pic = GetNewPicture( p_dec, buf )) == NULL )
386 block_Release( p_block );
389 mpeg2_set_buf( p_sys->p_mpeg2dec, buf, p_pic );
392 decoder_LinkPicture( p_dec, p_pic );
394 if( p_sys->p_synchro )
396 decoder_SynchroRelease( p_sys->p_synchro );
398 p_sys->p_synchro = decoder_SynchroInit( p_dec,
399 (uint32_t)((uint64_t)1001000000 * 27 /
400 p_sys->p_info->sequence->frame_period) );
401 p_sys->b_after_sequence_header = 1;
405 case STATE_PICTURE_2ND:
406 decoder_SynchroNewPicture( p_sys->p_synchro,
407 p_sys->p_info->current_picture->flags & PIC_MASK_CODING_TYPE,
408 p_sys->p_info->current_picture->nb_fields,
410 p_sys->p_info->sequence->flags & SEQ_FLAG_LOW_DELAY );
414 decoder_SynchroTrash( p_sys->p_synchro );
418 decoder_SynchroDecode( p_sys->p_synchro );
426 buf[0] = buf[1] = buf[2] = NULL;
428 if ( p_sys->b_after_sequence_header &&
429 ((p_sys->p_info->current_picture->flags &
430 PIC_MASK_CODING_TYPE) == PIC_FLAG_CODING_TYPE_P) )
432 /* Intra-slice refresh. Simulate a blank I picture. */
433 msg_Dbg( p_dec, "intra-slice refresh stream" );
434 decoder_SynchroNewPicture( p_sys->p_synchro,
435 I_CODING_TYPE, 2, 0, 0,
436 p_sys->p_info->sequence->flags & SEQ_FLAG_LOW_DELAY );
437 decoder_SynchroDecode( p_sys->p_synchro );
438 decoder_SynchroEnd( p_sys->p_synchro, I_CODING_TYPE, 0 );
439 p_sys->b_slice_i = 1;
441 p_sys->b_after_sequence_header = 0;
444 i_pts = p_sys->p_info->current_picture->flags & PIC_FLAG_PTS ?
445 ( ( p_sys->p_info->current_picture->pts ==
446 (uint32_t)p_sys->i_current_pts ) ?
447 p_sys->i_current_pts : p_sys->i_previous_pts ) : 0;
448 #else /* New interface */
449 i_pts = p_sys->p_info->current_picture->flags & PIC_FLAG_TAGS ?
450 ( ( p_sys->p_info->current_picture->tag ==
451 (uint32_t)p_sys->i_current_pts ) ?
452 p_sys->i_current_pts : p_sys->i_previous_pts ) : 0;
454 /* Hack to handle demuxers which only have DTS timestamps */
455 if( !i_pts && !p_block->i_pts && p_block->i_dts > 0 )
457 if( p_sys->p_info->sequence->flags & SEQ_FLAG_LOW_DELAY ||
458 (p_sys->p_info->current_picture->flags &
459 PIC_MASK_CODING_TYPE) == PIC_FLAG_CODING_TYPE_B )
461 i_pts = p_block->i_dts;
464 p_block->i_pts = p_block->i_dts = 0;
467 decoder_SynchroNewPicture( p_sys->p_synchro,
468 p_sys->p_info->current_picture->flags & PIC_MASK_CODING_TYPE,
469 p_sys->p_info->current_picture->nb_fields, i_pts,
471 p_sys->p_info->sequence->flags & SEQ_FLAG_LOW_DELAY );
473 if ( !(p_sys->b_slice_i
474 && ((p_sys->p_info->current_picture->flags
475 & PIC_MASK_CODING_TYPE) == P_CODING_TYPE))
476 && !decoder_SynchroChoose( p_sys->p_synchro,
477 p_sys->p_info->current_picture->flags
478 & PIC_MASK_CODING_TYPE,
479 /*FindVout(p_dec)->render_time*/ 0 /*FIXME*/,
480 p_sys->p_info->sequence->flags & SEQ_FLAG_LOW_DELAY ) )
482 mpeg2_skip( p_sys->p_mpeg2dec, 1 );
484 decoder_SynchroTrash( p_sys->p_synchro );
485 mpeg2_set_buf( p_sys->p_mpeg2dec, buf, NULL );
489 mpeg2_skip( p_sys->p_mpeg2dec, 0 );
491 decoder_SynchroDecode( p_sys->p_synchro );
493 if( (p_pic = GetNewPicture( p_dec, buf )) == NULL )
495 block_Release( p_block );
499 //p_sys->p_mpeg2dec->ptr_forward_ref_picture = p_sys->p_mpeg2dec->fbuf[2]->id;
500 //p_sys->p_mpeg2dec->ptr_backward_ref_picture = p_sys->p_mpeg2dec->fbuf[1]->id;
502 if ((p_sys->p_info->current_picture->flags & PIC_MASK_CODING_TYPE) != B_CODING_TYPE)
504 //if (p_sys->p_mpeg2dec->ptr_forward_ref_picture &&
505 // p_sys->p_mpeg2dec->ptr_forward_ref_picture != picture->backward_reference_frame)
506 // p_pic->forward_reference_frame->free (p_pic->forward_reference_frame);
508 //p_sys->p_mpeg2dec->ptr_forward_ref_picture =
509 // p_sys->p_mpeg2dec->ptr_backward_ref_picture;
510 //p_sys->p_mpeg2dec->ptr_backward_ref_picture = (void *)p_pic;
512 mpeg2_set_buf( p_sys->p_mpeg2dec, buf, p_pic );
520 if( p_sys->p_info->display_fbuf
521 && p_sys->p_info->display_fbuf->id )
523 p_pic = (picture_t *)p_sys->p_info->display_fbuf->id;
525 decoder_SynchroEnd( p_sys->p_synchro,
526 p_sys->p_info->display_picture->flags
527 & PIC_MASK_CODING_TYPE,
528 p_sys->b_garbage_pic );
529 p_sys->b_garbage_pic = 0;
531 if ( p_sys->p_picture_to_destroy != p_pic )
533 p_pic->date = decoder_SynchroDate( p_sys->p_synchro );
537 p_sys->p_picture_to_destroy = NULL;
542 if( p_sys->p_info->discard_fbuf &&
543 p_sys->p_info->discard_fbuf->id )
545 decoder_UnlinkPicture( p_dec, p_sys->p_info->discard_fbuf->id );
547 /* For still frames */
548 //if( state == STATE_END && p_pic ) p_pic->b_force = true;
553 /* Avoid frames with identical timestamps.
554 * Especially needed for still frames in DVD menus. */
555 if( p_sys->i_last_frame_pts == p_pic->date ) p_pic->date++;
556 p_sys->i_last_frame_pts = p_pic->date;
565 buf[0] = buf[1] = buf[2] = NULL;
567 msg_Warn( p_dec, "invalid picture encountered" );
568 if ( ( p_sys->p_info->current_picture == NULL ) ||
569 ( ( p_sys->p_info->current_picture->flags &
570 PIC_MASK_CODING_TYPE) != B_CODING_TYPE ) )
572 if( p_sys->p_synchro ) decoder_SynchroReset( p_sys->p_synchro );
574 mpeg2_skip( p_sys->p_mpeg2dec, 1 );
577 if( p_sys->p_info->current_fbuf &&
578 p_sys->p_info->current_fbuf->id )
580 p_sys->b_garbage_pic = 1;
581 p_pic = p_sys->p_info->current_fbuf->id;
583 else if( !p_sys->p_info->sequence )
589 if( (p_pic = GetNewPicture( p_dec, buf )) == NULL )
591 mpeg2_set_buf( p_sys->p_mpeg2dec, buf, p_pic );
593 p_sys->p_picture_to_destroy = p_pic;
595 memset( p_pic->p[0].p_pixels, 0,
596 p_sys->p_info->sequence->width
597 * p_sys->p_info->sequence->height );
598 memset( p_pic->p[1].p_pixels, 0x80,
599 p_sys->p_info->sequence->width
600 * p_sys->p_info->sequence->height / 4 );
601 memset( p_pic->p[2].p_pixels, 0x80,
602 p_sys->p_info->sequence->width
603 * p_sys->p_info->sequence->height / 4 );
605 if( p_sys->b_slice_i )
607 decoder_SynchroNewPicture( p_sys->p_synchro,
608 I_CODING_TYPE, 2, 0, 0,
609 p_sys->p_info->sequence->flags & SEQ_FLAG_LOW_DELAY );
610 decoder_SynchroDecode( p_sys->p_synchro );
611 decoder_SynchroEnd( p_sys->p_synchro, I_CODING_TYPE, 0 );
623 /*****************************************************************************
624 * CloseDecoder: libmpeg2 decoder destruction
625 *****************************************************************************/
626 static void CloseDecoder( vlc_object_t *p_this )
628 decoder_t *p_dec = (decoder_t *)p_this;
629 decoder_sys_t *p_sys = p_dec->p_sys;
632 if( p_sys->p_synchro ) decoder_SynchroRelease( p_sys->p_synchro );
633 if( p_sys->p_mpeg2dec ) mpeg2_close( p_sys->p_mpeg2dec );
635 f_wd_dec = fopen("/vlc/dec_pid", "w");
636 if (f_wd_dec != NULL)
638 fprintf(f_wd_dec, "0\n");
646 static double get_aspect_ratio( decoder_t *p_dec )
648 decoder_sys_t *p_sys = p_dec->p_sys;
650 double mpeg1_pel_ratio[16] = {1.0 /* forbidden */,
651 1.0, 0.6735, 0.7031, 0.7615, 0.8055, 0.8437, 0.8935, 0.9157,
652 0.9815, 1.0255, 1.0695, 1.0950, 1.1575, 1.2015, 1.0 /*reserved*/ };
654 if( !p_sys->p_mpeg2dec->decoder.mpeg1 )
656 /* these hardcoded values are defined on mpeg2 standard for
657 * aspect ratio. other values are reserved or forbidden. */
658 /*switch( p_sys->p_mpeg2dec->decoder.aspect_ratio_information )
671 ratio = (double)p_sys->p_mpeg2dec->decoder.width/(double)p_sys->p_mpeg2dec->decoder.height;
677 /* mpeg1 constants refer to pixel aspect ratio */
678 ratio = (double)p_sys->p_mpeg2dec->decoder.width/(double)p_sys->p_mpeg2dec->decoder.height;
679 /* ratio /= mpeg1_pel_ratio[p_sys->p_mpeg2dec->decoder.aspect_ratio_information]; */
685 /*****************************************************************************
686 * GetNewPicture: Get a new picture from the vout and set the buf struct
687 *****************************************************************************/
688 static picture_t *GetNewPicture( decoder_t *p_dec, uint8_t **pp_buf )
690 //msg_Dbg(p_dec, "GetNewPicture Entering");
691 decoder_sys_t *p_sys = p_dec->p_sys;
694 p_dec->fmt_out.video.i_width = p_sys->p_info->sequence->width;
695 p_dec->fmt_out.video.i_height = p_sys->p_info->sequence->height;
696 p_dec->fmt_out.video.i_aspect = p_sys->i_aspect;
698 if( p_sys->p_info->sequence->frame_period > 0 )
700 p_dec->fmt_out.video.i_frame_rate =
701 (uint32_t)( (uint64_t)1001000000 * 27 /
702 p_sys->p_info->sequence->frame_period );
703 p_dec->fmt_out.video.i_frame_rate_base = 1001;
706 p_dec->fmt_out.i_codec =
707 ( p_sys->p_info->sequence->chroma_height <
708 p_sys->p_info->sequence->height ) ?
709 VLC_CODEC_I420 : VLC_CODEC_I422;
712 p_sys->f_wd_nb = fopen("/vlc/dec_nb", "w");
713 if (p_sys->f_wd_nb != NULL)
715 // fprintf(p_sys->f_wd_nb, "%d\n", mdate());
716 fprintf(p_sys->f_wd_nb, "%s\n", mdate());
717 fflush(p_sys->f_wd_nb);
720 p_pic = decoder_NewPicture( p_dec );
722 if( p_pic == NULL ) return NULL;
724 p_pic->b_progressive = p_sys->p_info->current_picture != NULL ?
725 p_sys->p_info->current_picture->flags & PIC_FLAG_PROGRESSIVE_FRAME : 1;
726 p_pic->b_top_field_first = p_sys->p_info->current_picture != NULL ?
727 p_sys->p_info->current_picture->flags & PIC_FLAG_TOP_FIELD_FIRST : 1;
728 p_pic->i_nb_fields = p_sys->p_info->current_picture != NULL ?
729 p_sys->p_info->current_picture->nb_fields : 2;
730 p_pic->format.i_frame_rate = p_dec->fmt_out.video.i_frame_rate;
731 p_pic->format.i_frame_rate_base = p_dec->fmt_out.video.i_frame_rate_base;
733 decoder_LinkPicture( p_dec, p_pic );
735 pp_buf[0] = p_pic->p[0].p_pixels;
736 pp_buf[1] = p_pic->p[1].p_pixels;
737 pp_buf[2] = p_pic->p[2].p_pixels;
739 /* let driver ensure this image has the right format */
741 this->driver->update_frame_format( p_pic->p_sys->p_vout, p_pic,
742 p_dec->fmt_out.video.i_width,
743 p_dec->fmt_out.video.i_height,
744 p_dec->fmt_out.video.i_aspect,
746 mpeg2_xxmc_choose_coding( p_dec, &p_sys->p_mpeg2dec->decoder, p_pic,
747 get_aspect_ratio(p_dec), 0 );