|
|
/**************************************************************************
|
|
|
* Copyright (C) 2005 by Jean-Michel Petit *
|
|
|
* jm_petit@laposte.net *
|
|
|
* *
|
|
|
* This program is free software; you can redistribute it and/or modify *
|
|
|
* it under the terms of the GNU General Public License as published by *
|
|
|
* the Free Software Foundation; either version 2 of the License, or *
|
|
|
* (at your option) any later version. *
|
|
|
* *
|
|
|
* This program is distributed in the hope that it will be useful, *
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
|
* GNU General Public License for more details. *
|
|
|
* *
|
|
|
* You should have received a copy of the GNU General Public License *
|
|
|
* along with this program; if not, write to the *
|
|
|
* Free Software Foundation, Inc., *
|
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. *
|
|
|
***************************************************************************/
|
|
|
#include "k9cell.h"
|
|
|
#include "k9dvdtitleset.h"
|
|
|
|
|
|
int k9VobuList::compareItems ( TQPtrCollection::Item item1, TQPtrCollection::Item item2 ) {
|
|
|
return(((k9Vobu*)item1)->oldSector- ((k9Vobu*)item2)->oldSector);
|
|
|
}
|
|
|
|
|
|
k9Vobu * k9VobuList::findVobu(uint32_t sector) {
|
|
|
//int c=count();
|
|
|
//if (c >0) {
|
|
|
k9Vobu *v1,*v2;
|
|
|
v1=first();
|
|
|
v2=last();
|
|
|
if (v1==NULL)
|
|
|
return NULL;
|
|
|
|
|
|
if (sector >= v1->oldSector && sector <= v2->oldSector)
|
|
|
return findVobu(sector,0,count()-1);
|
|
|
else
|
|
|
return NULL;
|
|
|
//}
|
|
|
}
|
|
|
|
|
|
k9Vobu * k9VobuList::findVobu(uint32_t sector, uint32_t start, uint32_t end) {
|
|
|
|
|
|
long lstart=start;
|
|
|
long lend=end;
|
|
|
|
|
|
|
|
|
while (lstart <=lend) {
|
|
|
long m =(lstart+lend)/2;
|
|
|
k9Vobu *v = (k9Vobu *)at(m);
|
|
|
if ( v->oldSector == sector) {
|
|
|
return v;
|
|
|
} else if ( v->oldSector >sector) {
|
|
|
lend = m-1;
|
|
|
} else {
|
|
|
lstart=m+1;
|
|
|
}
|
|
|
}
|
|
|
return NULL;
|
|
|
}
|
|
|
|
|
|
uint32_t k9Cell::getnewSize() {
|
|
|
uint32_t size=0;
|
|
|
//for (k9Vobu * vobu=vobus.first();vobu;vobu=vobus.next())
|
|
|
for (uint i=0; i<vobus.count();i++) {
|
|
|
k9Vobu *vobu=vobus.at(i);
|
|
|
size+=vobu->size;
|
|
|
}
|
|
|
return size;
|
|
|
}
|
|
|
|
|
|
int k9Cell::getStreamID(int type) {
|
|
|
int abase=0;
|
|
|
if (type >= 0x80 && type <= 0x87) {
|
|
|
// AC3 audio
|
|
|
abase = 0x80;
|
|
|
} else if (type >= 0x88 && type <= 0x8f) {
|
|
|
// DTS audio
|
|
|
abase = 0x88;
|
|
|
} else if (type >= 0xa0 && type <= 0xbf) {
|
|
|
// LPCM audio
|
|
|
abase = 0xa0;
|
|
|
} else if (type >= 0xc0 && type <= 0xdf) {
|
|
|
// MPEG audio
|
|
|
abase = 0xc0;
|
|
|
} else if (type >=0x20 && type <=0x3f) {
|
|
|
//subpicture;
|
|
|
abase =0x20;
|
|
|
}
|
|
|
return (type-abase );
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
streamType_t k9Cell::identifyStream( uchar *buffer,int *packetType ) {
|
|
|
*packetType = buffer[17];
|
|
|
|
|
|
if( (*packetType >= 0xE0) && (*packetType <= 0xEF) ) { // video streams
|
|
|
return stVideo;
|
|
|
} else if( *packetType == 0xBB ) { // system header
|
|
|
return stOther;
|
|
|
} else if( *packetType == 0xBE ) { // padding
|
|
|
return stOther;
|
|
|
} else if( *packetType == 0xBF ) { // nav pack
|
|
|
return stOther;
|
|
|
} else if( (*packetType >= 0xC0) && (*packetType <= 0xDF) ) { // mpeg audio
|
|
|
return stAudio;
|
|
|
} else if( *packetType == 0xBD ) { // private stream, check content
|
|
|
*packetType = buffer[23+buffer[22]];
|
|
|
if (( (*packetType >=0x80) && (*packetType <=0x8f)) || ((*packetType >=0xa0) && (*packetType <=0xa7)) || ((*packetType >=0xc0) && (*packetType <=0xdf)))
|
|
|
return stAudio;
|
|
|
if ( (*packetType >=0x20) && (*packetType <=0x3f))
|
|
|
return stSubpicture;
|
|
|
return stOther;
|
|
|
} else
|
|
|
return stOther;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int k9Cell::isNavPack (uchar *_ptr) {
|
|
|
uint32_t start_code;
|
|
|
uchar *ptr=_ptr;
|
|
|
if ((ptr [0]!=0) || (ptr [1] !=0) || (ptr [2] != 0x01) || (ptr [3] != 0xba))
|
|
|
return 0;
|
|
|
|
|
|
if ((ptr [4] & 0xc0) != 0x40)
|
|
|
return 0;
|
|
|
|
|
|
// ptr += 14;
|
|
|
|
|
|
start_code = (uint32_t) (ptr [14]) << 24;
|
|
|
start_code |= (uint32_t) (ptr [15]) << 16;
|
|
|
start_code |= (uint32_t) (ptr [16]) << 8;
|
|
|
start_code |= (uint32_t) (ptr [17]);
|
|
|
|
|
|
if (start_code != 0x000001bb)
|
|
|
return 0;
|
|
|
|
|
|
// ptr += 24;
|
|
|
|
|
|
start_code = (uint32_t) (ptr [0x26]) << 24;
|
|
|
start_code |= (uint32_t) (ptr [0x27]) << 16;
|
|
|
start_code |= (uint32_t) (ptr [0x28]) << 8;
|
|
|
start_code |= (uint32_t) (ptr [0x29]);
|
|
|
|
|
|
if (start_code != 0x000001bf)
|
|
|
return 0;
|
|
|
|
|
|
// ptr += 986;
|
|
|
|
|
|
start_code = (uint32_t) (ptr [0x400]) << 24;
|
|
|
start_code |= (uint32_t) (ptr [0x401]) << 16;
|
|
|
start_code |= (uint32_t) (ptr [0x402]) << 8;
|
|
|
start_code |= (uint32_t) (ptr [0x403]);
|
|
|
|
|
|
if (start_code != 0x000001bf)
|
|
|
return 0;
|
|
|
|
|
|
return 1;
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
k9Cell::k9Cell(TQObject *parent, const char *name)
|
|
|
: TQObject(parent, name) {
|
|
|
vobus.setAutoDelete(true);
|
|
|
numVobu=0;
|
|
|
|
|
|
vts=0;
|
|
|
pgc=0;
|
|
|
vob=0;
|
|
|
startSector=0;
|
|
|
lastSector=0;
|
|
|
//TO REMOVE oldStartSector=0;
|
|
|
//TO REMOVE oldLastSector=0;
|
|
|
//TO REMOVE newSize=0;
|
|
|
id=0;
|
|
|
selected=false;
|
|
|
copied=false;
|
|
|
//TO REMOVE nbVideoNew=0;
|
|
|
//TO REMOVE nbVideoOld=0;
|
|
|
angleBlock=angleNone;
|
|
|
}
|
|
|
|
|
|
|
|
|
k9Cell::~k9Cell() {}
|
|
|
|
|
|
//add a title in the title list (titles using this cell)
|
|
|
void k9Cell::addTitle( k9DVDTitle *_title) {
|
|
|
m_titles.append( _title);
|
|
|
}
|
|
|
|
|
|
bool k9Cell::getforceFactor() {
|
|
|
bool m_forced=false;
|
|
|
for (uint i=0; i< m_titles.count() && !m_forced ;i++) {
|
|
|
k9DVDTitle *title=m_titles.at(i);
|
|
|
if (title->getforceFactor())
|
|
|
m_forced=true;
|
|
|
}
|
|
|
return m_forced;
|
|
|
}
|
|
|
|
|
|
// returns the shrink factor.
|
|
|
float k9Cell::getFactor() {
|
|
|
float factor=0;
|
|
|
for (uint i=0; i< m_titles.count() ;i++) {
|
|
|
k9DVDTitle *title=m_titles.at(i);
|
|
|
if ((title->getfactor()< factor || factor==0 ) && title->getforceFactor() )
|
|
|
factor=title->getfactor();
|
|
|
}
|
|
|
return factor;
|
|
|
}
|
|
|
|
|
|
|
|
|
//looking for the end reference frame adress block
|
|
|
void k9Cell::addRefStream(k9Vobu *_vobu,uchar *_buffer,uint32_t _position) {
|
|
|
if (_vobu->firstRefOk && _vobu->secondRefOk && _vobu->thirdRefOk)
|
|
|
return;
|
|
|
//offset of frame
|
|
|
uint start=0x16+_buffer[0x16]+1;
|
|
|
//PES Length
|
|
|
uint length=_buffer[0x12];
|
|
|
length=(length <<8 ) + _buffer[0x13];
|
|
|
|
|
|
//end of PES
|
|
|
uint end=0x14+length;
|
|
|
|
|
|
if ( _vobu->frameType!=0x18) {
|
|
|
if (!_vobu->firstRefOk)
|
|
|
_vobu->firstRef=_position;
|
|
|
else if (!_vobu->secondRefOk)
|
|
|
_vobu->secondRef=_position;
|
|
|
else if (!_vobu->thirdRefOk)
|
|
|
_vobu->thirdRef=_position;
|
|
|
}
|
|
|
|
|
|
for (uint i=start;i <end-7;i++) {
|
|
|
//if last picture is not a B-Frame, keep offset of last video
|
|
|
|
|
|
//looking for start code picture
|
|
|
if ((_buffer[i-5] ==0) &&( _buffer[i-4] ==0) &&( _buffer[i-3] ==1)&&( _buffer[i-2] ==0)) {
|
|
|
uchar frameType=_buffer[i] &0x38;
|
|
|
// if frameType != B-Frame and a reference picture was found
|
|
|
if ((_vobu->frameType!=0x18) && (_vobu->frameType!=0)) {
|
|
|
if (!_vobu->firstRefOk) {
|
|
|
_vobu->firstRefOk=true;
|
|
|
_vobu->firstRef=_position;
|
|
|
} else if (!_vobu->secondRefOk) {
|
|
|
_vobu->secondRefOk=true;
|
|
|
_vobu->secondRef=_position;
|
|
|
} else if (!_vobu->thirdRefOk) {
|
|
|
_vobu->thirdRefOk=true;
|
|
|
_vobu->thirdRef=_position;
|
|
|
}
|
|
|
}
|
|
|
//keep the current frame type
|
|
|
_vobu->frameType=frameType;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
k9Vobu * k9Cell::addVobu(uint32_t _sector) {
|
|
|
k9Vobu * vobu = new k9Vobu(this,_sector);
|
|
|
vobus.append(vobu);
|
|
|
return vobu;
|
|
|
}
|
|
|
|
|
|
|
|
|
void k9Cell::addNewVobus(char *_buffer,uint32_t _len,uint32_t _position,int _vobNum,long _vobPos) {
|
|
|
uint32_t start= _position ;//lastSector - _len ;
|
|
|
k9Vobu *vobu;
|
|
|
for (uint32_t i= 0 ; i<_len ;i+=DVD_BLOCK_LEN) {
|
|
|
if (isNavPack((uchar*)_buffer+i)) {
|
|
|
vobu=(k9Vobu*)vobus.at(numVobu);
|
|
|
vobu->newSector=i/DVD_BLOCK_LEN +start;
|
|
|
numVobu++;
|
|
|
vobu->vobNum=_vobNum;
|
|
|
vobu->vobPos=_vobPos;
|
|
|
//TQString c;
|
|
|
//c.sprintf("vobu : %d old: %d new :%d",numVobu-1,vobu->oldSector,vobu->newSector);
|
|
|
//qDebug (c.latin1());
|
|
|
|
|
|
} else {
|
|
|
streamType_t st;
|
|
|
int packetType,id;
|
|
|
st=k9Cell::identifyStream((uchar*)_buffer+i,&packetType);
|
|
|
vobu=(k9Vobu*)vobus.at(numVobu-1);
|
|
|
switch (st) {
|
|
|
case stAudio:
|
|
|
id=k9Cell::getStreamID(packetType);
|
|
|
if (vobu->firstAudio[id]==-1) {
|
|
|
vobu->firstAudio[id]= ((i/ DVD_BLOCK_LEN) + start) - vobu->newSector;
|
|
|
}
|
|
|
break;
|
|
|
case stSubpicture:
|
|
|
id=k9Cell::getStreamID(packetType);
|
|
|
if ((id >=0) && (id<32)) {
|
|
|
if (vobu->firstSubp[id]==-1) {
|
|
|
vobu->firstSubp[id]= ((i / DVD_BLOCK_LEN)+start) - vobu->newSector;
|
|
|
}
|
|
|
}
|
|
|
break;
|
|
|
case stVideo:
|
|
|
addRefStream(vobu,(uchar*)_buffer+i,(i/DVD_BLOCK_LEN +start) - vobu->newSector);
|
|
|
if (vobu->firstVideo==-1) {
|
|
|
vobu->firstVideo = ((i / DVD_BLOCK_LEN)+start) - vobu->newSector;
|
|
|
}
|
|
|
//TO REMOVE nbVideoNew++;
|
|
|
break;
|
|
|
}
|
|
|
|
|
|
}
|
|
|
vobu->size= _position-vobu->newSector;
|
|
|
// <20>v<EFBFBD>ifier
|
|
|
lastSector=_position;
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
k9Vobu::k9Vobu(k9Cell *_parent,uint32_t _oldSector)
|
|
|
: TQObject(_parent,"") {
|
|
|
parent=_parent;
|
|
|
oldSector = _oldSector;
|
|
|
newSector = 0;
|
|
|
size=0;
|
|
|
for (int i=0;i<8;i++)
|
|
|
firstAudio[i]=-1;
|
|
|
for (int i=0;i<32;i++)
|
|
|
firstSubp[i]=-1;
|
|
|
firstVideo=-1;
|
|
|
empty=false;
|
|
|
vobPos=0;
|
|
|
vobNum=0;
|
|
|
frameType=0;
|
|
|
firstRef=0;
|
|
|
secondRef=0;
|
|
|
thirdRef=0;
|
|
|
firstRefOk=false;
|
|
|
secondRefOk=false;
|
|
|
thirdRefOk=false;
|
|
|
}
|
|
|
|
|
|
|
|
|
/*!
|
|
|
\fn k9Cell::findVobu(int _oldSector)
|
|
|
*/
|
|
|
k9Vobu * k9Cell::findVobu(uint32_t _oldSector) {
|
|
|
return vobus.findVobu(_oldSector) ;
|
|
|
}
|
|
|
|
|
|
|
|
|
k9Vobu::~k9Vobu() {}
|
|
|
|
|
|
|
|
|
k9Cell* k9CellList::addCell(int _vts,int _pgc, int _vob) {
|
|
|
k9Cell* cell=new k9Cell;
|
|
|
cell->cellList=this;
|
|
|
cell->vts=_vts;
|
|
|
cell->vob=_vob;
|
|
|
cell->pgc=_pgc;
|
|
|
|
|
|
if (count()!=0) {
|
|
|
k9Cell *prev=(k9Cell*)getLast();
|
|
|
if (prev->vts==_vts) {
|
|
|
cell->startSector= prev->lastSector +1;
|
|
|
if (_pgc!=prev->pgc)
|
|
|
cell->id=1;
|
|
|
else
|
|
|
cell->id=prev->id+1;
|
|
|
}
|
|
|
} else {
|
|
|
cell->startSector = 0;
|
|
|
cell->id=1;
|
|
|
}
|
|
|
append (cell);
|
|
|
return(cell);
|
|
|
|
|
|
}
|
|
|
|
|
|
k9Vobu *k9CellList::findVobu(uint32_t sector) {
|
|
|
k9Vobu *vobu=NULL;
|
|
|
k9Cell *cell=findCell(sector);
|
|
|
if (cell==NULL)
|
|
|
return NULL;
|
|
|
else
|
|
|
return cell->findVobu(sector);
|
|
|
|
|
|
int nbCells=count();
|
|
|
for (int i=0;i <nbCells;i++) {
|
|
|
k9Cell * cell = (k9Cell*)at(i);
|
|
|
vobu = cell->findVobu(sector);
|
|
|
if (vobu !=NULL) {
|
|
|
return vobu;
|
|
|
}
|
|
|
cell = (k9Cell*)at(nbCells-1-i);
|
|
|
vobu = cell->findVobu(sector);
|
|
|
if (vobu !=NULL) {
|
|
|
return vobu;
|
|
|
}
|
|
|
|
|
|
|
|
|
}
|
|
|
return vobu;
|
|
|
}
|
|
|
|
|
|
k9Cell * k9CellList::findCell(uint32_t sector) {
|
|
|
return findCell(sector,0,count()-1);
|
|
|
}
|
|
|
|
|
|
k9Cell * k9CellList::findCell(uint32_t sector, uint32_t start, uint32_t end) {
|
|
|
|
|
|
long lstart=start;
|
|
|
long lend=end;
|
|
|
|
|
|
while (lstart <=lend) {
|
|
|
long m =(lstart+lend)/2;
|
|
|
k9Cell *cell = (k9Cell *)at(m);
|
|
|
k9Vobu *fv=cell->vobus.first();
|
|
|
k9Vobu *lv=cell->vobus.last();
|
|
|
|
|
|
|
|
|
if ( sector >=fv->oldSector && sector<= lv->oldSector) {
|
|
|
return cell;
|
|
|
} else if ( fv->oldSector >sector) {
|
|
|
lend = m-1;
|
|
|
} else {
|
|
|
lstart=m+1;
|
|
|
}
|
|
|
}
|
|
|
return NULL;
|
|
|
}
|
|
|
|
|
|
k9DVDTitleset * k9Cell::getTitleset() {
|
|
|
k9DVDTitle *title=m_titles.at(0);
|
|
|
return title->gettitleset();
|
|
|
}
|
|
|
|
|
|
|
|
|
#include "k9cell.moc"
|
