cvd.c
上传用户:riyaled888
上传日期:2009-03-27
资源大小:7338k
文件大小:10k
源码类别:

多媒体

开发平台:

MultiPlatform

  1. /*****************************************************************************
  2.  * cvd.c : CVD Subtitle decoder thread
  3.  *****************************************************************************
  4.  * Copyright (C) 2003, 2004 VideoLAN
  5.  * $Id: cvd.c 8709 2004-09-15 15:50:54Z gbazin $
  6.  *
  7.  * Authors: Rocky Bernstein
  8.  *   based on code from:
  9.  *       Julio Sanchez Fernandez (http://subhandler.sourceforge.net)
  10.  *       Samuel Hocevar <sam@zoy.org>
  11.  *       Laurent Aimar <fenrir@via.ecp.fr>
  12.  *
  13.  * This program is free software; you can redistribute it and/or modify
  14.  * it under the terms of the GNU General Public License as published by
  15.  * the Free Software Foundation; either version 2 of the License, or
  16.  * (at your option) any later version.
  17.  *
  18.  * This program is distributed in the hope that it will be useful,
  19.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21.  * GNU General Public License for more details.
  22.  *
  23.  * You should have received a copy of the GNU General Public License
  24.  * along with this program; if not, write to the Free Software
  25.  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  26.  *****************************************************************************/
  27. /*****************************************************************************
  28.  * Preamble
  29.  *****************************************************************************/
  30. #include <vlc/vlc.h>
  31. #include <vlc/vout.h>
  32. #include <vlc/decoder.h>
  33. #include "subtitle.h"
  34. #include "cvd.h"
  35. #include "common.h"
  36. /*****************************************************************************
  37.  * Module descriptor.
  38.  *****************************************************************************/
  39. static int  VCDSubOpen   ( vlc_object_t * );
  40. static int  PacketizerOpen( vlc_object_t * );
  41. vlc_module_begin();
  42.     set_description( _("CVD subtitle decoder") );
  43.     set_capability( "decoder", 50 );
  44.     set_callbacks( VCDSubOpen, VCDSubClose );
  45.     add_integer ( MODULE_STRING "-debug", 0, NULL,
  46.                   DEBUG_TEXT, DEBUG_LONGTEXT, VLC_TRUE );
  47.     add_integer ( MODULE_STRING "-horizontal-correct", 0, NULL,
  48.                   HORIZONTAL_CORRECT, HORIZONTAL_CORRECT_LONGTEXT, VLC_FALSE );
  49.     add_integer ( MODULE_STRING "-vertical-correct", 0, NULL,
  50.                   VERTICAL_CORRECT, VERTICAL_CORRECT_LONGTEXT, VLC_FALSE );
  51.     add_string( MODULE_STRING "-aspect-ratio", "", NULL,
  52.                 SUB_ASPECT_RATIO_TEXT, SUB_ASPECT_RATIO_LONGTEXT,
  53.                 VLC_TRUE );
  54.     add_integer( MODULE_STRING "-duration-scaling", 3, NULL,
  55.                  DURATION_SCALE_TEXT, DURATION_SCALE_LONGTEXT,
  56.                  VLC_TRUE );
  57.     add_submodule();
  58.     set_description( _("Chaoji VCD subtitle packetizer") );
  59.     set_capability( "packetizer", 50 );
  60.     set_callbacks( PacketizerOpen, VCDSubClose );
  61. vlc_module_end();
  62. /*****************************************************************************
  63.  * Local prototypes
  64.  *****************************************************************************/
  65. static block_t *Reassemble( decoder_t *, block_t ** );
  66. static subpicture_t *Decode( decoder_t *, block_t ** );
  67. static block_t *Packetize( decoder_t *, block_t ** );
  68. /*****************************************************************************
  69.  * VCDSubOpen
  70.  *****************************************************************************
  71.  * Tries to launch a decoder and return score so that the interface is able
  72.  * to chose.
  73.  *****************************************************************************/
  74. static int
  75. VCDSubOpen( vlc_object_t *p_this )
  76. {
  77.     decoder_t     *p_dec = (decoder_t*)p_this;
  78.     decoder_sys_t *p_sys;
  79.     if( p_dec->fmt_in.i_codec != VLC_FOURCC( 'c','v','d',' ' ) )
  80.     {
  81.         return VLC_EGENERIC;
  82.     }
  83.     p_dec->p_sys = p_sys = malloc( sizeof( decoder_sys_t ) );
  84.     p_sys->i_debug       = config_GetInt( p_this, MODULE_STRING "-debug" );
  85.     p_sys->b_packetizer  = VLC_FALSE;
  86.     p_sys->p_vout        = NULL;
  87.     p_sys->i_image       = -1;
  88.     p_sys->subtitle_data = NULL;
  89.     VCDSubInitSubtitleBlock( p_sys );
  90.     es_format_Init( &p_dec->fmt_out, SPU_ES, VLC_FOURCC( 'c','v','d',' ' ) );
  91.     p_dec->pf_decode_sub = Decode;
  92.     p_dec->pf_packetize  = Packetize;
  93.     dbg_print( (DECODE_DBG_CALL|DECODE_DBG_EXT) , "");
  94.     return VLC_SUCCESS;
  95. }
  96. /*****************************************************************************
  97.  * PacketizerOpen
  98.  *****************************************************************************
  99.  * Tries to launch a decoder and return score so that the interface is able
  100.  * to chose.
  101.  *****************************************************************************/
  102. static int PacketizerOpen( vlc_object_t *p_this )
  103. {
  104.     decoder_t *p_dec = (decoder_t*)p_this;
  105.     if( VCDSubOpen( p_this ) )
  106.     {
  107.         return VLC_EGENERIC;
  108.     }
  109.     p_dec->p_sys->b_packetizer = VLC_TRUE;
  110.     return VLC_SUCCESS;
  111. }
  112. /*****************************************************************************
  113.  * Decode:
  114.  *****************************************************************************/
  115. static subpicture_t *
  116. Decode ( decoder_t *p_dec, block_t **pp_block )
  117. {
  118.     decoder_sys_t *p_sys = p_dec->p_sys;
  119.     block_t       *p_spu = Reassemble( p_dec, pp_block );
  120.     vout_thread_t *p_last_vout = p_dec->p_sys->p_vout;
  121.     dbg_print( (DECODE_DBG_CALL) , "");
  122.     if( p_spu )
  123.     {
  124.         p_sys->i_spu = block_ChainExtract( p_spu, p_sys->buffer, 65536 );
  125.         p_sys->i_pts = p_spu->i_pts;
  126.         block_ChainRelease( p_spu );
  127.         if( ( p_sys->p_vout = VCDSubFindVout( p_dec ) ) )
  128.         {
  129.             if( p_last_vout != p_sys->p_vout )
  130.             {
  131.                 spu_Control( p_sys->p_vout->p_spu, SPU_CHANNEL_REGISTER,
  132.                              &p_sys->i_subpic_channel );
  133.             }
  134.             /* Parse and decode */
  135.             E_(ParsePacket)( p_dec );
  136.             vlc_object_release( p_sys->p_vout );
  137.         }
  138.         VCDSubInitSubtitleBlock ( p_sys );
  139.     }
  140.     return NULL;
  141. }
  142. /*****************************************************************************
  143.  * Packetize:
  144.  *****************************************************************************/
  145. static block_t *
  146. Packetize( decoder_t *p_dec, block_t **pp_block )
  147. {
  148.     decoder_sys_t *p_sys = p_dec->p_sys;
  149.     block_t       *p_spu = Reassemble( p_dec, pp_block );
  150.     if( p_spu )
  151.     {
  152.         p_spu->i_dts = p_spu->i_pts;
  153.         p_spu->i_length = 0;
  154.         VCDSubInitSubtitleBlock( p_sys );
  155.         return block_ChainGather( p_spu );
  156.     }
  157.     return NULL;
  158. }
  159. /* following functions are local */
  160. #define SPU_HEADER_LEN 1
  161. /*****************************************************************************
  162.  Reassemble:
  163.  Data for single screen subtitle may come in several non-contiguous
  164.  packets of a stream. This routine is called when the next packet in
  165.  the stream comes in. The job of this routine is to parse the header,
  166.  if this is the beginning, and combine the packets into one complete
  167.  subtitle unit.
  168.  If everything is complete, we will return a block. Otherwise return
  169.  NULL.
  170.  *****************************************************************************/
  171. static block_t *
  172. Reassemble( decoder_t *p_dec, block_t **pp_block )
  173. {
  174.     decoder_sys_t *p_sys = p_dec->p_sys;
  175.     block_t *p_block;
  176.     uint8_t *p_buffer;
  177.     if( pp_block == NULL || *pp_block == NULL )
  178.     {
  179.         return NULL;
  180.     }
  181.     p_block = *pp_block;
  182.     *pp_block = NULL;
  183.     if( p_block->i_buffer < SPU_HEADER_LEN )
  184.     {
  185.       msg_Dbg( p_dec, "invalid packet header (size %d < %d)" ,
  186.                p_block->i_buffer, SPU_HEADER_LEN );
  187.       block_Release( p_block );
  188.       return NULL;
  189.     }
  190.     p_buffer = p_block->p_buffer;
  191.     dbg_print( (DECODE_DBG_CALL|DECODE_DBG_PACKET),
  192.                "header: 0x%02x 0x%02x 0x%02x 0x%02x, 0x%02x, 0x%02x, size: %i",
  193.                p_buffer[1], p_buffer[2], p_buffer[3], p_buffer[4],
  194.                p_buffer[5], p_buffer[6],
  195.                p_block->i_buffer);
  196.     /* Attach to our input thread and see if subtitle is selected. */
  197.     {
  198.         vlc_object_t * p_input;
  199.         vlc_value_t val;
  200.         p_input = vlc_object_find( p_dec, VLC_OBJECT_INPUT, FIND_PARENT );
  201.         if( !p_input ) return NULL;
  202.         if( var_Get( p_input, "spu-channel", &val ) )
  203.         {
  204.           vlc_object_release( p_input );
  205.           return NULL;
  206.         }
  207.         vlc_object_release( p_input );
  208.         /* Number could be 0bd, 1bd, 2bd, 3bd for 0..3. If so
  209.            reduce it to 0..3.
  210.          */
  211.         if ( (val.i_int & 0xff) == 0xbd ) val.i_int >>= 8;
  212.         if( val.i_int == -1 || val.i_int != p_buffer[0] )
  213.           return NULL;
  214.     }
  215.     /* From the scant data on the format, there is only only way known
  216.        to detect the first packet in a subtitle.  The first packet
  217.        seems to have a valid PTS while later packets for the same
  218.        image don't. */
  219.     if ( p_sys->state == SUBTITLE_BLOCK_EMPTY && p_block->i_pts == 0 ) {
  220.       msg_Warn( p_dec,
  221.                 "first packet expected but no PTS present -- skippedn");
  222.       return NULL;
  223.     }
  224.     if ( p_sys->subtitle_data_pos == 0 ) {
  225.       /* First packet in the subtitle block */
  226.       E_(ParseHeader)( p_dec, p_buffer, p_block );
  227.       VCDSubInitSubtitleData(p_sys);
  228.     }
  229.     /* FIXME - remove append_data and use chainappend */
  230.     VCDSubAppendData( p_dec, p_buffer + SPU_HEADER_LEN,
  231.                       p_block->i_buffer - SPU_HEADER_LEN );
  232.     block_ChainAppend( &p_sys->p_block, p_block );
  233.     p_sys->i_spu += p_block->i_buffer - SPU_HEADER_LEN;
  234.     if ( p_sys->subtitle_data_pos == p_sys->i_spu_size ) {
  235.       E_(ParseMetaInfo)( p_dec );
  236.       return p_sys->p_block;
  237.     } else {
  238.       /* Not last block in subtitle, so wait for another. */
  239.       p_sys->state = SUBTITLE_BLOCK_PARTIAL;
  240.     }
  241.     return NULL;
  242. }