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 #include "../vobsub.h"
35 #include <vlc_codecs.h>
37 /* GetFourCC helper */
38 #define GetFOURCC( p ) __GetFOURCC( (uint8_t*)p )
39 static vlc_fourcc_t __GetFOURCC( uint8_t *p )
41 return VLC_FOURCC( p[0], p[1], p[2], p[3] );
45 matroska_segment_c::~matroska_segment_c()
47 for( size_t i_track = 0; i_track < tracks.size(); i_track++ )
49 delete tracks[i_track]->p_compression_data;
50 es_format_Clean( &tracks[i_track]->fmt );
51 free( tracks[i_track]->p_extra_data );
52 free( tracks[i_track]->psz_codec );
53 delete tracks[i_track];
56 free( psz_writing_application );
57 free( psz_muxing_application );
58 free( psz_segment_filename );
66 delete p_prev_segment_uid;
67 delete p_next_segment_uid;
69 std::vector<chapter_edition_c*>::iterator index = stored_editions.begin();
70 while ( index != stored_editions.end() )
75 std::vector<chapter_translation_c*>::iterator indext = translations.begin();
76 while ( indext != translations.end() )
81 std::vector<KaxSegmentFamily*>::iterator indexf = families.begin();
82 while ( indexf != families.end() )
90 /*****************************************************************************
92 * * LoadCues : load the cues element and update index
94 * * LoadTags : load ... the tags element
96 * * InformationCreate : create all information, load tags if present
98 *****************************************************************************/
99 void matroska_segment_c::LoadCues( KaxCues *cues )
106 msg_Err( &sys.demuxer, "There can be only 1 Cues per section." );
110 ep = new EbmlParser( &es, cues, &sys.demuxer );
111 while( ( el = ep->Get() ) != NULL )
113 if( MKV_IS_ID( el, KaxCuePoint ) )
115 #define idx p_indexes[i_index]
118 idx.i_block_number= -1;
124 while( ( el = ep->Get() ) != NULL )
126 if( MKV_IS_ID( el, KaxCueTime ) )
128 KaxCueTime &ctime = *(KaxCueTime*)el;
130 ctime.ReadData( es.I_O() );
132 idx.i_time = uint64( ctime ) * i_timescale / (mtime_t)1000;
134 else if( MKV_IS_ID( el, KaxCueTrackPositions ) )
137 while( ( el = ep->Get() ) != NULL )
139 if( MKV_IS_ID( el, KaxCueTrack ) )
141 KaxCueTrack &ctrack = *(KaxCueTrack*)el;
143 ctrack.ReadData( es.I_O() );
144 idx.i_track = uint16( ctrack );
146 else if( MKV_IS_ID( el, KaxCueClusterPosition ) )
148 KaxCueClusterPosition &ccpos = *(KaxCueClusterPosition*)el;
150 ccpos.ReadData( es.I_O() );
151 idx.i_position = segment->GetGlobalPosition( uint64( ccpos ) );
153 else if( MKV_IS_ID( el, KaxCueBlockNumber ) )
155 KaxCueBlockNumber &cbnum = *(KaxCueBlockNumber*)el;
157 cbnum.ReadData( es.I_O() );
158 idx.i_block_number = uint32( cbnum );
162 msg_Dbg( &sys.demuxer, " * Unknown (%s)", typeid(*el).name() );
169 msg_Dbg( &sys.demuxer, " * Unknown (%s)", typeid(*el).name() );
175 msg_Dbg( &sys.demuxer, " * added time=%"PRId64" pos=%"PRId64
176 " track=%d bnum=%d", idx.i_time, idx.i_position,
177 idx.i_track, idx.i_block_number );
181 if( i_index >= i_index_max )
184 p_indexes = (mkv_index_t*)xrealloc( p_indexes,
185 sizeof( mkv_index_t ) * i_index_max );
191 msg_Dbg( &sys.demuxer, " * Unknown (%s)", typeid(*el).name() );
196 msg_Dbg( &sys.demuxer, "| - loading cues done." );
200 #define PARSE_TAG( type ) \
202 msg_Dbg( &sys.demuxer, "| + " type ); \
204 while( ( el = ep->Get() ) != NULL ) \
206 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid( *el ).name() ); \
208 ep->Up(); } while( 0 )
210 void matroska_segment_c::LoadTags( KaxTags *tags )
212 /* Master elements */
213 EbmlParser *ep = new EbmlParser( &es, tags, &sys.demuxer );
216 while( ( el = ep->Get() ) != NULL )
218 if( MKV_IS_ID( el, KaxTag ) )
220 msg_Dbg( &sys.demuxer, "+ Tag" );
222 while( ( el = ep->Get() ) != NULL )
224 if( MKV_IS_ID( el, KaxTagTargets ) )
225 PARSE_TAG( "Targets" );
226 else if( MKV_IS_ID( el, KaxTagGeneral ) )
227 PARSE_TAG( "General" );
228 else if( MKV_IS_ID( el, KaxTagGenres ) )
229 PARSE_TAG( "Genres" );
230 else if( MKV_IS_ID( el, KaxTagAudioSpecific ) )
231 PARSE_TAG( "Audio Specific" );
232 else if( MKV_IS_ID( el, KaxTagImageSpecific ) )
233 PARSE_TAG( "Images Specific" );
234 else if( MKV_IS_ID( el, KaxTagMultiComment ) )
236 msg_Dbg( &sys.demuxer, "| + Multi Comment" );
238 else if( MKV_IS_ID( el, KaxTagMultiCommercial ) )
240 msg_Dbg( &sys.demuxer, "| + Multi Commercial" );
242 else if( MKV_IS_ID( el, KaxTagMultiDate ) )
244 msg_Dbg( &sys.demuxer, "| + Multi Date" );
246 else if( MKV_IS_ID( el, KaxTagMultiEntity ) )
248 msg_Dbg( &sys.demuxer, "| + Multi Entity" );
250 else if( MKV_IS_ID( el, KaxTagMultiIdentifier ) )
252 msg_Dbg( &sys.demuxer, "| + Multi Identifier" );
254 else if( MKV_IS_ID( el, KaxTagMultiLegal ) )
256 msg_Dbg( &sys.demuxer, "| + Multi Legal" );
258 else if( MKV_IS_ID( el, KaxTagMultiTitle ) )
260 msg_Dbg( &sys.demuxer, "| + Multi Title" );
264 msg_Dbg( &sys.demuxer, "| + LoadTag Unknown (%s)", typeid( *el ).name() );
271 msg_Dbg( &sys.demuxer, "+ Unknown (%s)", typeid( *el ).name() );
276 msg_Dbg( &sys.demuxer, "loading tags done." );
280 /*****************************************************************************
282 *****************************************************************************/
283 void matroska_segment_c::InformationCreate( )
285 sys.meta = vlc_meta_New();
289 vlc_meta_SetTitle( sys.meta, psz_title );
293 vlc_meta_SetDate( sys.meta, psz_date_utc );
296 if( psz_segment_filename )
298 fprintf( stderr, "***** WARNING: Unhandled meta - Use custom\n" );
300 if( psz_muxing_application )
302 fprintf( stderr, "***** WARNING: Unhandled meta - Use custom\n" );
304 if( psz_writing_application )
306 fprintf( stderr, "***** WARNING: Unhandled meta - Use custom\n" );
309 for( size_t i_track = 0; i_track < tracks.size(); i_track++ )
311 // mkv_track_t *tk = tracks[i_track];
312 // vlc_meta_t *mtk = vlc_meta_New();
313 fprintf( stderr, "***** WARNING: Unhandled child meta\n");
317 if( i_tags_position >= 0 )
321 stream_Control( sys.demuxer.s, STREAM_CAN_FASTSEEK, &b_seekable );
331 /*****************************************************************************
333 *****************************************************************************/
335 void matroska_segment_c::IndexAppendCluster( KaxCluster *cluster )
337 #define idx p_indexes[i_index]
339 idx.i_block_number= -1;
340 idx.i_position = cluster->GetElementPosition();
345 if( i_index >= i_index_max )
348 p_indexes = (mkv_index_t*)xrealloc( p_indexes,
349 sizeof( mkv_index_t ) * i_index_max );
355 bool matroska_segment_c::PreloadFamily( const matroska_segment_c & of_segment )
360 for (size_t i=0; i<families.size(); i++)
362 for (size_t j=0; j<of_segment.families.size(); j++)
364 if ( *(families[i]) == *(of_segment.families[j]) )
372 bool matroska_segment_c::CompareSegmentUIDs( const matroska_segment_c * p_item_a, const matroska_segment_c * p_item_b )
376 if ( p_item_a == NULL || p_item_b == NULL )
379 p_tmp = (EbmlBinary *)p_item_a->p_segment_uid;
380 if ( p_item_b->p_prev_segment_uid != NULL
381 && *p_tmp == *p_item_b->p_prev_segment_uid )
384 p_tmp = (EbmlBinary *)p_item_a->p_next_segment_uid;
388 if ( p_item_b->p_segment_uid != NULL
389 && *p_tmp == *p_item_b->p_segment_uid )
392 if ( p_item_b->p_prev_segment_uid != NULL
393 && *p_tmp == *p_item_b->p_prev_segment_uid )
399 bool matroska_segment_c::Preload( )
404 EbmlElement *el = NULL;
406 ep->Reset( &sys.demuxer );
408 while( ( el = ep->Get() ) != NULL )
410 if( MKV_IS_ID( el, KaxSeekHead ) )
412 /* Multiple allowed */
413 /* We bail at 10, to prevent possible recursion */
414 msg_Dbg( &sys.demuxer, "| + Seek head" );
415 if( i_seekhead_count < 10 )
417 i_seekhead_position = (int64_t) es.I_O().getFilePointer();
418 ParseSeekHead( static_cast<KaxSeekHead*>( el ) );
421 else if( MKV_IS_ID( el, KaxInfo ) )
423 /* Multiple allowed, mandatory */
424 msg_Dbg( &sys.demuxer, "| + Information" );
425 if( i_info_position < 0 ) // FIXME
426 ParseInfo( static_cast<KaxInfo*>( el ) );
427 i_info_position = (int64_t) es.I_O().getFilePointer();
429 else if( MKV_IS_ID( el, KaxTracks ) )
431 /* Multiple allowed */
432 msg_Dbg( &sys.demuxer, "| + Tracks" );
433 if( i_tracks_position < 0 ) // FIXME
434 ParseTracks( static_cast<KaxTracks*>( el ) );
435 if ( tracks.size() == 0 )
437 msg_Err( &sys.demuxer, "No tracks supported" );
440 i_tracks_position = (int64_t) es.I_O().getFilePointer();
442 else if( MKV_IS_ID( el, KaxCues ) )
444 msg_Dbg( &sys.demuxer, "| + Cues" );
445 if( i_cues_position < 0 )
446 LoadCues( static_cast<KaxCues*>( el ) );
447 i_cues_position = (int64_t) es.I_O().getFilePointer();
449 else if( MKV_IS_ID( el, KaxCluster ) )
451 msg_Dbg( &sys.demuxer, "| + Cluster" );
453 cluster = (KaxCluster*)el;
455 i_cluster_pos = i_start_pos = cluster->GetElementPosition();
459 /* stop pre-parsing the stream */
462 else if( MKV_IS_ID( el, KaxAttachments ) )
464 msg_Dbg( &sys.demuxer, "| + Attachments" );
465 if( i_attachments_position < 0 )
466 ParseAttachments( static_cast<KaxAttachments*>( el ) );
467 i_attachments_position = (int64_t) es.I_O().getFilePointer();
469 else if( MKV_IS_ID( el, KaxChapters ) )
471 msg_Dbg( &sys.demuxer, "| + Chapters" );
472 if( i_chapters_position < 0 )
473 ParseChapters( static_cast<KaxChapters*>( el ) );
474 i_chapters_position = (int64_t) es.I_O().getFilePointer();
476 else if( MKV_IS_ID( el, KaxTag ) )
478 msg_Dbg( &sys.demuxer, "| + Tags" );
479 if( i_tags_position < 0) // FIXME
480 ;//LoadTags( static_cast<KaxTags*>( el ) );
481 i_tags_position = (int64_t) es.I_O().getFilePointer();
484 msg_Dbg( &sys.demuxer, "| + Preload Unknown (%s)", typeid(*el).name() );
492 /* Here we try to load elements that were found in Seek Heads, but not yet parsed */
493 bool matroska_segment_c::LoadSeekHeadItem( const EbmlCallbacks & ClassInfos, int64_t i_element_position )
495 int64_t i_sav_position = (int64_t)es.I_O().getFilePointer();
498 es.I_O().setFilePointer( i_element_position, seek_beginning );
499 el = es.FindNextID( ClassInfos, 0xFFFFFFFFL);
503 msg_Err( &sys.demuxer, "cannot load some cues/chapters/tags etc. (broken seekhead or file)" );
504 es.I_O().setFilePointer( i_sav_position, seek_beginning );
508 if( MKV_IS_ID( el, KaxSeekHead ) )
510 /* Multiple allowed */
511 msg_Dbg( &sys.demuxer, "| + Seek head" );
512 if( i_seekhead_count < 10 )
514 i_seekhead_position = i_element_position;
515 ParseSeekHead( static_cast<KaxSeekHead*>( el ) );
518 else if( MKV_IS_ID( el, KaxInfo ) ) // FIXME
520 /* Multiple allowed, mandatory */
521 msg_Dbg( &sys.demuxer, "| + Information" );
522 if( i_info_position < 0 )
523 ParseInfo( static_cast<KaxInfo*>( el ) );
524 i_info_position = i_element_position;
526 else if( MKV_IS_ID( el, KaxTracks ) ) // FIXME
528 /* Multiple allowed */
529 msg_Dbg( &sys.demuxer, "| + Tracks" );
530 if( i_tracks_position < 0 )
531 ParseTracks( static_cast<KaxTracks*>( el ) );
532 if ( tracks.size() == 0 )
534 msg_Err( &sys.demuxer, "No tracks supported" );
536 es.I_O().setFilePointer( i_sav_position, seek_beginning );
539 i_tracks_position = i_element_position;
541 else if( MKV_IS_ID( el, KaxCues ) )
543 msg_Dbg( &sys.demuxer, "| + Cues" );
544 if( i_cues_position < 0 )
545 LoadCues( static_cast<KaxCues*>( el ) );
546 i_cues_position = i_element_position;
548 else if( MKV_IS_ID( el, KaxAttachments ) )
550 msg_Dbg( &sys.demuxer, "| + Attachments" );
551 if( i_attachments_position < 0 )
552 ParseAttachments( static_cast<KaxAttachments*>( el ) );
553 i_attachments_position = i_element_position;
555 else if( MKV_IS_ID( el, KaxChapters ) )
557 msg_Dbg( &sys.demuxer, "| + Chapters" );
558 if( i_chapters_position < 0 )
559 ParseChapters( static_cast<KaxChapters*>( el ) );
560 i_chapters_position = i_element_position;
562 else if( MKV_IS_ID( el, KaxTag ) ) // FIXME
564 msg_Dbg( &sys.demuxer, "| + Tags" );
565 if( i_tags_position < 0 )
566 ;//LoadTags( static_cast<KaxTags*>( el ) );
567 i_tags_position = i_element_position;
571 msg_Dbg( &sys.demuxer, "| + LoadSeekHeadItem Unknown (%s)", typeid(*el).name() );
575 es.I_O().setFilePointer( i_sav_position, seek_beginning );
579 void matroska_segment_c::Seek( mtime_t i_date, mtime_t i_time_offset, int64_t i_global_position )
582 KaxSimpleBlock *simpleblock;
583 int i_track_skipping;
584 int64_t i_block_duration;
586 int64_t i_seek_position = i_start_pos;
587 int64_t i_seek_time = i_start_time;
589 if( i_global_position >= 0 )
591 /* Special case for seeking in files with no cues */
592 EbmlElement *el = NULL;
593 es.I_O().setFilePointer( i_start_pos, seek_beginning );
595 ep = new EbmlParser( &es, segment, &sys.demuxer );
598 while( ( el = ep->Get() ) != NULL )
600 if( MKV_IS_ID( el, KaxCluster ) )
602 cluster = (KaxCluster *)el;
603 i_cluster_pos = cluster->GetElementPosition();
605 ( i_index > 0 && p_indexes[i_index - 1].i_position < (int64_t)cluster->GetElementPosition() ) )
607 IndexAppendCluster( cluster );
609 if( es.I_O().getFilePointer() >= i_global_position )
612 msg_Dbg( &sys.demuxer, "we found a cluster that is in the neighbourhood" );
617 msg_Err( &sys.demuxer, "This file has no cues, and we were unable to seek to the requested position by parsing." );
625 for( ; i_idx < i_index; i_idx++ )
627 if( p_indexes[i_idx].i_time + i_time_offset > i_date )
638 i_seek_position = p_indexes[i_idx].i_position;
639 i_seek_time = p_indexes[i_idx].i_time;
642 msg_Dbg( &sys.demuxer, "seek got %"PRId64" (%d%%)",
643 i_seek_time, (int)( 100 * i_seek_position / stream_Size( sys.demuxer.s ) ) );
645 es.I_O().setFilePointer( i_seek_position, seek_beginning );
648 ep = new EbmlParser( &es, segment, &sys.demuxer );
651 sys.i_start_pts = i_date;
653 /* now parse until key frame */
654 i_track_skipping = 0;
655 for( i_track = 0; i_track < tracks.size(); i_track++ )
657 if( tracks[i_track]->fmt.i_cat == VIDEO_ES )
659 tracks[i_track]->b_search_keyframe = true;
663 es_out_Control( sys.demuxer.out, ES_OUT_SET_NEXT_DISPLAY_TIME, i_date );
665 while( i_track_skipping > 0 )
668 bool b_discardable_picture;
669 if( BlockGet( block, simpleblock, &b_key_picture, &b_discardable_picture, &i_block_duration ) )
671 msg_Warn( &sys.demuxer, "cannot get block EOF?" );
676 for( i_track = 0; i_track < tracks.size(); i_track++ )
678 if( (simpleblock && tracks[i_track]->i_number == simpleblock->TrackNum()) ||
679 (block && tracks[i_track]->i_number == block->TrackNum()) )
686 sys.i_pts = (sys.i_chapter_time + simpleblock->GlobalTimecode()) / (mtime_t) 1000;
688 sys.i_pts = (sys.i_chapter_time + block->GlobalTimecode()) / (mtime_t) 1000;
690 if( i_track < tracks.size() )
692 if( sys.i_pts > sys.i_start_pts )
694 cluster = static_cast<KaxCluster*>(ep->UnGet( i_block_pos, i_cluster_pos ));
695 i_track_skipping = 0;
697 else if( tracks[i_track]->fmt.i_cat == VIDEO_ES )
699 if( b_key_picture && tracks[i_track]->b_search_keyframe )
701 tracks[i_track]->b_search_keyframe = false;
704 if( !tracks[i_track]->b_search_keyframe )
706 BlockDecode( &sys.demuxer, block, simpleblock, sys.i_pts, 0, b_key_picture || b_discardable_picture );
714 /* FIXME current ES_OUT_SET_NEXT_DISPLAY_TIME does not work that well if
715 * the delay is too high. */
716 if( sys.i_pts + 500*1000 < sys.i_start_pts )
718 sys.i_start_pts = sys.i_pts;
720 es_out_Control( sys.demuxer.out, ES_OUT_SET_NEXT_DISPLAY_TIME, sys.i_start_pts );
725 int matroska_segment_c::BlockFindTrackIndex( size_t *pi_track,
726 const KaxBlock *p_block, const KaxSimpleBlock *p_simpleblock )
730 for( i_track = 0; i_track < tracks.size(); i_track++ )
732 const mkv_track_t *tk = tracks[i_track];
734 if( ( p_block != NULL && tk->i_number == p_block->TrackNum() ) ||
735 ( p_simpleblock != NULL && tk->i_number == p_simpleblock->TrackNum() ) )
741 if( i_track >= tracks.size() )
749 bool matroska_segment_c::Select( mtime_t i_start_time )
754 msg_Dbg( &sys.demuxer, "found %d es", (int)tracks.size() );
755 sys.b_pci_packet_set = false;
757 for( i_track = 0; i_track < tracks.size(); i_track++ )
759 mkv_track_t *p_tk = tracks[i_track];
760 es_format_t *p_fmt = &p_tk->fmt;
762 if( p_fmt->i_cat == UNKNOWN_ES || !p_tk->psz_codec )
764 msg_Warn( &sys.demuxer, "invalid track[%d, n=%d]", (int)i_track, p_tk->i_number );
769 if( !strcmp( p_tk->psz_codec, "V_MS/VFW/FOURCC" ) )
771 if( p_tk->i_extra_data < (int)sizeof( BITMAPINFOHEADER ) )
773 msg_Err( &sys.demuxer, "missing/invalid BITMAPINFOHEADER" );
774 p_tk->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
778 BITMAPINFOHEADER *p_bih = (BITMAPINFOHEADER*)p_tk->p_extra_data;
780 p_tk->fmt.video.i_width = GetDWLE( &p_bih->biWidth );
781 p_tk->fmt.video.i_height= GetDWLE( &p_bih->biHeight );
782 p_tk->fmt.i_codec = GetFOURCC( &p_bih->biCompression );
784 p_tk->fmt.i_extra = GetDWLE( &p_bih->biSize ) - sizeof( BITMAPINFOHEADER );
785 if( p_tk->fmt.i_extra > 0 )
787 p_tk->fmt.p_extra = xmalloc( p_tk->fmt.i_extra );
788 memcpy( p_tk->fmt.p_extra, &p_bih[1], p_tk->fmt.i_extra );
791 p_tk->b_dts_only = true;
793 else if( !strcmp( p_tk->psz_codec, "V_MPEG1" ) ||
794 !strcmp( p_tk->psz_codec, "V_MPEG2" ) )
796 p_tk->fmt.i_codec = VLC_CODEC_MPGV;
798 else if( !strncmp( p_tk->psz_codec, "V_THEORA", 8 ) )
800 p_tk->fmt.i_codec = VLC_CODEC_THEORA;
801 p_tk->fmt.i_extra = p_tk->i_extra_data;
802 p_tk->fmt.p_extra = xmalloc( p_tk->i_extra_data );
803 memcpy( p_tk->fmt.p_extra,p_tk->p_extra_data, p_tk->i_extra_data );
804 p_tk->b_pts_only = true;
806 else if( !strncmp( p_tk->psz_codec, "V_REAL/RV", 9 ) )
808 if( !strcmp( p_tk->psz_codec, "V_REAL/RV10" ) )
809 p_fmt->i_codec = VLC_CODEC_RV10;
810 else if( !strcmp( p_tk->psz_codec, "V_REAL/RV20" ) )
811 p_fmt->i_codec = VLC_CODEC_RV20;
812 else if( !strcmp( p_tk->psz_codec, "V_REAL/RV30" ) )
813 p_fmt->i_codec = VLC_CODEC_RV30;
814 else if( !strcmp( p_tk->psz_codec, "V_REAL/RV40" ) )
815 p_fmt->i_codec = VLC_CODEC_RV40;
817 if( p_tk->i_extra_data > 26 )
819 p_fmt->p_extra = malloc( p_tk->i_extra_data - 26 );
822 p_fmt->i_extra = p_tk->i_extra_data - 26;
823 memcpy( p_fmt->p_extra, &p_tk->p_extra_data[26], p_fmt->i_extra );
826 p_tk->b_dts_only = true;
828 else if( !strncmp( p_tk->psz_codec, "V_DIRAC", 7 ) )
830 p_tk->fmt.i_codec = VLC_CODEC_DIRAC;
832 else if( !strncmp( p_tk->psz_codec, "V_VP8", 5 ) )
834 p_tk->fmt.i_codec = VLC_CODEC_VP8;
835 p_tk->b_pts_only = true;
837 else if( !strncmp( p_tk->psz_codec, "V_MPEG4", 7 ) )
839 if( !strcmp( p_tk->psz_codec, "V_MPEG4/MS/V3" ) )
841 p_tk->fmt.i_codec = VLC_CODEC_DIV3;
843 else if( !strncmp( p_tk->psz_codec, "V_MPEG4/ISO", 11 ) )
845 /* A MPEG 4 codec, SP, ASP, AP or AVC */
846 if( !strcmp( p_tk->psz_codec, "V_MPEG4/ISO/AVC" ) )
847 p_tk->fmt.i_codec = VLC_FOURCC( 'a', 'v', 'c', '1' );
849 p_tk->fmt.i_codec = VLC_CODEC_MP4V;
850 p_tk->fmt.i_extra = p_tk->i_extra_data;
851 p_tk->fmt.p_extra = xmalloc( p_tk->i_extra_data );
852 memcpy( p_tk->fmt.p_extra,p_tk->p_extra_data, p_tk->i_extra_data );
855 else if( !strcmp( p_tk->psz_codec, "V_QUICKTIME" ) )
857 MP4_Box_t *p_box = (MP4_Box_t*)xmalloc( sizeof( MP4_Box_t ) );
858 stream_t *p_mp4_stream = stream_MemoryNew( VLC_OBJECT(&sys.demuxer),
862 if( MP4_ReadBoxCommon( p_mp4_stream, p_box ) &&
863 MP4_ReadBox_sample_vide( p_mp4_stream, p_box ) )
865 p_tk->fmt.i_codec = p_box->i_type;
866 p_tk->fmt.video.i_width = p_box->data.p_sample_vide->i_width;
867 p_tk->fmt.video.i_height = p_box->data.p_sample_vide->i_height;
868 p_tk->fmt.i_extra = p_box->data.p_sample_vide->i_qt_image_description;
869 p_tk->fmt.p_extra = xmalloc( p_tk->fmt.i_extra );
870 memcpy( p_tk->fmt.p_extra, p_box->data.p_sample_vide->p_qt_image_description, p_tk->fmt.i_extra );
871 MP4_FreeBox_sample_vide( p_box );
877 stream_Delete( p_mp4_stream );
879 else if( !strcmp( p_tk->psz_codec, "A_MS/ACM" ) )
881 if( p_tk->i_extra_data < (int)sizeof( WAVEFORMATEX ) )
883 msg_Err( &sys.demuxer, "missing/invalid WAVEFORMATEX" );
884 p_tk->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
888 WAVEFORMATEX *p_wf = (WAVEFORMATEX*)p_tk->p_extra_data;
890 wf_tag_to_fourcc( GetWLE( &p_wf->wFormatTag ), &p_tk->fmt.i_codec, NULL );
892 p_tk->fmt.audio.i_channels = GetWLE( &p_wf->nChannels );
893 p_tk->fmt.audio.i_rate = GetDWLE( &p_wf->nSamplesPerSec );
894 p_tk->fmt.i_bitrate = GetDWLE( &p_wf->nAvgBytesPerSec ) * 8;
895 p_tk->fmt.audio.i_blockalign = GetWLE( &p_wf->nBlockAlign );;
896 p_tk->fmt.audio.i_bitspersample = GetWLE( &p_wf->wBitsPerSample );
898 p_tk->fmt.i_extra = GetWLE( &p_wf->cbSize );
899 if( p_tk->fmt.i_extra > 0 )
901 p_tk->fmt.p_extra = xmalloc( p_tk->fmt.i_extra );
902 memcpy( p_tk->fmt.p_extra, &p_wf[1], p_tk->fmt.i_extra );
906 else if( !strcmp( p_tk->psz_codec, "A_MPEG/L3" ) ||
907 !strcmp( p_tk->psz_codec, "A_MPEG/L2" ) ||
908 !strcmp( p_tk->psz_codec, "A_MPEG/L1" ) )
910 p_tk->fmt.i_codec = VLC_CODEC_MPGA;
912 else if( !strcmp( p_tk->psz_codec, "A_AC3" ) )
914 p_tk->fmt.i_codec = VLC_CODEC_A52;
916 else if( !strcmp( p_tk->psz_codec, "A_EAC3" ) )
918 p_tk->fmt.i_codec = VLC_CODEC_EAC3;
920 else if( !strcmp( p_tk->psz_codec, "A_DTS" ) )
922 p_tk->fmt.i_codec = VLC_CODEC_DTS;
924 else if( !strcmp( p_tk->psz_codec, "A_MLP" ) )
926 p_tk->fmt.i_codec = VLC_CODEC_MLP;
928 else if( !strcmp( p_tk->psz_codec, "A_TRUEHD" ) )
930 /* FIXME when more samples arrive */
931 p_tk->fmt.i_codec = VLC_CODEC_TRUEHD;
932 p_fmt->b_packetized = false;
934 else if( !strcmp( p_tk->psz_codec, "A_FLAC" ) )
936 p_tk->fmt.i_codec = VLC_CODEC_FLAC;
937 p_tk->fmt.i_extra = p_tk->i_extra_data;
938 p_tk->fmt.p_extra = xmalloc( p_tk->i_extra_data );
939 memcpy( p_tk->fmt.p_extra,p_tk->p_extra_data, p_tk->i_extra_data );
941 else if( !strcmp( p_tk->psz_codec, "A_VORBIS" ) )
943 p_tk->fmt.i_codec = VLC_CODEC_VORBIS;
944 p_tk->fmt.i_extra = p_tk->i_extra_data;
945 p_tk->fmt.p_extra = xmalloc( p_tk->i_extra_data );
946 memcpy( p_tk->fmt.p_extra,p_tk->p_extra_data, p_tk->i_extra_data );
948 else if( !strncmp( p_tk->psz_codec, "A_AAC/MPEG2/", strlen( "A_AAC/MPEG2/" ) ) ||
949 !strncmp( p_tk->psz_codec, "A_AAC/MPEG4/", strlen( "A_AAC/MPEG4/" ) ) )
951 int i_profile, i_srate, sbr = 0;
952 static const unsigned int i_sample_rates[] =
954 96000, 88200, 64000, 48000, 44100, 32000, 24000, 22050,
955 16000, 12000, 11025, 8000, 7350, 0, 0, 0
958 p_tk->fmt.i_codec = VLC_CODEC_MP4A;
959 /* create data for faad (MP4DecSpecificDescrTag)*/
961 if( !strcmp( &p_tk->psz_codec[12], "MAIN" ) )
965 else if( !strcmp( &p_tk->psz_codec[12], "LC" ) )
969 else if( !strcmp( &p_tk->psz_codec[12], "SSR" ) )
973 else if( !strcmp( &p_tk->psz_codec[12], "LC/SBR" ) )
983 for( i_srate = 0; i_srate < 13; i_srate++ )
985 if( i_sample_rates[i_srate] == p_tk->i_original_rate )
990 msg_Dbg( &sys.demuxer, "profile=%d srate=%d", i_profile, i_srate );
992 p_tk->fmt.i_extra = sbr ? 5 : 2;
993 p_tk->fmt.p_extra = xmalloc( p_tk->fmt.i_extra );
994 ((uint8_t*)p_tk->fmt.p_extra)[0] = ((i_profile + 1) << 3) | ((i_srate&0xe) >> 1);
995 ((uint8_t*)p_tk->fmt.p_extra)[1] = ((i_srate & 0x1) << 7) | (p_tk->fmt.audio.i_channels << 3);
998 int syncExtensionType = 0x2B7;
1000 for (iDSRI=0; iDSRI<13; iDSRI++)
1001 if( i_sample_rates[iDSRI] == p_tk->fmt.audio.i_rate )
1003 ((uint8_t*)p_tk->fmt.p_extra)[2] = (syncExtensionType >> 3) & 0xFF;
1004 ((uint8_t*)p_tk->fmt.p_extra)[3] = ((syncExtensionType & 0x7) << 5) | 5;
1005 ((uint8_t*)p_tk->fmt.p_extra)[4] = ((1 & 0x1) << 7) | (iDSRI << 3);
1008 else if( !strcmp( p_tk->psz_codec, "A_AAC" ) )
1010 p_tk->fmt.i_codec = VLC_CODEC_MP4A;
1011 p_tk->fmt.i_extra = p_tk->i_extra_data;
1012 p_tk->fmt.p_extra = xmalloc( p_tk->i_extra_data );
1013 memcpy( p_tk->fmt.p_extra, p_tk->p_extra_data, p_tk->i_extra_data );
1015 else if( !strcmp( p_tk->psz_codec, "A_WAVPACK4" ) )
1017 p_tk->fmt.i_codec = VLC_CODEC_WAVPACK;
1018 p_tk->fmt.i_extra = p_tk->i_extra_data;
1019 p_tk->fmt.p_extra = xmalloc( p_tk->i_extra_data );
1020 memcpy( p_tk->fmt.p_extra, p_tk->p_extra_data, p_tk->i_extra_data );
1022 else if( !strcmp( p_tk->psz_codec, "A_TTA1" ) )
1024 p_fmt->i_codec = VLC_CODEC_TTA;
1025 p_fmt->i_extra = p_tk->i_extra_data;
1026 if( p_fmt->i_extra > 0 )
1028 p_fmt->p_extra = xmalloc( p_tk->i_extra_data );
1029 memcpy( p_fmt->p_extra, p_tk->p_extra_data, p_tk->i_extra_data );
1033 p_fmt->i_extra = 30;
1034 p_fmt->p_extra = xmalloc( p_fmt->i_extra );
1035 uint8_t *p_extra = (uint8_t*)p_fmt->p_extra;
1036 memcpy( &p_extra[ 0], "TTA1", 4 );
1037 SetWLE( &p_extra[ 4], 1 );
1038 SetWLE( &p_extra[ 6], p_fmt->audio.i_channels );
1039 SetWLE( &p_extra[ 8], p_fmt->audio.i_bitspersample );
1040 SetDWLE( &p_extra[10], p_fmt->audio.i_rate );
1041 SetDWLE( &p_extra[14], 0xffffffff );
1042 memset( &p_extra[18], 0, 30 - 18 );
1045 else if( !strcmp( p_tk->psz_codec, "A_PCM/INT/BIG" ) ||
1046 !strcmp( p_tk->psz_codec, "A_PCM/INT/LIT" ) ||
1047 !strcmp( p_tk->psz_codec, "A_PCM/FLOAT/IEEE" ) )
1049 if( !strcmp( p_tk->psz_codec, "A_PCM/INT/BIG" ) )
1051 p_tk->fmt.i_codec = VLC_FOURCC( 't', 'w', 'o', 's' );
1055 p_tk->fmt.i_codec = VLC_FOURCC( 'a', 'r', 'a', 'w' );
1057 p_tk->fmt.audio.i_blockalign = ( p_tk->fmt.audio.i_bitspersample + 7 ) / 8 * p_tk->fmt.audio.i_channels;
1059 else if( !strncmp( p_tk->psz_codec, "A_REAL/", 7 ) )
1061 if( !strcmp( p_tk->psz_codec, "A_REAL/COOK" ) )
1062 p_tk->fmt.i_codec = VLC_CODEC_COOK;
1063 else if( !strcmp( p_tk->psz_codec, "A_REAL/ATRC" ) )
1064 p_tk->fmt.i_codec = VLC_CODEC_ATRAC3;
1065 else if( !strcmp( p_tk->psz_codec, "A_REAL/28_8" ) )
1066 p_tk->fmt.i_codec = VLC_CODEC_RA_288;
1067 /* FIXME 14_4, RALF and SIPR */
1068 p_tk->fmt.i_extra = p_tk->i_extra_data;
1069 p_tk->fmt.p_extra = xmalloc( p_tk->i_extra_data );
1070 memcpy( p_tk->fmt.p_extra,p_tk->p_extra_data, p_tk->i_extra_data );
1072 /* disabled due to the potential "S_KATE" namespace issue */
1073 else if( !strcmp( p_tk->psz_codec, "S_KATE" ) )
1075 int i, i_offset = 1, i_extra, num_headers, size_so_far;
1078 p_tk->fmt.i_codec = VLC_CODEC_KATE;
1079 p_tk->fmt.subs.psz_encoding = strdup( "UTF-8" );
1081 p_tk->fmt.i_extra = p_tk->i_extra_data;
1082 p_tk->fmt.p_extra = xmalloc( p_tk->i_extra_data );
1083 memcpy( p_tk->fmt.p_extra,p_tk->p_extra_data, p_tk->i_extra_data );
1085 else if( !strcmp( p_tk->psz_codec, "S_TEXT/ASCII" ) )
1087 p_fmt->i_codec = VLC_CODEC_SUBT;
1088 p_fmt->subs.psz_encoding = NULL; /* Is there a place where it is stored ? */
1090 else if( !strcmp( p_tk->psz_codec, "S_TEXT/UTF8" ) )
1092 p_tk->fmt.i_codec = VLC_CODEC_SUBT;
1093 p_tk->fmt.subs.psz_encoding = strdup( "UTF-8" );
1095 else if( !strcmp( p_tk->psz_codec, "S_TEXT/USF" ) )
1097 p_tk->fmt.i_codec = VLC_FOURCC( 'u', 's', 'f', ' ' );
1098 p_tk->fmt.subs.psz_encoding = strdup( "UTF-8" );
1099 if( p_tk->i_extra_data )
1101 p_tk->fmt.i_extra = p_tk->i_extra_data;
1102 p_tk->fmt.p_extra = xmalloc( p_tk->i_extra_data );
1103 memcpy( p_tk->fmt.p_extra, p_tk->p_extra_data, p_tk->i_extra_data );
1106 else if( !strcmp( p_tk->psz_codec, "S_TEXT/SSA" ) ||
1107 !strcmp( p_tk->psz_codec, "S_TEXT/ASS" ) ||
1108 !strcmp( p_tk->psz_codec, "S_SSA" ) ||
1109 !strcmp( p_tk->psz_codec, "S_ASS" ))
1111 p_tk->fmt.i_codec = VLC_CODEC_SSA;
1112 p_tk->fmt.subs.psz_encoding = strdup( "UTF-8" );
1113 if( p_tk->i_extra_data )
1115 p_tk->fmt.i_extra = p_tk->i_extra_data;
1116 p_tk->fmt.p_extra = xmalloc( p_tk->i_extra_data );
1117 memcpy( p_tk->fmt.p_extra, p_tk->p_extra_data, p_tk->i_extra_data );
1120 else if( !strcmp( p_tk->psz_codec, "S_VOBSUB" ) )
1122 p_tk->fmt.i_codec = VLC_CODEC_SPU;
1123 if( p_tk->i_extra_data )
1126 char *psz_buf = (char *)malloc( p_tk->i_extra_data + 1);
1127 if( psz_buf != NULL )
1129 memcpy( psz_buf, p_tk->p_extra_data , p_tk->i_extra_data );
1130 psz_buf[p_tk->i_extra_data] = '\0';
1132 psz_start = strstr( psz_buf, "size:" );
1134 vobsub_size_parse( psz_start,
1135 &p_tk->fmt.subs.spu.i_original_frame_width,
1136 &p_tk->fmt.subs.spu.i_original_frame_height ) == VLC_SUCCESS )
1138 msg_Dbg( &sys.demuxer, "original frame size vobsubs: %dx%d",
1139 p_tk->fmt.subs.spu.i_original_frame_width,
1140 p_tk->fmt.subs.spu.i_original_frame_height );
1144 msg_Warn( &sys.demuxer, "reading original frame size for vobsub failed" );
1147 psz_start = strstr( psz_buf, "palette:" );
1149 vobsub_palette_parse( psz_start, &p_tk->fmt.subs.spu.palette[1] ) == VLC_SUCCESS )
1151 p_tk->fmt.subs.spu.palette[0] = 0xBeef;
1152 msg_Dbg( &sys.demuxer, "vobsub palette read" );
1156 msg_Warn( &sys.demuxer, "reading original palette failed" );
1162 else if( !strcmp( p_tk->psz_codec, "S_HDMV/PGS" ) )
1164 p_tk->fmt.i_codec = VLC_CODEC_BD_PG;
1166 else if( !strcmp( p_tk->psz_codec, "B_VOBBTN" ) )
1168 p_tk->fmt.i_cat = NAV_ES;
1171 else if( !strcmp( p_tk->psz_codec, "A_REAL/14_4" ) )
1173 p_fmt->i_codec = VLC_CODEC_RA_144;
1174 p_fmt->audio.i_channels = 1;
1175 p_fmt->audio.i_rate = 8000;
1176 p_fmt->audio.i_blockalign = 0x14;
1180 msg_Err( &sys.demuxer, "unknown codec id=`%s'", p_tk->psz_codec );
1181 p_tk->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
1183 if( p_tk->b_default )
1185 p_tk->fmt.i_priority = 1000;
1188 p_tk->p_es = es_out_Add( sys.demuxer.out, &p_tk->fmt );
1190 /* Turn on a subtitles track if it has been flagged as default -
1191 * but only do this if no subtitles track has already been engaged,
1192 * either by an earlier 'default track' (??) or by default
1193 * language choice behaviour.
1195 if( p_tk->b_default )
1197 es_out_Control( sys.demuxer.out,
1198 ES_OUT_SET_ES_DEFAULT,
1202 es_out_Control( sys.demuxer.out, ES_OUT_SET_NEXT_DISPLAY_TIME, i_start_time );
1204 sys.i_start_pts = i_start_time;
1205 // reset the stream reading to the first cluster of the segment used
1206 es.I_O().setFilePointer( i_start_pos );
1209 ep = new EbmlParser( &es, segment, &sys.demuxer );
1214 void matroska_segment_c::UnSelect( )
1218 for( i_track = 0; i_track < tracks.size(); i_track++ )
1220 if ( tracks[i_track]->p_es != NULL )
1222 // es_format_Clean( &tracks[i_track]->fmt );
1223 es_out_Del( sys.demuxer.out, tracks[i_track]->p_es );
1224 tracks[i_track]->p_es = NULL;
1231 int matroska_segment_c::BlockGet( KaxBlock * & pp_block, KaxSimpleBlock * & pp_simpleblock, bool *pb_key_picture, bool *pb_discardable_picture, int64_t *pi_duration )
1233 pp_simpleblock = NULL;
1236 *pb_key_picture = true;
1237 *pb_discardable_picture = false;
1241 EbmlElement *el = NULL;
1245 return VLC_EGENERIC;
1247 if( pp_simpleblock != NULL || ((el = ep->Get()) == NULL && pp_block != NULL) )
1249 /* Check blocks validity to protect againts broken files */
1250 if( BlockFindTrackIndex( NULL, pp_block , pp_simpleblock ) )
1253 pp_simpleblock = NULL;
1257 if( pp_simpleblock != NULL )
1259 *pb_key_picture = pp_simpleblock->IsKeyframe();
1260 *pb_discardable_picture = pp_simpleblock->IsDiscardable();
1263 /* update the index */
1264 #define idx p_indexes[i_index - 1]
1265 if( i_index > 0 && idx.i_time == -1 )
1267 if ( pp_simpleblock != NULL )
1268 idx.i_time = pp_simpleblock->GlobalTimecode() / (mtime_t)1000;
1270 idx.i_time = (*pp_block).GlobalTimecode() / (mtime_t)1000;
1271 idx.b_key = *pb_key_picture;
1277 i_level = ep->GetLevel();
1286 msg_Warn( &sys.demuxer, "EOF" );
1287 return VLC_EGENERIC;
1290 /* Verify that we are still inside our cluster
1291 * It can happens whith broken files and when seeking
1295 if( cluster && !ep->IsTopPresent( cluster ) )
1297 msg_Warn( &sys.demuxer, "Unexpected escape from current cluster" );
1308 if( MKV_IS_ID( el, KaxCluster ) )
1310 cluster = (KaxCluster*)el;
1311 i_cluster_pos = cluster->GetElementPosition();
1313 /* add it to the index */
1315 ( i_index > 0 && p_indexes[i_index - 1].i_position < (int64_t)cluster->GetElementPosition() ) )
1317 IndexAppendCluster( cluster );
1320 // reset silent tracks
1321 for (size_t i=0; i<tracks.size(); i++)
1323 tracks[i]->b_silent = false;
1328 else if( MKV_IS_ID( el, KaxCues ) )
1330 msg_Warn( &sys.demuxer, "find KaxCues FIXME" );
1331 return VLC_EGENERIC;
1335 msg_Dbg( &sys.demuxer, "unknown (%s)", typeid( el ).name() );
1339 if( MKV_IS_ID( el, KaxClusterTimecode ) )
1341 KaxClusterTimecode &ctc = *(KaxClusterTimecode*)el;
1343 ctc.ReadData( es.I_O(), SCOPE_ALL_DATA );
1344 cluster->InitTimecode( uint64( ctc ), i_timescale );
1346 else if( MKV_IS_ID( el, KaxClusterSilentTracks ) )
1350 else if( MKV_IS_ID( el, KaxBlockGroup ) )
1352 i_block_pos = el->GetElementPosition();
1355 else if( MKV_IS_ID( el, KaxSimpleBlock ) )
1357 pp_simpleblock = (KaxSimpleBlock*)el;
1359 pp_simpleblock->ReadData( es.I_O() );
1360 pp_simpleblock->SetParent( *cluster );
1364 if( MKV_IS_ID( el, KaxBlock ) )
1366 pp_block = (KaxBlock*)el;
1368 pp_block->ReadData( es.I_O() );
1369 pp_block->SetParent( *cluster );
1373 else if( MKV_IS_ID( el, KaxBlockDuration ) )
1375 KaxBlockDuration &dur = *(KaxBlockDuration*)el;
1377 dur.ReadData( es.I_O() );
1378 *pi_duration = uint64( dur );
1380 else if( MKV_IS_ID( el, KaxReferenceBlock ) )
1382 KaxReferenceBlock &ref = *(KaxReferenceBlock*)el;
1384 ref.ReadData( es.I_O() );
1386 if( *pb_key_picture )
1387 *pb_key_picture = false;
1388 else if( int64( ref ) > 0 )
1389 *pb_discardable_picture = true;
1391 else if( MKV_IS_ID( el, KaxClusterSilentTrackNumber ) )
1393 KaxClusterSilentTrackNumber &track_num = *(KaxClusterSilentTrackNumber*)el;
1394 track_num.ReadData( es.I_O() );
1396 for (size_t i=0; i<tracks.size(); i++)
1398 if ( tracks[i]->i_number == uint32(track_num))
1400 tracks[i]->b_silent = true;
1407 msg_Err( &sys.demuxer, "invalid level = %d", i_level );
1408 return VLC_EGENERIC;