1 /*****************************************************************************
2 * mkv.cpp : matroska demuxer
3 *****************************************************************************
4 * Copyright (C) 2003-2004 the VideoLAN team
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8 * Steve Lhomme <steve.lhomme@free.fr>
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.
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.
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 *****************************************************************************/
25 #include "matroska_segment.hpp"
27 #include "chapters.hpp"
32 matroska_segment_c::~matroska_segment_c()
34 for( size_t i_track = 0; i_track < tracks.size(); i_track++ )
36 delete tracks[i_track]->p_compression_data;
37 es_format_Clean( &tracks[i_track]->fmt );
38 free( tracks[i_track]->p_extra_data );
39 free( tracks[i_track]->psz_codec );
40 delete tracks[i_track];
43 free( psz_writing_application );
44 free( psz_muxing_application );
45 free( psz_segment_filename );
53 delete p_prev_segment_uid;
54 delete p_next_segment_uid;
56 std::vector<chapter_edition_c*>::iterator index = stored_editions.begin();
57 while ( index != stored_editions.end() )
62 std::vector<chapter_translation_c*>::iterator indext = translations.begin();
63 while ( indext != translations.end() )
68 std::vector<KaxSegmentFamily*>::iterator indexf = families.begin();
69 while ( indexf != families.end() )
77 /*****************************************************************************
79 * * LoadCues : load the cues element and update index
81 * * LoadTags : load ... the tags element
83 * * InformationCreate : create all information, load tags if present
85 *****************************************************************************/
86 void matroska_segment_c::LoadCues( KaxCues *cues )
94 msg_Err( &sys.demuxer, "There can be only 1 Cues per section." );
98 ep = new EbmlParser( &es, cues, &sys.demuxer );
99 while( ( el = ep->Get() ) != NULL )
101 if( MKV_IS_ID( el, KaxCuePoint ) )
103 #define idx p_indexes[i_index]
106 idx.i_block_number= -1;
112 while( ( el = ep->Get() ) != NULL )
114 if( MKV_IS_ID( el, KaxCueTime ) )
116 KaxCueTime &ctime = *(KaxCueTime*)el;
118 ctime.ReadData( es.I_O() );
120 idx.i_time = uint64( ctime ) * i_timescale / (mtime_t)1000;
122 else if( MKV_IS_ID( el, KaxCueTrackPositions ) )
125 while( ( el = ep->Get() ) != NULL )
127 if( MKV_IS_ID( el, KaxCueTrack ) )
129 KaxCueTrack &ctrack = *(KaxCueTrack*)el;
131 ctrack.ReadData( es.I_O() );
132 idx.i_track = uint16( ctrack );
134 else if( MKV_IS_ID( el, KaxCueClusterPosition ) )
136 KaxCueClusterPosition &ccpos = *(KaxCueClusterPosition*)el;
138 ccpos.ReadData( es.I_O() );
139 idx.i_position = segment->GetGlobalPosition( uint64( ccpos ) );
141 else if( MKV_IS_ID( el, KaxCueBlockNumber ) )
143 KaxCueBlockNumber &cbnum = *(KaxCueBlockNumber*)el;
145 cbnum.ReadData( es.I_O() );
146 idx.i_block_number = uint32( cbnum );
150 msg_Dbg( &sys.demuxer, " * Unknown (%s)", typeid(*el).name() );
157 msg_Dbg( &sys.demuxer, " * Unknown (%s)", typeid(*el).name() );
163 msg_Dbg( &sys.demuxer, " * added time=%"PRId64" pos=%"PRId64
164 " track=%d bnum=%d", idx.i_time, idx.i_position,
165 idx.i_track, idx.i_block_number );
169 if( i_index >= i_index_max )
172 p_indexes = (mkv_index_t*)realloc( p_indexes, sizeof( mkv_index_t ) * i_index_max );
178 msg_Dbg( &sys.demuxer, " * Unknown (%s)", typeid(*el).name() );
183 msg_Dbg( &sys.demuxer, "| - loading cues done." );
186 void matroska_segment_c::LoadTags( KaxTags *tags )
192 /* Master elements */
193 ep = new EbmlParser( &es, tags, &sys.demuxer );
195 while( ( el = ep->Get() ) != NULL )
197 if( MKV_IS_ID( el, KaxTag ) )
199 msg_Dbg( &sys.demuxer, "+ Tag" );
201 while( ( el = ep->Get() ) != NULL )
203 if( MKV_IS_ID( el, KaxTagTargets ) )
205 msg_Dbg( &sys.demuxer, "| + Targets" );
207 while( ( el = ep->Get() ) != NULL )
209 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid( *el ).name() );
213 else if( MKV_IS_ID( el, KaxTagGeneral ) )
215 msg_Dbg( &sys.demuxer, "| + General" );
217 while( ( el = ep->Get() ) != NULL )
219 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid( *el ).name() );
223 else if( MKV_IS_ID( el, KaxTagGenres ) )
225 msg_Dbg( &sys.demuxer, "| + Genres" );
227 while( ( el = ep->Get() ) != NULL )
229 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid( *el ).name() );
233 else if( MKV_IS_ID( el, KaxTagAudioSpecific ) )
235 msg_Dbg( &sys.demuxer, "| + Audio Specific" );
237 while( ( el = ep->Get() ) != NULL )
239 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid( *el ).name() );
243 else if( MKV_IS_ID( el, KaxTagImageSpecific ) )
245 msg_Dbg( &sys.demuxer, "| + Images Specific" );
247 while( ( el = ep->Get() ) != NULL )
249 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid( *el ).name() );
253 else if( MKV_IS_ID( el, KaxTagMultiComment ) )
255 msg_Dbg( &sys.demuxer, "| + Multi Comment" );
257 else if( MKV_IS_ID( el, KaxTagMultiCommercial ) )
259 msg_Dbg( &sys.demuxer, "| + Multi Commercial" );
261 else if( MKV_IS_ID( el, KaxTagMultiDate ) )
263 msg_Dbg( &sys.demuxer, "| + Multi Date" );
265 else if( MKV_IS_ID( el, KaxTagMultiEntity ) )
267 msg_Dbg( &sys.demuxer, "| + Multi Entity" );
269 else if( MKV_IS_ID( el, KaxTagMultiIdentifier ) )
271 msg_Dbg( &sys.demuxer, "| + Multi Identifier" );
273 else if( MKV_IS_ID( el, KaxTagMultiLegal ) )
275 msg_Dbg( &sys.demuxer, "| + Multi Legal" );
277 else if( MKV_IS_ID( el, KaxTagMultiTitle ) )
279 msg_Dbg( &sys.demuxer, "| + Multi Title" );
283 msg_Dbg( &sys.demuxer, "| + LoadTag Unknown (%s)", typeid( *el ).name() );
290 msg_Dbg( &sys.demuxer, "+ Unknown (%s)", typeid( *el ).name() );
295 msg_Dbg( &sys.demuxer, "loading tags done." );
298 /*****************************************************************************
300 *****************************************************************************/
301 void matroska_segment_c::InformationCreate( )
303 sys.meta = vlc_meta_New();
307 vlc_meta_SetTitle( sys.meta, psz_title );
311 vlc_meta_SetDate( sys.meta, psz_date_utc );
314 if( psz_segment_filename )
316 fprintf( stderr, "***** WARNING: Unhandled meta - Use custom\n" );
318 if( psz_muxing_application )
320 fprintf( stderr, "***** WARNING: Unhandled meta - Use custom\n" );
322 if( psz_writing_application )
324 fprintf( stderr, "***** WARNING: Unhandled meta - Use custom\n" );
327 for( size_t i_track = 0; i_track < tracks.size(); i_track++ )
329 // mkv_track_t *tk = tracks[i_track];
330 // vlc_meta_t *mtk = vlc_meta_New();
331 fprintf( stderr, "***** WARNING: Unhandled child meta\n");
335 if( i_tags_position >= 0 )
339 stream_Control( sys.demuxer.s, STREAM_CAN_FASTSEEK, &b_seekable );
349 /*****************************************************************************
351 *****************************************************************************/
353 void matroska_segment_c::IndexAppendCluster( KaxCluster *cluster )
355 #define idx p_indexes[i_index]
357 idx.i_block_number= -1;
358 idx.i_position = cluster->GetElementPosition();
363 if( i_index >= i_index_max )
366 p_indexes = (mkv_index_t*)realloc( p_indexes, sizeof( mkv_index_t ) * i_index_max );
372 bool matroska_segment_c::PreloadFamily( const matroska_segment_c & of_segment )
377 for (size_t i=0; i<families.size(); i++)
379 for (size_t j=0; j<of_segment.families.size(); j++)
381 if ( *(families[i]) == *(of_segment.families[j]) )
389 bool matroska_segment_c::CompareSegmentUIDs( const matroska_segment_c * p_item_a, const matroska_segment_c * p_item_b )
393 if ( p_item_a == NULL || p_item_b == NULL )
396 p_tmp = (EbmlBinary *)p_item_a->p_segment_uid;
397 if ( p_item_b->p_prev_segment_uid != NULL
398 && *p_tmp == *p_item_b->p_prev_segment_uid )
401 p_tmp = (EbmlBinary *)p_item_a->p_next_segment_uid;
405 if ( p_item_b->p_segment_uid != NULL
406 && *p_tmp == *p_item_b->p_segment_uid )
409 if ( p_item_b->p_prev_segment_uid != NULL
410 && *p_tmp == *p_item_b->p_prev_segment_uid )
416 bool matroska_segment_c::Preload( )
421 EbmlElement *el = NULL;
423 ep->Reset( &sys.demuxer );
425 while( ( el = ep->Get() ) != NULL )
427 if( MKV_IS_ID( el, KaxSeekHead ) )
429 /* Multiple allowed */
430 /* We bail at 10, to prevent possible recursion */
431 msg_Dbg( &sys.demuxer, "| + Seek head" );
432 if( i_seekhead_count < 10 )
434 i_seekhead_position = (int64_t) es.I_O().getFilePointer();
435 ParseSeekHead( static_cast<KaxSeekHead*>( el ) );
438 else if( MKV_IS_ID( el, KaxInfo ) )
440 /* Multiple allowed, mandatory */
441 msg_Dbg( &sys.demuxer, "| + Information" );
442 if( i_info_position < 0 ) // FIXME
443 ParseInfo( static_cast<KaxInfo*>( el ) );
444 i_info_position = (int64_t) es.I_O().getFilePointer();
446 else if( MKV_IS_ID( el, KaxTracks ) )
448 /* Multiple allowed */
449 msg_Dbg( &sys.demuxer, "| + Tracks" );
450 if( i_tracks_position < 0 ) // FIXME
451 ParseTracks( static_cast<KaxTracks*>( el ) );
452 if ( tracks.size() == 0 )
454 msg_Err( &sys.demuxer, "No tracks supported" );
457 i_tracks_position = (int64_t) es.I_O().getFilePointer();
459 else if( MKV_IS_ID( el, KaxCues ) )
461 msg_Dbg( &sys.demuxer, "| + Cues" );
462 if( i_cues_position < 0 )
463 LoadCues( static_cast<KaxCues*>( el ) );
464 i_cues_position = (int64_t) es.I_O().getFilePointer();
466 else if( MKV_IS_ID( el, KaxCluster ) )
468 msg_Dbg( &sys.demuxer, "| + Cluster" );
470 cluster = (KaxCluster*)el;
472 i_cluster_pos = i_start_pos = cluster->GetElementPosition();
476 /* stop pre-parsing the stream */
479 else if( MKV_IS_ID( el, KaxAttachments ) )
481 msg_Dbg( &sys.demuxer, "| + Attachments" );
482 if( i_attachments_position < 0 )
483 ParseAttachments( static_cast<KaxAttachments*>( el ) );
484 i_attachments_position = (int64_t) es.I_O().getFilePointer();
486 else if( MKV_IS_ID( el, KaxChapters ) )
488 msg_Dbg( &sys.demuxer, "| + Chapters" );
489 if( i_chapters_position < 0 )
490 ParseChapters( static_cast<KaxChapters*>( el ) );
491 i_chapters_position = (int64_t) es.I_O().getFilePointer();
493 else if( MKV_IS_ID( el, KaxTag ) )
495 msg_Dbg( &sys.demuxer, "| + Tags" );
496 if( i_tags_position < 0) // FIXME
497 ;//LoadTags( static_cast<KaxTags*>( el ) );
498 i_tags_position = (int64_t) es.I_O().getFilePointer();
501 msg_Dbg( &sys.demuxer, "| + Preload Unknown (%s)", typeid(*el).name() );
509 /* Here we try to load elements that were found in Seek Heads, but not yet parsed */
510 bool matroska_segment_c::LoadSeekHeadItem( const EbmlCallbacks & ClassInfos, int64_t i_element_position )
512 int64_t i_sav_position = (int64_t)es.I_O().getFilePointer();
515 es.I_O().setFilePointer( i_element_position, seek_beginning );
516 el = es.FindNextID( ClassInfos, 0xFFFFFFFFL);
520 msg_Err( &sys.demuxer, "cannot load some cues/chapters/tags etc. (broken seekhead or file)" );
521 es.I_O().setFilePointer( i_sav_position, seek_beginning );
525 if( MKV_IS_ID( el, KaxSeekHead ) )
527 /* Multiple allowed */
528 msg_Dbg( &sys.demuxer, "| + Seek head" );
529 if( i_seekhead_count < 10 )
531 i_seekhead_position = i_element_position;
532 ParseSeekHead( static_cast<KaxSeekHead*>( el ) );
535 else if( MKV_IS_ID( el, KaxInfo ) ) // FIXME
537 /* Multiple allowed, mandatory */
538 msg_Dbg( &sys.demuxer, "| + Information" );
539 if( i_info_position < 0 )
540 ParseInfo( static_cast<KaxInfo*>( el ) );
541 i_info_position = i_element_position;
543 else if( MKV_IS_ID( el, KaxTracks ) ) // FIXME
545 /* Multiple allowed */
546 msg_Dbg( &sys.demuxer, "| + Tracks" );
547 if( i_tracks_position < 0 )
548 ParseTracks( static_cast<KaxTracks*>( el ) );
549 if ( tracks.size() == 0 )
551 msg_Err( &sys.demuxer, "No tracks supported" );
553 es.I_O().setFilePointer( i_sav_position, seek_beginning );
556 i_tracks_position = i_element_position;
558 else if( MKV_IS_ID( el, KaxCues ) )
560 msg_Dbg( &sys.demuxer, "| + Cues" );
561 if( i_cues_position < 0 )
562 LoadCues( static_cast<KaxCues*>( el ) );
563 i_cues_position = i_element_position;
565 else if( MKV_IS_ID( el, KaxAttachments ) )
567 msg_Dbg( &sys.demuxer, "| + Attachments" );
568 if( i_attachments_position < 0 )
569 ParseAttachments( static_cast<KaxAttachments*>( el ) );
570 i_attachments_position = i_element_position;
572 else if( MKV_IS_ID( el, KaxChapters ) )
574 msg_Dbg( &sys.demuxer, "| + Chapters" );
575 if( i_chapters_position < 0 )
576 ParseChapters( static_cast<KaxChapters*>( el ) );
577 i_chapters_position = i_element_position;
579 else if( MKV_IS_ID( el, KaxTag ) ) // FIXME
581 msg_Dbg( &sys.demuxer, "| + Tags" );
582 if( i_tags_position < 0 )
583 ;//LoadTags( static_cast<KaxTags*>( el ) );
584 i_tags_position = i_element_position;
588 msg_Dbg( &sys.demuxer, "| + LoadSeekHeadItem Unknown (%s)", typeid(*el).name() );
592 es.I_O().setFilePointer( i_sav_position, seek_beginning );
596 int matroska_segment_c::BlockFindTrackIndex( size_t *pi_track,
597 const KaxBlock *p_block, const KaxSimpleBlock *p_simpleblock )
603 for( i_track = 0; i_track < tracks.size(); i_track++ )
605 const mkv_track_t *tk = tracks[i_track];
607 if( ( p_block != NULL && tk->i_number == p_block->TrackNum() ) ||
608 ( p_simpleblock != NULL && tk->i_number == p_simpleblock->TrackNum() ) )
614 if( i_track >= tracks.size() )
622 bool matroska_segment_c::Select( mtime_t i_start_time )
627 msg_Dbg( &sys.demuxer, "found %d es", (int)tracks.size() );
628 sys.b_pci_packet_set = false;
630 for( i_track = 0; i_track < tracks.size(); i_track++ )
632 if( tracks[i_track]->fmt.i_cat == UNKNOWN_ES )
634 msg_Warn( &sys.demuxer, "invalid track[%d, n=%d]", (int)i_track, tracks[i_track]->i_number );
635 tracks[i_track]->p_es = NULL;
639 if( !strcmp( tracks[i_track]->psz_codec, "V_MS/VFW/FOURCC" ) )
641 if( tracks[i_track]->i_extra_data < (int)sizeof( BITMAPINFOHEADER ) )
643 msg_Err( &sys.demuxer, "missing/invalid BITMAPINFOHEADER" );
644 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
648 BITMAPINFOHEADER *p_bih = (BITMAPINFOHEADER*)tracks[i_track]->p_extra_data;
650 tracks[i_track]->fmt.video.i_width = GetDWLE( &p_bih->biWidth );
651 tracks[i_track]->fmt.video.i_height= GetDWLE( &p_bih->biHeight );
652 tracks[i_track]->fmt.i_codec = GetFOURCC( &p_bih->biCompression );
654 tracks[i_track]->fmt.i_extra = GetDWLE( &p_bih->biSize ) - sizeof( BITMAPINFOHEADER );
655 if( tracks[i_track]->fmt.i_extra > 0 )
657 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
658 memcpy( tracks[i_track]->fmt.p_extra, &p_bih[1], tracks[i_track]->fmt.i_extra );
662 else if( !strcmp( tracks[i_track]->psz_codec, "V_MPEG1" ) ||
663 !strcmp( tracks[i_track]->psz_codec, "V_MPEG2" ) )
665 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'm', 'p', 'g', 'v' );
667 else if( !strncmp( tracks[i_track]->psz_codec, "V_THEORA", 8 ) )
669 uint8_t *p_data = tracks[i_track]->p_extra_data;
670 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 't', 'h', 'e', 'o' );
671 if( tracks[i_track]->i_extra_data >= 4 ) {
672 if( p_data[0] == 2 ) {
674 int i_size1 = 0, i_size2 = 0;
676 /* read size of first header packet */
677 while( *p_data == 0xFF &&
678 i < tracks[i_track]->i_extra_data )
687 msg_Dbg( &sys.demuxer, "first theora header size %d", i_size1 );
688 /* read size of second header packet */
689 while( *p_data == 0xFF &&
690 i < tracks[i_track]->i_extra_data )
699 int i_size3 = tracks[i_track]->i_extra_data - i - i_size1
701 msg_Dbg( &sys.demuxer, "second theora header size %d", i_size2 );
702 msg_Dbg( &sys.demuxer, "third theora header size %d", i_size3 );
703 tracks[i_track]->fmt.i_extra = i_size1 + i_size2 + i_size3
705 if( i_size1 > 0 && i_size2 > 0 && i_size3 > 0 ) {
706 tracks[i_track]->fmt.p_extra =
707 malloc( tracks[i_track]->fmt.i_extra );
708 uint8_t *p_out = (uint8_t*)tracks[i_track]->fmt.p_extra;
709 *p_out++ = (i_size1>>8) & 0xFF;
710 *p_out++ = i_size1 & 0xFF;
711 memcpy( p_out, p_data, i_size1 );
715 *p_out++ = (i_size2>>8) & 0xFF;
716 *p_out++ = i_size2 & 0xFF;
717 memcpy( p_out, p_data, i_size2 );
721 *p_out++ = (i_size3>>8) & 0xFF;
722 *p_out++ = i_size3 & 0xFF;
723 memcpy( p_out, p_data, i_size3 );
729 msg_Err( &sys.demuxer, "inconsistant theora extradata" );
733 msg_Err( &sys.demuxer, "Wrong number of ogg packets with theora headers (%d)", p_data[0] + 1 );
737 else if( !strncmp( tracks[i_track]->psz_codec, "V_REAL/RV", 9 ) )
739 if( !strcmp( tracks[i_track]->psz_codec, "V_REAL/RV10" ) )
740 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'R', 'V', '1', '0' );
741 else if( !strcmp( tracks[i_track]->psz_codec, "V_REAL/RV20" ) )
742 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'R', 'V', '2', '0' );
743 else if( !strcmp( tracks[i_track]->psz_codec, "V_REAL/RV30" ) )
744 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'R', 'V', '3', '0' );
745 else if( !strcmp( tracks[i_track]->psz_codec, "V_REAL/RV40" ) )
746 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'R', 'V', '4', '0' );
748 else if( !strncmp( tracks[i_track]->psz_codec, "V_MPEG4", 7 ) )
750 if( !strcmp( tracks[i_track]->psz_codec, "V_MPEG4/MS/V3" ) )
752 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'D', 'I', 'V', '3' );
754 else if( !strncmp( tracks[i_track]->psz_codec, "V_MPEG4/ISO", 11 ) )
756 /* A MPEG 4 codec, SP, ASP, AP or AVC */
757 if( !strcmp( tracks[i_track]->psz_codec, "V_MPEG4/ISO/AVC" ) )
758 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'a', 'v', 'c', '1' );
760 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'v' );
761 tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
762 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
763 memcpy( tracks[i_track]->fmt.p_extra,tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
766 else if( !strcmp( tracks[i_track]->psz_codec, "V_QUICKTIME" ) )
768 MP4_Box_t *p_box = (MP4_Box_t*)malloc( sizeof( MP4_Box_t ) );
769 stream_t *p_mp4_stream = stream_MemoryNew( VLC_OBJECT(&sys.demuxer),
770 tracks[i_track]->p_extra_data,
771 tracks[i_track]->i_extra_data,
773 MP4_ReadBoxCommon( p_mp4_stream, p_box );
774 MP4_ReadBox_sample_vide( p_mp4_stream, p_box );
775 tracks[i_track]->fmt.i_codec = p_box->i_type;
776 tracks[i_track]->fmt.video.i_width = p_box->data.p_sample_vide->i_width;
777 tracks[i_track]->fmt.video.i_height = p_box->data.p_sample_vide->i_height;
778 tracks[i_track]->fmt.i_extra = p_box->data.p_sample_vide->i_qt_image_description;
779 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
780 memcpy( tracks[i_track]->fmt.p_extra, p_box->data.p_sample_vide->p_qt_image_description, tracks[i_track]->fmt.i_extra );
781 MP4_FreeBox_sample_vide( p_box );
782 stream_Delete( p_mp4_stream );
784 else if( !strcmp( tracks[i_track]->psz_codec, "A_MS/ACM" ) )
786 if( tracks[i_track]->i_extra_data < (int)sizeof( WAVEFORMATEX ) )
788 msg_Err( &sys.demuxer, "missing/invalid WAVEFORMATEX" );
789 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
793 WAVEFORMATEX *p_wf = (WAVEFORMATEX*)tracks[i_track]->p_extra_data;
795 wf_tag_to_fourcc( GetWLE( &p_wf->wFormatTag ), &tracks[i_track]->fmt.i_codec, NULL );
797 tracks[i_track]->fmt.audio.i_channels = GetWLE( &p_wf->nChannels );
798 tracks[i_track]->fmt.audio.i_rate = GetDWLE( &p_wf->nSamplesPerSec );
799 tracks[i_track]->fmt.i_bitrate = GetDWLE( &p_wf->nAvgBytesPerSec ) * 8;
800 tracks[i_track]->fmt.audio.i_blockalign = GetWLE( &p_wf->nBlockAlign );;
801 tracks[i_track]->fmt.audio.i_bitspersample = GetWLE( &p_wf->wBitsPerSample );
803 tracks[i_track]->fmt.i_extra = GetWLE( &p_wf->cbSize );
804 if( tracks[i_track]->fmt.i_extra > 0 )
806 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
807 memcpy( tracks[i_track]->fmt.p_extra, &p_wf[1], tracks[i_track]->fmt.i_extra );
811 else if( !strcmp( tracks[i_track]->psz_codec, "A_MPEG/L3" ) ||
812 !strcmp( tracks[i_track]->psz_codec, "A_MPEG/L2" ) ||
813 !strcmp( tracks[i_track]->psz_codec, "A_MPEG/L1" ) )
815 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'm', 'p', 'g', 'a' );
817 else if( !strcmp( tracks[i_track]->psz_codec, "A_AC3" ) )
819 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'a', '5', '2', ' ' );
821 else if( !strcmp( tracks[i_track]->psz_codec, "A_EAC3" ) )
823 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'e', 'a', 'c', '3' );
825 else if( !strcmp( tracks[i_track]->psz_codec, "A_DTS" ) )
827 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'd', 't', 's', ' ' );
829 else if( !strcmp( tracks[i_track]->psz_codec, "A_FLAC" ) )
831 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'f', 'l', 'a', 'c' );
832 tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
833 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
834 memcpy( tracks[i_track]->fmt.p_extra,tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
836 else if( !strcmp( tracks[i_track]->psz_codec, "A_VORBIS" ) )
838 int i, i_offset = 1, i_size[3], i_extra;
841 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'v', 'o', 'r', 'b' );
843 /* Split the 3 headers */
844 if( tracks[i_track]->p_extra_data[0] != 0x02 )
845 msg_Err( &sys.demuxer, "invalid vorbis header" );
847 for( i = 0; i < 2; i++ )
850 while( i_offset < tracks[i_track]->i_extra_data )
852 i_size[i] += tracks[i_track]->p_extra_data[i_offset];
853 if( tracks[i_track]->p_extra_data[i_offset++] != 0xff ) break;
857 i_size[0] = __MIN(i_size[0], tracks[i_track]->i_extra_data - i_offset);
858 i_size[1] = __MIN(i_size[1], tracks[i_track]->i_extra_data -i_offset -i_size[0]);
859 i_size[2] = tracks[i_track]->i_extra_data - i_offset - i_size[0] - i_size[1];
861 tracks[i_track]->fmt.i_extra = 3 * 2 + i_size[0] + i_size[1] + i_size[2];
862 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
863 p_extra = (uint8_t *)tracks[i_track]->fmt.p_extra; i_extra = 0;
864 for( i = 0; i < 3; i++ )
866 *(p_extra++) = i_size[i] >> 8;
867 *(p_extra++) = i_size[i] & 0xFF;
868 memcpy( p_extra, tracks[i_track]->p_extra_data + i_offset + i_extra,
870 p_extra += i_size[i];
871 i_extra += i_size[i];
874 else if( !strncmp( tracks[i_track]->psz_codec, "A_AAC/MPEG2/", strlen( "A_AAC/MPEG2/" ) ) ||
875 !strncmp( tracks[i_track]->psz_codec, "A_AAC/MPEG4/", strlen( "A_AAC/MPEG4/" ) ) )
877 int i_profile, i_srate, sbr = 0;
878 static const unsigned int i_sample_rates[] =
880 96000, 88200, 64000, 48000, 44100, 32000, 24000, 22050,
881 16000, 12000, 11025, 8000, 7350, 0, 0, 0
884 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'a' );
885 /* create data for faad (MP4DecSpecificDescrTag)*/
887 if( !strcmp( &tracks[i_track]->psz_codec[12], "MAIN" ) )
891 else if( !strcmp( &tracks[i_track]->psz_codec[12], "LC" ) )
895 else if( !strcmp( &tracks[i_track]->psz_codec[12], "SSR" ) )
899 else if( !strcmp( &tracks[i_track]->psz_codec[12], "LC/SBR" ) )
909 for( i_srate = 0; i_srate < 13; i_srate++ )
911 if( i_sample_rates[i_srate] == tracks[i_track]->i_original_rate )
916 msg_Dbg( &sys.demuxer, "profile=%d srate=%d", i_profile, i_srate );
918 tracks[i_track]->fmt.i_extra = sbr ? 5 : 2;
919 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
920 ((uint8_t*)tracks[i_track]->fmt.p_extra)[0] = ((i_profile + 1) << 3) | ((i_srate&0xe) >> 1);
921 ((uint8_t*)tracks[i_track]->fmt.p_extra)[1] = ((i_srate & 0x1) << 7) | (tracks[i_track]->fmt.audio.i_channels << 3);
924 int syncExtensionType = 0x2B7;
926 for (iDSRI=0; iDSRI<13; iDSRI++)
927 if( i_sample_rates[iDSRI] == tracks[i_track]->fmt.audio.i_rate )
929 ((uint8_t*)tracks[i_track]->fmt.p_extra)[2] = (syncExtensionType >> 3) & 0xFF;
930 ((uint8_t*)tracks[i_track]->fmt.p_extra)[3] = ((syncExtensionType & 0x7) << 5) | 5;
931 ((uint8_t*)tracks[i_track]->fmt.p_extra)[4] = ((1 & 0x1) << 7) | (iDSRI << 3);
934 else if( !strcmp( tracks[i_track]->psz_codec, "A_AAC" ) )
936 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'a' );
937 tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
938 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
939 memcpy( tracks[i_track]->fmt.p_extra, tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
941 else if( !strcmp( tracks[i_track]->psz_codec, "A_WAVPACK4" ) )
943 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'W', 'V', 'P', 'K' );
944 tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
945 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
946 memcpy( tracks[i_track]->fmt.p_extra, tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
948 else if( !strcmp( tracks[i_track]->psz_codec, "A_TTA1" ) )
950 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'T', 'T', 'A', '1' );
951 tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
952 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
953 memcpy( tracks[i_track]->fmt.p_extra, tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
955 else if( !strcmp( tracks[i_track]->psz_codec, "A_PCM/INT/BIG" ) ||
956 !strcmp( tracks[i_track]->psz_codec, "A_PCM/INT/LIT" ) ||
957 !strcmp( tracks[i_track]->psz_codec, "A_PCM/FLOAT/IEEE" ) )
959 if( !strcmp( tracks[i_track]->psz_codec, "A_PCM/INT/BIG" ) )
961 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 't', 'w', 'o', 's' );
965 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'a', 'r', 'a', 'w' );
967 tracks[i_track]->fmt.audio.i_blockalign = ( tracks[i_track]->fmt.audio.i_bitspersample + 7 ) / 8 * tracks[i_track]->fmt.audio.i_channels;
969 /* disabled due to the potential "S_KATE" namespace issue */
970 else if( !strcmp( tracks[i_track]->psz_codec, "S_KATE" ) )
972 int i, i_offset = 1, *i_size, i_extra, num_headers, size_so_far;
975 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'k', 'a', 't', 'e' );
976 tracks[i_track]->fmt.subs.psz_encoding = strdup( "UTF-8" );
978 /* Recover the number of headers to expect */
979 num_headers = tracks[i_track]->p_extra_data[0]+1;
980 msg_Dbg( &sys.demuxer, "kate in mkv detected: %d headers in %u bytes",
981 num_headers, tracks[i_track]->i_extra_data);
983 /* this won't overflow the stack as is can allocate only 1020 bytes max */
984 i_size = (int*)alloca(num_headers*sizeof(int));
986 /* Split the headers */
988 for( i = 0; i < num_headers-1; i++ )
991 while( i_offset < tracks[i_track]->i_extra_data )
993 i_size[i] += tracks[i_track]->p_extra_data[i_offset];
994 if( tracks[i_track]->p_extra_data[i_offset++] != 0xff ) break;
996 msg_Dbg( &sys.demuxer, "kate header %d is %d bytes", i, i_size[i]);
997 size_so_far += i_size[i];
999 i_size[num_headers-1] = tracks[i_track]->i_extra_data - (size_so_far+i_offset);
1000 msg_Dbg( &sys.demuxer, "kate last header (%d) is %d bytes", num_headers-1, i_size[num_headers-1]);
1002 tracks[i_track]->fmt.i_extra = 1 + num_headers * 2 + size_so_far + i_size[num_headers-1];
1003 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
1005 p_extra = (uint8_t *)tracks[i_track]->fmt.p_extra;
1007 *(p_extra++) = num_headers;
1009 for( i = 0; i < num_headers; i++ )
1011 *(p_extra++) = i_size[i] >> 8;
1012 *(p_extra++) = i_size[i] & 0xFF;
1013 memcpy( p_extra, tracks[i_track]->p_extra_data + i_offset + i_extra-1,
1015 p_extra += i_size[i];
1016 i_extra += i_size[i];
1019 else if( !strcmp( tracks[i_track]->psz_codec, "S_TEXT/UTF8" ) )
1021 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 's', 'u', 'b', 't' );
1022 tracks[i_track]->fmt.subs.psz_encoding = strdup( "UTF-8" );
1024 else if( !strcmp( tracks[i_track]->psz_codec, "S_TEXT/USF" ) )
1026 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'u', 's', 'f', ' ' );
1027 tracks[i_track]->fmt.subs.psz_encoding = strdup( "UTF-8" );
1028 if( tracks[i_track]->i_extra_data )
1030 tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
1031 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
1032 memcpy( tracks[i_track]->fmt.p_extra, tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
1035 else if( !strcmp( tracks[i_track]->psz_codec, "S_TEXT/SSA" ) ||
1036 !strcmp( tracks[i_track]->psz_codec, "S_TEXT/ASS" ) ||
1037 !strcmp( tracks[i_track]->psz_codec, "S_SSA" ) ||
1038 !strcmp( tracks[i_track]->psz_codec, "S_ASS" ))
1040 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 's', 's', 'a', ' ' );
1041 tracks[i_track]->fmt.subs.psz_encoding = strdup( "UTF-8" );
1042 if( tracks[i_track]->i_extra_data )
1044 tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
1045 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
1046 memcpy( tracks[i_track]->fmt.p_extra, tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
1049 else if( !strcmp( tracks[i_track]->psz_codec, "S_VOBSUB" ) )
1051 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 's','p','u',' ' );
1052 if( tracks[i_track]->i_extra_data )
1055 char *p_buf = (char *)malloc( tracks[i_track]->i_extra_data + 1);
1056 memcpy( p_buf, tracks[i_track]->p_extra_data , tracks[i_track]->i_extra_data );
1057 p_buf[tracks[i_track]->i_extra_data] = '\0';
1059 p_start = strstr( p_buf, "size:" );
1060 if( sscanf( p_start, "size: %dx%d",
1061 &tracks[i_track]->fmt.subs.spu.i_original_frame_width, &tracks[i_track]->fmt.subs.spu.i_original_frame_height ) == 2 )
1063 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 );
1067 msg_Warn( &sys.demuxer, "reading original frame size for vobsub failed" );
1072 else if( !strcmp( tracks[i_track]->psz_codec, "B_VOBBTN" ) )
1074 tracks[i_track]->fmt.i_cat = NAV_ES;
1079 msg_Err( &sys.demuxer, "unknown codec id=`%s'", tracks[i_track]->psz_codec );
1080 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
1082 if( tracks[i_track]->b_default )
1084 tracks[i_track]->fmt.i_priority = 1000;
1087 tracks[i_track]->p_es = es_out_Add( sys.demuxer.out, &tracks[i_track]->fmt );
1089 /* Turn on a subtitles track if it has been flagged as default -
1090 * but only do this if no subtitles track has already been engaged,
1091 * either by an earlier 'default track' (??) or by default
1092 * language choice behaviour.
1094 if( tracks[i_track]->b_default )
1096 es_out_Control( sys.demuxer.out,
1098 tracks[i_track]->p_es );
1101 es_out_Control( sys.demuxer.out, ES_OUT_SET_NEXT_DISPLAY_TIME, tracks[i_track]->p_es, i_start_time );
1104 sys.i_start_pts = i_start_time;
1105 // reset the stream reading to the first cluster of the segment used
1106 es.I_O().setFilePointer( i_start_pos );
1109 ep = new EbmlParser( &es, segment, &sys.demuxer );
1114 void matroska_segment_c::UnSelect( )
1118 for( i_track = 0; i_track < tracks.size(); i_track++ )
1120 if ( tracks[i_track]->p_es != NULL )
1122 // es_format_Clean( &tracks[i_track]->fmt );
1123 es_out_Del( sys.demuxer.out, tracks[i_track]->p_es );
1124 tracks[i_track]->p_es = NULL;
1131 int matroska_segment_c::BlockGet( KaxBlock * & pp_block, KaxSimpleBlock * & pp_simpleblock, int64_t *pi_ref1, int64_t *pi_ref2, int64_t *pi_duration )
1133 pp_simpleblock = NULL;
1140 EbmlElement *el = NULL;
1144 return VLC_EGENERIC;
1146 if( pp_simpleblock != NULL || ((el = ep->Get()) == NULL && pp_block != NULL) )
1148 /* Check blocks validity to protect againts broken files */
1149 if( BlockFindTrackIndex( NULL, pp_block , pp_simpleblock ) )
1152 pp_simpleblock = NULL;
1157 /* update the index */
1158 #define idx p_indexes[i_index - 1]
1159 if( i_index > 0 && idx.i_time == -1 )
1161 if ( pp_simpleblock != NULL )
1162 idx.i_time = pp_simpleblock->GlobalTimecode() / (mtime_t)1000;
1164 idx.i_time = (*pp_block).GlobalTimecode() / (mtime_t)1000;
1165 idx.b_key = *pi_ref1 == 0 ? true : false;
1171 i_level = ep->GetLevel();
1180 msg_Warn( &sys.demuxer, "EOF" );
1181 return VLC_EGENERIC;
1184 /* Verify that we are still inside our cluster
1185 * It can happens whith broken files and when seeking
1189 if( cluster && !ep->IsTopPresent( cluster ) )
1191 msg_Warn( &sys.demuxer, "Unexpected escape from current cluster" );
1202 if( MKV_IS_ID( el, KaxCluster ) )
1204 cluster = (KaxCluster*)el;
1205 i_cluster_pos = cluster->GetElementPosition();
1207 /* add it to the index */
1209 ( i_index > 0 && p_indexes[i_index - 1].i_position < (int64_t)cluster->GetElementPosition() ) )
1211 IndexAppendCluster( cluster );
1214 // reset silent tracks
1215 for (size_t i=0; i<tracks.size(); i++)
1217 tracks[i]->b_silent = false;
1222 else if( MKV_IS_ID( el, KaxCues ) )
1224 msg_Warn( &sys.demuxer, "find KaxCues FIXME" );
1225 return VLC_EGENERIC;
1229 msg_Dbg( &sys.demuxer, "unknown (%s)", typeid( el ).name() );
1233 if( MKV_IS_ID( el, KaxClusterTimecode ) )
1235 KaxClusterTimecode &ctc = *(KaxClusterTimecode*)el;
1237 ctc.ReadData( es.I_O(), SCOPE_ALL_DATA );
1238 cluster->InitTimecode( uint64( ctc ), i_timescale );
1240 else if( MKV_IS_ID( el, KaxClusterSilentTracks ) )
1244 else if( MKV_IS_ID( el, KaxBlockGroup ) )
1246 i_block_pos = el->GetElementPosition();
1249 else if( MKV_IS_ID( el, KaxSimpleBlock ) )
1251 pp_simpleblock = (KaxSimpleBlock*)el;
1253 pp_simpleblock->ReadData( es.I_O() );
1254 pp_simpleblock->SetParent( *cluster );
1258 if( MKV_IS_ID( el, KaxBlock ) )
1260 pp_block = (KaxBlock*)el;
1262 pp_block->ReadData( es.I_O() );
1263 pp_block->SetParent( *cluster );
1267 else if( MKV_IS_ID( el, KaxBlockDuration ) )
1269 KaxBlockDuration &dur = *(KaxBlockDuration*)el;
1271 dur.ReadData( es.I_O() );
1272 *pi_duration = uint64( dur );
1274 else if( MKV_IS_ID( el, KaxReferenceBlock ) )
1276 KaxReferenceBlock &ref = *(KaxReferenceBlock*)el;
1278 ref.ReadData( es.I_O() );
1281 *pi_ref1 = int64( ref ) * cluster->GlobalTimecodeScale();
1283 else if( *pi_ref2 == 0 )
1285 *pi_ref2 = int64( ref ) * cluster->GlobalTimecodeScale();
1288 else if( MKV_IS_ID( el, KaxClusterSilentTrackNumber ) )
1290 KaxClusterSilentTrackNumber &track_num = *(KaxClusterSilentTrackNumber*)el;
1291 track_num.ReadData( es.I_O() );
1293 for (size_t i=0; i<tracks.size(); i++)
1295 if ( tracks[i]->i_number == uint32(track_num))
1297 tracks[i]->b_silent = true;
1304 msg_Err( &sys.demuxer, "invalid level = %d", i_level );
1305 return VLC_EGENERIC;