100 lines
3.7 KiB
C++
100 lines
3.7 KiB
C++
#include "box.cpp"
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
struct afrt_fragmentrunentry {
|
|
uint32_t FirstFragment;
|
|
uint32_t FirstFragmentTimestamp; //write as uint64_t
|
|
uint32_t FragmentDuration;
|
|
uint8_t DiscontinuityIndicator;//if FragmentDuration == 0
|
|
};//afrt_fragmentrunentry
|
|
|
|
class Box_afrt {
|
|
public:
|
|
Box_afrt( );
|
|
~Box_afrt();
|
|
Box * GetBox();
|
|
void SetUpdate( bool Update = false );
|
|
void SetTimeScale( uint32_t Scale = 1000 );
|
|
void AddQualityEntry( std::string Quality = "", uint32_t Offset = 0 );
|
|
void AddFragmentRunEntry( uint32_t FirstFragment = 0, uint32_t FirstFragmentTimestamp = 0, uint32_t FragmentsDuration = 1, uint8_t Discontinuity = 0, uint32_t Offset = 0 );
|
|
void WriteContent( );
|
|
private:
|
|
void SetDefaults( );
|
|
bool isUpdate;
|
|
uint32_t curTimeScale;
|
|
std::vector<std::string> QualitySegmentUrlModifiers;
|
|
std::vector<afrt_fragmentrunentry> FragmentRunEntryTable;
|
|
Box * Container;
|
|
};//Box_ftyp Class
|
|
|
|
Box_afrt::Box_afrt( ) {
|
|
Container = new Box( 0x61667274 );
|
|
}
|
|
|
|
Box_afrt::~Box_afrt() {
|
|
delete Container;
|
|
}
|
|
|
|
Box * Box_afrt::GetBox() {
|
|
return Container;
|
|
}
|
|
|
|
void Box_afrt::SetUpdate( bool Update ) {
|
|
isUpdate = Update;
|
|
}
|
|
|
|
void Box_afrt::AddQualityEntry( std::string Quality, uint32_t Offset ) {
|
|
if(Offset >= QualitySegmentUrlModifiers.size()) {
|
|
QualitySegmentUrlModifiers.resize(Offset+1);
|
|
}
|
|
QualitySegmentUrlModifiers[Offset] = Quality;
|
|
}
|
|
|
|
void Box_afrt::AddFragmentRunEntry( uint32_t FirstFragment, uint32_t FirstFragmentTimestamp, uint32_t FragmentsDuration, uint8_t Discontinuity, uint32_t Offset ) {
|
|
if( Offset >= FragmentRunEntryTable.size() ) {
|
|
FragmentRunEntryTable.resize(Offset+1);
|
|
}
|
|
FragmentRunEntryTable[Offset].FirstFragment = FirstFragment;
|
|
FragmentRunEntryTable[Offset].FirstFragmentTimestamp = FirstFragmentTimestamp;
|
|
FragmentRunEntryTable[Offset].FragmentDuration = FragmentsDuration;
|
|
FragmentRunEntryTable[Offset].DiscontinuityIndicator = Discontinuity;
|
|
}
|
|
|
|
void Box_afrt::SetDefaults( ) {
|
|
SetUpdate( );
|
|
SetTimeScale( );
|
|
}
|
|
|
|
void Box_afrt::SetTimeScale( uint32_t Scale ) {
|
|
curTimeScale = Scale;
|
|
}
|
|
|
|
void Box_afrt::WriteContent( ) {
|
|
std::string serializedQualities = "";
|
|
std::string serializedFragmentEntries = "";
|
|
Container->ResetPayload( );
|
|
|
|
for( uint32_t i = 0; i < QualitySegmentUrlModifiers.size(); i++ ) {
|
|
serializedQualities.append(QualitySegmentUrlModifiers[i].c_str());
|
|
serializedQualities += '\0';
|
|
}
|
|
for( uint32_t i = 0; i < FragmentRunEntryTable.size(); i ++ ) {
|
|
serializedFragmentEntries.append((char*)Box::uint32_to_uint8(FragmentRunEntryTable[i].FirstFragment),4);
|
|
serializedFragmentEntries.append((char*)Box::uint32_to_uint8(0),4);
|
|
serializedFragmentEntries.append((char*)Box::uint32_to_uint8(FragmentRunEntryTable[i].FirstFragmentTimestamp),4);
|
|
serializedFragmentEntries.append((char*)Box::uint32_to_uint8(FragmentRunEntryTable[i].FragmentDuration)),4;
|
|
if(FragmentRunEntryTable[i].FragmentDuration == 0) {
|
|
serializedFragmentEntries.append((char*)Box::uint8_to_uint8(FragmentRunEntryTable[i].DiscontinuityIndicator),1);
|
|
}
|
|
}
|
|
|
|
uint32_t OffsetFragmentRunEntryCount = 9 + serializedQualities.size();
|
|
|
|
Container->SetPayload((uint32_t)serializedFragmentEntries.size(),(uint8_t*)serializedFragmentEntries.c_str(),OffsetFragmentRunEntryCount+4);
|
|
Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(FragmentRunEntryTable.size()),OffsetFragmentRunEntryCount);
|
|
Container->SetPayload((uint32_t)serializedQualities.size(),(uint8_t*)serializedQualities.c_str(),9);
|
|
Container->SetPayload((uint32_t)1,Box::uint8_to_uint8(QualitySegmentUrlModifiers.size()),8);
|
|
Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(curTimeScale),4);
|
|
Container->SetPayload((uint32_t)4,Box::uint32_to_uint8((isUpdate ? 1 : 0)));
|
|
}
|