?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
* : svn:keywords
[vlc.git] / modules / demux / mkv.cpp
blobcfaebc1c1a54b7794692f569c537a603dc01dc69
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 /*****************************************************************************
26 * Preamble
27 *****************************************************************************/
28 #include <stdlib.h> /* malloc(), free() */
29
30 #include <vlc/vlc.h>
31
32 #ifdef HAVE_TIME_H
33 # include <time.h> /* time() */
34 #endif
35
36 #include <vlc/input.h>
37
38 #include <codecs.h> /* BITMAPINFOHEADER, WAVEFORMATEX */
39 #include "iso_lang.h"
40 #include "vlc_meta.h"
41
42 #include <iostream>
43 #include <cassert>
44 #include <typeinfo>
45 #include <string>
46 #include <vector>
47 #include <algorithm>
48
49 #ifdef HAVE_DIRENT_H
50 # include <dirent.h>
51 #endif
52
53 /* libebml and matroska */
54 #include "ebml/EbmlHead.h"
55 #include "ebml/EbmlSubHead.h"
56 #include "ebml/EbmlStream.h"
57 #include "ebml/EbmlContexts.h"
58 #include "ebml/EbmlVoid.h"
59 #include "ebml/EbmlVersion.h"
60 #include "ebml/StdIOCallback.h"
61
62 #include "matroska/KaxAttachments.h"
63 #include "matroska/KaxBlock.h"
64 #include "matroska/KaxBlockData.h"
65 #include "matroska/KaxChapters.h"
66 #include "matroska/KaxCluster.h"
67 #include "matroska/KaxClusterData.h"
68 #include "matroska/KaxContexts.h"
69 #include "matroska/KaxCues.h"
70 #include "matroska/KaxCuesData.h"
71 #include "matroska/KaxInfo.h"
72 #include "matroska/KaxInfoData.h"
73 #include "matroska/KaxSeekHead.h"
74 #include "matroska/KaxSegment.h"
75 #include "matroska/KaxTag.h"
76 #include "matroska/KaxTags.h"
77 #include "matroska/KaxTagMulti.h"
78 #include "matroska/KaxTracks.h"
79 #include "matroska/KaxTrackAudio.h"
80 #include "matroska/KaxTrackVideo.h"
81 #include "matroska/KaxTrackEntryData.h"
82 #include "matroska/KaxContentEncoding.h"
83 #include "matroska/KaxVersion.h"
84
85 #include "ebml/StdIOCallback.h"
86
87 #if LIBMATROSKA_VERSION < 0x000706
88 START_LIBMATROSKA_NAMESPACE
89 extern const EbmlSemanticContext MATROSKA_DLL_API KaxMatroska_Context;
90 END_LIBMATROSKA_NAMESPACE
91 #endif
92
93 #include "vlc_keys.h"
94
95 extern "C" {
96 #include "mp4/libmp4.h"
97 }
98 #ifdef HAVE_ZLIB_H
99 # include <zlib.h>
100 #endif
101
102 #define MATROSKA_COMPRESSION_NONE -1
103 #define MATROSKA_COMPRESSION_ZLIB 0
104 #define MATROSKA_COMPRESSION_BLIB 1
105 #define MATROSKA_COMPRESSION_LZOX 2
106 #define MATROSKA_COMPRESSION_HEADER 3
107
108 #define MKVD_TIMECODESCALE 1000000
109
110 ////////////////////////////////////////////////////////////////////////////////////////////////////////////////
111 #undef ATTRIBUTE_PACKED
112 #undef PRAGMA_PACK_BEGIN
113 #undef PRAGMA_PACK_END
114
115 #if defined(__GNUC__)
116 #if __GNUC__ > 2 || (__GNUC__ == 2 && __GNUC_MINOR__ >= 95)
117 #define ATTRIBUTE_PACKED __attribute__ ((packed))
118 #define PRAGMA_PACK 0
119 #endif
120 #endif
121
122 #if !defined(ATTRIBUTE_PACKED)
123 #define ATTRIBUTE_PACKED
124 #define PRAGMA_PACK 1
125 #endif
126
127 #if PRAGMA_PACK
128 #pragma pack(1)
129 #endif
130
131 /*************************************
132 * taken from libdvdnav / libdvdread
133 **************************************/
134
135 /**
136 * DVD Time Information.
137 */
138 typedef struct {
139 uint8_t hour;
140 uint8_t minute;
141 uint8_t second;
142 uint8_t frame_u; /* The two high bits are the frame rate. */
143 } ATTRIBUTE_PACKED dvd_time_t;
144
145 /**
146 * User Operations.
147 */
148 typedef struct {
149 #ifdef WORDS_BIGENDIAN
150 unsigned char zero : 7; /* 25-31 */
151 unsigned char video_pres_mode_change : 1; /* 24 */
152
153 unsigned char karaoke_audio_pres_mode_change : 1; /* 23 */
154 unsigned char angle_change : 1;
155 unsigned char subpic_stream_change : 1;
156 unsigned char audio_stream_change : 1;
157 unsigned char pause_on : 1;
158 unsigned char still_off : 1;
159 unsigned char button_select_or_activate : 1;
160 unsigned char resume : 1; /* 16 */
161
162 unsigned char chapter_menu_call : 1; /* 15 */
163 unsigned char angle_menu_call : 1;
164 unsigned char audio_menu_call : 1;
165 unsigned char subpic_menu_call : 1;
166 unsigned char root_menu_call : 1;
167 unsigned char title_menu_call : 1;
168 unsigned char backward_scan : 1;
169 unsigned char forward_scan : 1; /* 8 */
170
171 unsigned char next_pg_search : 1; /* 7 */
172 unsigned char prev_or_top_pg_search : 1;
173 unsigned char time_or_chapter_search : 1;
174 unsigned char go_up : 1;
175 unsigned char stop : 1;
176 unsigned char title_play : 1;
177 unsigned char chapter_search_or_play : 1;
178 unsigned char title_or_time_play : 1; /* 0 */
179 #else
180 unsigned char video_pres_mode_change : 1; /* 24 */
181 unsigned char zero : 7; /* 25-31 */
182
183 unsigned char resume : 1; /* 16 */
184 unsigned char button_select_or_activate : 1;
185 unsigned char still_off : 1;
186 unsigned char pause_on : 1;
187 unsigned char audio_stream_change : 1;
188 unsigned char subpic_stream_change : 1;
189 unsigned char angle_change : 1;
190 unsigned char karaoke_audio_pres_mode_change : 1; /* 23 */
191
192 unsigned char forward_scan : 1; /* 8 */
193 unsigned char backward_scan : 1;
194 unsigned char title_menu_call : 1;
195 unsigned char root_menu_call : 1;
196 unsigned char subpic_menu_call : 1;
197 unsigned char audio_menu_call : 1;
198 unsigned char angle_menu_call : 1;
199 unsigned char chapter_menu_call : 1; /* 15 */
200
201 unsigned char title_or_time_play : 1; /* 0 */
202 unsigned char chapter_search_or_play : 1;
203 unsigned char title_play : 1;
204 unsigned char stop : 1;
205 unsigned char go_up : 1;
206 unsigned char time_or_chapter_search : 1;
207 unsigned char prev_or_top_pg_search : 1;
208 unsigned char next_pg_search : 1; /* 7 */
209 #endif
210 } ATTRIBUTE_PACKED user_ops_t;
211
212 /**
213 * Type to store per-command data.
214 */
215 typedef struct {
216 uint8_t bytes[8];
217 } ATTRIBUTE_PACKED vm_cmd_t;
218 #define COMMAND_DATA_SIZE 8
219
220 /**
221 * PCI General Information
222 */
223 typedef struct {
224 uint32_t nv_pck_lbn; /**< sector address of this nav pack */
225 uint16_t vobu_cat; /**< 'category' of vobu */
226 uint16_t zero1; /**< reserved */
227 user_ops_t vobu_uop_ctl; /**< UOP of vobu */
228 uint32_t vobu_s_ptm; /**< start presentation time of vobu */
229 uint32_t vobu_e_ptm; /**< end presentation time of vobu */
230 uint32_t vobu_se_e_ptm; /**< end ptm of sequence end in vobu */
231 dvd_time_t e_eltm; /**< Cell elapsed time */
232 char vobu_isrc[32];
233 } ATTRIBUTE_PACKED pci_gi_t;
234
235 /**
236 * Non Seamless Angle Information
237 */
238 typedef struct {
239 uint32_t nsml_agl_dsta[9]; /**< address of destination vobu in AGL_C#n */
240 } ATTRIBUTE_PACKED nsml_agli_t;
241
242 /**
243 * Highlight General Information
244 *
245 * For btngrX_dsp_ty the bits have the following meaning:
246 * 000b: normal 4/3 only buttons
247 * XX1b: wide (16/9) buttons
248 * X1Xb: letterbox buttons
249 * 1XXb: pan&scan buttons
250 */
251 typedef struct {
252 uint16_t hli_ss; /**< status, only low 2 bits 0: no buttons, 1: different 2: equal 3: eual except for button cmds */
253 uint32_t hli_s_ptm; /**< start ptm of hli */
254 uint32_t hli_e_ptm; /**< end ptm of hli */
255 uint32_t btn_se_e_ptm; /**< end ptm of button select */
256 #ifdef WORDS_BIGENDIAN
257 unsigned char zero1 : 2; /**< reserved */
258 unsigned char btngr_ns : 2; /**< number of button groups 1, 2 or 3 with 36/18/12 buttons */
259 unsigned char zero2 : 1; /**< reserved */
260 unsigned char btngr1_dsp_ty : 3; /**< display type of subpic stream for button group 1 */
261 unsigned char zero3 : 1; /**< reserved */
262 unsigned char btngr2_dsp_ty : 3; /**< display type of subpic stream for button group 2 */
263 unsigned char zero4 : 1; /**< reserved */
264 unsigned char btngr3_dsp_ty : 3; /**< display type of subpic stream for button group 3 */
265 #else
266 unsigned char btngr1_dsp_ty : 3;
267 unsigned char zero2 : 1;
268 unsigned char btngr_ns : 2;
269 unsigned char zero1 : 2;
270 unsigned char btngr3_dsp_ty : 3;
271 unsigned char zero4 : 1;
272 unsigned char btngr2_dsp_ty : 3;
273 unsigned char zero3 : 1;
274 #endif
275 uint8_t btn_ofn; /**< button offset number range 0-255 */
276 uint8_t btn_ns; /**< number of valid buttons <= 36/18/12 (low 6 bits) */
277 uint8_t nsl_btn_ns; /**< number of buttons selectable by U_BTNNi (low 6 bits) nsl_btn_ns <= btn_ns */
278 uint8_t zero5; /**< reserved */
279 uint8_t fosl_btnn; /**< forcedly selected button (low 6 bits) */
280 uint8_t foac_btnn; /**< forcedly activated button (low 6 bits) */
281 } ATTRIBUTE_PACKED hl_gi_t;
282
283
284 /**
285 * Button Color Information Table
286 * Each entry beeing a 32bit word that contains the color indexs and alpha
287 * values to use. They are all represented by 4 bit number and stored
288 * like this [Ci3, Ci2, Ci1, Ci0, A3, A2, A1, A0]. The actual palette
289 * that the indexes reference is in the PGC.
290 * \todo split the uint32_t into a struct
291 */
292 typedef struct {
293 uint32_t btn_coli[3][2]; /**< [button color number-1][select:0/action:1] */
294 } ATTRIBUTE_PACKED btn_colit_t;
295
296 /**
297 * Button Information
298 *
299 * NOTE: I've had to change the structure from the disk layout to get
300 * the packing to work with Sun's Forte C compiler.
301 * The 4 and 7 bytes are 'rotated' was: ABC DEF GHIJ is: ABCG DEFH IJ
302 */
303 typedef struct {
304 #ifdef WORDS_BIGENDIAN
305 uint32 btn_coln : 2; /**< button color number */
306 uint32 x_start : 10; /**< x start offset within the overlay */
307 uint32 zero1 : 2; /**< reserved */
308 uint32 x_end : 10; /**< x end offset within the overlay */
309
310 uint32 zero3 : 2; /**< reserved */
311 uint32 up : 6; /**< button index when pressing up */
312
313 uint32 auto_action_mode : 2; /**< 0: no, 1: activated if selected */
314 uint32 y_start : 10; /**< y start offset within the overlay */
315 uint32 zero2 : 2; /**< reserved */
316 uint32 y_end : 10; /**< y end offset within the overlay */
317
318 uint32 zero4 : 2; /**< reserved */
319 uint32 down : 6; /**< button index when pressing down */
320 unsigned char zero5 : 2; /**< reserved */
321 unsigned char left : 6; /**< button index when pressing left */
322 unsigned char zero6 : 2; /**< reserved */
323 unsigned char right : 6; /**< button index when pressing right */
324 #else
325 uint32 x_end : 10;
326 uint32 zero1 : 2;
327 uint32 x_start : 10;
328 uint32 btn_coln : 2;
329
330 uint32 up : 6;
331 uint32 zero3 : 2;
332
333 uint32 y_end : 10;
334 uint32 zero2 : 2;
335 uint32 y_start : 10;
336 uint32 auto_action_mode : 2;
337
338 uint32 down : 6;
339 uint32 zero4 : 2;
340 unsigned char left : 6;
341 unsigned char zero5 : 2;
342 unsigned char right : 6;
343 unsigned char zero6 : 2;
344 #endif
345 vm_cmd_t cmd;
346 } ATTRIBUTE_PACKED btni_t;
347
348 /**
349 * Highlight Information
350 */
351 typedef struct {
352 hl_gi_t hl_gi;
353 btn_colit_t btn_colit;
354 btni_t btnit[36];
355 } ATTRIBUTE_PACKED hli_t;
356
357 /**
358 * PCI packet
359 */
360 typedef struct {
361 pci_gi_t pci_gi;
362 nsml_agli_t nsml_agli;
363 hli_t hli;
364 uint8_t zero1[189];
365 } ATTRIBUTE_PACKED pci_t;
366
367
368 #if PRAGMA_PACK
369 #pragma pack()
370 #endif
371 ////////////////////////////////////////////////////////////////////////////////////////////////////////////////
372
373
374 /**
375 * What's between a directory and a filename?
376 */
377 #if defined( WIN32 )
378 #define DIRECTORY_SEPARATOR '\\'
379 #else
380 #define DIRECTORY_SEPARATOR '/'
381 #endif
382
383 using namespace LIBMATROSKA_NAMESPACE;
384 using namespace std;
385
386 /*****************************************************************************
387 * Module descriptor
388 *****************************************************************************/
389 static int Open ( vlc_object_t * );
390 static void Close( vlc_object_t * );
391
392 vlc_module_begin();
393 set_shortname( "Matroska" );
394 set_description( _("Matroska stream demuxer" ) );
395 set_capability( "demux2", 50 );
396 set_callbacks( Open, Close );
397 set_category( CAT_INPUT );
398 set_subcategory( SUBCAT_INPUT_DEMUX );
399
400 add_bool( "mkv-use-ordered-chapters", 1, NULL,
401 N_("Ordered chapters"),
402 N_("Play ordered chapters as specified in the segment."), VLC_TRUE );
403
404 add_bool( "mkv-use-chapter-codec", 1, NULL,
405 N_("Chapter codecs"),
406 N_("Use chapter codecs found in the segment."), VLC_TRUE );
407
408 add_bool( "mkv-preload-local-dir", 1, NULL,
409 N_("Preload Directory"),
410 N_("Preload matroska files from the same family in the same directory (not good for broken files)."), VLC_TRUE );
411
412 add_bool( "mkv-seek-percent", 0, NULL,
413 N_("Seek based on percent not time"),
414 N_("Seek based on percent not time."), VLC_TRUE );
415
416 add_bool( "mkv-use-dummy", 0, NULL,
417 N_("Dummy Elements"),
418 N_("Read and discard unknown EBML elements (not good for broken files)."), VLC_TRUE );
419
420 add_shortcut( "mka" );
421 add_shortcut( "mkv" );
422 vlc_module_end();
423
424 /*****************************************************************************
425 * Local prototypes
426 *****************************************************************************/
427 #ifdef HAVE_ZLIB_H
428 block_t *block_zlib_decompress( vlc_object_t *p_this, block_t *p_in_block ) {
429 int result, dstsize, n;
430 unsigned char *dst;
431 block_t *p_block;
432 z_stream d_stream;
433
434 d_stream.zalloc = (alloc_func)0;
435 d_stream.zfree = (free_func)0;
436 d_stream.opaque = (voidpf)0;
437 result = inflateInit(&d_stream);
438 if( result != Z_OK )
439 {
440 msg_Dbg( p_this, "inflateInit() failed. Result: %d", result );
441 return NULL;
442 }
443
444 d_stream.next_in = (Bytef *)p_in_block->p_buffer;
445 d_stream.avail_in = p_in_block->i_buffer;
446 n = 0;
447 p_block = block_New( p_this, 0 );
448 dst = NULL;
449 do
450 {
451 n++;
452 p_block = block_Realloc( p_block, 0, n * 1000 );
453 dst = (unsigned char *)p_block->p_buffer;
454 d_stream.next_out = (Bytef *)&dst[(n - 1) * 1000];
455 d_stream.avail_out = 1000;
456 result = inflate(&d_stream, Z_NO_FLUSH);
457 if( ( result != Z_OK ) && ( result != Z_STREAM_END ) )
458 {
459 msg_Dbg( p_this, "Zlib decompression failed. Result: %d", result );
460 return NULL;
461 }
462 }
463 while( ( d_stream.avail_out == 0 ) && ( d_stream.avail_in != 0 ) &&
464 ( result != Z_STREAM_END ) );
465
466 dstsize = d_stream.total_out;
467 inflateEnd( &d_stream );
468
469 p_block = block_Realloc( p_block, 0, dstsize );
470 p_block->i_buffer = dstsize;
471 block_Release( p_in_block );
472
473 return p_block;
474 }
475 #endif
476
477 /**
478 * Helper function to print the mkv parse tree
479 */
480 static void MkvTree( demux_t & demuxer, int i_level, char *psz_format, ... )
481 {
482 va_list args;
483 if( i_level > 9 )
484 {
485 msg_Err( &demuxer, "too deep tree" );
486 return;
487 }
488 va_start( args, psz_format );
489 static char *psz_foo = "| | | | | | | | | |";
490 char *psz_foo2 = (char*)malloc( ( i_level * 4 + 3 + strlen( psz_format ) ) * sizeof(char) );
491 strncpy( psz_foo2, psz_foo, 4 * i_level );
492 psz_foo2[ 4 * i_level ] = '+';
493 psz_foo2[ 4 * i_level + 1 ] = ' ';
494 strcpy( &psz_foo2[ 4 * i_level + 2 ], psz_format );
495 __msg_GenericVa( VLC_OBJECT(&demuxer), MSG_QUEUE_NORMAL, VLC_MSG_DBG, "mkv", psz_foo2, args );
496 free( psz_foo2 );
497 va_end( args );
498 }
499
500 /*****************************************************************************
501 * Stream managment
502 *****************************************************************************/
503 class vlc_stream_io_callback: public IOCallback
504 {
505 private:
506 stream_t *s;
507 vlc_bool_t mb_eof;
508 vlc_bool_t b_owner;
509
510 public:
511 vlc_stream_io_callback( stream_t *, vlc_bool_t );
512
513 virtual ~vlc_stream_io_callback()
514 {
515 if( b_owner )
516 stream_Delete( s );
517 }
518
519 virtual uint32 read ( void *p_buffer, size_t i_size);
520 virtual void setFilePointer ( int64_t i_offset, seek_mode mode = seek_beginning );
521 virtual size_t write ( const void *p_buffer, size_t i_size);
522 virtual uint64 getFilePointer ( void );
523 virtual void close ( void );
524 };
525
526 /*****************************************************************************
527 * Ebml Stream parser
528 *****************************************************************************/
529 class EbmlParser
530 {
531 public:
532 EbmlParser( EbmlStream *es, EbmlElement *el_start, demux_t *p_demux );
533 virtual ~EbmlParser( void );
534
535 void Up( void );
536 void Down( void );
537 void Reset( demux_t *p_demux );
538 EbmlElement *Get( void );
539 void Keep( void );
540 EbmlElement *UnGet( uint64 i_block_pos, uint64 i_cluster_pos );
541
542 int GetLevel( void );
543
544 private:
545 EbmlStream *m_es;
546 int mi_level;
547 EbmlElement *m_el[10];
548 int64_t mi_remain_size[10];
549
550 EbmlElement *m_got;
551
552 int mi_user_level;
553 vlc_bool_t mb_keep;
554 vlc_bool_t mb_dummy;
555 };
556
557
558 /*****************************************************************************
559 * Some functions to manipulate memory
560 *****************************************************************************/
561 #define GetFOURCC( p ) __GetFOURCC( (uint8_t*)p )
562 static vlc_fourcc_t __GetFOURCC( uint8_t *p )
563 {
564 return VLC_FOURCC( p[0], p[1], p[2], p[3] );
565 }
566
567 /*****************************************************************************
568 * definitions of structures and functions used by this plugins
569 *****************************************************************************/
570 typedef struct
571 {
572 // ~mkv_track_t();
573
574 vlc_bool_t b_default;
575 vlc_bool_t b_enabled;
576 unsigned int i_number;
577
578 int i_extra_data;
579 uint8_t *p_extra_data;
580
581 char *psz_codec;
582
583 uint64_t i_default_duration;
584 float f_timecodescale;
585 mtime_t i_last_dts;
586
587 /* video */
588 es_format_t fmt;
589 float f_fps;
590 es_out_id_t *p_es;
591
592 /* audio */
593 unsigned int i_original_rate;
594
595 vlc_bool_t b_inited;
596 /* data to be send first */
597 int i_data_init;
598 uint8_t *p_data_init;
599
600 /* hack : it's for seek */
601 vlc_bool_t b_search_keyframe;
602 vlc_bool_t b_silent;
603
604 /* informative */
605 char *psz_codec_name;
606 char *psz_codec_settings;
607 char *psz_codec_info_url;
608 char *psz_codec_download_url;
609
610 /* encryption/compression */
611 int i_compression_type;
612 KaxContentCompSettings *p_compression_data;
613
614 } mkv_track_t;
615
616 typedef struct
617 {
618 int i_track;
619 int i_block_number;
620
621 int64_t i_position;
622 int64_t i_time;
623
624 vlc_bool_t b_key;
625 } mkv_index_t;
626
627 class demux_sys_t;
628
629 const binary MATROSKA_DVD_LEVEL_SS = 0x30;
630 const binary MATROSKA_DVD_LEVEL_LU = 0x2A;
631 const binary MATROSKA_DVD_LEVEL_TT = 0x28;
632 const binary MATROSKA_DVD_LEVEL_PGC = 0x20;
633 const binary MATROSKA_DVD_LEVEL_PG = 0x18;
634 const binary MATROSKA_DVD_LEVEL_PTT = 0x10;
635 const binary MATROSKA_DVD_LEVEL_CN = 0x08;
636
637 class chapter_codec_cmds_c
638 {
639 public:
640 chapter_codec_cmds_c( demux_sys_t & demuxer, int codec_id = -1)
641 :p_private_data(NULL)
642 ,i_codec_id( codec_id )
643 ,sys( demuxer )
644 {}
645
646 virtual ~chapter_codec_cmds_c()
647 {
648 delete p_private_data;
649 std::vector<KaxChapterProcessData*>::iterator indexe = enter_cmds.begin();
650 while ( indexe != enter_cmds.end() )
651 {
652 delete (*indexe);
653 indexe++;
654 }
655 std::vector<KaxChapterProcessData*>::iterator indexl = leave_cmds.begin();
656 while ( indexl != leave_cmds.end() )
657 {
658 delete (*indexl);
659 indexl++;
660 }
661 std::vector<KaxChapterProcessData*>::iterator indexd = during_cmds.begin();
662 while ( indexd != during_cmds.end() )
663 {
664 delete (*indexd);
665 indexd++;
666 }
667 }
668
669 void SetPrivate( const KaxChapterProcessPrivate & private_data )
670 {
671 p_private_data = new KaxChapterProcessPrivate( private_data );
672 }
673
674 void AddCommand( const KaxChapterProcessCommand & command );
675
676 /// \return wether the codec has seeked in the files or not
677 virtual bool Enter() { return false; }
678 virtual bool Leave() { return false; }
679 virtual std::string GetCodecName( bool f_for_title = false ) const { return ""; }
680 virtual int16 GetTitleNumber() { return -1; }
681
682 KaxChapterProcessPrivate *p_private_data;
683
684 protected:
685 std::vector<KaxChapterProcessData*> enter_cmds;
686 std::vector<KaxChapterProcessData*> during_cmds;
687 std::vector<KaxChapterProcessData*> leave_cmds;
688
689 int i_codec_id;
690 demux_sys_t & sys;
691 };
692
693 class dvd_command_interpretor_c
694 {
695 public:
696 dvd_command_interpretor_c( demux_sys_t & demuxer )
697 :sys( demuxer )
698 {
699 memset( p_PRMs, 0, sizeof(p_PRMs) );
700 p_PRMs[ 0x80 + 1 ] = 15;
701 p_PRMs[ 0x80 + 2 ] = 62;
702 p_PRMs[ 0x80 + 3 ] = 1;
703 p_PRMs[ 0x80 + 4 ] = 1;
704 p_PRMs[ 0x80 + 7 ] = 1;
705 p_PRMs[ 0x80 + 8 ] = 1;
706 p_PRMs[ 0x80 + 16 ] = 0xFFFFu;
707 p_PRMs[ 0x80 + 18 ] = 0xFFFFu;
708 }
709
710 bool Interpret( const binary * p_command, size_t i_size = 8 );
711
712 uint16 GetPRM( size_t index ) const
713 {
714 if ( index < 256 )
715 return p_PRMs[ index ];
716 else return 0;
717 }
718
719 uint16 GetGPRM( size_t index ) const
720 {
721 if ( index >= 0 && index < 16 )
722 return p_PRMs[ index ];
723 else return 0;
724 }
725
726 uint16 GetSPRM( size_t index ) const
727 {
728 // 21,22,23 reserved for future use
729 if ( index >= 0x80 && index < 0x95 )
730 return p_PRMs[ index ];
731 else return 0;
732 }
733
734 bool SetPRM( size_t index, uint16 value )
735 {
736 if ( index >= 0 && index < 16 )
737 {
738 p_PRMs[ index ] = value;
739 return true;
740 }
741 return false;
742 }
743
744 bool SetGPRM( size_t index, uint16 value )
745 {
746 if ( index >= 0 && index < 16 )
747 {
748 p_PRMs[ index ] = value;
749 return true;
750 }
751 return false;
752 }
753
754 bool SetSPRM( size_t index, uint16 value )
755 {
756 if ( index > 0x80 && index <= 0x8D && index != 0x8C )
757 {
758 p_PRMs[ index ] = value;
759 return true;
760 }
761 return false;
762 }
763
764 protected:
765 std::string GetRegTypeName( bool b_value, uint16 value ) const
766 {
767 std::string result;
768 char s_value[6], s_reg_value[6];
769 sprintf( s_value, "%.5d", value );
770
771 if ( b_value )
772 {
773 result = "value (";
774 result += s_value;
775 result += ")";
776 }
777 else if ( value < 0x80 )
778 {
779 sprintf( s_reg_value, "%.5d", GetPRM( value ) );
780 result = "GPreg[";
781 result += s_value;
782 result += "] (";
783 result += s_reg_value;
784 result += ")";
785 }
786 else
787 {
788 sprintf( s_reg_value, "%.5d", GetPRM( value ) );
789 result = "SPreg[";
790 result += s_value;
791 result += "] (";
792 result += s_reg_value;
793 result += ")";
794 }
795 return result;
796 }
797
798 uint16 p_PRMs[256];
799 demux_sys_t & sys;
800
801 // DVD command IDs
802
803 // Tests
804 // wether it's a comparison on the value or register
805 static const uint16 CMD_DVD_TEST_VALUE = 0x80;
806 static const uint16 CMD_DVD_IF_GPREG_AND = (1 << 4);
807 static const uint16 CMD_DVD_IF_GPREG_EQUAL = (2 << 4);
808 static const uint16 CMD_DVD_IF_GPREG_NOT_EQUAL = (3 << 4);
809 static const uint16 CMD_DVD_IF_GPREG_SUP_EQUAL = (4 << 4);
810 static const uint16 CMD_DVD_IF_GPREG_SUP = (5 << 4);
811 static const uint16 CMD_DVD_IF_GPREG_INF_EQUAL = (6 << 4);
812 static const uint16 CMD_DVD_IF_GPREG_INF = (7 << 4);
813
814 static const uint16 CMD_DVD_NOP = 0x0000;
815 static const uint16 CMD_DVD_GOTO_LINE = 0x0001;
816 static const uint16 CMD_DVD_BREAK = 0x0002;
817 // Links
818 static const uint16 CMD_DVD_NOP2 = 0x2001;
819 static const uint16 CMD_DVD_LINKPGCN = 0x2004;
820 static const uint16 CMD_DVD_LINKPGN = 0x2006;
821 static const uint16 CMD_DVD_LINKCN = 0x2007;
822 static const uint16 CMD_DVD_JUMP_TT = 0x3002;
823 static const uint16 CMD_DVD_JUMPVTS_TT = 0x3003;
824 static const uint16 CMD_DVD_JUMPVTS_PTT = 0x3005;
825 static const uint16 CMD_DVD_JUMP_SS = 0x3006;
826 static const uint16 CMD_DVD_CALLSS_VTSM1 = 0x3008;
827 //
828 static const uint16 CMD_DVD_SET_HL_BTNN2 = 0x4600;
829 static const uint16 CMD_DVD_SET_HL_BTNN_LINKPGCN1 = 0x4604;
830 static const uint16 CMD_DVD_SET_STREAM = 0x5100;
831 static const uint16 CMD_DVD_SET_GPRMMD = 0x5300;
832 static const uint16 CMD_DVD_SET_HL_BTNN1 = 0x5600;
833 static const uint16 CMD_DVD_SET_HL_BTNN_LINKPGCN2 = 0x5604;
834 static const uint16 CMD_DVD_SET_HL_BTNN_LINKCN = 0x5607;
835 // Operations
836 static const uint16 CMD_DVD_MOV_SPREG_PREG = 0x6100;
837 static const uint16 CMD_DVD_GPREG_MOV_VALUE = 0x7100;
838 static const uint16 CMD_DVD_SUB_GPREG = 0x7400;
839 static const uint16 CMD_DVD_MULT_GPREG = 0x7500;
840 static const uint16 CMD_DVD_GPREG_DIV_VALUE = 0x7600;
841 static const uint16 CMD_DVD_GPREG_AND_VALUE = 0x7900;
842
843 // callbacks when browsing inside CodecPrivate
844 static bool MatchIsDomain ( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size );
845 static bool MatchIsVMG ( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size );
846 static bool MatchVTSNumber ( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size );
847 static bool MatchVTSMNumber ( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size );
848 static bool MatchTitleNumber ( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size );
849 static bool MatchPgcType ( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size );
850 static bool MatchPgcNumber ( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size );
851 static bool MatchChapterNumber( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size );
852 static bool MatchCellNumber ( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size );
853 };
854
855 class dvd_chapter_codec_c : public chapter_codec_cmds_c
856 {
857 public:
858 dvd_chapter_codec_c( demux_sys_t & sys )
859 :chapter_codec_cmds_c( sys, 1 )
860 {}
861
862 bool Enter();
863 bool Leave();
864 std::string GetCodecName( bool f_for_title = false ) const;
865 int16 GetTitleNumber();
866 };
867
868 class matroska_script_interpretor_c
869 {
870 public:
871 matroska_script_interpretor_c( demux_sys_t & demuxer )
872 :sys( demuxer )
873 {}
874
875 bool Interpret( const binary * p_command, size_t i_size );
876
877 // DVD command IDs
878 static const std::string CMD_MS_GOTO_AND_PLAY;
879
880 protected:
881 demux_sys_t & sys;
882 };
883
884 const std::string matroska_script_interpretor_c::CMD_MS_GOTO_AND_PLAY = "GotoAndPlay";
885
886
887 class matroska_script_codec_c : public chapter_codec_cmds_c
888 {
889 public:
890 matroska_script_codec_c( demux_sys_t & sys )
891 :chapter_codec_cmds_c( sys, 0 )
892 ,interpretor( sys )
893 {}
894
895 bool Enter();
896 bool Leave();
897
898 protected:
899 matroska_script_interpretor_c interpretor;
900 };
901
902 class chapter_translation_c
903 {
904 public:
905 chapter_translation_c()
906 :p_translated(NULL)
907 {}
908
909 ~chapter_translation_c()
910 {
911 delete p_translated;
912 }
913
914 KaxChapterTranslateID *p_translated;
915 unsigned int codec_id;
916 std::vector<uint64_t> editions;
917 };
918
919 class chapter_item_c
920 {
921 public:
922 chapter_item_c()
923 :i_start_time(0)
924 ,i_end_time(-1)
925 ,i_user_start_time(-1)
926 ,i_user_end_time(-1)
927 ,i_seekpoint_num(-1)
928 ,b_display_seekpoint(true)
929 ,b_user_display(false)
930 ,psz_parent(NULL)
931 ,b_is_leaving(false)
932 {}
933
934 virtual ~chapter_item_c()
935 {
936 std::vector<chapter_codec_cmds_c*>::iterator index = codecs.begin();
937 while ( index != codecs.end() )
938 {
939 delete (*index);
940 index++;
941 }
942 std::vector<chapter_item_c*>::iterator index_ = sub_chapters.begin();
943 while ( index_ != sub_chapters.end() )
944 {
945 delete (*index_);
946 index_++;
947 }
948 }
949
950 int64_t RefreshChapters( bool b_ordered, int64_t i_prev_user_time );
951 int PublishChapters( input_title_t & title, int & i_user_chapters, int i_level = 0 );
952 virtual chapter_item_c * FindTimecode( mtime_t i_timecode, const chapter_item_c * p_current, bool & b_found );
953 void Append( const chapter_item_c & edition );
954 chapter_item_c * FindChapter( int64_t i_find_uid );
955 virtual chapter_item_c *BrowseCodecPrivate( unsigned int codec_id,
956 bool (*match)(const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size ),
957 const void *p_cookie,
958 size_t i_cookie_size );
959 std::string GetCodecName( bool f_for_title = false ) const;
960 bool ParentOf( const chapter_item_c & item ) const;
961 int16 GetTitleNumber( ) const;
962
963 int64_t i_start_time, i_end_time;
964 int64_t i_user_start_time, i_user_end_time; /* the time in the stream when an edition is ordered */
965 std::vector<chapter_item_c*> sub_chapters;
966 int i_seekpoint_num;
967 int64_t i_uid;
968 bool b_display_seekpoint;
969 bool b_user_display;
970 std::string psz_name;
971 chapter_item_c *psz_parent;
972 bool b_is_leaving;
973
974 std::vector<chapter_codec_cmds_c*> codecs;
975
976 static bool CompareTimecode( const chapter_item_c * itemA, const chapter_item_c * itemB )
977 {
978 return ( itemA->i_user_start_time < itemB->i_user_start_time || (itemA->i_user_start_time == itemB->i_user_start_time && itemA->i_user_end_time < itemB->i_user_end_time) );
979 }
980
981 bool Enter( bool b_do_subchapters );
982 bool Leave( bool b_do_subchapters );
983 bool EnterAndLeave( chapter_item_c *p_item, bool b_enter = true );
984 };
985
986 class chapter_edition_c : public chapter_item_c
987 {
988 public:
989 chapter_edition_c()
990 :b_ordered(false)
991 {}
992
993 void RefreshChapters( );
994 mtime_t Duration() const;
995 std::string GetMainName() const;
996 chapter_item_c * FindTimecode( mtime_t i_timecode, const chapter_item_c * p_current );
997
998 bool b_ordered;
999 };
1000
1001 class matroska_segment_c
1002 {
1003 public:
1004 matroska_segment_c( demux_sys_t & demuxer, EbmlStream & estream )
1005 :segment(NULL)
1006 ,es(estream)
1007 ,i_timescale(MKVD_TIMECODESCALE)
1008 ,i_duration(-1)
1009 ,i_start_time(0)
1010 ,i_cues_position(-1)
1011 ,i_chapters_position(-1)
1012 ,i_tags_position(-1)
1013 ,cluster(NULL)
1014 ,i_block_pos(0)
1015 ,i_cluster_pos(0)
1016 ,i_start_pos(0)
1017 ,p_segment_uid(NULL)
1018 ,p_prev_segment_uid(NULL)
1019 ,p_next_segment_uid(NULL)
1020 ,b_cues(VLC_FALSE)
1021 ,i_index(0)
1022 ,i_index_max(1024)
1023 ,psz_muxing_application(NULL)
1024 ,psz_writing_application(NULL)
1025 ,psz_segment_filename(NULL)
1026 ,psz_title(NULL)
1027 ,psz_date_utc(NULL)
1028 ,i_default_edition(0)
1029 ,sys(demuxer)
1030 ,ep(NULL)
1031 ,b_preloaded(false)
1032 {
1033 p_indexes = (mkv_index_t*)malloc( sizeof( mkv_index_t ) * i_index_max );
1034 }
1035
1036 virtual ~matroska_segment_c()
1037 {
1038 for( size_t i_track = 0; i_track < tracks.size(); i_track++ )
1039 {
1040 if ( tracks[i_track]->p_compression_data )
1041 {
1042 delete tracks[i_track]->p_compression_data;
1043 }
1044 es_format_Clean( &tracks[i_track]->fmt );
1045 if( tracks[i_track]->p_extra_data )
1046 free( tracks[i_track]->p_extra_data );
1047 if( tracks[i_track]->psz_codec )
1048 free( tracks[i_track]->psz_codec );
1049 delete tracks[i_track];
1050 }
1051
1052 if( psz_writing_application )
1053 {
1054 free( psz_writing_application );
1055 }
1056 if( psz_muxing_application )
1057 {
1058 free( psz_muxing_application );
1059 }
1060 if( psz_segment_filename )
1061 {
1062 free( psz_segment_filename );
1063 }
1064 if( psz_title )
1065 {
1066 free( psz_title );
1067 }
1068 if( psz_date_utc )
1069 {
1070 free( psz_date_utc );
1071 }
1072 if ( p_indexes )
1073 free( p_indexes );
1074
1075 delete ep;
1076 delete segment;
1077 delete p_segment_uid;
1078 delete p_prev_segment_uid;
1079 delete p_next_segment_uid;
1080
1081 std::vector<chapter_edition_c*>::iterator index = stored_editions.begin();
1082 while ( index != stored_editions.end() )
1083 {
1084 delete (*index);
1085 index++;
1086 }
1087 std::vector<chapter_translation_c*>::iterator indext = translations.begin();
1088 while ( indext != translations.end() )
1089 {
1090 delete (*indext);
1091 indext++;
1092 }
1093 std::vector<KaxSegmentFamily*>::iterator indexf = families.begin();
1094 while ( indexf != families.end() )
1095 {
1096 delete (*indexf);
1097 indexf++;
1098 }
1099 }
1100
1101 KaxSegment *segment;
1102 EbmlStream & es;
1103
1104 /* time scale */
1105 uint64_t i_timescale;
1106
1107 /* duration of the segment */
1108 mtime_t i_duration;
1109 mtime_t i_start_time;
1110
1111 /* all tracks */
1112 std::vector<mkv_track_t*> tracks;
1113
1114 /* from seekhead */
1115 int64_t i_cues_position;
1116 int64_t i_chapters_position;
1117 int64_t i_tags_position;
1118
1119 KaxCluster *cluster;
1120 uint64 i_block_pos;
1121 uint64 i_cluster_pos;
1122 int64_t i_start_pos;
1123 KaxSegmentUID *p_segment_uid;
1124 KaxPrevUID *p_prev_segment_uid;
1125 KaxNextUID *p_next_segment_uid;
1126
1127 vlc_bool_t b_cues;
1128 int i_index;
1129 int i_index_max;
1130 mkv_index_t *p_indexes;
1131
1132 /* info */
1133 char *psz_muxing_application;
1134 char *psz_writing_application;
1135 char *psz_segment_filename;
1136 char *psz_title;
1137 char *psz_date_utc;
1138
1139 /* !!!!! GCC 3.3 bug on Darwin !!!!! */
1140 /* when you remove this variable the compiler issues an atomicity error */
1141 /* this variable only works when using std::vector<chapter_edition_c> */
1142 std::vector<chapter_edition_c*> stored_editions;
1143 int i_default_edition;
1144
1145 std::vector<chapter_translation_c*> translations;
1146 std::vector<KaxSegmentFamily*> families;
1147
1148 demux_sys_t & sys;
1149 EbmlParser *ep;
1150 bool b_preloaded;
1151
1152 bool Preload( );
1153 bool PreloadFamily( const matroska_segment_c & segment );
1154 void ParseInfo( KaxInfo *info );
1155 void ParseChapters( KaxChapters *chapters );
1156 void ParseSeekHead( KaxSeekHead *seekhead );
1157 void ParseTracks( KaxTracks *tracks );
1158 void ParseChapterAtom( int i_level, KaxChapterAtom *ca, chapter_item_c & chapters );
1159 void ParseTrackEntry( KaxTrackEntry *m );
1160 void ParseCluster( );
1161 void IndexAppendCluster( KaxCluster *cluster );
1162 void LoadCues( );
1163 void LoadTags( );
1164 void InformationCreate( );
1165 void Seek( mtime_t i_date, mtime_t i_time_offset );
1166 #if LIBMATROSKA_VERSION >= 0x000800
1167 int BlockGet( KaxBlock * &, KaxSimpleBlock * &, int64_t *, int64_t *, int64_t *);
1168 #else
1169 int BlockGet( KaxBlock * &, int64_t *, int64_t *, int64_t *);
1170 #endif
1171 bool Select( mtime_t i_start_time );
1172 void UnSelect( );
1173
1174 static bool CompareSegmentUIDs( const matroska_segment_c * item_a, const matroska_segment_c * item_b );
1175 };
1176
1177 // class holding hard-linked segment together in the playback order
1178 class virtual_segment_c
1179 {
1180 public:
1181 virtual_segment_c( matroska_segment_c *p_segment )
1182 :p_editions(NULL)
1183 ,i_sys_title(0)
1184 ,i_current_segment(0)
1185 ,i_current_edition(-1)
1186 ,psz_current_chapter(NULL)
1187 {
1188 linked_segments.push_back( p_segment );
1189
1190 AppendUID( p_segment->p_segment_uid );
1191 AppendUID( p_segment->p_prev_segment_uid );
1192 AppendUID( p_segment->p_next_segment_uid );
1193 }
1194
1195 void Sort();
1196 size_t AddSegment( matroska_segment_c *p_segment );
1197 void PreloadLinked( );
1198 mtime_t Duration( ) const;
1199 void LoadCues( );
1200 void Seek( demux_t & demuxer, mtime_t i_date, mtime_t i_time_offset, chapter_item_c *psz_chapter );
1201
1202 inline chapter_edition_c *Edition()
1203 {
1204 if ( i_current_edition >= 0 && size_t(i_current_edition) < p_editions->size() )
1205 return (*p_editions)[i_current_edition];
1206 return NULL;
1207 }
1208
1209 matroska_segment_c * Segment() const
1210 {
1211 if ( linked_segments.size() == 0 || i_current_segment >= linked_segments.size() )
1212 return NULL;
1213 return linked_segments[i_current_segment];
1214 }
1215
1216 inline chapter_item_c *CurrentChapter() {
1217 return psz_current_chapter;
1218 }
1219
1220 bool SelectNext()
1221 {
1222 if ( i_current_segment < linked_segments.size()-1 )
1223 {
1224 i_current_segment++;
1225 return true;
1226 }
1227 return false;
1228 }
1229
1230 bool FindUID( KaxSegmentUID & uid ) const
1231 {
1232 for ( size_t i=0; i<linked_uids.size(); i++ )
1233 {
1234 if ( linked_uids[i] == uid )
1235 return true;
1236 }
1237 return false;
1238 }
1239
1240 bool UpdateCurrentToChapter( demux_t & demux );
1241 void PrepareChapters( );
1242
1243 chapter_item_c *BrowseCodecPrivate( unsigned int codec_id,
1244 bool (*match)(const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size ),
1245 const void *p_cookie,
1246 size_t i_cookie_size );
1247 chapter_item_c *FindChapter( int64_t i_find_uid );
1248
1249 std::vector<chapter_edition_c*> *p_editions;
1250 int i_sys_title;
1251
1252 protected:
1253 std::vector<matroska_segment_c*> linked_segments;
1254 std::vector<KaxSegmentUID> linked_uids;
1255 size_t i_current_segment;
1256
1257 int i_current_edition;
1258 chapter_item_c *psz_current_chapter;
1259
1260 void AppendUID( const EbmlBinary * UID );
1261 };
1262
1263 class matroska_stream_c
1264 {
1265 public:
1266 matroska_stream_c( demux_sys_t & demuxer )
1267 :p_in(NULL)
1268 ,p_es(NULL)
1269 ,sys(demuxer)
1270 {}
1271
1272 virtual ~matroska_stream_c()
1273 {
1274 delete p_in;
1275 delete p_es;
1276 }
1277
1278 IOCallback *p_in;
1279 EbmlStream *p_es;
1280
1281 std::vector<matroska_segment_c*> segments;
1282
1283 demux_sys_t & sys;
1284 };
1285
1286 typedef struct
1287 {
1288 VLC_COMMON_MEMBERS
1289
1290 demux_t *p_demux;
1291 vlc_mutex_t lock;
1292
1293 vlc_bool_t b_moved;
1294 vlc_bool_t b_clicked;
1295 vlc_bool_t b_key;
1296
1297 } event_thread_t;
1298
1299 class demux_sys_t
1300 {
1301 public:
1302 demux_sys_t( demux_t & demux )
1303 :demuxer(demux)
1304 ,i_pts(0)
1305 ,i_start_pts(0)
1306 ,i_chapter_time(0)
1307 ,meta(NULL)
1308 ,i_current_title(0)
1309 ,p_current_segment(NULL)
1310 ,dvd_interpretor( *this )
1311 ,f_duration(-1.0)
1312 ,b_ui_hooked(false)
1313 ,p_input(NULL)
1314 ,b_pci_packet_set(false)
1315 ,p_ev(NULL)
1316 {
1317 vlc_mutex_init( &demuxer, &lock_demuxer );
1318 }
1319
1320 virtual ~demux_sys_t()
1321 {
1322 StopUiThread();
1323 size_t i;
1324 for ( i=0; i<streams.size(); i++ )
1325 delete streams[i];
1326 for ( i=0; i<opened_segments.size(); i++ )
1327 delete opened_segments[i];
1328 for ( i=0; i<used_segments.size(); i++ )
1329 delete used_segments[i];
1330 if( meta ) vlc_meta_Delete( meta );
1331
1332 while( titles.size() )
1333 { vlc_input_title_Delete( titles.back() ); titles.pop_back();}
1334
1335 vlc_mutex_destroy( &lock_demuxer );
1336 }
1337
1338 /* current data */
1339 demux_t & demuxer;
1340
1341 mtime_t i_pts;
1342 mtime_t i_start_pts;
1343 mtime_t i_chapter_time;
1344
1345 vlc_meta_t *meta;
1346
1347 std::vector<input_title_t*> titles; // matroska editions
1348 size_t i_current_title;
1349
1350 std::vector<matroska_stream_c*> streams;
1351 std::vector<matroska_segment_c*> opened_segments;
1352 std::vector<virtual_segment_c*> used_segments;
1353 virtual_segment_c *p_current_segment;
1354
1355 dvd_command_interpretor_c dvd_interpretor;
1356
1357 /* duration of the stream */
1358 float f_duration;
1359
1360 matroska_segment_c *FindSegment( const EbmlBinary & uid ) const;
1361 chapter_item_c *BrowseCodecPrivate( unsigned int codec_id,
1362 bool (*match)(const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size ),
1363 const void *p_cookie,
1364 size_t i_cookie_size,
1365 virtual_segment_c * & p_segment_found );
1366 chapter_item_c *FindChapter( int64_t i_find_uid, virtual_segment_c * & p_segment_found );
1367
1368 void PreloadFamily( const matroska_segment_c & of_segment );
1369 void PreloadLinked( matroska_segment_c *p_segment );
1370 bool PreparePlayback( virtual_segment_c *p_new_segment );
1371 matroska_stream_c *AnalyseAllSegmentsFound( demux_t *p_demux, EbmlStream *p_estream, bool b_initial = false );
1372 void JumpTo( virtual_segment_c & p_segment, chapter_item_c * p_chapter );
1373
1374 void StartUiThread();
1375 void StopUiThread();
1376 bool b_ui_hooked;
1377 inline void SwapButtons();
1378
1379 /* for spu variables */
1380 input_thread_t *p_input;
1381 pci_t pci_packet;
1382 bool b_pci_packet_set;
1383 uint8_t palette[4][4];
1384 vlc_mutex_t lock_demuxer;
1385
1386 /* event */
1387 event_thread_t *p_ev;
1388 static int EventThread( vlc_object_t *p_this );
1389 static int EventMouse( vlc_object_t *p_this, char const *psz_var,
1390 vlc_value_t oldval, vlc_value_t newval, void *p_data );
1391 static int EventKey( vlc_object_t *p_this, char const *psz_var,
1392 vlc_value_t oldval, vlc_value_t newval, void *p_data );
1393
1394
1395
1396 protected:
1397 virtual_segment_c *VirtualFromSegments( matroska_segment_c *p_segment ) const;
1398 bool IsUsedSegment( matroska_segment_c &p_segment ) const;
1399 };
1400
1401 static int Demux ( demux_t * );
1402 static int Control( demux_t *, int, va_list );
1403 static void Seek ( demux_t *, mtime_t i_date, double f_percent, chapter_item_c *psz_chapter );
1404
1405 #define MKV_IS_ID( el, C ) ( EbmlId( (*el) ) == C::ClassInfos.GlobalId )
1406
1407 static inline char * ToUTF8( const UTFstring &u )
1408 {
1409 return strdup( u.GetUTF8().c_str() );
1410 }
1411
1412 /*****************************************************************************
1413 * Open: initializes matroska demux structures
1414 *****************************************************************************/
1415 static int Open( vlc_object_t * p_this )
1416 {
1417 demux_t *p_demux = (demux_t*)p_this;
1418 demux_sys_t *p_sys;
1419 matroska_stream_c *p_stream;
1420 matroska_segment_c *p_segment;
1421 uint8_t *p_peek;
1422 std::string s_path, s_filename;
1423 vlc_stream_io_callback *p_io_callback;
1424 EbmlStream *p_io_stream;
1425
1426 /* peek the begining */
1427 if( stream_Peek( p_demux->s, &p_peek, 4 ) < 4 ) return VLC_EGENERIC;
1428
1429 /* is a valid file */
1430 if( p_peek[0] != 0x1a || p_peek[1] != 0x45 ||
1431 p_peek[2] != 0xdf || p_peek[3] != 0xa3 ) return VLC_EGENERIC;
1432
1433 /* Set the demux function */
1434 p_demux->pf_demux = Demux;
1435 p_demux->pf_control = Control;
1436 p_demux->p_sys = p_sys = new demux_sys_t( *p_demux );
1437
1438 p_io_callback = new vlc_stream_io_callback( p_demux->s, VLC_FALSE );
1439 p_io_stream = new EbmlStream( *p_io_callback );
1440
1441 if( p_io_stream == NULL )
1442 {
1443 msg_Err( p_demux, "failed to create EbmlStream" );
1444 delete p_io_callback;
1445 delete p_sys;
1446 return VLC_EGENERIC;
1447 }
1448
1449 p_stream = p_sys->AnalyseAllSegmentsFound( p_demux, p_io_stream, true );
1450 if( p_stream == NULL )
1451 {
1452 msg_Err( p_demux, "cannot find KaxSegment" );
1453 goto error;
1454 }
1455 p_sys->streams.push_back( p_stream );
1456
1457 p_stream->p_in = p_io_callback;
1458 p_stream->p_es = p_io_stream;
1459
1460 for (size_t i=0; i<p_stream->segments.size(); i++)
1461 {
1462 p_stream->segments[i]->Preload();
1463 }
1464
1465 p_segment = p_stream->segments[0];
1466 if( p_segment->cluster != NULL )
1467 {
1468 msg_Warn( p_demux, "cannot find any cluster, damaged file ?" );
1469
1470 // reset the stream reading to the first cluster of the segment used
1471 p_stream->p_in->setFilePointer( p_segment->cluster->GetElementPosition() );
1472 }
1473
1474 if (config_GetInt( p_demux, "mkv-preload-local-dir" ))
1475 {
1476 /* get the files from the same dir from the same family (based on p_demux->psz_path) */
1477 if (p_demux->psz_path[0] != '\0' && !strcmp(p_demux->psz_access, ""))
1478 {
1479 // assume it's a regular file
1480 // get the directory path
1481 s_path = p_demux->psz_path;
1482 if (s_path.at(s_path.length() - 1) == DIRECTORY_SEPARATOR)
1483 {
1484 s_path = s_path.substr(0,s_path.length()-1);
1485 }
1486 else
1487 {
1488 if (s_path.find_last_of(DIRECTORY_SEPARATOR) > 0)
1489 {
1490 s_path = s_path.substr(0,s_path.find_last_of(DIRECTORY_SEPARATOR));
1491 }
1492 }
1493
1494 struct dirent *p_file_item;
1495 DIR *p_src_dir = opendir(s_path.c_str());
1496
1497 if (p_src_dir != NULL)
1498 {
1499 while ((p_file_item = (dirent *) readdir(p_src_dir)))
1500 {
1501 if (strlen(p_file_item->d_name) > 4)
1502 {
1503 s_filename = s_path + DIRECTORY_SEPARATOR + p_file_item->d_name;
1504
1505 #ifdef WIN32
1506 if (!strcasecmp(s_filename.c_str(), p_demux->psz_path))
1507 #else
1508 if (!s_filename.compare(p_demux->psz_path))
1509 #endif
1510 continue; // don't reuse the original opened file
1511
1512 #if defined(__GNUC__) && (__GNUC__ < 3)
1513 if (!s_filename.compare("mkv", s_filename.length() - 3, 3) ||
1514 !s_filename.compare("mka", s_filename.length() - 3, 3))
1515 #else
1516 if (!s_filename.compare(s_filename.length() - 3, 3, "mkv") ||
1517 !s_filename.compare(s_filename.length() - 3, 3, "mka"))
1518 #endif
1519 {
1520 // test wether this file belongs to our family
1521 stream_t *p_file_stream = stream_UrlNew( p_demux, s_filename.c_str());
1522 if ( p_file_stream != NULL )
1523 {
1524 vlc_stream_io_callback *p_file_io = new vlc_stream_io_callback( p_file_stream, VLC_TRUE );
1525 EbmlStream *p_estream = new EbmlStream(*p_file_io);
1526
1527 p_stream = p_sys->AnalyseAllSegmentsFound( p_demux, p_estream );
1528
1529 if ( p_stream == NULL )
1530 {
1531 msg_Dbg( p_demux, "the file '%s' will not be used", s_filename.c_str() );
1532 delete p_estream;
1533 delete p_file_io;
1534 }
1535 else
1536 {
1537 p_stream->p_in = p_file_io;
1538 p_stream->p_es = p_estream;
1539 p_sys->streams.push_back( p_stream );
1540 }
1541 }
1542 else
1543 {
1544 msg_Dbg( p_demux, "the file '%s' cannot be opened", s_filename.c_str() );
1545 }
1546 }
1547 }
1548 }
1549 closedir( p_src_dir );
1550 }
1551 }
1552
1553 p_sys->PreloadFamily( *p_segment );
1554 }
1555
1556 p_sys->PreloadLinked( p_segment );
1557
1558 if ( !p_sys->PreparePlayback( NULL ) )
1559 {
1560 msg_Err( p_demux, "cannot use the segment" );
1561 goto error;
1562 }
1563
1564 p_sys->StartUiThread();
1565
1566 return VLC_SUCCESS;
1567
1568 error:
1569 delete p_sys;
1570 return VLC_EGENERIC;
1571 }
1572
1573 /*****************************************************************************
1574 * Close: frees unused data
1575 *****************************************************************************/
1576 static void Close( vlc_object_t *p_this )
1577 {
1578 demux_t *p_demux = (demux_t*)p_this;
1579 demux_sys_t *p_sys = p_demux->p_sys;
1580
1581 delete p_sys;
1582 }
1583
1584 /*****************************************************************************
1585 * Control:
1586 *****************************************************************************/
1587 static int Control( demux_t *p_demux, int i_query, va_list args )
1588 {
1589 demux_sys_t *p_sys = p_demux->p_sys;
1590 int64_t *pi64;
1591 double *pf, f;
1592 int i_skp;
1593 size_t i_idx;
1594
1595 vlc_meta_t **pp_meta;
1596
1597 switch( i_query )
1598 {
1599 case DEMUX_GET_META:
1600 pp_meta = (vlc_meta_t**)va_arg( args, vlc_meta_t** );
1601 *pp_meta = vlc_meta_Duplicate( p_sys->meta );
1602 return VLC_SUCCESS;
1603
1604 case DEMUX_GET_LENGTH:
1605 pi64 = (int64_t*)va_arg( args, int64_t * );
1606 if( p_sys->f_duration > 0.0 )
1607 {
1608 *pi64 = (int64_t)(p_sys->f_duration * 1000);
1609 return VLC_SUCCESS;
1610 }
1611 return VLC_EGENERIC;
1612
1613 case DEMUX_GET_POSITION:
1614 pf = (double*)va_arg( args, double * );
1615 if ( p_sys->f_duration > 0.0 )
1616 *pf = (double)(p_sys->i_pts >= p_sys->i_start_pts ? p_sys->i_pts : p_sys->i_start_pts ) / (1000.0 * p_sys->f_duration);
1617 return VLC_SUCCESS;
1618
1619 case DEMUX_SET_POSITION:
1620 f = (double)va_arg( args, double );
1621 Seek( p_demux, -1, f, NULL );
1622 return VLC_SUCCESS;
1623
1624 case DEMUX_GET_TIME:
1625 pi64 = (int64_t*)va_arg( args, int64_t * );
1626 *pi64 = p_sys->i_pts;
1627 return VLC_SUCCESS;
1628
1629 case DEMUX_GET_TITLE_INFO:
1630 if( p_sys->titles.size() )
1631 {
1632 input_title_t ***ppp_title = (input_title_t***)va_arg( args, input_title_t*** );
1633 int *pi_int = (int*)va_arg( args, int* );
1634
1635 *pi_int = p_sys->titles.size();
1636 *ppp_title = (input_title_t**)malloc( sizeof( input_title_t**) * p_sys->titles.size() );
1637
1638 for( size_t i = 0; i < p_sys->titles.size(); i++ )
1639 {
1640 (*ppp_title)[i] = vlc_input_title_Duplicate( p_sys->titles[i] );
1641 }
1642
1643 return VLC_SUCCESS;
1644 }
1645 return VLC_EGENERIC;
1646
1647 case DEMUX_SET_TITLE:
1648 /* TODO handle editions as titles */
1649 i_idx = (int)va_arg( args, int );
1650 if( i_idx < p_sys->used_segments.size() )
1651 {
1652 p_sys->JumpTo( *p_sys->used_segments[i_idx], NULL );
1653 return VLC_SUCCESS;
1654 }
1655 return VLC_EGENERIC;
1656
1657 case DEMUX_SET_SEEKPOINT:
1658 i_skp = (int)va_arg( args, int );
1659
1660 // TODO change the way it works with the << & >> buttons on the UI (+1/-1 instead of a number)
1661 if( p_sys->titles.size() && i_skp < p_sys->titles[p_sys->i_current_title]->i_seekpoint)
1662 {
1663 Seek( p_demux, (int64_t)p_sys->titles[p_sys->i_current_title]->seekpoint[i_skp]->i_time_offset, -1, NULL);
1664 p_demux->info.i_seekpoint |= INPUT_UPDATE_SEEKPOINT;
1665 p_demux->info.i_seekpoint = i_skp;
1666 return VLC_SUCCESS;
1667 }
1668 return VLC_EGENERIC;
1669
1670 case DEMUX_SET_TIME:
1671 case DEMUX_GET_FPS:
1672 default:
1673 return VLC_EGENERIC;
1674 }
1675 }
1676
1677 #if LIBMATROSKA_VERSION >= 0x000800
1678 int matroska_segment_c::BlockGet( KaxBlock * & pp_block, KaxSimpleBlock * & pp_simpleblock, int64_t *pi_ref1, int64_t *pi_ref2, int64_t *pi_duration )
1679 #else
1680 int matroska_segment_c::BlockGet( KaxBlock * & pp_block, int64_t *pi_ref1, int64_t *pi_ref2, int64_t *pi_duration )
1681 #endif
1682 {
1683 #if LIBMATROSKA_VERSION >= 0x000800
1684 pp_simpleblock = NULL;
1685 #endif
1686 pp_block = NULL;
1687 *pi_ref1 = 0;
1688 *pi_ref2 = 0;
1689
1690 for( ;; )
1691 {
1692 EbmlElement *el = NULL;
1693 int i_level;
1694
1695 if ( ep == NULL )
1696 return VLC_EGENERIC;
1697
1698 #if LIBMATROSKA_VERSION >= 0x000800
1699 if( pp_simpleblock != NULL || ((el = ep->Get()) == NULL && pp_block != NULL) )
1700 #else
1701 if( (el = ep->Get()) == NULL && pp_block != NULL )
1702 #endif
1703 {
1704 /* update the index */
1705 #define idx p_indexes[i_index - 1]
1706 if( i_index > 0 && idx.i_time == -1 )
1707 {
1708 #if LIBMATROSKA_VERSION >= 0x000800
1709 if ( pp_simpleblock != NULL )
1710 idx.i_time = pp_simpleblock->GlobalTimecode() / (mtime_t)1000;
1711 else
1712 #endif
1713 idx.i_time = (*pp_block).GlobalTimecode() / (mtime_t)1000;
1714 idx.b_key = *pi_ref1 == 0 ? VLC_TRUE : VLC_FALSE;
1715 }
1716 #undef idx
1717 return VLC_SUCCESS;
1718 }
1719
1720 i_level = ep->GetLevel();
1721
1722 if( el == NULL )
1723 {
1724 if( i_level > 1 )
1725 {
1726 ep->Up();
1727 continue;
1728 }
1729 msg_Warn( &sys.demuxer, "EOF" );
1730 return VLC_EGENERIC;
1731 }
1732
1733 /* do parsing */
1734 switch ( i_level )
1735 {
1736 case 1:
1737 if( MKV_IS_ID( el, KaxCluster ) )
1738 {
1739 cluster = (KaxCluster*)el;
1740 i_cluster_pos = cluster->GetElementPosition();
1741
1742 /* add it to the index */
1743 if( i_index == 0 ||
1744 ( i_index > 0 && p_indexes[i_index - 1].i_position < (int64_t)cluster->GetElementPosition() ) )
1745 {
1746 IndexAppendCluster( cluster );
1747 }
1748
1749 // reset silent tracks
1750 for (size_t i=0; i<tracks.size(); i++)
1751 {
1752 tracks[i]->b_silent = VLC_FALSE;
1753 }
1754
1755 ep->Down();
1756 }
1757 else if( MKV_IS_ID( el, KaxCues ) )
1758 {
1759 msg_Warn( &sys.demuxer, "find KaxCues FIXME" );
1760 return VLC_EGENERIC;
1761 }
1762 else
1763 {
1764 msg_Dbg( &sys.demuxer, "unknown (%s)", typeid( el ).name() );
1765 }
1766 break;
1767 case 2:
1768 if( MKV_IS_ID( el, KaxClusterTimecode ) )
1769 {
1770 KaxClusterTimecode &ctc = *(KaxClusterTimecode*)el;
1771
1772 ctc.ReadData( es.I_O(), SCOPE_ALL_DATA );
1773 cluster->InitTimecode( uint64( ctc ), i_timescale );
1774 }
1775 else if( MKV_IS_ID( el, KaxClusterSilentTracks ) )
1776 {
1777 ep->Down();
1778 }
1779 else if( MKV_IS_ID( el, KaxBlockGroup ) )
1780 {
1781 i_block_pos = el->GetElementPosition();
1782 ep->Down();
1783 }
1784 #if LIBMATROSKA_VERSION >= 0x000800
1785 else if( MKV_IS_ID( el, KaxSimpleBlock ) )
1786 {
1787 pp_simpleblock = (KaxSimpleBlock*)el;
1788
1789 pp_simpleblock->ReadData( es.I_O() );
1790 pp_simpleblock->SetParent( *cluster );
1791 }
1792 #endif
1793 break;
1794 case 3:
1795 if( MKV_IS_ID( el, KaxBlock ) )
1796 {
1797 pp_block = (KaxBlock*)el;
1798
1799 pp_block->ReadData( es.I_O() );
1800 pp_block->SetParent( *cluster );
1801
1802 ep->Keep();
1803 }
1804 else if( MKV_IS_ID( el, KaxBlockDuration ) )
1805 {
1806 KaxBlockDuration &dur = *(KaxBlockDuration*)el;
1807
1808 dur.ReadData( es.I_O() );
1809 *pi_duration = uint64( dur );
1810 }
1811 else if( MKV_IS_ID( el, KaxReferenceBlock ) )
1812 {
1813 KaxReferenceBlock &ref = *(KaxReferenceBlock*)el;
1814
1815 ref.ReadData( es.I_O() );
1816 if( *pi_ref1 == 0 )
1817 {
1818 *pi_ref1 = int64( ref ) * cluster->GlobalTimecodeScale();
1819 }
1820 else if( *pi_ref2 == 0 )
1821 {
1822 *pi_ref2 = int64( ref ) * cluster->GlobalTimecodeScale();
1823 }
1824 }
1825 else if( MKV_IS_ID( el, KaxClusterSilentTrackNumber ) )
1826 {
1827 KaxClusterSilentTrackNumber &track_num = *(KaxClusterSilentTrackNumber*)el;
1828 track_num.ReadData( es.I_O() );
1829 // find the track
1830 for (size_t i=0; i<tracks.size(); i++)
1831 {
1832 if ( tracks[i]->i_number == uint32(track_num))
1833 {
1834 tracks[i]->b_silent = VLC_TRUE;
1835 break;
1836 }
1837 }
1838 }
1839 break;
1840 default:
1841 msg_Err( &sys.demuxer, "invalid level = %d", i_level );
1842 return VLC_EGENERIC;
1843 }
1844 }
1845 }
1846
1847 static block_t *MemToBlock( demux_t *p_demux, uint8_t *p_mem, int i_mem, size_t offset)
1848 {
1849 block_t *p_block;
1850 if( !(p_block = block_New( p_demux, i_mem + offset ) ) ) return NULL;
1851 memcpy( p_block->p_buffer + offset, p_mem, i_mem );
1852 //p_block->i_rate = p_input->stream.control.i_rate;
1853 return p_block;
1854 }
1855
1856 #if LIBMATROSKA_VERSION >= 0x000800
1857 static void BlockDecode( demux_t *p_demux, KaxBlock *block, KaxSimpleBlock *simpleblock,
1858 mtime_t i_pts, mtime_t i_duration, bool f_mandatory )
1859 #else
1860 static void BlockDecode( demux_t *p_demux, KaxBlock *block, mtime_t i_pts,
1861 mtime_t i_duration, bool f_mandatory )
1862 #endif
1863 {
1864 demux_sys_t *p_sys = p_demux->p_sys;
1865 matroska_segment_c *p_segment = p_sys->p_current_segment->Segment();
1866
1867 size_t i_track;
1868 unsigned int i;
1869 vlc_bool_t b;
1870
1871 #define tk p_segment->tracks[i_track]
1872 for( i_track = 0; i_track < p_segment->tracks.size(); i_track++ )
1873 {
1874 #if LIBMATROSKA_VERSION >= 0x000800
1875 if( (block != NULL && tk->i_number == block->TrackNum()) ||
1876 (simpleblock != NULL && tk->i_number == simpleblock->TrackNum()))
1877 #else
1878 if( tk->i_number == block->TrackNum() )
1879 #endif
1880 {
1881 break;
1882 }
1883 }
1884
1885 if( i_track >= p_segment->tracks.size() )
1886 {
1887 msg_Err( p_demux, "invalid track number=%d", block->TrackNum() );
1888 return;
1889 }
1890 if( tk->fmt.i_cat != NAV_ES && tk->p_es == NULL )
1891 {
1892 msg_Err( p_demux, "unknown track number=%d", block->TrackNum() );
1893 return;
1894 }
1895 if( i_pts + i_duration < p_sys->i_start_pts && tk->fmt.i_cat == AUDIO_ES )
1896 {
1897 return; /* discard audio packets that shouldn't be rendered */
1898 }
1899
1900 if ( tk->fmt.i_cat != NAV_ES )
1901 {
1902 es_out_Control( p_demux->out, ES_OUT_GET_ES_STATE, tk->p_es, &b );
1903
1904 if( !b )
1905 {
1906 tk->b_inited = VLC_FALSE;
1907 return;
1908 }
1909 }
1910
1911
1912 /* First send init data */
1913 if( !tk->b_inited && tk->i_data_init > 0 )
1914 {
1915 block_t *p_init;
1916
1917 msg_Dbg( p_demux, "sending header (%d bytes)", tk->i_data_init );
1918 p_init = MemToBlock( p_demux, tk->p_data_init, tk->i_data_init, 0 );
1919 if( p_init ) es_out_Send( p_demux->out, tk->p_es, p_init );
1920 }
1921 tk->b_inited = VLC_TRUE;
1922
1923
1924 #if LIBMATROSKA_VERSION >= 0x000800
1925 for( i = 0;
1926 (block != NULL && i < block->NumberFrames()) || (simpleblock != NULL && i < simpleblock->NumberFrames());
1927 i++ )
1928 #else
1929 for( i = 0; i < block->NumberFrames(); i++ )
1930 #endif
1931 {
1932 block_t *p_block;
1933 DataBuffer *data;
1934 #if LIBMATROSKA_VERSION >= 0x000800
1935 if ( simpleblock != NULL )
1936 {
1937 data = &simpleblock->GetBuffer(i);
1938 // condition when the DTS is correct (keyframe or B frame == NOT P frame)
1939 f_mandatory = simpleblock->IsDiscardable() || simpleblock->IsKeyframe();
1940 }
1941 else
1942 #endif
1943 data = &block->GetBuffer(i);
1944
1945 if( tk->i_compression_type == MATROSKA_COMPRESSION_HEADER && tk->p_compression_data != NULL )
1946 p_block = MemToBlock( p_demux, data->Buffer(), data->Size(), tk->p_compression_data->GetSize() );
1947 else
1948 p_block = MemToBlock( p_demux, data->Buffer(), data->Size(), 0 );
1949
1950 if( p_block == NULL )
1951 {
1952 break;
1953 }
1954
1955 #if defined(HAVE_ZLIB_H)
1956 if( tk->i_compression_type == MATROSKA_COMPRESSION_ZLIB )
1957 {
1958 p_block = block_zlib_decompress( VLC_OBJECT(p_demux), p_block );
1959 }
1960 else
1961 #endif
1962 if( tk->i_compression_type == MATROSKA_COMPRESSION_HEADER )
1963 {
1964 memcpy( p_block->p_buffer, tk->p_compression_data->GetBuffer(), tk->p_compression_data->GetSize() );
1965 }
1966
1967 if ( tk->fmt.i_cat == NAV_ES )
1968 {
1969 // TODO handle the start/stop times of this packet
1970 if ( p_sys->b_ui_hooked )
1971 {
1972 vlc_mutex_lock( &p_sys->p_ev->lock );
1973 memcpy( &p_sys->pci_packet, &p_block->p_buffer[1], sizeof(pci_t) );
1974 p_sys->SwapButtons();
1975 p_sys->b_pci_packet_set = true;
1976 vlc_mutex_unlock( &p_sys->p_ev->lock );
1977 block_Release( p_block );
1978 }
1979 return;
1980 }
1981 // correct timestamping when B frames are used
1982 if( tk->fmt.i_cat != VIDEO_ES )
1983 {
1984 p_block->i_dts = p_block->i_pts = i_pts;
1985 }
1986 else
1987 {
1988 if( !strcmp( tk->psz_codec, "V_MS/VFW/FOURCC" ) )
1989 {
1990 // in VFW we have no idea about B frames
1991 p_block->i_pts = 0;
1992 p_block->i_dts = i_pts;
1993 }
1994 else
1995 {
1996 p_block->i_pts = i_pts;
1997 if ( f_mandatory )
1998 p_block->i_dts = p_block->i_pts;
1999 else
2000 p_block->i_dts = min( i_pts, tk->i_last_dts + (tk->i_default_duration >> 10));
2001 p_sys->i_pts = p_block->i_dts;
2002 }
2003 }
2004 tk->i_last_dts = p_block->i_dts;
2005
2006 #if 0
2007 msg_Dbg( p_demux, "block i_dts: "I64Fd" / i_pts: "I64Fd, p_block->i_dts, p_block->i_pts);
2008 #endif
2009 if( strcmp( tk->psz_codec, "S_VOBSUB" ) )
2010 {
2011 p_block->i_length = i_duration * 1000;
2012 }
2013
2014 es_out_Send( p_demux->out, tk->p_es, p_block );
2015
2016 /* use time stamp only for first block */
2017 i_pts = 0;
2018 }
2019
2020 #undef tk
2021 }
2022
2023 matroska_stream_c *demux_sys_t::AnalyseAllSegmentsFound( demux_t *p_demux, EbmlStream *p_estream, bool b_initial )
2024 {
2025 int i_upper_lvl = 0;
2026 size_t i;
2027 EbmlElement *p_l0, *p_l1, *p_l2;
2028 bool b_keep_stream = false, b_keep_segment;
2029
2030 // verify the EBML Header
2031 p_l0 = p_estream->FindNextID(EbmlHead::ClassInfos, 0xFFFFFFFFL);
2032 if (p_l0 == NULL)
2033 {
2034 msg_Err( p_demux, "No EBML header found" );
2035 return NULL;
2036 }
2037
2038 // verify we can read this Segment, we only support Matroska version 1 for now
2039 p_l0->Read(*p_estream, EbmlHead::ClassInfos.Context, i_upper_lvl, p_l0, true);
2040
2041 EDocType doc_type = GetChild<EDocType>(*static_cast<EbmlHead*>(p_l0));
2042 if (std::string(doc_type) != "matroska")
2043 {
2044 msg_Err( p_demux, "Not a Matroska file : DocType = %s ", std::string(doc_type).c_str());
2045 return NULL;
2046 }
2047
2048 EDocTypeReadVersion doc_read_version = GetChild<EDocTypeReadVersion>(*static_cast<EbmlHead*>(p_l0));
2049 #if LIBMATROSKA_VERSION >= 0x000800
2050 if (uint64(doc_read_version) > 2)
2051 {
2052 msg_Err( p_demux, "This matroska file is needs version "I64Fd" and this VLC only supports version 1 & 2", uint64(doc_read_version));
2053 return NULL;
2054 }
2055 #else
2056 if (uint64(doc_read_version) != 1)
2057 {
2058 msg_Err( p_demux, "This matroska file is needs version "I64Fd" and this VLC only supports version 1", uint64(doc_read_version));
2059 return NULL;
2060 }
2061 #endif
2062
2063 delete p_l0;
2064
2065
2066 // find all segments in this file
2067 p_l0 = p_estream->FindNextID(KaxSegment::ClassInfos, 0xFFFFFFFFFLL);
2068 if (p_l0 == NULL)
2069 {
2070 return NULL;
2071 }
2072
2073 matroska_stream_c *p_stream1 = new matroska_stream_c( *this );
2074
2075 while (p_l0 != 0)
2076 {
2077 if (EbmlId(*p_l0) == KaxSegment::ClassInfos.GlobalId)
2078 {
2079 EbmlParser *ep;
2080 matroska_segment_c *p_segment1 = new matroska_segment_c( *this, *p_estream );
2081 b_keep_segment = b_initial;
2082
2083 ep = new EbmlParser(p_estream, p_l0, &demuxer );
2084 p_segment1->ep = ep;
2085 p_segment1->segment = (KaxSegment*)p_l0;
2086
2087 while ((p_l1 = ep->Get()))
2088 {
2089 if (MKV_IS_ID(p_l1, KaxInfo))
2090 {
2091 // find the families of this segment
2092 KaxInfo *p_info = static_cast<KaxInfo*>(p_l1);
2093
2094 p_info->Read(*p_estream, KaxInfo::ClassInfos.Context, i_upper_lvl, p_l2, true);
2095 for( i = 0; i < p_info->ListSize(); i++ )
2096 {
2097 EbmlElement *l = (*p_info)[i];
2098
2099 if( MKV_IS_ID( l, KaxSegmentUID ) )
2100 {
2101 KaxSegmentUID *p_uid = static_cast<KaxSegmentUID*>(l);
2102 b_keep_segment = (FindSegment( *p_uid ) == NULL);
2103 if ( !b_keep_segment )
2104 break; // this segment is already known
2105 opened_segments.push_back( p_segment1 );
2106 delete p_segment1->p_segment_uid;
2107 p_segment1->p_segment_uid = new KaxSegmentUID(*p_uid);
2108 }
2109 else if( MKV_IS_ID( l, KaxPrevUID ) )
2110 {
2111 p_segment1->p_prev_segment_uid = new KaxPrevUID( *static_cast<KaxPrevUID*>(l) );
2112 }
2113 else if( MKV_IS_ID( l, KaxNextUID ) )
2114 {
2115 p_segment1->p_next_segment_uid = new KaxNextUID( *static_cast<KaxNextUID*>(l) );
2116 }
2117 else if( MKV_IS_ID( l, KaxSegmentFamily ) )
2118 {
2119 KaxSegmentFamily *p_fam = new KaxSegmentFamily( *static_cast<KaxSegmentFamily*>(l) );
2120 p_segment1->families.push_back( p_fam );
2121 }
2122 }
2123 break;
2124 }
2125 }
2126 if ( b_keep_segment )
2127 {
2128 b_keep_stream = true;
2129 p_stream1->segments.push_back( p_segment1 );
2130 }
2131 else
2132 delete p_segment1;
2133 }
2134 if (p_l0->IsFiniteSize() )
2135 {
2136 p_l0->SkipData(*p_estream, KaxMatroska_Context);
2137 p_l0 = p_estream->FindNextID(KaxSegment::ClassInfos, 0xFFFFFFFFL);
2138 }
2139 else
2140 p_l0 = p_l0->SkipData(*p_estream, KaxSegment_Context);
2141 }
2142
2143 if ( !b_keep_stream )
2144 {
2145 delete p_stream1;
2146 p_stream1 = NULL;
2147 }
2148
2149 return p_stream1;
2150 }
2151
2152 bool matroska_segment_c::Select( mtime_t i_start_time )
2153 {
2154 size_t i_track;
2155
2156 /* add all es */
2157 msg_Dbg( &sys.demuxer, "found %d es", (int)tracks.size() );
2158 sys.b_pci_packet_set = false;
2159
2160 for( i_track = 0; i_track < tracks.size(); i_track++ )
2161 {
2162 if( tracks[i_track]->fmt.i_cat == UNKNOWN_ES )
2163 {
2164 msg_Warn( &sys.demuxer, "invalid track[%d, n=%d]", (int)i_track, tracks[i_track]->i_number );
2165 tracks[i_track]->p_es = NULL;
2166 continue;
2167 }
2168
2169 if( !strcmp( tracks[i_track]->psz_codec, "V_MS/VFW/FOURCC" ) )
2170 {
2171 if( tracks[i_track]->i_extra_data < (int)sizeof( BITMAPINFOHEADER ) )
2172 {
2173 msg_Err( &sys.demuxer, "missing/invalid BITMAPINFOHEADER" );
2174 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
2175 }
2176 else
2177 {
2178 BITMAPINFOHEADER *p_bih = (BITMAPINFOHEADER*)tracks[i_track]->p_extra_data;
2179
2180 tracks[i_track]->fmt.video.i_width = GetDWLE( &p_bih->biWidth );
2181 tracks[i_track]->fmt.video.i_height= GetDWLE( &p_bih->biHeight );
2182 tracks[i_track]->fmt.i_codec = GetFOURCC( &p_bih->biCompression );
2183
2184 tracks[i_track]->fmt.i_extra = GetDWLE( &p_bih->biSize ) - sizeof( BITMAPINFOHEADER );
2185 if( tracks[i_track]->fmt.i_extra > 0 )
2186 {
2187 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
2188 memcpy( tracks[i_track]->fmt.p_extra, &p_bih[1], tracks[i_track]->fmt.i_extra );
2189 }
2190 }
2191 }
2192 else if( !strcmp( tracks[i_track]->psz_codec, "V_MPEG1" ) ||
2193 !strcmp( tracks[i_track]->psz_codec, "V_MPEG2" ) )
2194 {
2195 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'm', 'p', 'g', 'v' );
2196 }
2197 else if( !strncmp( tracks[i_track]->psz_codec, "V_MPEG4", 7 ) )
2198 {
2199 if( !strcmp( tracks[i_track]->psz_codec, "V_MPEG4/MS/V3" ) )
2200 {
2201 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'D', 'I', 'V', '3' );
2202 }
2203 else if( !strcmp( tracks[i_track]->psz_codec, "V_MPEG4/ISO/AVC" ) )
2204 {
2205 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'a', 'v', 'c', '1' );
2206 tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
2207 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
2208 memcpy( tracks[i_track]->fmt.p_extra,tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
2209 }
2210 else
2211 {
2212 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'v' );
2213 }
2214 }
2215 else if( !strcmp( tracks[i_track]->psz_codec, "V_QUICKTIME" ) )
2216 {
2217 MP4_Box_t *p_box = (MP4_Box_t*)malloc( sizeof( MP4_Box_t ) );
2218 stream_t *p_mp4_stream = stream_MemoryNew( VLC_OBJECT(&sys.demuxer),
2219 tracks[i_track]->p_extra_data,
2220 tracks[i_track]->i_extra_data,
2221 VLC_FALSE );
2222 MP4_ReadBoxCommon( p_mp4_stream, p_box );
2223 MP4_ReadBox_sample_vide( p_mp4_stream, p_box );
2224 tracks[i_track]->fmt.i_codec = p_box->i_type;
2225 tracks[i_track]->fmt.video.i_width = p_box->data.p_sample_vide->i_width;
2226 tracks[i_track]->fmt.video.i_height = p_box->data.p_sample_vide->i_height;
2227 tracks[i_track]->fmt.i_extra = p_box->data.p_sample_vide->i_qt_image_description;
2228 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
2229 memcpy( tracks[i_track]->fmt.p_extra, p_box->data.p_sample_vide->p_qt_image_description, tracks[i_track]->fmt.i_extra );
2230 MP4_FreeBox_sample_vide( p_box );
2231 stream_Delete( p_mp4_stream );
2232 }
2233 else if( !strcmp( tracks[i_track]->psz_codec, "A_MS/ACM" ) )
2234 {
2235 if( tracks[i_track]->i_extra_data < (int)sizeof( WAVEFORMATEX ) )
2236 {
2237 msg_Err( &sys.demuxer, "missing/invalid WAVEFORMATEX" );
2238 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
2239 }
2240 else
2241 {
2242 WAVEFORMATEX *p_wf = (WAVEFORMATEX*)tracks[i_track]->p_extra_data;
2243
2244 wf_tag_to_fourcc( GetWLE( &p_wf->wFormatTag ), &tracks[i_track]->fmt.i_codec, NULL );
2245
2246 tracks[i_track]->fmt.audio.i_channels = GetWLE( &p_wf->nChannels );
2247 tracks[i_track]->fmt.audio.i_rate = GetDWLE( &p_wf->nSamplesPerSec );
2248 tracks[i_track]->fmt.i_bitrate = GetDWLE( &p_wf->nAvgBytesPerSec ) * 8;
2249 tracks[i_track]->fmt.audio.i_blockalign = GetWLE( &p_wf->nBlockAlign );;
2250 tracks[i_track]->fmt.audio.i_bitspersample = GetWLE( &p_wf->wBitsPerSample );
2251
2252 tracks[i_track]->fmt.i_extra = GetWLE( &p_wf->cbSize );
2253 if( tracks[i_track]->fmt.i_extra > 0 )
2254 {
2255 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
2256 memcpy( tracks[i_track]->fmt.p_extra, &p_wf[1], tracks[i_track]->fmt.i_extra );
2257 }
2258 }
2259 }
2260 else if( !strcmp( tracks[i_track]->psz_codec, "A_MPEG/L3" ) ||
2261 !strcmp( tracks[i_track]->psz_codec, "A_MPEG/L2" ) ||
2262 !strcmp( tracks[i_track]->psz_codec, "A_MPEG/L1" ) )
2263 {
2264 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'm', 'p', 'g', 'a' );
2265 }
2266 else if( !strcmp( tracks[i_track]->psz_codec, "A_AC3" ) )
2267 {
2268 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'a', '5', '2', ' ' );
2269 }
2270 else if( !strcmp( tracks[i_track]->psz_codec, "A_DTS" ) )
2271 {
2272 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'd', 't', 's', ' ' );
2273 }
2274 else if( !strcmp( tracks[i_track]->psz_codec, "A_FLAC" ) )
2275 {
2276 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'f', 'l', 'a', 'c' );
2277 tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
2278 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
2279 memcpy( tracks[i_track]->fmt.p_extra,tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
2280 }
2281 else if( !strcmp( tracks[i_track]->psz_codec, "A_VORBIS" ) )
2282 {
2283 int i, i_offset = 1, i_size[3], i_extra;
2284 uint8_t *p_extra;
2285
2286 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'v', 'o', 'r', 'b' );
2287
2288 /* Split the 3 headers */
2289 if( tracks[i_track]->p_extra_data[0] != 0x02 )
2290 msg_Err( &sys.demuxer, "invalid vorbis header" );
2291
2292 for( i = 0; i < 2; i++ )
2293 {
2294 i_size[i] = 0;
2295 while( i_offset < tracks[i_track]->i_extra_data )
2296 {
2297 i_size[i] += tracks[i_track]->p_extra_data[i_offset];
2298 if( tracks[i_track]->p_extra_data[i_offset++] != 0xff ) break;
2299 }
2300 }
2301
2302 i_size[0] = __MIN(i_size[0], tracks[i_track]->i_extra_data - i_offset);
2303 i_size[1] = __MIN(i_size[1], tracks[i_track]->i_extra_data -i_offset -i_size[0]);
2304 i_size[2] = tracks[i_track]->i_extra_data - i_offset - i_size[0] - i_size[1];
2305
2306 tracks[i_track]->fmt.i_extra = 3 * 2 + i_size[0] + i_size[1] + i_size[2];
2307 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
2308 p_extra = (uint8_t *)tracks[i_track]->fmt.p_extra; i_extra = 0;
2309 for( i = 0; i < 3; i++ )
2310 {
2311 *(p_extra++) = i_size[i] >> 8;
2312 *(p_extra++) = i_size[i] & 0xFF;
2313 memcpy( p_extra, tracks[i_track]->p_extra_data + i_offset + i_extra,
2314 i_size[i] );
2315 p_extra += i_size[i];
2316 i_extra += i_size[i];
2317 }
2318 }
2319 else if( !strncmp( tracks[i_track]->psz_codec, "A_AAC/MPEG2/", strlen( "A_AAC/MPEG2/" ) ) ||
2320 !strncmp( tracks[i_track]->psz_codec, "A_AAC/MPEG4/", strlen( "A_AAC/MPEG4/" ) ) )
2321 {
2322 int i_profile, i_srate, sbr = 0;
2323 static unsigned int i_sample_rates[] =
2324 {
2325 96000, 88200, 64000, 48000, 44100, 32000, 24000, 22050,
2326 16000, 12000, 11025, 8000, 7350, 0, 0, 0
2327 };
2328
2329 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'a' );
2330 /* create data for faad (MP4DecSpecificDescrTag)*/
2331
2332 if( !strcmp( &tracks[i_track]->psz_codec[12], "MAIN" ) )
2333 {
2334 i_profile = 0;
2335 }
2336 else if( !strcmp( &tracks[i_track]->psz_codec[12], "LC" ) )
2337 {
2338 i_profile = 1;
2339 }
2340 else if( !strcmp( &tracks[i_track]->psz_codec[12], "SSR" ) )
2341 {
2342 i_profile = 2;
2343 }
2344 else if( !strcmp( &tracks[i_track]->psz_codec[12], "LC/SBR" ) )
2345 {
2346 i_profile = 1;
2347 sbr = 1;
2348 }
2349 else
2350 {
2351 i_profile = 3;
2352 }
2353
2354 for( i_srate = 0; i_srate < 13; i_srate++ )
2355 {
2356 if( i_sample_rates[i_srate] == tracks[i_track]->i_original_rate )
2357 {
2358 break;
2359 }
2360 }
2361 msg_Dbg( &sys.demuxer, "profile=%d srate=%d", i_profile, i_srate );
2362
2363 tracks[i_track]->fmt.i_extra = sbr ? 5 : 2;
2364 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->fmt.i_extra );
2365 ((uint8_t*)tracks[i_track]->fmt.p_extra)[0] = ((i_profile + 1) << 3) | ((i_srate&0xe) >> 1);
2366 ((uint8_t*)tracks[i_track]->fmt.p_extra)[1] = ((i_srate & 0x1) << 7) | (tracks[i_track]->fmt.audio.i_channels << 3);
2367 if (sbr != 0)
2368 {
2369 int syncExtensionType = 0x2B7;
2370 int iDSRI;
2371 for (iDSRI=0; iDSRI<13; iDSRI++)
2372 if( i_sample_rates[iDSRI] == tracks[i_track]->fmt.audio.i_rate )
2373 break;
2374 ((uint8_t*)tracks[i_track]->fmt.p_extra)[2] = (syncExtensionType >> 3) & 0xFF;
2375 ((uint8_t*)tracks[i_track]->fmt.p_extra)[3] = ((syncExtensionType & 0x7) << 5) | 5;
2376 ((uint8_t*)tracks[i_track]->fmt.p_extra)[4] = ((1 & 0x1) << 7) | (iDSRI << 3);
2377 }
2378 }
2379 else if( !strcmp( tracks[i_track]->psz_codec, "A_AAC" ) )
2380 {
2381 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'a' );
2382 tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
2383 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
2384 memcpy( tracks[i_track]->fmt.p_extra, tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
2385 }
2386 else if( !strcmp( tracks[i_track]->psz_codec, "A_PCM/INT/BIG" ) ||
2387 !strcmp( tracks[i_track]->psz_codec, "A_PCM/INT/LIT" ) ||
2388 !strcmp( tracks[i_track]->psz_codec, "A_PCM/FLOAT/IEEE" ) )
2389 {
2390 if( !strcmp( tracks[i_track]->psz_codec, "A_PCM/INT/BIG" ) )
2391 {
2392 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 't', 'w', 'o', 's' );
2393 }
2394 else
2395 {
2396 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'a', 'r', 'a', 'w' );
2397 }
2398 tracks[i_track]->fmt.audio.i_blockalign = ( tracks[i_track]->fmt.audio.i_bitspersample + 7 ) / 8 * tracks[i_track]->fmt.audio.i_channels;
2399 }
2400 else if( !strcmp( tracks[i_track]->psz_codec, "A_TTA1" ) )
2401 {
2402 /* FIXME: support this codec */
2403 msg_Err( &sys.demuxer, "TTA not supported yet[%d, n=%d]", (int)i_track, tracks[i_track]->i_number );
2404 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
2405 }
2406 else if( !strcmp( tracks[i_track]->psz_codec, "A_WAVPACK4" ) )
2407 {
2408 /* FIXME: support this codec */
2409 msg_Err( &sys.demuxer, "Wavpack not supported yet[%d, n=%d]", (int)i_track, tracks[i_track]->i_number );
2410 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
2411 }
2412 else if( !strcmp( tracks[i_track]->psz_codec, "S_TEXT/UTF8" ) )
2413 {
2414 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 's', 'u', 'b', 't' );
2415 tracks[i_track]->fmt.subs.psz_encoding = strdup( "UTF-8" );
2416 }
2417 else if( !strcmp( tracks[i_track]->psz_codec, "S_TEXT/SSA" ) ||
2418 !strcmp( tracks[i_track]->psz_codec, "S_TEXT/ASS" ) ||
2419 !strcmp( tracks[i_track]->psz_codec, "S_SSA" ) ||
2420 !strcmp( tracks[i_track]->psz_codec, "S_ASS" ))
2421 {
2422 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 's', 's', 'a', ' ' );
2423 tracks[i_track]->fmt.subs.psz_encoding = strdup( "UTF-8" );
2424 if( tracks[i_track]->i_extra_data )
2425 {
2426 tracks[i_track]->fmt.i_extra = tracks[i_track]->i_extra_data;
2427 tracks[i_track]->fmt.p_extra = malloc( tracks[i_track]->i_extra_data );
2428 memcpy( tracks[i_track]->fmt.p_extra, tracks[i_track]->p_extra_data, tracks[i_track]->i_extra_data );
2429 }
2430 }
2431 else if( !strcmp( tracks[i_track]->psz_codec, "S_VOBSUB" ) )
2432 {
2433 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 's','p','u',' ' );
2434 if( tracks[i_track]->i_extra_data )
2435 {
2436 char *p_start;
2437 char *p_buf = (char *)malloc( tracks[i_track]->i_extra_data + 1);
2438 memcpy( p_buf, tracks[i_track]->p_extra_data , tracks[i_track]->i_extra_data );
2439 p_buf[tracks[i_track]->i_extra_data] = '\0';
2440
2441 p_start = strstr( p_buf, "size:" );
2442 if( sscanf( p_start, "size: %dx%d",
2443 &tracks[i_track]->fmt.subs.spu.i_original_frame_width, &tracks[i_track]->fmt.subs.spu.i_original_frame_height ) == 2 )
2444 {
2445 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 );
2446 }
2447 else
2448 {
2449 msg_Warn( &sys.demuxer, "reading original frame size for vobsub failed" );
2450 }
2451 free( p_buf );
2452 }
2453 }
2454 else if( !strcmp( tracks[i_track]->psz_codec, "B_VOBBTN" ) )
2455 {
2456 tracks[i_track]->fmt.i_cat = NAV_ES;
2457 continue;
2458 }
2459 else
2460 {
2461 msg_Err( &sys.demuxer, "unknow codec id=`%s'", tracks[i_track]->psz_codec );
2462 tracks[i_track]->fmt.i_codec = VLC_FOURCC( 'u', 'n', 'd', 'f' );
2463 }
2464 if( tracks[i_track]->b_default )
2465 {
2466 tracks[i_track]->fmt.i_priority = 1000;
2467 }
2468
2469 tracks[i_track]->p_es = es_out_Add( sys.demuxer.out, &tracks[i_track]->fmt );
2470
2471 es_out_Control( sys.demuxer.out, ES_OUT_SET_NEXT_DISPLAY_TIME, tracks[i_track]->p_es, i_start_time );
2472 }
2473
2474 sys.i_start_pts = i_start_time;
2475 // reset the stream reading to the first cluster of the segment used
2476 es.I_O().setFilePointer( i_start_pos );
2477
2478 delete ep;
2479 ep = new EbmlParser( &es, segment, &sys.demuxer );
2480
2481 return true;
2482 }
2483
2484 void demux_sys_t::StartUiThread()
2485 {
2486 if ( !b_ui_hooked )
2487 {
2488 msg_Dbg( &demuxer, "Starting the UI Hook" );
2489 b_ui_hooked = true;
2490 /* FIXME hack hack hack hack FIXME */
2491 /* Get p_input and create variable */
2492 p_input = (input_thread_t *) vlc_object_find( &demuxer, VLC_OBJECT_INPUT, FIND_PARENT );
2493 var_Create( p_input, "x-start", VLC_VAR_INTEGER );
2494 var_Create( p_input, "y-start", VLC_VAR_INTEGER );
2495 var_Create( p_input, "x-end", VLC_VAR_INTEGER );
2496 var_Create( p_input, "y-end", VLC_VAR_INTEGER );
2497 var_Create( p_input, "color", VLC_VAR_ADDRESS );
2498 var_Create( p_input, "menu-palette", VLC_VAR_ADDRESS );
2499 var_Create( p_input, "highlight", VLC_VAR_BOOL );
2500 var_Create( p_input, "highlight-mutex", VLC_VAR_MUTEX );
2501
2502 /* Now create our event thread catcher */
2503 p_ev = (event_thread_t *) vlc_object_create( &demuxer, sizeof( event_thread_t ) );
2504 p_ev->p_demux = &demuxer;
2505 p_ev->b_die = VLC_FALSE;
2506 vlc_mutex_init( p_ev, &p_ev->lock );
2507 vlc_thread_create( p_ev, "mkv event thread handler", EventThread,
2508 VLC_THREAD_PRIORITY_LOW, VLC_FALSE );
2509 }
2510 }
2511
2512 void demux_sys_t::StopUiThread()
2513 {
2514 if ( b_ui_hooked )
2515 {
2516 p_ev->b_die = VLC_TRUE;
2517
2518 vlc_thread_join( p_ev );
2519 vlc_object_destroy( p_ev );
2520
2521 p_ev = NULL;
2522
2523 var_Destroy( p_input, "highlight-mutex" );
2524 var_Destroy( p_input, "highlight" );
2525 var_Destroy( p_input, "x-start" );
2526 var_Destroy( p_input, "x-end" );
2527 var_Destroy( p_input, "y-start" );
2528 var_Destroy( p_input, "y-end" );
2529 var_Destroy( p_input, "color" );
2530 var_Destroy( p_input, "menu-palette" );
2531
2532 vlc_object_release( p_input );
2533
2534 msg_Dbg( &demuxer, "Stopping the UI Hook" );
2535 }
2536 b_ui_hooked = false;
2537 }
2538
2539 int demux_sys_t::EventMouse( vlc_object_t *p_this, char const *psz_var,
2540 vlc_value_t oldval, vlc_value_t newval, void *p_data )
2541 {
2542 event_thread_t *p_ev = (event_thread_t *) p_data;
2543 vlc_mutex_lock( &p_ev->lock );
2544 if( psz_var[6] == 'c' )
2545 {
2546 p_ev->b_clicked = VLC_TRUE;
2547 msg_Dbg( p_this, "Event Mouse: clicked");
2548 }
2549 else if( psz_var[6] == 'm' )
2550 p_ev->b_moved = VLC_TRUE;
2551 vlc_mutex_unlock( &p_ev->lock );
2552
2553 return VLC_SUCCESS;
2554 }
2555
2556 int demux_sys_t::EventKey( vlc_object_t *p_this, char const *psz_var,
2557 vlc_value_t oldval, vlc_value_t newval, void *p_data )
2558 {
2559 event_thread_t *p_ev = (event_thread_t *) p_data;
2560 vlc_mutex_lock( &p_ev->lock );
2561 p_ev->b_key = VLC_TRUE;
2562 vlc_mutex_unlock( &p_ev->lock );
2563 msg_Dbg( p_this, "Event Key");
2564
2565 return VLC_SUCCESS;
2566 }
2567
2568 int demux_sys_t::EventThread( vlc_object_t *p_this )
2569 {
2570 event_thread_t *p_ev = (event_thread_t*)p_this;
2571 demux_sys_t *p_sys = p_ev->p_demux->p_sys;
2572 vlc_object_t *p_vout = NULL;
2573
2574 p_ev->b_moved = VLC_FALSE;
2575 p_ev->b_clicked = VLC_FALSE;
2576 p_ev->b_key = VLC_FALSE;
2577
2578 /* catch all key event */
2579 var_AddCallback( p_ev->p_vlc, "key-pressed", EventKey, p_ev );
2580
2581 /* main loop */
2582 while( !p_ev->b_die )
2583 {
2584 if ( !p_sys->b_pci_packet_set )
2585 {
2586 /* Wait 100ms */
2587 msleep( 100000 );
2588 continue;
2589 }
2590
2591 vlc_bool_t b_activated = VLC_FALSE;
2592
2593 /* KEY part */
2594 if( p_ev->b_key )
2595 {
2596 vlc_value_t valk;
2597 struct vlc_t::hotkey *p_hotkeys = p_ev->p_vlc->p_hotkeys;
2598 int i, i_action = -1;
2599
2600 msg_Dbg( p_ev->p_demux, "Handle Key Event");
2601
2602 vlc_mutex_lock( &p_ev->lock );
2603
2604 pci_t *pci = (pci_t *) &p_sys->pci_packet;
2605
2606 var_Get( p_ev->p_vlc, "key-pressed", &valk );
2607 for( i = 0; p_hotkeys[i].psz_action != NULL; i++ )
2608 {
2609 if( p_hotkeys[i].i_key == valk.i_int )
2610 {
2611 i_action = p_hotkeys[i].i_action;
2612 }
2613 }
2614
2615 uint16 i_curr_button = p_sys->dvd_interpretor.GetSPRM( 0x88 );
2616
2617 switch( i_action )
2618 {
2619 case ACTIONID_NAV_LEFT:
2620 if ( i_curr_button > 0 && i_curr_button <= pci->hli.hl_gi.btn_ns )
2621 {
2622 btni_t *p_button_ptr = &(pci->hli.btnit[i_curr_button-1]);
2623 if ( p_button_ptr->left > 0 && p_button_ptr->left <= pci->hli.hl_gi.btn_ns )
2624 {
2625 i_curr_button = p_button_ptr->left;
2626 p_sys->dvd_interpretor.SetSPRM( 0x88, i_curr_button );
2627 btni_t button_ptr = pci->hli.btnit[i_curr_button-1];
2628 if ( button_ptr.auto_action_mode )
2629 {
2630 vlc_mutex_unlock( &p_ev->lock );
2631 vlc_mutex_lock( &p_sys->lock_demuxer );
2632
2633 // process the button action
2634 p_sys->dvd_interpretor.Interpret( button_ptr.cmd.bytes, 8 );
2635
2636 vlc_mutex_unlock( &p_sys->lock_demuxer );
2637 vlc_mutex_lock( &p_ev->lock );
2638 }
2639 }
2640 }
2641 break;
2642 case ACTIONID_NAV_RIGHT:
2643 if ( i_curr_button > 0 && i_curr_button <= pci->hli.hl_gi.btn_ns )
2644 {
2645 btni_t *p_button_ptr = &(pci->hli.btnit[i_curr_button-1]);
2646 if ( p_button_ptr->right > 0 && p_button_ptr->right <= pci->hli.hl_gi.btn_ns )
2647 {
2648 i_curr_button = p_button_ptr->right;
2649 p_sys->dvd_interpretor.SetSPRM( 0x88, i_curr_button );
2650 btni_t button_ptr = pci->hli.btnit[i_curr_button-1];
2651 if ( button_ptr.auto_action_mode )
2652 {
2653 vlc_mutex_unlock( &p_ev->lock );
2654 vlc_mutex_lock( &p_sys->lock_demuxer );
2655
2656 // process the button action
2657 p_sys->dvd_interpretor.Interpret( button_ptr.cmd.bytes, 8 );
2658
2659 vlc_mutex_unlock( &p_sys->lock_demuxer );
2660 vlc_mutex_lock( &p_ev->lock );
2661 }
2662 }
2663 }
2664 break;
2665 case ACTIONID_NAV_UP:
2666 if ( i_curr_button > 0 && i_curr_button <= pci->hli.hl_gi.btn_ns )
2667 {
2668 btni_t *p_button_ptr = &(pci->hli.btnit[i_curr_button-1]);
2669 if ( p_button_ptr->up > 0 && p_button_ptr->up <= pci->hli.hl_gi.btn_ns )
2670 {
2671 i_curr_button = p_button_ptr->up;
2672 p_sys->dvd_interpretor.SetSPRM( 0x88, i_curr_button );
2673 btni_t button_ptr = pci->hli.btnit[i_curr_button-1];
2674 if ( button_ptr.auto_action_mode )
2675 {
2676 vlc_mutex_unlock( &p_ev->lock );
2677 vlc_mutex_lock( &p_sys->lock_demuxer );
2678
2679 // process the button action
2680 p_sys->dvd_interpretor.Interpret( button_ptr.cmd.bytes, 8 );
2681
2682 vlc_mutex_unlock( &p_sys->lock_demuxer );
2683 vlc_mutex_lock( &p_ev->lock );
2684 }
2685 }
2686 }
2687 break;
2688 case ACTIONID_NAV_DOWN:
2689 if ( i_curr_button > 0 && i_curr_button <= pci->hli.hl_gi.btn_ns )
2690 {
2691 btni_t *p_button_ptr = &(pci->hli.btnit[i_curr_button-1]);
2692 if ( p_button_ptr->down > 0 && p_button_ptr->down <= pci->hli.hl_gi.btn_ns )
2693 {
2694 i_curr_button = p_button_ptr->down;
2695 p_sys->dvd_interpretor.SetSPRM( 0x88, i_curr_button );
2696 btni_t button_ptr = pci->hli.btnit[i_curr_button-1];
2697 if ( button_ptr.auto_action_mode )
2698 {
2699 vlc_mutex_unlock( &p_ev->lock );
2700 vlc_mutex_lock( &p_sys->lock_demuxer );
2701
2702 // process the button action
2703 p_sys->dvd_interpretor.Interpret( button_ptr.cmd.bytes, 8 );
2704
2705 vlc_mutex_unlock( &p_sys->lock_demuxer );
2706 vlc_mutex_lock( &p_ev->lock );
2707 }
2708 }
2709 }
2710 break;
2711 case ACTIONID_NAV_ACTIVATE:
2712 b_activated = VLC_TRUE;
2713
2714 if ( i_curr_button > 0 && i_curr_button <= pci->hli.hl_gi.btn_ns )
2715 {
2716 btni_t button_ptr = pci->hli.btnit[i_curr_button-1];
2717
2718 vlc_mutex_unlock( &p_ev->lock );
2719 vlc_mutex_lock( &p_sys->lock_demuxer );
2720
2721 // process the button action
2722 p_sys->dvd_interpretor.Interpret( button_ptr.cmd.bytes, 8 );
2723
2724 vlc_mutex_unlock( &p_sys->lock_demuxer );
2725 vlc_mutex_lock( &p_ev->lock );
2726 }
2727 break;
2728 default:
2729 break;
2730 }
2731 p_ev->b_key = VLC_FALSE;
2732 vlc_mutex_unlock( &p_ev->lock );
2733 }
2734
2735 /* MOUSE part */
2736 if( p_vout && ( p_ev->b_moved || p_ev->b_clicked ) )
2737 {
2738 vlc_value_t valx, valy;
2739
2740 vlc_mutex_lock( &p_ev->lock );
2741 pci_t *pci = (pci_t *) &p_sys->pci_packet;
2742 var_Get( p_vout, "mouse-x", &valx );
2743 var_Get( p_vout, "mouse-y", &valy );
2744
2745 if( p_ev->b_clicked )
2746 {
2747 int32_t button;
2748 int32_t best,dist,d;
2749 int32_t mx,my,dx,dy;
2750
2751 msg_Dbg( p_ev->p_demux, "Handle Mouse Event: Mouse clicked x(%d)*y(%d)", (unsigned)valx.i_int, (unsigned)valy.i_int);
2752
2753 b_activated = VLC_TRUE;
2754 // get current button
2755 best = 0;
2756 dist = 0x08000000; /* >> than (720*720)+(567*567); */
2757 for(button = 1; button <= pci->hli.hl_gi.btn_ns; button++)
2758 {
2759 btni_t *button_ptr = &(pci->hli.btnit[button-1]);
2760
2761 if(((unsigned)valx.i_int >= button_ptr->x_start)
2762 && ((unsigned)valx.i_int <= button_ptr->x_end)
2763 && ((unsigned)valy.i_int >= button_ptr->y_start)
2764 && ((unsigned)valy.i_int <= button_ptr->y_end))
2765 {
2766 mx = (button_ptr->x_start + button_ptr->x_end)/2;
2767 my = (button_ptr->y_start + button_ptr->y_end)/2;
2768 dx = mx - valx.i_int;
2769 dy = my - valy.i_int;
2770 d = (dx*dx) + (dy*dy);
2771 /* If the mouse is within the button and the mouse is closer
2772 * to the center of this button then it is the best choice. */
2773 if(d < dist) {
2774 dist = d;
2775 best = button;
2776 }
2777 }
2778 }
2779
2780 if ( best != 0)
2781 {
2782 btni_t button_ptr = pci->hli.btnit[best-1];
2783 uint16 i_curr_button = p_sys->dvd_interpretor.GetSPRM( 0x88 );
2784
2785 msg_Dbg( &p_sys->demuxer, "Clicked button %d", best );
2786 vlc_mutex_unlock( &p_ev->lock );
2787 vlc_mutex_lock( &p_sys->lock_demuxer );
2788
2789 // process the button action
2790 p_sys->dvd_interpretor.SetSPRM( 0x88, best );
2791 p_sys->dvd_interpretor.Interpret( button_ptr.cmd.bytes, 8 );
2792
2793 msg_Dbg( &p_sys->demuxer, "Processed button %d", best );
2794
2795 // select new button
2796 if ( best != i_curr_button )
2797 {
2798 vlc_value_t val;
2799
2800 if( var_Get( p_sys->p_input, "highlight-mutex", &val ) == VLC_SUCCESS )
2801 {
2802 vlc_mutex_t *p_mutex = (vlc_mutex_t *) val.p_address;
2803 uint32_t i_palette;
2804
2805 if(button_ptr.btn_coln != 0) {
2806 i_palette = pci->hli.btn_colit.btn_coli[button_ptr.btn_coln-1][1];
2807 } else {
2808 i_palette = 0;
2809 }
2810
2811 for( int i = 0; i < 4; i++ )
2812 {
2813 uint32_t i_yuv = 0xFF;//p_sys->clut[(hl.palette>>(16+i*4))&0x0f];
2814 uint8_t i_alpha = (i_palette>>(i*4))&0x0f;
2815 i_alpha = i_alpha == 0xf ? 0xff : i_alpha << 4;
2816
2817 p_sys->palette[i][0] = (i_yuv >> 16) & 0xff;
2818 p_sys->palette[i][1] = (i_yuv >> 0) & 0xff;
2819 p_sys->palette[i][2] = (i_yuv >> 8) & 0xff;
2820 p_sys->palette[i][3] = i_alpha;
2821 }
2822
2823 vlc_mutex_lock( p_mutex );
2824 val.i_int = button_ptr.x_start; var_Set( p_sys->p_input, "x-start", val );
2825 val.i_int = button_ptr.x_end; var_Set( p_sys->p_input, "x-end", val );
2826 val.i_int = button_ptr.y_start; var_Set( p_sys->p_input, "y-start", val );
2827 val.i_int = button_ptr.y_end; var_Set( p_sys->p_input, "y-end", val );
2828
2829 val.p_address = (void *)p_sys->palette;
2830 var_Set( p_sys->p_input, "menu-palette", val );
2831
2832 val.b_bool = VLC_TRUE; var_Set( p_sys->p_input, "highlight", val );
2833 vlc_mutex_unlock( p_mutex );
2834 }
2835 }
2836 vlc_mutex_unlock( &p_sys->lock_demuxer );
2837 vlc_mutex_lock( &p_ev->lock );
2838 }
2839 }
2840 else if( p_ev->b_moved )
2841 {
2842 // dvdnav_mouse_select( NULL, pci, valx.i_int, valy.i_int );
2843 }
2844
2845 p_ev->b_moved = VLC_FALSE;
2846 p_ev->b_clicked = VLC_FALSE;
2847 vlc_mutex_unlock( &p_ev->lock );
2848 }
2849
2850 /* VOUT part */
2851 if( p_vout && p_vout->b_die )
2852 {
2853 var_DelCallback( p_vout, "mouse-moved", EventMouse, p_ev );
2854 var_DelCallback( p_vout, "mouse-clicked", EventMouse, p_ev );
2855 vlc_object_release( p_vout );
2856 p_vout = NULL;
2857 }
2858
2859 else if( p_vout == NULL )
2860 {
2861 p_vout = (vlc_object_t*) vlc_object_find( p_sys->p_input, VLC_OBJECT_VOUT,
2862 FIND_CHILD );
2863 if( p_vout)
2864 {
2865 var_AddCallback( p_vout, "mouse-moved", EventMouse, p_ev );
2866 var_AddCallback( p_vout, "mouse-clicked", EventMouse, p_ev );
2867 }
2868 }
2869
2870 /* Wait a bit, 10ms */
2871 msleep( 10000 );
2872 }
2873
2874 /* Release callback */
2875 if( p_vout )
2876 {
2877 var_DelCallback( p_vout, "mouse-moved", EventMouse, p_ev );
2878 var_DelCallback( p_vout, "mouse-clicked", EventMouse, p_ev );
2879 vlc_object_release( p_vout );
2880 }
2881 var_DelCallback( p_ev->p_vlc, "key-pressed", EventKey, p_ev );
2882
2883 vlc_mutex_destroy( &p_ev->lock );
2884
2885 return VLC_SUCCESS;
2886 }
2887
2888 void matroska_segment_c::UnSelect( )
2889 {
2890 size_t i_track;
2891
2892 for( i_track = 0; i_track < tracks.size(); i_track++ )
2893 {
2894 if ( tracks[i_track]->p_es != NULL )
2895 {
2896 // es_format_Clean( &tracks[i_track]->fmt );
2897 es_out_Del( sys.demuxer.out, tracks[i_track]->p_es );
2898 tracks[i_track]->p_es = NULL;
2899 }
2900 }
2901 delete ep;
2902 ep = NULL;
2903 }
2904
2905 void virtual_segment_c::PrepareChapters( )
2906 {
2907 if ( linked_segments.size() == 0 )
2908 return;
2909
2910 // !!! should be called only once !!!
2911 matroska_segment_c *p_segment;
2912 size_t i, j;
2913
2914 // copy editions from the first segment
2915 p_segment = linked_segments[0];
2916 p_editions = &p_segment->stored_editions;
2917
2918 for ( i=1 ; i<linked_segments.size(); i++ )
2919 {
2920 p_segment = linked_segments[i];
2921 // FIXME assume we have the same editions in all segments
2922 for (j=0; j<p_segment->stored_editions.size(); j++)
2923 (*p_editions)[j]->Append( *p_segment->stored_editions[j] );
2924 }
2925 }
2926
2927 std::string chapter_edition_c::GetMainName() const
2928 {
2929 if ( sub_chapters.size() )
2930 {
2931 return sub_chapters[0]->GetCodecName( true );
2932 }
2933 return "";
2934 }
2935
2936 int chapter_item_c::PublishChapters( input_title_t & title, int & i_user_chapters, int i_level )
2937 {
2938 // add support for meta-elements from codec like DVD Titles
2939 if ( !b_display_seekpoint || psz_name == "" )
2940 {
2941 psz_name = GetCodecName();
2942 if ( psz_name != "" )
2943 b_display_seekpoint = true;
2944 }
2945
2946 if (b_display_seekpoint)
2947 {
2948 seekpoint_t *sk = vlc_seekpoint_New();
2949
2950 sk->i_level = i_level;
2951 sk->i_time_offset = i_start_time;
2952 sk->psz_name = strdup( psz_name.c_str() );
2953
2954 // A start time of '0' is ok. A missing ChapterTime element is ok, too, because '0' is its default value.
2955 title.i_seekpoint++;
2956 title.seekpoint = (seekpoint_t**)realloc( title.seekpoint, title.i_seekpoint * sizeof( seekpoint_t* ) );
2957 title.seekpoint[title.i_seekpoint-1] = sk;
2958
2959 if ( b_user_display )
2960 i_user_chapters++;
2961 }
2962
2963 for ( size_t i=0; i<sub_chapters.size() ; i++)
2964 {
2965 sub_chapters[i]->PublishChapters( title, i_user_chapters, i_level+1 );
2966 }
2967
2968 i_seekpoint_num = i_user_chapters;
2969
2970 return i_user_chapters;
2971 }
2972
2973 bool virtual_segment_c::UpdateCurrentToChapter( demux_t & demux )
2974 {
2975 demux_sys_t & sys = *demux.p_sys;
2976 chapter_item_c *psz_curr_chapter;
2977 bool b_has_seeked = false;
2978
2979 /* update current chapter/seekpoint */
2980 if ( p_editions->size() )
2981 {
2982 /* 1st, we need to know in which chapter we are */
2983 psz_curr_chapter = (*p_editions)[i_current_edition]->FindTimecode( sys.i_pts, psz_current_chapter );
2984
2985 /* we have moved to a new chapter */
2986 if (psz_curr_chapter != NULL && psz_current_chapter != psz_curr_chapter)
2987 {
2988 if ( (*p_editions)[i_current_edition]->b_ordered )
2989 {
2990 // Leave/Enter up to the link point
2991 b_has_seeked = psz_curr_chapter->EnterAndLeave( psz_current_chapter );
2992 if ( !b_has_seeked )
2993 {
2994 // only physically seek if necessary
2995 if ( psz_current_chapter == NULL || (psz_current_chapter->i_end_time != psz_curr_chapter->i_start_time) )
2996 Seek( demux, sys.i_pts, 0, psz_curr_chapter );
2997 }
2998 }
2999
3000 if ( !b_has_seeked )
3001 {
3002 psz_current_chapter = psz_curr_chapter;
3003 if ( psz_curr_chapter->i_seekpoint_num > 0 )
3004 {
3005 demux.info.i_update |= INPUT_UPDATE_TITLE | INPUT_UPDATE_SEEKPOINT;
3006 demux.info.i_title = sys.i_current_title = i_sys_title;
3007 demux.info.i_seekpoint = psz_curr_chapter->i_seekpoint_num - 1;
3008 }
3009 }
3010
3011 return true;
3012 }
3013 else if (psz_curr_chapter == NULL)
3014 {
3015 // out of the scope of the data described by chapters, leave the edition
3016 if ( (*p_editions)[i_current_edition]->b_ordered && psz_current_chapter != NULL )
3017 {
3018 if ( !(*p_editions)[i_current_edition]->EnterAndLeave( psz_current_chapter, false ) )
3019 psz_current_chapter = NULL;
3020 else
3021 return true;
3022 }
3023 }
3024 }
3025 return false;
3026 }
3027
3028 chapter_item_c *virtual_segment_c::BrowseCodecPrivate( unsigned int codec_id,
3029 bool (*match)(const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size ),
3030 const void *p_cookie,
3031 size_t i_cookie_size )
3032 {
3033 // FIXME don't assume it is the first edition
3034 std::vector<chapter_edition_c*>::iterator index = p_editions->begin();
3035 if ( index != p_editions->end() )
3036 {
3037 chapter_item_c *p_result = (*index)->BrowseCodecPrivate( codec_id, match, p_cookie, i_cookie_size );
3038 if ( p_result != NULL )
3039 return p_result;
3040 }
3041 return NULL;
3042 }
3043
3044 chapter_item_c *virtual_segment_c::FindChapter( int64_t i_find_uid )
3045 {
3046 // FIXME don't assume it is the first edition
3047 std::vector<chapter_edition_c*>::iterator index = p_editions->begin();
3048 if ( index != p_editions->end() )
3049 {
3050 chapter_item_c *p_result = (*index)->FindChapter( i_find_uid );
3051 if ( p_result != NULL )
3052 return p_result;
3053 }
3054 return NULL;
3055 }
3056
3057 chapter_item_c *chapter_item_c::BrowseCodecPrivate( unsigned int codec_id,
3058 bool (*match)(const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size ),
3059 const void *p_cookie,
3060 size_t i_cookie_size )
3061 {
3062 // this chapter
3063 std::vector<chapter_codec_cmds_c*>::const_iterator index = codecs.begin();
3064 while ( index != codecs.end() )
3065 {
3066 if ( match( **index ,p_cookie, i_cookie_size ) )
3067 return this;
3068 index++;
3069 }
3070
3071 // sub-chapters
3072 chapter_item_c *p_result = NULL;
3073 std::vector<chapter_item_c*>::const_iterator index2 = sub_chapters.begin();
3074 while ( index2 != sub_chapters.end() )
3075 {
3076 p_result = (*index2)->BrowseCodecPrivate( codec_id, match, p_cookie, i_cookie_size );
3077 if ( p_result != NULL )
3078 return p_result;
3079 index2++;
3080 }
3081
3082 return p_result;
3083 }
3084
3085 void chapter_item_c::Append( const chapter_item_c & chapter )
3086 {
3087 // we are appending content for the same chapter UID
3088 size_t i;
3089 chapter_item_c *p_chapter;
3090
3091 for ( i=0; i<chapter.sub_chapters.size(); i++ )
3092 {
3093 p_chapter = FindChapter( chapter.sub_chapters[i]->i_uid );
3094 if ( p_chapter != NULL )
3095 {
3096 p_chapter->Append( *chapter.sub_chapters[i] );
3097 }
3098 else
3099 {
3100 sub_chapters.push_back( chapter.sub_chapters[i] );
3101 }
3102 }
3103
3104 i_user_start_time = min( i_user_start_time, chapter.i_user_start_time );
3105 i_user_end_time = max( i_user_end_time, chapter.i_user_end_time );
3106 }
3107
3108 chapter_item_c * chapter_item_c::FindChapter( int64_t i_find_uid )
3109 {
3110 size_t i;
3111 chapter_item_c *p_result = NULL;
3112
3113 if ( i_uid == i_find_uid )
3114 return this;
3115
3116 for ( i=0; i<sub_chapters.size(); i++)
3117 {
3118 p_result = sub_chapters[i]->FindChapter( i_find_uid );
3119 if ( p_result != NULL )
3120 break;
3121 }
3122 return p_result;
3123 }
3124
3125 std::string chapter_item_c::GetCodecName( bool f_for_title ) const
3126 {
3127 std::string result;
3128
3129 std::vector<chapter_codec_cmds_c*>::const_iterator index = codecs.begin();
3130 while ( index != codecs.end() )
3131 {
3132 result = (*index)->GetCodecName( f_for_title );
3133 if ( result != "" )
3134 break;
3135 index++;
3136 }
3137
3138 return result;
3139 }
3140
3141 std::string dvd_chapter_codec_c::GetCodecName( bool f_for_title ) const
3142 {
3143 std::string result;
3144 if ( p_private_data->GetSize() >= 3)
3145 {
3146 const binary* p_data = p_private_data->GetBuffer();
3147 /* if ( p_data[0] == MATROSKA_DVD_LEVEL_TT )
3148 {
3149 uint16_t i_title = (p_data[1] << 8) + p_data[2];
3150 char psz_str[11];
3151 sprintf( psz_str, " %d ---", i_title );
3152 result = N_("--- DVD Title");
3153 result += psz_str;
3154 }
3155 else */ if ( p_data[0] == MATROSKA_DVD_LEVEL_LU )
3156 {
3157 char psz_str[11];
3158 sprintf( psz_str, " (%c%c) ---", p_data[1], p_data[2] );
3159 result = N_("--- DVD Menu");
3160 result += psz_str;
3161 }
3162 else if ( p_data[0] == MATROSKA_DVD_LEVEL_SS && f_for_title )
3163 {
3164 if ( p_data[1] == 0x00 )
3165 result = N_("First Played");
3166 else if ( p_data[1] == 0xC0 )
3167 result = N_("Video Manager");
3168 else if ( p_data[1] == 0x80 )
3169 {
3170 uint16_t i_title = (p_data[2] << 8) + p_data[3];
3171 char psz_str[20];
3172 sprintf( psz_str, " %d -----", i_title );
3173 result = N_("----- Title");
3174 result += psz_str;
3175 }
3176 }
3177 }
3178
3179 return result;
3180 }
3181
3182 int16 chapter_item_c::GetTitleNumber( ) const
3183 {
3184 int result = -1;
3185
3186 std::vector<chapter_codec_cmds_c*>::const_iterator index = codecs.begin();
3187 while ( index != codecs.end() )
3188 {
3189 result = (*index)->GetTitleNumber( );
3190 if ( result >= 0 )
3191 break;
3192 index++;
3193 }
3194
3195 return result;
3196 }
3197
3198 int16 dvd_chapter_codec_c::GetTitleNumber()
3199 {
3200 if ( p_private_data->GetSize() >= 3)
3201 {
3202 const binary* p_data = p_private_data->GetBuffer();
3203 if ( p_data[0] == MATROSKA_DVD_LEVEL_SS )
3204 {
3205 return int16( (p_data[2] << 8) + p_data[3] );
3206 }
3207 }
3208 return -1;
3209 }
3210
3211 static void Seek( demux_t *p_demux, mtime_t i_date, double f_percent, chapter_item_c *psz_chapter )
3212 {
3213 demux_sys_t *p_sys = p_demux->p_sys;
3214 virtual_segment_c *p_vsegment = p_sys->p_current_segment;
3215 matroska_segment_c *p_segment = p_vsegment->Segment();
3216 mtime_t i_time_offset = 0;
3217
3218 int i_index;
3219
3220 msg_Dbg( p_demux, "seek request to "I64Fd" (%f%%)", i_date, f_percent );
3221 if( i_date < 0 && f_percent < 0 )
3222 {
3223 msg_Warn( p_demux, "cannot seek nowhere !" );
3224 return;
3225 }
3226 if( f_percent > 1.0 )
3227 {
3228 msg_Warn( p_demux, "cannot seek so far !" );
3229 return;
3230 }
3231
3232 /* seek without index or without date */
3233 if( f_percent >= 0 && (config_GetInt( p_demux, "mkv-seek-percent" ) || !p_segment->b_cues || i_date < 0 ))
3234 {
3235 if (p_sys->f_duration >= 0)
3236 {
3237 i_date = int64_t( f_percent * p_sys->f_duration * 1000.0 );
3238 }
3239 else
3240 {
3241 int64_t i_pos = int64_t( f_percent * stream_Size( p_demux->s ) );
3242
3243 msg_Dbg( p_demux, "inacurate way of seeking" );
3244 for( i_index = 0; i_index < p_segment->i_index; i_index++ )
3245 {
3246 if( p_segment->p_indexes[i_index].i_position >= i_pos)
3247 {
3248 break;
3249 }
3250 }
3251 if( i_index == p_segment->i_index )
3252 {
3253 i_index--;
3254 }
3255
3256 i_date = p_segment->p_indexes[i_index].i_time;
3257
3258 #if 0
3259 if( p_segment->p_indexes[i_index].i_position < i_pos )
3260 {
3261 EbmlElement *el;
3262
3263 msg_Warn( p_demux, "searching for cluster, could take some time" );
3264
3265 /* search a cluster */
3266 while( ( el = p_sys->ep->Get() ) != NULL )
3267 {
3268 if( MKV_IS_ID( el, KaxCluster ) )
3269 {
3270 KaxCluster *cluster = (KaxCluster*)el;
3271
3272 /* add it to the index */
3273 p_segment->IndexAppendCluster( cluster );
3274
3275 if( (int64_t)cluster->GetElementPosition() >= i_pos )
3276 {
3277 p_sys->cluster = cluster;
3278 p_sys->ep->Down();
3279 break;
3280 }
3281 }
3282 }
3283 }
3284 #endif
3285 }
3286 }
3287
3288 p_vsegment->Seek( *p_demux, i_date, i_time_offset, psz_chapter );
3289 }
3290
3291 /*****************************************************************************
3292 * Demux: reads and demuxes data packets
3293 *****************************************************************************
3294 * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
3295 *****************************************************************************/
3296 static int Demux( demux_t *p_demux)
3297 {
3298 demux_sys_t *p_sys = p_demux->p_sys;
3299
3300 vlc_mutex_lock( &p_sys->lock_demuxer );
3301
3302 virtual_segment_c *p_vsegment = p_sys->p_current_segment;
3303 matroska_segment_c *p_segment = p_vsegment->Segment();
3304 if ( p_segment == NULL ) return 0;
3305 int i_block_count = 0;
3306 int i_return = 0;
3307
3308 for( ;; )
3309 {
3310 if ( p_sys->demuxer.b_die )
3311 break;
3312
3313 if( p_sys->i_pts >= p_sys->i_start_pts )
3314 if ( p_vsegment->UpdateCurrentToChapter( *p_demux ) )
3315 {
3316 i_return = 1;
3317 break;
3318 }
3319
3320 if ( p_vsegment->Edition() && p_vsegment->Edition()->b_ordered && p_vsegment->CurrentChapter() == NULL )
3321 {
3322 /* nothing left to read in this ordered edition */
3323 if ( !p_vsegment->SelectNext() )
3324 break;
3325 p_segment->UnSelect( );
3326
3327 es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
3328
3329 /* switch to the next segment */
3330 p_segment = p_vsegment->Segment();
3331 if ( !p_segment->Select( 0 ) )
3332 {
3333 msg_Err( p_demux, "Failed to select new segment" );
3334 break;
3335 }
3336 continue;
3337 }
3338
3339 KaxBlock *block;
3340 int64_t i_block_duration = 0;
3341 int64_t i_block_ref1;
3342 int64_t i_block_ref2;
3343
3344 #if LIBMATROSKA_VERSION >= 0x000800
3345 KaxSimpleBlock *simpleblock;
3346
3347 if( p_segment->BlockGet( block, simpleblock, &i_block_ref1, &i_block_ref2, &i_block_duration ) )
3348 #else
3349 if( p_segment->BlockGet( block, &i_block_ref1, &i_block_ref2, &i_block_duration ) )
3350 #endif
3351 {
3352 if ( p_vsegment->Edition() && p_vsegment->Edition()->b_ordered )
3353 {
3354 const chapter_item_c *p_chap = p_vsegment->CurrentChapter();
3355 // check if there are more chapters to read
3356 if ( p_chap != NULL )
3357 {
3358 /* TODO handle successive chapters with the same user_start_time/user_end_time
3359 if ( p_chap->i_user_start_time == p_chap->i_user_start_time )
3360 p_vsegment->SelectNext();
3361 */
3362 p_sys->i_pts = p_chap->i_user_end_time;
3363 p_sys->i_pts++; // trick to avoid staying on segments with no duration and no content
3364
3365 i_return = 1;
3366 }
3367
3368 break;
3369 }
3370 else
3371 {
3372 msg_Warn( p_demux, "cannot get block EOF?" );
3373 p_segment->UnSelect( );
3374
3375 es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
3376
3377 /* switch to the next segment */
3378 if ( !p_vsegment->SelectNext() )
3379 // no more segments in this stream
3380 break;
3381 p_segment = p_vsegment->Segment();
3382 if ( !p_segment->Select( 0 ) )
3383 {
3384 msg_Err( p_demux, "Failed to select new segment" );
3385 break;
3386 }
3387
3388 continue;
3389 }
3390 }
3391
3392 #if LIBMATROSKA_VERSION >= 0x000800
3393 if ( simpleblock != NULL )
3394 p_sys->i_pts = (p_sys->i_chapter_time + simpleblock->GlobalTimecode()) / (mtime_t) 1000;
3395 else
3396 #endif
3397 p_sys->i_pts = (p_sys->i_chapter_time + block->GlobalTimecode()) / (mtime_t) 1000;
3398
3399 if( p_sys->i_pts >= p_sys->i_start_pts )
3400 {
3401 es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_pts );
3402
3403 if ( p_vsegment->UpdateCurrentToChapter( *p_demux ) )
3404 {
3405 i_return = 1;
3406 delete block;
3407 break;
3408 }
3409 }
3410
3411 if ( p_vsegment->Edition() && p_vsegment->Edition()->b_ordered && p_vsegment->CurrentChapter() == NULL )
3412 {
3413 /* nothing left to read in this ordered edition */
3414 if ( !p_vsegment->SelectNext() )
3415 {
3416 delete block;
3417 break;
3418 }
3419 p_segment->UnSelect( );
3420
3421 es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
3422
3423 /* switch to the next segment */
3424 p_segment = p_vsegment->Segment();
3425 if ( !p_segment->Select( 0 ) )
3426 {
3427 msg_Err( p_demux, "Failed to select new segment" );
3428 delete block;
3429 break;
3430 }
3431 delete block;
3432 continue;
3433 }
3434
3435 #if LIBMATROSKA_VERSION >= 0x000800
3436 BlockDecode( p_demux, block, simpleblock, p_sys->i_pts, i_block_duration, i_block_ref1 >= 0 || i_block_ref2 > 0 );
3437 #else
3438 BlockDecode( p_demux, block, p_sys->i_pts, i_block_duration, i_block_ref1 >= 0 || i_block_ref2 > 0 );
3439 #endif
3440
3441 delete block;
3442 i_block_count++;
3443
3444 // TODO optimize when there is need to leave or when seeking has been called
3445 if( i_block_count > 5 )
3446 {
3447 i_return = 1;
3448 break;
3449 }
3450 }
3451
3452 vlc_mutex_unlock( &p_sys->lock_demuxer );
3453
3454 return i_return;
3455 }
3456
3457
3458
3459 /*****************************************************************************
3460 * Stream managment
3461 *****************************************************************************/
3462 vlc_stream_io_callback::vlc_stream_io_callback( stream_t *s_, vlc_bool_t b_owner_ )
3463 {
3464 s = s_;
3465 b_owner = b_owner_;
3466 mb_eof = VLC_FALSE;
3467 }
3468
3469 uint32 vlc_stream_io_callback::read( void *p_buffer, size_t i_size )
3470 {
3471 if( i_size <= 0 || mb_eof )
3472 {
3473 return 0;
3474 }
3475
3476 return stream_Read( s, p_buffer, i_size );
3477 }
3478 void vlc_stream_io_callback::setFilePointer(int64_t i_offset, seek_mode mode )
3479 {
3480 int64_t i_pos;
3481
3482 switch( mode )
3483 {
3484 case seek_beginning:
3485 i_pos = i_offset;
3486 break;
3487 case seek_end:
3488 i_pos = stream_Size( s ) - i_offset;
3489 break;
3490 default:
3491 i_pos= stream_Tell( s ) + i_offset;
3492 break;
3493 }
3494
3495 if( i_pos < 0 || i_pos >= stream_Size( s ) )
3496 {
3497 mb_eof = VLC_TRUE;
3498 return;
3499 }
3500
3501 mb_eof = VLC_FALSE;
3502 if( stream_Seek( s, i_pos ) )
3503 {
3504 mb_eof = VLC_TRUE;
3505 }
3506 return;
3507 }
3508 size_t vlc_stream_io_callback::write( const void *p_buffer, size_t i_size )
3509 {
3510 return 0;
3511 }
3512 uint64 vlc_stream_io_callback::getFilePointer( void )
3513 {
3514 if ( s == NULL )
3515 return 0;
3516 return stream_Tell( s );
3517 }
3518 void vlc_stream_io_callback::close( void )
3519 {
3520 return;
3521 }
3522
3523
3524 /*****************************************************************************
3525 * Ebml Stream parser
3526 *****************************************************************************/
3527 EbmlParser::EbmlParser( EbmlStream *es, EbmlElement *el_start, demux_t *p_demux )
3528 {
3529 int i;
3530
3531 m_es = es;
3532 m_got = NULL;
3533 m_el[0] = el_start;
3534 mi_remain_size[0] = el_start->GetSize();
3535
3536 for( i = 1; i < 6; i++ )
3537 {
3538 m_el[i] = NULL;
3539 }
3540 mi_level = 1;
3541 mi_user_level = 1;
3542 mb_keep = VLC_FALSE;
3543 mb_dummy = config_GetInt( p_demux, "mkv-use-dummy" );
3544 }
3545
3546 EbmlParser::~EbmlParser( void )
3547 {
3548 int i;
3549
3550 for( i = 1; i < mi_level; i++ )
3551 {
3552 if( !mb_keep )
3553 {
3554 delete m_el[i];
3555 }
3556 mb_keep = VLC_FALSE;
3557 }
3558 }
3559
3560 EbmlElement* EbmlParser::UnGet( uint64 i_block_pos, uint64 i_cluster_pos )
3561 {
3562 if ( mi_user_level > mi_level )
3563 {
3564 while ( mi_user_level != mi_level )
3565 {
3566 delete m_el[mi_user_level];
3567 m_el[mi_user_level] = NULL;
3568 mi_user_level--;
3569 }
3570 }
3571 m_got = NULL;
3572 mb_keep = VLC_FALSE;
3573 if ( m_el[1]->GetElementPosition() == i_cluster_pos )
3574 {
3575 m_es->I_O().setFilePointer( i_block_pos, seek_beginning );
3576 return (EbmlMaster*) m_el[1];
3577 }
3578 else
3579 {
3580 // seek to the previous Cluster
3581 m_es->I_O().setFilePointer( i_cluster_pos, seek_beginning );
3582 mi_level--;
3583 mi_user_level--;
3584 delete m_el[mi_level];
3585 m_el[mi_level] = NULL;
3586 return NULL;
3587 }
3588 }
3589
3590 void EbmlParser::Up( void )
3591 {
3592 if( mi_user_level == mi_level )
3593 {
3594 fprintf( stderr," arrrrrrrrrrrrrg Up cannot escape itself\n" );
3595 }
3596
3597 mi_user_level--;
3598 }
3599
3600 void EbmlParser::Down( void )
3601 {
3602 mi_user_level++;
3603 mi_level++;
3604 }
3605
3606 void EbmlParser::Keep( void )
3607 {
3608 mb_keep = VLC_TRUE;
3609 }
3610
3611 int EbmlParser::GetLevel( void )
3612 {
3613 return mi_user_level;
3614 }
3615
3616 void EbmlParser::Reset( demux_t *p_demux )
3617 {
3618 while ( mi_level > 0)
3619 {
3620 delete m_el[mi_level];
3621 m_el[mi_level] = NULL;
3622 mi_level--;
3623 }
3624 mi_user_level = mi_level = 1;
3625 #if LIBEBML_VERSION >= 0x000704
3626 // a little faster and cleaner
3627 m_es->I_O().setFilePointer( static_cast<KaxSegment*>(m_el[0])->GetGlobalPosition(0) );
3628 #else
3629 m_es->I_O().setFilePointer( m_el[0]->GetElementPosition() + m_el[0]->ElementSize(true) - m_el[0]->GetSize() );
3630 #endif
3631 mb_dummy = config_GetInt( p_demux, "mkv-use-dummy" );
3632 }
3633
3634 EbmlElement *EbmlParser::Get( void )
3635 {
3636 int i_ulev = 0;
3637
3638 if( mi_user_level != mi_level )
3639 {
3640 return NULL;
3641 }
3642 if( m_got )
3643 {
3644 EbmlElement *ret = m_got;
3645 m_got = NULL;
3646
3647 return ret;
3648 }
3649
3650 if( m_el[mi_level] )
3651 {
3652 m_el[mi_level]->SkipData( *m_es, m_el[mi_level]->Generic().Context );
3653 if( !mb_keep )
3654 {
3655 delete m_el[mi_level];
3656 }
3657 mb_keep = VLC_FALSE;
3658 }
3659
3660 m_el[mi_level] = m_es->FindNextElement( m_el[mi_level - 1]->Generic().Context, i_ulev, 0xFFFFFFFFL, mb_dummy != 0, 1 );
3661 // mi_remain_size[mi_level] = m_el[mi_level]->GetSize();
3662 if( i_ulev > 0 )
3663 {
3664 while( i_ulev > 0 )
3665 {
3666 if( mi_level == 1 )
3667 {
3668 mi_level = 0;
3669 return NULL;
3670 }
3671
3672 delete m_el[mi_level - 1];
3673 m_got = m_el[mi_level -1] = m_el[mi_level];
3674 m_el[mi_level] = NULL;
3675
3676 mi_level--;
3677 i_ulev--;
3678 }
3679 return NULL;
3680 }
3681 else if( m_el[mi_level] == NULL )
3682 {
3683 fprintf( stderr," m_el[mi_level] == NULL\n" );
3684 }
3685
3686 return m_el[mi_level];
3687 }
3688
3689
3690 /*****************************************************************************
3691 * Tools
3692 * * LoadCues : load the cues element and update index
3693 *
3694 * * LoadTags : load ... the tags element
3695 *
3696 * * InformationCreate : create all information, load tags if present
3697 *
3698 *****************************************************************************/
3699 void matroska_segment_c::LoadCues( )
3700 {
3701 int64_t i_sav_position = es.I_O().getFilePointer();
3702 EbmlParser *ep;
3703 EbmlElement *el, *cues;
3704
3705 /* *** Load the cue if found *** */
3706 if( i_cues_position < 0 )
3707 {
3708 msg_Warn( &sys.demuxer, "no cues/empty cues found->seek won't be precise" );
3709
3710 // IndexAppendCluster( cluster );
3711 }
3712
3713 vlc_bool_t b_seekable;
3714
3715 stream_Control( sys.demuxer.s, STREAM_CAN_FASTSEEK, &b_seekable );
3716 if( !b_seekable )
3717 return;
3718
3719 msg_Dbg( &sys.demuxer, "loading cues" );
3720 es.I_O().setFilePointer( i_cues_position, seek_beginning );
3721 cues = es.FindNextID( KaxCues::ClassInfos, 0xFFFFFFFFL);
3722
3723 if( cues == NULL )
3724 {
3725 msg_Err( &sys.demuxer, "cannot load cues (broken seekhead or file)" );
3726 es.I_O().setFilePointer( i_sav_position, seek_beginning );
3727 return;
3728 }
3729
3730 ep = new EbmlParser( &es, cues, &sys.demuxer );
3731 while( ( el = ep->Get() ) != NULL )
3732 {
3733 if( MKV_IS_ID( el, KaxCuePoint ) )
3734 {
3735 #define idx p_indexes[i_index]
3736
3737 idx.i_track = -1;
3738 idx.i_block_number= -1;
3739 idx.i_position = -1;
3740 idx.i_time = 0;
3741 idx.b_key = VLC_TRUE;
3742
3743 ep->Down();
3744 while( ( el = ep->Get() ) != NULL )
3745 {
3746 if( MKV_IS_ID( el, KaxCueTime ) )
3747 {
3748 KaxCueTime &ctime = *(KaxCueTime*)el;
3749
3750 ctime.ReadData( es.I_O() );
3751
3752 idx.i_time = uint64( ctime ) * i_timescale / (mtime_t)1000;
3753 }
3754 else if( MKV_IS_ID( el, KaxCueTrackPositions ) )
3755 {
3756 ep->Down();
3757 while( ( el = ep->Get() ) != NULL )
3758 {
3759 if( MKV_IS_ID( el, KaxCueTrack ) )
3760 {
3761 KaxCueTrack &ctrack = *(KaxCueTrack*)el;
3762
3763 ctrack.ReadData( es.I_O() );
3764 idx.i_track = uint16( ctrack );
3765 }
3766 else if( MKV_IS_ID( el, KaxCueClusterPosition ) )
3767 {
3768 KaxCueClusterPosition &ccpos = *(KaxCueClusterPosition*)el;
3769
3770 ccpos.ReadData( es.I_O() );
3771 idx.i_position = segment->GetGlobalPosition( uint64( ccpos ) );
3772 }
3773 else if( MKV_IS_ID( el, KaxCueBlockNumber ) )
3774 {
3775 KaxCueBlockNumber &cbnum = *(KaxCueBlockNumber*)el;
3776
3777 cbnum.ReadData( es.I_O() );
3778 idx.i_block_number = uint32( cbnum );
3779 }
3780 else
3781 {
3782 msg_Dbg( &sys.demuxer, " * Unknown (%s)", typeid(*el).name() );
3783 }
3784 }
3785 ep->Up();
3786 }
3787 else
3788 {
3789 msg_Dbg( &sys.demuxer, " * Unknown (%s)", typeid(*el).name() );
3790 }
3791 }
3792 ep->Up();
3793
3794 #if 0
3795 msg_Dbg( &sys.demuxer, " * added time="I64Fd" pos="I64Fd
3796 " track=%d bnum=%d", idx.i_time, idx.i_position,
3797 idx.i_track, idx.i_block_number );
3798 #endif
3799
3800 i_index++;
3801 if( i_index >= i_index_max )
3802 {
3803 i_index_max += 1024;
3804 p_indexes = (mkv_index_t*)realloc( p_indexes, sizeof( mkv_index_t ) * i_index_max );
3805 }
3806 #undef idx
3807 }
3808 else
3809 {
3810 msg_Dbg( &sys.demuxer, " * Unknown (%s)", typeid(*el).name() );
3811 }
3812 }
3813 delete ep;
3814 delete cues;
3815
3816 b_cues = VLC_TRUE;
3817
3818 msg_Dbg( &sys.demuxer, "loading cues done." );
3819 es.I_O().setFilePointer( i_sav_position, seek_beginning );
3820 }
3821
3822 void matroska_segment_c::LoadTags( )
3823 {
3824 int64_t i_sav_position = es.I_O().getFilePointer();
3825 EbmlParser *ep;
3826 EbmlElement *el, *tags;
3827
3828 msg_Dbg( &sys.demuxer, "loading tags" );
3829 es.I_O().setFilePointer( i_tags_position, seek_beginning );
3830 tags = es.FindNextID( KaxTags::ClassInfos, 0xFFFFFFFFL);
3831
3832 if( tags == NULL )
3833 {
3834 msg_Err( &sys.demuxer, "cannot load tags (broken seekhead or file)" );
3835 es.I_O().setFilePointer( i_sav_position, seek_beginning );
3836 return;
3837 }
3838
3839 msg_Dbg( &sys.demuxer, "Tags" );
3840 ep = new EbmlParser( &es, tags, &sys.demuxer );
3841 while( ( el = ep->Get() ) != NULL )
3842 {
3843 if( MKV_IS_ID( el, KaxTag ) )
3844 {
3845 msg_Dbg( &sys.demuxer, "+ Tag" );
3846 ep->Down();
3847 while( ( el = ep->Get() ) != NULL )
3848 {
3849 if( MKV_IS_ID( el, KaxTagTargets ) )
3850 {
3851 msg_Dbg( &sys.demuxer, "| + Targets" );
3852 ep->Down();
3853 while( ( el = ep->Get() ) != NULL )
3854 {
3855 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid( *el ).name() );
3856 }
3857 ep->Up();
3858 }
3859 else if( MKV_IS_ID( el, KaxTagGeneral ) )
3860 {
3861 msg_Dbg( &sys.demuxer, "| + General" );
3862 ep->Down();
3863 while( ( el = ep->Get() ) != NULL )
3864 {
3865 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid( *el ).name() );
3866 }
3867 ep->Up();
3868 }
3869 else if( MKV_IS_ID( el, KaxTagGenres ) )
3870 {
3871 msg_Dbg( &sys.demuxer, "| + Genres" );
3872 ep->Down();
3873 while( ( el = ep->Get() ) != NULL )
3874 {
3875 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid( *el ).name() );
3876 }
3877 ep->Up();
3878 }
3879 else if( MKV_IS_ID( el, KaxTagAudioSpecific ) )
3880 {
3881 msg_Dbg( &sys.demuxer, "| + Audio Specific" );
3882 ep->Down();
3883 while( ( el = ep->Get() ) != NULL )
3884 {
3885 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid( *el ).name() );
3886 }
3887 ep->Up();
3888 }
3889 else if( MKV_IS_ID( el, KaxTagImageSpecific ) )
3890 {
3891 msg_Dbg( &sys.demuxer, "| + Images Specific" );
3892 ep->Down();
3893 while( ( el = ep->Get() ) != NULL )
3894 {
3895 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid( *el ).name() );
3896 }
3897 ep->Up();
3898 }
3899 else if( MKV_IS_ID( el, KaxTagMultiComment ) )
3900 {
3901 msg_Dbg( &sys.demuxer, "| + Multi Comment" );
3902 }
3903 else if( MKV_IS_ID( el, KaxTagMultiCommercial ) )
3904 {
3905 msg_Dbg( &sys.demuxer, "| + Multi Commercial" );
3906 }
3907 else if( MKV_IS_ID( el, KaxTagMultiDate ) )
3908 {
3909 msg_Dbg( &sys.demuxer, "| + Multi Date" );
3910 }
3911 else if( MKV_IS_ID( el, KaxTagMultiEntity ) )
3912 {
3913 msg_Dbg( &sys.demuxer, "| + Multi Entity" );
3914 }
3915 else if( MKV_IS_ID( el, KaxTagMultiIdentifier ) )
3916 {
3917 msg_Dbg( &sys.demuxer, "| + Multi Identifier" );
3918 }
3919 else if( MKV_IS_ID( el, KaxTagMultiLegal ) )
3920 {
3921 msg_Dbg( &sys.demuxer, "| + Multi Legal" );
3922 }
3923 else if( MKV_IS_ID( el, KaxTagMultiTitle ) )
3924 {
3925 msg_Dbg( &sys.demuxer, "| + Multi Title" );
3926 }
3927 else
3928 {
3929 msg_Dbg( &sys.demuxer, "| + Unknown (%s)", typeid( *el ).name() );
3930 }
3931 }
3932 ep->Up();
3933 }
3934 else
3935 {
3936 msg_Dbg( &sys.demuxer, "+ Unknown (%s)", typeid( *el ).name() );
3937 }
3938 }
3939 delete ep;
3940 delete tags;
3941
3942 msg_Dbg( &sys.demuxer, "loading tags done." );
3943 es.I_O().setFilePointer( i_sav_position, seek_beginning );
3944 }
3945
3946 /*****************************************************************************
3947 * ParseSeekHead:
3948 *****************************************************************************/
3949 void matroska_segment_c::ParseSeekHead( KaxSeekHead *seekhead )
3950 {
3951 EbmlElement *el;
3952 size_t i, j;
3953 int i_upper_level = 0;
3954
3955 msg_Dbg( &sys.demuxer, "| + Seek head" );
3956
3957 /* Master elements */
3958 seekhead->Read( es, seekhead->Generic().Context, i_upper_level, el, true );
3959
3960 for( i = 0; i < seekhead->ListSize(); i++ )
3961 {
3962 EbmlElement *l = (*seekhead)[i];
3963
3964 if( MKV_IS_ID( l, KaxSeek ) )
3965 {
3966 EbmlMaster *sk = static_cast<EbmlMaster *>(l);
3967 EbmlId id = EbmlVoid::ClassInfos.GlobalId;
3968 int64_t i_pos = -1;
3969
3970 for( j = 0; j < sk->ListSize(); j++ )
3971 {
3972 EbmlElement *l = (*sk)[j];
3973
3974 if( MKV_IS_ID( l, KaxSeekID ) )
3975 {
3976 KaxSeekID &sid = *(KaxSeekID*)l;
3977 id = EbmlId( sid.GetBuffer(), sid.GetSize() );
3978 }
3979 else if( MKV_IS_ID( l, KaxSeekPosition ) )
3980 {
3981 KaxSeekPosition &spos = *(KaxSeekPosition*)l;
3982 i_pos = uint64( spos );
3983 }
3984 else
3985 {
3986 msg_Dbg( &sys.demuxer, "| | | + Unknown (%s)", typeid(*l).name() );
3987 }
3988 }
3989
3990 if( i_pos >= 0 )
3991 {
3992 if( id == KaxCues::ClassInfos.GlobalId )
3993 {
3994 msg_Dbg( &sys.demuxer, "| | | = cues at "I64Fd, i_pos );
3995 i_cues_position = segment->GetGlobalPosition( i_pos );
3996 }
3997 else if( id == KaxChapters::ClassInfos.GlobalId )
3998 {
3999 msg_Dbg( &sys.demuxer, "| | | = chapters at "I64Fd, i_pos );
4000 i_chapters_position = segment->GetGlobalPosition( i_pos );
4001 }
4002 else if( id == KaxTags::ClassInfos.GlobalId )
4003 {
4004 msg_Dbg( &sys.demuxer, "| | | = tags at "I64Fd, i_pos );
4005 i_tags_position = segment->GetGlobalPosition( i_pos );
4006 }
4007 }
4008 }
4009 else
4010 {
4011 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid(*l).name() );
4012 }
4013 }
4014 }
4015
4016 /*****************************************************************************
4017 * ParseTrackEntry:
4018 *****************************************************************************/
4019 void matroska_segment_c::ParseTrackEntry( KaxTrackEntry *m )
4020 {
4021 size_t i, j, k, n;
4022 bool bSupported = true;
4023
4024 mkv_track_t *tk;
4025
4026 msg_Dbg( &sys.demuxer, "| | + Track Entry" );
4027
4028 tk = new mkv_track_t();
4029
4030 /* Init the track */
4031 memset( tk, 0, sizeof( mkv_track_t ) );
4032
4033 es_format_Init( &tk->fmt, UNKNOWN_ES, 0 );
4034 tk->fmt.psz_language = strdup("English");
4035 tk->fmt.psz_description = NULL;
4036
4037 tk->b_default = VLC_TRUE;
4038 tk->b_enabled = VLC_TRUE;
4039 tk->b_silent = VLC_FALSE;
4040 tk->i_number = tracks.size() - 1;
4041 tk->i_extra_data = 0;
4042 tk->p_extra_data = NULL;
4043 tk->psz_codec = NULL;
4044 tk->i_default_duration = 0;
4045 tk->f_timecodescale = 1.0;
4046
4047 tk->b_inited = VLC_FALSE;
4048 tk->i_data_init = 0;
4049 tk->p_data_init = NULL;
4050
4051 tk->psz_codec_name = NULL;
4052 tk->psz_codec_settings = NULL;
4053 tk->psz_codec_info_url = NULL;
4054 tk->psz_codec_download_url = NULL;
4055
4056 tk->i_compression_type = MATROSKA_COMPRESSION_NONE;
4057 tk->p_compression_data = NULL;
4058
4059 for( i = 0; i < m->ListSize(); i++ )
4060 {
4061 EbmlElement *l = (*m)[i];
4062
4063 if( MKV_IS_ID( l, KaxTrackNumber ) )
4064 {
4065 KaxTrackNumber &tnum = *(KaxTrackNumber*)l;
4066
4067 tk->i_number = uint32( tnum );
4068 msg_Dbg( &sys.demuxer, "| | | + Track Number=%u", uint32( tnum ) );
4069 }
4070 else if( MKV_IS_ID( l, KaxTrackUID ) )
4071 {
4072 KaxTrackUID &tuid = *(KaxTrackUID*)l;
4073
4074 msg_Dbg( &sys.demuxer, "| | | + Track UID=%u", uint32( tuid ) );
4075 }
4076 else if( MKV_IS_ID( l, KaxTrackType ) )
4077 {
4078 char *psz_type;
4079 KaxTrackType &ttype = *(KaxTrackType*)l;
4080
4081 switch( uint8(ttype) )
4082 {
4083 case track_audio:
4084 psz_type = "audio";
4085 tk->fmt.i_cat = AUDIO_ES;
4086 break;
4087 case track_video:
4088 psz_type = "video";
4089 tk->fmt.i_cat = VIDEO_ES;
4090 break;
4091 case track_subtitle:
4092 psz_type = "subtitle";
4093 tk->fmt.i_cat = SPU_ES;
4094 break;
4095 case track_buttons:
4096 psz_type = "buttons";
4097 tk->fmt.i_cat = SPU_ES;
4098 break;
4099 default:
4100 psz_type = "unknown";
4101 tk->fmt.i_cat = UNKNOWN_ES;
4102 break;
4103 }
4104
4105 msg_Dbg( &sys.demuxer, "| | | + Track Type=%s", psz_type );
4106 }
4107 // else if( EbmlId( *l ) == KaxTrackFlagEnabled::ClassInfos.GlobalId )
4108 // {
4109 // KaxTrackFlagEnabled &fenb = *(KaxTrackFlagEnabled*)l;
4110
4111 // tk->b_enabled = uint32( fenb );
4112 // msg_Dbg( &sys.demuxer, "| | | + Track Enabled=%u",
4113 // uint32( fenb ) );
4114 // }
4115 else if( MKV_IS_ID( l, KaxTrackFlagDefault ) )
4116 {
4117 KaxTrackFlagDefault &fdef = *(KaxTrackFlagDefault*)l;
4118
4119 tk->b_default = uint32( fdef );
4120 msg_Dbg( &sys.demuxer, "| | | + Track Default=%u", uint32( fdef ) );
4121 }
4122 else if( MKV_IS_ID( l, KaxTrackFlagLacing ) )
4123 {
4124 KaxTrackFlagLacing &lac = *(KaxTrackFlagLacing*)l;
4125
4126 msg_Dbg( &sys.demuxer, "| | | + Track Lacing=%d", uint32( lac ) );
4127 }
4128 else if( MKV_IS_ID( l, KaxTrackMinCache ) )
4129 {
4130 KaxTrackMinCache &cmin = *(KaxTrackMinCache*)l;
4131
4132 msg_Dbg( &sys.demuxer, "| | | + Track MinCache=%d", uint32( cmin ) );
4133 }
4134 else if( MKV_IS_ID( l, KaxTrackMaxCache ) )
4135 {
4136 KaxTrackMaxCache &cmax = *(KaxTrackMaxCache*)l;
4137
4138 msg_Dbg( &sys.demuxer, "| | | + Track MaxCache=%d", uint32( cmax ) );
4139 }
4140 else if( MKV_IS_ID( l, KaxTrackDefaultDuration ) )
4141 {
4142 KaxTrackDefaultDuration &defd = *(KaxTrackDefaultDuration*)l;
4143
4144 tk->i_default_duration = uint64(defd);
4145 msg_Dbg( &sys.demuxer, "| | | + Track Default Duration="I64Fd, uint64(defd) );
4146 }
4147 else if( MKV_IS_ID( l, KaxTrackTimecodeScale ) )
4148 {
4149 KaxTrackTimecodeScale &ttcs = *(KaxTrackTimecodeScale*)l;
4150
4151 tk->f_timecodescale = float( ttcs );
4152 msg_Dbg( &sys.demuxer, "| | | + Track TimeCodeScale=%f", tk->f_timecodescale );
4153 }
4154 else if( MKV_IS_ID( l, KaxTrackName ) )
4155 {
4156 KaxTrackName &tname = *(KaxTrackName*)l;
4157
4158 tk->fmt.psz_description = ToUTF8( UTFstring( tname ) );
4159 msg_Dbg( &sys.demuxer, "| | | + Track Name=%s", tk->fmt.psz_description );
4160 }
4161 else if( MKV_IS_ID( l, KaxTrackLanguage ) )
4162 {
4163 KaxTrackLanguage &lang = *(KaxTrackLanguage*)l;
4164
4165 if ( tk->fmt.psz_language != NULL )
4166 free( tk->fmt.psz_language );
4167 tk->fmt.psz_language = strdup( string( lang ).c_str() );
4168 msg_Dbg( &sys.demuxer,
4169 "| | | + Track Language=`%s'", tk->fmt.psz_language );
4170 }
4171 else if( MKV_IS_ID( l, KaxCodecID ) )
4172 {
4173 KaxCodecID &codecid = *(KaxCodecID*)l;
4174
4175 tk->psz_codec = strdup( string( codecid ).c_str() );
4176 msg_Dbg( &sys.demuxer, "| | | + Track CodecId=%s", string( codecid ).c_str() );
4177 }
4178 else if( MKV_IS_ID( l, KaxCodecPrivate ) )
4179 {
4180 KaxCodecPrivate &cpriv = *(KaxCodecPrivate*)l;
4181
4182 tk->i_extra_data = cpriv.GetSize();
4183 if( tk->i_extra_data > 0 )
4184 {
4185 tk->p_extra_data = (uint8_t*)malloc( tk->i_extra_data );
4186 memcpy( tk->p_extra_data, cpriv.GetBuffer(), tk->i_extra_data );
4187 }
4188 msg_Dbg( &sys.demuxer, "| | | + Track CodecPrivate size="I64Fd, cpriv.GetSize() );
4189 }
4190 else if( MKV_IS_ID( l, KaxCodecName ) )
4191 {
4192 KaxCodecName &cname = *(KaxCodecName*)l;
4193
4194 tk->psz_codec_name = ToUTF8( UTFstring( cname ) );
4195 msg_Dbg( &sys.demuxer, "| | | + Track Codec Name=%s", tk->psz_codec_name );
4196 }
4197 else if( MKV_IS_ID( l, KaxContentEncodings ) )
4198 {
4199 EbmlMaster *cencs = static_cast<EbmlMaster*>(l);
4200 MkvTree( sys.demuxer, 3, "Content Encodings" );
4201 if ( cencs->ListSize() > 1 )
4202 {
4203 msg_Err( &sys.demuxer, "Multiple Compression method not supported" );
4204 bSupported = false;
4205 }
4206 for( j = 0; j < cencs->ListSize(); j++ )
4207 {
4208 EbmlElement *l2 = (*cencs)[j];
4209 if( MKV_IS_ID( l2, KaxContentEncoding ) )
4210 {
4211 MkvTree( sys.demuxer, 4, "Content Encoding" );
4212 EbmlMaster *cenc = static_cast<EbmlMaster*>(l2);
4213 for( k = 0; k < cenc->ListSize(); k++ )
4214 {
4215 EbmlElement *l3 = (*cenc)[k];
4216 if( MKV_IS_ID( l3, KaxContentEncodingOrder ) )
4217 {
4218 KaxContentEncodingOrder &encord = *(KaxContentEncodingOrder*)l3;
4219 MkvTree( sys.demuxer, 5, "Order: %i", uint32( encord ) );
4220 }
4221 else if( MKV_IS_ID( l3, KaxContentEncodingScope ) )
4222 {
4223 KaxContentEncodingScope &encscope = *(KaxContentEncodingScope*)l3;
4224 MkvTree( sys.demuxer, 5, "Scope: %i", uint32( encscope ) );
4225 }
4226 else if( MKV_IS_ID( l3, KaxContentEncodingType ) )
4227 {
4228 KaxContentEncodingType &enctype = *(KaxContentEncodingType*)l3;
4229 MkvTree( sys.demuxer, 5, "Type: %i", uint32( enctype ) );
4230 }
4231 else if( MKV_IS_ID( l3, KaxContentCompression ) )
4232 {
4233 EbmlMaster *compr = static_cast<EbmlMaster*>(l3);
4234 MkvTree( sys.demuxer, 5, "Content Compression" );
4235 for( n = 0; n < compr->ListSize(); n++ )
4236 {
4237 EbmlElement *l4 = (*compr)[n];
4238 if( MKV_IS_ID( l4, KaxContentCompAlgo ) )
4239 {
4240 KaxContentCompAlgo &compalg = *(KaxContentCompAlgo*)l4;
4241 MkvTree( sys.demuxer, 6, "Compression Algorithm: %i", uint32(compalg) );
4242 tk->i_compression_type = uint32( compalg );
4243 if ( ( tk->i_compression_type != MATROSKA_COMPRESSION_ZLIB ) &&
4244 ( tk->i_compression_type != MATROSKA_COMPRESSION_HEADER ) )
4245 {
4246 msg_Err( &sys.demuxer, "Track Compression method %d not supported", tk->i_compression_type );
4247 bSupported = false;
4248 }
4249 }
4250 else if( MKV_IS_ID( l4, KaxContentCompSettings ) )
4251 {
4252 tk->p_compression_data = new KaxContentCompSettings( *(KaxContentCompSettings*)l4 );
4253 }
4254 else
4255 {
4256 MkvTree( sys.demuxer, 6, "Unknown (%s)", typeid(*l4).name() );
4257 }
4258 }
4259 }
4260
4261 else
4262 {
4263 MkvTree( sys.demuxer, 5, "Unknown (%s)", typeid(*l3).name() );
4264 }
4265 }
4266
4267 }
4268 else
4269 {
4270 MkvTree( sys.demuxer, 4, "Unknown (%s)", typeid(*l2).name() );
4271 }
4272 }
4273
4274 }
4275 // else if( EbmlId( *l ) == KaxCodecSettings::ClassInfos.GlobalId )
4276 // {
4277 // KaxCodecSettings &cset = *(KaxCodecSettings*)l;
4278
4279 // tk->psz_codec_settings = ToUTF8( UTFstring( cset ) );
4280 // msg_Dbg( &sys.demuxer, "| | | + Track Codec Settings=%s", tk->psz_codec_settings );
4281 // }
4282 // else if( EbmlId( *l ) == KaxCodecInfoURL::ClassInfos.GlobalId )
4283 // {
4284 // KaxCodecInfoURL &ciurl = *(KaxCodecInfoURL*)l;
4285
4286 // tk->psz_codec_info_url = strdup( string( ciurl ).c_str() );
4287 // msg_Dbg( &sys.demuxer, "| | | + Track Codec Info URL=%s", tk->psz_codec_info_url );
4288 // }
4289 // else if( EbmlId( *l ) == KaxCodecDownloadURL::ClassInfos.GlobalId )
4290 // {
4291 // KaxCodecDownloadURL &cdurl = *(KaxCodecDownloadURL*)l;
4292
4293 // tk->psz_codec_download_url = strdup( string( cdurl ).c_str() );
4294 // msg_Dbg( &sys.demuxer, "| | | + Track Codec Info URL=%s", tk->psz_codec_download_url );
4295 // }
4296 // else if( EbmlId( *l ) == KaxCodecDecodeAll::ClassInfos.GlobalId )
4297 // {
4298 // KaxCodecDecodeAll &cdall = *(KaxCodecDecodeAll*)l;
4299
4300 // msg_Dbg( &sys.demuxer, "| | | + Track Codec Decode All=%u <== UNUSED", uint8( cdall ) );
4301 // }
4302 // else if( EbmlId( *l ) == KaxTrackOverlay::ClassInfos.GlobalId )
4303 // {
4304 // KaxTrackOverlay &tovr = *(KaxTrackOverlay*)l;
4305
4306 // msg_Dbg( &sys.demuxer, "| | | + Track Overlay=%u <== UNUSED", uint32( tovr ) );
4307 // }
4308 else if( MKV_IS_ID( l, KaxTrackVideo ) )
4309 {
4310 EbmlMaster *tkv = static_cast<EbmlMaster*>(l);
4311 unsigned int j;
4312
4313 msg_Dbg( &sys.demuxer, "| | | + Track Video" );
4314 tk->f_fps = 0.0;
4315
4316 tk->fmt.video.i_frame_rate_base = (unsigned int)(tk->i_default_duration / 1000);
4317 tk->fmt.video.i_frame_rate = 1000000;
4318
4319 for( j = 0; j < tkv->ListSize(); j++ )
4320 {
4321 EbmlElement *l = (*tkv)[j];
4322 // if( EbmlId( *el4 ) == KaxVideoFlagInterlaced::ClassInfos.GlobalId )
4323 // {
4324 // KaxVideoFlagInterlaced &fint = *(KaxVideoFlagInterlaced*)el4;
4325
4326 // msg_Dbg( &sys.demuxer, "| | | | + Track Video Interlaced=%u", uint8( fint ) );
4327 // }
4328 // else if( EbmlId( *el4 ) == KaxVideoStereoMode::ClassInfos.GlobalId )
4329 // {
4330 // KaxVideoStereoMode &stereo = *(KaxVideoStereoMode*)el4;
4331
4332 // msg_Dbg( &sys.demuxer, "| | | | + Track Video Stereo Mode=%u", uint8( stereo ) );
4333 // }
4334 // else
4335 if( MKV_IS_ID( l, KaxVideoPixelWidth ) )
4336 {
4337 KaxVideoPixelWidth &vwidth = *(KaxVideoPixelWidth*)l;
4338
4339 tk->fmt.video.i_width += uint16( vwidth );
4340 msg_Dbg( &sys.demuxer, "| | | | + width=%d", uint16( vwidth ) );
4341 }
4342 else if( MKV_IS_ID( l, KaxVideoPixelHeight ) )
4343 {
4344 KaxVideoPixelWidth &vheight = *(KaxVideoPixelWidth*)l;
4345
4346 tk->fmt.video.i_height += uint16( vheight );
4347 msg_Dbg( &sys.demuxer, "| | | | + height=%d", uint16( vheight ) );
4348 }
4349 else if( MKV_IS_ID( l, KaxVideoDisplayWidth ) )
4350 {
4351 KaxVideoDisplayWidth &vwidth = *(KaxVideoDisplayWidth*)l;
4352
4353 tk->fmt.video.i_visible_width = uint16( vwidth );
4354 msg_Dbg( &sys.demuxer, "| | | | + display width=%d", uint16( vwidth ) );
4355 }
4356 else if( MKV_IS_ID( l, KaxVideoDisplayHeight ) )
4357 {
4358 KaxVideoDisplayWidth &vheight = *(KaxVideoDisplayWidth*)l;
4359
4360 tk->fmt.video.i_visible_height = uint16( vheight );
4361 msg_Dbg( &sys.demuxer, "| | | | + display height=%d", uint16( vheight ) );
4362 }
4363 else if( MKV_IS_ID( l, KaxVideoPixelCropBottom ) )
4364 {
4365 KaxVideoPixelCropBottom &cropval = *(KaxVideoPixelCropBottom*)l;
4366
4367 tk->fmt.video.i_height -= uint16( cropval );
4368 msg_Dbg( &sys.demuxer, "| | | | + crop pixel bottom=%d", uint16( cropval ) );
4369 }
4370 else if( MKV_IS_ID( l, KaxVideoPixelCropTop ) )
4371 {
4372 KaxVideoPixelCropTop &cropval = *(KaxVideoPixelCropTop*)l;
4373
4374 tk->fmt.video.i_height -= uint16( cropval );
4375 tk->fmt.video.i_y_offset += uint16( cropval );
4376 msg_Dbg( &sys.demuxer, "| | | | + crop pixel top=%d", uint16( cropval ) );
4377 }
4378 else if( MKV_IS_ID( l, KaxVideoPixelCropRight ) )
4379 {
4380 KaxVideoPixelCropRight &cropval = *(KaxVideoPixelCropRight*)l;
4381
4382 tk->fmt.video.i_width -= uint16( cropval );
4383 msg_Dbg( &sys.demuxer, "| | | | + crop pixel right=%d", uint16( cropval ) );
4384 }
4385 else if( MKV_IS_ID( l, KaxVideoPixelCropLeft ) )
4386 {
4387 KaxVideoPixelCropLeft &cropval = *(KaxVideoPixelCropLeft*)l;
4388
4389 tk->fmt.video.i_width -= uint16( cropval );
4390 tk->fmt.video.i_x_offset += uint16( cropval );
4391 msg_Dbg( &sys.demuxer, "| | | | + crop pixel left=%d", uint16( cropval ) );
4392 }
4393 else if( MKV_IS_ID( l, KaxVideoFrameRate ) )
4394 {
4395 KaxVideoFrameRate &vfps = *(KaxVideoFrameRate*)l;
4396
4397 tk->f_fps = float( vfps );
4398 msg_Dbg( &sys.demuxer, " | | | + fps=%f", float( vfps ) );
4399 }
4400 // else if( EbmlId( *l ) == KaxVideoDisplayUnit::ClassInfos.GlobalId )
4401 // {
4402 // KaxVideoDisplayUnit &vdmode = *(KaxVideoDisplayUnit*)l;
4403
4404 // msg_Dbg( &sys.demuxer, "| | | | + Track Video Display Unit=%s",
4405 // uint8( vdmode ) == 0 ? "pixels" : ( uint8( vdmode ) == 1 ? "centimeters": "inches" ) );
4406 // }
4407 // else if( EbmlId( *l ) == KaxVideoAspectRatio::ClassInfos.GlobalId )
4408 // {
4409 // KaxVideoAspectRatio &ratio = *(KaxVideoAspectRatio*)l;
4410
4411 // msg_Dbg( &sys.demuxer, " | | | + Track Video Aspect Ratio Type=%u", uint8( ratio ) );
4412 // }
4413 // else if( EbmlId( *l ) == KaxVideoGamma::ClassInfos.GlobalId )
4414 // {
4415 // KaxVideoGamma &gamma = *(KaxVideoGamma*)l;
4416
4417 // msg_Dbg( &sys.demuxer, " | | | + fps=%f", float( gamma ) );
4418 // }
4419 else
4420 {
4421 msg_Dbg( &sys.demuxer, "| | | | + Unknown (%s)", typeid(*l).name() );
4422 }
4423 }
4424 if ( tk->fmt.video.i_visible_height && tk->fmt.video.i_visible_width )
4425 tk->fmt.video.i_aspect = VOUT_ASPECT_FACTOR * tk->fmt.video.i_visible_width / tk->fmt.video.i_visible_height;
4426 }
4427 else if( MKV_IS_ID( l, KaxTrackAudio ) )
4428 {
4429 EbmlMaster *tka = static_cast<EbmlMaster*>(l);
4430 unsigned int j;
4431
4432 msg_Dbg( &sys.demuxer, "| | | + Track Audio" );
4433
4434 for( j = 0; j < tka->ListSize(); j++ )
4435 {
4436 EbmlElement *l = (*tka)[j];
4437
4438 if( MKV_IS_ID( l, KaxAudioSamplingFreq ) )
4439 {
4440 KaxAudioSamplingFreq &afreq = *(KaxAudioSamplingFreq*)l;
4441
4442 tk->i_original_rate = tk->fmt.audio.i_rate = (int)float( afreq );
4443 msg_Dbg( &sys.demuxer, "| | | | + afreq=%d", tk->fmt.audio.i_rate );
4444 }
4445 else if( MKV_IS_ID( l, KaxAudioOutputSamplingFreq ) )
4446 {
4447 KaxAudioOutputSamplingFreq &afreq = *(KaxAudioOutputSamplingFreq*)l;
4448
4449 tk->fmt.audio.i_rate = (int)float( afreq );
4450 msg_Dbg( &sys.demuxer, "| | | | + aoutfreq=%d", tk->fmt.audio.i_rate );
4451 }
4452 else if( MKV_IS_ID( l, KaxAudioChannels ) )
4453 {
4454 KaxAudioChannels &achan = *(KaxAudioChannels*)l;
4455
4456 tk->fmt.audio.i_channels = uint8( achan );
4457 msg_Dbg( &sys.demuxer, "| | | | + achan=%u", uint8( achan ) );
4458 }
4459 else if( MKV_IS_ID( l, KaxAudioBitDepth ) )
4460 {
4461 KaxAudioBitDepth &abits = *(KaxAudioBitDepth*)l;
4462
4463 tk->fmt.audio.i_bitspersample = uint8( abits );
4464 msg_Dbg( &sys.demuxer, "| | | | + abits=%u", uint8( abits ) );
4465 }
4466 else
4467 {
4468 msg_Dbg( &sys.demuxer, "| | | | + Unknown (%s)", typeid(*l).name() );
4469 }
4470 }
4471 }
4472 else
4473 {
4474 msg_Dbg( &sys.demuxer, "| | | + Unknown (%s)",
4475 typeid(*l).name() );
4476 }
4477 }
4478
4479 if ( bSupported )
4480 {
4481 tracks.push_back( tk );
4482 }
4483 else
4484 {
4485 msg_Err( &sys.demuxer, "Track Entry %d not supported", tk->i_number );
4486 delete tk;
4487 }
4488 }
4489
4490 /*****************************************************************************
4491 * ParseTracks:
4492 *****************************************************************************/
4493 void matroska_segment_c::ParseTracks( KaxTracks *tracks )
4494 {
4495 EbmlElement *el;
4496 unsigned int i;
4497 int i_upper_level = 0;
4498
4499 msg_Dbg( &sys.demuxer, "| + Tracks" );
4500
4501 /* Master elements */
4502 tracks->Read( es, tracks->Generic().Context, i_upper_level, el, true );
4503
4504 for( i = 0; i < tracks->ListSize(); i++ )
4505 {
4506 EbmlElement *l = (*tracks)[i];
4507
4508 if( MKV_IS_ID( l, KaxTrackEntry ) )
4509 {
4510 ParseTrackEntry( static_cast<KaxTrackEntry *>(l) );
4511 }
4512 else
4513 {
4514 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid(*l).name() );
4515 }
4516 }
4517 }
4518
4519 /*****************************************************************************
4520 * ParseInfo:
4521 *****************************************************************************/
4522 void matroska_segment_c::ParseInfo( KaxInfo *info )
4523 {
4524 EbmlElement *el;
4525 EbmlMaster *m;
4526 size_t i, j;
4527 int i_upper_level = 0;
4528
4529 msg_Dbg( &sys.demuxer, "| + Information" );
4530
4531 /* Master elements */
4532 m = static_cast<EbmlMaster *>(info);
4533 m->Read( es, info->Generic().Context, i_upper_level, el, true );
4534
4535 for( i = 0; i < m->ListSize(); i++ )
4536 {
4537 EbmlElement *l = (*m)[i];
4538
4539 if( MKV_IS_ID( l, KaxSegmentUID ) )
4540 {
4541 if ( p_segment_uid == NULL )
4542 p_segment_uid = new KaxSegmentUID(*static_cast<KaxSegmentUID*>(l));
4543
4544 msg_Dbg( &sys.demuxer, "| | + UID=%d", *(uint32*)p_segment_uid->GetBuffer() );
4545 }
4546 else if( MKV_IS_ID( l, KaxPrevUID ) )
4547 {
4548 if ( p_prev_segment_uid == NULL )
4549 p_prev_segment_uid = new KaxPrevUID(*static_cast<KaxPrevUID*>(l));
4550
4551 msg_Dbg( &sys.demuxer, "| | + PrevUID=%d", *(uint32*)p_prev_segment_uid->GetBuffer() );
4552 }
4553 else if( MKV_IS_ID( l, KaxNextUID ) )
4554 {
4555 if ( p_next_segment_uid == NULL )
4556 p_next_segment_uid = new KaxNextUID(*static_cast<KaxNextUID*>(l));
4557
4558 msg_Dbg( &sys.demuxer, "| | + NextUID=%d", *(uint32*)p_next_segment_uid->GetBuffer() );
4559 }
4560 else if( MKV_IS_ID( l, KaxTimecodeScale ) )
4561 {
4562 KaxTimecodeScale &tcs = *(KaxTimecodeScale*)l;
4563
4564 i_timescale = uint64(tcs);
4565
4566 msg_Dbg( &sys.demuxer, "| | + TimecodeScale="I64Fd,
4567 i_timescale );
4568 }
4569 else if( MKV_IS_ID( l, KaxDuration ) )
4570 {
4571 KaxDuration &dur = *(KaxDuration*)l;
4572
4573 i_duration = mtime_t( double( dur ) );
4574
4575 msg_Dbg( &sys.demuxer, "| | + Duration="I64Fd,
4576 i_duration );
4577 }
4578 else if( MKV_IS_ID( l, KaxMuxingApp ) )
4579 {
4580 KaxMuxingApp &mapp = *(KaxMuxingApp*)l;
4581
4582 psz_muxing_application = ToUTF8( UTFstring( mapp ) );
4583
4584 msg_Dbg( &sys.demuxer, "| | + Muxing Application=%s",
4585 psz_muxing_application );
4586 }
4587 else if( MKV_IS_ID( l, KaxWritingApp ) )
4588 {
4589 KaxWritingApp &wapp = *(KaxWritingApp*)l;
4590
4591 psz_writing_application = ToUTF8( UTFstring( wapp ) );
4592
4593 msg_Dbg( &sys.demuxer, "| | + Writing Application=%s",
4594 psz_writing_application );
4595 }
4596 else if( MKV_IS_ID( l, KaxSegmentFilename ) )
4597 {
4598 KaxSegmentFilename &sfn = *(KaxSegmentFilename*)l;
4599
4600 psz_segment_filename = ToUTF8( UTFstring( sfn ) );
4601
4602 msg_Dbg( &sys.demuxer, "| | + Segment Filename=%s",
4603 psz_segment_filename );
4604 }
4605 else if( MKV_IS_ID( l, KaxTitle ) )
4606 {
4607 KaxTitle &title = *(KaxTitle*)l;
4608
4609 psz_title = ToUTF8( UTFstring( title ) );
4610
4611 msg_Dbg( &sys.demuxer, "| | + Title=%s", psz_title );
4612 }
4613 else if( MKV_IS_ID( l, KaxSegmentFamily ) )
4614 {
4615 KaxSegmentFamily *uid = static_cast<KaxSegmentFamily*>(l);
4616
4617 families.push_back( new KaxSegmentFamily(*uid) );
4618
4619 msg_Dbg( &sys.demuxer, "| | + family=%d", *(uint32*)uid->GetBuffer() );
4620 }
4621 #if defined( HAVE_GMTIME_R ) && !defined( __APPLE__ )
4622 else if( MKV_IS_ID( l, KaxDateUTC ) )
4623 {
4624 KaxDateUTC &date = *(KaxDateUTC*)l;
4625 time_t i_date;
4626 struct tm tmres;
4627 char buffer[256];
4628
4629 i_date = date.GetEpochDate();
4630 memset( buffer, 0, 256 );
4631 if( gmtime_r( &i_date, &tmres ) &&
4632 asctime_r( &tmres, buffer ) )
4633 {
4634 buffer[strlen( buffer)-1]= '\0';
4635 psz_date_utc = strdup( buffer );
4636 msg_Dbg( &sys.demuxer, "| | + Date=%s", psz_date_utc );
4637 }
4638 }
4639 #endif
4640 #if LIBMATROSKA_VERSION >= 0x000704
4641 else if( MKV_IS_ID( l, KaxChapterTranslate ) )
4642 {
4643 KaxChapterTranslate *p_trans = static_cast<KaxChapterTranslate*>( l );
4644 chapter_translation_c *p_translate = new chapter_translation_c();
4645
4646 p_trans->Read( es, p_trans->Generic().Context, i_upper_level, el, true );
4647 for( j = 0; j < p_trans->ListSize(); j++ )
4648 {
4649 EbmlElement *l = (*p_trans)[j];
4650
4651 if( MKV_IS_ID( l, KaxChapterTranslateEditionUID ) )
4652 {
4653 p_translate->editions.push_back( uint64( *static_cast<KaxChapterTranslateEditionUID*>( l ) ) );
4654 }
4655 else if( MKV_IS_ID( l, KaxChapterTranslateCodec ) )
4656 {
4657 p_translate->codec_id = uint32( *static_cast<KaxChapterTranslateCodec*>( l ) );
4658 }
4659 else if( MKV_IS_ID( l, KaxChapterTranslateID ) )
4660 {
4661 p_translate->p_translated = new KaxChapterTranslateID( *static_cast<KaxChapterTranslateID*>( l ) );
4662 }
4663 }
4664
4665 translations.push_back( p_translate );
4666 }
4667 #endif
4668 else
4669 {
4670 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid(*l).name() );
4671 }
4672 }
4673
4674 double f_dur = double(i_duration) * double(i_timescale) / 1000000.0;
4675 i_duration = mtime_t(f_dur);
4676 }
4677
4678
4679 /*****************************************************************************
4680 * ParseChapterAtom
4681 *****************************************************************************/
4682 void matroska_segment_c::ParseChapterAtom( int i_level, KaxChapterAtom *ca, chapter_item_c & chapters )
4683 {
4684 size_t i, j;
4685
4686 msg_Dbg( &sys.demuxer, "| | | + ChapterAtom (level=%d)", i_level );
4687 for( i = 0; i < ca->ListSize(); i++ )
4688 {
4689 EbmlElement *l = (*ca)[i];
4690
4691 if( MKV_IS_ID( l, KaxChapterUID ) )
4692 {
4693 chapters.i_uid = uint64_t(*(KaxChapterUID*)l);
4694 msg_Dbg( &sys.demuxer, "| | | | + ChapterUID: %lld", chapters.i_uid );
4695 }
4696 else if( MKV_IS_ID( l, KaxChapterFlagHidden ) )
4697 {
4698 KaxChapterFlagHidden &flag =*(KaxChapterFlagHidden*)l;
4699 chapters.b_display_seekpoint = uint8( flag ) == 0;
4700
4701 msg_Dbg( &sys.demuxer, "| | | | + ChapterFlagHidden: %s", chapters.b_display_seekpoint ? "no":"yes" );
4702 }
4703 else if( MKV_IS_ID( l, KaxChapterTimeStart ) )
4704 {
4705 KaxChapterTimeStart &start =*(KaxChapterTimeStart*)l;
4706 chapters.i_start_time = uint64( start ) / I64C(1000);
4707
4708 msg_Dbg( &sys.demuxer, "| | | | + ChapterTimeStart: %lld", chapters.i_start_time );
4709 }
4710 else if( MKV_IS_ID( l, KaxChapterTimeEnd ) )
4711 {
4712 KaxChapterTimeEnd &end =*(KaxChapterTimeEnd*)l;
4713 chapters.i_end_time = uint64( end ) / I64C(1000);
4714
4715 msg_Dbg( &sys.demuxer, "| | | | + ChapterTimeEnd: %lld", chapters.i_end_time );
4716 }
4717 else if( MKV_IS_ID( l, KaxChapterDisplay ) )
4718 {
4719 EbmlMaster *cd = static_cast<EbmlMaster *>(l);
4720
4721 msg_Dbg( &sys.demuxer, "| | | | + ChapterDisplay" );
4722 for( j = 0; j < cd->ListSize(); j++ )
4723 {
4724 EbmlElement *l= (*cd)[j];
4725
4726 if( MKV_IS_ID( l, KaxChapterString ) )
4727 {
4728 int k;
4729
4730 KaxChapterString &name =*(KaxChapterString*)l;
4731 for (k = 0; k < i_level; k++)
4732 chapters.psz_name += '+';
4733 chapters.psz_name += ' ';
4734 char *psz_tmp_utf8 = ToUTF8( UTFstring( name ) );
4735 chapters.psz_name += psz_tmp_utf8;
4736 chapters.b_user_display = true;
4737
4738 msg_Dbg( &sys.demuxer, "| | | | | + ChapterString '%s'", psz_tmp_utf8 );
4739 free( psz_tmp_utf8 );
4740 }
4741 else if( MKV_IS_ID( l, KaxChapterLanguage ) )
4742 {
4743 KaxChapterLanguage &lang =*(KaxChapterLanguage*)l;
4744 const char *psz = string( lang ).c_str();
4745
4746 msg_Dbg( &sys.demuxer, "| | | | | + ChapterLanguage '%s'", psz );
4747 }
4748 else if( MKV_IS_ID( l, KaxChapterCountry ) )
4749 {
4750 KaxChapterCountry &ct =*(KaxChapterCountry*)l;
4751 const char *psz = string( ct ).c_str();
4752
4753 msg_Dbg( &sys.demuxer, "| | | | | + ChapterCountry '%s'", psz );
4754 }
4755 }
4756 }
4757 else if( MKV_IS_ID( l, KaxChapterProcess ) )
4758 {
4759 msg_Dbg( &sys.demuxer, "| | | | + ChapterProcess" );
4760
4761 KaxChapterProcess *cp = static_cast<KaxChapterProcess *>(l);
4762 chapter_codec_cmds_c *p_ccodec = NULL;
4763
4764 for( j = 0; j < cp->ListSize(); j++ )
4765 {
4766 EbmlElement *k= (*cp)[j];
4767
4768 if( MKV_IS_ID( k, KaxChapterProcessCodecID ) )
4769 {
4770 KaxChapterProcessCodecID *p_codec_id = static_cast<KaxChapterProcessCodecID*>( k );
4771 if ( uint32(*p_codec_id) == 0 )
4772 p_ccodec = new matroska_script_codec_c( sys );
4773 else if ( uint32(*p_codec_id) == 1 )
4774 p_ccodec = new dvd_chapter_codec_c( sys );
4775 break;
4776 }
4777 }
4778
4779 if ( p_ccodec != NULL )
4780 {
4781 for( j = 0; j < cp->ListSize(); j++ )
4782 {
4783 EbmlElement *k= (*cp)[j];
4784
4785 if( MKV_IS_ID( k, KaxChapterProcessPrivate ) )
4786 {
4787 KaxChapterProcessPrivate * p_private = static_cast<KaxChapterProcessPrivate*>( k );
4788 p_ccodec->SetPrivate( *p_private );
4789 }
4790 else if( MKV_IS_ID( k, KaxChapterProcessCommand ) )
4791 {
4792 p_ccodec->AddCommand( *static_cast<KaxChapterProcessCommand*>( k ) );
4793 }
4794 }
4795 chapters.codecs.push_back( p_ccodec );
4796 }
4797 }
4798 else if( MKV_IS_ID( l, KaxChapterAtom ) )
4799 {
4800 chapter_item_c *new_sub_chapter = new chapter_item_c();
4801 ParseChapterAtom( i_level+1, static_cast<KaxChapterAtom *>(l), *new_sub_chapter );
4802 new_sub_chapter->psz_parent = &chapters;
4803 chapters.sub_chapters.push_back( new_sub_chapter );
4804 }
4805 }
4806 }
4807
4808 /*****************************************************************************
4809 * ParseChapters:
4810 *****************************************************************************/
4811 void matroska_segment_c::ParseChapters( KaxChapters *chapters )
4812 {
4813 EbmlElement *el;
4814 size_t i;
4815 int i_upper_level = 0;
4816 mtime_t i_dur;
4817
4818 /* Master elements */
4819 chapters->Read( es, chapters->Generic().Context, i_upper_level, el, true );
4820
4821 for( i = 0; i < chapters->ListSize(); i++ )
4822 {
4823 EbmlElement *l = (*chapters)[i];
4824
4825 if( MKV_IS_ID( l, KaxEditionEntry ) )
4826 {
4827 chapter_edition_c *p_edition = new chapter_edition_c();
4828
4829 EbmlMaster *E = static_cast<EbmlMaster *>(l );
4830 size_t j;
4831 msg_Dbg( &sys.demuxer, "| | + EditionEntry" );
4832 for( j = 0; j < E->ListSize(); j++ )
4833 {
4834 EbmlElement *l = (*E)[j];
4835
4836 if( MKV_IS_ID( l, KaxChapterAtom ) )
4837 {
4838 chapter_item_c *new_sub_chapter = new chapter_item_c();
4839 ParseChapterAtom( 0, static_cast<KaxChapterAtom *>(l), *new_sub_chapter );
4840 p_edition->sub_chapters.push_back( new_sub_chapter );
4841 }
4842 else if( MKV_IS_ID( l, KaxEditionUID ) )
4843 {
4844 p_edition->i_uid = uint64(*static_cast<KaxEditionUID *>( l ));
4845 }
4846 else if( MKV_IS_ID( l, KaxEditionFlagOrdered ) )
4847 {
4848 p_edition->b_ordered = config_GetInt( &sys.demuxer, "mkv-use-ordered-chapters" ) ? (uint8(*static_cast<KaxEditionFlagOrdered *>( l )) != 0) : 0;
4849 }
4850 else if( MKV_IS_ID( l, KaxEditionFlagDefault ) )
4851 {
4852 if (uint8(*static_cast<KaxEditionFlagDefault *>( l )) != 0)
4853 i_default_edition = stored_editions.size();
4854 }
4855 else
4856 {
4857 msg_Dbg( &sys.demuxer, "| | | + Unknown (%s)", typeid(*l).name() );
4858 }
4859 }
4860 stored_editions.push_back( p_edition );
4861 }
4862 else
4863 {
4864 msg_Dbg( &sys.demuxer, "| | + Unknown (%s)", typeid(*l).name() );
4865 }
4866 }
4867
4868 for( i = 0; i < stored_editions.size(); i++ )
4869 {
4870 stored_editions[i]->RefreshChapters( );
4871 }
4872
4873 if ( stored_editions.size() != 0 && stored_editions[i_default_edition]->b_ordered )
4874 {
4875 /* update the duration of the segment according to the sum of all sub chapters */
4876 i_dur = stored_editions[i_default_edition]->Duration() / I64C(1000);
4877 if (i_dur > 0)
4878 i_duration = i_dur;
4879 }
4880 }
4881
4882 void matroska_segment_c::ParseCluster( )
4883 {
4884 EbmlElement *el;
4885 EbmlMaster *m;
4886 unsigned int i;
4887 int i_upper_level = 0;
4888
4889 /* Master elements */
4890 m = static_cast<EbmlMaster *>( cluster );
4891 m->Read( es, cluster->Generic().Context, i_upper_level, el, true );
4892
4893 for( i = 0; i < m->ListSize(); i++ )
4894 {
4895 EbmlElement *l = (*m)[i];
4896
4897 if( MKV_IS_ID( l, KaxClusterTimecode ) )
4898 {
4899 KaxClusterTimecode &ctc = *(KaxClusterTimecode*)l;
4900
4901 cluster->InitTimecode( uint64( ctc ), i_timescale );
4902 break;
4903 }
4904 }
4905
4906 i_start_time = cluster->GlobalTimecode() / 1000;
4907 }
4908
4909 /*****************************************************************************
4910 * InformationCreate:
4911 *****************************************************************************/
4912 void matroska_segment_c::InformationCreate( )
4913 {
4914 size_t i_track;
4915
4916 sys.meta = vlc_meta_New();
4917
4918 if( psz_title )
4919 {
4920 vlc_meta_Add( sys.meta, VLC_META_TITLE, psz_title );
4921 }
4922 if( psz_date_utc )
4923 {
4924 vlc_meta_Add( sys.meta, VLC_META_DATE, psz_date_utc );
4925 }
4926 if( psz_segment_filename )
4927 {
4928 vlc_meta_Add( sys.meta, _("Segment filename"), psz_segment_filename );
4929 }
4930 if( psz_muxing_application )
4931 {
4932 vlc_meta_Add( sys.meta, _("Muxing application"), psz_muxing_application );
4933 }
4934 if( psz_writing_application )
4935 {
4936 vlc_meta_Add( sys.meta, _("Writing application"), psz_writing_application );
4937 }
4938
4939 for( i_track = 0; i_track < tracks.size(); i_track++ )
4940 {
4941 mkv_track_t *tk = tracks[i_track];
4942 vlc_meta_t *mtk = vlc_meta_New();
4943
4944 sys.meta->track = (vlc_meta_t**)realloc( sys.meta->track,
4945 sizeof( vlc_meta_t * ) * ( sys.meta->i_track + 1 ) );
4946 sys.meta->track[sys.meta->i_track++] = mtk;
4947
4948 if( tk->fmt.psz_description )
4949 {
4950 vlc_meta_Add( sys.meta, VLC_META_DESCRIPTION, tk->fmt.psz_description );
4951 }
4952 if( tk->psz_codec_name )
4953 {
4954 vlc_meta_Add( sys.meta, VLC_META_CODEC_NAME, tk->psz_codec_name );
4955 }
4956 if( tk->psz_codec_settings )
4957 {
4958 vlc_meta_Add( sys.meta, VLC_META_SETTING, tk->psz_codec_settings );
4959 }
4960 if( tk->psz_codec_info_url )
4961 {
4962 vlc_meta_Add( sys.meta, VLC_META_CODEC_DESCRIPTION, tk->psz_codec_info_url );
4963 }
4964 if( tk->psz_codec_download_url )
4965 {
4966 vlc_meta_Add( sys.meta, VLC_META_URL, tk->psz_codec_download_url );
4967 }
4968 }
4969
4970 if( i_tags_position >= 0 )
4971 {
4972 vlc_bool_t b_seekable;
4973
4974 stream_Control( sys.demuxer.s, STREAM_CAN_FASTSEEK, &b_seekable );
4975 if( b_seekable )
4976 {
4977 LoadTags( );
4978 }
4979 }
4980 }
4981
4982
4983 /*****************************************************************************
4984 * Divers
4985 *****************************************************************************/
4986
4987 void matroska_segment_c::IndexAppendCluster( KaxCluster *cluster )
4988 {
4989 #define idx p_indexes[i_index]
4990 idx.i_track = -1;
4991 idx.i_block_number= -1;
4992 idx.i_position = cluster->GetElementPosition();
4993 idx.i_time = -1;
4994 idx.b_key = VLC_TRUE;
4995
4996 i_index++;
4997 if( i_index >= i_index_max )
4998 {
4999 i_index_max += 1024;
5000 p_indexes = (mkv_index_t*)realloc( p_indexes, sizeof( mkv_index_t ) * i_index_max );
5001 }
5002 #undef idx
5003 }
5004
5005 void chapter_edition_c::RefreshChapters( )
5006 {
5007 chapter_item_c::RefreshChapters( b_ordered, -1 );
5008 b_display_seekpoint = false;
5009 }
5010
5011 int64_t chapter_item_c::RefreshChapters( bool b_ordered, int64_t i_prev_user_time )
5012 {
5013 int64_t i_user_time = i_prev_user_time;
5014
5015 // first the sub-chapters, and then ourself
5016 std::vector<chapter_item_c*>::iterator index = sub_chapters.begin();
5017 while ( index != sub_chapters.end() )
5018 {
5019 i_user_time = (*index)->RefreshChapters( b_ordered, i_user_time );
5020 index++;
5021 }
5022
5023 if ( b_ordered )
5024 {
5025 // the ordered chapters always start at zero
5026 if ( i_prev_user_time == -1 )
5027 {
5028 if ( i_user_time == -1 )
5029 i_user_time = 0;
5030 i_prev_user_time = 0;
5031 }
5032
5033 i_user_start_time = i_prev_user_time;
5034 if ( i_end_time != -1 && i_user_time == i_prev_user_time )
5035 {
5036 i_user_end_time = i_user_start_time - i_start_time + i_end_time;
5037 }
5038 else
5039 {
5040 i_user_end_time = i_user_time;
5041 }
5042 }
5043 else
5044 {
5045 if ( sub_chapters.begin() != sub_chapters.end() )
5046 std::sort( sub_chapters.begin(), sub_chapters.end(), chapter_item_c::CompareTimecode );
5047 i_user_start_time = i_start_time;
5048 if ( i_end_time != -1 )
5049 i_user_end_time = i_end_time;
5050 else if ( i_user_time != -1 )
5051 i_user_end_time = i_user_time;
5052 else
5053 i_user_end_time = i_user_start_time;
5054 }
5055
5056 return i_user_end_time;
5057 }
5058
5059 mtime_t chapter_edition_c::Duration() const
5060 {
5061 mtime_t i_result = 0;
5062
5063 if ( sub_chapters.size() )
5064 {
5065 std::vector<chapter_item_c*>::const_iterator index = sub_chapters.end();
5066 index--;
5067 i_result = (*index)->i_user_end_time;
5068 }
5069
5070 return i_result;
5071 }
5072
5073 chapter_item_c * chapter_edition_c::FindTimecode( mtime_t i_timecode, const chapter_item_c * p_current )
5074 {
5075 if ( !b_ordered )
5076 p_current = NULL;
5077 bool b_found_current = false;
5078 return chapter_item_c::FindTimecode( i_timecode, p_current, b_found_current );
5079 }
5080
5081 chapter_item_c *chapter_item_c::FindTimecode( mtime_t i_user_timecode, const chapter_item_c * p_current, bool & b_found )
5082 {
5083 chapter_item_c *psz_result = NULL;
5084
5085 if ( p_current == this )
5086 b_found = true;
5087
5088 if ( i_user_timecode >= i_user_start_time &&
5089 ( i_user_timecode < i_user_end_time ||
5090 ( i_user_start_time == i_user_end_time && i_user_timecode == i_user_end_time )))
5091 {
5092 std::vector<chapter_item_c*>::iterator index = sub_chapters.begin();
5093 while ( index != sub_chapters.end() && ((p_current == NULL && psz_result == NULL) || (p_current != NULL && (!b_found || psz_result == NULL))))
5094 {
5095 psz_result = (*index)->FindTimecode( i_user_timecode, p_current, b_found );
5096 index++;
5097 }
5098
5099 if ( psz_result == NULL )
5100 psz_result = this;
5101 }
5102
5103 return psz_result;
5104 }
5105
5106 bool chapter_item_c::ParentOf( const chapter_item_c & item ) const
5107 {
5108 if ( &item == this )
5109 return true;
5110
5111 std::vector<chapter_item_c*>::const_iterator index = sub_chapters.begin();
5112 while ( index != sub_chapters.end() )
5113 {
5114 if ( (*index)->ParentOf( item ) )
5115 return true;
5116 index++;
5117 }
5118
5119 return false;
5120 }
5121
5122 void demux_sys_t::PreloadFamily( const matroska_segment_c & of_segment )
5123 {
5124 for (size_t i=0; i<opened_segments.size(); i++)
5125 {
5126 opened_segments[i]->PreloadFamily( of_segment );
5127 }
5128 }
5129 bool matroska_segment_c::PreloadFamily( const matroska_segment_c & of_segment )
5130 {
5131 if ( b_preloaded )
5132 return false;
5133
5134 for (size_t i=0; i<families.size(); i++)
5135 {
5136 for (size_t j=0; j<of_segment.families.size(); j++)
5137 {
5138 if ( *(families[i]) == *(of_segment.families[j]) )
5139 return Preload( );
5140 }
5141 }
5142
5143 return false;
5144 }
5145
5146 // preload all the linked segments for all preloaded segments
5147 void demux_sys_t::PreloadLinked( matroska_segment_c *p_segment )
5148 {
5149 size_t i_preloaded, i, j;
5150 virtual_segment_c *p_seg;
5151
5152 p_current_segment = VirtualFromSegments( p_segment );
5153
5154 used_segments.push_back( p_current_segment );
5155
5156 // create all the other virtual segments of the family
5157 do {
5158 i_preloaded = 0;
5159 for ( i=0; i< opened_segments.size(); i++ )
5160 {
5161 if ( opened_segments[i]->b_preloaded && !IsUsedSegment( *opened_segments[i] ) )
5162 {
5163 p_seg = VirtualFromSegments( opened_segments[i] );
5164 used_segments.push_back( p_seg );
5165 i_preloaded++;
5166 }
5167 }
5168 } while ( i_preloaded ); // worst case: will stop when all segments are found as family related
5169
5170 // publish all editions of all usable segment
5171 for ( i=0; i< used_segments.size(); i++ )
5172 {
5173 p_seg = used_segments[i];
5174 if ( p_seg->p_editions != NULL )
5175 {
5176 std::string sz_name;
5177 input_title_t *p_title = vlc_input_title_New();
5178 p_seg->i_sys_title = i;
5179 int i_chapters;
5180
5181 // TODO use a name for each edition, let the TITLE deal with a codec name
5182 for ( j=0; j<p_seg->p_editions->size(); j++ )
5183 {
5184 if ( p_title->psz_name == NULL )
5185 {
5186 sz_name = (*p_seg->p_editions)[j]->GetMainName();
5187 if ( sz_name != "" )
5188 p_title->psz_name = strdup( sz_name.c_str() );
5189 }
5190
5191 chapter_edition_c *p_edition = (*p_seg->p_editions)[j];
5192
5193 i_chapters = 0;
5194 p_edition->PublishChapters( *p_title, i_chapters, 0 );
5195 }
5196
5197 // create a name if there is none
5198 if ( p_title->psz_name == NULL )
5199 {
5200 sz_name = N_("Segment");
5201 char psz_str[6];
5202 sprintf( psz_str, " %d", (int)i );
5203 sz_name += psz_str;
5204 p_title->psz_name = strdup( sz_name.c_str() );
5205 }
5206
5207 titles.push_back( p_title );
5208 }
5209 }
5210
5211 // TODO decide which segment should be first used (VMG for DVD)
5212 }
5213
5214 bool demux_sys_t::IsUsedSegment( matroska_segment_c &segment ) const
5215 {
5216 for ( size_t i=0; i< used_segments.size(); i++ )
5217 {
5218 if ( used_segments[i]->FindUID( *segment.p_segment_uid ) )
5219 return true;
5220 }
5221 return false;
5222 }
5223
5224 virtual_segment_c *demux_sys_t::VirtualFromSegments( matroska_segment_c *p_segment ) const
5225 {
5226 size_t i_preloaded, i;
5227
5228 virtual_segment_c *p_result = new virtual_segment_c( p_segment );
5229
5230 // fill our current virtual segment with all hard linked segments
5231 do {
5232 i_preloaded = 0;
5233 for ( i=0; i< opened_segments.size(); i++ )
5234 {
5235 i_preloaded += p_result->AddSegment( opened_segments[i] );
5236 }
5237 } while ( i_preloaded ); // worst case: will stop when all segments are found as linked
5238
5239 p_result->Sort( );
5240
5241 p_result->PreloadLinked( );
5242
5243 p_result->PrepareChapters( );
5244
5245 return p_result;
5246 }
5247
5248 bool demux_sys_t::PreparePlayback( virtual_segment_c *p_new_segment )
5249 {
5250 if ( p_new_segment != NULL && p_new_segment != p_current_segment )
5251 {
5252 if ( p_current_segment != NULL && p_current_segment->Segment() != NULL )
5253 p_current_segment->Segment()->UnSelect();
5254
5255 p_current_segment = p_new_segment;
5256 i_current_title = p_new_segment->i_sys_title;
5257 }
5258
5259 p_current_segment->LoadCues();
5260 f_duration = p_current_segment->Duration();
5261
5262 /* add information */
5263 p_current_segment->Segment()->InformationCreate( );
5264
5265 p_current_segment->Segment()->Select( 0 );
5266
5267 return true;
5268 }
5269
5270 void demux_sys_t::JumpTo( virtual_segment_c & vsegment, chapter_item_c * p_chapter )
5271 {
5272 // if the segment is not part of the current segment, select the new one
5273 if ( &vsegment != p_current_segment )
5274 {
5275 PreparePlayback( &vsegment );
5276 }
5277
5278 if ( p_chapter != NULL )
5279 {
5280 if ( !p_chapter->Enter( true ) )
5281 {
5282 // jump to the location in the found segment
5283 vsegment.Seek( demuxer, p_chapter->i_user_start_time, -1, p_chapter );
5284 }
5285 }
5286
5287 }
5288
5289 bool matroska_segment_c::CompareSegmentUIDs( const matroska_segment_c * p_item_a, const matroska_segment_c * p_item_b )
5290 {
5291 if ( p_item_a == NULL || p_item_b == NULL )
5292 return false;
5293
5294 EbmlBinary * p_itema = (EbmlBinary *)(p_item_a->p_segment_uid);
5295 if ( p_item_b->p_prev_segment_uid != NULL && *p_itema == *p_item_b->p_prev_segment_uid )
5296 return true;
5297
5298 p_itema = (EbmlBinary *)(&p_item_a->p_next_segment_uid);
5299 if ( p_item_b->p_segment_uid != NULL && *p_itema == *p_item_b->p_segment_uid )
5300 return true;
5301
5302 if ( p_item_b->p_prev_segment_uid != NULL && *p_itema == *p_item_b->p_prev_segment_uid )
5303 return true;
5304
5305 return false;
5306 }
5307
5308 bool matroska_segment_c::Preload( )
5309 {
5310 if ( b_preloaded )
5311 return false;
5312
5313 EbmlElement *el = NULL;
5314
5315 ep->Reset( &sys.demuxer );
5316
5317 while( ( el = ep->Get() ) != NULL )
5318 {
5319 if( MKV_IS_ID( el, KaxInfo ) )
5320 {
5321 ParseInfo( static_cast<KaxInfo*>( el ) );
5322 }
5323 else if( MKV_IS_ID( el, KaxTracks ) )
5324 {
5325 ParseTracks( static_cast<KaxTracks*>( el ) );
5326 if ( tracks.size() == 0 )
5327 {
5328 msg_Err( &sys.demuxer, "No tracks supported" );
5329 return false;
5330 }
5331 }
5332 else if( MKV_IS_ID( el, KaxSeekHead ) )
5333 {
5334 ParseSeekHead( static_cast<KaxSeekHead*>( el ) );
5335 }
5336 else if( MKV_IS_ID( el, KaxCues ) )
5337 {
5338 msg_Dbg( &sys.demuxer, "| + Cues" );
5339 }
5340 else if( MKV_IS_ID( el, KaxCluster ) )
5341 {
5342 msg_Dbg( &sys.demuxer, "| + Cluster" );
5343
5344 cluster = (KaxCluster*)el;
5345
5346 i_cluster_pos = i_start_pos = cluster->GetElementPosition();
5347 ParseCluster( );
5348
5349 ep->Down();
5350 /* stop parsing the stream */
5351 break;
5352 }
5353 else if( MKV_IS_ID( el, KaxAttachments ) )
5354 {
5355 msg_Dbg( &sys.demuxer, "| + Attachments FIXME (but probably never supported)" );
5356 }
5357 else if( MKV_IS_ID( el, KaxChapters ) )
5358 {
5359 msg_Dbg( &sys.demuxer, "| + Chapters" );
5360 ParseChapters( static_cast<KaxChapters*>( el ) );
5361 }
5362 else if( MKV_IS_ID( el, KaxTag ) )
5363 {
5364 msg_Dbg( &sys.demuxer, "| + Tags FIXME TODO" );
5365 }
5366 else
5367 {
5368 msg_Dbg( &sys.demuxer, "| + Unknown (%s)", typeid(*el).name() );
5369 }
5370 }
5371
5372 b_preloaded = true;
5373
5374 return true;
5375 }
5376
5377 matroska_segment_c *demux_sys_t::FindSegment( const EbmlBinary & uid ) const
5378 {
5379 for (size_t i=0; i<opened_segments.size(); i++)
5380 {
5381 if ( *opened_segments[i]->p_segment_uid == uid )
5382 return opened_segments[i];
5383 }
5384 return NULL;
5385 }
5386
5387 chapter_item_c *demux_sys_t::BrowseCodecPrivate( unsigned int codec_id,
5388 bool (*match)(const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size ),
5389 const void *p_cookie,
5390 size_t i_cookie_size,
5391 virtual_segment_c * &p_segment_found )
5392 {
5393 chapter_item_c *p_result = NULL;
5394 for (size_t i=0; i<used_segments.size(); i++)
5395 {
5396 p_result = used_segments[i]->BrowseCodecPrivate( codec_id, match, p_cookie, i_cookie_size );
5397 if ( p_result != NULL )
5398 {
5399 p_segment_found = used_segments[i];
5400 break;
5401 }
5402 }
5403 return p_result;
5404 }
5405
5406 chapter_item_c *demux_sys_t::FindChapter( int64_t i_find_uid, virtual_segment_c * & p_segment_found )
5407 {
5408 chapter_item_c *p_result = NULL;
5409 for (size_t i=0; i<used_segments.size(); i++)
5410 {
5411 p_result = used_segments[i]->FindChapter( i_find_uid );
5412 if ( p_result != NULL )
5413 {
5414 p_segment_found = used_segments[i];
5415 break;
5416 }
5417 }
5418 return p_result;
5419 }
5420
5421 void virtual_segment_c::Sort()
5422 {
5423 // keep the current segment index
5424 matroska_segment_c *p_segment = linked_segments[i_current_segment];
5425
5426 std::sort( linked_segments.begin(), linked_segments.end(), matroska_segment_c::CompareSegmentUIDs );
5427
5428 for ( i_current_segment=0; i_current_segment<linked_segments.size(); i_current_segment++)
5429 if ( linked_segments[i_current_segment] == p_segment )
5430 break;
5431 }
5432
5433 size_t virtual_segment_c::AddSegment( matroska_segment_c *p_segment )
5434 {
5435 size_t i;
5436 // check if it's not already in here
5437 for ( i=0; i<linked_segments.size(); i++ )
5438 {
5439 if ( linked_segments[i]->p_segment_uid != NULL
5440 && p_segment->p_segment_uid != NULL
5441 && *p_segment->p_segment_uid == *linked_segments[i]->p_segment_uid )
5442 return 0;
5443 }
5444
5445 // find possible mates
5446 for ( i=0; i<linked_uids.size(); i++ )
5447 {
5448 if ( (p_segment->p_segment_uid != NULL && *p_segment->p_segment_uid == linked_uids[i])
5449 || (p_segment->p_prev_segment_uid != NULL && *p_segment->p_prev_segment_uid == linked_uids[i])
5450 || (p_segment->p_next_segment_uid !=NULL && *p_segment->p_next_segment_uid == linked_uids[i]) )
5451 {
5452 linked_segments.push_back( p_segment );
5453
5454 AppendUID( p_segment->p_prev_segment_uid );
5455 AppendUID( p_segment->p_next_segment_uid );
5456
5457 return 1;
5458 }
5459 }
5460 return 0;
5461 }
5462
5463 void virtual_segment_c::PreloadLinked( )
5464 {
5465 for ( size_t i=0; i<linked_segments.size(); i++ )
5466 {
5467 linked_segments[i]->Preload( );
5468 }
5469 i_current_edition = linked_segments[0]->i_default_edition;
5470 }
5471
5472 mtime_t virtual_segment_c::Duration() const
5473 {
5474 mtime_t i_duration;
5475 if ( linked_segments.size() == 0 )
5476 i_duration = 0;
5477 else {
5478 matroska_segment_c *p_last_segment = linked_segments[linked_segments.size()-1];
5479 // p_last_segment->ParseCluster( );
5480
5481 i_duration = p_last_segment->i_start_time / 1000 + p_last_segment->i_duration;
5482 }
5483 return i_duration;
5484 }
5485
5486 void virtual_segment_c::LoadCues( )
5487 {
5488 for ( size_t i=0; i<linked_segments.size(); i++ )
5489 {
5490 linked_segments[i]->LoadCues();
5491 }
5492 }
5493
5494 void virtual_segment_c::AppendUID( const EbmlBinary * p_UID )
5495 {
5496 if ( p_UID == NULL )
5497 return;
5498 if ( p_UID->GetBuffer() == NULL )
5499 return;
5500
5501 for (size_t i=0; i<linked_uids.size(); i++)
5502 {
5503 if ( *p_UID == linked_uids[i] )
5504 return;
5505 }
5506 linked_uids.push_back( *(KaxSegmentUID*)(p_UID) );
5507 }
5508
5509 void matroska_segment_c::Seek( mtime_t i_date, mtime_t i_time_offset )
5510 {
5511 KaxBlock *block;
5512 #if LIBMATROSKA_VERSION >= 0x000800
5513 KaxSimpleBlock *simpleblock;
5514 #endif
5515 int i_track_skipping;
5516 int64_t i_block_duration;
5517 int64_t i_block_ref1;
5518 int64_t i_block_ref2;
5519 size_t i_track;
5520 int64_t i_seek_position = i_start_pos;
5521 int64_t i_seek_time = i_start_time;
5522
5523 if ( i_index > 0 )
5524 {
5525 int i_idx = 0;
5526
5527 for( ; i_idx < i_index; i_idx++ )
5528 {
5529 if( p_indexes[i_idx].i_time + i_time_offset > i_date )
5530 {
5531 break;
5532 }
5533 }
5534
5535 if( i_idx > 0 )
5536 {
5537 i_idx--;
5538 }
5539
5540 i_seek_position = p_indexes[i_idx].i_position;
5541 i_seek_time = p_indexes[i_idx].i_time;
5542 }
5543
5544 msg_Dbg( &sys.demuxer, "seek got "I64Fd" (%d%%)",
5545 i_seek_time, (int)( 100 * i_seek_position / stream_Size( sys.demuxer.s ) ) );
5546
5547 es.I_O().setFilePointer( i_seek_position, seek_beginning );
5548
5549 delete ep;
5550 ep = new EbmlParser( &es, segment, &sys.demuxer );
5551 cluster = NULL;
5552
5553 sys.i_start_pts = i_date;
5554
5555 es_out_Control( sys.demuxer.out, ES_OUT_RESET_PCR );
5556
5557 /* now parse until key frame */
5558 i_track_skipping = 0;
5559 for( i_track = 0; i_track < tracks.size(); i_track++ )
5560 {
5561 if( tracks[i_track]->fmt.i_cat == VIDEO_ES )
5562 {
5563 tracks[i_track]->b_search_keyframe = VLC_TRUE;
5564 i_track_skipping++;
5565 }
5566 es_out_Control( sys.demuxer.out, ES_OUT_SET_NEXT_DISPLAY_TIME, tracks[i_track]->p_es, i_date );
5567 }
5568
5569
5570 while( i_track_skipping > 0 )
5571 {
5572 #if LIBMATROSKA_VERSION >= 0x000800
5573 if( BlockGet( block, simpleblock, &i_block_ref1, &i_block_ref2, &i_block_duration ) )
5574 #else
5575 if( BlockGet( block, &i_block_ref1, &i_block_ref2, &i_block_duration ) )
5576 #endif
5577 {
5578 msg_Warn( &sys.demuxer, "cannot get block EOF?" );
5579
5580 return;
5581 }
5582 ep->Down();
5583
5584 for( i_track = 0; i_track < tracks.size(); i_track++ )
5585 {
5586 if( tracks[i_track]->i_number == block->TrackNum() )
5587 {
5588 break;
5589 }
5590 }
5591
5592 sys.i_pts = (sys.i_chapter_time + block->GlobalTimecode()) / (mtime_t) 1000;
5593
5594 if( i_track < tracks.size() )
5595 {
5596 if( sys.i_pts >= sys.i_start_pts )
5597 {
5598 cluster = static_cast<KaxCluster*>(ep->UnGet( i_block_pos, i_cluster_pos ));
5599 i_track_skipping = 0;
5600 }
5601 else if( tracks[i_track]->fmt.i_cat == VIDEO_ES )
5602 {
5603 if( i_block_ref1 == 0 && tracks[i_track]->b_search_keyframe )
5604 {
5605 tracks[i_track]->b_search_keyframe = VLC_FALSE;
5606 i_track_skipping--;
5607 }
5608 if( !tracks[i_track]->b_search_keyframe )
5609 {
5610 #if LIBMATROSKA_VERSION >= 0x000800
5611 BlockDecode( &sys.demuxer, block, simpleblock, sys.i_pts, 0, i_block_ref1 >= 0 || i_block_ref2 > 0 );
5612 #else
5613 BlockDecode( &sys.demuxer, block, sys.i_pts, 0, i_block_ref1 >= 0 || i_block_ref2 > 0 );
5614 #endif
5615 }
5616 }
5617 }
5618
5619 delete block;
5620 }
5621 }
5622
5623 void virtual_segment_c::Seek( demux_t & demuxer, mtime_t i_date, mtime_t i_time_offset, chapter_item_c *psz_chapter )
5624 {
5625 demux_sys_t *p_sys = demuxer.p_sys;
5626 size_t i;
5627
5628 // find the actual time for an ordered edition
5629 if ( psz_chapter == NULL )
5630 {
5631 if ( Edition() && Edition()->b_ordered )
5632 {
5633 /* 1st, we need to know in which chapter we are */
5634 psz_chapter = (*p_editions)[i_current_edition]->FindTimecode( i_date, psz_current_chapter );
5635 }
5636 }
5637
5638 if ( psz_chapter != NULL )
5639 {
5640 psz_current_chapter = psz_chapter;
5641 p_sys->i_chapter_time = i_time_offset = psz_chapter->i_user_start_time - psz_chapter->i_start_time;
5642 if ( psz_chapter->i_seekpoint_num > 0 )
5643 {
5644 demuxer.info.i_update |= INPUT_UPDATE_TITLE | INPUT_UPDATE_SEEKPOINT;
5645 demuxer.info.i_title = p_sys->i_current_title = i_sys_title;
5646 demuxer.info.i_seekpoint = psz_chapter->i_seekpoint_num - 1;
5647 }
5648 }
5649
5650 // find the best matching segment
5651 for ( i=0; i<linked_segments.size(); i++ )
5652 {
5653 if ( i_date < linked_segments[i]->i_start_time )
5654 break;
5655 }
5656
5657 if ( i > 0 )
5658 i--;
5659
5660 if ( i_current_segment != i )
5661 {
5662 linked_segments[i_current_segment]->UnSelect();
5663 linked_segments[i]->Select( i_date );
5664 i_current_segment = i;
5665 }
5666
5667 linked_segments[i]->Seek( i_date, i_time_offset );
5668 }
5669
5670 void chapter_codec_cmds_c::AddCommand( const KaxChapterProcessCommand & command )
5671 {
5672 size_t i;
5673
5674 uint32 codec_time = uint32(-1);
5675 for( i = 0; i < command.ListSize(); i++ )
5676 {
5677 const EbmlElement *k = command[i];
5678
5679 if( MKV_IS_ID( k, KaxChapterProcessTime ) )
5680 {
5681 codec_time = uint32( *static_cast<const KaxChapterProcessTime*>( k ) );
5682 break;
5683 }
5684 }
5685
5686 for( i = 0; i < command.ListSize(); i++ )
5687 {
5688 const EbmlElement *k = command[i];
5689
5690 if( MKV_IS_ID( k, KaxChapterProcessData ) )
5691 {
5692 KaxChapterProcessData *p_data = new KaxChapterProcessData( *static_cast<const KaxChapterProcessData*>( k ) );
5693 switch ( codec_time )
5694 {
5695 case 0:
5696 during_cmds.push_back( p_data );
5697 break;
5698 case 1:
5699 enter_cmds.push_back( p_data );
5700 break;
5701 case 2:
5702 leave_cmds.push_back( p_data );
5703 break;
5704 default:
5705 delete p_data;
5706 }
5707 }
5708 }
5709 }
5710
5711 bool chapter_item_c::Enter( bool b_do_subs )
5712 {
5713 bool f_result = false;
5714 std::vector<chapter_codec_cmds_c*>::iterator index = codecs.begin();
5715 while ( index != codecs.end() )
5716 {
5717 f_result |= (*index)->Enter();
5718 index++;
5719 }
5720
5721 if ( b_do_subs )
5722 {
5723 // sub chapters
5724 std::vector<chapter_item_c*>::iterator index_ = sub_chapters.begin();
5725 while ( index_ != sub_chapters.end() )
5726 {
5727 f_result |= (*index_)->Enter( true );
5728 index_++;
5729 }
5730 }
5731 return f_result;
5732 }
5733
5734 bool chapter_item_c::Leave( bool b_do_subs )
5735 {
5736 bool f_result = false;
5737 b_is_leaving = true;
5738 std::vector<chapter_codec_cmds_c*>::iterator index = codecs.begin();
5739 while ( index != codecs.end() )
5740 {
5741 f_result |= (*index)->Leave();
5742 index++;
5743 }
5744
5745 if ( b_do_subs )
5746 {
5747 // sub chapters
5748 std::vector<chapter_item_c*>::iterator index_ = sub_chapters.begin();
5749 while ( index_ != sub_chapters.end() )
5750 {
5751 f_result |= (*index_)->Leave( true );
5752 index_++;
5753 }
5754 }
5755 b_is_leaving = false;
5756 return f_result;
5757 }
5758
5759 bool chapter_item_c::EnterAndLeave( chapter_item_c *p_item, bool b_final_enter )
5760 {
5761 chapter_item_c *p_common_parent = p_item;
5762
5763 // leave, up to a common parent
5764 while ( p_common_parent != NULL && !p_common_parent->ParentOf( *this ) )
5765 {
5766 if ( !p_common_parent->b_is_leaving && p_common_parent->Leave( false ) )
5767 return true;
5768 p_common_parent = p_common_parent->psz_parent;
5769 }
5770
5771 // enter from the parent to <this>
5772 if ( p_common_parent != NULL )
5773 {
5774 do
5775 {
5776 if ( p_common_parent == this )
5777 return Enter( true );
5778
5779 for ( size_t i = 0; i<p_common_parent->sub_chapters.size(); i++ )
5780 {
5781 if ( p_common_parent->sub_chapters[i]->ParentOf( *this ) )
5782 {
5783 p_common_parent = p_common_parent->sub_chapters[i];
5784 if ( p_common_parent != this )
5785 if ( p_common_parent->Enter( false ) )
5786 return true;
5787
5788 break;
5789 }
5790 }
5791 } while ( 1 );
5792 }
5793
5794 if ( b_final_enter )
5795 return Enter( true );
5796 else
5797 return false;
5798 }
5799
5800 bool dvd_chapter_codec_c::Enter()
5801 {
5802 bool f_result = false;
5803 std::vector<KaxChapterProcessData*>::iterator index = enter_cmds.begin();
5804 while ( index != enter_cmds.end() )
5805 {
5806 if ( (*index)->GetSize() )
5807 {
5808 binary *p_data = (*index)->GetBuffer();
5809 size_t i_size = *p_data++;
5810 // avoid reading too much from the buffer
5811 i_size = min( i_size, ((*index)->GetSize() - 1) >> 3 );
5812 for ( ; i_size > 0; i_size--, p_data += 8 )
5813 {
5814 msg_Dbg( &sys.demuxer, "Matroska DVD enter command" );
5815 f_result |= sys.dvd_interpretor.Interpret( p_data );
5816 }
5817 }
5818 index++;
5819 }
5820 return f_result;
5821 }
5822
5823 bool dvd_chapter_codec_c::Leave()
5824 {
5825 bool f_result = false;
5826 std::vector<KaxChapterProcessData*>::iterator index = leave_cmds.begin();
5827 while ( index != leave_cmds.end() )
5828 {
5829 if ( (*index)->GetSize() )
5830 {
5831 binary *p_data = (*index)->GetBuffer();
5832 size_t i_size = *p_data++;
5833 // avoid reading too much from the buffer
5834 i_size = min( i_size, ((*index)->GetSize() - 1) >> 3 );
5835 for ( ; i_size > 0; i_size--, p_data += 8 )
5836 {
5837 msg_Dbg( &sys.demuxer, "Matroska DVD leave command" );
5838 f_result |= sys.dvd_interpretor.Interpret( p_data );
5839 }
5840 }
5841 index++;
5842 }
5843 return f_result;
5844 }
5845
5846 // see http://www.dvd-replica.com/DVD/vmcmdset.php for a description of DVD commands
5847 bool dvd_command_interpretor_c::Interpret( const binary * p_command, size_t i_size )
5848 {
5849 if ( i_size != 8 )
5850 return false;
5851
5852 virtual_segment_c *p_segment = NULL;
5853 chapter_item_c *p_chapter = NULL;
5854 bool f_result = false;
5855 uint16 i_command = ( p_command[0] << 8 ) + p_command[1];
5856
5857 // handle register tests if there are some
5858 if ( (i_command & 0xF0) != 0 )
5859 {
5860 bool b_test_positive = true;//(i_command & CMD_DVD_IF_NOT) == 0;
5861 bool b_test_value = (i_command & CMD_DVD_TEST_VALUE) != 0;
5862 uint8 i_test = i_command & 0x70;
5863 uint16 i_value;
5864
5865 // see http://dvd.sourceforge.net/dvdinfo/vmi.html
5866 uint8 i_cr1;
5867 uint16 i_cr2;
5868 switch ( i_command >> 12 )
5869 {
5870 default:
5871 i_cr1 = p_command[3];
5872 i_cr2 = (p_command[4] << 8) + p_command[5];
5873 break;
5874 case 3:
5875 case 4:
5876 case 5:
5877 i_cr1 = p_command[6];
5878 i_cr2 = p_command[7];
5879 b_test_value = false;
5880 break;
5881 case 6:
5882 case 7:
5883 if ( ((p_command[1] >> 4) & 0x7) == 0)
5884 {
5885 i_cr1 = p_command[2];
5886 i_cr2 = (p_command[6] << 8) + p_command[7];
5887 }
5888 else
5889 {
5890 i_cr1 = p_command[2];
5891 i_cr2 = (p_command[6] << 8) + p_command[7];
5892 }
5893 break;
5894 }
5895
5896 if ( b_test_value )
5897 i_value = i_cr2;
5898 else
5899 i_value = GetPRM( i_cr2 );
5900
5901 switch ( i_test )
5902 {
5903 case CMD_DVD_IF_GPREG_EQUAL:
5904 // if equals
5905 msg_Dbg( &sys.demuxer, "IF %s EQUALS %s", GetRegTypeName( false, i_cr1 ).c_str(), GetRegTypeName( b_test_value, i_value ).c_str() );
5906 if (!( GetPRM( i_cr1 ) == i_value ))
5907 {
5908 b_test_positive = false;
5909 }
5910 break;
5911 case CMD_DVD_IF_GPREG_NOT_EQUAL:
5912 // if not equals
5913 msg_Dbg( &sys.demuxer, "IF %s NOT EQUALS %s", GetRegTypeName( false, i_cr1 ).c_str(), GetRegTypeName( b_test_value, i_value ).c_str() );
5914 if (!( GetPRM( i_cr1 ) != i_value ))
5915 {
5916 b_test_positive = false;
5917 }
5918 break;
5919 case CMD_DVD_IF_GPREG_INF:
5920 // if inferior
5921 msg_Dbg( &sys.demuxer, "IF %s < %s", GetRegTypeName( false, p_command[3] ).c_str(), GetRegTypeName( b_test_value, i_value ).c_str() );
5922 if (!( GetPRM( i_cr1 ) < i_value ))
5923 {
5924 b_test_positive = false;
5925 }
5926 break;
5927 case CMD_DVD_IF_GPREG_INF_EQUAL:
5928 // if inferior or equal
5929 msg_Dbg( &sys.demuxer, "IF %s < %s", GetRegTypeName( false, p_command[3] ).c_str(), GetRegTypeName( b_test_value, i_value ).c_str() );
5930 if (!( GetPRM( i_cr1 ) <= i_value ))
5931 {
5932 b_test_positive = false;
5933 }
5934 break;
5935 case CMD_DVD_IF_GPREG_AND:
5936 // if logical and
5937 msg_Dbg( &sys.demuxer, "IF %s & %s", GetRegTypeName( false, p_command[3] ).c_str(), GetRegTypeName( b_test_value, i_value ).c_str() );
5938 if (!( GetPRM( i_cr1 ) & i_value ))
5939 {
5940 b_test_positive = false;
5941 }
5942 break;
5943 case CMD_DVD_IF_GPREG_SUP:
5944 // if superior
5945 msg_Dbg( &sys.demuxer, "IF %s >= %s", GetRegTypeName( false, p_command[3] ).c_str(), GetRegTypeName( b_test_value, i_value ).c_str() );
5946 if (!( GetPRM( i_cr1 ) > i_value ))
5947 {
5948 b_test_positive = false;
5949 }
5950 break;
5951 case CMD_DVD_IF_GPREG_SUP_EQUAL:
5952 // if superior or equal
5953 msg_Dbg( &sys.demuxer, "IF %s >= %s", GetRegTypeName( false, p_command[3] ).c_str(), GetRegTypeName( b_test_value, i_value ).c_str() );
5954 if (!( GetPRM( i_cr1 ) >= i_value ))
5955 {
5956 b_test_positive = false;
5957 }
5958 break;
5959 }
5960
5961 if ( !b_test_positive )
5962 return false;
5963 }
5964
5965 // strip the test command
5966 i_command &= 0xFF0F;
5967
5968 switch ( i_command )
5969 {
5970 case CMD_DVD_NOP:
5971 case CMD_DVD_NOP2:
5972 {
5973 msg_Dbg( &sys.demuxer, "NOP" );
5974 break;
5975 }
5976 case CMD_DVD_BREAK:
5977 {
5978 msg_Dbg( &sys.demuxer, "Break" );
5979 // TODO
5980 break;
5981 }
5982 case CMD_DVD_JUMP_TT:
5983 {
5984 uint8 i_title = p_command[5];
5985 msg_Dbg( &sys.demuxer, "JumpTT %d", i_title );
5986
5987 // find in the ChapProcessPrivate matching this Title level
5988 p_chapter = sys.BrowseCodecPrivate( 1, MatchTitleNumber, &i_title, sizeof(i_title), p_segment );
5989 if ( p_segment != NULL )
5990 {
5991 sys.JumpTo( *p_segment, p_chapter );
5992 f_result = true;
5993 }
5994
5995 break;
5996 }
5997 case CMD_DVD_CALLSS_VTSM1:
5998 {
5999 msg_Dbg( &sys.demuxer, "CallSS" );
6000 binary p_type;
6001 switch( (p_command[6] & 0xC0) >> 6 ) {
6002 case 0:
6003 p_type = p_command[5] & 0x0F;
6004 switch ( p_type )
6005 {
6006 case 0x00:
6007 msg_Dbg( &sys.demuxer, "CallSS PGC (rsm_cell %x)", p_command[4]);
6008 break;
6009 case 0x02:
6010 msg_Dbg( &sys.demuxer, "CallSS Title Entry (rsm_cell %x)", p_command[4]);
6011 break;
6012 case 0x03:
6013 msg_Dbg( &sys.demuxer, "CallSS Root Menu (rsm_cell %x)", p_command[4]);
6014 break;
6015 case 0x04:
6016 msg_Dbg( &sys.demuxer, "CallSS Subpicture Menu (rsm_cell %x)", p_command[4]);
6017 break;
6018 case 0x05:
6019 msg_Dbg( &sys.demuxer, "CallSS Audio Menu (rsm_cell %x)", p_command[4]);
6020 break;
6021 case 0x06:
6022 msg_Dbg( &sys.demuxer, "CallSS Angle Menu (rsm_cell %x)", p_command[4]);
6023 break;
6024 case 0x07:
6025 msg_Dbg( &sys.demuxer, "CallSS Chapter Menu (rsm_cell %x)", p_command[4]);
6026 break;
6027 default:
6028 msg_Dbg( &sys.demuxer, "CallSS <unknown> (rsm_cell %x)", p_command[4]);
6029 break;
6030 }
6031 p_chapter = sys.BrowseCodecPrivate( 1, MatchPgcType, &p_type, 1, p_segment );
6032 if ( p_segment != NULL )
6033 {
6034 sys.JumpTo( *p_segment, p_chapter );
6035 f_result = true;
6036 }
6037 break;
6038 case 1:
6039 msg_Dbg( &sys.demuxer, "CallSS VMGM (menu %d, rsm_cell %x)", p_command[5] & 0x0F, p_command[4]);
6040 break;
6041 case 2:
6042 msg_Dbg( &sys.demuxer, "CallSS VTSM (menu %d, rsm_cell %x)", p_command[5] & 0x0F, p_command[4]);
6043 break;
6044 case 3:
6045 msg_Dbg( &sys.demuxer, "CallSS VMGM (pgc %d, rsm_cell %x)", (p_command[2] << 8) + p_command[3], p_command[4]);
6046 break;
6047 }
6048 break;
6049 }
6050 case CMD_DVD_JUMP_SS:
6051 {
6052 msg_Dbg( &sys.demuxer, "JumpSS");
6053 binary p_type;
6054 switch( (p_command[5] & 0xC0) >> 6 ) {
6055 case 0:
6056 msg_Dbg( &sys.demuxer, "JumpSS FP");
6057 break;
6058 case 1:
6059 p_type = p_command[5] & 0x0F;
6060 switch ( p_type )
6061 {
6062 case 0x02:
6063 msg_Dbg( &sys.demuxer, "JumpSS VMGM Title Entry");
6064 break;
6065 case 0x03:
6066 msg_Dbg( &sys.demuxer, "JumpSS VMGM Root Menu");
6067 break;
6068 case 0x04:
6069 msg_Dbg( &sys.demuxer, "JumpSS VMGM Subpicture Menu");
6070 break;
6071 case 0x05:
6072 msg_Dbg( &sys.demuxer, "JumpSS VMGM Audio Menu");
6073 break;
6074 case 0x06:
6075 msg_Dbg( &sys.demuxer, "JumpSS VMGM Angle Menu");
6076 break;
6077 case 0x07:
6078 msg_Dbg( &sys.demuxer, "JumpSS VMGM Chapter Menu");
6079 break;
6080 default:
6081 msg_Dbg( &sys.demuxer, "JumpSS <unknown>");
6082 break;
6083 }
6084 // find the VMG
6085 p_chapter = sys.BrowseCodecPrivate( 1, MatchIsVMG, NULL, 0, p_segment );
6086 if ( p_segment != NULL )
6087 {
6088 p_chapter = p_segment->BrowseCodecPrivate( 1, MatchPgcType, &p_type, 1 );
6089 if ( p_chapter != NULL )
6090 {
6091 sys.JumpTo( *p_segment, p_chapter );
6092 f_result = true;
6093 }
6094 }
6095 break;
6096 case 2:
6097 p_type = p_command[5] & 0x0F;
6098 switch ( p_type )
6099 {
6100 case 0x02:
6101 msg_Dbg( &sys.demuxer, "JumpSS VTSM (vts %d, ttn %d) Title Entry", p_command[4], p_command[3]);
6102 break;
6103 case 0x03:
6104 msg_Dbg( &sys.demuxer, "JumpSS VTSM (vts %d, ttn %d) Root Menu", p_command[4], p_command[3]);
6105 break;
6106 case 0x04:
6107 msg_Dbg( &sys.demuxer, "JumpSS VTSM (vts %d, ttn %d) Subpicture Menu", p_command[4], p_command[3]);
6108 break;
6109 case 0x05:
6110 msg_Dbg( &sys.demuxer, "JumpSS VTSM (vts %d, ttn %d) Audio Menu", p_command[4], p_command[3]);
6111 break;
6112 case 0x06:
6113 msg_Dbg( &sys.demuxer, "JumpSS VTSM (vts %d, ttn %d) Angle Menu", p_command[4], p_command[3]);
6114 break;
6115 case 0x07:
6116 msg_Dbg( &sys.demuxer, "JumpSS VTSM (vts %d, ttn %d) Chapter Menu", p_command[4], p_command[3]);
6117 break;
6118 default:
6119 msg_Dbg( &sys.demuxer, "JumpSS VTSM (vts %d, ttn %d) <unknown>", p_command[4], p_command[3]);
6120 break;
6121 }
6122
6123 p_chapter = sys.BrowseCodecPrivate( 1, MatchVTSMNumber, &p_command[4], 1, p_segment );
6124
6125 if ( p_segment != NULL && p_chapter != NULL )
6126 {
6127 // find the title in the VTS
6128 p_chapter = p_chapter->BrowseCodecPrivate( 1, MatchTitleNumber, &p_command[3], 1 );
6129 if ( p_chapter != NULL )
6130 {
6131 // find the specified menu in the VTSM
6132 p_chapter = p_segment->BrowseCodecPrivate( 1, MatchPgcType, &p_type, 1 );
6133 if ( p_chapter != NULL )
6134 {
6135 sys.JumpTo( *p_segment, p_chapter );
6136 f_result = true;
6137 }
6138 }
6139 else
6140 msg_Dbg( &sys.demuxer, "Title (%d) does not exist in this VTS", p_command[3] );
6141 }
6142 else
6143 msg_Dbg( &sys.demuxer, "DVD Domain VTS (%d) not found", p_command[4] );
6144 break;
6145 case 3:
6146 msg_Dbg( &sys.demuxer, "JumpSS VMGM (pgc %d)", (p_command[2] << 8) + p_command[3]);
6147 break;
6148 }
6149 break;
6150 }
6151 case CMD_DVD_JUMPVTS_PTT:
6152 {
6153 uint8 i_title = p_command[5];
6154 uint8 i_ptt = p_command[3];
6155
6156 msg_Dbg( &sys.demuxer, "JumpVTS Title (%d) PTT (%d)", i_title, i_ptt);
6157
6158 // find the current VTS content segment
6159 p_chapter = sys.p_current_segment->BrowseCodecPrivate( 1, MatchIsDomain, NULL, 0 );
6160 if ( p_chapter != NULL )
6161 {
6162 int16 i_curr_title = p_chapter->GetTitleNumber( );
6163 if ( i_curr_title > 0 )
6164 {
6165 p_chapter = sys.BrowseCodecPrivate( 1, MatchVTSNumber, &i_curr_title, sizeof(i_curr_title), p_segment );
6166
6167 if ( p_segment != NULL && p_chapter != NULL )
6168 {
6169 // find the title in the VTS
6170 p_chapter = p_chapter->BrowseCodecPrivate( 1, MatchTitleNumber, &i_title, sizeof(i_title) );
6171 if ( p_chapter != NULL )
6172 {
6173 // find the chapter in the title
6174 p_chapter = p_chapter->BrowseCodecPrivate( 1, MatchChapterNumber, &i_ptt, sizeof(i_ptt) );
6175 if ( p_chapter != NULL )
6176 {
6177 sys.JumpTo( *p_segment, p_chapter );
6178 f_result = true;
6179 }
6180 }
6181 else
6182 msg_Dbg( &sys.demuxer, "Title (%d) does not exist in this VTS", i_title );
6183 }
6184 else
6185 msg_Dbg( &sys.demuxer, "DVD Domain VTS (%d) not found", i_curr_title );
6186 }
6187 else
6188 msg_Dbg( &sys.demuxer, "JumpVTS_PTT command found but not in a VTS(M)");
6189 }
6190 else
6191 msg_Dbg( &sys.demuxer, "JumpVTS_PTT command but the DVD domain wasn't found");
6192 break;
6193 }
6194 case CMD_DVD_SET_GPRMMD:
6195 {
6196 msg_Dbg( &sys.demuxer, "Set GPRMMD [%d]=%d", (p_command[4] << 8) + p_command[5], (p_command[2] << 8) + p_command[3]);
6197
6198 if ( !SetGPRM( (p_command[4] << 8) + p_command[5], (p_command[2] << 8) + p_command[3] ) )
6199 msg_Dbg( &sys.demuxer, "Set GPRMMD failed" );
6200 break;
6201 }
6202 case CMD_DVD_LINKPGCN:
6203 {
6204 uint16 i_pgcn = (p_command[6] << 8) + p_command[7];
6205
6206 msg_Dbg( &sys.demuxer, "Link PGCN(%d)", i_pgcn );
6207 p_chapter = sys.p_current_segment->BrowseCodecPrivate( 1, MatchPgcNumber, &i_pgcn, 2 );
6208 if ( p_chapter != NULL )
6209 {
6210 if ( !p_chapter->Enter( true ) )
6211 // jump to the location in the found segment
6212 sys.p_current_segment->Seek( sys.demuxer, p_chapter->i_user_start_time, -1, p_chapter );
6213
6214 f_result = true;
6215 }
6216 break;
6217 }
6218 case CMD_DVD_LINKCN:
6219 {
6220 uint8 i_cn = p_command[7];
6221
6222 p_chapter = sys.p_current_segment->CurrentChapter();
6223
6224 msg_Dbg( &sys.demuxer, "LinkCN (cell %d)", i_cn );
6225 p_chapter = p_chapter->BrowseCodecPrivate( 1, MatchCellNumber, &i_cn, 1 );
6226 if ( p_chapter != NULL )
6227 {
6228 if ( !p_chapter->Enter( true ) )
6229 // jump to the location in the found segment
6230 sys.p_current_segment->Seek( sys.demuxer, p_chapter->i_user_start_time, -1, p_chapter );
6231
6232 f_result = true;
6233 }
6234 break;
6235 }
6236 case CMD_DVD_GOTO_LINE:
6237 {
6238 msg_Dbg( &sys.demuxer, "GotoLine (%d)", (p_command[6] << 8) + p_command[7] );
6239 // TODO
6240 break;
6241 }
6242 case CMD_DVD_SET_HL_BTNN1:
6243 {
6244 msg_Dbg( &sys.demuxer, "SetHL_BTN (%d)", p_command[4] );
6245 SetSPRM( 0x88, p_command[4] );
6246 break;
6247 }
6248 default:
6249 {
6250 msg_Dbg( &sys.demuxer, "unsupported command : %02X %02X %02X %02X %02X %02X %02X %02X"
6251 ,p_command[0]
6252 ,p_command[1]
6253 ,p_command[2]
6254 ,p_command[3]
6255 ,p_command[4]
6256 ,p_command[5]
6257 ,p_command[6]
6258 ,p_command[7]);
6259 break;
6260 }
6261 }
6262
6263 return f_result;
6264 }
6265
6266 bool dvd_command_interpretor_c::MatchIsDomain( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size )
6267 {
6268 return ( data.p_private_data != NULL && data.p_private_data->GetBuffer()[0] == MATROSKA_DVD_LEVEL_SS );
6269 }
6270
6271 bool dvd_command_interpretor_c::MatchIsVMG( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size )
6272 {
6273 if ( data.p_private_data == NULL || data.p_private_data->GetSize() < 2 )
6274 return false;
6275
6276 return ( data.p_private_data->GetBuffer()[0] == MATROSKA_DVD_LEVEL_SS && data.p_private_data->GetBuffer()[1] == 0xC0);
6277 }
6278
6279 bool dvd_command_interpretor_c::MatchVTSNumber( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size )
6280 {
6281 if ( i_cookie_size != 2 || data.p_private_data == NULL || data.p_private_data->GetSize() < 4 )
6282 return false;
6283
6284 if ( data.p_private_data->GetBuffer()[0] != MATROSKA_DVD_LEVEL_SS || data.p_private_data->GetBuffer()[1] != 0x80 )
6285 return false;
6286
6287 uint16 i_gtitle = (data.p_private_data->GetBuffer()[2] << 8 ) + data.p_private_data->GetBuffer()[3];
6288 uint16 i_title = *(uint16*)p_cookie;
6289
6290 return (i_gtitle == i_title);
6291 }
6292
6293 bool dvd_command_interpretor_c::MatchVTSMNumber( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size )
6294 {
6295 if ( i_cookie_size != 1 || data.p_private_data == NULL || data.p_private_data->GetSize() < 4 )
6296 return false;
6297
6298 if ( data.p_private_data->GetBuffer()[0] != MATROSKA_DVD_LEVEL_SS || data.p_private_data->GetBuffer()[1] != 0x40 )
6299 return false;
6300
6301 uint8 i_gtitle = data.p_private_data->GetBuffer()[3];
6302 uint8 i_title = *(uint8*)p_cookie;
6303
6304 return (i_gtitle == i_title);
6305 }
6306
6307 bool dvd_command_interpretor_c::MatchTitleNumber( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size )
6308 {
6309 if ( i_cookie_size != 1 || data.p_private_data == NULL || data.p_private_data->GetSize() < 4 )
6310 return false;
6311
6312 if ( data.p_private_data->GetBuffer()[0] != MATROSKA_DVD_LEVEL_TT )
6313 return false;
6314
6315 uint16 i_gtitle = (data.p_private_data->GetBuffer()[1] << 8 ) + data.p_private_data->GetBuffer()[2];
6316 uint8 i_title = *(uint8*)p_cookie;
6317
6318 return (i_gtitle == i_title);
6319 }
6320
6321 bool dvd_command_interpretor_c::MatchPgcType( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size )
6322 {
6323 if ( i_cookie_size != 1 || data.p_private_data == NULL || data.p_private_data->GetSize() < 8 )
6324 return false;
6325
6326 if ( data.p_private_data->GetBuffer()[0] != MATROSKA_DVD_LEVEL_PGC )
6327 return false;
6328
6329 uint8 i_pgc_type = data.p_private_data->GetBuffer()[3] & 0x0F;
6330 uint8 i_pgc = *(uint8*)p_cookie;
6331
6332 return (i_pgc_type == i_pgc);
6333 }
6334
6335 bool dvd_command_interpretor_c::MatchPgcNumber( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size )
6336 {
6337 if ( i_cookie_size != 2 || data.p_private_data == NULL || data.p_private_data->GetSize() < 8 )
6338 return false;
6339
6340 if ( data.p_private_data->GetBuffer()[0] != MATROSKA_DVD_LEVEL_PGC )
6341 return false;
6342
6343 uint16 *i_pgc_n = (uint16 *)p_cookie;
6344 uint16 i_pgc_num = (data.p_private_data->GetBuffer()[1] << 8) + data.p_private_data->GetBuffer()[2];
6345
6346 return (i_pgc_num == *i_pgc_n);
6347 }
6348
6349 bool dvd_command_interpretor_c::MatchChapterNumber( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size )
6350 {
6351 if ( i_cookie_size != 1 || data.p_private_data == NULL || data.p_private_data->GetSize() < 2 )
6352 return false;
6353
6354 if ( data.p_private_data->GetBuffer()[0] != MATROSKA_DVD_LEVEL_PTT )
6355 return false;
6356
6357 uint8 i_chapter = data.p_private_data->GetBuffer()[1];
6358 uint8 i_ptt = *(uint8*)p_cookie;
6359
6360 return (i_chapter == i_ptt);
6361 }
6362
6363 bool dvd_command_interpretor_c::MatchCellNumber( const chapter_codec_cmds_c &data, const void *p_cookie, size_t i_cookie_size )
6364 {
6365 if ( i_cookie_size != 1 || data.p_private_data == NULL || data.p_private_data->GetSize() < 5 )
6366 return false;
6367
6368 if ( data.p_private_data->GetBuffer()[0] != MATROSKA_DVD_LEVEL_CN )
6369 return false;
6370
6371 uint8 *i_cell_n = (uint8 *)p_cookie;
6372 uint8 i_cell_num = data.p_private_data->GetBuffer()[3];
6373
6374 return (i_cell_num == *i_cell_n);
6375 }
6376
6377 bool matroska_script_codec_c::Enter()
6378 {
6379 bool f_result = false;
6380 std::vector<KaxChapterProcessData*>::iterator index = enter_cmds.begin();
6381 while ( index != enter_cmds.end() )
6382 {
6383 if ( (*index)->GetSize() )
6384 {
6385 msg_Dbg( &sys.demuxer, "Matroska Script enter command" );
6386 f_result |= interpretor.Interpret( (*index)->GetBuffer(), (*index)->GetSize() );
6387 }
6388 index++;
6389 }
6390 return f_result;
6391 }
6392
6393 bool matroska_script_codec_c::Leave()
6394 {
6395 bool f_result = false;
6396 std::vector<KaxChapterProcessData*>::iterator index = leave_cmds.begin();
6397 while ( index != leave_cmds.end() )
6398 {
6399 if ( (*index)->GetSize() )
6400 {
6401 msg_Dbg( &sys.demuxer, "Matroska Script leave command" );
6402 f_result |= interpretor.Interpret( (*index)->GetBuffer(), (*index)->GetSize() );
6403 }
6404 index++;
6405 }
6406 return f_result;
6407 }
6408
6409 // see http://www.matroska.org/technical/specs/chapters/index.html#mscript
6410 // for a description of existing commands
6411 bool matroska_script_interpretor_c::Interpret( const binary * p_command, size_t i_size )
6412 {
6413 bool b_result = false;
6414
6415 char *psz_str = (char*) malloc( i_size + 1 );
6416 memcpy( psz_str, p_command, i_size );
6417 psz_str[ i_size ] = '\0';
6418
6419 std::string sz_command = psz_str;
6420 free( psz_str );
6421
6422 msg_Dbg( &sys.demuxer, "command : %s", sz_command.c_str() );
6423
6424 #if defined(__GNUC__) && (__GNUC__ < 3)
6425 if ( sz_command.compare( CMD_MS_GOTO_AND_PLAY, 0, CMD_MS_GOTO_AND_PLAY.size() ) == 0 )
6426 #else
6427 if ( sz_command.compare( 0, CMD_MS_GOTO_AND_PLAY.size(), CMD_MS_GOTO_AND_PLAY ) == 0 )
6428 #endif
6429 {
6430 size_t i,j;
6431
6432 // find the (
6433 for ( i=CMD_MS_GOTO_AND_PLAY.size(); i<sz_command.size(); i++)
6434 {
6435 if ( sz_command[i] == '(' )
6436 {
6437 i++;
6438 break;
6439 }
6440 }
6441 // find the )
6442 for ( j=i; j<sz_command.size(); j++)
6443 {
6444 if ( sz_command[j] == ')' )
6445 {
6446 i--;
6447 break;
6448 }
6449 }
6450
6451 std::string st = sz_command.substr( i+1, j-i-1 );
6452 int64_t i_chapter_uid = atoi( st.c_str() );
6453
6454 virtual_segment_c *p_segment;
6455 chapter_item_c *p_chapter = sys.FindChapter( i_chapter_uid, p_segment );
6456
6457 if ( p_chapter == NULL )
6458 msg_Dbg( &sys.demuxer, "Chapter "I64Fd" not found", i_chapter_uid);
6459 else
6460 {
6461 if ( !p_chapter->EnterAndLeave( sys.p_current_segment->CurrentChapter() ) )
6462 p_segment->Seek( sys.demuxer, p_chapter->i_user_start_time, -1, p_chapter );
6463 b_result = true;
6464 }
6465 }
6466
6467 return b_result;
6468 }
6469
6470 void demux_sys_t::SwapButtons()
6471 {
6472 #ifndef WORDS_BIGENDIAN
6473 uint8_t button, i, j;
6474
6475 for( button = 1; button <= pci_packet.hli.hl_gi.btn_ns; button++) {
6476 btni_t *button_ptr = &(pci_packet.hli.btnit[button-1]);
6477 binary *p_data = (binary*) button_ptr;
6478
6479 uint16 i_x_start = ((p_data[0] & 0x3F) << 4 ) + ( p_data[1] >> 4 );
6480 uint16 i_x_end = ((p_data[1] & 0x03) << 8 ) + p_data[2];
6481 uint16 i_y_start = ((p_data[3] & 0x3F) << 4 ) + ( p_data[4] >> 4 );
6482 uint16 i_y_end = ((p_data[4] & 0x03) << 8 ) + p_data[5];
6483 button_ptr->x_start = i_x_start;
6484 button_ptr->x_end = i_x_end;
6485 button_ptr->y_start = i_y_start;
6486 button_ptr->y_end = i_y_end;
6487
6488 }
6489 for ( i = 0; i<3; i++ )
6490 {
6491 for ( j = 0; j<2; j++ )
6492 {
6493 pci_packet.hli.btn_colit.btn_coli[i][j] = U32_AT( &pci_packet.hli.btn_colit.btn_coli[i][j] );
6494 }
6495 }
6496 #endif
6497 }
VLC Development tree mirror