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

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * dirac.c : Dirac Video demuxer
  3.  *****************************************************************************
  4.  * Copyright (C) 2002-2008 the VideoLAN team
  5.  * $Id: 253454e5212b9b718ec18d2dae2fb4fe9f2ffa29 $
  6.  *
  7.  * Authors: David Flynn <davidf@rd.bbc.co.uk>
  8.  * Based on vc1.c by: Laurent Aimar <fenrir@via.ecp.fr>
  9.  *
  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.
  14.  *
  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.
  19.  *
  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., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  23.  *****************************************************************************/
  24. /*****************************************************************************
  25.  * Preamble
  26.  *****************************************************************************/
  27. #ifdef HAVE_CONFIG_H
  28. # include "config.h"
  29. #endif
  30. #include <vlc_common.h>
  31. #include <vlc_plugin.h>
  32. #include <vlc_demux.h>
  33. #include "vlc_codec.h"
  34. #define DEMUX_CFG_PREFIX "dirac-"
  35. #define DEMUX_DTSOFFSET "dts-offset"
  36. #define DEMUX_DTSOFFSET_TEXT N_("Value to adjust dts by")
  37. #define DEMUX_DTSOFFSET_LONGTEXT DEMUX_DTSOFFSET_TEXT
  38. /*****************************************************************************
  39.  * Module descriptor
  40.  *****************************************************************************/
  41. static int  Open ( vlc_object_t * );
  42. static void Close( vlc_object_t * );
  43. vlc_module_begin();
  44.     set_shortname( "Dirac");
  45.     set_category( CAT_INPUT );
  46.     set_subcategory( SUBCAT_INPUT_DEMUX );
  47.     set_description( N_("Dirac video demuxer" ) );
  48.     set_capability( "demux", 50 );
  49.     add_integer( DEMUX_CFG_PREFIX DEMUX_DTSOFFSET, 0, NULL,
  50.                  DEMUX_DTSOFFSET_TEXT, DEMUX_DTSOFFSET_LONGTEXT, false )
  51.     set_callbacks( Open, Close );
  52. vlc_module_end();
  53. /*****************************************************************************
  54.  * Local prototypes
  55.  *****************************************************************************/
  56. struct demux_sys_t
  57. {
  58.     mtime_t i_dts;
  59.     mtime_t i_dtsoffset;
  60.     mtime_t i_pts_offset_lowtide;
  61.     es_out_id_t *p_es;
  62.     enum {
  63.         /* demuxer states, do not reorder (++ is used) */
  64.         DIRAC_DEMUX_DISCONT = 0, /* signal a discontinuity to packetizer */
  65.         DIRAC_DEMUX_FIRST, /* provide an origin timestamp for the packetizer */
  66.         DIRAC_DEMUX_STEADY, /* normal operation */
  67.     } i_state;
  68.     decoder_t *p_packetizer;
  69. };
  70. static int Demux( demux_t * );
  71. static int Control( demux_t *, int, va_list );
  72. #define DIRAC_PACKET_SIZE 4096
  73. /*****************************************************************************
  74.  * Open: initializes demux structures
  75.  *****************************************************************************/
  76. static int Open( vlc_object_t * p_this )
  77. {
  78.     demux_t     *p_demux = (demux_t*)p_this;
  79.     demux_sys_t *p_sys;
  80.     const uint8_t *p_peek;
  81.     es_format_t fmt;
  82.     if( stream_Peek( p_demux->s, &p_peek, 5 ) < 5 ) return VLC_EGENERIC;
  83.     if( p_peek[0] != 'B' || p_peek[1] != 'B' ||
  84.         p_peek[2] != 'C' || p_peek[3] != 'D') /* start of ParseInfo */
  85.     {
  86.         if( !p_demux->b_force ) return VLC_EGENERIC;
  87.         msg_Err( p_demux, "This doesn't look like a Dirac stream (incorrect parsecode)" );
  88.         msg_Warn( p_demux, "continuing anyway" );
  89.     }
  90.     p_demux->pf_demux = Demux;
  91.     p_demux->pf_control = Control;
  92.     p_demux->p_sys = p_sys = calloc( 1, sizeof( demux_sys_t ) );
  93.     if( !p_sys ) return VLC_ENOMEM;
  94.     p_sys->i_pts_offset_lowtide = INT64_MAX;
  95.     p_sys->i_state = DIRAC_DEMUX_FIRST;
  96.     p_sys->i_dtsoffset = var_CreateGetInteger( p_demux, DEMUX_CFG_PREFIX DEMUX_DTSOFFSET );
  97.     /* Load the packetizer */
  98.     es_format_Init( &fmt, VIDEO_ES, VLC_FOURCC( 'd','r','a','c' ) );
  99.     p_sys->p_packetizer = demux_PacketizerNew( p_demux, &fmt, "dirac" );
  100.     if( !p_sys->p_packetizer )
  101.     {
  102.         free( p_sys );
  103.         return VLC_EGENERIC;
  104.     }
  105.     return VLC_SUCCESS;
  106. }
  107. /*****************************************************************************
  108.  * Close: frees unused data
  109.  *****************************************************************************/
  110. static void Close( vlc_object_t * p_this )
  111. {
  112.     demux_t     *p_demux = (demux_t*)p_this;
  113.     demux_sys_t *p_sys = p_demux->p_sys;
  114.     demux_PacketizerDestroy( p_sys->p_packetizer );
  115.     if( p_sys->i_pts_offset_lowtide < INT64_MAX &&
  116.         p_sys->i_pts_offset_lowtide > 0 )
  117.     {
  118.         msg_Warn( p_demux, "For all packets seen, pts-dts (%"PRId64") could be reduced to 0",
  119.                   p_sys->i_pts_offset_lowtide );
  120.     }
  121.     free( p_sys );
  122. }
  123. /*****************************************************************************
  124.  * Demux: reads and demuxes data packets
  125.  *****************************************************************************
  126.  * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
  127.  *****************************************************************************/
  128. static int Demux( demux_t *p_demux)
  129. {
  130.     demux_sys_t *p_sys = p_demux->p_sys;
  131.     block_t *p_block_in, *p_block_out;
  132.     if( p_sys->i_state == DIRAC_DEMUX_DISCONT )
  133.     {
  134.         p_sys->i_state++;
  135.         p_block_in = block_Alloc( 128 );
  136.         if( p_block_in )
  137.         {
  138.             p_block_in->i_flags = BLOCK_FLAG_DISCONTINUITY | BLOCK_FLAG_CORRUPTED;
  139.         }
  140.     }
  141.     else
  142.     {
  143.         p_block_in = stream_Block( p_demux->s, DIRAC_PACKET_SIZE );
  144.         if( !p_block_in )
  145.         {
  146.             return 0;
  147.         }
  148.         if ( p_sys->i_state == DIRAC_DEMUX_FIRST)
  149.         {
  150.             p_sys->i_state++;
  151.             /* by default, timestamps are invalid.
  152.              * Except when we need an anchor point */
  153. #if VLC_TS_INVALID == 0
  154.             /* xxx: to be removed in 1.1 */
  155.             p_block_in->i_dts = 1;
  156. #else
  157.             p_block_in->i_dts = 0;
  158. #endif
  159.         }
  160.     }
  161.     while( (p_block_out = p_sys->p_packetizer->pf_packetize( p_sys->p_packetizer, &p_block_in )) )
  162.     {
  163.         while( p_block_out )
  164.         {
  165.             block_t *p_next = p_block_out->p_next;
  166.             p_block_out->p_next = NULL;
  167.             if( p_sys->p_es == NULL )
  168.                 p_sys->p_es = es_out_Add( p_demux->out, &p_sys->p_packetizer->fmt_out);
  169.             p_block_out->i_dts += p_sys->i_dtsoffset;
  170.             p_sys->i_dts = p_block_out->i_dts;
  171.             /* track low watermark for pts_offset -- can be used to show
  172.              * when it is too large */
  173.             mtime_t i_delay = p_block_out->i_pts - p_block_out->i_dts;
  174.             if( p_sys->i_pts_offset_lowtide > i_delay )
  175.             {
  176.                 p_sys->i_pts_offset_lowtide = i_delay;
  177.             }
  178.             es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_block_out->i_dts );
  179.             es_out_Send( p_demux->out, p_sys->p_es, p_block_out );
  180.             p_block_out = p_next;
  181.         }
  182.     }
  183.     return 1;
  184. }
  185. /*****************************************************************************
  186.  * Control:
  187.  *****************************************************************************/
  188. static int Control( demux_t *p_demux, int i_query, va_list args )
  189. {
  190.     demux_sys_t *p_sys  = p_demux->p_sys;
  191.     if( DEMUX_GET_TIME == i_query )
  192.     {
  193.         int64_t *pi64 = (int64_t*)va_arg( args, int64_t * );
  194.         *pi64 = p_sys->i_dts;
  195.         return VLC_SUCCESS;
  196.     }
  197.     else if( DEMUX_GET_FPS == i_query )
  198.     {
  199.         if( !p_sys->p_packetizer->fmt_out.video.i_frame_rate )
  200.         {
  201.             return VLC_EGENERIC;
  202.         }
  203.         double *pd = (double*)va_arg( args, double * );
  204.         *pd = (float) p_sys->p_packetizer->fmt_out.video.i_frame_rate
  205.             / p_sys->p_packetizer->fmt_out.video.i_frame_rate_base;
  206.         return VLC_SUCCESS;
  207.     }
  208.     else
  209.     {
  210.         if( DEMUX_SET_POSITION == i_query || DEMUX_SET_TIME == i_query )
  211.         {
  212.             p_sys->i_state = DIRAC_DEMUX_DISCONT;
  213.         }
  214.         return demux_vaControlHelper( p_demux->s, 0, -1, 0, 1, i_query, args );
  215.     }
  216. }