git.sesse.net Git - vlc/blob - modules/demux/mkv.cpp

   1 /*****************************************************************************
   2  * mkv.cpp : matroska demuxer
   3  *****************************************************************************
   4  * Copyright (C) 2003-2004 VideoLAN
   5  * $Id$
   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., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  23  *****************************************************************************/
  24
  25 /*****************************************************************************
  26  * Preamble
  27  *****************************************************************************/
  28 #include <stdlib.h>                                      /* malloc(), free() */
  29
  30 #include <vlc/vlc.h>
  31
  32 #ifdef HAVE_TIME_H
  33 #   include <time.h>                                               /* time() */
  34 #endif
  35
  36 #include <vlc/input.h>
  37
  38 #include <codecs.h>                        /* BITMAPINFOHEADER, WAVEFORMATEX */
  39 #include "iso_lang.h"
  40 #include "vlc_meta.h"
  41
  42 #include <iostream>
  43 #include <cassert>
  44 #include <typeinfo>
  45 #include <string>
  46 #include <vector>
  47 #include <algorithm>
  48
  49 #ifdef HAVE_DIRENT_H
  50 #   include <dirent.h>
  51 #endif
  52
  53 /* libebml and matroska */
  54 #include "ebml/EbmlHead.h"
  55 #include "ebml/EbmlSubHead.h"
  56 #include "ebml/EbmlStream.h"
  57 #include "ebml/EbmlContexts.h"
  58 #include "ebml/EbmlVoid.h"
  59 #include "ebml/EbmlVersion.h"
  60 #include "ebml/StdIOCallback.h"
  61
  62 #include "matroska/KaxAttachments.h"
  63 #include "matroska/KaxBlock.h"
  64 #include "matroska/KaxBlockData.h"
  65 #include "matroska/KaxChapters.h"
  66 #include "matroska/KaxCluster.h"
  67 #include "matroska/KaxClusterData.h"
  68 #include "matroska/KaxContexts.h"
  69 #include "matroska/KaxCues.h"
  70 #include "matroska/KaxCuesData.h"
  71 #include "matroska/KaxInfo.h"
  72 #include "matroska/KaxInfoData.h"
  73 #include "matroska/KaxSeekHead.h"
  74 #include "matroska/KaxSegment.h"
  75 #include "matroska/KaxTag.h"
  76 #include "matroska/KaxTags.h"
  77 #include "matroska/KaxTagMulti.h"
  78 #include "matroska/KaxTracks.h"
  79 #include "matroska/KaxTrackAudio.h"
  80 #include "matroska/KaxTrackVideo.h"
  81 #include "matroska/KaxTrackEntryData.h"
  82 #include "matroska/KaxContentEncoding.h"
  83 #include "matroska/KaxVersion.h"
  84
  85 #include "ebml/StdIOCallback.h"
  86
  87 extern "C" {
  88    #include "mp4/libmp4.h"
  89 }
  90 #ifdef HAVE_ZLIB_H
  91 #   include <zlib.h>
  92 #endif
  93
  94 #define MATROSKA_COMPRESSION_NONE 0
  95 #define MATROSKA_COMPRESSION_ZLIB 1
  96
  97 #define MKVD_TIMECODESCALE 1000000
  98
  99 /**
 100  * What's between a directory and a filename?
 101  */
 102 #if defined( WIN32 )
 103     #define DIRECTORY_SEPARATOR '\\'
 104 #else
 105     #define DIRECTORY_SEPARATOR '/'
 106 #endif
 107
 108 using namespace LIBMATROSKA_NAMESPACE;
 109 using namespace std;
 110
 111 /*****************************************************************************
 112  * Module descriptor
 113  *****************************************************************************/
 114 static int  Open ( vlc_object_t * );
 115 static void Close( vlc_object_t * );
 116
 117 vlc_module_begin();
 118     set_shortname( _("Matroska") );
 119     set_description( _("Matroska stream demuxer" ) );
 120     set_capability( "demux2", 50 );
 121     set_callbacks( Open, Close );
 122     set_category( CAT_INPUT );
 123     set_subcategory( SUBCAT_INPUT_DEMUX );
 124
 125     add_bool( "mkv-use-ordered-chapters", 1, NULL,
 126             N_("Ordered chapters"),
 127             N_("Play ordered chapters as specified in the segment"), VLC_TRUE );
 128
 129     add_bool( "mkv-use-chapter-codec", 1, NULL,
 130             N_("Chapter codecs"),
 131             N_("Use chapter codecs found in the segment"), VLC_TRUE );
 132
 133     add_bool( "mkv-seek-percent", 0, NULL,
 134             N_("Seek based on percent not time"),
 135             N_("Seek based on percent not time"), VLC_TRUE );
 136
 137     add_bool( "mkv-use-dummy", 0, NULL,
 138             N_("Dummy Elements"),
 139             N_("Read and discard unknown EBML elements (not good for broken files)"), VLC_TRUE );
 140
 141     add_shortcut( "mka" );
 142     add_shortcut( "mkv" );
 143 vlc_module_end();
 144
 145 /*****************************************************************************
 146  * Local prototypes
 147  *****************************************************************************/
 148 #ifdef HAVE_ZLIB_H
 149 block_t *block_zlib_decompress( vlc_object_t *p_this, block_t *p_in_block ) {
 150     int result, dstsize, n;
 151     unsigned char *dst;
 152     block_t *p_block;
 153     z_stream d_stream;
 154
 155     d_stream.zalloc = (alloc_func)0;
 156     d_stream.zfree = (free_func)0;
 157     d_stream.opaque = (voidpf)0;
 158     result = inflateInit(&d_stream);
 159     if( result != Z_OK )
 160     {
 161         msg_Dbg( p_this, "inflateInit() failed. Result: %d", result );
 162         return NULL;
 163     }
 164
 165     d_stream.next_in = (Bytef *)p_in_block->p_buffer;
 166     d_stream.avail_in = p_in_block->i_buffer;
 167     n = 0;
 168     p_block = block_New( p_this, 0 );
 169     dst = NULL;
 170     do
 171     {
 172         n++;
 173         p_block = block_Realloc( p_block, 0, n * 1000 );
 174         dst = (unsigned char *)p_block->p_buffer;
 175         d_stream.next_out = (Bytef *)&dst[(n - 1) * 1000];
 176         d_stream.avail_out = 1000;
 177         result = inflate(&d_stream, Z_NO_FLUSH);
 178         if( ( result != Z_OK ) && ( result != Z_STREAM_END ) )
 179         {
 180             msg_Dbg( p_this, "Zlib decompression failed. Result: %d", result );
 181             return NULL;
 182         }
 183     }
 184     while( ( d_stream.avail_out == 0 ) && ( d_stream.avail_in != 0 ) &&
 185            ( result != Z_STREAM_END ) );
 186
 187     dstsize = d_stream.total_out;
 188     inflateEnd( &d_stream );
 189
 190     p_block = block_Realloc( p_block, 0, dstsize );
 191     p_block->i_buffer = dstsize;
 192     block_Release( p_in_block );
 193
 194     return p_block;
 195 }
 196 #endif
 197
 198 /**
 199  * Helper function to print the mkv parse tree
 200  */
 201 static void MkvTree( demux_t & demuxer, int i_level, char *psz_format, ... )
 202 {
 203     va_list args;
 204     if( i_level > 9 )
 205     {
 206         msg_Err( &demuxer, "too deep tree" );
 207         return;
 208     }
 209     va_start( args, psz_format );
 210     static char *psz_foo = "|   |   |   |   |   |   |   |   |   |";
 211     char *psz_foo2 = (char*)malloc( ( i_level * 4 + 3 + strlen( psz_format ) ) * sizeof(char) );
 212     strncpy( psz_foo2, psz_foo, 4 * i_level );
 213     psz_foo2[ 4 * i_level ] = '+';
 214     psz_foo2[ 4 * i_level + 1 ] = ' ';
 215     strcpy( &psz_foo2[ 4 * i_level + 2 ], psz_format );
 216     __msg_GenericVa( VLC_OBJECT(&demuxer), VLC_MSG_DBG, "mkv", psz_foo2, args );
 217     free( psz_foo2 );
 218     va_end( args );
 219 }
 220
 221 /*****************************************************************************
 222  * Stream managment
 223  *****************************************************************************/
 224 class vlc_stream_io_callback: public IOCallback
 225 {
 226   private:
 227     stream_t       *s;
 228     vlc_bool_t     mb_eof;
 229
 230   public:
 231     vlc_stream_io_callback( stream_t * );
 232
 233     virtual uint32   read            ( void *p_buffer, size_t i_size);
 234     virtual void     setFilePointer  ( int64_t i_offset, seek_mode mode = seek_beginning );
 235     virtual size_t   write           ( const void *p_buffer, size_t i_size);
 236     virtual uint64   getFilePointer  ( void );
 237     virtual void     close           ( void );
 238 };
 239
 240 /*****************************************************************************
 241  * Ebml Stream parser
 242  *****************************************************************************/
 243 class EbmlParser
 244 {
 245   public:
 246     EbmlParser( EbmlStream *es, EbmlElement *el_start, demux_t *p_demux );
 247     virtual ~EbmlParser( void );
 248
 249     void Up( void );
 250     void Down( void );
 251     void Reset( demux_t *p_demux );
 252     EbmlElement *Get( void );
 253     void        Keep( void );
 254
 255     int GetLevel( void );
 256
 257   private:
 258     EbmlStream  *m_es;
 259     int         mi_level;
 260     EbmlElement *m_el[10];
 261     int64_t      mi_remain_size[10];
 262
 263     EbmlElement *m_got;
 264
 265     int         mi_user_level;
 266     vlc_bool_t  mb_keep;
 267     vlc_bool_t  mb_dummy;
 268 };
 269
 270
 271 /*****************************************************************************
 272  * Some functions to manipulate memory
 273  *****************************************************************************/
 274 #define GetFOURCC( p )  __GetFOURCC( (uint8_t*)p )
 275 static vlc_fourcc_t __GetFOURCC( uint8_t *p )
 276 {
 277     return VLC_FOURCC( p[0], p[1], p[2], p[3] );
 278 }
 279
 280 /*****************************************************************************
 281  * definitions of structures and functions used by this plugins
 282  *****************************************************************************/
 283 typedef struct
 284 {
 285     vlc_bool_t   b_default;
 286     vlc_bool_t   b_enabled;
 287     unsigned int i_number;
 288
 289     int          i_extra_data;
 290     uint8_t      *p_extra_data;
 291
 292     char         *psz_codec;
 293
 294     uint64_t     i_default_duration;
 295     float        f_timecodescale;
 296
 297     /* video */
 298     es_format_t fmt;
 299     float       f_fps;
 300     es_out_id_t *p_es;
 301
 302     vlc_bool_t      b_inited;
 303     /* data to be send first */
 304     int             i_data_init;
 305     uint8_t         *p_data_init;
 306
 307     /* hack : it's for seek */
 308     vlc_bool_t      b_search_keyframe;
 309     vlc_bool_t      b_silent;
 310
 311     /* informative */
 312     char         *psz_codec_name;
 313     char         *psz_codec_settings;
 314     char         *psz_codec_info_url;
 315     char         *psz_codec_download_url;
 316
 317     /* encryption/compression */
 318     int           i_compression_type;
 319
 320 } mkv_track_t;
 321
 322 typedef struct
 323 {
 324     int     i_track;
 325     int     i_block_number;
 326
 327     int64_t i_position;
 328     int64_t i_time;
 329
 330     vlc_bool_t b_key;
 331 } mkv_index_t;
 332
 333 class demux_sys_t;
 334
 335 const binary MATROSKA_DVD_LEVEL_SS   = 0x30;
 336 const binary MATROSKA_DVD_LEVEL_LU   = 0x2A;
 337 const binary MATROSKA_DVD_LEVEL_TT   = 0x28;
 338 const binary MATROSKA_DVD_LEVEL_PGC  = 0x20;
 339 const binary MATROSKA_DVD_LEVEL_PG   = 0x18;
 340 const binary MATROSKA_DVD_LEVEL_PTT  = 0x10;
 341 const binary MATROSKA_DVD_LEVEL_CN   = 0x08;
 342
 343 class chapter_codec_cmds_c
 344 {
 345 public:
 346     chapter_codec_cmds_c( demux_sys_t & demuxer, int codec_id = -1)
 347     :i_codec_id( codec_id )
 348     ,sys( demuxer )
 349     {}
 350
 351     virtual ~chapter_codec_cmds_c() {}
 352
 353     void SetPrivate( const KaxChapterProcessPrivate & private_data )
 354     {
 355         m_private_data = *( new KaxChapterProcessPrivate( private_data ) );
 356     }
 357
 358     void AddCommand( const KaxChapterProcessCommand & command );
 359
 360     /// \return wether the codec has seeked in the files or not
 361     virtual bool Enter() { return false; }
 362     virtual bool Leave() { return false; }
 363     virtual std::string GetCodecName( bool f_for_title = false ) const { return ""; }
 364
 365     KaxChapterProcessPrivate m_private_data;
 366
 367 protected:
 368     std::vector<KaxChapterProcessData> enter_cmds;
 369     std::vector<KaxChapterProcessData> during_cmds;
 370     std::vector<KaxChapterProcessData> leave_cmds;
 371
 372     int i_codec_id;
 373     demux_sys_t & sys;
 374 };
 375
 376 class dvd_command_interpretor_c
 377 {
 378 public:
 379     dvd_command_interpretor_c( demux_sys_t & demuxer )
 380     :sys( demuxer )
 381     {
 382         memset( p_PRMs, 0, sizeof(p_PRMs) );
 383         p_PRMs[ 0x80 + 1 ] = 15;
 384         p_PRMs[ 0x80 + 2 ] = 62;
 385         p_PRMs[ 0x80 + 3 ] = 1;
 386         p_PRMs[ 0x80 + 4 ] = 1;
 387         p_PRMs[ 0x80 + 7 ] = 1;
 388         p_PRMs[ 0x80 + 8 ] = 1;
 389         p_PRMs[ 0x80 + 16 ] = 0xFFFFu;
 390         p_PRMs[ 0x80 + 18 ] = 0xFFFFu;
 391     }
 392
 393     bool Interpret( const binary * p_command, size_t i_size = 8 );
 394
 395 protected:
 396     uint16 GetPRM( size_t index ) const
 397     {
 398         if ( index < 256 )
 399             return p_PRMs[ index ];
 400         else return 0;
 401     }
 402
 403     uint16 GetGPRM( size_t index ) const
 404     {
 405         if ( index >= 0 && index < 16 )
 406             return p_PRMs[ index ];
 407         else return 0;
 408     }
 409
 410     uint16 GetSPRM( size_t index ) const
 411     {
 412         // 21,22,23 reserved for future use
 413         if ( index >= 0x80 && index < 0x95 )
 414             return p_PRMs[ index ];
 415         else return 0;
 416     }
 417
 418     bool SetPRM( size_t index, uint16 value )
 419     {
 420         if ( index >= 0 && index < 16 )
 421         {
 422             p_PRMs[ index ] = value;
 423             return true;
 424         }
 425         return false;
 426     }
 427
 428     bool SetGPRM( size_t index, uint16 value )
 429     {
 430         if ( index >= 0 && index < 16 )
 431         {
 432             p_PRMs[ index ] = value;
 433             return true;
 434         }
 435         return false;
 436     }
 437
 438     bool SetSPRM( size_t index, uint16 value )
 439     {
 440         if ( index > 0x80 && index <= 0x8D && index != 0x8C )
 441         {
 442             p_PRMs[ index ] = value;
 443             return true;
 444         }
 445         return false;
 446     }
 447
 448     std::string GetRegTypeName( bool b_value, uint16 value ) const
 449     {
 450         std::string result;
 451         char s_value[6], s_reg_value[6];
 452         sprintf( s_value, "%.5d", value );
 453
 454         if ( b_value )
 455         {
 456             result = "value (";
 457             result += s_value;
 458             result += ")";
 459         }
 460         else if ( value < 0x80 )
 461         {
 462             sprintf( s_reg_value, "%.5d", GetPRM( value ) );
 463             result = "GPreg[";
 464             result += s_value;
 465             result += "] (";
 466             result += s_reg_value;
 467             result += ")";
 468         }
 469         else
 470         {
 471             sprintf( s_reg_value, "%.5d", GetPRM( value ) );
 472             result = "SPreg[";
 473             result += s_value;
 474             result += "] (";
 475             result += s_reg_value;
 476             result += ")";
 477         }
 478         return result;
 479     }
 480
 481     uint16       p_PRMs[256];
 482     demux_sys_t  & sys;
 483
 484     // DVD command IDs
 485
 486     // Tests
 487     // wether the test has to be positive or not
 488     static const uint16 CMD_DVD_IF_NOT              = 0x10;
 489     // wether it's a comparison on the value or register
 490     static const uint16 CMD_DVD_TEST_VALUE          = 0x80;
 491     static const uint16 CMD_DVD_IF_GPREG_AND        = (0 << 5);
 492     static const uint16 CMD_DVD_IF_GPREG_EQUAL      = (1 << 5);
 493     static const uint16 CMD_DVD_IF_GPREG_SUP_EQUAL  = (2 << 5);
 494     static const uint16 CMD_DVD_IF_GPREG_INF        = (3 << 5);
 495
 496     static const uint16 CMD_DVD_NOP                    = 0x0000;
 497     static const uint16 CMD_DVD_GOTO_LINE              = 0x0001;
 498     static const uint16 CMD_DVD_BREAK                  = 0x0002;
 499     // Links
 500     static const uint16 CMD_DVD_NOP2                   = 0x2001;
 501     static const uint16 CMD_DVD_LINKPGCN               = 0x2004;
 502     static const uint16 CMD_DVD_LINKPGN                = 0x2006;
 503     static const uint16 CMD_DVD_LINKCN                 = 0x2007;
 504     static const uint16 CMD_DVD_JUMP_TT                = 0x3002;
 505     static const uint16 CMD_DVD_JUMPVTS_TT             = 0x3003;
 506     static const uint16 CMD_DVD_JUMP_SS                = 0x3006;
 507     static const uint16 CMD_DVD_CALLSS_VTSM1           = 0x3008;
 508     //
 509     static const uint16 CMD_DVD_SET_HL_BTNN2           = 0x4600;
 510     static const uint16 CMD_DVD_SET_HL_BTNN_LINKPGCN1  = 0x4604;
 511     static const uint16 CMD_DVD_SET_AUDIO              = 0x5100;
 512     static const uint16 CMD_DVD_SET_GPRMMD             = 0x5300;
 513     static const uint16 CMD_DVD_SET_HL_BTNN1           = 0x5600;
 514     static const uint16 CMD_DVD_SET_HL_BTNN_LINKPGCN2  = 0x5604;
 515     static const uint16 CMD_DVD_SET_HL_BTNN_LINKCN     = 0x5607;
 516     // Operations
 517     static const uint16 CMD_DVD_MOV_SPREG_PREG         = 0x6100;
 518     static const uint16 CMD_DVD_GPREG_MOV_VALUE        = 0x7100;
 519     static const uint16 CMD_DVD_SUB_GPREG              = 0x7400;
 520     static const uint16 CMD_DVD_MULT_GPREG             = 0x7500;
 521     static const uint16 CMD_DVD_GPREG_DIV_VALUE        = 0x7600;
 522     static const uint16 CMD_DVD_GPREG_AND_VALUE        = 0x7900;
 523
 524     // callbacks when browsing inside CodecPrivate
 525     static bool MatchTitleNumber( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size );
 526     static bool MatchPgcType    ( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size );
 527 };
 528
 529 class dvd_chapter_codec_c : public chapter_codec_cmds_c
 530 {
 531 public:
 532     dvd_chapter_codec_c( demux_sys_t & sys )
 533     :chapter_codec_cmds_c( sys, 1 )
 534     ,interpretor( sys )
 535     {}
 536
 537     bool Enter();
 538     bool Leave();
 539     std::string GetCodecName( bool f_for_title = false ) const;
 540
 541 protected:
 542     dvd_command_interpretor_c interpretor;
 543 };
 544
 545 class matroska_script_interpretor_c
 546 {
 547 public:
 548     matroska_script_interpretor_c( demux_sys_t & demuxer )
 549     :sys( demuxer )
 550     {}
 551
 552     bool Interpret( const binary * p_command, size_t i_size );
 553
 554     // DVD command IDs
 555     static const std::string CMD_MS_GOTO_AND_PLAY;
 556
 557 protected:
 558     demux_sys_t  & sys;
 559 };
 560
 561 const std::string matroska_script_interpretor_c::CMD_MS_GOTO_AND_PLAY = "GotoAndPlay";
 562
 563
 564 class matroska_script_codec_c : public chapter_codec_cmds_c
 565 {
 566 public:
 567     matroska_script_codec_c( demux_sys_t & sys )
 568     :chapter_codec_cmds_c( sys, 0 )
 569     ,interpretor( sys )
 570     {}
 571
 572     bool Enter();
 573     bool Leave();
 574
 575 protected:
 576     matroska_script_interpretor_c interpretor;
 577 };
 578
 579 class chapter_translation_c
 580 {
 581 public:
 582     KaxChapterTranslateID  translated;
 583     unsigned int           codec_id;
 584     std::vector<uint64_t>  editions;
 585 };
 586
 587 class chapter_item_c
 588 {
 589 public:
 590     chapter_item_c()
 591     :i_start_time(0)
 592     ,i_end_time(-1)
 593     ,i_user_start_time(-1)
 594     ,i_user_end_time(-1)
 595     ,i_seekpoint_num(-1)
 596     ,b_display_seekpoint(true)
 597     ,b_user_display(false)
 598     ,psz_parent(NULL)
 599     ,b_is_leaving(false)
 600     {}
 601
 602     virtual ~chapter_item_c()
 603     {
 604         std::vector<chapter_codec_cmds_c*>::iterator index = codecs.begin();
 605         while ( index != codecs.end() )
 606         {
 607             delete (*index);
 608             index++;
 609         }
 610         std::vector<chapter_item_c*>::iterator index_ = sub_chapters.begin();
 611         while ( index_ != sub_chapters.end() )
 612         {
 613             delete (*index_);
 614             index_++;
 615         }
 616     }
 617
 618     int64_t RefreshChapters( bool b_ordered, int64_t i_prev_user_time );
 619     int PublishChapters( input_title_t & title, int & i_user_chapters, int i_level = 0 );
 620     chapter_item_c * FindTimecode( mtime_t i_timecode );
 621     void Append( const chapter_item_c & edition );
 622     chapter_item_c * FindChapter( int64_t i_find_uid );
 623     virtual chapter_item_c *BrowseCodecPrivate( unsigned int codec_id,
 624                                     bool (*match)(const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size ),
 625                                     const void *p_cookie,
 626                                     size_t i_cookie_size );
 627     std::string                 GetCodecName( bool f_for_title = false ) const;
 628     bool                        ParentOf( const chapter_item_c & item ) const;
 629
 630     int64_t                     i_start_time, i_end_time;
 631     int64_t                     i_user_start_time, i_user_end_time; /* the time in the stream when an edition is ordered */
 632     std::vector<chapter_item_c*> sub_chapters;
 633     int                         i_seekpoint_num;
 634     int64_t                     i_uid;
 635     bool                        b_display_seekpoint;
 636     bool                        b_user_display;
 637     std::string                 psz_name;
 638     chapter_item_c              *psz_parent;
 639     bool                        b_is_leaving;
 640
 641     std::vector<chapter_codec_cmds_c*> codecs;
 642
 643     bool operator<( const chapter_item_c & item ) const
 644     {
 645         return ( i_user_start_time < item.i_user_start_time || (i_user_start_time == item.i_user_start_time && i_user_end_time < item.i_user_end_time) );
 646     }
 647
 648     bool Enter( bool b_do_subchapters );
 649     bool Leave( bool b_do_subchapters );
 650     bool EnterAndLeave( chapter_item_c *p_item );
 651 };
 652
 653 class chapter_edition_c : public chapter_item_c
 654 {
 655 public:
 656     chapter_edition_c()
 657     :b_ordered(false)
 658     {}
 659
 660     void RefreshChapters( );
 661     mtime_t Duration() const;
 662     std::string GetMainName() const;
 663
 664     bool                        b_ordered;
 665 };
 666
 667 class matroska_segment_c
 668 {
 669 public:
 670     matroska_segment_c( demux_sys_t & demuxer, EbmlStream & estream )
 671         :segment(NULL)
 672         ,es(estream)
 673         ,i_timescale(MKVD_TIMECODESCALE)
 674         ,i_duration(-1)
 675         ,i_start_time(0)
 676         ,i_cues_position(-1)
 677         ,i_chapters_position(-1)
 678         ,i_tags_position(-1)
 679         ,cluster(NULL)
 680         ,i_start_pos(0)
 681         ,b_cues(VLC_FALSE)
 682         ,i_index(0)
 683         ,i_index_max(1024)
 684         ,psz_muxing_application(NULL)
 685         ,psz_writing_application(NULL)
 686         ,psz_segment_filename(NULL)
 687         ,psz_title(NULL)
 688         ,psz_date_utc(NULL)
 689         ,i_default_edition(0)
 690         ,sys(demuxer)
 691         ,ep(NULL)
 692         ,b_preloaded(false)
 693     {
 694         index = (mkv_index_t*)malloc( sizeof( mkv_index_t ) * i_index_max );
 695     }
 696
 697     virtual ~matroska_segment_c()
 698     {
 699         for( size_t i_track = 0; i_track < tracks.size(); i_track++ )
 700         {
 701 #define tk  tracks[i_track]
 702             if( tk->fmt.psz_description )
 703             {
 704                 free( tk->fmt.psz_description );
 705             }
 706             if( tk->psz_codec )
 707             {
 708                 free( tk->psz_codec );
 709             }
 710             if( tk->fmt.psz_language )
 711             {
 712                 free( tk->fmt.psz_language );
 713             }
 714             delete tk;
 715 #undef tk
 716         }
 717
 718         if( psz_writing_application )
 719         {
 720             free( psz_writing_application );
 721         }
 722         if( psz_muxing_application )
 723         {
 724             free( psz_muxing_application );
 725         }
 726         if( psz_segment_filename )
 727         {
 728             free( psz_segment_filename );
 729         }
 730         if( psz_title )
 731         {
 732             free( psz_title );
 733         }
 734         if( psz_date_utc )
 735         {
 736             free( psz_date_utc );
 737         }
 738         if ( index )
 739             free( index );
 740
 741         delete ep;
 742
 743         std::vector<chapter_edition_c*>::iterator index = stored_editions.begin();
 744         while ( index != stored_editions.end() )
 745         {
 746             delete (*index);
 747             index++;
 748         }
 749     }
 750
 751     KaxSegment              *segment;
 752     EbmlStream              & es;
 753
 754     /* time scale */
 755     uint64_t                i_timescale;
 756
 757     /* duration of the segment */
 758     mtime_t                 i_duration;
 759     mtime_t                 i_start_time;
 760
 761     /* all tracks */
 762     std::vector<mkv_track_t*> tracks;
 763
 764     /* from seekhead */
 765     int64_t                 i_cues_position;
 766     int64_t                 i_chapters_position;
 767     int64_t                 i_tags_position;
 768
 769     KaxCluster              *cluster;
 770     int64_t                 i_start_pos;
 771     KaxSegmentUID           segment_uid;
 772     KaxPrevUID              prev_segment_uid;
 773     KaxNextUID              next_segment_uid;
 774
 775     vlc_bool_t              b_cues;
 776     int                     i_index;
 777     int                     i_index_max;
 778     mkv_index_t             *index;
 779
 780     /* info */
 781     char                    *psz_muxing_application;
 782     char                    *psz_writing_application;
 783     char                    *psz_segment_filename;
 784     char                    *psz_title;
 785     char                    *psz_date_utc;
 786
 787     /* !!!!! GCC 3.3 bug on Darwin !!!!! */
 788     /* when you remove this variable the compiler issues an atomicity error */
 789     /* this variable only works when using std::vector<chapter_edition_c> */
 790     std::vector<chapter_edition_c*> stored_editions;
 791     int                             i_default_edition;
 792
 793     std::vector<chapter_translation_c> translations;
 794     std::vector<KaxSegmentFamily>  families;
 795
 796     demux_sys_t                    & sys;
 797     EbmlParser                     *ep;
 798     bool                           b_preloaded;
 799
 800     bool Preload( );
 801     bool PreloadFamily( const matroska_segment_c & segment );
 802     void ParseInfo( KaxInfo *info );
 803     void ParseChapters( KaxChapters *chapters );
 804     void ParseSeekHead( KaxSeekHead *seekhead );
 805     void ParseTracks( KaxTracks *tracks );
 806     void ParseChapterAtom( int i_level, KaxChapterAtom *ca, chapter_item_c & chapters );
 807     void ParseTrackEntry( KaxTrackEntry *m );
 808     void ParseCluster( );
 809     void IndexAppendCluster( KaxCluster *cluster );
 810     void LoadCues( );
 811     void LoadTags( );
 812     void InformationCreate( );
 813     void Seek( mtime_t i_date, mtime_t i_time_offset );
 814     int BlockGet( KaxBlock **pp_block, int64_t *pi_ref1, int64_t *pi_ref2, int64_t *pi_duration );
 815     bool Select( mtime_t i_start_time );
 816     void UnSelect( );
 817
 818     static bool CompareSegmentUIDs( const matroska_segment_c * item_a, const matroska_segment_c * item_b );
 819 };
 820
 821 // class holding hard-linked segment together in the playback order
 822 class virtual_segment_c
 823 {
 824 public:
 825     virtual_segment_c( matroska_segment_c *p_segment )
 826         :p_editions(NULL)
 827         ,i_sys_title(0)
 828         ,i_current_segment(0)
 829         ,i_current_edition(-1)
 830         ,psz_current_chapter(NULL)
 831     {
 832         linked_segments.push_back( p_segment );
 833
 834         AppendUID( p_segment->segment_uid );
 835         AppendUID( p_segment->prev_segment_uid );
 836         AppendUID( p_segment->next_segment_uid );
 837     }
 838
 839     void Sort();
 840     size_t AddSegment( matroska_segment_c *p_segment );
 841     void PreloadLinked( );
 842     mtime_t Duration( ) const;
 843     void LoadCues( );
 844     void Seek( demux_t & demuxer, mtime_t i_date, mtime_t i_time_offset, chapter_item_c *psz_chapter );
 845
 846     inline chapter_edition_c *Edition()
 847     {
 848         if ( i_current_edition >= 0 && size_t(i_current_edition) < p_editions->size() )
 849             return (*p_editions)[i_current_edition];
 850         return NULL;
 851     }
 852
 853     matroska_segment_c * Segment() const
 854     {
 855         if ( linked_segments.size() == 0 || i_current_segment >= linked_segments.size() )
 856             return NULL;
 857         return linked_segments[i_current_segment];
 858     }
 859
 860     inline const chapter_item_c *CurrentChapter() const {
 861         return psz_current_chapter;
 862     }
 863
 864     bool SelectNext()
 865     {
 866         if ( i_current_segment < linked_segments.size()-1 )
 867         {
 868             i_current_segment++;
 869             return true;
 870         }
 871         return false;
 872     }
 873
 874     bool FindUID( KaxSegmentUID & uid ) const
 875     {
 876         for ( size_t i=0; i<linked_uids.size(); i++ )
 877         {
 878             if ( linked_uids[i] == uid )
 879                 return true;
 880         }
 881         return false;
 882     }
 883
 884     bool UpdateCurrentToChapter( demux_t & demux );
 885     void PrepareChapters( );
 886
 887     chapter_item_c *BrowseCodecPrivate( unsigned int codec_id,
 888                                         bool (*match)(const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size ),
 889                                         const void *p_cookie,
 890                                         size_t i_cookie_size );
 891     chapter_item_c *FindChapter( int64_t i_find_uid );
 892
 893     std::vector<chapter_edition_c*>  *p_editions;
 894     int                              i_sys_title;
 895
 896 protected:
 897     std::vector<matroska_segment_c*> linked_segments;
 898     std::vector<KaxSegmentUID>       linked_uids;
 899     size_t                           i_current_segment;
 900
 901     int                              i_current_edition;
 902     chapter_item_c                   *psz_current_chapter;
 903
 904     void                             AppendUID( const EbmlBinary & UID );
 905 };
 906
 907 class matroska_stream_c
 908 {
 909 public:
 910     matroska_stream_c( demux_sys_t & demuxer )
 911         :p_in(NULL)
 912         ,p_es(NULL)
 913         ,sys(demuxer)
 914     {}
 915
 916     virtual ~matroska_stream_c()
 917     {
 918         delete p_in;
 919         delete p_es;
 920     }
 921
 922     IOCallback         *p_in;
 923     EbmlStream         *p_es;
 924
 925     std::vector<matroska_segment_c*> segments;
 926
 927     demux_sys_t                      & sys;
 928 };
 929
 930 class demux_sys_t
 931 {
 932 public:
 933     demux_sys_t( demux_t & demux )
 934         :demuxer(demux)
 935         ,i_pts(0)
 936         ,i_start_pts(0)
 937         ,i_chapter_time(0)
 938         ,meta(NULL)
 939         ,i_current_title(0)
 940         ,p_current_segment(NULL)
 941         ,f_duration(-1.0)
 942     {}
 943
 944     virtual ~demux_sys_t()
 945     {
 946         size_t i;
 947         for ( i=0; i<streams.size(); i++ )
 948             delete streams[i];
 949         for ( i=0; i<opened_segments.size(); i++ )
 950             delete opened_segments[i];
 951         for ( i=0; i<used_segments.size(); i++ )
 952             delete used_segments[i];
 953     }
 954
 955     /* current data */
 956     demux_t                 & demuxer;
 957
 958     mtime_t                 i_pts;
 959     mtime_t                 i_start_pts;
 960     mtime_t                 i_chapter_time;
 961
 962     vlc_meta_t              *meta;
 963
 964     std::vector<input_title_t>       titles; // matroska editions
 965     size_t                           i_current_title;
 966
 967     std::vector<matroska_stream_c*>  streams;
 968     std::vector<matroska_segment_c*> opened_segments;
 969     std::vector<virtual_segment_c*>  used_segments;
 970     virtual_segment_c                *p_current_segment;
 971
 972     /* duration of the stream */
 973     float                   f_duration;
 974
 975     matroska_segment_c *FindSegment( const EbmlBinary & uid ) const;
 976     chapter_item_c *BrowseCodecPrivate( unsigned int codec_id,
 977                                         bool (*match)(const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size ),
 978                                         const void *p_cookie,
 979                                         size_t i_cookie_size,
 980                                         virtual_segment_c * & p_segment_found );
 981     chapter_item_c *FindChapter( int64_t i_find_uid, virtual_segment_c * & p_segment_found );
 982
 983     void PreloadFamily( const matroska_segment_c & of_segment );
 984     void PreloadLinked( matroska_segment_c *p_segment );
 985     bool PreparePlayback( virtual_segment_c *p_new_segment );
 986     matroska_stream_c *AnalyseAllSegmentsFound( EbmlStream *p_estream );
 987
 988 protected:
 989     virtual_segment_c *VirtualFromSegments( matroska_segment_c *p_segment ) const;
 990     bool IsUsedSegment( matroska_segment_c &p_segment ) const;
 991 };
 992
 993 static int  Demux  ( demux_t * );
 994 static int  Control( demux_t *, int, va_list );
 995 static void Seek   ( demux_t *, mtime_t i_date, double f_percent, chapter_item_c *psz_chapter );
 996
 997 #define MKV_IS_ID( el, C ) ( EbmlId( (*el) ) == C::ClassInfos.GlobalId )
 998
 999 static char *UTF8ToStr          ( const UTFstring &u );
1000
1001 /*****************************************************************************
1002  * Open: initializes matroska demux structures
1003  *****************************************************************************/
1004 static int Open( vlc_object_t * p_this )
1005 {
1006     demux_t            *p_demux = (demux_t*)p_this;
1007     demux_sys_t        *p_sys;
1008     matroska_stream_c  *p_stream;
1009     matroska_segment_c *p_segment;
1010     uint8_t            *p_peek;
1011     std::string        s_path, s_filename;
1012     vlc_stream_io_callback *p_io_callback;
1013     EbmlStream         *p_io_stream;
1014
1015     /* peek the begining */
1016     if( stream_Peek( p_demux->s, &p_peek, 4 ) < 4 ) return VLC_EGENERIC;
1017
1018     /* is a valid file */
1019     if( p_peek[0] != 0x1a || p_peek[1] != 0x45 ||
1020         p_peek[2] != 0xdf || p_peek[3] != 0xa3 ) return VLC_EGENERIC;
1021
1022     /* Set the demux function */
1023     p_demux->pf_demux   = Demux;
1024     p_demux->pf_control = Control;
1025     p_demux->p_sys      = p_sys = new demux_sys_t( *p_demux );
1026
1027     p_io_callback = new vlc_stream_io_callback( p_demux->s );
1028     p_io_stream = new EbmlStream( *p_io_callback );
1029
1030     if( p_io_stream == NULL )
1031     {
1032         msg_Err( p_demux, "failed to create EbmlStream" );
1033         delete p_io_callback;
1034         delete p_sys;
1035         return VLC_EGENERIC;
1036     }
1037
1038     p_stream = p_sys->AnalyseAllSegmentsFound( p_io_stream );
1039     if( p_stream == NULL )
1040     {
1041         msg_Err( p_demux, "cannot find KaxSegment" );
1042         goto error;
1043     }
1044     p_sys->streams.push_back( p_stream );
1045
1046     p_stream->p_in = p_io_callback;
1047     p_stream->p_es = p_io_stream;
1048
1049     for (size_t i=0; i<p_stream->segments.size(); i++)
1050     {
1051         p_stream->segments[i]->Preload();
1052     }
1053
1054     p_segment = p_stream->segments[0];
1055     if( p_segment->cluster != NULL )
1056     {
1057         msg_Warn( p_demux, "cannot find any cluster, damaged file ?" );
1058
1059         // reset the stream reading to the first cluster of the segment used
1060         p_stream->p_in->setFilePointer( p_segment->cluster->GetElementPosition() );
1061     }
1062
1063     /* get the files from the same dir from the same family (based on p_demux->psz_path) */
1064     if (p_demux->psz_path[0] != '\0' && !strcmp(p_demux->psz_access, ""))
1065     {
1066         // assume it's a regular file
1067         // get the directory path
1068         s_path = p_demux->psz_path;
1069         if (s_path.at(s_path.length() - 1) == DIRECTORY_SEPARATOR)
1070         {
1071             s_path = s_path.substr(0,s_path.length()-1);
1072         }
1073         else
1074         {
1075             if (s_path.find_last_of(DIRECTORY_SEPARATOR) > 0)
1076             {
1077                 s_path = s_path.substr(0,s_path.find_last_of(DIRECTORY_SEPARATOR));
1078             }
1079         }
1080
1081         struct dirent *p_file_item;
1082         DIR *p_src_dir = opendir(s_path.c_str());
1083
1084         if (p_src_dir != NULL)
1085         {
1086             while ((p_file_item = (dirent *) readdir(p_src_dir)))
1087             {
1088                 if (strlen(p_file_item->d_name) > 4)
1089                 {
1090                     s_filename = s_path + DIRECTORY_SEPARATOR + p_file_item->d_name;
1091
1092                     if (!s_filename.compare(p_demux->psz_path))
1093                         continue; // don't reuse the original opened file
1094
1095 #if defined(__GNUC__) && (__GNUC__ < 3)
1096                     if (!s_filename.compare("mkv", s_filename.length() - 3, 3) ||
1097                         !s_filename.compare("mka", s_filename.length() - 3, 3))
1098 #else
1099                     if (!s_filename.compare(s_filename.length() - 3, 3, "mkv") ||
1100                         !s_filename.compare(s_filename.length() - 3, 3, "mka"))
1101 #endif
1102                     {
1103                         // test wether this file belongs to the our family
1104                         StdIOCallback *p_file_io = new StdIOCallback(s_filename.c_str(), MODE_READ);
1105                         EbmlStream *p_estream = new EbmlStream(*p_file_io);
1106
1107                         p_stream = p_sys->AnalyseAllSegmentsFound( p_estream );
1108                         if ( p_stream == NULL )
1109                         {
1110                             msg_Dbg( p_demux, "the file '%s' will not be used", s_filename.c_str() );
1111                             delete p_estream;
1112                             delete p_file_io;
1113                         }
1114                         else
1115                         {
1116                             p_stream->p_in = p_file_io;
1117                             p_stream->p_es = p_estream;
1118                             p_sys->streams.push_back( p_stream );
1119                         }
1120                     }
1121                 }
1122             }
1123             closedir( p_src_dir );
1124         }
1125     }
1126
1127     p_sys->PreloadFamily( *p_segment );
1128     p_sys->PreloadLinked( p_segment );
1129     if ( !p_sys->PreparePlayback( NULL ) )
1130     {
1131         msg_Err( p_demux, "cannot use the segment" );
1132         goto error;
1133     }
1134
1135     return VLC_SUCCESS;
1136
1137 error:
1138     delete p_sys;
1139     return VLC_EGENERIC;
1140 }
1141
1142 /*****************************************************************************
1143  * Close: frees unused data
1144  *****************************************************************************/
1145 static void Close( vlc_object_t *p_this )
1146 {
1147     demux_t     *p_demux = (demux_t*)p_this;
1148     demux_sys_t *p_sys   = p_demux->p_sys;
1149
1150     delete p_sys;
1151 }
1152
1153 /*****************************************************************************
1154  * Control:
1155  *****************************************************************************/
1156 static int Control( demux_t *p_demux, int i_query, va_list args )
1157 {
1158     demux_sys_t        *p_sys = p_demux->p_sys;
1159     int64_t     *pi64;
1160     double      *pf, f;
1161     int         i_skp;
1162     size_t      i_idx;
1163
1164     vlc_meta_t **pp_meta;
1165
1166     switch( i_query )
1167     {
1168         case DEMUX_GET_META:
1169             pp_meta = (vlc_meta_t**)va_arg( args, vlc_meta_t** );
1170             *pp_meta = vlc_meta_Duplicate( p_sys->meta );
1171             return VLC_SUCCESS;
1172
1173         case DEMUX_GET_LENGTH:
1174             pi64 = (int64_t*)va_arg( args, int64_t * );
1175             if( p_sys->f_duration > 0.0 )
1176             {
1177                 *pi64 = (int64_t)(p_sys->f_duration * 1000);
1178                 return VLC_SUCCESS;
1179             }
1180             return VLC_EGENERIC;
1181
1182         case DEMUX_GET_POSITION:
1183             pf = (double*)va_arg( args, double * );
1184             if ( p_sys->f_duration > 0.0 )
1185                 *pf = (double)(p_sys->i_pts >= p_sys->i_start_pts ? p_sys->i_pts : p_sys->i_start_pts ) / (1000.0 * p_sys->f_duration);
1186             return VLC_SUCCESS;
1187
1188         case DEMUX_SET_POSITION:
1189             f = (double)va_arg( args, double );
1190             Seek( p_demux, -1, f, NULL );
1191             return VLC_SUCCESS;
1192
1193         case DEMUX_GET_TIME:
1194             pi64 = (int64_t*)va_arg( args, int64_t * );
1195             *pi64 = p_sys->i_pts;
1196             return VLC_SUCCESS;
1197
1198         case DEMUX_GET_TITLE_INFO:
1199             if( p_sys->titles.size() )
1200             {
1201                 input_title_t ***ppp_title = (input_title_t***)va_arg( args, input_title_t*** );
1202                 int *pi_int    = (int*)va_arg( args, int* );
1203
1204                 *pi_int = p_sys->titles.size();
1205                 *ppp_title = (input_title_t**)malloc( sizeof( input_title_t**) * p_sys->titles.size() );
1206
1207                 for( size_t i = 0; i < p_sys->titles.size(); i++ )
1208                 {
1209                     (*ppp_title)[i] = vlc_input_title_Duplicate( &p_sys->titles[i] );
1210                 }
1211
1212                 return VLC_SUCCESS;
1213             }
1214             return VLC_EGENERIC;
1215
1216         case DEMUX_SET_TITLE:
1217             /* TODO handle editions as titles */
1218             i_idx = (int)va_arg( args, int );
1219             if( i_idx < p_sys->used_segments.size() )
1220             {
1221                 p_sys->PreparePlayback( p_sys->used_segments[i_idx] );
1222                 return VLC_SUCCESS;
1223             }
1224             return VLC_EGENERIC;
1225
1226         case DEMUX_SET_SEEKPOINT:
1227             i_skp = (int)va_arg( args, int );
1228
1229             if( p_sys->titles.size() && i_skp < p_sys->titles[p_sys->i_current_title].i_seekpoint)
1230             {
1231                 Seek( p_demux, (int64_t)p_sys->titles[p_sys->i_current_title].seekpoint[i_skp]->i_time_offset, -1, NULL);
1232                 p_demux->info.i_seekpoint |= INPUT_UPDATE_SEEKPOINT;
1233                 p_demux->info.i_seekpoint = i_skp;
1234                 return VLC_SUCCESS;
1235             }
1236             return VLC_EGENERIC;
1237
1238         case DEMUX_SET_TIME:
1239         case DEMUX_GET_FPS:
1240         default:
1241             return VLC_EGENERIC;
1242     }
1243 }
1244
1245 int matroska_segment_c::BlockGet( KaxBlock **pp_block, int64_t *pi_ref1, int64_t *pi_ref2, int64_t *pi_duration )
1246 {
1247     *pp_block = NULL;
1248     *pi_ref1  = -1;
1249     *pi_ref2  = -1;
1250
1251     for( ;; )
1252     {
1253         EbmlElement *el;
1254         int         i_level;
1255
1256         if ( ep == NULL )
1257             return VLC_EGENERIC;
1258
1259         el = ep->Get();
1260         i_level = ep->GetLevel();
1261
1262         if( el == NULL && *pp_block != NULL )
1263         {
1264             /* update the index */
1265 #define idx index[i_index - 1]
1266             if( i_index > 0 && idx.i_time == -1 )
1267             {
1268                 idx.i_time        = (*pp_block)->GlobalTimecode() / (mtime_t)1000;
1269                 idx.b_key         = *pi_ref1 == -1 ? VLC_TRUE : VLC_FALSE;
1270             }
1271 #undef idx
1272             return VLC_SUCCESS;
1273         }
1274
1275         if( el == NULL )
1276         {
1277             if( ep->GetLevel() > 1 )
1278             {
1279                 ep->Up();
1280                 continue;
1281             }
1282             msg_Warn( &sys.demuxer, "EOF" );
1283             return VLC_EGENERIC;
1284         }
1285
1286         /* do parsing */
1287         if( i_level == 1 )
1288         {
1289             if( MKV_IS_ID( el, KaxCluster ) )
1290             {
1291                 cluster = (KaxCluster*)el;
1292
1293                 /* add it to the index */
1294                 if( i_index == 0 ||
1295                     ( i_index > 0 && index[i_index - 1].i_position < (int64_t)cluster->GetElementPosition() ) )
1296                 {
1297                     IndexAppendCluster( cluster );
1298                 }
1299
1300                 // reset silent tracks
1301                 for (size_t i=0; i<tracks.size(); i++)
1302                 {
1303                     tracks[i]->b_silent = VLC_FALSE;
1304                 }
1305
1306                 ep->Down();
1307             }
1308             else if( MKV_IS_ID( el, KaxCues ) )
1309             {
1310                 msg_Warn( &sys.demuxer, "find KaxCues FIXME" );
1311                 return VLC_EGENERIC;
1312             }
1313             else
1314             {
1315                 msg_Dbg( &sys.demuxer, "unknown (%s)", typeid( el ).name() );
1316             }
1317         }
1318         else if( i_level == 2 )
1319         {
1320             if( MKV_IS_ID( el, KaxClusterTimecode ) )
1321             {
1322                 KaxClusterTimecode &ctc = *(KaxClusterTimecode*)el;
1323
1324                 ctc.ReadData( es.I_O(), SCOPE_ALL_DATA );
1325                 cluster->InitTimecode( uint64( ctc ), i_timescale );
1326             }
1327             else if( MKV_IS_ID( el, KaxClusterSilentTracks ) )
1328             {
1329                 ep->Down();
1330             }
1331             else if( MKV_IS_ID( el, KaxBlockGroup ) )
1332             {
1333                 ep->Down();
1334             }
1335         }
1336         else if( i_level == 3 )
1337         {
1338             if( MKV_IS_ID( el, KaxBlock ) )
1339             {
1340                 *pp_block = (KaxBlock*)el;
1341
1342                 (*pp_block)->ReadData( es.I_O() );
1343                 (*pp_block)->SetParent( *cluster );
1344
1345                 ep->Keep();
1346             }
1347             else if( MKV_IS_ID( el, KaxBlockDuration ) )
1348             {
1349                 KaxBlockDuration &dur = *(KaxBlockDuration*)el;
1350
1351                 dur.ReadData( es.I_O() );
1352                 *pi_duration = uint64( dur );
1353             }
1354             else if( MKV_IS_ID( el, KaxReferenceBlock ) )
1355             {
1356                 KaxReferenceBlock &ref = *(KaxReferenceBlock*)el;
1357
1358                 ref.ReadData( es.I_O() );
1359                 if( *pi_ref1 == -1 )
1360                 {
1361                     *pi_ref1 = int64( ref );
1362                 }
1363                 else
1364                 {
1365                     *pi_ref2 = int64( ref );
1366                 }
1367             }
1368             else if( MKV_IS_ID( el, KaxClusterSilentTrackNumber ) )
1369             {
1370                 KaxClusterSilentTrackNumber &track_num = *(KaxClusterSilentTrackNumber*)el;
1371                 track_num.ReadData( es.I_O() );
1372                 // find the track
1373                 for (size_t i=0; i<tracks.size(); i++)
1374                 {
1375                     if ( tracks[i]->i_number == uint32(track_num))
1376                     {
1377                         tracks[i]->b_silent = VLC_TRUE;
1378                         break;
1379                     }
1380                 }
1381             }
1382         }
1383         else
1384         {
1385             msg_Err( &sys.demuxer, "invalid level = %d", i_level );
1386             return VLC_EGENERIC;
1387         }
1388     }
1389 }
1390
1391 static block_t *MemToBlock( demux_t *p_demux, uint8_t *p_mem, int i_mem)
1392 {
1393     block_t *p_block;
1394     if( !(p_block = block_New( p_demux, i_mem ) ) ) return NULL;
1395     memcpy( p_block->p_buffer, p_mem, i_mem );
1396     //p_block->i_rate = p_input->stream.control.i_rate;
1397     return p_block;
1398 }
1399
1400 static void BlockDecode( demux_t *p_demux, KaxBlock *block, mtime_t i_pts,
1401                          mtime_t i_duration )
1402 {
1403     demux_sys_t        *p_sys = p_demux->p_sys;
1404     matroska_segment_c *p_segment = p_sys->p_current_segment->Segment();
1405
1406     size_t          i_track;
1407     unsigned int    i;
1408     vlc_bool_t      b;
1409
1410 #define tk  p_segment->tracks[i_track]
1411     for( i_track = 0; i_track < p_segment->tracks.size(); i_track++ )
1412     {
1413         if( tk->i_number == block->TrackNum() )
1414         {
1415             break;
1416         }
1417     }
1418
1419     if( i_track >= p_segment->tracks.size() )
1420     {
1421         msg_Err( p_demux, "invalid track number=%d", block->TrackNum() );
1422         return;
1423     }
1424     if( tk->p_es == NULL )
1425     {
1426         msg_Err( p_demux, "unknown track number=%d", block->TrackNum() );
1427         return;
1428     }
1429     if( i_pts < p_sys->i_start_pts && tk->fmt.i_cat == AUDIO_ES )
1430     {
1431         return; /* discard audio packets that shouldn't be rendered */
1432     }
1433
1434     es_out_Control( p_demux->out, ES_OUT_GET_ES_STATE, tk->p_es, &b );
1435     if( !b )
1436     {
1437         tk->b_inited = VLC_FALSE;
1438         return;
1439     }
1440
1441     /* First send init data */
1442     if( !tk->b_inited && tk->i_data_init > 0 )
1443     {
1444         block_t *p_init;
1445
1446         msg_Dbg( p_demux, "sending header (%d bytes)", tk->i_data_init );
1447         p_init = MemToBlock( p_demux, tk->p_data_init, tk->i_data_init );
1448         if( p_init ) es_out_Send( p_demux->out, tk->p_es, p_init );
1449     }
1450     tk->b_inited = VLC_TRUE;
1451
1452
1453     for( i = 0; i < block->NumberFrames(); i++ )
1454     {
1455         block_t *p_block;
1456         DataBuffer &data = block->GetBuffer(i);
1457
1458         p_block = MemToBlock( p_demux, data.Buffer(), data.Size() );
1459
1460         if( p_block == NULL )
1461         {
1462             break;
1463         }
1464
1465 #if defined(HAVE_ZLIB_H)
1466         if( tk->i_compression_type )
1467         {
1468             p_block = block_zlib_decompress( VLC_OBJECT(p_demux), p_block );
1469         }
1470 #endif
1471
1472         // TODO implement correct timestamping when B frames are used
1473         if( tk->fmt.i_cat != VIDEO_ES )
1474         {
1475             p_block->i_dts = p_block->i_pts = i_pts;
1476         }
1477         else
1478         {
1479             p_block->i_dts = i_pts;
1480             p_block->i_pts = 0;
1481         }
1482
1483         if( tk->fmt.i_cat == SPU_ES && strcmp( tk->psz_codec, "S_VOBSUB" ) )
1484         {
1485             p_block->i_length = i_duration * 1000;
1486         }
1487
1488         es_out_Send( p_demux->out, tk->p_es, p_block );
1489
1490         /* use time stamp only for first block */
1491         i_pts = 0;
1492     }
1493
1494 #undef tk
1495 }
1496
1497 matroska_stream_c *demux_sys_t::AnalyseAllSegmentsFound( EbmlStream *p_estream )
1498 {
1499     int i_upper_lvl = 0;
1500     size_t i;
1501     EbmlElement *p_l0, *p_l1, *p_l2;
1502     bool b_keep_stream = false, b_keep_segment;
1503
1504     // verify the EBML Header
1505     p_l0 = p_estream->FindNextID(EbmlHead::ClassInfos, 0xFFFFFFFFL);
1506     if (p_l0 == NULL)
1507     {
1508         return NULL;
1509     }
1510     p_l0->SkipData(*p_estream, EbmlHead_Context);
1511     delete p_l0;
1512
1513     // find all segments in this file
1514     p_l0 = p_estream->FindNextID(KaxSegment::ClassInfos, 0xFFFFFFFFL);
1515     if (p_l0 == NULL)
1516     {
1517         return NULL;
1518     }
1519
1520     matroska_stream_c *p_stream1 = new matroska_stream_c( *this );
1521
1522     while (p_l0 != 0)
1523     {
1524         if (EbmlId(*p_l0) == KaxSegment::ClassInfos.GlobalId)
1525         {
1526             EbmlParser  *ep;
1527             matroska_segment_c *p_segment1 = new matroska_segment_c( *this, *p_estream );
1528             b_keep_segment = false;
1529
1530             ep = new EbmlParser(p_estream, p_l0, &demuxer );
1531             p_segment1->ep = ep;
1532             p_segment1->segment = (KaxSegment*)p_l0;
1533
1534             while ((p_l1 = ep->Get()))
1535             {
1536                 if (MKV_IS_ID(p_l1, KaxInfo))
1537                 {
1538                     // find the families of this segment
1539                     KaxInfo *p_info = static_cast<KaxInfo*>(p_l1);
1540
1541                     p_info->Read(*p_estream, KaxInfo::ClassInfos.Context, i_upper_lvl, p_l2, true);
1542                     for( i = 0; i < p_info->ListSize(); i++ )
1543                     {
1544                         EbmlElement *l = (*p_info)[i];
1545
1546                         if( MKV_IS_ID( l, KaxSegmentUID ) )
1547                         {
1548                             KaxSegmentUID *p_uid = static_cast<KaxSegmentUID*>(l);
1549                             b_keep_segment = (FindSegment( *p_uid ) == NULL);
1550                             if ( !b_keep_segment )
1551                                 break; // this segment is already known
1552                             opened_segments.push_back( p_segment1 );
1553                             p_segment1->segment_uid = *( new KaxSegmentUID(*p_uid) );
1554                         }
1555                         else if( MKV_IS_ID( l, KaxPrevUID ) )
1556                         {
1557                             p_segment1->prev_segment_uid = *( new KaxPrevUID( *static_cast<KaxPrevUID*>(l) ) );
1558                         }
1559                         else if( MKV_IS_ID( l, KaxNextUID ) )
1560                         {
1561                             p_segment1->next_segment_uid = *( new KaxNextUID( *static_cast<KaxNextUID*>(l) ) );
1562                         }
1563                         else if( MKV_IS_ID( l, KaxSegmentFamily ) )
1564                         {
1565                             KaxSegmentFamily *p_fam = new KaxSegmentFamily( *static_cast<KaxSegmentFamily*>(l) );
1566                             p_segment1->families.push_back( *p_fam );
1567                         }
1568                     }
1569                     break;
1570                 }
1571             }
1572             if ( b_keep_segment )
1573             {
1574                 b_keep_stream = true;
1575                 p_stream1->segments.push_back( p_segment1 );
1576             }
1577             else
1578                 delete p_segment1;
1579         }
1580
1581         p_l0->SkipData(*p_estream, EbmlHead_Context);
1582         p_l0 = p_estream->FindNextID(KaxSegment::ClassInfos, 0xFFFFFFFFL);
1583     }
1584
1585     if ( !b_keep_stream )
1586     {
1587         delete p_stream1;
1588         p_stream1 = NULL;
1589     }
1590
1591     return p_stream1;
1592 }
1593
1594 bool matroska_segment_c::Select( mtime_t i_start_time )
1595 {
1596     size_t i_track;
1597
1598     /* add all es */
1599     msg_Dbg( &sys.demuxer, "found %d es", tracks.size() );
1600     for( i_track = 0; i_track < tracks.size(); i_track++ )
1601     {
1602 #define tk  tracks[i_track]
1603         if( tk->fmt.i_cat == UNKNOWN_ES )
1604         {
1605             msg_Warn( &sys.demuxer, "invalid track[%d, n=%d]", i_track, tk->i_number );
1606             tk->p_es = NULL;
1607             continue;
1608         }
1609
1610         if( !strcmp( tk->psz_codec, "V_MS/VFW/FOURCC" ) )
1611         {
1612             if( tk->i_extra_data < (int)sizeof( BITMAPINFOHEADER ) )
1613             {
1614                 msg_Err( &sys.demuxer, "missing/invalid BITMAPINFOHEADER" );
1615                 tk->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
1616             }
1617             else
1618             {
1619                 BITMAPINFOHEADER *p_bih = (BITMAPINFOHEADER*)tk->p_extra_data;
1620
1621                 tk->fmt.video.i_width = GetDWLE( &p_bih->biWidth );
1622                 tk->fmt.video.i_height= GetDWLE( &p_bih->biHeight );
1623                 tk->fmt.i_codec       = GetFOURCC( &p_bih->biCompression );
1624
1625                 tk->fmt.i_extra       = GetDWLE( &p_bih->biSize ) - sizeof( BITMAPINFOHEADER );
1626                 if( tk->fmt.i_extra > 0 )
1627                 {
1628                     tk->fmt.p_extra = malloc( tk->fmt.i_extra );
1629                     memcpy( tk->fmt.p_extra, &p_bih[1], tk->fmt.i_extra );
1630                 }
1631             }
1632         }
1633         else if( !strcmp( tk->psz_codec, "V_MPEG1" ) ||
1634                  !strcmp( tk->psz_codec, "V_MPEG2" ) )
1635         {
1636             tk->fmt.i_codec = VLC_FOURCC( 'm', 'p', 'g', 'v' );
1637         }
1638         else if( !strncmp( tk->psz_codec, "V_MPEG4", 7 ) )
1639         {
1640             if( !strcmp( tk->psz_codec, "V_MPEG4/MS/V3" ) )
1641             {
1642                 tk->fmt.i_codec = VLC_FOURCC( 'D', 'I', 'V', '3' );
1643             }
1644             else if( !strcmp( tk->psz_codec, "V_MPEG4/ISO/AVC" ) )
1645             {
1646                 tk->fmt.i_codec = VLC_FOURCC( 'a', 'v', 'c', '1' );
1647                 tk->fmt.b_packetized = VLC_FALSE;
1648                 tk->fmt.i_extra = tk->i_extra_data;
1649                 tk->fmt.p_extra = malloc( tk->i_extra_data );
1650                 memcpy( tk->fmt.p_extra,tk->p_extra_data, tk->i_extra_data );
1651             }
1652             else
1653             {
1654                 tk->fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'v' );
1655             }
1656         }
1657         else if( !strcmp( tk->psz_codec, "V_QUICKTIME" ) )
1658         {
1659             MP4_Box_t *p_box = (MP4_Box_t*)malloc( sizeof( MP4_Box_t ) );
1660 #ifdef VSLHC
1661             stream_t *p_mp4_stream = stream_MemoryNew( VLC_OBJECT(&sys.demuxer),
1662                                                        tk->p_extra_data,
1663                                                        tk->i_extra_data );
1664 #else
1665             stream_t *p_mp4_stream = stream_MemoryNew( VLC_OBJECT(&sys.demuxer),
1666                                                        tk->p_extra_data,
1667                                                        tk->i_extra_data,
1668                                                        VLC_FALSE );
1669 #endif
1670             MP4_ReadBoxCommon( p_mp4_stream, p_box );
1671             MP4_ReadBox_sample_vide( p_mp4_stream, p_box );
1672             tk->fmt.i_codec = p_box->i_type;
1673             tk->fmt.video.i_width = p_box->data.p_sample_vide->i_width;
1674             tk->fmt.video.i_height = p_box->data.p_sample_vide->i_height;
1675             tk->fmt.i_extra = p_box->data.p_sample_vide->i_qt_image_description;
1676             tk->fmt.p_extra = malloc( tk->fmt.i_extra );
1677             memcpy( tk->fmt.p_extra, p_box->data.p_sample_vide->p_qt_image_description, tk->fmt.i_extra );
1678             MP4_FreeBox_sample_vide( p_box );
1679 #ifdef VSLHC
1680             stream_MemoryDelete( p_mp4_stream, VLC_TRUE );
1681 #else
1682             stream_Delete( p_mp4_stream );
1683 #endif
1684         }
1685         else if( !strcmp( tk->psz_codec, "A_MS/ACM" ) )
1686         {
1687             if( tk->i_extra_data < (int)sizeof( WAVEFORMATEX ) )
1688             {
1689                 msg_Err( &sys.demuxer, "missing/invalid WAVEFORMATEX" );
1690                 tk->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
1691             }
1692             else
1693             {
1694                 WAVEFORMATEX *p_wf = (WAVEFORMATEX*)tk->p_extra_data;
1695
1696                 wf_tag_to_fourcc( GetWLE( &p_wf->wFormatTag ), &tk->fmt.i_codec, NULL );
1697
1698                 tk->fmt.audio.i_channels   = GetWLE( &p_wf->nChannels );
1699                 tk->fmt.audio.i_rate = GetDWLE( &p_wf->nSamplesPerSec );
1700                 tk->fmt.i_bitrate    = GetDWLE( &p_wf->nAvgBytesPerSec ) * 8;
1701                 tk->fmt.audio.i_blockalign = GetWLE( &p_wf->nBlockAlign );;
1702                 tk->fmt.audio.i_bitspersample = GetWLE( &p_wf->wBitsPerSample );
1703
1704                 tk->fmt.i_extra            = GetWLE( &p_wf->cbSize );
1705                 if( tk->fmt.i_extra > 0 )
1706                 {
1707                     tk->fmt.p_extra = malloc( tk->fmt.i_extra );
1708                     memcpy( tk->fmt.p_extra, &p_wf[1], tk->fmt.i_extra );
1709                 }
1710             }
1711         }
1712         else if( !strcmp( tk->psz_codec, "A_MPEG/L3" ) ||
1713                  !strcmp( tk->psz_codec, "A_MPEG/L2" ) ||
1714                  !strcmp( tk->psz_codec, "A_MPEG/L1" ) )
1715         {
1716             tk->fmt.i_codec = VLC_FOURCC( 'm', 'p', 'g', 'a' );
1717         }
1718         else if( !strcmp( tk->psz_codec, "A_AC3" ) )
1719         {
1720             tk->fmt.i_codec = VLC_FOURCC( 'a', '5', '2', ' ' );
1721         }
1722         else if( !strcmp( tk->psz_codec, "A_DTS" ) )
1723         {
1724             tk->fmt.i_codec = VLC_FOURCC( 'd', 't', 's', ' ' );
1725         }
1726         else if( !strcmp( tk->psz_codec, "A_FLAC" ) )
1727         {
1728             tk->fmt.i_codec = VLC_FOURCC( 'f', 'l', 'a', 'c' );
1729             tk->fmt.i_extra = tk->i_extra_data;
1730             tk->fmt.p_extra = malloc( tk->i_extra_data );
1731             memcpy( tk->fmt.p_extra,tk->p_extra_data, tk->i_extra_data );
1732         }
1733         else if( !strcmp( tk->psz_codec, "A_VORBIS" ) )
1734         {
1735             int i, i_offset = 1, i_size[3], i_extra;
1736             uint8_t *p_extra;
1737
1738             tk->fmt.i_codec = VLC_FOURCC( 'v', 'o', 'r', 'b' );
1739
1740             /* Split the 3 headers */
1741             if( tk->p_extra_data[0] != 0x02 )
1742                 msg_Err( &sys.demuxer, "invalid vorbis header" );
1743
1744             for( i = 0; i < 2; i++ )
1745             {
1746                 i_size[i] = 0;
1747                 while( i_offset < tk->i_extra_data )
1748                 {
1749                     i_size[i] += tk->p_extra_data[i_offset];
1750                     if( tk->p_extra_data[i_offset++] != 0xff ) break;
1751                 }
1752             }
1753
1754             i_size[0] = __MIN(i_size[0], tk->i_extra_data - i_offset);
1755             i_size[1] = __MIN(i_size[1], tk->i_extra_data -i_offset -i_size[0]);
1756             i_size[2] = tk->i_extra_data - i_offset - i_size[0] - i_size[1];
1757
1758             tk->fmt.i_extra = 3 * 2 + i_size[0] + i_size[1] + i_size[2];
1759             tk->fmt.p_extra = malloc( tk->fmt.i_extra );
1760             p_extra = (uint8_t *)tk->fmt.p_extra; i_extra = 0;
1761             for( i = 0; i < 3; i++ )
1762             {
1763                 *(p_extra++) = i_size[i] >> 8;
1764                 *(p_extra++) = i_size[i] & 0xFF;
1765                 memcpy( p_extra, tk->p_extra_data + i_offset + i_extra,
1766                         i_size[i] );
1767                 p_extra += i_size[i];
1768                 i_extra += i_size[i];
1769             }
1770         }
1771         else if( !strncmp( tk->psz_codec, "A_AAC/MPEG2/", strlen( "A_AAC/MPEG2/" ) ) ||
1772                  !strncmp( tk->psz_codec, "A_AAC/MPEG4/", strlen( "A_AAC/MPEG4/" ) ) )
1773         {
1774             int i_profile, i_srate;
1775             static unsigned int i_sample_rates[] =
1776             {
1777                     96000, 88200, 64000, 48000, 44100, 32000, 24000, 22050,
1778                         16000, 12000, 11025, 8000,  7350,  0,     0,     0
1779             };
1780
1781             tk->fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'a' );
1782             /* create data for faad (MP4DecSpecificDescrTag)*/
1783
1784             if( !strcmp( &tk->psz_codec[12], "MAIN" ) )
1785             {
1786                 i_profile = 0;
1787             }
1788             else if( !strcmp( &tk->psz_codec[12], "LC" ) )
1789             {
1790                 i_profile = 1;
1791             }
1792             else if( !strcmp( &tk->psz_codec[12], "SSR" ) )
1793             {
1794                 i_profile = 2;
1795             }
1796             else
1797             {
1798                 i_profile = 3;
1799             }
1800
1801             for( i_srate = 0; i_srate < 13; i_srate++ )
1802             {
1803                 if( i_sample_rates[i_srate] == tk->fmt.audio.i_rate )
1804                 {
1805                     break;
1806                 }
1807             }
1808             msg_Dbg( &sys.demuxer, "profile=%d srate=%d", i_profile, i_srate );
1809
1810             tk->fmt.i_extra = 2;
1811             tk->fmt.p_extra = malloc( tk->fmt.i_extra );
1812             ((uint8_t*)tk->fmt.p_extra)[0] = ((i_profile + 1) << 3) | ((i_srate&0xe) >> 1);
1813             ((uint8_t*)tk->fmt.p_extra)[1] = ((i_srate & 0x1) << 7) | (tk->fmt.audio.i_channels << 3);
1814         }
1815         else if( !strcmp( tk->psz_codec, "A_PCM/INT/BIG" ) ||
1816                  !strcmp( tk->psz_codec, "A_PCM/INT/LIT" ) ||
1817                  !strcmp( tk->psz_codec, "A_PCM/FLOAT/IEEE" ) )
1818         {
1819             if( !strcmp( tk->psz_codec, "A_PCM/INT/BIG" ) )
1820             {
1821                 tk->fmt.i_codec = VLC_FOURCC( 't', 'w', 'o', 's' );
1822             }
1823             else
1824             {
1825                 tk->fmt.i_codec = VLC_FOURCC( 'a', 'r', 'a', 'w' );
1826             }
1827             tk->fmt.audio.i_blockalign = ( tk->fmt.audio.i_bitspersample + 7 ) / 8 * tk->fmt.audio.i_channels;
1828         }
1829         else if( !strcmp( tk->psz_codec, "A_TTA1" ) )
1830         {
1831             /* FIXME: support this codec */
1832             msg_Err( &sys.demuxer, "TTA not supported yet[%d, n=%d]", i_track, tk->i_number );
1833             tk->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
1834         }
1835         else if( !strcmp( tk->psz_codec, "A_WAVPACK4" ) )
1836         {
1837             /* FIXME: support this codec */
1838             msg_Err( &sys.demuxer, "Wavpack not supported yet[%d, n=%d]", i_track, tk->i_number );
1839             tk->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
1840         }
1841         else if( !strcmp( tk->psz_codec, "S_TEXT/UTF8" ) )
1842         {
1843             tk->fmt.i_codec = VLC_FOURCC( 's', 'u', 'b', 't' );
1844             tk->fmt.subs.psz_encoding = strdup( "UTF-8" );
1845         }
1846         else if( !strcmp( tk->psz_codec, "S_TEXT/SSA" ) ||
1847                  !strcmp( tk->psz_codec, "S_TEXT/ASS" ) ||
1848                  !strcmp( tk->psz_codec, "S_SSA" ) ||
1849                  !strcmp( tk->psz_codec, "S_ASS" ))
1850         {
1851             tk->fmt.i_codec = VLC_FOURCC( 's', 's', 'a', ' ' );
1852             tk->fmt.subs.psz_encoding = strdup( "UTF-8" );
1853         }
1854         else if( !strcmp( tk->psz_codec, "S_VOBSUB" ) )
1855         {
1856             tk->fmt.i_codec = VLC_FOURCC( 's','p','u',' ' );
1857             if( tk->i_extra_data )
1858             {
1859                 char *p_start;
1860                 char *p_buf = (char *)malloc( tk->i_extra_data + 1);
1861                 memcpy( p_buf, tk->p_extra_data , tk->i_extra_data );
1862                 p_buf[tk->i_extra_data] = '\0';
1863
1864                 p_start = strstr( p_buf, "size:" );
1865                 if( sscanf( p_start, "size: %dx%d",
1866                         &tk->fmt.subs.spu.i_original_frame_width, &tk->fmt.subs.spu.i_original_frame_height ) == 2 )
1867                 {
1868                     msg_Dbg( &sys.demuxer, "original frame size vobsubs: %dx%d", tk->fmt.subs.spu.i_original_frame_width, tk->fmt.subs.spu.i_original_frame_height );
1869                 }
1870                 else
1871                 {
1872                     msg_Warn( &sys.demuxer, "reading original frame size for vobsub failed" );
1873                 }
1874                 free( p_buf );
1875             }
1876         }
1877         else if( !strcmp( tk->psz_codec, "B_VOBBTN" ) )
1878         {
1879             /* FIXME: support this codec */
1880             msg_Err( &sys.demuxer, "Vob Buttons not supported yet[%d, n=%d]", i_track, tk->i_number );
1881             tk->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
1882         }
1883         else
1884         {
1885             msg_Err( &sys.demuxer, "unknow codec id=`%s'", tk->psz_codec );
1886             tk->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
1887         }
1888         if( tk->b_default )
1889         {
1890             tk->fmt.i_priority = 1000;
1891         }
1892
1893         tk->p_es = es_out_Add( sys.demuxer.out, &tk->fmt );
1894
1895         es_out_Control( sys.demuxer.out, ES_OUT_SET_NEXT_DISPLAY_TIME, tk->p_es, i_start_time );
1896 #undef tk
1897     }
1898
1899     sys.i_start_pts = i_start_time;
1900     // reset the stream reading to the first cluster of the segment used
1901     es.I_O().setFilePointer( i_start_pos );
1902
1903     delete ep;
1904     ep = new EbmlParser( &es, segment, &sys.demuxer );
1905
1906     return true;
1907 }
1908
1909 void matroska_segment_c::UnSelect( )
1910 {
1911     size_t i_track;
1912
1913     for( i_track = 0; i_track < tracks.size(); i_track++ )
1914     {
1915 #define tk  tracks[i_track]
1916         if ( tk->p_es != NULL )
1917         {
1918             es_out_Del( sys.demuxer.out, tk->p_es );
1919             tk->p_es = NULL;
1920         }
1921 #undef tk
1922     }
1923     delete ep;
1924     ep = NULL;
1925 }
1926
1927 void virtual_segment_c::PrepareChapters( )
1928 {
1929     if ( linked_segments.size() == 0 )
1930         return;
1931
1932     // !!! should be called only once !!!
1933     matroska_segment_c *p_segment;
1934     size_t i, j;
1935
1936     // copy editions from the first segment
1937     p_segment = linked_segments[0];
1938     p_editions = &p_segment->stored_editions;
1939
1940     for ( i=1 ; i<linked_segments.size(); i++ )
1941     {
1942         p_segment = linked_segments[i];
1943         // FIXME assume we have the same editions in all segments
1944         for (j=0; j<p_segment->stored_editions.size(); j++)
1945             (*p_editions)[j]->Append( *p_segment->stored_editions[j] );
1946     }
1947 }
1948
1949 std::string chapter_edition_c::GetMainName() const
1950 {
1951     if ( sub_chapters.size() )
1952     {
1953         return sub_chapters[0]->GetCodecName( true );
1954     }
1955     return "";
1956 }
1957
1958 int chapter_item_c::PublishChapters( input_title_t & title, int & i_user_chapters, int i_level )
1959 {
1960     // add support for meta-elements from codec like DVD Titles
1961     if ( !b_display_seekpoint || psz_name == "" )
1962     {
1963         psz_name = GetCodecName();
1964         if ( psz_name != "" )
1965             b_display_seekpoint = true;
1966     }
1967
1968     if (b_display_seekpoint)
1969     {
1970         seekpoint_t *sk = vlc_seekpoint_New();
1971
1972         sk->i_level = i_level;
1973         sk->i_time_offset = i_start_time;
1974         sk->psz_name = strdup( psz_name.c_str() );
1975
1976         // A start time of '0' is ok. A missing ChapterTime element is ok, too, because '0' is its default value.
1977         title.i_seekpoint++;
1978         title.seekpoint = (seekpoint_t**)realloc( title.seekpoint, title.i_seekpoint * sizeof( seekpoint_t* ) );
1979         title.seekpoint[title.i_seekpoint-1] = sk;
1980
1981         if ( b_user_display )
1982             i_user_chapters++;
1983     }
1984
1985     for ( size_t i=0; i<sub_chapters.size() ; i++)
1986     {
1987         sub_chapters[i]->PublishChapters( title, i_user_chapters, i_level+1 );
1988     }
1989
1990     i_seekpoint_num = i_user_chapters;
1991
1992     return i_user_chapters;
1993 }
1994
1995 bool virtual_segment_c::UpdateCurrentToChapter( demux_t & demux )
1996 {
1997     demux_sys_t & sys = *demux.p_sys;
1998     chapter_item_c *psz_curr_chapter;
1999
2000     /* update current chapter/seekpoint */
2001     if ( p_editions->size() )
2002     {
2003         /* 1st, we need to know in which chapter we are */
2004         psz_curr_chapter = (*p_editions)[i_current_edition]->FindTimecode( sys.i_pts );
2005
2006         /* we have moved to a new chapter */
2007         if (psz_curr_chapter != NULL && psz_current_chapter != psz_curr_chapter)
2008         {
2009             if ( (*p_editions)[i_current_edition]->b_ordered )
2010             {
2011                 // Leave/Enter up to the link point
2012                 if ( !psz_curr_chapter->EnterAndLeave( psz_current_chapter ) )
2013                 {
2014                     // only seek if necessary
2015                     if ( psz_current_chapter == NULL || (psz_current_chapter->i_end_time != psz_curr_chapter->i_start_time) )
2016                         Seek( demux, sys.i_pts, 0, psz_curr_chapter );
2017                     psz_current_chapter = psz_curr_chapter;
2018                 }
2019             }
2020             else if ( psz_curr_chapter->i_seekpoint_num > 0 )
2021             {
2022                 demux.info.i_update |= INPUT_UPDATE_TITLE | INPUT_UPDATE_SEEKPOINT;
2023                 demux.info.i_title = sys.i_current_title = i_sys_title;
2024                 demux.info.i_seekpoint = psz_curr_chapter->i_seekpoint_num - 1;
2025                 psz_current_chapter = psz_curr_chapter;
2026             }
2027             else
2028             {
2029                 psz_current_chapter = psz_curr_chapter;
2030             }
2031
2032             return true;
2033         }
2034     }
2035     return false;
2036 }
2037
2038 chapter_item_c *virtual_segment_c::BrowseCodecPrivate( unsigned int codec_id,
2039                                     bool (*match)(const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size ),
2040                                     const void *p_cookie,
2041                                     size_t i_cookie_size )
2042 {
2043     // FIXME don't assume it is the first edition
2044     std::vector<chapter_edition_c*>::iterator index = p_editions->begin();
2045     if ( index != p_editions->end() )
2046     {
2047         chapter_item_c *p_result = (*index)->BrowseCodecPrivate( codec_id, match, p_cookie, i_cookie_size );
2048         if ( p_result != NULL )
2049             return p_result;
2050     }
2051     return NULL;
2052 }
2053
2054 chapter_item_c *virtual_segment_c::FindChapter( int64_t i_find_uid )
2055 {
2056     // FIXME don't assume it is the first edition
2057     std::vector<chapter_edition_c*>::iterator index = p_editions->begin();
2058     if ( index != p_editions->end() )
2059     {
2060         chapter_item_c *p_result = (*index)->FindChapter( i_find_uid );
2061         if ( p_result != NULL )
2062             return p_result;
2063     }
2064     return NULL;
2065 }
2066
2067 chapter_item_c *chapter_item_c::BrowseCodecPrivate( unsigned int codec_id,
2068                                     bool (*match)(const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size ),
2069                                     const void *p_cookie,
2070                                     size_t i_cookie_size )
2071 {
2072     // this chapter
2073     std::vector<chapter_codec_cmds_c*>::const_iterator index = codecs.begin();
2074     while ( index != codecs.end() )
2075     {
2076         if ( match( **index ,p_cookie, i_cookie_size ) )
2077             return this;
2078         index++;
2079     }
2080
2081     // sub-chapters
2082     chapter_item_c *p_result = NULL;
2083     std::vector<chapter_item_c*>::const_iterator index2 = sub_chapters.begin();
2084     while ( index2 != sub_chapters.end() )
2085     {
2086         p_result = (*index2)->BrowseCodecPrivate( codec_id, match, p_cookie, i_cookie_size );
2087         if ( p_result != NULL )
2088             return p_result;
2089         index2++;
2090     }
2091
2092     return p_result;
2093 }
2094
2095 void chapter_item_c::Append( const chapter_item_c & chapter )
2096 {
2097     // we are appending content for the same chapter UID
2098     size_t i;
2099     chapter_item_c *p_chapter;
2100
2101     for ( i=0; i<chapter.sub_chapters.size(); i++ )
2102     {
2103         p_chapter = FindChapter( chapter.sub_chapters[i]->i_uid );
2104         if ( p_chapter != NULL )
2105         {
2106             p_chapter->Append( *chapter.sub_chapters[i] );
2107         }
2108         else
2109         {
2110             sub_chapters.push_back( chapter.sub_chapters[i] );
2111         }
2112     }
2113
2114     i_user_start_time = min( i_user_start_time, chapter.i_user_start_time );
2115     i_user_end_time = max( i_user_end_time, chapter.i_user_end_time );
2116 }
2117
2118 chapter_item_c * chapter_item_c::FindChapter( int64_t i_find_uid )
2119 {
2120     size_t i;
2121     chapter_item_c *p_result = NULL;
2122
2123     if ( i_uid == i_find_uid )
2124         return this;
2125
2126     for ( i=0; i<sub_chapters.size(); i++)
2127     {
2128         p_result = sub_chapters[i]->FindChapter( i_find_uid );
2129         if ( p_result != NULL )
2130             break;
2131     }
2132     return p_result;
2133 }
2134
2135 std::string chapter_item_c::GetCodecName( bool f_for_title ) const
2136 {
2137     std::string result;
2138
2139     std::vector<chapter_codec_cmds_c*>::const_iterator index = codecs.begin();
2140     while ( index != codecs.end() )
2141     {
2142         result = (*index)->GetCodecName( f_for_title );
2143         if ( result != "" )
2144             break;
2145         index++;
2146     }
2147
2148     return result;
2149 }
2150
2151 std::string dvd_chapter_codec_c::GetCodecName( bool f_for_title ) const
2152 {
2153     std::string result;
2154     if ( m_private_data.GetSize() >= 3)
2155     {
2156         const binary* p_data = m_private_data.GetBuffer();
2157 /*        if ( p_data[0] == MATROSKA_DVD_LEVEL_TT )
2158         {
2159             uint16_t i_title = (p_data[1] << 8) + p_data[2];
2160             char psz_str[11];
2161             sprintf( psz_str, " %d  ---", i_title );
2162             result = N_("---  DVD Title");
2163             result += psz_str;
2164         }
2165         else */ if ( p_data[0] == MATROSKA_DVD_LEVEL_LU )
2166         {
2167             char psz_str[11];
2168             sprintf( psz_str, " (%c%c)  ---", p_data[1], p_data[2] );
2169             result = N_("---  DVD Menu");
2170             result += psz_str;
2171         }
2172         else if ( p_data[0] == MATROSKA_DVD_LEVEL_SS && f_for_title )
2173         {
2174             if ( p_data[1] == 0x00 )
2175                 result = N_("First Played");
2176             else if ( p_data[1] == 0xC0 )
2177                 result = N_("Video Manager");
2178             else if ( p_data[1] == 0x80 )
2179             {
2180                 uint16_t i_title = (p_data[2] << 8) + p_data[3];
2181                 char psz_str[20];
2182                 sprintf( psz_str, " %d -----", i_title );
2183                 result = N_("----- Title");
2184                 result += psz_str;
2185             }
2186         }
2187     }
2188
2189     return result;
2190 }
2191
2192 static void Seek( demux_t *p_demux, mtime_t i_date, double f_percent, chapter_item_c *psz_chapter )
2193 {
2194     demux_sys_t        *p_sys = p_demux->p_sys;
2195     virtual_segment_c  *p_vsegment = p_sys->p_current_segment;
2196     matroska_segment_c *p_segment = p_vsegment->Segment();
2197     mtime_t            i_time_offset = 0;
2198
2199     int         i_index;
2200
2201     msg_Dbg( p_demux, "seek request to "I64Fd" (%f%%)", i_date, f_percent );
2202     if( i_date < 0 && f_percent < 0 )
2203     {
2204         msg_Warn( p_demux, "cannot seek nowhere !" );
2205         return;
2206     }
2207     if( f_percent > 1.0 )
2208     {
2209         msg_Warn( p_demux, "cannot seek so far !" );
2210         return;
2211     }
2212
2213     /* seek without index or without date */
2214     if( f_percent >= 0 && (config_GetInt( p_demux, "mkv-seek-percent" ) || !p_segment->b_cues || i_date < 0 ))
2215     {
2216         if (p_sys->f_duration >= 0)
2217         {
2218             i_date = int64_t( f_percent * p_sys->f_duration * 1000.0 );
2219         }
2220         else
2221         {
2222             int64_t i_pos = int64_t( f_percent * stream_Size( p_demux->s ) );
2223
2224             msg_Dbg( p_demux, "inacurate way of seeking" );
2225             for( i_index = 0; i_index < p_segment->i_index; i_index++ )
2226             {
2227                 if( p_segment->index[i_index].i_position >= i_pos)
2228                 {
2229                     break;
2230                 }
2231             }
2232             if( i_index == p_segment->i_index )
2233             {
2234                 i_index--;
2235             }
2236
2237             i_date = p_segment->index[i_index].i_time;
2238
2239 #if 0
2240             if( p_segment->index[i_index].i_position < i_pos )
2241             {
2242                 EbmlElement *el;
2243
2244                 msg_Warn( p_demux, "searching for cluster, could take some time" );
2245
2246                 /* search a cluster */
2247                 while( ( el = p_sys->ep->Get() ) != NULL )
2248                 {
2249                     if( MKV_IS_ID( el, KaxCluster ) )
2250                     {
2251                         KaxCluster *cluster = (KaxCluster*)el;
2252
2253                         /* add it to the index */
2254                         p_segment->IndexAppendCluster( cluster );
2255
2256                         if( (int64_t)cluster->GetElementPosition() >= i_pos )
2257                         {
2258                             p_sys->cluster = cluster;
2259                             p_sys->ep->Down();
2260                             break;
2261                         }
2262                     }
2263                 }
2264             }
2265 #endif
2266         }
2267     }
2268
2269     p_vsegment->Seek( *p_demux, i_date, i_time_offset, psz_chapter );
2270 }
2271
2272 /*****************************************************************************
2273  * Demux: reads and demuxes data packets
2274  *****************************************************************************
2275  * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
2276  *****************************************************************************/
2277 static int Demux( demux_t *p_demux)
2278 {
2279     demux_sys_t        *p_sys = p_demux->p_sys;
2280     virtual_segment_c  *p_vsegment = p_sys->p_current_segment;
2281     matroska_segment_c *p_segmet = p_vsegment->Segment();
2282     if ( p_segmet == NULL ) return 0;
2283     int                i_block_count = 0;
2284
2285     KaxBlock *block;
2286     int64_t i_block_duration;
2287     int64_t i_block_ref1;
2288     int64_t i_block_ref2;
2289
2290     for( ;; )
2291     {
2292         if ( p_sys->demuxer.b_die )
2293             return 0;
2294
2295         if( p_sys->i_pts >= p_sys->i_start_pts  )
2296             if ( p_vsegment->UpdateCurrentToChapter( *p_demux ) )
2297                 return 1;
2298
2299         if ( p_vsegment->Edition() && p_vsegment->Edition()->b_ordered && p_vsegment->CurrentChapter() == NULL )
2300         {
2301             /* nothing left to read in this ordered edition */
2302             if ( !p_vsegment->SelectNext() )
2303                 return 0;
2304             p_segmet->UnSelect( );
2305
2306             es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
2307
2308             /* switch to the next segment */
2309             p_segmet = p_vsegment->Segment();
2310             if ( !p_segmet->Select( 0 ) )
2311             {
2312                 msg_Err( p_demux, "Failed to select new segment" );
2313                 return 0;
2314             }
2315             continue;
2316         }
2317
2318
2319         if( p_segmet->BlockGet( &block, &i_block_ref1, &i_block_ref2, &i_block_duration ) )
2320         {
2321             if ( p_vsegment->Edition() && p_vsegment->Edition()->b_ordered )
2322             {
2323                 const chapter_item_c *p_chap = p_vsegment->CurrentChapter();
2324                 // check if there are more chapters to read
2325                 if ( p_chap != NULL )
2326                 {
2327                     /* TODO handle successive chapters with the same user_start_time/user_end_time
2328                     if ( p_chap->i_user_start_time == p_chap->i_user_start_time )
2329                         p_vsegment->SelectNext();
2330                     */
2331                     p_sys->i_pts = p_chap->i_user_end_time;
2332                     p_sys->i_pts++; // trick to avoid staying on segments with no duration and no content
2333
2334                     return 1;
2335                 }
2336
2337                 return 0;
2338             }
2339             msg_Warn( p_demux, "cannot get block EOF?" );
2340             p_segmet->UnSelect( );
2341
2342             es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
2343
2344             /* switch to the next segment */
2345             if ( !p_vsegment->SelectNext() )
2346                 // no more segments in this stream
2347                 return 0;
2348             p_segmet = p_vsegment->Segment();
2349             if ( !p_segmet->Select( 0 ) )
2350             {
2351                 msg_Err( p_demux, "Failed to select new segment" );
2352                 return 0;
2353             }
2354
2355             continue;
2356         }
2357
2358         p_sys->i_pts = p_sys->i_chapter_time + block->GlobalTimecode() / (mtime_t) 1000;
2359
2360         if( p_sys->i_pts >= p_sys->i_start_pts  )
2361         {
2362             es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_pts );
2363         }
2364
2365         BlockDecode( p_demux, block, p_sys->i_pts, i_block_duration );
2366
2367         delete block;
2368         i_block_count++;
2369
2370         // TODO optimize when there is need to leave or when seeking has been called
2371         if( i_block_count > 5 )
2372         {
2373             return 1;
2374         }
2375     }
2376 }
2377
2378
2379
2380 /*****************************************************************************
2381  * Stream managment
2382  *****************************************************************************/
2383 vlc_stream_io_callback::vlc_stream_io_callback( stream_t *s_ )
2384 {
2385     s = s_;
2386     mb_eof = VLC_FALSE;
2387 }
2388
2389 uint32 vlc_stream_io_callback::read( void *p_buffer, size_t i_size )
2390 {
2391     if( i_size <= 0 || mb_eof )
2392     {
2393         return 0;
2394     }
2395
2396     return stream_Read( s, p_buffer, i_size );
2397 }
2398 void vlc_stream_io_callback::setFilePointer(int64_t i_offset, seek_mode mode )
2399 {
2400     int64_t i_pos;
2401
2402     switch( mode )
2403     {
2404         case seek_beginning:
2405             i_pos = i_offset;
2406             break;
2407         case seek_end:
2408             i_pos = stream_Size( s ) - i_offset;
2409             break;
2410         default:
2411             i_pos= stream_Tell( s ) + i_offset;
2412             break;
2413     }
2414
2415     if( i_pos < 0 || i_pos >= stream_Size( s ) )
2416     {
2417         mb_eof = VLC_TRUE;
2418         return;
2419     }
2420
2421     mb_eof = VLC_FALSE;
2422     if( stream_Seek( s, i_pos ) )
2423     {
2424         mb_eof = VLC_TRUE;
2425     }
2426     return;
2427 }
2428 size_t vlc_stream_io_callback::write( const void *p_buffer, size_t i_size )
2429 {
2430     return 0;
2431 }
2432 uint64 vlc_stream_io_callback::getFilePointer( void )
2433 {
2434     return stream_Tell( s );
2435 }
2436 void vlc_stream_io_callback::close( void )
2437 {
2438     return;
2439 }
2440
2441
2442 /*****************************************************************************
2443  * Ebml Stream parser
2444  *****************************************************************************/
2445 EbmlParser::EbmlParser( EbmlStream *es, EbmlElement *el_start, demux_t *p_demux )
2446 {
2447     int i;
2448
2449     m_es = es;
2450     m_got = NULL;
2451     m_el[0] = el_start;
2452     mi_remain_size[0] = el_start->GetSize();
2453
2454     for( i = 1; i < 6; i++ )
2455     {
2456         m_el[i] = NULL;
2457     }
2458     mi_level = 1;
2459     mi_user_level = 1;
2460     mb_keep = VLC_FALSE;
2461     mb_dummy = config_GetInt( p_demux, "mkv-use-dummy" );
2462 }
2463
2464 EbmlParser::~EbmlParser( void )
2465 {
2466     int i;
2467
2468     for( i = 1; i < mi_level; i++ )
2469     {
2470         if( !mb_keep )
2471         {
2472             delete m_el[i];
2473         }
2474         mb_keep = VLC_FALSE;
2475     }
2476 }
2477
2478 void EbmlParser::Up( void )
2479 {
2480     if( mi_user_level == mi_level )
2481     {
2482         fprintf( stderr," arrrrrrrrrrrrrg Up cannot escape itself\n" );
2483     }
2484
2485     mi_user_level--;
2486 }
2487
2488 void EbmlParser::Down( void )
2489 {
2490     mi_user_level++;
2491     mi_level++;
2492 }
2493
2494 void EbmlParser::Keep( void )
2495 {
2496     mb_keep = VLC_TRUE;
2497 }
2498
2499 int EbmlParser::GetLevel( void )
2500 {
2501     return mi_user_level;
2502 }
2503
2504 void EbmlParser::Reset( demux_t *p_demux )
2505 {
2506     while ( mi_level > 0)
2507     {
2508         delete m_el[mi_level];
2509         m_el[mi_level] = NULL;
2510         mi_level--;
2511     }
2512     mi_user_level = mi_level = 1;
2513 #if LIBEBML_VERSION >= 0x000704
2514     // a little faster and cleaner
2515     m_es->I_O().setFilePointer( static_cast<KaxSegment*>(m_el[0])->GetGlobalPosition(0) );
2516 #else
2517     m_es->I_O().setFilePointer( m_el[0]->GetElementPosition() + m_el[0]->ElementSize(true) - m_el[0]->GetSize() );
2518 #endif
2519     mb_dummy = config_GetInt( p_demux, "mkv-use-dummy" );
2520 }
2521
2522 EbmlElement *EbmlParser::Get( void )
2523 {
2524     int i_ulev = 0;
2525
2526     if( mi_user_level != mi_level )
2527     {
2528         return NULL;
2529     }
2530     if( m_got )
2531     {
2532         EbmlElement *ret = m_got;
2533         m_got = NULL;
2534
2535         return ret;
2536     }
2537
2538     if( m_el[mi_level] )
2539     {
2540         m_el[mi_level]->SkipData( *m_es, m_el[mi_level]->Generic().Context );
2541         if( !mb_keep )
2542         {
2543             delete m_el[mi_level];
2544         }
2545         mb_keep = VLC_FALSE;
2546     }
2547
2548     m_el[mi_level] = m_es->FindNextElement( m_el[mi_level - 1]->Generic().Context, i_ulev, 0xFFFFFFFFL, mb_dummy, 1 );
2549 //    mi_remain_size[mi_level] = m_el[mi_level]->GetSize();
2550     if( i_ulev > 0 )
2551     {
2552         while( i_ulev > 0 )
2553         {
2554             if( mi_level == 1 )
2555             {
2556                 mi_level = 0;
2557                 return NULL;
2558             }
2559
2560             delete m_el[mi_level - 1];
2561             m_got = m_el[mi_level -1] = m_el[mi_level];
2562             m_el[mi_level] = NULL;
2563
2564             mi_level--;
2565             i_ulev--;
2566         }
2567         return NULL;
2568     }
2569     else if( m_el[mi_level] == NULL )
2570     {
2571         fprintf( stderr," m_el[mi_level] == NULL\n" );
2572     }
2573
2574     return m_el[mi_level];
2575 }
2576
2577
2578 /*****************************************************************************
2579  * Tools
2580  *  * LoadCues : load the cues element and update index
2581  *
2582  *  * LoadTags : load ... the tags element
2583  *
2584  *  * InformationCreate : create all information, load tags if present
2585  *
2586  *****************************************************************************/
2587 void matroska_segment_c::LoadCues( )
2588 {
2589     int64_t     i_sav_position = es.I_O().getFilePointer();
2590     EbmlParser  *ep;
2591     EbmlElement *el, *cues;
2592
2593     /* *** Load the cue if found *** */
2594     if( i_cues_position < 0 )
2595     {
2596         msg_Warn( &sys.demuxer, "no cues/empty cues found->seek won't be precise" );
2597
2598 //        IndexAppendCluster( cluster );
2599     }
2600
2601     vlc_bool_t b_seekable;
2602
2603     stream_Control( sys.demuxer.s, STREAM_CAN_FASTSEEK, &b_seekable );
2604     if( !b_seekable )
2605         return;
2606
2607     msg_Dbg( &sys.demuxer, "loading cues" );
2608     es.I_O().setFilePointer( i_cues_position, seek_beginning );
2609     cues = es.FindNextID( KaxCues::ClassInfos, 0xFFFFFFFFL);
2610
2611     if( cues == NULL )
2612     {
2613         msg_Err( &sys.demuxer, "cannot load cues (broken seekhead or file)" );
2614         es.I_O().setFilePointer( i_sav_position, seek_beginning );
2615         return;
2616     }
2617
2618     ep = new EbmlParser( &es, cues, &sys.demuxer );
2619     while( ( el = ep->Get() ) != NULL )
2620     {
2621         if( MKV_IS_ID( el, KaxCuePoint ) )
2622         {
2623 #define idx index[i_index]
2624
2625             idx.i_track       = -1;
2626             idx.i_block_number= -1;
2627             idx.i_position    = -1;
2628             idx.i_time        = 0;
2629             idx.b_key         = VLC_TRUE;
2630
2631             ep->Down();
2632             while( ( el = ep->Get() ) != NULL )
2633             {
2634                 if( MKV_IS_ID( el, KaxCueTime ) )
2635                 {
2636                     KaxCueTime &ctime = *(KaxCueTime*)el;
2637
2638                     ctime.ReadData( es.I_O() );
2639
2640                     idx.i_time = uint64( ctime ) * i_timescale / (mtime_t)1000;
2641                 }
2642                 else if( MKV_IS_ID( el, KaxCueTrackPositions ) )
2643                 {
2644                     ep->Down();
2645                     while( ( el = ep->Get() ) != NULL )
2646                     {
2647                         if( MKV_IS_ID( el, KaxCueTrack ) )
2648                         {
2649                             KaxCueTrack &ctrack = *(KaxCueTrack*)el;
2650
2651                             ctrack.ReadData( es.I_O() );
2652                             idx.i_track = uint16( ctrack );
2653                         }
2654                         else if( MKV_IS_ID( el, KaxCueClusterPosition ) )
2655                         {
2656                             KaxCueClusterPosition &ccpos = *(KaxCueClusterPosition*)el;
2657
2658                             ccpos.ReadData( es.I_O() );
2659                             idx.i_position = segment->GetGlobalPosition( uint64( ccpos ) );
2660                         }
2661                         else if( MKV_IS_ID( el, KaxCueBlockNumber ) )
2662                         {
2663                             KaxCueBlockNumber &cbnum = *(KaxCueBlockNumber*)el;
2664
2665                             cbnum.ReadData( es.I_O() );
2666                             idx.i_block_number = uint32( cbnum );
2667                         }
2668                         else
2669                         {
2670                             msg_Dbg( &sys.demuxer, "         * Unknown (%s)", typeid(*el).name() );
2671                         }
2672                     }
2673                     ep->Up();
2674                 }
2675                 else
2676                 {
2677                     msg_Dbg( &sys.demuxer, "     * Unknown (%s)", typeid(*el).name() );
2678                 }
2679             }
2680             ep->Up();
2681
2682 #if 0
2683             msg_Dbg( &sys.demuxer, " * added time="I64Fd" pos="I64Fd
2684                      " track=%d bnum=%d", idx.i_time, idx.i_position,
2685                      idx.i_track, idx.i_block_number );
2686 #endif
2687
2688             i_index++;
2689             if( i_index >= i_index_max )
2690             {
2691                 i_index_max += 1024;
2692                 index = (mkv_index_t*)realloc( index, sizeof( mkv_index_t ) * i_index_max );
2693             }
2694 #undef idx
2695         }
2696         else
2697         {
2698             msg_Dbg( &sys.demuxer, " * Unknown (%s)", typeid(*el).name() );
2699         }
2700     }
2701     delete ep;
2702     delete cues;
2703
2704     b_cues = VLC_TRUE;
2705
2706     msg_Dbg( &sys.demuxer, "loading cues done." );
2707     es.I_O().setFilePointer( i_sav_position, seek_beginning );
2708 }
2709
2710 void matroska_segment_c::LoadTags( )
2711 {
2712     int64_t     i_sav_position = es.I_O().getFilePointer();
2713     EbmlParser  *ep;
2714     EbmlElement *el, *tags;
2715
2716     msg_Dbg( &sys.demuxer, "loading tags" );
2717     es.I_O().setFilePointer( i_tags_position, seek_beginning );
2718     tags = es.FindNextID( KaxTags::ClassInfos, 0xFFFFFFFFL);
2719
2720     if( tags == NULL )
2721     {
2722         msg_Err( &sys.demuxer, "cannot load tags (broken seekhead or file)" );
2723         es.I_O().setFilePointer( i_sav_position, seek_beginning );
2724         return;
2725     }
2726
2727     msg_Dbg( &sys.demuxer, "Tags" );
2728     ep = new EbmlParser( &es, tags, &sys.demuxer );
2729     while( ( el = ep->Get() ) != NULL )
2730     {
2731         if( MKV_IS_ID( el, KaxTag ) )
2732         {
2733             msg_Dbg( &sys.demuxer, "+ Tag" );
2734             ep->Down();
2735             while( ( el = ep->Get() ) != NULL )
2736             {
2737                 if( MKV_IS_ID( el, KaxTagTargets ) )
2738                 {
2739                     msg_Dbg( &sys.demuxer, "|   + Targets" );
2740                     ep->Down();
2741                     while( ( el = ep->Get() ) != NULL )
2742                     {
2743                         msg_Dbg( &sys.demuxer, "|   |   + Unknown (%s)", typeid( *el ).name() );
2744                     }
2745                     ep->Up();
2746                 }
2747                 else if( MKV_IS_ID( el, KaxTagGeneral ) )
2748                 {
2749                     msg_Dbg( &sys.demuxer, "|   + General" );
2750                     ep->Down();
2751                     while( ( el = ep->Get() ) != NULL )
2752                     {
2753                         msg_Dbg( &sys.demuxer, "|   |   + Unknown (%s)", typeid( *el ).name() );
2754                     }
2755                     ep->Up();
2756                 }
2757                 else if( MKV_IS_ID( el, KaxTagGenres ) )
2758                 {
2759                     msg_Dbg( &sys.demuxer, "|   + Genres" );
2760                     ep->Down();
2761                     while( ( el = ep->Get() ) != NULL )
2762                     {
2763                         msg_Dbg( &sys.demuxer, "|   |   + Unknown (%s)", typeid( *el ).name() );
2764                     }
2765                     ep->Up();
2766                 }
2767                 else if( MKV_IS_ID( el, KaxTagAudioSpecific ) )
2768                 {
2769                     msg_Dbg( &sys.demuxer, "|   + Audio Specific" );
2770                     ep->Down();
2771                     while( ( el = ep->Get() ) != NULL )
2772                     {
2773                         msg_Dbg( &sys.demuxer, "|   |   + Unknown (%s)", typeid( *el ).name() );
2774                     }
2775                     ep->Up();
2776                 }
2777                 else if( MKV_IS_ID( el, KaxTagImageSpecific ) )
2778                 {
2779                     msg_Dbg( &sys.demuxer, "|   + Images Specific" );
2780                     ep->Down();
2781                     while( ( el = ep->Get() ) != NULL )
2782                     {
2783                         msg_Dbg( &sys.demuxer, "|   |   + Unknown (%s)", typeid( *el ).name() );
2784                     }
2785                     ep->Up();
2786                 }
2787                 else if( MKV_IS_ID( el, KaxTagMultiComment ) )
2788                 {
2789                     msg_Dbg( &sys.demuxer, "|   + Multi Comment" );
2790                 }
2791                 else if( MKV_IS_ID( el, KaxTagMultiCommercial ) )
2792                 {
2793                     msg_Dbg( &sys.demuxer, "|   + Multi Commercial" );
2794                 }
2795                 else if( MKV_IS_ID( el, KaxTagMultiDate ) )
2796                 {
2797                     msg_Dbg( &sys.demuxer, "|   + Multi Date" );
2798                 }
2799                 else if( MKV_IS_ID( el, KaxTagMultiEntity ) )
2800                 {
2801                     msg_Dbg( &sys.demuxer, "|   + Multi Entity" );
2802                 }
2803                 else if( MKV_IS_ID( el, KaxTagMultiIdentifier ) )
2804                 {
2805                     msg_Dbg( &sys.demuxer, "|   + Multi Identifier" );
2806                 }
2807                 else if( MKV_IS_ID( el, KaxTagMultiLegal ) )
2808                 {
2809                     msg_Dbg( &sys.demuxer, "|   + Multi Legal" );
2810                 }
2811                 else if( MKV_IS_ID( el, KaxTagMultiTitle ) )
2812                 {
2813                     msg_Dbg( &sys.demuxer, "|   + Multi Title" );
2814                 }
2815                 else
2816                 {
2817                     msg_Dbg( &sys.demuxer, "|   + Unknown (%s)", typeid( *el ).name() );
2818                 }
2819             }
2820             ep->Up();
2821         }
2822         else
2823         {
2824             msg_Dbg( &sys.demuxer, "+ Unknown (%s)", typeid( *el ).name() );
2825         }
2826     }
2827     delete ep;
2828     delete tags;
2829
2830     msg_Dbg( &sys.demuxer, "loading tags done." );
2831     es.I_O().setFilePointer( i_sav_position, seek_beginning );
2832 }
2833
2834 /*****************************************************************************
2835  * ParseSeekHead:
2836  *****************************************************************************/
2837 void matroska_segment_c::ParseSeekHead( KaxSeekHead *seekhead )
2838 {
2839     EbmlElement *el;
2840     size_t i, j;
2841     int i_upper_level = 0;
2842
2843     msg_Dbg( &sys.demuxer, "|   + Seek head" );
2844
2845     /* Master elements */
2846     seekhead->Read( es, seekhead->Generic().Context, i_upper_level, el, true );
2847
2848     for( i = 0; i < seekhead->ListSize(); i++ )
2849     {
2850         EbmlElement *l = (*seekhead)[i];
2851
2852         if( MKV_IS_ID( l, KaxSeek ) )
2853         {
2854             EbmlMaster *sk = static_cast<EbmlMaster *>(l);
2855             EbmlId id = EbmlVoid::ClassInfos.GlobalId;
2856             int64_t i_pos = -1;
2857
2858             for( j = 0; j < sk->ListSize(); j++ )
2859             {
2860                 EbmlElement *l = (*sk)[j];
2861
2862                 if( MKV_IS_ID( l, KaxSeekID ) )
2863                 {
2864                     KaxSeekID &sid = *(KaxSeekID*)l;
2865                     id = EbmlId( sid.GetBuffer(), sid.GetSize() );
2866                 }
2867                 else if( MKV_IS_ID( l, KaxSeekPosition ) )
2868                 {
2869                     KaxSeekPosition &spos = *(KaxSeekPosition*)l;
2870                     i_pos = uint64( spos );
2871                 }
2872                 else
2873                 {
2874                     msg_Dbg( &sys.demuxer, "|   |   |   + Unknown (%s)", typeid(*l).name() );
2875                 }
2876             }
2877
2878             if( i_pos >= 0 )
2879             {
2880                 if( id == KaxCues::ClassInfos.GlobalId )
2881                 {
2882                     msg_Dbg( &sys.demuxer, "|   |   |   = cues at "I64Fd, i_pos );
2883                     i_cues_position = segment->GetGlobalPosition( i_pos );
2884                 }
2885                 else if( id == KaxChapters::ClassInfos.GlobalId )
2886                 {
2887                     msg_Dbg( &sys.demuxer, "|   |   |   = chapters at "I64Fd, i_pos );
2888                     i_chapters_position = segment->GetGlobalPosition( i_pos );
2889                 }
2890                 else if( id == KaxTags::ClassInfos.GlobalId )
2891                 {
2892                     msg_Dbg( &sys.demuxer, "|   |   |   = tags at "I64Fd, i_pos );
2893                     i_tags_position = segment->GetGlobalPosition( i_pos );
2894                 }
2895             }
2896         }
2897         else
2898         {
2899             msg_Dbg( &sys.demuxer, "|   |   + Unknown (%s)", typeid(*l).name() );
2900         }
2901     }
2902 }
2903
2904 /*****************************************************************************
2905  * ParseTrackEntry:
2906  *****************************************************************************/
2907 void matroska_segment_c::ParseTrackEntry( KaxTrackEntry *m )
2908 {
2909     size_t i, j, k, n;
2910
2911     mkv_track_t *tk;
2912
2913     msg_Dbg( &sys.demuxer, "|   |   + Track Entry" );
2914
2915     tk = new mkv_track_t();
2916     tracks.push_back( tk );
2917
2918     /* Init the track */
2919     memset( tk, 0, sizeof( mkv_track_t ) );
2920
2921     es_format_Init( &tk->fmt, UNKNOWN_ES, 0 );
2922     tk->fmt.psz_language = strdup("English");
2923     tk->fmt.psz_description = NULL;
2924
2925     tk->b_default = VLC_TRUE;
2926     tk->b_enabled = VLC_TRUE;
2927     tk->b_silent = VLC_FALSE;
2928     tk->i_number = tracks.size() - 1;
2929     tk->i_extra_data = 0;
2930     tk->p_extra_data = NULL;
2931     tk->psz_codec = NULL;
2932     tk->i_default_duration = 0;
2933     tk->f_timecodescale = 1.0;
2934
2935     tk->b_inited = VLC_FALSE;
2936     tk->i_data_init = 0;
2937     tk->p_data_init = NULL;
2938
2939     tk->psz_codec_name = NULL;
2940     tk->psz_codec_settings = NULL;
2941     tk->psz_codec_info_url = NULL;
2942     tk->psz_codec_download_url = NULL;
2943
2944     tk->i_compression_type = MATROSKA_COMPRESSION_NONE;
2945
2946     for( i = 0; i < m->ListSize(); i++ )
2947     {
2948         EbmlElement *l = (*m)[i];
2949
2950         if( MKV_IS_ID( l, KaxTrackNumber ) )
2951         {
2952             KaxTrackNumber &tnum = *(KaxTrackNumber*)l;
2953
2954             tk->i_number = uint32( tnum );
2955             msg_Dbg( &sys.demuxer, "|   |   |   + Track Number=%u", uint32( tnum ) );
2956         }
2957         else  if( MKV_IS_ID( l, KaxTrackUID ) )
2958         {
2959             KaxTrackUID &tuid = *(KaxTrackUID*)l;
2960
2961             msg_Dbg( &sys.demuxer, "|   |   |   + Track UID=%u",  uint32( tuid ) );
2962         }
2963         else  if( MKV_IS_ID( l, KaxTrackType ) )
2964         {
2965             char *psz_type;
2966             KaxTrackType &ttype = *(KaxTrackType*)l;
2967
2968             switch( uint8(ttype) )
2969             {
2970                 case track_audio:
2971                     psz_type = "audio";
2972                     tk->fmt.i_cat = AUDIO_ES;
2973                     break;
2974                 case track_video:
2975                     psz_type = "video";
2976                     tk->fmt.i_cat = VIDEO_ES;
2977                     break;
2978                 case track_subtitle:
2979                     psz_type = "subtitle";
2980                     tk->fmt.i_cat = SPU_ES;
2981                     break;
2982                 default:
2983                     psz_type = "unknown";
2984                     tk->fmt.i_cat = UNKNOWN_ES;
2985                     break;
2986             }
2987
2988             msg_Dbg( &sys.demuxer, "|   |   |   + Track Type=%s", psz_type );
2989         }
2990 //        else  if( EbmlId( *l ) == KaxTrackFlagEnabled::ClassInfos.GlobalId )
2991 //        {
2992 //            KaxTrackFlagEnabled &fenb = *(KaxTrackFlagEnabled*)l;
2993
2994 //            tk->b_enabled = uint32( fenb );
2995 //            msg_Dbg( &sys.demuxer, "|   |   |   + Track Enabled=%u",
2996 //                     uint32( fenb )  );
2997 //        }
2998         else  if( MKV_IS_ID( l, KaxTrackFlagDefault ) )
2999         {
3000             KaxTrackFlagDefault &fdef = *(KaxTrackFlagDefault*)l;
3001
3002             tk->b_default = uint32( fdef );
3003             msg_Dbg( &sys.demuxer, "|   |   |   + Track Default=%u", uint32( fdef )  );
3004         }
3005         else  if( MKV_IS_ID( l, KaxTrackFlagLacing ) )
3006         {
3007             KaxTrackFlagLacing &lac = *(KaxTrackFlagLacing*)l;
3008
3009             msg_Dbg( &sys.demuxer, "|   |   |   + Track Lacing=%d", uint32( lac ) );
3010         }
3011         else  if( MKV_IS_ID( l, KaxTrackMinCache ) )
3012         {
3013             KaxTrackMinCache &cmin = *(KaxTrackMinCache*)l;
3014
3015             msg_Dbg( &sys.demuxer, "|   |   |   + Track MinCache=%d", uint32( cmin ) );
3016         }
3017         else  if( MKV_IS_ID( l, KaxTrackMaxCache ) )
3018         {
3019             KaxTrackMaxCache &cmax = *(KaxTrackMaxCache*)l;
3020
3021             msg_Dbg( &sys.demuxer, "|   |   |   + Track MaxCache=%d", uint32( cmax ) );
3022         }
3023         else  if( MKV_IS_ID( l, KaxTrackDefaultDuration ) )
3024         {
3025             KaxTrackDefaultDuration &defd = *(KaxTrackDefaultDuration*)l;
3026
3027             tk->i_default_duration = uint64(defd);
3028             msg_Dbg( &sys.demuxer, "|   |   |   + Track Default Duration="I64Fd, uint64(defd) );
3029         }
3030         else  if( MKV_IS_ID( l, KaxTrackTimecodeScale ) )
3031         {
3032             KaxTrackTimecodeScale &ttcs = *(KaxTrackTimecodeScale*)l;
3033
3034             tk->f_timecodescale = float( ttcs );
3035             msg_Dbg( &sys.demuxer, "|   |   |   + Track TimeCodeScale=%f", tk->f_timecodescale );
3036         }
3037         else if( MKV_IS_ID( l, KaxTrackName ) )
3038         {
3039             KaxTrackName &tname = *(KaxTrackName*)l;
3040
3041             tk->fmt.psz_description = UTF8ToStr( UTFstring( tname ) );
3042             msg_Dbg( &sys.demuxer, "|   |   |   + Track Name=%s", tk->fmt.psz_description );
3043         }
3044         else  if( MKV_IS_ID( l, KaxTrackLanguage ) )
3045         {
3046             KaxTrackLanguage &lang = *(KaxTrackLanguage*)l;
3047
3048             tk->fmt.psz_language = strdup( string( lang ).c_str() );
3049             msg_Dbg( &sys.demuxer,
3050                      "|   |   |   + Track Language=`%s'", tk->fmt.psz_language );
3051         }
3052         else  if( MKV_IS_ID( l, KaxCodecID ) )
3053         {
3054             KaxCodecID &codecid = *(KaxCodecID*)l;
3055
3056             tk->psz_codec = strdup( string( codecid ).c_str() );
3057             msg_Dbg( &sys.demuxer, "|   |   |   + Track CodecId=%s", string( codecid ).c_str() );
3058         }
3059         else  if( MKV_IS_ID( l, KaxCodecPrivate ) )
3060         {
3061             KaxCodecPrivate &cpriv = *(KaxCodecPrivate*)l;
3062
3063             tk->i_extra_data = cpriv.GetSize();
3064             if( tk->i_extra_data > 0 )
3065             {
3066                 tk->p_extra_data = (uint8_t*)malloc( tk->i_extra_data );
3067                 memcpy( tk->p_extra_data, cpriv.GetBuffer(), tk->i_extra_data );
3068             }
3069             msg_Dbg( &sys.demuxer, "|   |   |   + Track CodecPrivate size="I64Fd, cpriv.GetSize() );
3070         }
3071         else if( MKV_IS_ID( l, KaxCodecName ) )
3072         {
3073             KaxCodecName &cname = *(KaxCodecName*)l;
3074
3075             tk->psz_codec_name = UTF8ToStr( UTFstring( cname ) );
3076             msg_Dbg( &sys.demuxer, "|   |   |   + Track Codec Name=%s", tk->psz_codec_name );
3077         }
3078         else if( MKV_IS_ID( l, KaxContentEncodings ) )
3079         {
3080             EbmlMaster *cencs = static_cast<EbmlMaster*>(l);
3081             MkvTree( sys.demuxer, 3, "Content Encodings" );
3082             for( j = 0; j < cencs->ListSize(); j++ )
3083             {
3084                 EbmlElement *l2 = (*cencs)[j];
3085                 if( MKV_IS_ID( l2, KaxContentEncoding ) )
3086                 {
3087                     MkvTree( sys.demuxer, 4, "Content Encoding" );
3088                     EbmlMaster *cenc = static_cast<EbmlMaster*>(l2);
3089                     for( k = 0; k < cenc->ListSize(); k++ )
3090                     {
3091                         EbmlElement *l3 = (*cenc)[k];
3092                         if( MKV_IS_ID( l3, KaxContentEncodingOrder ) )
3093                         {
3094                             KaxContentEncodingOrder &encord = *(KaxContentEncodingOrder*)l3;
3095                             MkvTree( sys.demuxer, 5, "Order: %i", uint32( encord ) );
3096                         }
3097                         else if( MKV_IS_ID( l3, KaxContentEncodingScope ) )
3098                         {
3099                             KaxContentEncodingScope &encscope = *(KaxContentEncodingScope*)l3;
3100                             MkvTree( sys.demuxer, 5, "Scope: %i", uint32( encscope ) );
3101                         }
3102                         else if( MKV_IS_ID( l3, KaxContentEncodingType ) )
3103                         {
3104                             KaxContentEncodingType &enctype = *(KaxContentEncodingType*)l3;
3105                             MkvTree( sys.demuxer, 5, "Type: %i", uint32( enctype ) );
3106                         }
3107                         else if( MKV_IS_ID( l3, KaxContentCompression ) )
3108                         {
3109                             EbmlMaster *compr = static_cast<EbmlMaster*>(l3);
3110                             MkvTree( sys.demuxer, 5, "Content Compression" );
3111                             for( n = 0; n < compr->ListSize(); n++ )
3112                             {
3113                                 EbmlElement *l4 = (*compr)[n];
3114                                 if( MKV_IS_ID( l4, KaxContentCompAlgo ) )
3115                                 {
3116                                     KaxContentCompAlgo &compalg = *(KaxContentCompAlgo*)l4;
3117                                     MkvTree( sys.demuxer, 6, "Compression Algorithm: %i", uint32(compalg) );
3118                                     if( uint32( compalg ) == 0 )
3119                                     {
3120                                         tk->i_compression_type = MATROSKA_COMPRESSION_ZLIB;
3121                                     }
3122                                 }
3123                                 else
3124                                 {
3125                                     MkvTree( sys.demuxer, 6, "Unknown (%s)", typeid(*l4).name() );
3126                                 }
3127                             }
3128                         }
3129
3130                         else
3131                         {
3132                             MkvTree( sys.demuxer, 5, "Unknown (%s)", typeid(*l3).name() );
3133                         }
3134                     }
3135
3136                 }
3137                 else
3138                 {
3139                     MkvTree( sys.demuxer, 4, "Unknown (%s)", typeid(*l2).name() );
3140                 }
3141             }
3142
3143         }
3144 //        else if( EbmlId( *l ) == KaxCodecSettings::ClassInfos.GlobalId )
3145 //        {
3146 //            KaxCodecSettings &cset = *(KaxCodecSettings*)l;
3147
3148 //            tk->psz_codec_settings = UTF8ToStr( UTFstring( cset ) );
3149 //            msg_Dbg( &sys.demuxer, "|   |   |   + Track Codec Settings=%s", tk->psz_codec_settings );
3150 //        }
3151 //        else if( EbmlId( *l ) == KaxCodecInfoURL::ClassInfos.GlobalId )
3152 //        {
3153 //            KaxCodecInfoURL &ciurl = *(KaxCodecInfoURL*)l;
3154
3155 //            tk->psz_codec_info_url = strdup( string( ciurl ).c_str() );
3156 //            msg_Dbg( &sys.demuxer, "|   |   |   + Track Codec Info URL=%s", tk->psz_codec_info_url );
3157 //        }
3158 //        else if( EbmlId( *l ) == KaxCodecDownloadURL::ClassInfos.GlobalId )
3159 //        {
3160 //            KaxCodecDownloadURL &cdurl = *(KaxCodecDownloadURL*)l;
3161
3162 //            tk->psz_codec_download_url = strdup( string( cdurl ).c_str() );
3163 //            msg_Dbg( &sys.demuxer, "|   |   |   + Track Codec Info URL=%s", tk->psz_codec_download_url );
3164 //        }
3165 //        else if( EbmlId( *l ) == KaxCodecDecodeAll::ClassInfos.GlobalId )
3166 //        {
3167 //            KaxCodecDecodeAll &cdall = *(KaxCodecDecodeAll*)l;
3168
3169 //            msg_Dbg( &sys.demuxer, "|   |   |   + Track Codec Decode All=%u <== UNUSED", uint8( cdall ) );
3170 //        }
3171 //        else if( EbmlId( *l ) == KaxTrackOverlay::ClassInfos.GlobalId )
3172 //        {
3173 //            KaxTrackOverlay &tovr = *(KaxTrackOverlay*)l;
3174
3175 //            msg_Dbg( &sys.demuxer, "|   |   |   + Track Overlay=%u <== UNUSED", uint32( tovr ) );
3176 //        }
3177         else  if( MKV_IS_ID( l, KaxTrackVideo ) )
3178         {
3179             EbmlMaster *tkv = static_cast<EbmlMaster*>(l);
3180             unsigned int j;
3181
3182             msg_Dbg( &sys.demuxer, "|   |   |   + Track Video" );
3183             tk->f_fps = 0.0;
3184
3185             for( j = 0; j < tkv->ListSize(); j++ )
3186             {
3187                 EbmlElement *l = (*tkv)[j];
3188 //                if( EbmlId( *el4 ) == KaxVideoFlagInterlaced::ClassInfos.GlobalId )
3189 //                {
3190 //                    KaxVideoFlagInterlaced &fint = *(KaxVideoFlagInterlaced*)el4;
3191
3192 //                    msg_Dbg( &sys.demuxer, "|   |   |   |   + Track Video Interlaced=%u", uint8( fint ) );
3193 //                }
3194 //                else if( EbmlId( *el4 ) == KaxVideoStereoMode::ClassInfos.GlobalId )
3195 //                {
3196 //                    KaxVideoStereoMode &stereo = *(KaxVideoStereoMode*)el4;
3197
3198 //                    msg_Dbg( &sys.demuxer, "|   |   |   |   + Track Video Stereo Mode=%u", uint8( stereo ) );
3199 //                }
3200 //                else
3201                 if( MKV_IS_ID( l, KaxVideoPixelWidth ) )
3202                 {
3203                     KaxVideoPixelWidth &vwidth = *(KaxVideoPixelWidth*)l;
3204
3205                     tk->fmt.video.i_width = uint16( vwidth );
3206                     msg_Dbg( &sys.demuxer, "|   |   |   |   + width=%d", uint16( vwidth ) );
3207                 }
3208                 else if( MKV_IS_ID( l, KaxVideoPixelHeight ) )
3209                 {
3210                     KaxVideoPixelWidth &vheight = *(KaxVideoPixelWidth*)l;
3211
3212                     tk->fmt.video.i_height = uint16( vheight );
3213                     msg_Dbg( &sys.demuxer, "|   |   |   |   + height=%d", uint16( vheight ) );
3214                 }
3215                 else if( MKV_IS_ID( l, KaxVideoDisplayWidth ) )
3216                 {
3217                     KaxVideoDisplayWidth &vwidth = *(KaxVideoDisplayWidth*)l;
3218
3219                     tk->fmt.video.i_visible_width = uint16( vwidth );
3220                     msg_Dbg( &sys.demuxer, "|   |   |   |   + display width=%d", uint16( vwidth ) );
3221                 }
3222                 else if( MKV_IS_ID( l, KaxVideoDisplayHeight ) )
3223                 {
3224                     KaxVideoDisplayWidth &vheight = *(KaxVideoDisplayWidth*)l;
3225
3226                     tk->fmt.video.i_visible_height = uint16( vheight );
3227                     msg_Dbg( &sys.demuxer, "|   |   |   |   + display height=%d", uint16( vheight ) );
3228                 }
3229                 else if( MKV_IS_ID( l, KaxVideoFrameRate ) )
3230                 {
3231                     KaxVideoFrameRate &vfps = *(KaxVideoFrameRate*)l;
3232
3233                     tk->f_fps = float( vfps );
3234                     msg_Dbg( &sys.demuxer, "   |   |   |   + fps=%f", float( vfps ) );
3235                 }
3236 //                else if( EbmlId( *l ) == KaxVideoDisplayUnit::ClassInfos.GlobalId )
3237 //                {
3238 //                     KaxVideoDisplayUnit &vdmode = *(KaxVideoDisplayUnit*)l;
3239
3240 //                    msg_Dbg( &sys.demuxer, "|   |   |   |   + Track Video Display Unit=%s",
3241 //                             uint8( vdmode ) == 0 ? "pixels" : ( uint8( vdmode ) == 1 ? "centimeters": "inches" ) );
3242 //                }
3243 //                else if( EbmlId( *l ) == KaxVideoAspectRatio::ClassInfos.GlobalId )
3244 //                {
3245 //                    KaxVideoAspectRatio &ratio = *(KaxVideoAspectRatio*)l;
3246
3247 //                    msg_Dbg( &sys.demuxer, "   |   |   |   + Track Video Aspect Ratio Type=%u", uint8( ratio ) );
3248 //                }
3249 //                else if( EbmlId( *l ) == KaxVideoGamma::ClassInfos.GlobalId )
3250 //                {
3251 //                    KaxVideoGamma &gamma = *(KaxVideoGamma*)l;
3252
3253 //                    msg_Dbg( &sys.demuxer, "   |   |   |   + fps=%f", float( gamma ) );
3254 //                }
3255                 else
3256                 {
3257                     msg_Dbg( &sys.demuxer, "|   |   |   |   + Unknown (%s)", typeid(*l).name() );
3258                 }
3259             }
3260             if ( tk->fmt.video.i_visible_height && tk->fmt.video.i_visible_width )
3261                 tk->fmt.video.i_aspect = VOUT_ASPECT_FACTOR * tk->fmt.video.i_visible_width / tk->fmt.video.i_visible_height;
3262         }
3263         else  if( MKV_IS_ID( l, KaxTrackAudio ) )
3264         {
3265             EbmlMaster *tka = static_cast<EbmlMaster*>(l);
3266             unsigned int j;
3267
3268             msg_Dbg( &sys.demuxer, "|   |   |   + Track Audio" );
3269
3270             for( j = 0; j < tka->ListSize(); j++ )
3271             {
3272                 EbmlElement *l = (*tka)[j];
3273
3274                 if( MKV_IS_ID( l, KaxAudioSamplingFreq ) )
3275                 {
3276                     KaxAudioSamplingFreq &afreq = *(KaxAudioSamplingFreq*)l;
3277
3278                     tk->fmt.audio.i_rate = (int)float( afreq );
3279                     msg_Dbg( &sys.demuxer, "|   |   |   |   + afreq=%d", tk->fmt.audio.i_rate );
3280                 }
3281                 else if( MKV_IS_ID( l, KaxAudioChannels ) )
3282                 {
3283                     KaxAudioChannels &achan = *(KaxAudioChannels*)l;
3284
3285                     tk->fmt.audio.i_channels = uint8( achan );
3286                     msg_Dbg( &sys.demuxer, "|   |   |   |   + achan=%u", uint8( achan ) );
3287                 }
3288                 else if( MKV_IS_ID( l, KaxAudioBitDepth ) )
3289                 {
3290                     KaxAudioBitDepth &abits = *(KaxAudioBitDepth*)l;
3291
3292                     tk->fmt.audio.i_bitspersample = uint8( abits );
3293                     msg_Dbg( &sys.demuxer, "|   |   |   |   + abits=%u", uint8( abits ) );
3294                 }
3295                 else
3296                 {
3297                     msg_Dbg( &sys.demuxer, "|   |   |   |   + Unknown (%s)", typeid(*l).name() );
3298                 }
3299             }
3300         }
3301         else
3302         {
3303             msg_Dbg( &sys.demuxer, "|   |   |   + Unknown (%s)",
3304                      typeid(*l).name() );
3305         }
3306     }
3307 }
3308
3309 /*****************************************************************************
3310  * ParseTracks:
3311  *****************************************************************************/
3312 void matroska_segment_c::ParseTracks( KaxTracks *tracks )
3313 {
3314     EbmlElement *el;
3315     unsigned int i;
3316     int i_upper_level = 0;
3317
3318     msg_Dbg( &sys.demuxer, "|   + Tracks" );
3319
3320     /* Master elements */
3321     tracks->Read( es, tracks->Generic().Context, i_upper_level, el, true );
3322
3323     for( i = 0; i < tracks->ListSize(); i++ )
3324     {
3325         EbmlElement *l = (*tracks)[i];
3326
3327         if( MKV_IS_ID( l, KaxTrackEntry ) )
3328         {
3329             ParseTrackEntry( static_cast<KaxTrackEntry *>(l) );
3330         }
3331         else
3332         {
3333             msg_Dbg( &sys.demuxer, "|   |   + Unknown (%s)", typeid(*l).name() );
3334         }
3335     }
3336 }
3337
3338 /*****************************************************************************
3339  * ParseInfo:
3340  *****************************************************************************/
3341 void matroska_segment_c::ParseInfo( KaxInfo *info )
3342 {
3343     EbmlElement *el;
3344     EbmlMaster  *m;
3345     size_t i, j;
3346     int i_upper_level = 0;
3347
3348     msg_Dbg( &sys.demuxer, "|   + Information" );
3349
3350     /* Master elements */
3351     m = static_cast<EbmlMaster *>(info);
3352     m->Read( es, info->Generic().Context, i_upper_level, el, true );
3353
3354     for( i = 0; i < m->ListSize(); i++ )
3355     {
3356         EbmlElement *l = (*m)[i];
3357
3358         if( MKV_IS_ID( l, KaxSegmentUID ) )
3359         {
3360             segment_uid = *(new KaxSegmentUID(*static_cast<KaxSegmentUID*>(l)));
3361
3362             msg_Dbg( &sys.demuxer, "|   |   + UID=%d", *(uint32*)segment_uid.GetBuffer() );
3363         }
3364         else if( MKV_IS_ID( l, KaxPrevUID ) )
3365         {
3366             prev_segment_uid = *(new KaxPrevUID(*static_cast<KaxPrevUID*>(l)));
3367
3368             msg_Dbg( &sys.demuxer, "|   |   + PrevUID=%d", *(uint32*)prev_segment_uid.GetBuffer() );
3369         }
3370         else if( MKV_IS_ID( l, KaxNextUID ) )
3371         {
3372             next_segment_uid = *(new KaxNextUID(*static_cast<KaxNextUID*>(l)));
3373
3374             msg_Dbg( &sys.demuxer, "|   |   + NextUID=%d", *(uint32*)next_segment_uid.GetBuffer() );
3375         }
3376         else if( MKV_IS_ID( l, KaxTimecodeScale ) )
3377         {
3378             KaxTimecodeScale &tcs = *(KaxTimecodeScale*)l;
3379
3380             i_timescale = uint64(tcs);
3381
3382             msg_Dbg( &sys.demuxer, "|   |   + TimecodeScale="I64Fd,
3383                      i_timescale );
3384         }
3385         else if( MKV_IS_ID( l, KaxDuration ) )
3386         {
3387             KaxDuration &dur = *(KaxDuration*)l;
3388
3389             i_duration = mtime_t( double( dur ) );
3390
3391             msg_Dbg( &sys.demuxer, "|   |   + Duration="I64Fd,
3392                      i_duration );
3393         }
3394         else if( MKV_IS_ID( l, KaxMuxingApp ) )
3395         {
3396             KaxMuxingApp &mapp = *(KaxMuxingApp*)l;
3397
3398             psz_muxing_application = UTF8ToStr( UTFstring( mapp ) );
3399
3400             msg_Dbg( &sys.demuxer, "|   |   + Muxing Application=%s",
3401                      psz_muxing_application );
3402         }
3403         else if( MKV_IS_ID( l, KaxWritingApp ) )
3404         {
3405             KaxWritingApp &wapp = *(KaxWritingApp*)l;
3406
3407             psz_writing_application = UTF8ToStr( UTFstring( wapp ) );
3408
3409             msg_Dbg( &sys.demuxer, "|   |   + Writing Application=%s",
3410                      psz_writing_application );
3411         }
3412         else if( MKV_IS_ID( l, KaxSegmentFilename ) )
3413         {
3414             KaxSegmentFilename &sfn = *(KaxSegmentFilename*)l;
3415
3416             psz_segment_filename = UTF8ToStr( UTFstring( sfn ) );
3417
3418             msg_Dbg( &sys.demuxer, "|   |   + Segment Filename=%s",
3419                      psz_segment_filename );
3420         }
3421         else if( MKV_IS_ID( l, KaxTitle ) )
3422         {
3423             KaxTitle &title = *(KaxTitle*)l;
3424
3425             psz_title = UTF8ToStr( UTFstring( title ) );
3426
3427             msg_Dbg( &sys.demuxer, "|   |   + Title=%s", psz_title );
3428         }
3429         else if( MKV_IS_ID( l, KaxSegmentFamily ) )
3430         {
3431             KaxSegmentFamily *uid = static_cast<KaxSegmentFamily*>(l);
3432
3433             families.push_back(*uid);
3434
3435             msg_Dbg( &sys.demuxer, "|   |   + family=%d", *(uint32*)uid->GetBuffer() );
3436         }
3437 #if defined( HAVE_GMTIME_R ) && !defined( SYS_DARWIN )
3438         else if( MKV_IS_ID( l, KaxDateUTC ) )
3439         {
3440             KaxDateUTC &date = *(KaxDateUTC*)l;
3441             time_t i_date;
3442             struct tm tmres;
3443             char   buffer[256];
3444
3445             i_date = date.GetEpochDate();
3446             memset( buffer, 0, 256 );
3447             if( gmtime_r( &i_date, &tmres ) &&
3448                 asctime_r( &tmres, buffer ) )
3449             {
3450                 buffer[strlen( buffer)-1]= '\0';
3451                 psz_date_utc = strdup( buffer );
3452                 msg_Dbg( &sys.demuxer, "|   |   + Date=%s", psz_date_utc );
3453             }
3454         }
3455 #endif
3456 #if LIBMATROSKA_VERSION >= 0x000704
3457         else if( MKV_IS_ID( l, KaxChapterTranslate ) )
3458         {
3459             KaxChapterTranslate *p_trans = static_cast<KaxChapterTranslate*>( l );
3460             chapter_translation_c translated;
3461
3462             p_trans->Read( es, p_trans->Generic().Context, i_upper_level, el, true );
3463             for( j = 0; j < p_trans->ListSize(); j++ )
3464             {
3465                 EbmlElement *l = (*p_trans)[j];
3466
3467                 if( MKV_IS_ID( l, KaxChapterTranslateEditionUID ) )
3468                 {
3469                     translated.editions.push_back( uint64( *static_cast<KaxChapterTranslateEditionUID*>( l ) ) );
3470                 }
3471                 else if( MKV_IS_ID( l, KaxChapterTranslateCodec ) )
3472                 {
3473                     translated.codec_id = uint32( *static_cast<KaxChapterTranslateCodec*>( l ) );
3474                 }
3475                 else if( MKV_IS_ID( l, KaxChapterTranslateID ) )
3476                 {
3477                     translated.translated = *( new KaxChapterTranslateID( *static_cast<KaxChapterTranslateID*>( l ) ) );
3478                 }
3479             }
3480
3481             translations.push_back( translated );
3482         }
3483 #endif
3484         else
3485         {
3486             msg_Dbg( &sys.demuxer, "|   |   + Unknown (%s)", typeid(*l).name() );
3487         }
3488     }
3489
3490     double f_dur = double(i_duration) * double(i_timescale) / 1000000.0;
3491     i_duration = mtime_t(f_dur);
3492 }
3493
3494
3495 /*****************************************************************************
3496  * ParseChapterAtom
3497  *****************************************************************************/
3498 void matroska_segment_c::ParseChapterAtom( int i_level, KaxChapterAtom *ca, chapter_item_c & chapters )
3499 {
3500     size_t i, j;
3501
3502     msg_Dbg( &sys.demuxer, "|   |   |   + ChapterAtom (level=%d)", i_level );
3503     for( i = 0; i < ca->ListSize(); i++ )
3504     {
3505         EbmlElement *l = (*ca)[i];
3506
3507         if( MKV_IS_ID( l, KaxChapterUID ) )
3508         {
3509             chapters.i_uid = uint64_t(*(KaxChapterUID*)l);
3510             msg_Dbg( &sys.demuxer, "|   |   |   |   + ChapterUID: %lld", chapters.i_uid );
3511         }
3512         else if( MKV_IS_ID( l, KaxChapterFlagHidden ) )
3513         {
3514             KaxChapterFlagHidden &flag =*(KaxChapterFlagHidden*)l;
3515             chapters.b_display_seekpoint = uint8( flag ) == 0;
3516
3517             msg_Dbg( &sys.demuxer, "|   |   |   |   + ChapterFlagHidden: %s", chapters.b_display_seekpoint ? "no":"yes" );
3518         }
3519         else if( MKV_IS_ID( l, KaxChapterTimeStart ) )
3520         {
3521             KaxChapterTimeStart &start =*(KaxChapterTimeStart*)l;
3522             chapters.i_start_time = uint64( start ) / I64C(1000);
3523
3524             msg_Dbg( &sys.demuxer, "|   |   |   |   + ChapterTimeStart: %lld", chapters.i_start_time );
3525         }
3526         else if( MKV_IS_ID( l, KaxChapterTimeEnd ) )
3527         {
3528             KaxChapterTimeEnd &end =*(KaxChapterTimeEnd*)l;
3529             chapters.i_end_time = uint64( end ) / I64C(1000);
3530
3531             msg_Dbg( &sys.demuxer, "|   |   |   |   + ChapterTimeEnd: %lld", chapters.i_end_time );
3532         }
3533         else if( MKV_IS_ID( l, KaxChapterDisplay ) )
3534         {
3535             EbmlMaster *cd = static_cast<EbmlMaster *>(l);
3536
3537             msg_Dbg( &sys.demuxer, "|   |   |   |   + ChapterDisplay" );
3538             for( j = 0; j < cd->ListSize(); j++ )
3539             {
3540                 EbmlElement *l= (*cd)[j];
3541
3542                 if( MKV_IS_ID( l, KaxChapterString ) )
3543                 {
3544                     int k;
3545
3546                     KaxChapterString &name =*(KaxChapterString*)l;
3547                     for (k = 0; k < i_level; k++)
3548                         chapters.psz_name += '+';
3549                     chapters.psz_name += ' ';
3550                     chapters.psz_name += UTF8ToStr( UTFstring( name ) );
3551                     chapters.b_user_display = true;
3552
3553                     msg_Dbg( &sys.demuxer, "|   |   |   |   |    + ChapterString '%s'", UTF8ToStr(UTFstring(name)) );
3554                 }
3555                 else if( MKV_IS_ID( l, KaxChapterLanguage ) )
3556                 {
3557                     KaxChapterLanguage &lang =*(KaxChapterLanguage*)l;
3558                     const char *psz = string( lang ).c_str();
3559
3560                     msg_Dbg( &sys.demuxer, "|   |   |   |   |    + ChapterLanguage '%s'", psz );
3561                 }
3562                 else if( MKV_IS_ID( l, KaxChapterCountry ) )
3563                 {
3564                     KaxChapterCountry &ct =*(KaxChapterCountry*)l;
3565                     const char *psz = string( ct ).c_str();
3566
3567                     msg_Dbg( &sys.demuxer, "|   |   |   |   |    + ChapterCountry '%s'", psz );
3568                 }
3569             }
3570         }
3571         else if( MKV_IS_ID( l, KaxChapterProcess ) )
3572         {
3573             msg_Dbg( &sys.demuxer, "|   |   |   |   + ChapterProcess" );
3574
3575             KaxChapterProcess *cp = static_cast<KaxChapterProcess *>(l);
3576             chapter_codec_cmds_c *p_ccodec = NULL;
3577
3578             for( j = 0; j < cp->ListSize(); j++ )
3579             {
3580                 EbmlElement *k= (*cp)[j];
3581
3582                 if( MKV_IS_ID( k, KaxChapterProcessCodecID ) )
3583                 {
3584                     KaxChapterProcessCodecID *p_codec_id = static_cast<KaxChapterProcessCodecID*>( k );
3585                     if ( uint32(*p_codec_id) == 0 )
3586                         p_ccodec = new matroska_script_codec_c( sys );
3587                     else if ( uint32(*p_codec_id) == 1 )
3588                         p_ccodec = new dvd_chapter_codec_c( sys );
3589                     break;
3590                 }
3591             }
3592
3593             if ( p_ccodec != NULL )
3594             {
3595                 for( j = 0; j < cp->ListSize(); j++ )
3596                 {
3597                     EbmlElement *k= (*cp)[j];
3598
3599                     if( MKV_IS_ID( k, KaxChapterProcessPrivate ) )
3600                     {
3601                         KaxChapterProcessPrivate * p_private = static_cast<KaxChapterProcessPrivate*>( k );
3602                         p_ccodec->SetPrivate( *p_private );
3603                     }
3604                     else if( MKV_IS_ID( k, KaxChapterProcessCommand ) )
3605                     {
3606                         p_ccodec->AddCommand( *static_cast<KaxChapterProcessCommand*>( k ) );
3607                     }
3608                 }
3609                 chapters.codecs.push_back( p_ccodec );
3610             }
3611         }
3612         else if( MKV_IS_ID( l, KaxChapterAtom ) )
3613         {
3614             chapter_item_c *new_sub_chapter = new chapter_item_c();
3615             ParseChapterAtom( i_level+1, static_cast<KaxChapterAtom *>(l), *new_sub_chapter );
3616             new_sub_chapter->psz_parent = &chapters;
3617             chapters.sub_chapters.push_back( new_sub_chapter );
3618         }
3619     }
3620 }
3621
3622 /*****************************************************************************
3623  * ParseChapters:
3624  *****************************************************************************/
3625 void matroska_segment_c::ParseChapters( KaxChapters *chapters )
3626 {
3627     EbmlElement *el;
3628     size_t i;
3629     int i_upper_level = 0;
3630     mtime_t i_dur;
3631
3632     /* Master elements */
3633     chapters->Read( es, chapters->Generic().Context, i_upper_level, el, true );
3634
3635     for( i = 0; i < chapters->ListSize(); i++ )
3636     {
3637         EbmlElement *l = (*chapters)[i];
3638
3639         if( MKV_IS_ID( l, KaxEditionEntry ) )
3640         {
3641             chapter_edition_c *p_edition = new chapter_edition_c();
3642
3643             EbmlMaster *E = static_cast<EbmlMaster *>(l );
3644             size_t j;
3645             msg_Dbg( &sys.demuxer, "|   |   + EditionEntry" );
3646             for( j = 0; j < E->ListSize(); j++ )
3647             {
3648                 EbmlElement *l = (*E)[j];
3649
3650                 if( MKV_IS_ID( l, KaxChapterAtom ) )
3651                 {
3652                     chapter_item_c *new_sub_chapter = new chapter_item_c();
3653                     ParseChapterAtom( 0, static_cast<KaxChapterAtom *>(l), *new_sub_chapter );
3654                     p_edition->sub_chapters.push_back( new_sub_chapter );
3655                 }
3656                 else if( MKV_IS_ID( l, KaxEditionUID ) )
3657                 {
3658                     p_edition->i_uid = uint64(*static_cast<KaxEditionUID *>( l ));
3659                 }
3660                 else if( MKV_IS_ID( l, KaxEditionFlagOrdered ) )
3661                 {
3662                     p_edition->b_ordered = config_GetInt( &sys.demuxer, "mkv-use-ordered-chapters" ) ? (uint8(*static_cast<KaxEditionFlagOrdered *>( l )) != 0) : 0;
3663                 }
3664                 else if( MKV_IS_ID( l, KaxEditionFlagDefault ) )
3665                 {
3666                     if (uint8(*static_cast<KaxEditionFlagDefault *>( l )) != 0)
3667                         i_default_edition = stored_editions.size();
3668                 }
3669                 else
3670                 {
3671                     msg_Dbg( &sys.demuxer, "|   |   |   + Unknown (%s)", typeid(*l).name() );
3672                 }
3673             }
3674             stored_editions.push_back( p_edition );
3675         }
3676         else
3677         {
3678             msg_Dbg( &sys.demuxer, "|   |   + Unknown (%s)", typeid(*l).name() );
3679         }
3680     }
3681
3682     for( i = 0; i < stored_editions.size(); i++ )
3683     {
3684         stored_editions[i]->RefreshChapters( );
3685     }
3686
3687     if ( stored_editions[i_default_edition]->b_ordered )
3688     {
3689         /* update the duration of the segment according to the sum of all sub chapters */
3690         i_dur = stored_editions[i_default_edition]->Duration() / I64C(1000);
3691         if (i_dur > 0)
3692             i_duration = i_dur;
3693     }
3694 }
3695
3696 void matroska_segment_c::ParseCluster( )
3697 {
3698     EbmlElement *el;
3699     EbmlMaster  *m;
3700     unsigned int i;
3701     int i_upper_level = 0;
3702
3703     /* Master elements */
3704     m = static_cast<EbmlMaster *>( cluster );
3705     m->Read( es, cluster->Generic().Context, i_upper_level, el, true );
3706
3707     for( i = 0; i < m->ListSize(); i++ )
3708     {
3709         EbmlElement *l = (*m)[i];
3710
3711         if( MKV_IS_ID( l, KaxClusterTimecode ) )
3712         {
3713             KaxClusterTimecode &ctc = *(KaxClusterTimecode*)l;
3714
3715             cluster->InitTimecode( uint64( ctc ), i_timescale );
3716             break;
3717         }
3718     }
3719
3720     i_start_time = cluster->GlobalTimecode() / 1000;
3721 }
3722
3723 /*****************************************************************************
3724  * InformationCreate:
3725  *****************************************************************************/
3726 void matroska_segment_c::InformationCreate( )
3727 {
3728     size_t      i_track;
3729
3730     sys.meta = vlc_meta_New();
3731
3732     if( psz_title )
3733     {
3734         vlc_meta_Add( sys.meta, VLC_META_TITLE, psz_title );
3735     }
3736     if( psz_date_utc )
3737     {
3738         vlc_meta_Add( sys.meta, VLC_META_DATE, psz_date_utc );
3739     }
3740     if( psz_segment_filename )
3741     {
3742         vlc_meta_Add( sys.meta, _("Segment filename"), psz_segment_filename );
3743     }
3744     if( psz_muxing_application )
3745     {
3746         vlc_meta_Add( sys.meta, _("Muxing application"), psz_muxing_application );
3747     }
3748     if( psz_writing_application )
3749     {
3750         vlc_meta_Add( sys.meta, _("Writing application"), psz_writing_application );
3751     }
3752
3753     for( i_track = 0; i_track < tracks.size(); i_track++ )
3754     {
3755         mkv_track_t *tk = tracks[i_track];
3756         vlc_meta_t *mtk = vlc_meta_New();
3757
3758         sys.meta->track = (vlc_meta_t**)realloc( sys.meta->track,
3759                                                     sizeof( vlc_meta_t * ) * ( sys.meta->i_track + 1 ) );
3760         sys.meta->track[sys.meta->i_track++] = mtk;
3761
3762         if( tk->fmt.psz_description )
3763         {
3764             vlc_meta_Add( sys.meta, VLC_META_DESCRIPTION, tk->fmt.psz_description );
3765         }
3766         if( tk->psz_codec_name )
3767         {
3768             vlc_meta_Add( sys.meta, VLC_META_CODEC_NAME, tk->psz_codec_name );
3769         }
3770         if( tk->psz_codec_settings )
3771         {
3772             vlc_meta_Add( sys.meta, VLC_META_SETTING, tk->psz_codec_settings );
3773         }
3774         if( tk->psz_codec_info_url )
3775         {
3776             vlc_meta_Add( sys.meta, VLC_META_CODEC_DESCRIPTION, tk->psz_codec_info_url );
3777         }
3778         if( tk->psz_codec_download_url )
3779         {
3780             vlc_meta_Add( sys.meta, VLC_META_URL, tk->psz_codec_download_url );
3781         }
3782     }
3783
3784     if( i_tags_position >= 0 )
3785     {
3786         vlc_bool_t b_seekable;
3787
3788         stream_Control( sys.demuxer.s, STREAM_CAN_FASTSEEK, &b_seekable );
3789         if( b_seekable )
3790         {
3791             LoadTags( );
3792         }
3793     }
3794 }
3795
3796
3797 /*****************************************************************************
3798  * Divers
3799  *****************************************************************************/
3800
3801 void matroska_segment_c::IndexAppendCluster( KaxCluster *cluster )
3802 {
3803 #define idx index[i_index]
3804     idx.i_track       = -1;
3805     idx.i_block_number= -1;
3806     idx.i_position    = cluster->GetElementPosition();
3807     idx.i_time        = -1;
3808     idx.b_key         = VLC_TRUE;
3809
3810     i_index++;
3811     if( i_index >= i_index_max )
3812     {
3813         i_index_max += 1024;
3814         index = (mkv_index_t*)realloc( index, sizeof( mkv_index_t ) * i_index_max );
3815     }
3816 #undef idx
3817 }
3818
3819 static char * UTF8ToStr( const UTFstring &u )
3820 {
3821     int     i_src;
3822     const wchar_t *src;
3823     char *dst, *p;
3824
3825     i_src = u.length();
3826     src   = u.c_str();
3827
3828     p = dst = (char*)malloc( i_src + 1);
3829     while( i_src > 0 )
3830     {
3831         if( *src < 255 )
3832         {
3833             *p++ = (char)*src;
3834         }
3835         else
3836         {
3837             *p++ = '?';
3838         }
3839         src++;
3840         i_src--;
3841     }
3842     *p++= '\0';
3843
3844     return dst;
3845 }
3846
3847 void chapter_edition_c::RefreshChapters( )
3848 {
3849     chapter_item_c::RefreshChapters( b_ordered, -1 );
3850     b_display_seekpoint = false;
3851 }
3852
3853 int64_t chapter_item_c::RefreshChapters( bool b_ordered, int64_t i_prev_user_time )
3854 {
3855     int64_t i_user_time = i_prev_user_time;
3856
3857     // first the sub-chapters, and then ourself
3858     std::vector<chapter_item_c*>::iterator index = sub_chapters.begin();
3859     while ( index != sub_chapters.end() )
3860     {
3861         i_user_time = (*index)->RefreshChapters( b_ordered, i_user_time );
3862         index++;
3863     }
3864
3865     if ( b_ordered )
3866     {
3867         // the ordered chapters always start at zero
3868         if ( i_prev_user_time == -1 )
3869         {
3870             if ( i_user_time == -1 )
3871                 i_user_time = 0;
3872             i_prev_user_time = 0;
3873         }
3874
3875         i_user_start_time = i_prev_user_time;
3876         if ( i_end_time != -1 && i_user_time == i_prev_user_time )
3877         {
3878             i_user_end_time = i_user_start_time - i_start_time + i_end_time;
3879         }
3880         else
3881         {
3882             i_user_end_time = i_user_time;
3883         }
3884     }
3885     else
3886     {
3887         std::sort( sub_chapters.begin(), sub_chapters.end() );
3888         i_user_start_time = i_start_time;
3889         if ( i_end_time != -1 )
3890             i_user_end_time = i_end_time;
3891         else if ( i_user_time != -1 )
3892             i_user_end_time = i_user_time;
3893         else
3894             i_user_end_time = i_user_start_time;
3895     }
3896
3897     return i_user_end_time;
3898 }
3899
3900 mtime_t chapter_edition_c::Duration() const
3901 {
3902     mtime_t i_result = 0;
3903
3904     if ( sub_chapters.size() )
3905     {
3906         std::vector<chapter_item_c*>::const_iterator index = sub_chapters.end();
3907         index--;
3908         i_result = (*index)->i_user_end_time;
3909     }
3910
3911     return i_result;
3912 }
3913
3914 chapter_item_c *chapter_item_c::FindTimecode( mtime_t i_user_timecode )
3915 {
3916     chapter_item_c *psz_result = NULL;
3917
3918     if ( i_user_timecode >= i_user_start_time &&
3919         ( i_user_timecode < i_user_end_time ||
3920           ( i_user_start_time == i_user_end_time && i_user_timecode == i_user_end_time )))
3921     {
3922         std::vector<chapter_item_c*>::iterator index = sub_chapters.begin();
3923         while ( index != sub_chapters.end() && psz_result == NULL )
3924         {
3925             psz_result = (*index)->FindTimecode( i_user_timecode );
3926             index++;
3927         }
3928
3929         if ( psz_result == NULL )
3930             psz_result = this;
3931     }
3932
3933     return psz_result;
3934 }
3935
3936 bool chapter_item_c::ParentOf( const chapter_item_c & item ) const
3937 {
3938     if ( &item == this )
3939         return true;
3940
3941     std::vector<chapter_item_c*>::const_iterator index = sub_chapters.begin();
3942     while ( index != sub_chapters.end() )
3943     {
3944         if ( (*index)->ParentOf( item ) )
3945             return true;
3946         index++;
3947     }
3948
3949     return false;
3950 }
3951
3952 void demux_sys_t::PreloadFamily( const matroska_segment_c & of_segment )
3953 {
3954     for (size_t i=0; i<opened_segments.size(); i++)
3955     {
3956         opened_segments[i]->PreloadFamily( of_segment );
3957     }
3958 }
3959 bool matroska_segment_c::PreloadFamily( const matroska_segment_c & of_segment )
3960 {
3961     if ( b_preloaded )
3962         return false;
3963
3964     for (size_t i=0; i<families.size(); i++)
3965     {
3966         for (size_t j=0; j<of_segment.families.size(); j++)
3967         {
3968             if ( families[i] == of_segment.families[j] )
3969                 return Preload( );
3970         }
3971     }
3972
3973     return false;
3974 }
3975
3976 // preload all the linked segments for all preloaded segments
3977 void demux_sys_t::PreloadLinked( matroska_segment_c *p_segment )
3978 {
3979     size_t i_preloaded, i, j;
3980     virtual_segment_c *p_seg;
3981
3982     p_current_segment = VirtualFromSegments( p_segment );
3983
3984     used_segments.push_back( p_current_segment );
3985
3986     // create all the other virtual segments of the family
3987     do {
3988         i_preloaded = 0;
3989         for ( i=0; i< opened_segments.size(); i++ )
3990         {
3991             if ( opened_segments[i]->b_preloaded && !IsUsedSegment( *opened_segments[i] ) )
3992             {
3993                 p_seg = VirtualFromSegments( opened_segments[i] );
3994                 used_segments.push_back( p_seg );
3995                 i_preloaded++;
3996             }
3997         }
3998     } while ( i_preloaded ); // worst case: will stop when all segments are found as family related
3999
4000     // publish all editions of all usable segment
4001     for ( i=0; i< used_segments.size(); i++ )
4002     {
4003         p_seg = used_segments[i];
4004         if ( p_seg->p_editions != NULL )
4005         {
4006             std::string sz_name;
4007             input_title_t *p_title = vlc_input_title_New();
4008             p_seg->i_sys_title = i;
4009             int i_chapters;
4010
4011             // TODO use a name for each edition, let the TITLE deal with a codec name
4012             for ( j=0; j<p_seg->p_editions->size(); j++ )
4013             {
4014                 if ( p_title->psz_name == NULL )
4015                 {
4016                     sz_name = (*p_seg->p_editions)[j]->GetMainName();
4017                     if ( sz_name != "" )
4018                         p_title->psz_name = strdup( sz_name.c_str() );
4019                 }
4020
4021                 chapter_edition_c *p_edition = (*p_seg->p_editions)[j];
4022
4023                 i_chapters = 0;
4024                 p_edition->PublishChapters( *p_title, i_chapters, 0 );
4025             }
4026
4027             // create a name if there is none
4028             if ( p_title->psz_name == NULL )
4029             {
4030                 sz_name = N_("Segment ");
4031                 char psz_str[6];
4032                 sprintf( psz_str, "%d", i );
4033                 sz_name += psz_str;
4034                 p_title->psz_name = strdup( sz_name.c_str() );
4035             }
4036
4037             titles.push_back( *p_title );
4038         }
4039     }
4040 }
4041
4042 bool demux_sys_t::IsUsedSegment( matroska_segment_c &segment ) const
4043 {
4044     for ( size_t i=0; i< used_segments.size(); i++ )
4045     {
4046         if ( used_segments[i]->FindUID( segment.segment_uid ) )
4047             return true;
4048     }
4049     return false;
4050 }
4051
4052 virtual_segment_c *demux_sys_t::VirtualFromSegments( matroska_segment_c *p_segment ) const
4053 {
4054     size_t i_preloaded, i;
4055
4056     virtual_segment_c *p_result = new virtual_segment_c( p_segment );
4057
4058     // fill our current virtual segment with all hard linked segments
4059     do {
4060         i_preloaded = 0;
4061         for ( i=0; i< opened_segments.size(); i++ )
4062         {
4063             i_preloaded += p_result->AddSegment( opened_segments[i] );
4064         }
4065     } while ( i_preloaded ); // worst case: will stop when all segments are found as linked
4066
4067     p_result->Sort( );
4068
4069     p_result->PreloadLinked( );
4070
4071     p_result->PrepareChapters( );
4072
4073     return p_result;
4074 }
4075
4076 bool demux_sys_t::PreparePlayback( virtual_segment_c *p_new_segment )
4077 {
4078     if ( p_new_segment != NULL && p_new_segment != p_current_segment )
4079     {
4080         if ( p_current_segment != NULL && p_current_segment->Segment() != NULL )
4081             p_current_segment->Segment()->UnSelect();
4082
4083         p_current_segment = p_new_segment;
4084         i_current_title = p_new_segment->i_sys_title;
4085     }
4086
4087     p_current_segment->LoadCues();
4088     f_duration = p_current_segment->Duration();
4089
4090     /* add information */
4091     p_current_segment->Segment()->InformationCreate( );
4092
4093     p_current_segment->Segment()->Select( 0 );
4094
4095     return true;
4096 }
4097
4098 bool matroska_segment_c::CompareSegmentUIDs( const matroska_segment_c * p_item_a, const matroska_segment_c * p_item_b )
4099 {
4100     EbmlBinary * p_itema = (EbmlBinary *)(&p_item_a->segment_uid);
4101     if ( *p_itema == p_item_b->prev_segment_uid )
4102         return true;
4103
4104     p_itema = (EbmlBinary *)(&p_item_a->next_segment_uid);
4105     if ( *p_itema == p_item_b->segment_uid )
4106         return true;
4107
4108     if ( *p_itema == p_item_b->prev_segment_uid )
4109         return true;
4110
4111     return false;
4112 }
4113
4114 bool matroska_segment_c::Preload( )
4115 {
4116     if ( b_preloaded )
4117         return false;
4118
4119     EbmlElement *el = NULL;
4120
4121     ep->Reset( &sys.demuxer );
4122
4123     while( ( el = ep->Get() ) != NULL )
4124     {
4125         if( MKV_IS_ID( el, KaxInfo ) )
4126         {
4127             ParseInfo( static_cast<KaxInfo*>( el ) );
4128         }
4129         else if( MKV_IS_ID( el, KaxTracks ) )
4130         {
4131             ParseTracks( static_cast<KaxTracks*>( el ) );
4132         }
4133         else if( MKV_IS_ID( el, KaxSeekHead ) )
4134         {
4135             ParseSeekHead( static_cast<KaxSeekHead*>( el ) );
4136         }
4137         else if( MKV_IS_ID( el, KaxCues ) )
4138         {
4139             msg_Dbg( &sys.demuxer, "|   + Cues" );
4140         }
4141         else if( MKV_IS_ID( el, KaxCluster ) )
4142         {
4143             msg_Dbg( &sys.demuxer, "|   + Cluster" );
4144
4145             cluster = (KaxCluster*)el;
4146
4147             i_start_pos = cluster->GetElementPosition();
4148             ParseCluster( );
4149
4150             ep->Down();
4151             /* stop parsing the stream */
4152             break;
4153         }
4154         else if( MKV_IS_ID( el, KaxAttachments ) )
4155         {
4156             msg_Dbg( &sys.demuxer, "|   + Attachments FIXME (but probably never supported)" );
4157         }
4158         else if( MKV_IS_ID( el, KaxChapters ) )
4159         {
4160             msg_Dbg( &sys.demuxer, "|   + Chapters" );
4161             ParseChapters( static_cast<KaxChapters*>( el ) );
4162         }
4163         else if( MKV_IS_ID( el, KaxTag ) )
4164         {
4165             msg_Dbg( &sys.demuxer, "|   + Tags FIXME TODO" );
4166         }
4167         else
4168         {
4169             msg_Dbg( &sys.demuxer, "|   + Unknown (%s)", typeid(*el).name() );
4170         }
4171     }
4172
4173     b_preloaded = true;
4174
4175     return true;
4176 }
4177
4178 matroska_segment_c *demux_sys_t::FindSegment( const EbmlBinary & uid ) const
4179 {
4180     for (size_t i=0; i<opened_segments.size(); i++)
4181     {
4182         if ( opened_segments[i]->segment_uid == uid )
4183             return opened_segments[i];
4184     }
4185     return NULL;
4186 }
4187
4188 chapter_item_c *demux_sys_t::BrowseCodecPrivate( unsigned int codec_id,
4189                                         bool (*match)(const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size ),
4190                                         const void *p_cookie,
4191                                         size_t i_cookie_size,
4192                                         virtual_segment_c * &p_segment_found )
4193 {
4194     chapter_item_c *p_result = NULL;
4195     for (size_t i=0; i<used_segments.size(); i++)
4196     {
4197         p_result = used_segments[i]->BrowseCodecPrivate( codec_id, match, p_cookie, i_cookie_size );
4198         if ( p_result != NULL )
4199         {
4200             p_segment_found = used_segments[i];
4201             break;
4202         }
4203     }
4204     return p_result;
4205 }
4206
4207 chapter_item_c *demux_sys_t::FindChapter( int64_t i_find_uid, virtual_segment_c * & p_segment_found )
4208 {
4209     chapter_item_c *p_result = NULL;
4210     for (size_t i=0; i<used_segments.size(); i++)
4211     {
4212         p_result = used_segments[i]->FindChapter( i_find_uid );
4213         if ( p_result != NULL )
4214         {
4215             p_segment_found = used_segments[i];
4216             break;
4217         }
4218     }
4219     return p_result;
4220 }
4221
4222 void virtual_segment_c::Sort()
4223 {
4224     // keep the current segment index
4225     matroska_segment_c *p_segment = linked_segments[i_current_segment];
4226
4227     std::sort( linked_segments.begin(), linked_segments.end(), matroska_segment_c::CompareSegmentUIDs );
4228
4229     for ( i_current_segment=0; i_current_segment<linked_segments.size(); i_current_segment++)
4230         if ( linked_segments[i_current_segment] == p_segment )
4231             break;
4232 }
4233
4234 size_t virtual_segment_c::AddSegment( matroska_segment_c *p_segment )
4235 {
4236     size_t i;
4237     // check if it's not already in here
4238     for ( i=0; i<linked_segments.size(); i++ )
4239     {
4240         if ( p_segment->segment_uid == linked_segments[i]->segment_uid )
4241             return 0;
4242     }
4243
4244     // find possible mates
4245     for ( i=0; i<linked_uids.size(); i++ )
4246     {
4247         if (   p_segment->segment_uid == linked_uids[i]
4248             || p_segment->prev_segment_uid == linked_uids[i]
4249             || p_segment->next_segment_uid == linked_uids[i] )
4250         {
4251             linked_segments.push_back( p_segment );
4252
4253             AppendUID( p_segment->prev_segment_uid );
4254             AppendUID( p_segment->next_segment_uid );
4255
4256             return 1;
4257         }
4258     }
4259     return 0;
4260 }
4261
4262 void virtual_segment_c::PreloadLinked( )
4263 {
4264     for ( size_t i=0; i<linked_segments.size(); i++ )
4265     {
4266         linked_segments[i]->Preload( );
4267     }
4268     i_current_edition = linked_segments[0]->i_default_edition;
4269 }
4270
4271 mtime_t virtual_segment_c::Duration() const
4272 {
4273     mtime_t i_duration;
4274     if ( linked_segments.size() == 0 )
4275         i_duration = 0;
4276     else {
4277         matroska_segment_c *p_last_segment = linked_segments[linked_segments.size()-1];
4278 //        p_last_segment->ParseCluster( );
4279
4280         i_duration = p_last_segment->i_start_time / 1000 + p_last_segment->i_duration;
4281     }
4282     return i_duration;
4283 }
4284
4285 void virtual_segment_c::LoadCues( )
4286 {
4287     for ( size_t i=0; i<linked_segments.size(); i++ )
4288     {
4289         linked_segments[i]->LoadCues();
4290     }
4291 }
4292
4293 void virtual_segment_c::AppendUID( const EbmlBinary & UID )
4294 {
4295     if ( UID.GetBuffer() == NULL )
4296         return;
4297
4298     for (size_t i=0; i<linked_uids.size(); i++)
4299     {
4300         if ( UID == linked_uids[i] )
4301             return;
4302     }
4303     linked_uids.push_back( *(KaxSegmentUID*)(&UID) );
4304 }
4305
4306 void matroska_segment_c::Seek( mtime_t i_date, mtime_t i_time_offset )
4307 {
4308     KaxBlock    *block;
4309     int         i_track_skipping;
4310     int64_t     i_block_duration;
4311     int64_t     i_block_ref1;
4312     int64_t     i_block_ref2;
4313     size_t      i_track;
4314     int64_t     i_seek_position = i_start_pos;
4315     int64_t     i_seek_time = i_start_time;
4316
4317     if ( i_index > 0 )
4318     {
4319         int i_idx = 0;
4320
4321         for( ; i_idx < i_index; i_idx++ )
4322         {
4323             if( index[i_idx].i_time + i_time_offset > i_date )
4324             {
4325                 break;
4326             }
4327         }
4328
4329         if( i_idx > 0 )
4330         {
4331             i_idx--;
4332         }
4333
4334         i_seek_position = index[i_idx].i_position;
4335         i_seek_time = index[i_idx].i_time;
4336     }
4337
4338     msg_Dbg( &sys.demuxer, "seek got "I64Fd" (%d%%)",
4339                 i_seek_time, (int)( 100 * i_seek_position / stream_Size( sys.demuxer.s ) ) );
4340
4341     es.I_O().setFilePointer( i_seek_position, seek_beginning );
4342
4343     delete ep;
4344     ep = new EbmlParser( &es, segment, &sys.demuxer );
4345     cluster = NULL;
4346
4347     sys.i_start_pts = i_date;
4348
4349     es_out_Control( sys.demuxer.out, ES_OUT_RESET_PCR );
4350
4351     /* now parse until key frame */
4352 #define tk  tracks[i_track]
4353     i_track_skipping = 0;
4354     for( i_track = 0; i_track < tracks.size(); i_track++ )
4355     {
4356         if( tk->fmt.i_cat == VIDEO_ES )
4357         {
4358             tk->b_search_keyframe = VLC_TRUE;
4359             i_track_skipping++;
4360         }
4361         es_out_Control( sys.demuxer.out, ES_OUT_SET_NEXT_DISPLAY_TIME, tk->p_es, i_date );
4362     }
4363
4364
4365     while( i_track_skipping > 0 )
4366     {
4367         if( BlockGet( &block, &i_block_ref1, &i_block_ref2, &i_block_duration ) )
4368         {
4369             msg_Warn( &sys.demuxer, "cannot get block EOF?" );
4370
4371             return;
4372         }
4373
4374         for( i_track = 0; i_track < tracks.size(); i_track++ )
4375         {
4376             if( tk->i_number == block->TrackNum() )
4377             {
4378                 break;
4379             }
4380         }
4381
4382         sys.i_pts = sys.i_chapter_time + block->GlobalTimecode() / (mtime_t) 1000;
4383
4384         if( i_track < tracks.size() )
4385         {
4386             if( sys.i_pts >= sys.i_start_pts )
4387             {
4388                 BlockDecode( &sys.demuxer, block, sys.i_pts, 0 );
4389                 i_track_skipping = 0;
4390             }
4391             else if( tk->fmt.i_cat == VIDEO_ES )
4392             {
4393                 if( i_block_ref1 == -1 && tk->b_search_keyframe )
4394                 {
4395                     tk->b_search_keyframe = VLC_FALSE;
4396                     i_track_skipping--;
4397                 }
4398                 if( !tk->b_search_keyframe )
4399                 {
4400                     BlockDecode( &sys.demuxer, block, sys.i_pts, 0 );
4401                 }
4402             }
4403         }
4404
4405         delete block;
4406     }
4407 #undef tk
4408 }
4409
4410 void virtual_segment_c::Seek( demux_t & demuxer, mtime_t i_date, mtime_t i_time_offset, chapter_item_c *psz_chapter )
4411 {
4412     demux_sys_t *p_sys = demuxer.p_sys;
4413     size_t i;
4414
4415     // find the actual time for an ordered edition
4416     if ( psz_chapter == NULL )
4417     {
4418         if ( Edition() && Edition()->b_ordered )
4419         {
4420             /* 1st, we need to know in which chapter we are */
4421             psz_chapter = (*p_editions)[i_current_edition]->FindTimecode( i_date );
4422         }
4423     }
4424
4425     if ( psz_chapter != NULL )
4426     {
4427         psz_current_chapter = psz_chapter;
4428         p_sys->i_chapter_time = i_time_offset = psz_chapter->i_user_start_time - psz_chapter->i_start_time;
4429         if ( psz_chapter->i_seekpoint_num > 0 )
4430         {
4431             demuxer.info.i_update |= INPUT_UPDATE_TITLE | INPUT_UPDATE_SEEKPOINT;
4432             demuxer.info.i_title = p_sys->i_current_title = i_sys_title;
4433             demuxer.info.i_seekpoint = psz_chapter->i_seekpoint_num - 1;
4434         }
4435     }
4436
4437     // find the best matching segment
4438     for ( i=0; i<linked_segments.size(); i++ )
4439     {
4440         if ( i_date < linked_segments[i]->i_start_time )
4441             break;
4442     }
4443
4444     if ( i > 0 )
4445         i--;
4446
4447     if ( i_current_segment != i  )
4448     {
4449         linked_segments[i_current_segment]->UnSelect();
4450         linked_segments[i]->Select( i_date );
4451         i_current_segment = i;
4452     }
4453
4454     linked_segments[i]->Seek( i_date, i_time_offset );
4455 }
4456
4457 void chapter_codec_cmds_c::AddCommand( const KaxChapterProcessCommand & command )
4458 {
4459     size_t i;
4460
4461     uint32 codec_time = uint32(-1);
4462     for( i = 0; i < command.ListSize(); i++ )
4463     {
4464         const EbmlElement *k = command[i];
4465
4466         if( MKV_IS_ID( k, KaxChapterProcessTime ) )
4467         {
4468             codec_time = uint32( *static_cast<const KaxChapterProcessTime*>( k ) );
4469             break;
4470         }
4471     }
4472
4473     for( i = 0; i < command.ListSize(); i++ )
4474     {
4475         const EbmlElement *k = command[i];
4476
4477         if( MKV_IS_ID( k, KaxChapterProcessData ) )
4478         {
4479             KaxChapterProcessData *p_data =  new KaxChapterProcessData( *static_cast<const KaxChapterProcessData*>( k ) );
4480             switch ( codec_time )
4481             {
4482             case 0:
4483                 during_cmds.push_back( *p_data );
4484                 break;
4485             case 1:
4486                 enter_cmds.push_back( *p_data );
4487                 break;
4488             case 2:
4489                 leave_cmds.push_back( *p_data );
4490                 break;
4491             default:
4492                 delete p_data;
4493             }
4494         }
4495     }
4496 }
4497
4498 bool chapter_item_c::Enter( bool b_do_subs )
4499 {
4500     bool f_result = false;
4501     std::vector<chapter_codec_cmds_c*>::iterator index = codecs.begin();
4502     while ( index != codecs.end() )
4503     {
4504         f_result |= (*index)->Enter();
4505         index++;
4506     }
4507
4508     if ( b_do_subs )
4509     {
4510         // sub chapters
4511         std::vector<chapter_item_c*>::iterator index_ = sub_chapters.begin();
4512         while ( index_ != sub_chapters.end() )
4513         {
4514             f_result |= (*index_)->Enter( true );
4515             index_++;
4516         }
4517     }
4518     return f_result;
4519 }
4520
4521 bool chapter_item_c::Leave( bool b_do_subs )
4522 {
4523     bool f_result = false;
4524     b_is_leaving = true;
4525     std::vector<chapter_codec_cmds_c*>::iterator index = codecs.begin();
4526     while ( index != codecs.end() )
4527     {
4528         f_result |= (*index)->Leave();
4529         index++;
4530     }
4531
4532     if ( b_do_subs )
4533     {
4534         // sub chapters
4535         std::vector<chapter_item_c*>::iterator index_ = sub_chapters.begin();
4536         while ( index_ != sub_chapters.end() )
4537         {
4538             f_result |= (*index_)->Leave( true );
4539             index_++;
4540         }
4541     }
4542     b_is_leaving = false;
4543     return f_result;
4544 }
4545
4546 bool chapter_item_c::EnterAndLeave( chapter_item_c *p_item )
4547 {
4548     chapter_item_c *p_common_parent = p_item;
4549
4550     // leave, up to a common parent
4551     while ( p_common_parent != NULL && !p_common_parent->ParentOf( *this ) )
4552     {
4553         if ( !p_common_parent->b_is_leaving && p_common_parent->Leave( false ) )
4554             return true;
4555         p_common_parent = p_common_parent->psz_parent;
4556     }
4557
4558     // enter from the parent to <this>
4559     if ( p_common_parent != NULL )
4560     {
4561         do
4562         {
4563             for ( size_t i = 0; i<p_common_parent->sub_chapters.size(); i++ )
4564             {
4565                 if ( p_common_parent->sub_chapters[i]->ParentOf( *this ) )
4566                 {
4567                     p_common_parent = p_common_parent->sub_chapters[i];
4568                     break;
4569                 }
4570             }
4571
4572             if ( p_common_parent == this )
4573                 break;
4574
4575             if ( p_common_parent->Enter( false ) )
4576                 return true;
4577         } while ( 1 );
4578     }
4579
4580     return Enter( true );
4581 }
4582
4583 bool dvd_chapter_codec_c::Enter()
4584 {
4585     bool f_result = false;
4586     std::vector<KaxChapterProcessData>::iterator index = enter_cmds.begin();
4587     while ( index != enter_cmds.end() )
4588     {
4589         if ( (*index).GetSize() )
4590         {
4591             binary *p_data = (*index).GetBuffer();
4592             size_t i_size = *p_data++;
4593             // avoid reading too much from the buffer
4594             i_size = min( i_size, ((*index).GetSize() - 1) >> 3 );
4595             for ( ; i_size > 0; i_size--, p_data += 8 )
4596             {
4597                 msg_Dbg( &sys.demuxer, "Matroska DVD enter command" );
4598                 f_result |= interpretor.Interpret( p_data );
4599             }
4600         }
4601         index++;
4602     }
4603     return f_result;
4604 }
4605
4606 bool dvd_chapter_codec_c::Leave()
4607 {
4608     bool f_result = false;
4609     std::vector<KaxChapterProcessData>::iterator index = leave_cmds.begin();
4610     while ( index != leave_cmds.end() )
4611     {
4612         if ( (*index).GetSize() )
4613         {
4614             binary *p_data = (*index).GetBuffer();
4615             size_t i_size = *p_data++;
4616             // avoid reading too much from the buffer
4617             i_size = min( i_size, ((*index).GetSize() - 1) >> 3 );
4618             for ( ; i_size > 0; i_size--, p_data += 8 )
4619             {
4620                 msg_Dbg( &sys.demuxer, "Matroska DVD leave command" );
4621                 f_result |= interpretor.Interpret( p_data );
4622             }
4623         }
4624         index++;
4625     }
4626     return f_result;
4627 }
4628
4629 // see http://www.dvd-replica.com/DVD/vmcmdset.php for a description of DVD commands
4630 bool dvd_command_interpretor_c::Interpret( const binary * p_command, size_t i_size )
4631 {
4632     if ( i_size != 8 )
4633         return false;
4634
4635     virtual_segment_c *p_segment;
4636     chapter_item_c *p_chapter;
4637     bool f_result = false;
4638     uint16 i_command = ( p_command[0] << 8 ) + p_command[1];
4639
4640     // handle register tests if there are some
4641     if ( (i_command & 0xF0) != 0 )
4642     {
4643         bool b_test_positive = (i_command & CMD_DVD_IF_NOT) == 0;
4644         bool b_test_value    = (i_command & CMD_DVD_TEST_VALUE) != 0;
4645         uint8 i_test = i_command & 0x70;
4646         uint16 i_value;
4647
4648         if ( b_test_value )
4649             i_value = ( p_command[4] << 8 ) + p_command[5];
4650         else
4651             i_value = GetPRM( p_command[7] );
4652
4653         switch ( i_test )
4654         {
4655         case CMD_DVD_IF_GPREG_EQUAL:
4656             // if equals
4657             msg_Dbg( &sys.demuxer, "IF %s EQUALS %s", GetRegTypeName( false, p_command[3] ).c_str(), GetRegTypeName( b_test_value, i_value ).c_str() );
4658             if (!( GetPRM( p_command[3] ) == i_value ))
4659             {
4660                 b_test_positive = !b_test_positive;
4661             }
4662             break;
4663         case CMD_DVD_IF_GPREG_INF:
4664             // if inferior
4665             msg_Dbg( &sys.demuxer, "IF %s < %s", GetRegTypeName( false, p_command[3] ).c_str(), GetRegTypeName( b_test_value, i_value ).c_str() );
4666             if (!( GetPRM( p_command[3] ) < i_value ))
4667             {
4668                 b_test_positive = !b_test_positive;
4669             }
4670             break;
4671         case CMD_DVD_IF_GPREG_AND:
4672             // if logical and
4673             msg_Dbg( &sys.demuxer, "IF %s & %s", GetRegTypeName( false, p_command[3] ).c_str(), GetRegTypeName( b_test_value, i_value ).c_str() );
4674             if (!( GetPRM( p_command[3] ) & i_value ))
4675             {
4676                 b_test_positive = !b_test_positive;
4677             }
4678             break;
4679         case CMD_DVD_IF_GPREG_SUP_EQUAL:
4680             // if superior or equal
4681             msg_Dbg( &sys.demuxer, "IF %s >= %s", GetRegTypeName( false, p_command[3] ).c_str(), GetRegTypeName( b_test_value, i_value ).c_str() );
4682             if (!( GetPRM( p_command[3] ) >= i_value ))
4683             {
4684                 b_test_positive = !b_test_positive;
4685             }
4686             break;
4687         }
4688
4689         if ( !b_test_positive )
4690             return false;
4691     }
4692
4693     // strip the test command
4694     i_command &= 0xFF0F;
4695
4696     switch ( i_command )
4697     {
4698     case CMD_DVD_NOP:
4699     case CMD_DVD_NOP2:
4700         {
4701             msg_Dbg( &sys.demuxer, "NOP" );
4702             break;
4703         }
4704     case CMD_DVD_BREAK:
4705         {
4706             msg_Dbg( &sys.demuxer, "Break" );
4707             // TODO
4708             break;
4709         }
4710     case CMD_DVD_JUMP_TT:
4711         {
4712             uint8 i_title = p_command[5];
4713             msg_Dbg( &sys.demuxer, "JumpTT %d", i_title );
4714
4715             // find in the ChapProcessPrivate matching this Title level
4716             p_chapter = sys.BrowseCodecPrivate( 1, MatchTitleNumber, &i_title, sizeof(i_title), p_segment );
4717             if ( p_chapter != NULL )
4718             {
4719                 // if the segment is not part of the current segment, select the new one
4720                 if ( p_segment != sys.p_current_segment )
4721                 {
4722                     sys.PreparePlayback( p_segment );
4723                 }
4724
4725                 // jump to the location in the found segment
4726                 p_segment->Seek( sys.demuxer, p_chapter->i_user_start_time, -1, NULL );
4727                 p_chapter->Enter( true );
4728
4729                 f_result = true;
4730             }
4731
4732             break;
4733         }
4734     case CMD_DVD_CALLSS_VTSM1:
4735         {
4736             msg_Dbg( &sys.demuxer, "CallSS VTSM" );
4737             switch( (p_command[6] & 0xC0) >> 6 ) {
4738                 case 0:
4739                     switch ( p_command[5] )
4740                     {
4741                     case 0x00:
4742                         msg_Dbg( &sys.demuxer, "CallSS PGC (rsm_cell %x)", p_command[5]);
4743                         break;
4744                     case 0x82:
4745                         msg_Dbg( &sys.demuxer, "CallSS Title Entry (rsm_cell %x)", p_command[5]);
4746                         break;
4747                     case 0x83:
4748                         msg_Dbg( &sys.demuxer, "CallSS Root Menu (rsm_cell %x)", p_command[5]);
4749                         break;
4750                     case 0x84:
4751                         msg_Dbg( &sys.demuxer, "CallSS Subpicture Menu (rsm_cell %x)", p_command[5]);
4752                         break;
4753                     case 0x85:
4754                         msg_Dbg( &sys.demuxer, "CallSS Audio Menu (rsm_cell %x)", p_command[5]);
4755                         break;
4756                     case 0x86:
4757                         msg_Dbg( &sys.demuxer, "CallSS Angle Menu (rsm_cell %x)", p_command[5]);
4758                         break;
4759                     case 0x87:
4760                         msg_Dbg( &sys.demuxer, "CallSS Chapter Menu (rsm_cell %x)", p_command[5]);
4761                         break;
4762                     default:
4763                         msg_Dbg( &sys.demuxer, "CallSS <unknown> (rsm_cell %x)", p_command[5]);
4764                         break;
4765                     }
4766                     p_chapter = sys.BrowseCodecPrivate( 1, MatchPgcType, &p_command[5], 1, p_segment );
4767                     if ( p_chapter != NULL )
4768                     {
4769                         // if the segment is not part of the current segment, select the new one
4770                         if ( p_segment != sys.p_current_segment )
4771                         {
4772                             sys.PreparePlayback( p_segment );
4773                         }
4774
4775                         p_chapter->Enter( true );
4776
4777                         // jump to the location in the found segment
4778                         p_segment->Seek( sys.demuxer, p_chapter->i_user_start_time, -1, p_chapter );
4779                         f_result = true;
4780                     }
4781                 break;
4782                 case 1:
4783                     msg_Dbg( &sys.demuxer, "CallSS VMGM (menu %d, rsm_cell %x)", p_command[6] & 0x0F, p_command[5]);
4784                 break;
4785                 case 2:
4786                     msg_Dbg( &sys.demuxer, "CallSS VTSM (menu %d, rsm_cell %x)", p_command[6] & 0x0F, p_command[5]);
4787                 break;
4788                 case 3:
4789                     msg_Dbg( &sys.demuxer, "CallSS VMGM (pgc %d, rsm_cell %x)", (p_command[3] << 8) + p_command[4], p_command[5]);
4790                 break;
4791             }
4792             break;
4793         }
4794     case CMD_DVD_SET_GPRMMD:
4795         {
4796             msg_Dbg( &sys.demuxer, "Set GPRMMD [%d]=%d", (p_command[4] << 8) + p_command[5], (p_command[2] << 8) + p_command[3]);
4797
4798             if ( !SetGPRM( (p_command[4] << 8) + p_command[5], (p_command[2] << 8) + p_command[3] ) )
4799                 msg_Dbg( &sys.demuxer, "Set GPRMMD failed" );
4800             break;
4801         }
4802     case CMD_DVD_LINKPGCN:
4803         {
4804             uint16 i_pgcn = (p_command[6] << 8) + p_command[7];
4805
4806             msg_Dbg( &sys.demuxer, "Link PGCN(%d)", i_pgcn );
4807             // TODO
4808             break;
4809         }
4810     case CMD_DVD_LINKCN:
4811         {
4812             msg_Dbg( &sys.demuxer, "LinkCN (cell %d)", (p_command[6] << 8) + p_command[7] );
4813             // TODO
4814             f_result = true;
4815             break;
4816         }
4817     case CMD_DVD_GOTO_LINE:
4818         {
4819             msg_Dbg( &sys.demuxer, "GotoLine (%d)", (p_command[6] << 8) + p_command[7] );
4820             // TODO
4821             break;
4822         }
4823     default:
4824         {
4825             msg_Dbg( &sys.demuxer, "unsupported command : %02X %02X %02X %02X %02X %02X %02X %02X"
4826                      ,p_command[0]
4827                      ,p_command[1]
4828                      ,p_command[2]
4829                      ,p_command[3]
4830                      ,p_command[4]
4831                      ,p_command[5]
4832                      ,p_command[6]
4833                      ,p_command[7]);
4834             break;
4835         }
4836     }
4837
4838     return f_result;
4839 }
4840
4841 bool dvd_command_interpretor_c::MatchTitleNumber( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size )
4842 {
4843     if ( i_cookie_size != 1 || data.m_private_data.GetSize() < 4 )
4844         return false;
4845
4846     if ( data.m_private_data.GetBuffer()[0] != MATROSKA_DVD_LEVEL_TT )
4847         return false;
4848
4849     uint16 i_gtitle = (data.m_private_data.GetBuffer()[1] << 8 ) + data.m_private_data.GetBuffer()[2];
4850     uint8 i_title = *(uint8*)p_cookie;
4851
4852     return (i_gtitle == i_title);
4853 }
4854
4855 bool dvd_command_interpretor_c::MatchPgcType( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size )
4856 {
4857     if ( i_cookie_size != 1 || data.m_private_data.GetSize() < 7 )
4858         return false;
4859
4860     if ( data.m_private_data.GetBuffer()[0] != MATROSKA_DVD_LEVEL_PGC )
4861         return false;
4862
4863     uint8 i_pgc_type = data.m_private_data.GetBuffer()[3];
4864     uint8 i_pgc = *(uint8*)p_cookie;
4865
4866     return (i_pgc_type == i_pgc);
4867 }
4868
4869 bool matroska_script_codec_c::Enter()
4870 {
4871     bool f_result = false;
4872     std::vector<KaxChapterProcessData>::iterator index = enter_cmds.begin();
4873     while ( index != enter_cmds.end() )
4874     {
4875         if ( (*index).GetSize() )
4876         {
4877             msg_Dbg( &sys.demuxer, "Matroska Script enter command" );
4878             f_result |= interpretor.Interpret( (*index).GetBuffer(), (*index).GetSize() );
4879         }
4880         index++;
4881     }
4882     return f_result;
4883 }
4884
4885 bool matroska_script_codec_c::Leave()
4886 {
4887     bool f_result = false;
4888     std::vector<KaxChapterProcessData>::iterator index = leave_cmds.begin();
4889     while ( index != leave_cmds.end() )
4890     {
4891         if ( (*index).GetSize() )
4892         {
4893             msg_Dbg( &sys.demuxer, "Matroska Script leave command" );
4894             f_result |= interpretor.Interpret( (*index).GetBuffer(), (*index).GetSize() );
4895         }
4896         index++;
4897     }
4898     return f_result;
4899 }
4900
4901 // see http://www.matroska.org/technical/specs/chapters/index.html#mscript
4902 //  for a description of existing commands
4903 bool matroska_script_interpretor_c::Interpret( const binary * p_command, size_t i_size )
4904 {
4905     bool b_result = false;
4906
4907     char *psz_str = (char*) malloc( i_size + 1 );
4908     memcpy( psz_str, p_command, i_size );
4909     psz_str[ i_size ] = '\0';
4910
4911     std::string sz_command = psz_str;
4912     free( psz_str );
4913
4914     msg_Dbg( &sys.demuxer, "command : %s", sz_command.c_str() );
4915
4916     if ( sz_command.compare( 0, CMD_MS_GOTO_AND_PLAY.size(), CMD_MS_GOTO_AND_PLAY ) == 0 )
4917     {
4918         size_t i,j;
4919
4920         // find the (
4921         for ( i=CMD_MS_GOTO_AND_PLAY.size(); i<sz_command.size(); i++)
4922         {
4923             if ( sz_command[i] == '(' )
4924             {
4925                 i++;
4926                 break;
4927             }
4928         }
4929         // find the )
4930         for ( j=i; j<sz_command.size(); j++)
4931         {
4932             if ( sz_command[j] == ')' )
4933             {
4934                 i--;
4935                 break;
4936             }
4937         }
4938
4939         std::string st = sz_command.substr( i+1, j-i-1 );
4940         int64_t i_chapter_uid = atoi( st.c_str() );
4941
4942         virtual_segment_c *p_segment;
4943         chapter_item_c *p_chapter = sys.FindChapter( i_chapter_uid, p_segment );
4944
4945         if ( p_chapter == NULL )
4946             msg_Dbg( &sys.demuxer, "Chapter "I64Fd" not found", i_chapter_uid);
4947         else
4948         {
4949             p_segment->Seek( sys.demuxer, p_chapter->i_user_start_time, -1, p_chapter );
4950             b_result = true;
4951         }
4952     }
4953
4954     return b_result;
4955 }