]> git.sesse.net Git - vlc/blob - modules/demux/mkv/matroska_segment.cpp
Fix compilation
[vlc] / 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             MP4_ReadBoxCommon( p_mp4_stream, p_box );
794             MP4_ReadBox_sample_vide( p_mp4_stream, p_box );
795             tracks[i_track]->fmt.i_codec = p_box->i_type;
796             tracks[i_track]->fmt.video.i_width = p_box->data.p_sample_vide->i_width;
797             tracks[i_track]->fmt.video.i_height = p_box->data.p_sample_vide->i_height;
798             tracks[i_track]->fmt.i_extra = p_box->data.p_sample_vide->i_qt_image_description;
799             tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
800             memcpy( tracks[i_track]->fmt.p_extra, p_box->data.p_sample_vide->p_qt_image_description, tracks[i_track]->fmt.i_extra );
801             MP4_FreeBox_sample_vide( p_box );
802             stream_Delete( p_mp4_stream );
803         }
804         else if( !strcmp( tracks[i_track]->psz_codec, "A_MS/ACM" ) )
805         {
806             if( tracks[i_track]->i_extra_data < (int)sizeof( WAVEFORMATEX ) )
807             {
808                 msg_Err( &sys.demuxer, "missing/invalid WAVEFORMATEX" );
809                 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
810             }
811             else
812             {
813                 WAVEFORMATEX *p_wf = (WAVEFORMATEX*)tracks[i_track]->p_extra_data;
814
815                 wf_tag_to_fourcc( GetWLE( &p_wf->wFormatTag ), &tracks[i_track]->fmt.i_codec, NULL );
816
817                 tracks[i_track]->fmt.audio.i_channels   = GetWLE( &p_wf->nChannels );
818                 tracks[i_track]->fmt.audio.i_rate = GetDWLE( &p_wf->nSamplesPerSec );
819                 tracks[i_track]->fmt.i_bitrate    = GetDWLE( &p_wf->nAvgBytesPerSec ) * 8;
820                 tracks[i_track]->fmt.audio.i_blockalign = GetWLE( &p_wf->nBlockAlign );;
821                 tracks[i_track]->fmt.audio.i_bitspersample = GetWLE( &p_wf->wBitsPerSample );
822
823                 tracks[i_track]->fmt.i_extra            = GetWLE( &p_wf->cbSize );
824                 if( tracks[i_track]->fmt.i_extra > 0 )
825                 {
826                     tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
827                     memcpy( tracks[i_track]->fmt.p_extra, &p_wf[1], tracks[i_track]->fmt.i_extra );
828                 }
829             }
830         }
831         else if( !strcmp( tracks[i_track]->psz_codec, "A_MPEG/L3" ) ||
832                  !strcmp( tracks[i_track]->psz_codec, "A_MPEG/L2" ) ||
833                  !strcmp( tracks[i_track]->psz_codec, "A_MPEG/L1" ) )
834         {
835             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'm', 'p', 'g', 'a' );
836         }
837         else if( !strcmp( tracks[i_track]->psz_codec, "A_AC3" ) )
838         {
839             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'a', '5', '2', ' ' );
840         }
841         else if( !strcmp( tracks[i_track]->psz_codec, "A_EAC3" ) )
842         {
843             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'e', 'a', 'c', '3' );
844         }
845         else if( !strcmp( tracks[i_track]->psz_codec, "A_DTS" ) )
846         {
847             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'd', 't', 's', ' ' );
848         }
849         else if( !strcmp( tracks[i_track]->psz_codec, "A_FLAC" ) )
850         {
851             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'f', 'l', 'a', 'c' );
852             tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
853             tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
854             memcpy( tracks[i_track]->fmt.p_extra,tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
855         }
856         else if( !strcmp( tracks[i_track]->psz_codec, "A_VORBIS" ) )
857         {
858             int i, i_offset = 1, i_size[3], i_extra;
859             uint8_t *p_extra;
860
861             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'v', 'o', 'r', 'b' );
862
863             /* Split the 3 headers */
864             if( tracks[i_track]->p_extra_data[0] != 0x02 )
865                 msg_Err( &sys.demuxer, "invalid vorbis header" );
866
867             for( i = 0; i < 2; i++ )
868             {
869                 i_size[i] = 0;
870                 while( i_offset < tracks[i_track]->i_extra_data )
871                 {
872                     i_size[i] += tracks[i_track]->p_extra_data[i_offset];
873                     if( tracks[i_track]->p_extra_data[i_offset++] != 0xff ) break;
874                 }
875             }
876
877             i_size[0] = __MIN(i_size[0], tracks[i_track]->i_extra_data - i_offset);
878             i_size[1] = __MIN(i_size[1], tracks[i_track]->i_extra_data -i_offset -i_size[0]);
879             i_size[2] = tracks[i_track]->i_extra_data - i_offset - i_size[0] - i_size[1];
880
881             tracks[i_track]->fmt.i_extra = 3 * 2 + i_size[0] + i_size[1] + i_size[2];
882             tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
883             p_extra = (uint8_t *)tracks[i_track]->fmt.p_extra; i_extra = 0;
884             for( i = 0; i < 3; i++ )
885             {
886                 *(p_extra++) = i_size[i] >> 8;
887                 *(p_extra++) = i_size[i] & 0xFF;
888                 memcpy( p_extra, tracks[i_track]->p_extra_data + i_offset + i_extra,
889                         i_size[i] );
890                 p_extra += i_size[i];
891                 i_extra += i_size[i];
892             }
893         }
894         else if( !strncmp( tracks[i_track]->psz_codec, "A_AAC/MPEG2/", strlen( "A_AAC/MPEG2/" ) ) ||
895                  !strncmp( tracks[i_track]->psz_codec, "A_AAC/MPEG4/", strlen( "A_AAC/MPEG4/" ) ) )
896         {
897             int i_profile, i_srate, sbr = 0;
898             static const unsigned int i_sample_rates[] =
899             {
900                     96000, 88200, 64000, 48000, 44100, 32000, 24000, 22050,
901                         16000, 12000, 11025, 8000,  7350,  0,     0,     0
902             };
903
904             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'a' );
905             /* create data for faad (MP4DecSpecificDescrTag)*/
906
907             if( !strcmp( &tracks[i_track]->psz_codec[12], "MAIN" ) )
908             {
909                 i_profile = 0;
910             }
911             else if( !strcmp( &tracks[i_track]->psz_codec[12], "LC" ) )
912             {
913                 i_profile = 1;
914             }
915             else if( !strcmp( &tracks[i_track]->psz_codec[12], "SSR" ) )
916             {
917                 i_profile = 2;
918             }
919             else if( !strcmp( &tracks[i_track]->psz_codec[12], "LC/SBR" ) )
920             {
921                 i_profile = 1;
922                 sbr = 1;
923             }
924             else
925             {
926                 i_profile = 3;
927             }
928
929             for( i_srate = 0; i_srate < 13; i_srate++ )
930             {
931                 if( i_sample_rates[i_srate] == tracks[i_track]->i_original_rate )
932                 {
933                     break;
934                 }
935             }
936             msg_Dbg( &sys.demuxer, "profile=%d srate=%d", i_profile, i_srate );
937
938             tracks[i_track]->fmt.i_extra = sbr ? 5 : 2;
939             tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
940             ((uint8_t*)tracks[i_track]->fmt.p_extra)[0] = ((i_profile + 1) << 3) | ((i_srate&0xe) >> 1);
941             ((uint8_t*)tracks[i_track]->fmt.p_extra)[1] = ((i_srate & 0x1) << 7) | (tracks[i_track]->fmt.audio.i_channels << 3);
942             if (sbr != 0)
943             {
944                 int syncExtensionType = 0x2B7;
945                 int iDSRI;
946                 for (iDSRI=0; iDSRI<13; iDSRI++)
947                     if( i_sample_rates[iDSRI] == tracks[i_track]->fmt.audio.i_rate )
948                         break;
949                 ((uint8_t*)tracks[i_track]->fmt.p_extra)[2] = (syncExtensionType >> 3) & 0xFF;
950                 ((uint8_t*)tracks[i_track]->fmt.p_extra)[3] = ((syncExtensionType & 0x7) << 5) | 5;
951                 ((uint8_t*)tracks[i_track]->fmt.p_extra)[4] = ((1 & 0x1) << 7) | (iDSRI << 3);
952             }
953         }
954         else if( !strcmp( tracks[i_track]->psz_codec, "A_AAC" ) )
955         {
956             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'a' );
957             tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
958             tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
959             memcpy( tracks[i_track]->fmt.p_extra, tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
960         }
961         else if( !strcmp( tracks[i_track]->psz_codec, "A_WAVPACK4" ) )
962         {
963             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'W', 'V', 'P', 'K' );
964             tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
965             tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
966             memcpy( tracks[i_track]->fmt.p_extra, tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
967         }
968         else if( !strcmp( tracks[i_track]->psz_codec, "A_TTA1" ) )
969         {
970             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'T', 'T', 'A', '1' );
971             tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
972             tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
973             memcpy( tracks[i_track]->fmt.p_extra, tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
974         }
975         else if( !strcmp( tracks[i_track]->psz_codec, "A_PCM/INT/BIG" ) ||
976                  !strcmp( tracks[i_track]->psz_codec, "A_PCM/INT/LIT" ) ||
977                  !strcmp( tracks[i_track]->psz_codec, "A_PCM/FLOAT/IEEE" ) )
978         {
979             if( !strcmp( tracks[i_track]->psz_codec, "A_PCM/INT/BIG" ) )
980             {
981                 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 't', 'w', 'o', 's' );
982             }
983             else
984             {
985                 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'a', 'r', 'a', 'w' );
986             }
987             tracks[i_track]->fmt.audio.i_blockalign = ( tracks[i_track]->fmt.audio.i_bitspersample + 7 ) / 8 * tracks[i_track]->fmt.audio.i_channels;
988         }
989         /* disabled due to the potential "S_KATE" namespace issue */
990         else if( !strcmp( tracks[i_track]->psz_codec, "S_KATE" ) )
991         {
992             int i, i_offset = 1, *i_size, i_extra, num_headers, size_so_far;
993             uint8_t *p_extra;
994
995             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'k', 'a', 't', 'e' );
996             tracks[i_track]->fmt.subs.psz_encoding = strdup( "UTF-8" );
997
998             /* Recover the number of headers to expect */
999             num_headers = tracks[i_track]->p_extra_data[0]+1;
1000             msg_Dbg( &sys.demuxer, "kate in mkv detected: %d headers in %u bytes",
1001                 num_headers, tracks[i_track]->i_extra_data);
1002
1003             /* this won't overflow the stack as is can allocate only 1020 bytes max */
1004             i_size = (int*)alloca(num_headers*sizeof(int));
1005
1006             /* Split the headers */
1007             size_so_far = 0;
1008             for( i = 0; i < num_headers-1; i++ )
1009             {
1010                 i_size[i] = 0;
1011                 while( i_offset < tracks[i_track]->i_extra_data )
1012                 {
1013                     i_size[i] += tracks[i_track]->p_extra_data[i_offset];
1014                     if( tracks[i_track]->p_extra_data[i_offset++] != 0xff ) break;
1015                 }
1016                 msg_Dbg( &sys.demuxer, "kate header %d is %d bytes", i, i_size[i]);
1017                 size_so_far += i_size[i];
1018             }
1019             i_size[num_headers-1] = tracks[i_track]->i_extra_data - (size_so_far+i_offset);
1020             msg_Dbg( &sys.demuxer, "kate last header (%d) is %d bytes", num_headers-1, i_size[num_headers-1]);
1021
1022             tracks[i_track]->fmt.i_extra = 1 + num_headers * 2 + size_so_far + i_size[num_headers-1];
1023             tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
1024
1025             p_extra = (uint8_t *)tracks[i_track]->fmt.p_extra;
1026             i_extra = 0;
1027             *(p_extra++) = num_headers;
1028             ++i_extra;
1029             for( i = 0; i < num_headers; i++ )
1030             {
1031                 *(p_extra++) = i_size[i] >> 8;
1032                 *(p_extra++) = i_size[i] & 0xFF;
1033                 memcpy( p_extra, tracks[i_track]->p_extra_data + i_offset + i_extra-1,
1034                         i_size[i] );
1035                 p_extra += i_size[i];
1036                 i_extra += i_size[i];
1037             }
1038         }
1039         else if( !strcmp( tracks[i_track]->psz_codec, "S_TEXT/ASCII" ) )
1040         {
1041             p_fmt->i_codec = VLC_FOURCC( 's', 'u', 'b', 't' );
1042             p_fmt->subs.psz_encoding = NULL; /* Is there a place where it is stored ? */
1043         }
1044         else if( !strcmp( tracks[i_track]->psz_codec, "S_TEXT/UTF8" ) )
1045         {
1046             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 's', 'u', 'b', 't' );
1047             tracks[i_track]->fmt.subs.psz_encoding = strdup( "UTF-8" );
1048         }
1049         else if( !strcmp( tracks[i_track]->psz_codec, "S_TEXT/USF" ) )
1050         {
1051             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'u', 's', 'f', ' ' );
1052             tracks[i_track]->fmt.subs.psz_encoding = strdup( "UTF-8" );
1053             if( tracks[i_track]->i_extra_data )
1054             {
1055                 tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
1056                 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
1057                 memcpy( tracks[i_track]->fmt.p_extra, tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
1058             }
1059         }
1060         else if( !strcmp( tracks[i_track]->psz_codec, "S_TEXT/SSA" ) ||
1061                  !strcmp( tracks[i_track]->psz_codec, "S_TEXT/ASS" ) ||
1062                  !strcmp( tracks[i_track]->psz_codec, "S_SSA" ) ||
1063                  !strcmp( tracks[i_track]->psz_codec, "S_ASS" ))
1064         {
1065             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 's', 's', 'a', ' ' );
1066             tracks[i_track]->fmt.subs.psz_encoding = strdup( "UTF-8" );
1067             if( tracks[i_track]->i_extra_data )
1068             {
1069                 tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
1070                 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
1071                 memcpy( tracks[i_track]->fmt.p_extra, tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
1072             }
1073         }
1074         else if( !strcmp( tracks[i_track]->psz_codec, "S_VOBSUB" ) )
1075         {
1076             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 's','p','u',' ' );
1077             if( tracks[i_track]->i_extra_data )
1078             {
1079                 char *p_start;
1080                 char *p_buf = (char *)malloc( tracks[i_track]->i_extra_data + 1);
1081                 memcpy( p_buf, tracks[i_track]->p_extra_data , tracks[i_track]->i_extra_data );
1082                 p_buf[tracks[i_track]->i_extra_data] = '\0';
1083  
1084                 p_start = strstr( p_buf, "size:" );
1085                 if( sscanf( p_start, "size: %dx%d",
1086                         &tracks[i_track]->fmt.subs.spu.i_original_frame_width, &tracks[i_track]->fmt.subs.spu.i_original_frame_height ) == 2 )
1087                 {
1088                     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 );
1089                 }
1090                 else
1091                 {
1092                     msg_Warn( &sys.demuxer, "reading original frame size for vobsub failed" );
1093                 }
1094                 free( p_buf );
1095             }
1096         }
1097         else if( !strcmp( tracks[i_track]->psz_codec, "B_VOBBTN" ) )
1098         {
1099             tracks[i_track]->fmt.i_cat = NAV_ES;
1100             continue;
1101         }
1102         else if( !strcmp( p_tk->psz_codec, "A_REAL/14_4" ) )
1103         {
1104             p_fmt->i_codec = VLC_FOURCC( '1', '4', '_', '4');
1105             p_fmt->audio.i_channels = 1;
1106             p_fmt->audio.i_rate = 8000;
1107             p_fmt->audio.i_blockalign = 0x14;
1108         }
1109         else
1110         {
1111             msg_Err( &sys.demuxer, "unknown codec id=`%s'", tracks[i_track]->psz_codec );
1112             tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
1113         }
1114         if( tracks[i_track]->b_default )
1115         {
1116             tracks[i_track]->fmt.i_priority = 1000;
1117         }
1118
1119         tracks[i_track]->p_es = es_out_Add( sys.demuxer.out, &tracks[i_track]->fmt );
1120
1121         /* Turn on a subtitles track if it has been flagged as default -
1122          * but only do this if no subtitles track has already been engaged,
1123          * either by an earlier 'default track' (??) or by default
1124          * language choice behaviour.
1125          */
1126         if( tracks[i_track]->b_default )
1127         {
1128             es_out_Control( sys.demuxer.out,
1129                             ES_OUT_SET_ES_DEFAULT,
1130                             tracks[i_track]->p_es );
1131         }
1132     }
1133     es_out_Control( sys.demuxer.out, ES_OUT_SET_NEXT_DISPLAY_TIME, i_start_time );
1134  
1135     sys.i_start_pts = i_start_time;
1136     // reset the stream reading to the first cluster of the segment used
1137     es.I_O().setFilePointer( i_start_pos );
1138
1139     delete ep;
1140     ep = new EbmlParser( &es, segment, &sys.demuxer );
1141
1142     return true;
1143 }
1144
1145 void matroska_segment_c::UnSelect( )
1146 {
1147     size_t i_track;
1148
1149     for( i_track = 0; i_track < tracks.size(); i_track++ )
1150     {
1151         if ( tracks[i_track]->p_es != NULL )
1152         {
1153 //            es_format_Clean( &tracks[i_track]->fmt );
1154             es_out_Del( sys.demuxer.out, tracks[i_track]->p_es );
1155             tracks[i_track]->p_es = NULL;
1156         }
1157     }
1158     delete ep;
1159     ep = NULL;
1160 }
1161
1162 int matroska_segment_c::BlockGet( KaxBlock * & pp_block, KaxSimpleBlock * & pp_simpleblock, int64_t *pi_ref1, int64_t *pi_ref2, int64_t *pi_duration )
1163 {
1164     pp_simpleblock = NULL;
1165     pp_block = NULL;
1166     *pi_ref1  = 0;
1167     *pi_ref2  = 0;
1168
1169     for( ;; )
1170     {
1171         EbmlElement *el = NULL;
1172         int         i_level;
1173
1174         if ( ep == NULL )
1175             return VLC_EGENERIC;
1176
1177         if( pp_simpleblock != NULL || ((el = ep->Get()) == NULL && pp_block != NULL) )
1178         {
1179             /* Check blocks validity to protect againts broken files */
1180             if( BlockFindTrackIndex( NULL, pp_block , pp_simpleblock ) )
1181             {
1182                 delete pp_block;
1183                 pp_simpleblock = NULL;
1184                 pp_block = NULL;
1185                 continue;
1186             }
1187
1188             /* update the index */
1189 #define idx p_indexes[i_index - 1]
1190             if( i_index > 0 && idx.i_time == -1 )
1191             {
1192                 if ( pp_simpleblock != NULL )
1193                     idx.i_time        = pp_simpleblock->GlobalTimecode() / (mtime_t)1000;
1194                 else
1195                     idx.i_time        = (*pp_block).GlobalTimecode() / (mtime_t)1000;
1196                 idx.b_key         = *pi_ref1 == 0 ? true : false;
1197             }
1198 #undef idx
1199             return VLC_SUCCESS;
1200         }
1201
1202         i_level = ep->GetLevel();
1203
1204         if( el == NULL )
1205         {
1206             if( i_level > 1 )
1207             {
1208                 ep->Up();
1209                 continue;
1210             }
1211             msg_Warn( &sys.demuxer, "EOF" );
1212             return VLC_EGENERIC;
1213         }
1214
1215         /* Verify that we are still inside our cluster
1216          * It can happens whith broken files and when seeking
1217          * without index */
1218         if( i_level > 1 )
1219         {
1220             if( cluster && !ep->IsTopPresent( cluster ) )
1221             {
1222                 msg_Warn( &sys.demuxer, "Unexpected escape from current cluster" );
1223                 cluster = NULL;
1224             }
1225             if( !cluster )
1226                 continue;
1227         }
1228
1229         /* do parsing */
1230         switch ( i_level )
1231         {
1232         case 1:
1233             if( MKV_IS_ID( el, KaxCluster ) )
1234             {
1235                 cluster = (KaxCluster*)el;
1236                 i_cluster_pos = cluster->GetElementPosition();
1237
1238                 /* add it to the index */
1239                 if( i_index == 0 ||
1240                     ( i_index > 0 && p_indexes[i_index - 1].i_position < (int64_t)cluster->GetElementPosition() ) )
1241                 {
1242                     IndexAppendCluster( cluster );
1243                 }
1244
1245                 // reset silent tracks
1246                 for (size_t i=0; i<tracks.size(); i++)
1247                 {
1248                     tracks[i]->b_silent = false;
1249                 }
1250
1251                 ep->Down();
1252             }
1253             else if( MKV_IS_ID( el, KaxCues ) )
1254             {
1255                 msg_Warn( &sys.demuxer, "find KaxCues FIXME" );
1256                 return VLC_EGENERIC;
1257             }
1258             else
1259             {
1260                 msg_Dbg( &sys.demuxer, "unknown (%s)", typeid( el ).name() );
1261             }
1262             break;
1263         case 2:
1264             if( MKV_IS_ID( el, KaxClusterTimecode ) )
1265             {
1266                 KaxClusterTimecode &ctc = *(KaxClusterTimecode*)el;
1267
1268                 ctc.ReadData( es.I_O(), SCOPE_ALL_DATA );
1269                 cluster->InitTimecode( uint64( ctc ), i_timescale );
1270             }
1271             else if( MKV_IS_ID( el, KaxClusterSilentTracks ) )
1272             {
1273                 ep->Down();
1274             }
1275             else if( MKV_IS_ID( el, KaxBlockGroup ) )
1276             {
1277                 i_block_pos = el->GetElementPosition();
1278                 ep->Down();
1279             }
1280             else if( MKV_IS_ID( el, KaxSimpleBlock ) )
1281             {
1282                 pp_simpleblock = (KaxSimpleBlock*)el;
1283
1284                 pp_simpleblock->ReadData( es.I_O() );
1285                 pp_simpleblock->SetParent( *cluster );
1286             }
1287             break;
1288         case 3:
1289             if( MKV_IS_ID( el, KaxBlock ) )
1290             {
1291                 pp_block = (KaxBlock*)el;
1292
1293                 pp_block->ReadData( es.I_O() );
1294                 pp_block->SetParent( *cluster );
1295
1296                 ep->Keep();
1297             }
1298             else if( MKV_IS_ID( el, KaxBlockDuration ) )
1299             {
1300                 KaxBlockDuration &dur = *(KaxBlockDuration*)el;
1301
1302                 dur.ReadData( es.I_O() );
1303                 *pi_duration = uint64( dur );
1304             }
1305             else if( MKV_IS_ID( el, KaxReferenceBlock ) )
1306             {
1307                 KaxReferenceBlock &ref = *(KaxReferenceBlock*)el;
1308
1309                 ref.ReadData( es.I_O() );
1310                 if( *pi_ref1 == 0 )
1311                 {
1312                     *pi_ref1 = int64( ref ) * cluster->GlobalTimecodeScale();
1313                 }
1314                 else if( *pi_ref2 == 0 )
1315                 {
1316                     *pi_ref2 = int64( ref ) * cluster->GlobalTimecodeScale();
1317                 }
1318             }
1319             else if( MKV_IS_ID( el, KaxClusterSilentTrackNumber ) )
1320             {
1321                 KaxClusterSilentTrackNumber &track_num = *(KaxClusterSilentTrackNumber*)el;
1322                 track_num.ReadData( es.I_O() );
1323                 // find the track
1324                 for (size_t i=0; i<tracks.size(); i++)
1325                 {
1326                     if ( tracks[i]->i_number == uint32(track_num))
1327                     {
1328                         tracks[i]->b_silent = true;
1329                         break;
1330                     }
1331                 }
1332             }
1333             break;
1334         default:
1335             msg_Err( &sys.demuxer, "invalid level = %d", i_level );
1336             return VLC_EGENERIC;
1337         }
1338     }
1339 }
1340