abst box rewritten

This commit is contained in:
Erik Zandvliet 2012-09-17 16:44:57 +02:00 committed by Thulinma
parent 07faa64075
commit 42ab565381
2 changed files with 316 additions and 234 deletions

View file

@ -8,17 +8,18 @@
namespace MP4{
Box::Box(size_t size) {
isUpdated = false;
data.resize( size + 8 );
memset( (char*)data.c_str(), 0, size + 8 );
}
Box::Box( char* BoxType, size_t size ) {
Box::Box( const char* BoxType, size_t size ) {
isUpdated = false;
data.resize( size + 8 );
memset( (char*)data.c_str(), 0, size + 8 );
memcpy( (char*)data.c_str() + 4, BoxType, 4 );
}
Box::Box( std::string & newData ) {
isUpdated = false;
if( !read( newData ) ) {
clear();
}
@ -55,6 +56,9 @@ namespace MP4{
}
std::string & Box::asBox() {
if( isUpdated ) {
regenerate( );
}
((int*)data.c_str())[0] = htonl( data.size() );
return data;
}
@ -75,6 +79,14 @@ namespace MP4{
data[index] = newData;
}
char Box::getInt8( size_t index ) {
index += 8;
if( index > data.size() ) {
data.resize( index );
}
return data[index];
}
void Box::setInt16( short newData, size_t index ) {
index += 8;
if( index + 1 > data.size() ) {
@ -84,6 +96,39 @@ namespace MP4{
memcpy( (char*)data.c_str() + index, (char*)newData, 2 );
}
short Box::getInt16( size_t index ) {
index += 8;
if( index + 1 > data.size() ) {
data.resize( index + 1 );
}
short result;
memcpy( (char*)result, (char*)data.c_str() + index, 2 );
return ntohs(result);
}
void Box::setInt24( long newData, size_t index ) {
index += 8;
if( index + 2 > data.size() ) {
data.resize( index + 2 );
}
data[index] = (newData & 0x00FF0000) >> 16;
data[index+1] = (newData & 0x0000FF00) >> 8;
data[index+2] = (newData & 0x000000FF);
}
long Box::getInt24( size_t index ) {
index += 8;
if( index + 2 > data.size() ) {
data.resize( index + 2 );
}
long result = data[index];
result <<= 8;
result += data[index+1];
result <<= 8;
result += data[index+2];
return result;
}
void Box::setInt32( long newData, size_t index ) {
index += 8;
if( index + 3 > data.size() ) {
@ -93,6 +138,17 @@ namespace MP4{
memcpy( (char*)data.c_str() + index, (char*)newData, 4 );
}
long Box::getInt32( size_t index ) {
index += 8;
if( index + 3 > data.size() ) {
data.resize( index + 3 );
}
long result;
memcpy( (char*)result, (char*)data.c_str() + index, 4 );
return ntohl(result);
}
void Box::setInt64( long long int newData, size_t index ) {
index += 8;
if( index + 7 > data.size() ) {
@ -108,6 +164,30 @@ namespace MP4{
data[index+7] = ( newData * 0x00000000000000FF );
}
long long int Box::getInt64( size_t index ) {
index += 8;
if( index + 7 > data.size() ) {
data.resize( index + 7 );
}
long result = data[index];
result <<= 8;
result += data[index+1];
result <<= 8;
result += data[index+2];
result <<= 8;
result += data[index+3];
result <<= 8;
result += data[index+4];
result <<= 8;
result += data[index+5];
result <<= 8;
result += data[index+6];
result <<= 8;
result += data[index+7];
return result;
}
void Box::setString(std::string newData, size_t index ) {
setString( (char*)newData.c_str(), newData.size(), index );
}
@ -120,198 +200,219 @@ namespace MP4{
memcpy( (char*)data.c_str() + index, newData, size );
}
/*
void ABST::SetBootstrapVersion( uint32_t Version ) {
curBootstrapInfoVersion = Version;
void Box::regenerate() {
std::cerr << "Regenerate() not implemented for this box type\n";
}
void ABST::SetProfile( uint8_t Profile ) {
curProfile = Profile;
ABST::ABST( ) : Box("abst") {
setVersion( 0 );
setFlags( 0 );
setBootstrapinfoVersion( 0 );
setProfile( 0 );
setLive( 1 );
setUpdate( 0 );
setTimeScale( 1000 );
setCurrentMediaTime( 0 );
setSmpteTimeCodeOffset( 0 );
setMovieIdentifier( "" );
setDrmData( "" );
setMetaData( "" );
}
void ABST::SetLive( bool Live ) {
isLive = Live;
void ABST::setVersion( char newVersion ) {
setInt8( newVersion, 0 );
}
void ABST::SetUpdate( bool Update ) {
isUpdate = Update;
void ABST::setFlags( long newFlags ) {
setInt24( newFlags, 1 );
}
void ABST::SetTimeScale( uint32_t Scale ) {
curTimeScale = Scale;
void ABST::setBootstrapinfoVersion( long newVersion ) {
setInt32( newVersion, 4 );
}
void ABST::SetMediaTime( uint32_t Time ) {
curMediatime = Time;
void ABST::setProfile( char newProfile ) {
setInt8( ( getInt8(5) & 0x3F ) + ( ( newProfile & 0x02 ) << 6 ), 5 );
}
void ABST::SetSMPTE( uint32_t Smpte ) {
curSMPTE = Smpte;
void ABST::setLive( char newLive ) {
setInt8( ( getInt8(5) & 0xDF ) + ( ( newLive & 0x01 ) << 5 ), 5 );
}
void ABST::SetMovieIdentifier( std::string Identifier ) {
curMovieIdentifier = Identifier;
void ABST::setUpdate( char newUpdate ) {
setInt8( ( getInt8(5) & 0xEF ) + ( ( newUpdate & 0x01 ) << 4 ), 5 );
}
void ABST::SetDRM( std::string Drm ) {
curDRM = Drm;
void ABST::setTimeScale( long newScale ) {
setInt32( newScale, 6 );
}
void ABST::SetMetaData( std::string MetaData ) {
curMetaData = MetaData;
void ABST::setCurrentMediaTime( long long int newTime ) {
setInt64( newTime, 10 );
}
void ABST::AddServerEntry( std::string Url, uint32_t Offset ) {
if(Offset >= Servers.size()) {
Servers.resize(Offset+1);
}
Servers[Offset].ServerBaseUrl = Url;
void ABST::setSmpteTimeCodeOffset( long long int newTime ) {
setInt64( newTime, 18 );
}
void ABST::AddQualityEntry( std::string Quality, uint32_t Offset ) {
if(Offset >= Qualities.size()) {
Qualities.resize(Offset+1);
}
Qualities[Offset].QualityModifier = Quality;
void ABST::setMovieIdentifier( std::string newIdentifier ) {
movieIdentifier = newIdentifier;
}
void ABST::AddSegmentRunTable( Box * newSegment, uint32_t Offset ) {
if( Offset >= SegmentRunTables.size() ) {
SegmentRunTables.resize(Offset+1);
void ABST::addServerEntry( std::string newEntry ) {
if( std::find( Servers.begin(), Servers.end(), newEntry ) == Servers.end() ) {
Servers.push_back( newEntry );
isUpdated = true;
}
if( SegmentRunTables[Offset] ) {
delete SegmentRunTables[Offset];
}
SegmentRunTables[Offset] = newSegment;
}
void ABST::AddFragmentRunTable( Box * newFragment, uint32_t Offset ) {
if( Offset >= FragmentRunTables.size() ) {
FragmentRunTables.resize(Offset+1);
void ABST::delServerEntry( std::string delEntry ) {
for( std::deque<std::string>::iterator it = Servers.begin(); it != Servers.end(); it++ ) {
if( (*it) == delEntry ) {
Servers.erase( it );
isUpdated = true;
break;
}
if( FragmentRunTables[Offset] ) {
delete FragmentRunTables[Offset];
}
FragmentRunTables[Offset] = newFragment;
}
void ABST::SetDefaults( ) {
SetProfile( );
SetLive( );
SetUpdate( );
SetTimeScale( );
SetMediaTime( );
SetSMPTE( );
SetMovieIdentifier( );
SetDRM( );
SetMetaData( );
SetVersion( );
void ABST::addQualityEntry( std::string newEntry ) {
if( std::find( Qualities.begin(), Qualities.end(), newEntry ) == Qualities.end() ) {
Servers.push_back( newEntry );
isUpdated = true;
}
}
void ABST::SetVersion( bool NewVersion) {
Version = NewVersion;
void ABST::delQualityEntry( std::string delEntry ) {
for( std::deque<std::string>::iterator it = Qualities.begin(); it != Qualities.end(); it++ ) {
if( (*it) == delEntry ) {
Qualities.erase( it );
isUpdated = true;
break;
}
}
}
void ABST::SetReserved( ) {
SetInt32(0);
void ABST::setDrmData( std::string newDrm ) {
drmData = newDrm;
isUpdated = true;
}
void ABST::WriteContent( ) {
Box * current;
std::string serializedServers = "";
std::string serializedQualities = "";
std::string serializedSegments = "";
std::string serializedFragments = "";
int SegmentAmount = 0;
int FragmentAmount = 0;
uint8_t * temp = new uint8_t[1];
void ABST::setMetaData( std::string newMetaData ) {
metaData = newMetaData;
isUpdated = true;
}
clear( );
SetReserved( );
void ABST::addSegmentRunTable( Box * newSegment ) {
segmentTables.push_back(newSegment);
isUpdated = true;
}
for( uint32_t i = 0; i < Servers.size(); i++ ) {
serializedServers.append(Servers[i].ServerBaseUrl.c_str());
serializedServers += '\0';
void ABST::addFragmentRunTable( Box * newFragment ) {
fragmentTables.push_back(newFragment);
isUpdated = true;
}
for( uint32_t i = 0; i < Qualities.size(); i++ ) {
serializedQualities.append(Qualities[i].QualityModifier.c_str());
serializedQualities += '\0';
}
for( uint32_t i = 0; i < SegmentRunTables.size(); i++ ) {
current=SegmentRunTables[i];
if( current ) {
SegmentAmount ++;
serializedSegments.append((char*)current->GetBoxedData(),current->GetBoxedDataSize());
}
}
for( uint32_t i = 0; i < FragmentRunTables.size(); i++ ) {
current=FragmentRunTables[i];
if( current ) {
FragmentAmount ++;
serializedFragments.append((char*)current->GetBoxedData(),current->GetBoxedDataSize());
}
}
uint32_t OffsetServerEntryCount = 29 + curMovieIdentifier.size() + 1;
uint32_t OffsetQualityEntryCount = OffsetServerEntryCount + 1 + serializedServers.size();
uint32_t OffsetDrmData = OffsetQualityEntryCount + 1 + serializedQualities.size();
uint32_t OffsetMetaData = OffsetDrmData + curDRM.size() + 1;
uint32_t OffsetSegmentRuntableCount = OffsetMetaData + curMetaData.size() + 1;
uint32_t OffsetFragmentRuntableCount = OffsetSegmentRuntableCount + 1 + serializedSegments.size();
temp[0] = 0 + ( curProfile << 6 ) + ( (uint8_t)isLive << 7 ) + ( (uint8_t)isUpdate << 7 );
SetPayload((uint32_t)serializedFragments.size(),(uint8_t*)serializedFragments.c_str(),OffsetFragmentRuntableCount+1);
SetPayload((uint32_t)1,Box::uint8_to_uint8(FragmentAmount),OffsetFragmentRuntableCount);
SetPayload((uint32_t)serializedSegments.size(),(uint8_t*)serializedSegments.c_str(),OffsetSegmentRuntableCount+1);
SetPayload((uint32_t)1,Box::uint8_to_uint8(SegmentAmount),OffsetSegmentRuntableCount);
SetPayload((uint32_t)curMetaData.size()+1,(uint8_t*)curMetaData.c_str(),OffsetMetaData);
SetPayload((uint32_t)curDRM.size()+1,(uint8_t*)curDRM.c_str(),OffsetDrmData);
SetPayload((uint32_t)serializedQualities.size(),(uint8_t*)serializedQualities.c_str(),OffsetQualityEntryCount+1);
SetPayload((uint32_t)1,Box::uint8_to_uint8(Qualities.size()),OffsetQualityEntryCount);
SetPayload((uint32_t)serializedServers.size(),(uint8_t*)serializedServers.c_str(),OffsetServerEntryCount+1);
SetPayload((uint32_t)1,Box::uint8_to_uint8(Servers.size()),OffsetServerEntryCount);
SetPayload((uint32_t)curMovieIdentifier.size()+1,(uint8_t*)curMovieIdentifier.c_str(),29);//+1 for \0-terminated string...
SetPayload((uint32_t)4,Box::uint32_to_uint8(curSMPTE),25);
SetPayload((uint32_t)4,Box::uint32_to_uint8(0),21);
SetPayload((uint32_t)4,Box::uint32_to_uint8(curMediatime),17);
SetPayload((uint32_t)4,Box::uint32_to_uint8(0),13);
SetPayload((uint32_t)4,Box::uint32_to_uint8(curTimeScale),9);
SetPayload((uint32_t)1,temp,8);
SetPayload((uint32_t)4,Box::uint32_to_uint8(curBootstrapInfoVersion),4);
void ABST::regenerate( ) {
int myOffset = 26;
//0-terminated movieIdentifier
memcpy( (char*)data.c_str() + myOffset, movieIdentifier.c_str(), movieIdentifier.size() + 1);
myOffset += movieIdentifier.size() + 1;
//8-bit server amount
setInt8( Servers.size(), myOffset );
myOffset ++;
//0-terminated string for each entry
for( std::deque<std::string>::iterator it = Servers.begin(); it != Servers.end(); it++ ) {
memcpy( (char*)data.c_str() + myOffset, (*it).c_str(), (*it).size() + 1);
myOffset += (*it).size() + 1;
}
//8-bit quality amount
setInt8( Qualities.size(), myOffset );
myOffset ++;
//0-terminated string for each entry
for( std::deque<std::string>::iterator it = Qualities.begin();it != Qualities.end(); it++ ) {
memcpy( (char*)data.c_str() + myOffset, (*it).c_str(), (*it).size() + 1);
myOffset += (*it).size() + 1;
}
//0-terminated DrmData
memcpy( (char*)data.c_str() + myOffset, drmData.c_str(), drmData.size() + 1);
myOffset += drmData.size() + 1;
//0-terminated MetaData
memcpy( (char*)data.c_str() + myOffset, metaData.c_str(), metaData.size() + 1);
myOffset += metaData.size() + 1;
//8-bit segment run amount
setInt8( segmentTables.size(), myOffset );
myOffset ++;
//retrieve box for each entry
for( std::deque<Box*>::iterator it = segmentTables.begin(); it != segmentTables.end(); it++ ) {
memcpy( (char*)data.c_str() + myOffset, (*it)->asBox().c_str(), (*it)->boxedSize() );
myOffset += (*it)->boxedSize();
}
//8-bit fragment run amount
setInt8( fragmentTables.size(), myOffset );
myOffset ++;
//retrieve box for each entry
for( std::deque<Box*>::iterator it = fragmentTables.begin(); it != fragmentTables.end(); it++ ) {
memcpy( (char*)data.c_str() + myOffset, (*it)->asBox().c_str(), (*it)->boxedSize() + 1);
myOffset += (*it)->boxedSize();
}
isUpdated = false;
}
std::string ABST::toPrettyString( int indent ) {
std::string r;
r += std::string(indent, ' ')+"Bootstrap Info\n";
if (isUpdate){
if( getInt8(5) & 0x10 ) {
r += std::string(indent+1, ' ')+"Update\n";
} else {
r += std::string(indent+1, ' ')+"Replacement or new table\n";
}
if (isLive){
if( getInt8(5) & 0x20 ) {
r += std::string(indent+1, ' ' )+"Live\n";
}else{
r += std::string(indent+1, ' ' )+"Recorded\n";
}
r += std::string(indent+1, ' ')+"Profile "+JSON::Value((long long int)curProfile).asString()+"\n";
r += std::string(indent+1, ' ')+"Timescale "+JSON::Value((long long int)curTimeScale).asString()+"\n";
r += std::string(indent+1, ' ')+"CurrMediaTime "+JSON::Value((long long int)curMediatime).asString()+"\n";
r += std::string(indent+1, ' ')+"Segment Run Tables "+JSON::Value((long long int)SegmentRunTables.size()).asString()+"\n";
for( uint32_t i = 0; i < SegmentRunTables.size(); i++ ) {
r += ((ASRT*)SegmentRunTables[i])->toPrettyString(indent+2)+"\n";
r += std::string(indent+1, ' ')+"Profile "+JSON::Value((long long int)( getInt8(5) & 0xC0 ) ).asString()+"\n";
r += std::string(indent+1, ' ')+"Timescale "+JSON::Value((long long int)getInt64(10)).asString()+"\n";
r += std::string(indent+1, ' ')+"CurrMediaTime "+JSON::Value((long long int)getInt32(6)).asString()+"\n";
r += std::string(indent+1, ' ')+"Segment Run Tables "+JSON::Value((long long int)segmentTables.size()).asString()+"\n";
for( uint32_t i = 0; i < segmentTables.size(); i++ ) {
r += ((ASRT*)segmentTables[i])->toPrettyString(indent+2)+"\n";
}
r += std::string(indent+1, ' ')+"Fragment Run Tables "+JSON::Value((long long int)FragmentRunTables.size()).asString()+"\n";
for( uint32_t i = 0; i < FragmentRunTables.size(); i++ ) {
r += ((AFRT*)FragmentRunTables[i])->toPrettyString(indent+2)+"\n";
r += std::string(indent+1, ' ')+"Fragment Run Tables "+JSON::Value((long long int)fragmentTables.size()).asString()+"\n";
for( uint32_t i = 0; i < fragmentTables.size(); i++ ) {
r += ((AFRT*)fragmentTables[i])->toPrettyString(indent+2)+"\n";
}
return r;
}
void AFRT::SetUpdate( bool Update ) {
isUpdate = Update;
AFTR::AFRT() : Box(0x61667274){
setVersion( 0 );
setUpdate( 0 );
setTimeScale( 1000 );
}
void AFRT::setVersion( char newVersion ) {
setInt8( newVersion ,0 )
}
void AFRT::setUpdate( long newUpdate ) {
setInt24( newUpdate, 1 );
}
void AFRT::setTimeScale( long newScale ) {
setInt32( newScale, 4 );
}
void AFRT::addQualityEntry( std::string newQuality ) {
qualityModifiers.push_back( newQuality );
}
/*
void AFRT::AddQualityEntry( std::string Quality, uint32_t Offset ) {
if(Offset >= QualitySegmentUrlModifiers.size()) {
QualitySegmentUrlModifiers.resize(Offset+1);
@ -456,16 +557,6 @@ namespace MP4{
return r;
}
std::string mdatFold(std::string data){
std::string Result;
unsigned int t_int;
t_int = htonl(data.size()+8);
Result.append((char*)&t_int, 4);
t_int = htonl(0x6D646174);
Result.append((char*)&t_int, 4);
Result.append(data);
return Result;
}
*/
};

117
lib/mp4.h
View file

@ -1,7 +1,10 @@
#pragma once
#include <string>
#include <iostream>
#include <stdint.h>
#include <vector>
#include <deque>
#include <vector> ///\todo remove this include
#include <algorithm>
#include "json.h"
/// Contains all MP4 format related code.
@ -10,7 +13,7 @@ namespace MP4{
class Box {
public:
Box( size_t size = 0);
Box(char* boxType, size_t size = 0 );
Box( const char* boxType, size_t size = 0 );
Box( std::string & newData );
~Box();
std::string getType();
@ -19,67 +22,58 @@ namespace MP4{
size_t boxedSize();
size_t payloadSize();
std::string & asBox();
void regenerate();
void clear();
std::string toPrettyString( int indent = 0 );
protected:
void setInt8( char newData, size_t index = 0);
void setInt16( short newData, size_t index = 0);
void setInt32( long newData, size_t index = 0);
void setInt64( long long int newData, size_t index = 0);
void setString(std::string newData, size_t index = 0);
void setString(char* newData, size_t size, size_t index = 0);
void setInt8( char newData, size_t index );
char getInt8( size_t index );
void setInt16( short newData, size_t index );
short getInt16( size_t index );
void setInt24( long newData, size_t index );
long getInt24( size_t index );
void setInt32( long newData, size_t index );
long getInt32( size_t index );
void setInt64( long long int newData, size_t index );
long long int getInt64( size_t index );
void setString(std::string newData, size_t index );
void setString(char* newData, size_t size, size_t index );
std::string data;
bool isUpdated;
};//Box Class
//Tot hier rewritten
struct abst_serverentry {
std::string ServerBaseUrl;
};//abst_serverentry
struct abst_qualityentry {
std::string QualityModifier;
};//abst_qualityentry
/// ABST Box class
class ABST: public Box {
public:
ABST() : Box(0x61627374){};
void SetBootstrapVersion( uint32_t Version = 1 );
void SetProfile( uint8_t Profile = 0 );
void SetLive( bool Live = true );
void SetUpdate( bool Update = false );
void SetTimeScale( uint32_t Scale = 1000 );
void SetMediaTime( uint32_t Time = 0 );
void SetSMPTE( uint32_t Smpte = 0 );
void SetMovieIdentifier( std::string Identifier = "" );
void SetDRM( std::string Drm = "" );
void SetMetaData( std::string MetaData = "" );
void AddServerEntry( std::string Url = "", uint32_t Offset = 0 );
void AddQualityEntry( std::string Quality = "", uint32_t Offset = 0 );
void AddSegmentRunTable( Box * newSegment, uint32_t Offset = 0 );
void AddFragmentRunTable( Box * newFragment, uint32_t Offset = 0 );
void SetVersion( bool NewVersion = 0 );
void WriteContent( );
ABST();
void setVersion( char newVersion );
void setFlags( long newFlags );
void setBootstrapinfoVersion( long newVersion );
void setProfile( char newProfile );
void setLive( char newLive );
void setUpdate( char newUpdate );
void setTimeScale( long newTimeScale );
void setCurrentMediaTime( long long int newTime );
void setSmpteTimeCodeOffset( long long int newTime );
void setMovieIdentifier( std::string newIdentifier );
void addServerEntry( std::string newEntry );
void delServerEntry( std::string delEntry );
void addQualityEntry( std::string newEntry );
void delQualityEntry( std::string delEntry );
void setDrmData( std::string newDrm );
void setMetaData( std::string newMetaData );
void addSegmentRunTable( Box * newSegment );
void addFragmentRunTable( Box * newFragment );
std::string toPrettyString(int indent = 0);
private:
void SetDefaults( );
void SetReserved( );
uint32_t curBootstrapInfoVersion;
uint8_t curProfile;
bool isLive;
bool isUpdate;
bool Version;
uint32_t curTimeScale;
uint32_t curMediatime;//write as uint64_t
uint32_t curSMPTE;//write as uint64_t
std::string curMovieIdentifier;
std::string curDRM;
std::string curMetaData;
std::vector<abst_serverentry> Servers;
std::vector<abst_qualityentry> Qualities;
std::vector<Box *> SegmentRunTables;
std::vector<Box *> FragmentRunTables;
void regenerate();
std::string movieIdentifier;
std::deque<std::string> Servers;
std::deque<std::string> Qualities;
std::string drmData;
std::string metaData;
std::deque<Box *> segmentTables;
std::deque<Box *> fragmentTables;
};//ABST Box
struct afrt_fragmentrunentry {
@ -93,19 +87,18 @@ namespace MP4{
/// AFRT Box class
class AFRT : public Box {
public:
AFRT() : Box(0x61667274){};
void SetUpdate( bool Update = false );
void SetTimeScale( uint32_t Scale = 1000 );
void AddQualityEntry( std::string Quality = "", uint32_t Offset = 0 );
AFRT();
void setVersion( char newVersion );
void setUpdate( long newUpdate );
void setTimeScale( long newScale );
void addQualityEntry( std::string newQuality );
void AddFragmentRunEntry( uint32_t FirstFragment = 0, uint32_t FirstFragmentTimestamp = 0, uint32_t FragmentsDuration = 1, uint8_t Discontinuity = 0, uint32_t Offset = 0 );
void WriteContent( );
std::string toPrettyString(int indent = 0);
private:
void SetDefaults( );
bool isUpdate;
uint32_t curTimeScale;
std::vector<std::string> QualitySegmentUrlModifiers;
std::vector<afrt_fragmentrunentry> FragmentRunEntryTable;
std::deque<std::string> qualityModifiers;
std::deque<afrt_fragmentrunentry> FragmentRunEntryTable;
};//AFRT Box
struct asrt_segmentrunentry {
@ -132,6 +125,4 @@ namespace MP4{
Box * Container;
};//ASRT Box
std::string mdatFold(std::string data);
};