Skip to content

V
vlc-gpu
Commit 336db7b9 authored by Steve Lhomme's avatar Steve Lhomme
Browse files
Options

mkv.cpp: keep the interresting segments somewhere in memory (not functional yet)

parent a778a891
Show whitespace changes
Inline Side-by-side
Showing with 65 additions and 41 deletions
modules/demux/mkv.cpp
View file @ 336db7b9
...@@ -362,10 +362,12 @@ public: ...@@ -362,10 +362,12 @@ public:
bool b_ordered; bool b_ordered;
}; };
class demux_sys_t;
class matroska_segment_t class matroska_segment_t
{ {
public: public:
matroska_segment_t() matroska_segment_t( demux_sys_t *p_demuxer )
:segment(NULL) :segment(NULL)
,i_timescale(0) ,i_timescale(0)
,f_duration(0.0) ,f_duration(0.0)
...@@ -386,8 +388,15 @@ public: ...@@ -386,8 +388,15 @@ public:
,psz_date_utc(NULL) ,psz_date_utc(NULL)
,i_current_edition(0) ,i_current_edition(0)
,psz_current_chapter(NULL) ,psz_current_chapter(NULL)
,p_sys(p_demuxer)
,ep(NULL)
{} {}
~matroska_segment_t()
{
delete ep;
}
KaxSegment *segment; KaxSegment *segment;
/* time scale */ /* time scale */
...@@ -406,7 +415,7 @@ public: ...@@ -406,7 +415,7 @@ public:
int64_t i_tags_position; int64_t i_tags_position;
KaxCluster *cluster; KaxCluster *cluster;
KaxSegmentUID segment_uid; KaxSegmentUID segment_uid, prev_segment_uid, next_segment_uid;
vlc_bool_t b_cues; vlc_bool_t b_cues;
int i_index; int i_index;
...@@ -426,7 +435,10 @@ public: ...@@ -426,7 +435,10 @@ public:
std::vector<KaxSegmentFamily> families; std::vector<KaxSegmentFamily> families;
chapter_edition_t *Edition() demux_sys_t *p_sys;
EbmlParser *ep;
inline chapter_edition_t *Edition()
{ {
if ( i_current_edition != -1 && i_current_edition < editions.size() ) if ( i_current_edition != -1 && i_current_edition < editions.size() )
return &editions[i_current_edition]; return &editions[i_current_edition];
...@@ -437,38 +449,37 @@ public: ...@@ -437,38 +449,37 @@ public:
class matroska_stream_t class matroska_stream_t
{ {
public: public:
matroska_stream_t() matroska_stream_t( demux_sys_t *p_demuxer )
:in(NULL) :in(NULL)
,es(NULL) ,es(NULL)
,ep(NULL)
,i_current_segment(-1) ,i_current_segment(-1)
,p_sys(p_demuxer)
{} {}
~matroska_stream_t() ~matroska_stream_t()
{ {
for ( int i=0; i<segments.size(); i++ ) for ( int i=0; i<segments.size(); i++ )
delete segments[i]; delete segments[i];
if ( in )
delete in; delete in;
if ( es )
delete es; delete es;
if ( ep )
delete ep;
} }
vlc_stream_io_callback *in; vlc_stream_io_callback *in;
EbmlStream *es; EbmlStream *es;
EbmlParser *ep;
std::vector<matroska_segment_t*> segments; std::vector<matroska_segment_t*> segments;
int i_current_segment; int i_current_segment;
matroska_segment_t *Segment() demux_sys_t *p_sys;
inline matroska_segment_t *Segment()
{ {
if ( i_current_segment != -1 && i_current_segment < segments.size() ) if ( i_current_segment != -1 && i_current_segment < segments.size() )
return segments[i_current_segment]; return segments[i_current_segment];
return NULL; return NULL;
} }
matroska_segment_t *FindSegment( KaxSegmentUID & i_uid ) const;
}; };
class demux_sys_t class demux_sys_t
...@@ -501,12 +512,14 @@ public: ...@@ -501,12 +512,14 @@ public:
std::vector<matroska_stream_t*> streams; std::vector<matroska_stream_t*> streams;
int i_current_stream; int i_current_stream;
matroska_stream_t *Stream() inline matroska_stream_t *Stream()
{ {
if ( i_current_stream != -1 && i_current_stream < streams.size() ) if ( i_current_stream != -1 && i_current_stream < streams.size() )
return streams[i_current_stream]; return streams[i_current_stream];
return NULL; return NULL;
} }
matroska_segment_t *FindSegment( KaxSegmentUID & i_uid ) const;
}; };
static int Demux ( demux_t * ); static int Demux ( demux_t * );
...@@ -557,8 +570,8 @@ static int Open( vlc_object_t * p_this ) ...@@ -557,8 +570,8 @@ static int Open( vlc_object_t * p_this )
p_demux->pf_control = Control; p_demux->pf_control = Control;
p_demux->p_sys = p_sys = new demux_sys_t; p_demux->p_sys = p_sys = new demux_sys_t;
p_stream = new matroska_stream_t; p_stream = new matroska_stream_t( p_sys );
p_segment = new matroska_segment_t; p_segment = new matroska_segment_t( p_sys );
p_sys->streams.push_back( p_stream ); p_sys->streams.push_back( p_stream );
p_sys->i_current_stream = 0; p_sys->i_current_stream = 0;
...@@ -620,9 +633,9 @@ static int Open( vlc_object_t * p_this ) ...@@ -620,9 +633,9 @@ static int Open( vlc_object_t * p_this )
p_segment->segment = (KaxSegment*)el; p_segment->segment = (KaxSegment*)el;
p_segment->cluster = NULL; p_segment->cluster = NULL;
p_stream->ep = new EbmlParser( p_stream->es, el ); p_segment->ep = new EbmlParser( p_stream->es, el );
while( ( el1 = p_stream->ep->Get() ) != NULL ) while( ( el1 = p_segment->ep->Get() ) != NULL )
{ {
if( MKV_IS_ID( el1, KaxInfo ) ) if( MKV_IS_ID( el1, KaxInfo ) )
{ {
...@@ -646,7 +659,7 @@ static int Open( vlc_object_t * p_this ) ...@@ -646,7 +659,7 @@ static int Open( vlc_object_t * p_this )
p_segment->cluster = (KaxCluster*)el1; p_segment->cluster = (KaxCluster*)el1;
p_stream->ep->Down(); p_segment->ep->Down();
/* stop parsing the stream */ /* stop parsing the stream */
break; break;
} }
...@@ -713,26 +726,29 @@ static int Open( vlc_object_t * p_this ) ...@@ -713,26 +726,29 @@ static int Open( vlc_object_t * p_this )
// test wether this file belongs to the our family // test wether this file belongs to the our family
bool b_keep_file_opened = false; bool b_keep_file_opened = false;
StdIOCallback *p_file_io = new StdIOCallback(s_filename.c_str(), MODE_READ); StdIOCallback *p_file_io = new StdIOCallback(s_filename.c_str(), MODE_READ);
EbmlStream *p_stream = new EbmlStream(*p_file_io); EbmlStream *p_estream = new EbmlStream(*p_file_io);
EbmlElement *p_l0, *p_l1, *p_l2; EbmlElement *p_l0, *p_l1, *p_l2;
// verify the EBML Header // verify the EBML Header
p_l0 = p_stream->FindNextID(EbmlHead::ClassInfos, 0xFFFFFFFFL); p_l0 = p_estream->FindNextID(EbmlHead::ClassInfos, 0xFFFFFFFFL);
if (p_l0 == NULL) if (p_l0 == NULL)
{ {
delete p_stream; delete p_estream;
delete p_file_io; delete p_file_io;
continue; continue;
} }
p_l0->SkipData(*p_stream, EbmlHead_Context); matroska_stream_t *p_stream1 = new matroska_stream_t( p_sys );
p_sys->streams.push_back( p_stream1 );
p_l0->SkipData(*p_estream, EbmlHead_Context);
delete p_l0; delete p_l0;
// find all segments in this file // find all segments in this file
p_l0 = p_stream->FindNextID(KaxSegment::ClassInfos, 0xFFFFFFFFL); p_l0 = p_estream->FindNextID(KaxSegment::ClassInfos, 0xFFFFFFFFL);
if (p_l0 == NULL) if (p_l0 == NULL)
{ {
delete p_stream; delete p_estream;
delete p_file_io; delete p_file_io;
continue; continue;
} }
...@@ -745,8 +761,13 @@ static int Open( vlc_object_t * p_this ) ...@@ -745,8 +761,13 @@ static int Open( vlc_object_t * p_this )
{ {
EbmlParser *ep; EbmlParser *ep;
KaxSegmentUID *p_uid = NULL; KaxSegmentUID *p_uid = NULL;
matroska_segment_t *p_segment1 = new matroska_segment_t( p_sys );
p_stream1->segments.push_back( p_segment1 );
ep = new EbmlParser(p_estream, p_l0);
p_segment1->ep = ep;
ep = new EbmlParser(p_stream, p_l0);
bool b_this_segment_matches = false; bool b_this_segment_matches = false;
while ((p_l1 = ep->Get())) while ((p_l1 = ep->Get()))
{ {
...@@ -755,7 +776,7 @@ static int Open( vlc_object_t * p_this ) ...@@ -755,7 +776,7 @@ static int Open( vlc_object_t * p_this )
// find the families of this segment // find the families of this segment
KaxInfo *p_info = static_cast<KaxInfo*>(p_l1); KaxInfo *p_info = static_cast<KaxInfo*>(p_l1);
p_info->Read(*p_stream, KaxInfo::ClassInfos.Context, i_upper_lvl, p_l2, true); p_info->Read(*p_estream, KaxInfo::ClassInfos.Context, i_upper_lvl, p_l2, true);
for( i = 0; i < p_info->ListSize() && !b_this_segment_matches; i++ ) for( i = 0; i < p_info->ListSize() && !b_this_segment_matches; i++ )
{ {
EbmlElement *l = (*p_info)[i]; EbmlElement *l = (*p_info)[i];
...@@ -765,21 +786,24 @@ static int Open( vlc_object_t * p_this ) ...@@ -765,21 +786,24 @@ static int Open( vlc_object_t * p_this )
p_uid = static_cast<KaxSegmentUID*>(l); p_uid = static_cast<KaxSegmentUID*>(l);
if (p_segment->segment_uid == *p_uid) if (p_segment->segment_uid == *p_uid)
break; break;
p_segment1->segment_uid = *p_uid;
} }
else if( MKV_IS_ID( l, KaxSegmentFamily ) ) else if( MKV_IS_ID( l, KaxSegmentFamily ) )
{ {
KaxSegmentFamily *p_fam = static_cast<KaxSegmentFamily*>(l); KaxSegmentFamily *p_fam = static_cast<KaxSegmentFamily*>(l);
std::vector<KaxSegmentFamily>::iterator iter; std::vector<KaxSegmentFamily>::iterator iter;
for( iter = p_segment->families.begin(); p_segment1->families.push_back( *p_fam );
/* for( iter = p_segment->families.begin();
iter != p_segment->families.end(); iter != p_segment->families.end();
iter++ ) iter++ )
{ {
if( *iter == *p_fam ) if( *iter == *p_fam )
{ {
b_this_segment_matches = true; b_this_segment_matches = true;
p_segment1->families.push_back( *p_fam );
break; break;
} }
} }*/
} }
} }
break; break;
...@@ -792,14 +816,14 @@ static int Open( vlc_object_t * p_this ) ...@@ -792,14 +816,14 @@ static int Open( vlc_object_t * p_this )
} }
} }
p_l0->SkipData(*p_stream, EbmlHead_Context); p_l0->SkipData(*p_estream, EbmlHead_Context);
delete p_l0; delete p_l0;
p_l0 = p_stream->FindNextID(KaxSegment::ClassInfos, 0xFFFFFFFFL); p_l0 = p_estream->FindNextID(KaxSegment::ClassInfos, 0xFFFFFFFFL);
} }
if (!b_keep_file_opened) if (!b_keep_file_opened)
{ {
delete p_stream; delete p_estream;
delete p_file_io; delete p_file_io;
} }
} }
...@@ -1287,8 +1311,8 @@ static int BlockGet( demux_t *p_demux, KaxBlock **pp_block, int64_t *pi_ref1, in ...@@ -1287,8 +1311,8 @@ static int BlockGet( demux_t *p_demux, KaxBlock **pp_block, int64_t *pi_ref1, in
return VLC_EGENERIC; return VLC_EGENERIC;
} }
el = p_stream->ep->Get(); el = p_segment->ep->Get();
i_level = p_stream->ep->GetLevel(); i_level = p_segment->ep->GetLevel();
if( el == NULL && *pp_block != NULL ) if( el == NULL && *pp_block != NULL )
{ {
...@@ -1305,9 +1329,9 @@ static int BlockGet( demux_t *p_demux, KaxBlock **pp_block, int64_t *pi_ref1, in ...@@ -1305,9 +1329,9 @@ static int BlockGet( demux_t *p_demux, KaxBlock **pp_block, int64_t *pi_ref1, in
if( el == NULL ) if( el == NULL )
{ {
if( p_stream->ep->GetLevel() > 1 ) if( p_segment->ep->GetLevel() > 1 )
{ {
p_stream->ep->Up(); p_segment->ep->Up();
continue; continue;
} }
msg_Warn( p_demux, "EOF" ); msg_Warn( p_demux, "EOF" );
...@@ -1328,7 +1352,7 @@ static int BlockGet( demux_t *p_demux, KaxBlock **pp_block, int64_t *pi_ref1, in ...@@ -1328,7 +1352,7 @@ static int BlockGet( demux_t *p_demux, KaxBlock **pp_block, int64_t *pi_ref1, in
IndexAppendCluster( p_demux, p_segment->cluster ); IndexAppendCluster( p_demux, p_segment->cluster );
} }
p_stream->ep->Down(); p_segment->ep->Down();
} }
else if( MKV_IS_ID( el, KaxCues ) ) else if( MKV_IS_ID( el, KaxCues ) )
{ {
...@@ -1351,7 +1375,7 @@ static int BlockGet( demux_t *p_demux, KaxBlock **pp_block, int64_t *pi_ref1, in ...@@ -1351,7 +1375,7 @@ static int BlockGet( demux_t *p_demux, KaxBlock **pp_block, int64_t *pi_ref1, in
} }
else if( MKV_IS_ID( el, KaxBlockGroup ) ) else if( MKV_IS_ID( el, KaxBlockGroup ) )
{ {
p_stream->ep->Down(); p_segment->ep->Down();
} }
} }
else if( i_level == 3 ) else if( i_level == 3 )
...@@ -1363,7 +1387,7 @@ static int BlockGet( demux_t *p_demux, KaxBlock **pp_block, int64_t *pi_ref1, in ...@@ -1363,7 +1387,7 @@ static int BlockGet( demux_t *p_demux, KaxBlock **pp_block, int64_t *pi_ref1, in
(*pp_block)->ReadData( p_stream->es->I_O() ); (*pp_block)->ReadData( p_stream->es->I_O() );
(*pp_block)->SetParent( *p_segment->cluster ); (*pp_block)->SetParent( *p_segment->cluster );
p_stream->ep->Keep(); p_segment->ep->Keep();
} }
else if( MKV_IS_ID( el, KaxBlockDuration ) ) else if( MKV_IS_ID( el, KaxBlockDuration ) )
{ {
...@@ -1567,8 +1591,8 @@ static void Seek( demux_t *p_demux, mtime_t i_date, double f_percent, const chap ...@@ -1567,8 +1591,8 @@ static void Seek( demux_t *p_demux, mtime_t i_date, double f_percent, const chap
return; return;
} }
delete p_stream->ep; delete p_segment->ep;
p_stream->ep = new EbmlParser( p_stream->es, p_segment->segment ); p_segment->ep = new EbmlParser( p_stream->es, p_segment->segment );
p_segment->cluster = NULL; p_segment->cluster = NULL;
/* seek without index or without date */ /* seek without index or without date */
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment