vc1.c
上传用户:kjfoods
上传日期:2020-07-06
资源大小:29949k
文件大小:7k
源码类别:

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * vc1.c : VC1 Video demuxer
  3.  *****************************************************************************
  4.  * Copyright (C) 2002-2004 the VideoLAN team
  5.  * $Id: 8c3af591962a6efe26fcc7bc1bd0a0f4b8ae88d4 $
  6.  *
  7.  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
  8.  *
  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.
  13.  *
  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.
  18.  *
  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., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22.  *****************************************************************************/
  23. /*****************************************************************************
  24.  * Preamble
  25.  *****************************************************************************/
  26. #ifdef HAVE_CONFIG_H
  27. # include "config.h"
  28. #endif
  29. #include <vlc_common.h>
  30. #include <vlc_plugin.h>
  31. #include <vlc_demux.h>
  32. #include "vlc_codec.h"
  33. /*****************************************************************************
  34.  * Module descriptor
  35.  *****************************************************************************/
  36. static int  Open ( vlc_object_t * );
  37. static void Close( vlc_object_t * );
  38. #define FPS_TEXT N_("Frames per Second")
  39. #define FPS_LONGTEXT N_("Desired frame rate for the VC-1 stream.")
  40. vlc_module_begin ()
  41.     set_shortname( "VC-1")
  42.     set_category( CAT_INPUT )
  43.     set_subcategory( SUBCAT_INPUT_DEMUX )
  44.     set_description( N_("VC1 video demuxer" ) )
  45.     set_capability( "demux", 0 )
  46.     add_float( "vc1-fps", 25.0, NULL, FPS_TEXT, FPS_LONGTEXT, true )
  47.     set_callbacks( Open, Close )
  48.     add_shortcut( "vc1" )
  49. vlc_module_end ()
  50. /*****************************************************************************
  51.  * Local prototypes
  52.  *****************************************************************************/
  53. struct demux_sys_t
  54. {
  55.     mtime_t     i_dts;
  56.     es_out_id_t *p_es;
  57.     float       f_fps;
  58.     decoder_t *p_packetizer;
  59. };
  60. static int Demux( demux_t * );
  61. static int Control( demux_t *, int, va_list );
  62. #define VC1_PACKET_SIZE 4096
  63. /*****************************************************************************
  64.  * Open: initializes demux structures
  65.  *****************************************************************************/
  66. static int Open( vlc_object_t * p_this )
  67. {
  68.     demux_t     *p_demux = (demux_t*)p_this;
  69.     demux_sys_t *p_sys;
  70.     const uint8_t *p_peek;
  71.     es_format_t fmt;
  72.     if( stream_Peek( p_demux->s, &p_peek, 4 ) < 4 ) return VLC_EGENERIC;
  73.     if( p_peek[0] != 0x00 || p_peek[1] != 0x00 ||
  74.         p_peek[2] != 0x01 || p_peek[3] != 0x0f ) /* Sequence header */
  75.     {
  76.         if( !p_demux->b_force )
  77.         {
  78.             msg_Warn( p_demux, "vc-1 module discarded (no startcode)" );
  79.             return VLC_EGENERIC;
  80.         }
  81.         msg_Err( p_demux, "this doesn't look like a VC-1 ES stream, "
  82.                  "continuing anyway" );
  83.     }
  84.     p_demux->pf_demux  = Demux;
  85.     p_demux->pf_control= Control;
  86.     p_demux->p_sys     = p_sys = malloc( sizeof( demux_sys_t ) );
  87.     p_sys->p_es        = NULL;
  88.     p_sys->i_dts       = 1;
  89.     p_sys->f_fps = var_CreateGetFloat( p_demux, "vc1-fps" );
  90.     if( p_sys->f_fps < 0.001 )
  91.         p_sys->f_fps = 0.0;
  92.     /* Load the packetizer */
  93.     es_format_Init( &fmt, VIDEO_ES, VLC_FOURCC( 'W', 'V', 'C', '1' ) );
  94.     p_sys->p_packetizer = demux_PacketizerNew( p_demux, &fmt, "VC-1" );
  95.     if( !p_sys->p_packetizer )
  96.     {
  97.         free( p_sys );
  98.         return VLC_EGENERIC;
  99.     }
  100.     return VLC_SUCCESS;
  101. }
  102. /*****************************************************************************
  103.  * Close: frees unused data
  104.  *****************************************************************************/
  105. static void Close( vlc_object_t * p_this )
  106. {
  107.     demux_t     *p_demux = (demux_t*)p_this;
  108.     demux_sys_t *p_sys = p_demux->p_sys;
  109.     demux_PacketizerDestroy( p_sys->p_packetizer );
  110.     free( p_sys );
  111. }
  112. /*****************************************************************************
  113.  * Demux: reads and demuxes data packets
  114.  *****************************************************************************
  115.  * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
  116.  *****************************************************************************/
  117. static int Demux( demux_t *p_demux)
  118. {
  119.     demux_sys_t *p_sys = p_demux->p_sys;
  120.     block_t *p_block_in, *p_block_out;
  121.     if( ( p_block_in = stream_Block( p_demux->s, VC1_PACKET_SIZE ) ) == NULL )
  122.         return 0;
  123.     /*  */
  124.     p_block_in->i_dts = 1;
  125.     p_block_in->i_pts = 1;
  126.     while( (p_block_out = p_sys->p_packetizer->pf_packetize( p_sys->p_packetizer, &p_block_in )) )
  127.     {
  128.         while( p_block_out )
  129.         {
  130.             block_t *p_next = p_block_out->p_next;
  131.             p_block_out->p_next = NULL;
  132.             if( p_sys->p_es == NULL )
  133.             {
  134.                 p_sys->p_packetizer->fmt_out.b_packetized = true;
  135.                 p_sys->p_es = es_out_Add( p_demux->out, &p_sys->p_packetizer->fmt_out);
  136.             }
  137.             es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_dts );
  138.             p_block_out->i_dts = p_sys->i_dts;
  139.             p_block_out->i_pts = p_sys->i_dts;
  140.             es_out_Send( p_demux->out, p_sys->p_es, p_block_out );
  141.             p_block_out = p_next;
  142.             if( p_sys->p_packetizer->fmt_out.video.i_frame_rate > 0 &&
  143.                 p_sys->p_packetizer->fmt_out.video.i_frame_rate_base > 0 )
  144.                 p_sys->i_dts += INT64_C(1000000) *
  145.                     p_sys->p_packetizer->fmt_out.video.i_frame_rate_base /
  146.                     p_sys->p_packetizer->fmt_out.video.i_frame_rate;
  147.             else if( p_sys->f_fps > 0.001 )
  148.                 p_sys->i_dts += (int64_t)((double)1000000.0 / p_sys->f_fps);
  149.             else
  150.                 p_sys->i_dts += INT64_C(1000000) / 25;
  151.         }
  152.     }
  153.     return 1;
  154. }
  155. /*****************************************************************************
  156.  * Control:
  157.  *****************************************************************************/
  158. static int Control( demux_t *p_demux, int i_query, va_list args )
  159. {
  160.     /* demux_sys_t *p_sys  = p_demux->p_sys; */
  161.     /* FIXME calculate the bitrate */
  162.     if( i_query == DEMUX_SET_TIME )
  163.         return VLC_EGENERIC;
  164.     else
  165.         return demux_vaControlHelper( p_demux->s,
  166.                                        0, -1,
  167.                                        0, 1, i_query, args );
  168. }