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