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

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