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

   1 /*****************************************************************************
   2  * mkv.cpp : matroska demuxer
   3  *****************************************************************************
   4  * Copyright (C) 2003-2010 the VideoLAN team
   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., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  23  *****************************************************************************/
  24
  25 #include "matroska_segment.hpp"
  26
  27 #include "chapters.hpp"
  28
  29 #include "demux.hpp"
  30
  31 extern "C" {
  32 #include "../vobsub.h"
  33 }
  34
  35 #include <vlc_codecs.h>
  36
  37 /* GetFourCC helper */
  38 #define GetFOURCC( p )  __GetFOURCC( (uint8_t*)p )
  39 static vlc_fourcc_t __GetFOURCC( uint8_t *p )
  40 {
  41     return VLC_FOURCC( p[0], p[1], p[2], p[3] );
  42 }
  43
  44 /* Destructor */
  45 matroska_segment_c::~matroska_segment_c()
  46 {
  47     for( size_t i_track = 0; i_track < tracks.size(); i_track++ )
  48     {
  49         delete tracks[i_track]->p_compression_data;
  50         es_format_Clean( &tracks[i_track]->fmt );
  51         free( tracks[i_track]->p_extra_data );
  52         free( tracks[i_track]->psz_codec );
  53         delete tracks[i_track];
  54     }
  55
  56     free( psz_writing_application );
  57     free( psz_muxing_application );
  58     free( psz_segment_filename );
  59     free( psz_title );
  60     free( psz_date_utc );
  61     free( p_indexes );
  62
  63     delete ep;
  64     delete segment;
  65     delete p_segment_uid;
  66     delete p_prev_segment_uid;
  67     delete p_next_segment_uid;
  68
  69     std::vector<chapter_edition_c*>::iterator index = stored_editions.begin();
  70     while ( index != stored_editions.end() )
  71     {
  72         delete (*index);
  73         index++;
  74     }
  75     std::vector<chapter_translation_c*>::iterator indext = translations.begin();
  76     while ( indext != translations.end() )
  77     {
  78         delete (*indext);
  79         indext++;
  80     }
  81     std::vector<KaxSegmentFamily*>::iterator indexf = families.begin();
  82     while ( indexf != families.end() )
  83     {
  84         delete (*indexf);
  85         indexf++;
  86    }
  87 }
  88
  89
  90 /*****************************************************************************
  91  * Tools                                                                     *
  92  *****************************************************************************
  93  *  * LoadCues : load the cues element and update index
  94  *  * LoadTags : load ... the tags element
  95  *  * InformationCreate : create all information, load tags if present
  96  *****************************************************************************/
  97 void matroska_segment_c::LoadCues( KaxCues *cues )
  98 {
  99     EbmlParser  *ep;
 100     EbmlElement *el;
 101
 102     if( b_cues )
 103     {
 104         msg_Err( &sys.demuxer, "There can be only 1 Cues per section." );
 105         return;
 106     }
 107
 108     ep = new EbmlParser( &es, cues, &sys.demuxer );
 109     while( ( el = ep->Get() ) != NULL )
 110     {
 111         if( MKV_IS_ID( el, KaxCuePoint ) )
 112         {
 113 #define idx p_indexes[i_index]
 114
 115             idx.i_track       = -1;
 116             idx.i_block_number= -1;
 117             idx.i_position    = -1;
 118             idx.i_time        = 0;
 119             idx.b_key         = true;
 120
 121             ep->Down();
 122             while( ( el = ep->Get() ) != NULL )
 123             {
 124                 if( MKV_IS_ID( el, KaxCueTime ) )
 125                 {
 126                     KaxCueTime &ctime = *(KaxCueTime*)el;
 127
 128                     ctime.ReadData( es.I_O() );
 129
 130                     idx.i_time = uint64( ctime ) * i_timescale / (mtime_t)1000;
 131                 }
 132                 else if( MKV_IS_ID( el, KaxCueTrackPositions ) )
 133                 {
 134                     ep->Down();
 135                     while( ( el = ep->Get() ) != NULL )
 136                     {
 137                         if( MKV_IS_ID( el, KaxCueTrack ) )
 138                         {
 139                             KaxCueTrack &ctrack = *(KaxCueTrack*)el;
 140
 141                             ctrack.ReadData( es.I_O() );
 142                             idx.i_track = uint16( ctrack );
 143                         }
 144                         else if( MKV_IS_ID( el, KaxCueClusterPosition ) )
 145                         {
 146                             KaxCueClusterPosition &ccpos = *(KaxCueClusterPosition*)el;
 147
 148                             ccpos.ReadData( es.I_O() );
 149                             idx.i_position = segment->GetGlobalPosition( uint64( ccpos ) );
 150                         }
 151                         else if( MKV_IS_ID( el, KaxCueBlockNumber ) )
 152                         {
 153                             KaxCueBlockNumber &cbnum = *(KaxCueBlockNumber*)el;
 154
 155                             cbnum.ReadData( es.I_O() );
 156                             idx.i_block_number = uint32( cbnum );
 157                         }
 158                         else
 159                         {
 160                             msg_Dbg( &sys.demuxer, "         * Unknown (%s)", typeid(*el).name() );
 161                         }
 162                     }
 163                     ep->Up();
 164                 }
 165                 else
 166                 {
 167                     msg_Dbg( &sys.demuxer, "     * Unknown (%s)", typeid(*el).name() );
 168                 }
 169             }
 170             ep->Up();
 171
 172 #if 0
 173             msg_Dbg( &sys.demuxer, " * added time=%"PRId64" pos=%"PRId64
 174                      " track=%d bnum=%d", idx.i_time, idx.i_position,
 175                      idx.i_track, idx.i_block_number );
 176 #endif
 177
 178             i_index++;
 179             if( i_index >= i_index_max )
 180             {
 181                 i_index_max += 1024;
 182                 p_indexes = (mkv_index_t*)xrealloc( p_indexes,
 183                                         sizeof( mkv_index_t ) * i_index_max );
 184             }
 185 #undef idx
 186         }
 187         else
 188         {
 189             msg_Dbg( &sys.demuxer, " * Unknown (%s)", typeid(*el).name() );
 190         }
 191     }
 192     delete ep;
 193     b_cues = true;
 194     msg_Dbg( &sys.demuxer, "|   - loading cues done." );
 195 }
 196
 197
 198 #define PARSE_TAG( type ) \
 199     do { \
 200     msg_Dbg( &sys.demuxer, "|   + " type ); \
 201     ep->Down();                             \
 202     while( ( el = ep->Get() ) != NULL )     \
 203     {                                       \
 204         msg_Dbg( &sys.demuxer, "|   |   + Unknown (%s)", typeid( *el ).name() ); \
 205     }                                      \
 206     ep->Up(); } while( 0 )
 207
 208 static const struct {
 209     vlc_meta_type_t type;
 210     const char *key;
 211 } metadata_map[] = { {vlc_meta_Title,       "TITLE"},
 212                      {vlc_meta_Artist,      "ARTIST"},
 213                      {vlc_meta_Genre,       "GENRE"},
 214                      {vlc_meta_Copyright,   "COPYRIGHT"},
 215                      {vlc_meta_Description, "DESCRIPTION"},
 216                      {vlc_meta_Publisher,   "PUBLISHER"},
 217                      {vlc_meta_URL,         "URL"},
 218                      {vlc_meta_Title,       NULL},
 219 };
 220
 221 void matroska_segment_c::ParseSimpleTags( KaxTagSimple *tag )
 222 {
 223     EbmlElement *el;
 224     EbmlParser *ep = new EbmlParser( &es, tag, &sys.demuxer );
 225     char *k = NULL, *v = NULL;
 226
 227     if( !sys.meta )
 228         sys.meta = vlc_meta_New();
 229
 230     msg_Dbg( &sys.demuxer, "|   + Simple Tag ");
 231     while( ( el = ep->Get() ) != NULL )
 232     {
 233         if( MKV_IS_ID( el, KaxTagName ) )
 234         {
 235             KaxTagName &key = *(KaxTagName*)el;
 236             key.ReadData( es.I_O(), SCOPE_ALL_DATA );
 237             k = strdup( UTFstring( key ).GetUTF8().c_str() );
 238         }
 239         if( MKV_IS_ID( el, KaxTagString ) )
 240         {
 241             KaxTagString &value = *(KaxTagString*)el;
 242             value.ReadData( es.I_O(), SCOPE_ALL_DATA );
 243             v = strdup( UTFstring( value ).GetUTF8().c_str() );
 244         }
 245     }
 246     delete ep;
 247
 248     if( !k || !v )
 249     {
 250         msg_Warn( &sys.demuxer, "Invalid MKV SimpleTag found.");
 251         return;
 252     }
 253
 254     for( int i = 0; metadata_map[i].key; i++ )
 255     {
 256         if( !strcmp( k, metadata_map[i].key ) )
 257         {
 258             vlc_meta_Set( sys.meta, metadata_map[i].type, v );
 259             goto done;
 260         }
 261     }
 262     vlc_meta_AddExtra( sys.meta, k, v );
 263 done:
 264     free( k );
 265     free( v );
 266     return;
 267 }
 268
 269 void matroska_segment_c::LoadTags( KaxTags *tags )
 270 {
 271     /* Master elements */
 272     EbmlParser *ep = new EbmlParser( &es, tags, &sys.demuxer );
 273     EbmlElement *el;
 274
 275     while( ( el = ep->Get() ) != NULL )
 276     {
 277         if( MKV_IS_ID( el, KaxTag ) )
 278         {
 279             msg_Dbg( &sys.demuxer, "+ Tag" );
 280             ep->Down();
 281             while( ( el = ep->Get() ) != NULL )
 282             {
 283                 if( MKV_IS_ID( el, KaxTagTargets ) )
 284                     PARSE_TAG( "Targets" );
 285                 else if( MKV_IS_ID( el, KaxTagGeneral ) )
 286                     PARSE_TAG( "General" );
 287                 else if( MKV_IS_ID( el, KaxTagGenres ) )
 288                     PARSE_TAG( "Genres" );
 289                 else if( MKV_IS_ID( el, KaxTagAudioSpecific ) )
 290                     PARSE_TAG( "Audio Specific" );
 291                 else if( MKV_IS_ID( el, KaxTagImageSpecific ) )
 292                     PARSE_TAG( "Images Specific" );
 293                 else if( MKV_IS_ID( el, KaxTagMultiComment ) )
 294                 {
 295                     msg_Dbg( &sys.demuxer, "|   + Multi Comment" );
 296                 }
 297                 else if( MKV_IS_ID( el, KaxTagMultiCommercial ) )
 298                 {
 299                     msg_Dbg( &sys.demuxer, "|   + Multi Commercial" );
 300                 }
 301                 else if( MKV_IS_ID( el, KaxTagMultiDate ) )
 302                 {
 303                     msg_Dbg( &sys.demuxer, "|   + Multi Date" );
 304                 }
 305                 else if( MKV_IS_ID( el, KaxTagMultiEntity ) )
 306                 {
 307                     msg_Dbg( &sys.demuxer, "|   + Multi Entity" );
 308                 }
 309                 else if( MKV_IS_ID( el, KaxTagMultiIdentifier ) )
 310                 {
 311                     msg_Dbg( &sys.demuxer, "|   + Multi Identifier" );
 312                 }
 313                 else if( MKV_IS_ID( el, KaxTagMultiLegal ) )
 314                 {
 315                     msg_Dbg( &sys.demuxer, "|   + Multi Legal" );
 316                 }
 317                 else if( MKV_IS_ID( el, KaxTagMultiTitle ) )
 318                 {
 319                     msg_Dbg( &sys.demuxer, "|   + Multi Title" );
 320                 }
 321                 else if( MKV_IS_ID( el, KaxTagSimple ) )
 322                     ParseSimpleTags( static_cast<KaxTagSimple*>( el ) );
 323                 else
 324                 {
 325                     msg_Dbg( &sys.demuxer, "|   + LoadTag Unknown (%s)", typeid( *el ).name() );
 326                 }
 327             }
 328             ep->Up();
 329         }
 330         else
 331         {
 332             msg_Dbg( &sys.demuxer, "+ Unknown (%s)", typeid( *el ).name() );
 333         }
 334     }
 335     delete ep;
 336
 337     msg_Dbg( &sys.demuxer, "loading tags done." );
 338 }
 339 #undef PARSE_TAG
 340
 341 /*****************************************************************************
 342  * InformationCreate:
 343  *****************************************************************************/
 344 void matroska_segment_c::InformationCreate( )
 345 {
 346 #if 0
 347     sys.meta = vlc_meta_New();
 348
 349     if( psz_title )
 350     {
 351         vlc_meta_SetTitle( sys.meta, psz_title );
 352     }
 353     if( psz_date_utc )
 354     {
 355         vlc_meta_SetDate( sys.meta, psz_date_utc );
 356     }
 357
 358     if( psz_segment_filename )
 359     {
 360         fprintf( stderr, "***** WARNING: Unhandled meta - Use custom\n" );
 361     }
 362     if( psz_muxing_application )
 363     {
 364         fprintf( stderr, "***** WARNING: Unhandled meta - Use custom\n" );
 365     }
 366     if( psz_writing_application )
 367     {
 368         fprintf( stderr, "***** WARNING: Unhandled meta - Use custom\n" );
 369     }
 370
 371     for( size_t i_track = 0; i_track < tracks.size(); i_track++ )
 372     {
 373 //        mkv_track_t *tk = tracks[i_track];
 374 //        vlc_meta_t *mtk = vlc_meta_New();
 375         fprintf( stderr, "***** WARNING: Unhandled child meta\n");
 376     }
 377 #endif
 378 #if 0
 379     if( i_tags_position >= 0 )
 380     {
 381         bool b_seekable;
 382
 383         stream_Control( sys.demuxer.s, STREAM_CAN_FASTSEEK, &b_seekable );
 384         if( b_seekable )
 385         {
 386             LoadTags( );
 387         }
 388     }
 389 #endif
 390 }
 391
 392
 393 /*****************************************************************************
 394  * Misc
 395  *****************************************************************************/
 396
 397 void matroska_segment_c::IndexAppendCluster( KaxCluster *cluster )
 398 {
 399 #define idx p_indexes[i_index]
 400     idx.i_track       = -1;
 401     idx.i_block_number= -1;
 402     idx.i_position    = cluster->GetElementPosition();
 403     idx.i_time        = -1;
 404     idx.b_key         = true;
 405
 406     i_index++;
 407     if( i_index >= i_index_max )
 408     {
 409         i_index_max += 1024;
 410         p_indexes = (mkv_index_t*)xrealloc( p_indexes,
 411                                         sizeof( mkv_index_t ) * i_index_max );
 412     }
 413 #undef idx
 414 }
 415
 416 bool matroska_segment_c::PreloadFamily( const matroska_segment_c & of_segment )
 417 {
 418     if ( b_preloaded )
 419         return false;
 420
 421     for (size_t i=0; i<families.size(); i++)
 422     {
 423         for (size_t j=0; j<of_segment.families.size(); j++)
 424         {
 425             if ( *(families[i]) == *(of_segment.families[j]) )
 426                 return Preload( );
 427         }
 428     }
 429
 430     return false;
 431 }
 432
 433 bool matroska_segment_c::CompareSegmentUIDs( const matroska_segment_c * p_item_a, const matroska_segment_c * p_item_b )
 434 {
 435     EbmlBinary *p_tmp;
 436
 437     if ( p_item_a == NULL || p_item_b == NULL )
 438         return false;
 439
 440     p_tmp = (EbmlBinary *)p_item_a->p_segment_uid;
 441     if ( p_item_b->p_prev_segment_uid != NULL
 442           && *p_tmp == *p_item_b->p_prev_segment_uid )
 443         return true;
 444
 445     p_tmp = (EbmlBinary *)p_item_a->p_next_segment_uid;
 446     if ( !p_tmp )
 447         return false;
 448
 449     if ( p_item_b->p_segment_uid != NULL
 450           && *p_tmp == *p_item_b->p_segment_uid )
 451         return true;
 452
 453     if ( p_item_b->p_prev_segment_uid != NULL
 454           && *p_tmp == *p_item_b->p_prev_segment_uid )
 455         return true;
 456
 457     return false;
 458 }
 459
 460 bool matroska_segment_c::Preload( )
 461 {
 462     if ( b_preloaded )
 463         return false;
 464
 465     EbmlElement *el = NULL;
 466
 467     ep->Reset( &sys.demuxer );
 468
 469     while( ( el = ep->Get() ) != NULL )
 470     {
 471         if( MKV_IS_ID( el, KaxSeekHead ) )
 472         {
 473             /* Multiple allowed */
 474             /* We bail at 10, to prevent possible recursion */
 475             msg_Dbg(  &sys.demuxer, "|   + Seek head" );
 476             if( i_seekhead_count < 10 )
 477             {
 478                 i_seekhead_position = (int64_t) es.I_O().getFilePointer();
 479                 ParseSeekHead( static_cast<KaxSeekHead*>( el ) );
 480             }
 481         }
 482         else if( MKV_IS_ID( el, KaxInfo ) )
 483         {
 484             /* Multiple allowed, mandatory */
 485             msg_Dbg(  &sys.demuxer, "|   + Information" );
 486             if( i_info_position < 0 ) // FIXME
 487                 ParseInfo( static_cast<KaxInfo*>( el ) );
 488             i_info_position = (int64_t) es.I_O().getFilePointer();
 489         }
 490         else if( MKV_IS_ID( el, KaxTracks ) )
 491         {
 492             /* Multiple allowed */
 493             msg_Dbg(  &sys.demuxer, "|   + Tracks" );
 494             if( i_tracks_position < 0 ) // FIXME
 495                 ParseTracks( static_cast<KaxTracks*>( el ) );
 496             if ( tracks.size() == 0 )
 497             {
 498                 msg_Err( &sys.demuxer, "No tracks supported" );
 499                 return false;
 500             }
 501             i_tracks_position = (int64_t) es.I_O().getFilePointer();
 502         }
 503         else if( MKV_IS_ID( el, KaxCues ) )
 504         {
 505             msg_Dbg(  &sys.demuxer, "|   + Cues" );
 506             if( i_cues_position < 0 )
 507                 LoadCues( static_cast<KaxCues*>( el ) );
 508             i_cues_position = (int64_t) es.I_O().getFilePointer();
 509         }
 510         else if( MKV_IS_ID( el, KaxCluster ) )
 511         {
 512             msg_Dbg( &sys.demuxer, "|   + Cluster" );
 513
 514             cluster = (KaxCluster*)el;
 515
 516             i_cluster_pos = i_start_pos = cluster->GetElementPosition();
 517             ParseCluster( );
 518
 519             ep->Down();
 520             /* stop pre-parsing the stream */
 521             break;
 522         }
 523         else if( MKV_IS_ID( el, KaxAttachments ) )
 524         {
 525             msg_Dbg( &sys.demuxer, "|   + Attachments" );
 526             if( i_attachments_position < 0 )
 527                 ParseAttachments( static_cast<KaxAttachments*>( el ) );
 528             i_attachments_position = (int64_t) es.I_O().getFilePointer();
 529         }
 530         else if( MKV_IS_ID( el, KaxChapters ) )
 531         {
 532             msg_Dbg( &sys.demuxer, "|   + Chapters" );
 533             if( i_chapters_position < 0 )
 534                 ParseChapters( static_cast<KaxChapters*>( el ) );
 535             i_chapters_position = (int64_t) es.I_O().getFilePointer();
 536         }
 537         else if( MKV_IS_ID( el, KaxTag ) )
 538         {
 539             msg_Dbg( &sys.demuxer, "|   + Tags" );
 540             if( i_tags_position < 0) // FIXME
 541                 ;//LoadTags( static_cast<KaxTags*>( el ) );
 542             i_tags_position = (int64_t) es.I_O().getFilePointer();
 543         }
 544         else if( MKV_IS_ID( el, EbmlVoid ) )
 545             msg_Dbg( &sys.demuxer, "|   + Void" );
 546         else
 547             msg_Dbg( &sys.demuxer, "|   + Preload Unknown (%s)", typeid(*el).name() );
 548     }
 549
 550     b_preloaded = true;
 551
 552     return true;
 553 }
 554
 555 /* Here we try to load elements that were found in Seek Heads, but not yet parsed */
 556 bool matroska_segment_c::LoadSeekHeadItem( const EbmlCallbacks & ClassInfos, int64_t i_element_position )
 557 {
 558     int64_t     i_sav_position = (int64_t)es.I_O().getFilePointer();
 559     EbmlElement *el;
 560
 561     es.I_O().setFilePointer( i_element_position, seek_beginning );
 562     el = es.FindNextID( ClassInfos, 0xFFFFFFFFL);
 563
 564     if( el == NULL )
 565     {
 566         msg_Err( &sys.demuxer, "cannot load some cues/chapters/tags etc. (broken seekhead or file)" );
 567         es.I_O().setFilePointer( i_sav_position, seek_beginning );
 568         return false;
 569     }
 570
 571     if( MKV_IS_ID( el, KaxSeekHead ) )
 572     {
 573         /* Multiple allowed */
 574         msg_Dbg( &sys.demuxer, "|   + Seek head" );
 575         if( i_seekhead_count < 10 )
 576         {
 577             i_seekhead_position = i_element_position;
 578             ParseSeekHead( static_cast<KaxSeekHead*>( el ) );
 579         }
 580     }
 581     else if( MKV_IS_ID( el, KaxInfo ) ) // FIXME
 582     {
 583         /* Multiple allowed, mandatory */
 584         msg_Dbg( &sys.demuxer, "|   + Information" );
 585         if( i_info_position < 0 )
 586             ParseInfo( static_cast<KaxInfo*>( el ) );
 587         i_info_position = i_element_position;
 588     }
 589     else if( MKV_IS_ID( el, KaxTracks ) ) // FIXME
 590     {
 591         /* Multiple allowed */
 592         msg_Dbg( &sys.demuxer, "|   + Tracks" );
 593         if( i_tracks_position < 0 )
 594             ParseTracks( static_cast<KaxTracks*>( el ) );
 595         if ( tracks.size() == 0 )
 596         {
 597             msg_Err( &sys.demuxer, "No tracks supported" );
 598             delete el;
 599             es.I_O().setFilePointer( i_sav_position, seek_beginning );
 600             return false;
 601         }
 602         i_tracks_position = i_element_position;
 603     }
 604     else if( MKV_IS_ID( el, KaxCues ) )
 605     {
 606         msg_Dbg( &sys.demuxer, "|   + Cues" );
 607         if( i_cues_position < 0 )
 608             LoadCues( static_cast<KaxCues*>( el ) );
 609         i_cues_position = i_element_position;
 610     }
 611     else if( MKV_IS_ID( el, KaxAttachments ) )
 612     {
 613         msg_Dbg( &sys.demuxer, "|   + Attachments" );
 614         if( i_attachments_position < 0 )
 615             ParseAttachments( static_cast<KaxAttachments*>( el ) );
 616         i_attachments_position = i_element_position;
 617     }
 618     else if( MKV_IS_ID( el, KaxChapters ) )
 619     {
 620         msg_Dbg( &sys.demuxer, "|   + Chapters" );
 621         if( i_chapters_position < 0 )
 622             ParseChapters( static_cast<KaxChapters*>( el ) );
 623         i_chapters_position = i_element_position;
 624     }
 625     else if( MKV_IS_ID( el, KaxTags ) )
 626     {
 627         msg_Dbg( &sys.demuxer, "|   + Tags" );
 628         if( i_tags_position < 0 )
 629             LoadTags( static_cast<KaxTags*>( el ) );
 630         i_tags_position = i_element_position;
 631     }
 632     else
 633     {
 634         msg_Dbg( &sys.demuxer, "|   + LoadSeekHeadItem Unknown (%s)", typeid(*el).name() );
 635     }
 636     delete el;
 637
 638     es.I_O().setFilePointer( i_sav_position, seek_beginning );
 639     return true;
 640 }
 641
 642 void matroska_segment_c::Seek( mtime_t i_date, mtime_t i_time_offset, int64_t i_global_position )
 643 {
 644     KaxBlock    *block;
 645     KaxSimpleBlock *simpleblock;
 646     int         i_track_skipping;
 647     int64_t     i_block_duration;
 648     size_t      i_track;
 649     int64_t     i_seek_position = i_start_pos;
 650     int64_t     i_seek_time = i_start_time;
 651
 652     if( i_global_position >= 0 )
 653     {
 654         /* Special case for seeking in files with no cues */
 655         EbmlElement *el = NULL;
 656         es.I_O().setFilePointer( i_start_pos, seek_beginning );
 657         delete ep;
 658         ep = new EbmlParser( &es, segment, &sys.demuxer );
 659         cluster = NULL;
 660
 661         while( ( el = ep->Get() ) != NULL )
 662         {
 663             if( MKV_IS_ID( el, KaxCluster ) )
 664             {
 665                 cluster = (KaxCluster *)el;
 666                 i_cluster_pos = cluster->GetElementPosition();
 667                 if( i_index == 0 ||
 668                         ( i_index > 0 && p_indexes[i_index - 1].i_position < (int64_t)cluster->GetElementPosition() ) )
 669                 {
 670                     IndexAppendCluster( cluster );
 671                 }
 672                 if( es.I_O().getFilePointer() >= i_global_position )
 673                 {
 674                     ParseCluster();
 675                     msg_Dbg( &sys.demuxer, "we found a cluster that is in the neighbourhood" );
 676                     return;
 677                 }
 678             }
 679         }
 680         msg_Err( &sys.demuxer, "This file has no cues, and we were unable to seek to the requested position by parsing." );
 681         return;
 682     }
 683
 684     if ( i_index > 0 )
 685     {
 686         int i_idx = 0;
 687
 688         for( ; i_idx < i_index; i_idx++ )
 689         {
 690             if( p_indexes[i_idx].i_time + i_time_offset > i_date )
 691             {
 692                 break;
 693             }
 694         }
 695
 696         if( i_idx > 0 )
 697         {
 698             i_idx--;
 699         }
 700
 701         i_seek_position = p_indexes[i_idx].i_position;
 702         i_seek_time = p_indexes[i_idx].i_time;
 703     }
 704
 705     msg_Dbg( &sys.demuxer, "seek got %"PRId64" (%d%%)",
 706                 i_seek_time, (int)( 100 * i_seek_position / stream_Size( sys.demuxer.s ) ) );
 707
 708     es.I_O().setFilePointer( i_seek_position, seek_beginning );
 709
 710     delete ep;
 711     ep = new EbmlParser( &es, segment, &sys.demuxer );
 712     cluster = NULL;
 713
 714     sys.i_start_pts = i_date;
 715
 716     /* now parse until key frame */
 717     i_track_skipping = 0;
 718     for( i_track = 0; i_track < tracks.size(); i_track++ )
 719     {
 720         if( tracks[i_track]->fmt.i_cat == VIDEO_ES )
 721         {
 722             tracks[i_track]->b_search_keyframe = true;
 723             i_track_skipping++;
 724         }
 725     }
 726     es_out_Control( sys.demuxer.out, ES_OUT_SET_NEXT_DISPLAY_TIME, i_date );
 727
 728     while( i_track_skipping > 0 )
 729     {
 730         bool b_key_picture;
 731         bool b_discardable_picture;
 732         if( BlockGet( block, simpleblock, &b_key_picture, &b_discardable_picture, &i_block_duration ) )
 733         {
 734             msg_Warn( &sys.demuxer, "cannot get block EOF?" );
 735
 736             return;
 737         }
 738
 739         for( i_track = 0; i_track < tracks.size(); i_track++ )
 740         {
 741             if( (simpleblock && tracks[i_track]->i_number == simpleblock->TrackNum()) ||
 742                 (block && tracks[i_track]->i_number == block->TrackNum()) )
 743             {
 744                 break;
 745             }
 746         }
 747
 748         if( simpleblock )
 749             sys.i_pts = (sys.i_chapter_time + simpleblock->GlobalTimecode()) / (mtime_t) 1000;
 750         else
 751             sys.i_pts = (sys.i_chapter_time + block->GlobalTimecode()) / (mtime_t) 1000;
 752
 753         if( i_track < tracks.size() )
 754         {
 755             if( sys.i_pts > sys.i_start_pts )
 756             {
 757                 cluster = static_cast<KaxCluster*>(ep->UnGet( i_block_pos, i_cluster_pos ));
 758                 i_track_skipping = 0;
 759             }
 760             else if( tracks[i_track]->fmt.i_cat == VIDEO_ES )
 761             {
 762                 if( b_key_picture && tracks[i_track]->b_search_keyframe )
 763                 {
 764                     tracks[i_track]->b_search_keyframe = false;
 765                     i_track_skipping--;
 766                 }
 767                 if( !tracks[i_track]->b_search_keyframe )
 768                 {
 769                     BlockDecode( &sys.demuxer, block, simpleblock, sys.i_pts, 0, b_key_picture || b_discardable_picture );
 770                 }
 771             }
 772         }
 773
 774         delete block;
 775     }
 776
 777     /* FIXME current ES_OUT_SET_NEXT_DISPLAY_TIME does not work that well if
 778      * the delay is too high. */
 779     if( sys.i_pts + 500*1000 < sys.i_start_pts )
 780     {
 781         sys.i_start_pts = sys.i_pts;
 782
 783         es_out_Control( sys.demuxer.out, ES_OUT_SET_NEXT_DISPLAY_TIME, sys.i_start_pts );
 784     }
 785 }
 786
 787 int matroska_segment_c::BlockFindTrackIndex( size_t *pi_track,
 788                                              const KaxBlock *p_block, const KaxSimpleBlock *p_simpleblock )
 789 {
 790     size_t i_track;
 791     for( i_track = 0; i_track < tracks.size(); i_track++ )
 792     {
 793         const mkv_track_t *tk = tracks[i_track];
 794
 795         if( ( p_block != NULL && tk->i_number == p_block->TrackNum() ) ||
 796             ( p_simpleblock != NULL && tk->i_number == p_simpleblock->TrackNum() ) )
 797         {
 798             break;
 799         }
 800     }
 801
 802     if( i_track >= tracks.size() )
 803         return VLC_EGENERIC;
 804
 805     if( pi_track )
 806         *pi_track = i_track;
 807     return VLC_SUCCESS;
 808 }
 809
 810 static inline void fill_extra_data( mkv_track_t *p_tk )
 811 {
 812       p_tk->fmt.i_extra = p_tk->i_extra_data;
 813       p_tk->fmt.p_extra = xmalloc( p_tk->i_extra_data );
 814       memcpy( p_tk->fmt.p_extra, p_tk->p_extra_data, p_tk->i_extra_data );
 815 }
 816
 817 bool matroska_segment_c::Select( mtime_t i_start_time )
 818 {
 819     /* add all es */
 820     msg_Dbg( &sys.demuxer, "found %d es", (int)tracks.size() );
 821     sys.b_pci_packet_set = false;
 822
 823     for( size_t i_track = 0; i_track < tracks.size(); i_track++ )
 824     {
 825         mkv_track_t *p_tk = tracks[i_track];
 826         es_format_t *p_fmt = &p_tk->fmt;
 827
 828         if( p_fmt->i_cat == UNKNOWN_ES || !p_tk->psz_codec )
 829         {
 830             msg_Warn( &sys.demuxer, "invalid track[%d, n=%d]", (int)i_track, p_tk->i_number );
 831             p_tk->p_es = NULL;
 832             continue;
 833         }
 834
 835         if( !strcmp( p_tk->psz_codec, "V_MS/VFW/FOURCC" ) )
 836         {
 837             if( p_tk->i_extra_data < (int)sizeof( BITMAPINFOHEADER ) )
 838             {
 839                 msg_Err( &sys.demuxer, "missing/invalid BITMAPINFOHEADER" );
 840                 p_tk->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
 841             }
 842             else
 843             {
 844                 BITMAPINFOHEADER *p_bih = (BITMAPINFOHEADER*)p_tk->p_extra_data;
 845
 846                 p_tk->fmt.video.i_width = GetDWLE( &p_bih->biWidth );
 847                 p_tk->fmt.video.i_height= GetDWLE( &p_bih->biHeight );
 848                 p_tk->fmt.i_codec       = GetFOURCC( &p_bih->biCompression );
 849
 850                 p_tk->fmt.i_extra       = GetDWLE( &p_bih->biSize ) - sizeof( BITMAPINFOHEADER );
 851                 if( p_tk->fmt.i_extra > 0 )
 852                 {
 853                     p_tk->fmt.p_extra = xmalloc( p_tk->fmt.i_extra );
 854                     memcpy( p_tk->fmt.p_extra, &p_bih[1], p_tk->fmt.i_extra );
 855                 }
 856             }
 857             p_tk->b_dts_only = true;
 858         }
 859         else if( !strcmp( p_tk->psz_codec, "V_MPEG1" ) ||
 860                  !strcmp( p_tk->psz_codec, "V_MPEG2" ) )
 861         {
 862             p_tk->fmt.i_codec = VLC_CODEC_MPGV;
 863             if( p_tk->i_extra_data )
 864                 fill_extra_data( p_tk );
 865         }
 866         else if( !strncmp( p_tk->psz_codec, "V_THEORA", 8 ) )
 867         {
 868             p_tk->fmt.i_codec = VLC_CODEC_THEORA;
 869             fill_extra_data( p_tk );
 870             p_tk->b_pts_only = true;
 871         }
 872         else if( !strncmp( p_tk->psz_codec, "V_REAL/RV", 9 ) )
 873         {
 874             if( !strcmp( p_tk->psz_codec, "V_REAL/RV10" ) )
 875                 p_fmt->i_codec = VLC_CODEC_RV10;
 876             else if( !strcmp( p_tk->psz_codec, "V_REAL/RV20" ) )
 877                 p_fmt->i_codec = VLC_CODEC_RV20;
 878             else if( !strcmp( p_tk->psz_codec, "V_REAL/RV30" ) )
 879                 p_fmt->i_codec = VLC_CODEC_RV30;
 880             else if( !strcmp( p_tk->psz_codec, "V_REAL/RV40" ) )
 881                 p_fmt->i_codec = VLC_CODEC_RV40;
 882
 883             if( p_tk->i_extra_data > 26 )
 884             {
 885                 p_fmt->p_extra = malloc( p_tk->i_extra_data - 26 );
 886                 if( p_fmt->p_extra )
 887                 {
 888                     p_fmt->i_extra = p_tk->i_extra_data - 26;
 889                     memcpy( p_fmt->p_extra, &p_tk->p_extra_data[26], p_fmt->i_extra );
 890                 }
 891             }
 892             p_tk->b_dts_only = true;
 893         }
 894         else if( !strncmp( p_tk->psz_codec, "V_DIRAC", 7 ) )
 895         {
 896             p_tk->fmt.i_codec = VLC_CODEC_DIRAC;
 897         }
 898         else if( !strncmp( p_tk->psz_codec, "V_VP8", 5 ) )
 899         {
 900             p_tk->fmt.i_codec = VLC_CODEC_VP8;
 901             p_tk->b_pts_only = true;
 902         }
 903         else if( !strncmp( p_tk->psz_codec, "V_MPEG4", 7 ) )
 904         {
 905             if( !strcmp( p_tk->psz_codec, "V_MPEG4/MS/V3" ) )
 906             {
 907                 p_tk->fmt.i_codec = VLC_CODEC_DIV3;
 908             }
 909             else if( !strncmp( p_tk->psz_codec, "V_MPEG4/ISO", 11 ) )
 910             {
 911                 /* A MPEG 4 codec, SP, ASP, AP or AVC */
 912                 if( !strcmp( p_tk->psz_codec, "V_MPEG4/ISO/AVC" ) )
 913                     p_tk->fmt.i_codec = VLC_FOURCC( 'a', 'v', 'c', '1' );
 914                 else
 915                     p_tk->fmt.i_codec = VLC_CODEC_MP4V;
 916                 fill_extra_data( p_tk );
 917             }
 918         }
 919         else if( !strcmp( p_tk->psz_codec, "V_QUICKTIME" ) )
 920         {
 921             MP4_Box_t *p_box = (MP4_Box_t*)xmalloc( sizeof( MP4_Box_t ) );
 922             stream_t *p_mp4_stream = stream_MemoryNew( VLC_OBJECT(&sys.demuxer),
 923                                                        p_tk->p_extra_data,
 924                                                        p_tk->i_extra_data,
 925                                                        true );
 926             if( MP4_ReadBoxCommon( p_mp4_stream, p_box ) &&
 927                 MP4_ReadBox_sample_vide( p_mp4_stream, p_box ) )
 928             {
 929                 p_tk->fmt.i_codec = p_box->i_type;
 930                 p_tk->fmt.video.i_width = p_box->data.p_sample_vide->i_width;
 931                 p_tk->fmt.video.i_height = p_box->data.p_sample_vide->i_height;
 932                 p_tk->fmt.i_extra = p_box->data.p_sample_vide->i_qt_image_description;
 933                 p_tk->fmt.p_extra = xmalloc( p_tk->fmt.i_extra );
 934                 memcpy( p_tk->fmt.p_extra, p_box->data.p_sample_vide->p_qt_image_description, p_tk->fmt.i_extra );
 935                 MP4_FreeBox_sample_vide( p_box );
 936             }
 937             else
 938             {
 939                 free( p_box );
 940             }
 941             stream_Delete( p_mp4_stream );
 942         }
 943         else if( !strcmp( p_tk->psz_codec, "V_MJPEG" ) )
 944         {
 945             p_tk->fmt.i_codec = VLC_CODEC_MJPG;
 946         }
 947         else if( !strcmp( p_tk->psz_codec, "A_MS/ACM" ) )
 948         {
 949             if( p_tk->i_extra_data < (int)sizeof( WAVEFORMATEX ) )
 950             {
 951                 msg_Err( &sys.demuxer, "missing/invalid WAVEFORMATEX" );
 952                 p_tk->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
 953             }
 954             else
 955             {
 956                 WAVEFORMATEX *p_wf = (WAVEFORMATEX*)p_tk->p_extra_data;
 957
 958                 wf_tag_to_fourcc( GetWLE( &p_wf->wFormatTag ), &p_tk->fmt.i_codec, NULL );
 959
 960                 p_tk->fmt.audio.i_channels   = GetWLE( &p_wf->nChannels );
 961                 p_tk->fmt.audio.i_rate = GetDWLE( &p_wf->nSamplesPerSec );
 962                 p_tk->fmt.i_bitrate    = GetDWLE( &p_wf->nAvgBytesPerSec ) * 8;
 963                 p_tk->fmt.audio.i_blockalign = GetWLE( &p_wf->nBlockAlign );;
 964                 p_tk->fmt.audio.i_bitspersample = GetWLE( &p_wf->wBitsPerSample );
 965
 966                 p_tk->fmt.i_extra            = GetWLE( &p_wf->cbSize );
 967                 if( p_tk->fmt.i_extra > 0 )
 968                 {
 969                     p_tk->fmt.p_extra = xmalloc( p_tk->fmt.i_extra );
 970                     memcpy( p_tk->fmt.p_extra, &p_wf[1], p_tk->fmt.i_extra );
 971                 }
 972             }
 973         }
 974         else if( !strcmp( p_tk->psz_codec, "A_MPEG/L3" ) ||
 975                  !strcmp( p_tk->psz_codec, "A_MPEG/L2" ) ||
 976                  !strcmp( p_tk->psz_codec, "A_MPEG/L1" ) )
 977         {
 978             p_tk->fmt.i_codec = VLC_CODEC_MPGA;
 979         }
 980         else if( !strcmp( p_tk->psz_codec, "A_AC3" ) )
 981         {
 982             p_tk->fmt.i_codec = VLC_CODEC_A52;
 983         }
 984         else if( !strcmp( p_tk->psz_codec, "A_EAC3" ) )
 985         {
 986             p_tk->fmt.i_codec = VLC_CODEC_EAC3;
 987         }
 988         else if( !strcmp( p_tk->psz_codec, "A_DTS" ) )
 989         {
 990             p_tk->fmt.i_codec = VLC_CODEC_DTS;
 991         }
 992         else if( !strcmp( p_tk->psz_codec, "A_MLP" ) )
 993         {
 994             p_tk->fmt.i_codec = VLC_CODEC_MLP;
 995         }
 996         else if( !strcmp( p_tk->psz_codec, "A_TRUEHD" ) )
 997         {
 998             /* FIXME when more samples arrive */
 999             p_tk->fmt.i_codec = VLC_CODEC_TRUEHD;
1000             p_fmt->b_packetized = false;
1001         }
1002         else if( !strcmp( p_tk->psz_codec, "A_FLAC" ) )
1003         {
1004             p_tk->fmt.i_codec = VLC_CODEC_FLAC;
1005             fill_extra_data( p_tk );
1006         }
1007         else if( !strcmp( p_tk->psz_codec, "A_VORBIS" ) )
1008         {
1009             p_tk->fmt.i_codec = VLC_CODEC_VORBIS;
1010             fill_extra_data( p_tk );
1011         }
1012         else if( !strncmp( p_tk->psz_codec, "A_AAC/MPEG2/", strlen( "A_AAC/MPEG2/" ) ) ||
1013                  !strncmp( p_tk->psz_codec, "A_AAC/MPEG4/", strlen( "A_AAC/MPEG4/" ) ) )
1014         {
1015             int i_profile, i_srate, sbr = 0;
1016             static const unsigned int i_sample_rates[] =
1017             {
1018                     96000, 88200, 64000, 48000, 44100, 32000, 24000, 22050,
1019                     16000, 12000, 11025,  8000,  7350,     0,     0,     0
1020             };
1021
1022             p_tk->fmt.i_codec = VLC_CODEC_MP4A;
1023             /* create data for faad (MP4DecSpecificDescrTag)*/
1024
1025             if( !strcmp( &p_tk->psz_codec[12], "MAIN" ) )
1026             {
1027                 i_profile = 0;
1028             }
1029             else if( !strcmp( &p_tk->psz_codec[12], "LC" ) )
1030             {
1031                 i_profile = 1;
1032             }
1033             else if( !strcmp( &p_tk->psz_codec[12], "SSR" ) )
1034             {
1035                 i_profile = 2;
1036             }
1037             else if( !strcmp( &p_tk->psz_codec[12], "LC/SBR" ) )
1038             {
1039                 i_profile = 1;
1040                 sbr = 1;
1041             }
1042             else
1043             {
1044                 i_profile = 3;
1045             }
1046
1047             for( i_srate = 0; i_srate < 13; i_srate++ )
1048             {
1049                 if( i_sample_rates[i_srate] == p_tk->i_original_rate )
1050                 {
1051                     break;
1052                 }
1053             }
1054             msg_Dbg( &sys.demuxer, "profile=%d srate=%d", i_profile, i_srate );
1055
1056             p_tk->fmt.i_extra = sbr ? 5 : 2;
1057             p_tk->fmt.p_extra = xmalloc( p_tk->fmt.i_extra );
1058             ((uint8_t*)p_tk->fmt.p_extra)[0] = ((i_profile + 1) << 3) | ((i_srate&0xe) >> 1);
1059             ((uint8_t*)p_tk->fmt.p_extra)[1] = ((i_srate & 0x1) << 7) | (p_tk->fmt.audio.i_channels << 3);
1060             if (sbr != 0)
1061             {
1062                 int syncExtensionType = 0x2B7;
1063                 int iDSRI;
1064                 for (iDSRI=0; iDSRI<13; iDSRI++)
1065                     if( i_sample_rates[iDSRI] == p_tk->fmt.audio.i_rate )
1066                         break;
1067                 ((uint8_t*)p_tk->fmt.p_extra)[2] = (syncExtensionType >> 3) & 0xFF;
1068                 ((uint8_t*)p_tk->fmt.p_extra)[3] = ((syncExtensionType & 0x7) << 5) | 5;
1069                 ((uint8_t*)p_tk->fmt.p_extra)[4] = ((1 & 0x1) << 7) | (iDSRI << 3);
1070             }
1071         }
1072         else if( !strcmp( p_tk->psz_codec, "A_AAC" ) )
1073         {
1074             p_tk->fmt.i_codec = VLC_CODEC_MP4A;
1075             fill_extra_data( p_tk );
1076         }
1077         else if( !strcmp( p_tk->psz_codec, "A_WAVPACK4" ) )
1078         {
1079             p_tk->fmt.i_codec = VLC_CODEC_WAVPACK;
1080             fill_extra_data( p_tk );
1081         }
1082         else if( !strcmp( p_tk->psz_codec, "A_TTA1" ) )
1083         {
1084             p_fmt->i_codec = VLC_CODEC_TTA;
1085             if( p_tk->i_extra_data > 0 )
1086             {
1087                 fill_extra_data( p_tk );
1088             }
1089             else
1090             {
1091                 p_fmt->i_extra = 30;
1092                 p_fmt->p_extra = xmalloc( p_fmt->i_extra );
1093                 uint8_t *p_extra = (uint8_t*)p_fmt->p_extra;
1094                 memcpy( &p_extra[ 0], "TTA1", 4 );
1095                 SetWLE( &p_extra[ 4], 1 );
1096                 SetWLE( &p_extra[ 6], p_fmt->audio.i_channels );
1097                 SetWLE( &p_extra[ 8], p_fmt->audio.i_bitspersample );
1098                 SetDWLE( &p_extra[10], p_fmt->audio.i_rate );
1099                 SetDWLE( &p_extra[14], 0xffffffff );
1100                 memset( &p_extra[18], 0, 30  - 18 );
1101             }
1102         }
1103         else if( !strcmp( p_tk->psz_codec, "A_PCM/INT/BIG" ) ||
1104                  !strcmp( p_tk->psz_codec, "A_PCM/INT/LIT" ) ||
1105                  !strcmp( p_tk->psz_codec, "A_PCM/FLOAT/IEEE" ) )
1106         {
1107             if( !strcmp( p_tk->psz_codec, "A_PCM/INT/BIG" ) )
1108             {
1109                 p_tk->fmt.i_codec = VLC_FOURCC( 't', 'w', 'o', 's' );
1110             }
1111             else
1112             {
1113                 p_tk->fmt.i_codec = VLC_FOURCC( 'a', 'r', 'a', 'w' );
1114             }
1115             p_tk->fmt.audio.i_blockalign = ( p_tk->fmt.audio.i_bitspersample + 7 ) / 8 * p_tk->fmt.audio.i_channels;
1116         }
1117         else if( !strncmp( p_tk->psz_codec, "A_REAL/", 7 ) )
1118         {
1119             if( !strcmp( p_tk->psz_codec, "A_REAL/COOK" ) )
1120                 p_tk->fmt.i_codec = VLC_CODEC_COOK;
1121             else if( !strcmp( p_tk->psz_codec, "A_REAL/ATRC" ) )
1122                 p_tk->fmt.i_codec = VLC_CODEC_ATRAC3;
1123             else if( !strcmp( p_tk->psz_codec, "A_REAL/28_8" ) )
1124                 p_tk->fmt.i_codec = VLC_CODEC_RA_288;
1125             /* FIXME 14_4, RALF and SIPR */
1126             fill_extra_data( p_tk );
1127         }
1128         else if( !strcmp( p_tk->psz_codec, "A_REAL/14_4" ) )
1129         {
1130             p_fmt->i_codec = VLC_CODEC_RA_144;
1131             p_fmt->audio.i_channels = 1;
1132             p_fmt->audio.i_rate = 8000;
1133             p_fmt->audio.i_blockalign = 0x14;
1134         }
1135         /* disabled due to the potential "S_KATE" namespace issue */
1136         else if( !strcmp( p_tk->psz_codec, "S_KATE" ) )
1137         {
1138             int i, i_offset = 1, i_extra, num_headers, size_so_far;
1139             uint8_t *p_extra;
1140
1141             p_tk->fmt.i_codec = VLC_CODEC_KATE;
1142             p_tk->fmt.subs.psz_encoding = strdup( "UTF-8" );
1143
1144             fill_extra_data( p_tk );
1145         }
1146         else if( !strcmp( p_tk->psz_codec, "S_TEXT/ASCII" ) )
1147         {
1148             p_fmt->i_codec = VLC_CODEC_SUBT;
1149             p_fmt->subs.psz_encoding = NULL; /* Is there a place where it is stored ? */
1150         }
1151         else if( !strcmp( p_tk->psz_codec, "S_TEXT/UTF8" ) )
1152         {
1153             p_tk->fmt.i_codec = VLC_CODEC_SUBT;
1154             p_tk->fmt.subs.psz_encoding = strdup( "UTF-8" );
1155         }
1156         else if( !strcmp( p_tk->psz_codec, "S_TEXT/USF" ) )
1157         {
1158             p_tk->fmt.i_codec = VLC_FOURCC( 'u', 's', 'f', ' ' );
1159             p_tk->fmt.subs.psz_encoding = strdup( "UTF-8" );
1160             if( p_tk->i_extra_data )
1161             {
1162                 fill_extra_data( p_tk );
1163             }
1164         }
1165         else if( !strcmp( p_tk->psz_codec, "S_TEXT/SSA" ) ||
1166                  !strcmp( p_tk->psz_codec, "S_TEXT/ASS" ) ||
1167                  !strcmp( p_tk->psz_codec, "S_SSA" ) ||
1168                  !strcmp( p_tk->psz_codec, "S_ASS" ))
1169         {
1170             p_tk->fmt.i_codec = VLC_CODEC_SSA;
1171             p_tk->fmt.subs.psz_encoding = strdup( "UTF-8" );
1172             if( p_tk->i_extra_data )
1173             {
1174                 fill_extra_data( p_tk );
1175             }
1176         }
1177         else if( !strcmp( p_tk->psz_codec, "S_VOBSUB" ) )
1178         {
1179             p_tk->fmt.i_codec = VLC_CODEC_SPU;
1180             if( p_tk->i_extra_data )
1181             {
1182                 char *psz_start;
1183                 char *psz_buf = (char *)malloc( p_tk->i_extra_data + 1);
1184                 if( psz_buf != NULL )
1185                 {
1186                     memcpy( psz_buf, p_tk->p_extra_data , p_tk->i_extra_data );
1187                     psz_buf[p_tk->i_extra_data] = '\0';
1188
1189                     psz_start = strstr( psz_buf, "size:" );
1190                     if( psz_start &&
1191                         vobsub_size_parse( psz_start,
1192                                            &p_tk->fmt.subs.spu.i_original_frame_width,
1193                                            &p_tk->fmt.subs.spu.i_original_frame_height ) == VLC_SUCCESS )
1194                     {
1195                         msg_Dbg( &sys.demuxer, "original frame size vobsubs: %dx%d",
1196                                  p_tk->fmt.subs.spu.i_original_frame_width,
1197                                  p_tk->fmt.subs.spu.i_original_frame_height );
1198                     }
1199                     else
1200                     {
1201                         msg_Warn( &sys.demuxer, "reading original frame size for vobsub failed" );
1202                     }
1203
1204                     psz_start = strstr( psz_buf, "palette:" );
1205                     if( psz_start &&
1206                         vobsub_palette_parse( psz_start, &p_tk->fmt.subs.spu.palette[1] ) == VLC_SUCCESS )
1207                     {
1208                         p_tk->fmt.subs.spu.palette[0] =  0xBeef;
1209                         msg_Dbg( &sys.demuxer, "vobsub palette read" );
1210                     }
1211                     else
1212                     {
1213                         msg_Warn( &sys.demuxer, "reading original palette failed" );
1214                     }
1215                     free( psz_buf );
1216                 }
1217             }
1218         }
1219         else if( !strcmp( p_tk->psz_codec, "S_HDMV/PGS" ) )
1220         {
1221             p_tk->fmt.i_codec = VLC_CODEC_BD_PG;
1222         }
1223         else if( !strcmp( p_tk->psz_codec, "B_VOBBTN" ) )
1224         {
1225             p_tk->fmt.i_cat = NAV_ES;
1226             continue;
1227         }
1228         else
1229         {
1230             msg_Err( &sys.demuxer, "unknown codec id=`%s'", p_tk->psz_codec );
1231             p_tk->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
1232         }
1233         if( p_tk->b_default )
1234         {
1235             p_tk->fmt.i_priority = 1000;
1236         }
1237
1238         p_tk->p_es = es_out_Add( sys.demuxer.out, &p_tk->fmt );
1239
1240         /* Turn on a subtitles track if it has been flagged as default -
1241          * but only do this if no subtitles track has already been engaged,
1242          * either by an earlier 'default track' (??) or by default
1243          * language choice behaviour.
1244          */
1245         if( p_tk->b_default )
1246         {
1247             es_out_Control( sys.demuxer.out,
1248                             ES_OUT_SET_ES_DEFAULT,
1249                             p_tk->p_es );
1250         }
1251     }
1252     es_out_Control( sys.demuxer.out, ES_OUT_SET_NEXT_DISPLAY_TIME, i_start_time );
1253
1254     sys.i_start_pts = i_start_time;
1255     // reset the stream reading to the first cluster of the segment used
1256     es.I_O().setFilePointer( i_start_pos );
1257
1258     delete ep;
1259     ep = new EbmlParser( &es, segment, &sys.demuxer );
1260
1261     return true;
1262 }
1263
1264 void matroska_segment_c::UnSelect( )
1265 {
1266     for( size_t i_track = 0; i_track < tracks.size(); i_track++ )
1267     {
1268         if ( tracks[i_track]->p_es != NULL )
1269         {
1270 //            es_format_Clean( &tracks[i_track]->fmt );
1271             es_out_Del( sys.demuxer.out, tracks[i_track]->p_es );
1272             tracks[i_track]->p_es = NULL;
1273         }
1274     }
1275     delete ep;
1276     ep = NULL;
1277 }
1278
1279 int matroska_segment_c::BlockGet( KaxBlock * & pp_block, KaxSimpleBlock * & pp_simpleblock, bool *pb_key_picture, bool *pb_discardable_picture, int64_t *pi_duration )
1280 {
1281     pp_simpleblock = NULL;
1282     pp_block = NULL;
1283
1284     *pb_key_picture         = true;
1285     *pb_discardable_picture = false;
1286
1287     for( ;; )
1288     {
1289         EbmlElement *el = NULL;
1290         int         i_level;
1291
1292         if ( ep == NULL )
1293             return VLC_EGENERIC;
1294
1295         if( pp_simpleblock != NULL || ((el = ep->Get()) == NULL && pp_block != NULL) )
1296         {
1297             /* Check blocks validity to protect againts broken files */
1298             if( BlockFindTrackIndex( NULL, pp_block , pp_simpleblock ) )
1299             {
1300                 delete pp_block;
1301                 pp_simpleblock = NULL;
1302                 pp_block = NULL;
1303                 continue;
1304             }
1305             if( pp_simpleblock != NULL )
1306             {
1307                 *pb_key_picture         = pp_simpleblock->IsKeyframe();
1308                 *pb_discardable_picture = pp_simpleblock->IsDiscardable();
1309             }
1310
1311             /* update the index */
1312 #define idx p_indexes[i_index - 1]
1313             if( i_index > 0 && idx.i_time == -1 )
1314             {
1315                 if ( pp_simpleblock != NULL )
1316                     idx.i_time        = pp_simpleblock->GlobalTimecode() / (mtime_t)1000;
1317                 else
1318                     idx.i_time        = (*pp_block).GlobalTimecode() / (mtime_t)1000;
1319                 idx.b_key         = *pb_key_picture;
1320             }
1321 #undef idx
1322             return VLC_SUCCESS;
1323         }
1324
1325         i_level = ep->GetLevel();
1326
1327         if( el == NULL )
1328         {
1329             if( i_level > 1 )
1330             {
1331                 ep->Up();
1332                 continue;
1333             }
1334             msg_Warn( &sys.demuxer, "EOF" );
1335             return VLC_EGENERIC;
1336         }
1337
1338         /* Verify that we are still inside our cluster
1339          * It can happens whith broken files and when seeking
1340          * without index */
1341         if( i_level > 1 )
1342         {
1343             if( cluster && !ep->IsTopPresent( cluster ) )
1344             {
1345                 msg_Warn( &sys.demuxer, "Unexpected escape from current cluster" );
1346                 cluster = NULL;
1347             }
1348             if( !cluster )
1349                 continue;
1350         }
1351
1352         /* do parsing */
1353         switch ( i_level )
1354         {
1355         case 1:
1356             if( MKV_IS_ID( el, KaxCluster ) )
1357             {
1358                 cluster = (KaxCluster*)el;
1359                 i_cluster_pos = cluster->GetElementPosition();
1360
1361                 /* add it to the index */
1362                 if( i_index == 0 ||
1363                     ( i_index > 0 && p_indexes[i_index - 1].i_position < (int64_t)cluster->GetElementPosition() ) )
1364                 {
1365                     IndexAppendCluster( cluster );
1366                 }
1367
1368                 // reset silent tracks
1369                 for (size_t i=0; i<tracks.size(); i++)
1370                 {
1371                     tracks[i]->b_silent = false;
1372                 }
1373
1374                 ep->Down();
1375             }
1376             else if( MKV_IS_ID( el, KaxCues ) )
1377             {
1378                 msg_Warn( &sys.demuxer, "find KaxCues FIXME" );
1379                 return VLC_EGENERIC;
1380             }
1381             else
1382             {
1383                 msg_Dbg( &sys.demuxer, "unknown (%s)", typeid( el ).name() );
1384             }
1385             break;
1386         case 2:
1387             if( MKV_IS_ID( el, KaxClusterTimecode ) )
1388             {
1389                 KaxClusterTimecode &ctc = *(KaxClusterTimecode*)el;
1390
1391                 ctc.ReadData( es.I_O(), SCOPE_ALL_DATA );
1392                 cluster->InitTimecode( uint64( ctc ), i_timescale );
1393             }
1394             else if( MKV_IS_ID( el, KaxClusterSilentTracks ) )
1395             {
1396                 ep->Down();
1397             }
1398             else if( MKV_IS_ID( el, KaxBlockGroup ) )
1399             {
1400                 i_block_pos = el->GetElementPosition();
1401                 ep->Down();
1402             }
1403             else if( MKV_IS_ID( el, KaxSimpleBlock ) )
1404             {
1405                 pp_simpleblock = (KaxSimpleBlock*)el;
1406
1407                 pp_simpleblock->ReadData( es.I_O() );
1408                 pp_simpleblock->SetParent( *cluster );
1409             }
1410             break;
1411         case 3:
1412             if( MKV_IS_ID( el, KaxBlock ) )
1413             {
1414                 pp_block = (KaxBlock*)el;
1415
1416                 pp_block->ReadData( es.I_O() );
1417                 pp_block->SetParent( *cluster );
1418
1419                 ep->Keep();
1420             }
1421             else if( MKV_IS_ID( el, KaxBlockDuration ) )
1422             {
1423                 KaxBlockDuration &dur = *(KaxBlockDuration*)el;
1424
1425                 dur.ReadData( es.I_O() );
1426                 *pi_duration = uint64( dur );
1427             }
1428             else if( MKV_IS_ID( el, KaxReferenceBlock ) )
1429             {
1430                 KaxReferenceBlock &ref = *(KaxReferenceBlock*)el;
1431
1432                 ref.ReadData( es.I_O() );
1433
1434                 if( *pb_key_picture )
1435                     *pb_key_picture = false;
1436                 else if( int64( ref ) > 0 )
1437                     *pb_discardable_picture = true;
1438             }
1439             else if( MKV_IS_ID( el, KaxClusterSilentTrackNumber ) )
1440             {
1441                 KaxClusterSilentTrackNumber &track_num = *(KaxClusterSilentTrackNumber*)el;
1442                 track_num.ReadData( es.I_O() );
1443                 // find the track
1444                 for (size_t i=0; i<tracks.size(); i++)
1445                 {
1446                     if ( tracks[i]->i_number == uint32(track_num))
1447                     {
1448                         tracks[i]->b_silent = true;
1449                         break;
1450                     }
1451                 }
1452             }
1453             break;
1454         default:
1455             msg_Err( &sys.demuxer, "invalid level = %d", i_level );
1456             return VLC_EGENERIC;
1457         }
1458     }
1459 }
1460