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

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * mkv.cpp : matroska demuxer
  3.  *****************************************************************************
  4.  * Copyright (C) 2003-2004 the VideoLAN team
  5.  * $Id: e9b01cdd03e5f7cfe06c38da84e90295dd6f9c7f $
  6.  *
  7.  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
  8.  *          Steve Lhomme <steve.lhomme@free.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. #include "Ebml_parser.hpp"
  25. /*****************************************************************************
  26.  * Ebml Stream parser
  27.  *****************************************************************************/
  28. EbmlParser::EbmlParser( EbmlStream *es, EbmlElement *el_start, demux_t *p_demux )
  29. {
  30.     int i;
  31.     m_es = es;
  32.     m_got = NULL;
  33.     m_el[0] = el_start;
  34.     mi_remain_size[0] = el_start->GetSize();
  35.     for( i = 1; i < 6; i++ )
  36.     {
  37.         m_el[i] = NULL;
  38.     }
  39.     mi_level = 1;
  40.     mi_user_level = 1;
  41.     mb_keep = false;
  42.     mb_dummy = config_GetInt( p_demux, "mkv-use-dummy" );
  43. }
  44. EbmlParser::~EbmlParser( void )
  45. {
  46.     int i;
  47.     for( i = 1; i < mi_level; i++ )
  48.     {
  49.         if( !mb_keep )
  50.         {
  51.             delete m_el[i];
  52.         }
  53.         mb_keep = false;
  54.     }
  55. }
  56. EbmlElement* EbmlParser::UnGet( uint64 i_block_pos, uint64 i_cluster_pos )
  57. {
  58.     if ( mi_user_level > mi_level )
  59.     {
  60.         while ( mi_user_level != mi_level )
  61.         {
  62.             delete m_el[mi_user_level];
  63.             m_el[mi_user_level] = NULL;
  64.             mi_user_level--;
  65.         }
  66.     }
  67.     m_got = NULL;
  68.     mb_keep = false;
  69.     if ( m_el[1]->GetElementPosition() == i_cluster_pos )
  70.     {
  71.         m_es->I_O().setFilePointer( i_block_pos, seek_beginning );
  72.         return (EbmlMaster*) m_el[1];
  73.     }
  74.     else
  75.     {
  76.         // seek to the previous Cluster
  77.         m_es->I_O().setFilePointer( i_cluster_pos, seek_beginning );
  78.         mi_level--;
  79.         mi_user_level--;
  80.         delete m_el[mi_level];
  81.         m_el[mi_level] = NULL;
  82.         return NULL;
  83.     }
  84. }
  85. void EbmlParser::Up( void )
  86. {
  87.     if( mi_user_level == mi_level )
  88.     {
  89.         fprintf( stderr," arrrrrrrrrrrrrg Up cannot escape itselfn" );
  90.     }
  91.     mi_user_level--;
  92. }
  93. void EbmlParser::Down( void )
  94. {
  95.     mi_user_level++;
  96.     mi_level++;
  97. }
  98. void EbmlParser::Keep( void )
  99. {
  100.     mb_keep = true;
  101. }
  102. int EbmlParser::GetLevel( void )
  103. {
  104.     return mi_user_level;
  105. }
  106. void EbmlParser::Reset( demux_t *p_demux )
  107. {
  108.     while ( mi_level > 0)
  109.     {
  110.         delete m_el[mi_level];
  111.         m_el[mi_level] = NULL;
  112.         mi_level--;
  113.     }
  114.     mi_user_level = mi_level = 1;
  115.     // a little faster and cleaner
  116.     m_es->I_O().setFilePointer( static_cast<KaxSegment*>(m_el[0])->GetGlobalPosition(0) );
  117.     mb_dummy = config_GetInt( p_demux, "mkv-use-dummy" );
  118. }
  119. /* This function workarounds a bug in KaxBlockVirtual implementation */
  120. class KaxBlockVirtualWorkaround : public KaxBlockVirtual
  121. {
  122. public:
  123.     void Fix()
  124.     {
  125.         if( Data == DataBlock )
  126.             SetBuffer( NULL, 0 );
  127.     }
  128. };
  129. EbmlElement *EbmlParser::Get( void )
  130. {
  131.     int i_ulev = 0;
  132.     if( mi_user_level != mi_level )
  133.     {
  134.         return NULL;
  135.     }
  136.     if( m_got )
  137.     {
  138.         EbmlElement *ret = m_got;
  139.         m_got = NULL;
  140.         return ret;
  141.     }
  142.     if( m_el[mi_level] )
  143.     {
  144.         m_el[mi_level]->SkipData( *m_es, m_el[mi_level]->Generic().Context );
  145.         if( !mb_keep )
  146.         {
  147.             if( MKV_IS_ID( m_el[mi_level], KaxBlockVirtual ) )
  148.                 static_cast<KaxBlockVirtualWorkaround*>(m_el[mi_level])->Fix();
  149.             delete m_el[mi_level];
  150.         }
  151.         mb_keep = false;
  152.     }
  153.     m_el[mi_level] = m_es->FindNextElement( m_el[mi_level - 1]->Generic().Context, i_ulev, 0xFFFFFFFFL, mb_dummy != 0, 1 );
  154. //    mi_remain_size[mi_level] = m_el[mi_level]->GetSize();
  155.     if( i_ulev > 0 )
  156.     {
  157.         while( i_ulev > 0 )
  158.         {
  159.             if( mi_level == 1 )
  160.             {
  161.                 mi_level = 0;
  162.                 return NULL;
  163.             }
  164.             delete m_el[mi_level - 1];
  165.             m_got = m_el[mi_level -1] = m_el[mi_level];
  166.             m_el[mi_level] = NULL;
  167.             mi_level--;
  168.             i_ulev--;
  169.         }
  170.         return NULL;
  171.     }
  172.     else if( m_el[mi_level] == NULL )
  173.     {
  174.         fprintf( stderr," m_el[mi_level] == NULLn" );
  175.     }
  176.     return m_el[mi_level];
  177. }
  178. bool EbmlParser::IsTopPresent( EbmlElement *el )
  179. {
  180.     for( int i = 0; i < mi_level; i++ )
  181.     {
  182.         if( m_el[i] && m_el[i] == el )
  183.             return true;
  184.     }
  185.     return false;
  186. }