241 lines
8.3 KiB
C++
241 lines
8.3 KiB
C++
#include "interface.h"
|
|
|
|
Interface::Interface() {
|
|
//Initializing local data
|
|
Width = 0;
|
|
Height = 0;
|
|
Duration = 0;
|
|
UnitsPerSecond = 0;
|
|
//Creating the boxes
|
|
ftyp = new Box_ftyp();
|
|
moov = new Box_moov();
|
|
mvhd = new Box_mvhd();
|
|
trak_vide = new Box_trak();
|
|
tkhd_vide = new Box_tkhd();
|
|
mdia_vide = new Box_mdia();
|
|
mdhd_vide = new Box_mdhd();
|
|
hdlr_vide = new Box_hdlr();
|
|
minf_vide = new Box_minf();
|
|
vmhd_vide = new Box_vmhd();
|
|
dinf_vide = new Box_dinf();
|
|
dref_vide = new Box_dref();
|
|
url_vide = new Box_url();
|
|
stbl_vide = new Box_stbl();
|
|
stts_vide = new Box_stts();
|
|
stsc_vide = new Box_stsc();
|
|
stco_vide = new Box_stco();
|
|
stsd_vide = new Box_stsd();
|
|
avcC_vide = new Box_avcC();
|
|
trak_soun = new Box_trak();
|
|
tkhd_soun = new Box_tkhd();
|
|
mdia_soun = new Box_mdia();
|
|
mdhd_soun = new Box_mdhd();
|
|
hdlr_soun = new Box_hdlr();
|
|
minf_soun = new Box_minf();
|
|
smhd_soun = new Box_smhd();
|
|
dinf_soun = new Box_dinf();
|
|
dref_soun = new Box_dref();
|
|
url_soun = new Box_url();
|
|
stbl_soun = new Box_stbl();
|
|
stts_soun = new Box_stts();
|
|
stsc_soun = new Box_stsc();
|
|
stco_soun = new Box_stco();
|
|
stsd_soun = new Box_stsd();
|
|
esds_soun = new Box_esds();
|
|
//Set some values we already know won't change once the boxes have been created
|
|
SetStaticDefaults();
|
|
}
|
|
|
|
Interface::~Interface() {
|
|
//Deleting the boxes if they still exist.
|
|
if( esds_soun ) { delete esds_soun; esds_soun = NULL; }
|
|
if( stsd_soun ) { delete stsd_soun; stsd_soun = NULL; }
|
|
if( stco_soun ) { delete stco_soun; stco_soun = NULL; }
|
|
if( stsc_soun ) { delete stsc_soun; stsc_soun = NULL; }
|
|
if( stts_soun ) { delete stts_soun; stts_soun = NULL; }
|
|
if( stbl_soun ) { delete stbl_soun; stbl_soun = NULL; }
|
|
if( url_soun ) { delete url_soun; url_soun = NULL; }
|
|
if( dref_soun ) { delete dref_soun; dref_soun = NULL; }
|
|
if( dinf_soun ) { delete dinf_soun; dinf_soun = NULL; }
|
|
if( minf_soun ) { delete minf_soun; minf_soun = NULL; }
|
|
if( hdlr_soun ) { delete hdlr_soun; hdlr_soun = NULL; }
|
|
if( mdhd_soun ) { delete mdhd_soun; mdhd_soun = NULL; }
|
|
if( mdia_soun ) { delete mdia_soun; mdia_soun = NULL; }
|
|
if( tkhd_soun ) { delete tkhd_soun; tkhd_soun = NULL; }
|
|
if( trak_soun ) { delete trak_soun; trak_soun = NULL; }
|
|
if( avcC_vide ) { delete avcC_vide; avcC_vide = NULL; }
|
|
if( stsd_vide ) { delete stsd_vide; stsd_vide = NULL; }
|
|
if( stco_vide ) { delete stco_vide; stco_vide = NULL; }
|
|
if( stsc_vide ) { delete stsc_vide; stsc_vide = NULL; }
|
|
if( stts_vide ) { delete stts_vide; stts_vide = NULL; }
|
|
if( stbl_vide ) { delete stbl_vide; stbl_vide = NULL; }
|
|
if( url_vide ) { delete url_vide; url_vide = NULL; }
|
|
if( dref_vide ) { delete dref_vide; dref_vide = NULL; }
|
|
if( dinf_vide ) { delete dinf_vide; dinf_vide = NULL; }
|
|
if( minf_vide ) { delete minf_vide; minf_vide = NULL; }
|
|
if( hdlr_vide ) { delete hdlr_vide; hdlr_vide = NULL; }
|
|
if( mdhd_vide ) { delete mdhd_vide; mdhd_vide = NULL; }
|
|
if( mdia_vide ) { delete mdia_vide; mdia_vide = NULL; }
|
|
if( tkhd_vide ) { delete tkhd_vide; tkhd_vide = NULL; }
|
|
if( trak_vide ) { delete trak_vide; trak_vide = NULL; }
|
|
if( mvhd ) { delete mvhd; mvhd = NULL; }
|
|
if( moov ) { delete moov; moov = NULL; }
|
|
if( ftyp ) { delete ftyp; ftyp = NULL; }
|
|
}
|
|
|
|
void Interface::link( ) {
|
|
//Linking Video Track
|
|
stsd_vide->AddContent(avcC_vide->GetBox());
|
|
stbl_vide->AddContent(stsd_vide->GetBox(),3);
|
|
stbl_vide->AddContent(stco_vide->GetBox(),2);
|
|
stbl_vide->AddContent(stsc_vide->GetBox(),1);
|
|
stbl_vide->AddContent(stts_vide->GetBox());
|
|
dref_vide->AddContent(url_vide->GetBox());
|
|
dinf_vide->AddContent(dref_vide->GetBox());
|
|
minf_vide->AddContent(stbl_vide->GetBox(),2);
|
|
minf_vide->AddContent(dinf_vide->GetBox(),1);
|
|
minf_vide->AddContent(vmhd_vide->GetBox());
|
|
mdia_vide->AddContent(minf_vide->GetBox(),2);
|
|
mdia_vide->AddContent(hdlr_vide->GetBox(),1);
|
|
mdia_vide->AddContent(mdhd_vide->GetBox());
|
|
trak_vide->AddContent(mdia_vide->GetBox(),1);
|
|
trak_vide->AddContent(tkhd_vide->GetBox());
|
|
|
|
//Linking Sound Track
|
|
stsd_soun->AddContent(esds_soun->GetBox());
|
|
stbl_soun->AddContent(stsd_soun->GetBox(),3);
|
|
stbl_soun->AddContent(stco_soun->GetBox(),2);
|
|
stbl_soun->AddContent(stsc_soun->GetBox(),1);
|
|
stbl_soun->AddContent(stts_soun->GetBox());
|
|
dref_soun->AddContent(url_soun->GetBox());
|
|
dinf_soun->AddContent(dref_soun->GetBox());
|
|
minf_soun->AddContent(stbl_soun->GetBox(),2);
|
|
minf_soun->AddContent(dinf_soun->GetBox(),1);
|
|
minf_soun->AddContent(smhd_soun->GetBox());
|
|
mdia_soun->AddContent(minf_soun->GetBox(),2);
|
|
mdia_soun->AddContent(hdlr_soun->GetBox(),1);
|
|
mdia_soun->AddContent(mdhd_soun->GetBox());
|
|
trak_soun->AddContent(mdia_soun->GetBox(),1);
|
|
trak_soun->AddContent(tkhd_soun->GetBox());
|
|
|
|
//Linking total file
|
|
moov->AddContent(trak_soun->GetBox(),2);
|
|
moov->AddContent(trak_vide->GetBox(),1);
|
|
moov->AddContent(mvhd->GetBox());
|
|
}
|
|
|
|
uint32_t Interface::GetContentSize( ) {
|
|
return ftyp->GetBox( )->GetBoxedDataSize( ) + moov->GetBox( )->GetBoxedDataSize( );
|
|
}
|
|
|
|
uint8_t * Interface::GetContents( ) {
|
|
uint8_t * Result = new uint8_t[GetContentSize( )];
|
|
memcpy(Result,ftyp->GetBox( )->GetBoxedData( ),ftyp->GetBox( )->GetBoxedDataSize( ));
|
|
memcpy(&Result[ftyp->GetBox( )->GetBoxedDataSize( )],moov->GetBox( )->GetBoxedData( ),moov->GetBox( )->GetBoxedDataSize( ));
|
|
return Result;
|
|
}
|
|
|
|
void Interface::UpdateContents( ) {
|
|
if( !Width ) { fprintf(stderr,"WARNING: Width not set!\n"); }
|
|
if( !Height ) { fprintf(stderr,"WARNING: Height not set!\n"); }
|
|
if( !Duration ) { fprintf(stderr,"WARNING: Duration not set!\n"); }
|
|
if( !UnitsPerSecond ) { fprintf(stderr,"WARNING: Timescale not set!\n"); }
|
|
if( stts.size() == 0 ) {
|
|
fprintf(stderr,"WARNING: No stts available!\n");
|
|
} else {
|
|
WriteSTTS();
|
|
}
|
|
stsd_vide->WriteContent( );
|
|
stco_vide->WriteContent( );
|
|
stsc_vide->WriteContent( );
|
|
stts_vide->WriteContent( );
|
|
stbl_vide->WriteContent( );
|
|
dref_vide->WriteContent( );
|
|
dinf_vide->WriteContent( );
|
|
minf_vide->WriteContent( );
|
|
mdia_vide->WriteContent( );
|
|
|
|
stsd_soun->WriteContent( );
|
|
stco_soun->WriteContent( );
|
|
stsc_soun->WriteContent( );
|
|
stts_soun->WriteContent( );
|
|
stbl_soun->WriteContent( );
|
|
dref_soun->WriteContent( );
|
|
dinf_soun->WriteContent( );
|
|
minf_soun->WriteContent( );
|
|
mdia_soun->WriteContent( );
|
|
|
|
trak_vide->WriteContent( );
|
|
|
|
moov->WriteContent( );
|
|
}
|
|
|
|
bool Interface::AllBoxesExist() {
|
|
return ( ftyp && moov && mvhd && trak_vide && tkhd_vide && mdia_vide && mdhd_vide && hdlr_vide &&
|
|
minf_vide && vmhd_vide && dinf_vide && dref_vide && url_vide && stbl_vide && stts_vide && stsc_vide &&
|
|
stco_vide && stsd_vide && avcC_vide && trak_soun && tkhd_soun && mdia_soun && mdhd_soun && hdlr_soun &&
|
|
minf_soun && smhd_soun && dinf_soun && dref_soun && url_soun && stbl_soun && stts_soun && stsc_soun &&
|
|
stco_soun && stsd_soun && esds_soun );
|
|
}
|
|
|
|
void Interface::SetWidth( uint16_t NewWidth ) {
|
|
if( Width != NewWidth ) {
|
|
Width = NewWidth;
|
|
avcC_vide->SetWidth( Width );
|
|
tkhd_vide->SetWidth( Width );
|
|
}
|
|
}
|
|
|
|
void Interface::SetHeight( uint16_t NewHeight ) {
|
|
if( Height != NewHeight ) {
|
|
Height = NewHeight;
|
|
avcC_vide->SetHeight( Height );
|
|
tkhd_vide->SetHeight( Height );
|
|
}
|
|
}
|
|
|
|
void Interface::SetDurationTime( uint32_t NewDuration ) {
|
|
if( Duration != NewDuration ) {
|
|
Duration = NewDuration;
|
|
mvhd->SetDurationTime( Duration );
|
|
mdhd_vide->SetDurationTime( Duration );
|
|
tkhd_vide->SetDurationTime( Duration );
|
|
mdhd_soun->SetDurationTime( Duration );
|
|
tkhd_soun->SetDurationTime( Duration );
|
|
}
|
|
}
|
|
void Interface::SetTimeScale( uint32_t NewUnitsPerSecond ) {
|
|
if( UnitsPerSecond != NewUnitsPerSecond ) {
|
|
UnitsPerSecond = NewUnitsPerSecond;
|
|
mvhd->SetTimeScale( UnitsPerSecond );
|
|
mdhd_vide->SetTimeScale( UnitsPerSecond );
|
|
mdhd_soun->SetTimeScale( UnitsPerSecond );
|
|
}
|
|
}
|
|
|
|
void Interface::SetStaticDefaults() {
|
|
// 'vide' = 0x76696465
|
|
hdlr_vide->SetHandlerType( 0x76696465 );
|
|
hdlr_vide->SetName( "Video Track" );
|
|
// 'soun' = 0x736F756E
|
|
hdlr_soun->SetHandlerType( 0x736F756E );
|
|
hdlr_vide->SetName( "Audio Track" );
|
|
}
|
|
|
|
void Interface::AddSTTSEntry( uint32_t SampleCount, uint32_t SampleDelta ) {
|
|
stts_record temp;
|
|
temp.SampleCount = SampleCount;
|
|
temp.SampleDelta = SampleDelta;
|
|
stts.push_back(temp);
|
|
}
|
|
|
|
void Interface::EmptySTTS() {
|
|
stts.clear();
|
|
}
|
|
|
|
void Interface::WriteSTTS( ) {
|
|
for( int i = stts.size() -1; i > 0; i -- ) {
|
|
stts_vide->AddEntry(stts[i].SampleCount,stts[i].SampleDelta,i);
|
|
stts_soun->AddEntry(stts[i].SampleCount,stts[i].SampleDelta,i);
|
|
}
|
|
}
|