mistserver/lib/mp4_adobe.cpp
2014-06-18 10:39:27 +02:00

929 lines
29 KiB
C++

#include "mp4_adobe.h"
namespace MP4 {
ABST::ABST() {
memcpy(data + 4, "abst", 4);
setVersion(0);
setFlags(0);
setBootstrapinfoVersion(0);
setProfile(0);
setLive(1);
setUpdate(0);
setTimeScale(1000);
setCurrentMediaTime(0);
setSmpteTimeCodeOffset(0);
std::string empty;
setMovieIdentifier(empty);
setInt8(0, 30); //set serverentrycount to 0
setInt8(0, 31); //set qualityentrycount to 0
setDrmData(empty);
setMetaData(empty);
}
void ABST::setVersion(char newVersion) {
setInt8(newVersion, 0);
}
char ABST::getVersion() {
return getInt8(0);
}
void ABST::setFlags(uint32_t newFlags) {
setInt24(newFlags, 1);
}
uint32_t ABST::getFlags() {
return getInt24(1);
}
void ABST::setBootstrapinfoVersion(uint32_t newVersion) {
setInt32(newVersion, 4);
}
uint32_t ABST::getBootstrapinfoVersion() {
return getInt32(4);
}
void ABST::setProfile(char newProfile) {
//profile = bit 1 and 2 of byte 8.
setInt8((getInt8(8) & 0x3F) + ((newProfile & 0x03) << 6), 8);
}
char ABST::getProfile() {
return (getInt8(8) & 0xC0);
}
;
void ABST::setLive(bool newLive) {
//live = bit 4 of byte 8.
setInt8((getInt8(8) & 0xDF) + (newLive ? 0x10 : 0), 8);
}
bool ABST::getLive() {
return (getInt8(8) & 0x10);
}
void ABST::setUpdate(bool newUpdate) {
//update = bit 5 of byte 8.
setInt8((getInt8(8) & 0xEF) + (newUpdate ? 0x08 : 0), 8);
}
bool ABST::getUpdate() {
return (getInt8(8) & 0x08);
}
void ABST::setTimeScale(uint32_t newScale) {
setInt32(newScale, 9);
}
uint32_t ABST::getTimeScale() {
return getInt32(9);
}
void ABST::setCurrentMediaTime(uint64_t newTime) {
setInt64(newTime, 13);
}
uint64_t ABST::getCurrentMediaTime() {
return getInt64(13);
}
void ABST::setSmpteTimeCodeOffset(uint64_t newTime) {
setInt64(newTime, 21);
}
uint64_t ABST::getSmpteTimeCodeOffset() {
return getInt64(21);
}
void ABST::setMovieIdentifier(std::string & newIdentifier) {
setString(newIdentifier, 29);
}
char * ABST::getMovieIdentifier() {
return getString(29);
}
uint32_t ABST::getServerEntryCount() {
int countLoc = 29 + getStringLen(29) + 1;
return getInt8(countLoc);
}
void ABST::setServerEntry(std::string & newEntry, uint32_t no) {
int countLoc = 29 + getStringLen(29) + 1;
int tempLoc = countLoc + 1;
//attempt to reach the wanted position
unsigned int i;
for (i = 0; i < getInt8(countLoc) && i < no; ++i) {
tempLoc += getStringLen(tempLoc) + 1;
}
//we are now either at the end, or at the right position
//let's reserve any unreserved space...
if (no + 1 > getInt8(countLoc)) {
int amount = no + 1 - getInt8(countLoc);
if (!reserve(payloadOffset + tempLoc, 0, amount)) {
return;
};
memset(data + payloadOffset + tempLoc, 0, amount);
setInt8(no + 1, countLoc); //set new qualityEntryCount
tempLoc += no - i;
}
//now, tempLoc is at position for string number no, and we have at least 1 byte reserved.
setString(newEntry, tempLoc);
}
///\return Empty string if no > serverEntryCount(), serverEntry[no] otherwise.
const char * ABST::getServerEntry(uint32_t no) {
if (no + 1 > getServerEntryCount()) {
return "";
}
int tempLoc = 29 + getStringLen(29) + 1 + 1; //position of first entry
for (unsigned int i = 0; i < no; i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
return getString(tempLoc);
}
uint32_t ABST::getQualityEntryCount() {
int countLoc = 29 + getStringLen(29) + 1 + 1;
for (unsigned int i = 0; i < getServerEntryCount(); i++) {
countLoc += getStringLen(countLoc) + 1;
}
return getInt8(countLoc);
}
void ABST::setQualityEntry(std::string & newEntry, uint32_t no) {
int countLoc = 29 + getStringLen(29) + 1 + 1;
for (unsigned int i = 0; i < getServerEntryCount(); i++) {
countLoc += getStringLen(countLoc) + 1;
}
int tempLoc = countLoc + 1;
//attempt to reach the wanted position
unsigned int i;
for (i = 0; i < getInt8(countLoc) && i < no; ++i) {
tempLoc += getStringLen(tempLoc) + 1;
}
//we are now either at the end, or at the right position
//let's reserve any unreserved space...
if (no + 1 > getInt8(countLoc)) {
int amount = no + 1 - getInt8(countLoc);
if (!reserve(payloadOffset + tempLoc, 0, amount)) {
return;
};
memset(data + payloadOffset + tempLoc, 0, amount);
setInt8(no + 1, countLoc); //set new qualityEntryCount
tempLoc += no - i;
}
//now, tempLoc is at position for string number no, and we have at least 1 byte reserved.
setString(newEntry, tempLoc);
}
const char * ABST::getQualityEntry(uint32_t no) {
if (no > getQualityEntryCount()) {
return "";
}
int tempLoc = 29 + getStringLen(29) + 1 + 1; //position of serverentries;
for (unsigned int i = 0; i < getServerEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc += 1; //first qualityentry
for (unsigned int i = 0; i < no; i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
return getString(tempLoc);
}
void ABST::setDrmData(std::string newDrm) {
uint32_t tempLoc = 29 + getStringLen(29) + 1 + 1;
for (unsigned int i = 0; i < getServerEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc++;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
setString(newDrm, tempLoc);
}
char * ABST::getDrmData() {
uint32_t tempLoc = 29 + getStringLen(29) + 1 + 1;
for (unsigned int i = 0; i < getServerEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc++;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
return getString(tempLoc);
}
void ABST::setMetaData(std::string newMetaData) {
uint32_t tempLoc = 29 + getStringLen(29) + 1 + 1;
for (unsigned int i = 0; i < getServerEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc++;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc += getStringLen(tempLoc) + 1;
setString(newMetaData, tempLoc);
}
char * ABST::getMetaData() {
uint32_t tempLoc = 29 + getStringLen(29) + 1 + 1;
for (unsigned int i = 0; i < getServerEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc++;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc += getStringLen(tempLoc) + 1;
return getString(tempLoc);
}
uint32_t ABST::getSegmentRunTableCount() {
uint32_t tempLoc = 29 + getStringLen(29) + 1 + 1;
for (unsigned int i = 0; i < getServerEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc++;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc += getStringLen(tempLoc) + 1; //DrmData
tempLoc += getStringLen(tempLoc) + 1; //MetaData
return getInt8(tempLoc);
}
void ABST::setSegmentRunTable(ASRT & newSegment, uint32_t no) {
uint32_t tempLoc = 29 + getStringLen(29) + 1 + 1;
for (unsigned int i = 0; i < getServerEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc++;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc += getStringLen(tempLoc) + 1; //DrmData
tempLoc += getStringLen(tempLoc) + 1; //MetaData
int countLoc = tempLoc;
tempLoc++; //skip segmentRuntableCount
//attempt to reach the wanted position
unsigned int i;
for (i = 0; i < getInt8(countLoc) && i < no; ++i) {
tempLoc += getBoxLen(tempLoc);
}
//we are now either at the end, or at the right position
//let's reserve any unreserved space...
if (no + 1 > getInt8(countLoc)) {
int amount = no + 1 - getInt8(countLoc);
if (!reserve(payloadOffset + tempLoc, 0, amount * 8)) {
return;
};
//set empty erro boxes as contents
for (int j = 0; j < amount; ++j) {
memcpy(data + payloadOffset + tempLoc + j * 8, "\000\000\000\010erro", 8);
}
setInt8(no + 1, countLoc); //set new count
tempLoc += (no - i) * 8;
}
//now, tempLoc is at position for string number no, and we have at least an erro box reserved.
setBox(newSegment, tempLoc);
}
ASRT & ABST::getSegmentRunTable(uint32_t no) {
static Box result;
if (no > getSegmentRunTableCount()) {
static Box res;
return (ASRT &)res;
}
uint32_t tempLoc = 29 + getStringLen(29) + 1 + 1;
for (unsigned int i = 0; i < getServerEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc++;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc += getStringLen(tempLoc) + 1; //DrmData
tempLoc += getStringLen(tempLoc) + 1; //MetaData
tempLoc++; //segmentRuntableCount
for (unsigned int i = 0; i < no; ++i) {
tempLoc += getBoxLen(tempLoc);
}
return (ASRT &)getBox(tempLoc);
}
uint32_t ABST::getFragmentRunTableCount() {
uint32_t tempLoc = 29 + getStringLen(29) + 1 + 1;
for (unsigned int i = 0; i < getServerEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc++;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc += getStringLen(tempLoc) + 1; //DrmData
tempLoc += getStringLen(tempLoc) + 1; //MetaData
for (unsigned int i = getInt8(tempLoc++); i != 0; --i) {
tempLoc += getBoxLen(tempLoc);
}
return getInt8(tempLoc);
}
void ABST::setFragmentRunTable(AFRT & newFragment, uint32_t no) {
uint32_t tempLoc = 29 + getStringLen(29) + 1 + 1;
for (unsigned int i = 0; i < getServerEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc++;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc += getStringLen(tempLoc) + 1; //DrmData
tempLoc += getStringLen(tempLoc) + 1; //MetaData
for (unsigned int i = getInt8(tempLoc++); i != 0; --i) {
tempLoc += getBoxLen(tempLoc);
}
int countLoc = tempLoc;
tempLoc++;
//attempt to reach the wanted position
unsigned int i;
for (i = 0; i < getInt8(countLoc) && i < no; ++i) {
tempLoc += getBoxLen(tempLoc);
}
//we are now either at the end, or at the right position
//let's reserve any unreserved space...
if (no + 1 > getInt8(countLoc)) {
unsigned int amount = no + 1 - getInt8(countLoc);
if (!reserve(payloadOffset + tempLoc, 0, amount * 8)) {
return;
};
//set empty erro boxes as contents
for (unsigned int j = 0; j < amount; ++j) {
memcpy(data + payloadOffset + tempLoc + j * 8, "\000\000\000\010erro", 8);
}
setInt8(no + 1, countLoc); //set new count
tempLoc += (no - i) * 8;
}
//now, tempLoc is at position for string number no, and we have at least 1 byte reserved.
setBox(newFragment, tempLoc);
}
AFRT & ABST::getFragmentRunTable(uint32_t no) {
static Box result;
if (no >= getFragmentRunTableCount()) {
static Box res;
return (AFRT &)res;
}
uint32_t tempLoc = 29 + getStringLen(29) + 1 + 1;
for (unsigned int i = 0; i < getServerEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc++;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc += getStringLen(tempLoc) + 1; //DrmData
tempLoc += getStringLen(tempLoc) + 1; //MetaData
for (unsigned int i = getInt8(tempLoc++); i != 0; --i) {
tempLoc += getBoxLen(tempLoc);
}
tempLoc++;
for (unsigned int i = 0; i < no; i++) {
tempLoc += getBoxLen(tempLoc);
}
return (AFRT &)getBox(tempLoc);
}
std::string ABST::toPrettyString(uint32_t indent) {
std::stringstream r;
r << std::string(indent, ' ') << "[abst] Bootstrap Info (" << boxedSize() << ")" << std::endl;
r << std::string(indent + 1, ' ') << "Version " << (int)getVersion() << std::endl;
r << std::string(indent + 1, ' ') << "BootstrapinfoVersion " << getBootstrapinfoVersion() << std::endl;
r << std::string(indent + 1, ' ') << "Profile " << (int)getProfile() << std::endl;
if (getLive()) {
r << std::string(indent + 1, ' ') << "Live" << std::endl;
} else {
r << std::string(indent + 1, ' ') << "Recorded" << std::endl;
}
if (getUpdate()) {
r << std::string(indent + 1, ' ') << "Update" << std::endl;
} else {
r << std::string(indent + 1, ' ') << "Replacement or new table" << std::endl;
}
r << std::string(indent + 1, ' ') << "Timescale " << getTimeScale() << std::endl;
r << std::string(indent + 1, ' ') << "CurrMediaTime " << getCurrentMediaTime() << std::endl;
r << std::string(indent + 1, ' ') << "SmpteTimeCodeOffset " << getSmpteTimeCodeOffset() << std::endl;
r << std::string(indent + 1, ' ') << "MovieIdentifier " << getMovieIdentifier() << std::endl;
r << std::string(indent + 1, ' ') << "ServerEntryTable (" << getServerEntryCount() << ")" << std::endl;
for (unsigned int i = 0; i < getServerEntryCount(); i++) {
r << std::string(indent + 2, ' ') << i << ": " << getServerEntry(i) << std::endl;
}
r << std::string(indent + 1, ' ') << "QualityEntryTable (" << getQualityEntryCount() << ")" << std::endl;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
r << std::string(indent + 2, ' ') << i << ": " << getQualityEntry(i) << std::endl;
}
r << std::string(indent + 1, ' ') << "DrmData " << getDrmData() << std::endl;
r << std::string(indent + 1, ' ') << "MetaData " << getMetaData() << std::endl;
r << std::string(indent + 1, ' ') << "SegmentRunTableEntries (" << getSegmentRunTableCount() << ")" << std::endl;
for (uint32_t i = 0; i < getSegmentRunTableCount(); i++) {
r << ((Box)getSegmentRunTable(i)).toPrettyString(indent + 2);
}
r << std::string(indent + 1, ' ') + "FragmentRunTableEntries (" << getFragmentRunTableCount() << ")" << std::endl;
for (uint32_t i = 0; i < getFragmentRunTableCount(); i++) {
r << ((Box)getFragmentRunTable(i)).toPrettyString(indent + 2);
}
return r.str();
}
AFRT::AFRT() {
memcpy(data + 4, "afrt", 4);
setVersion(0);
setUpdate(0);
setTimeScale(1000);
}
void AFRT::setVersion(char newVersion) {
setInt8(newVersion, 0);
}
uint32_t AFRT::getVersion() {
return getInt8(0);
}
void AFRT::setUpdate(uint32_t newUpdate) {
setInt24(newUpdate, 1);
}
uint32_t AFRT::getUpdate() {
return getInt24(1);
}
void AFRT::setTimeScale(uint32_t newScale) {
setInt32(newScale, 4);
}
uint32_t AFRT::getTimeScale() {
return getInt32(4);
}
uint32_t AFRT::getQualityEntryCount() {
return getInt8(8);
}
void AFRT::setQualityEntry(std::string & newEntry, uint32_t no) {
int countLoc = 8;
int tempLoc = countLoc + 1;
//attempt to reach the wanted position
unsigned int i;
for (i = 0; i < getQualityEntryCount() && i < no; ++i) {
tempLoc += getStringLen(tempLoc) + 1;
}
//we are now either at the end, or at the right position
//let's reserve any unreserved space...
if (no + 1 > getQualityEntryCount()) {
int amount = no + 1 - getQualityEntryCount();
if (!reserve(payloadOffset + tempLoc, 0, amount)) {
return;
};
memset(data + payloadOffset + tempLoc, 0, amount);
setInt8(no + 1, countLoc); //set new qualityEntryCount
tempLoc += no - i;
}
//now, tempLoc is at position for string number no, and we have at least 1 byte reserved.
setString(newEntry, tempLoc);
}
const char * AFRT::getQualityEntry(uint32_t no) {
if (no + 1 > getQualityEntryCount()) {
return "";
}
int tempLoc = 9; //position of first quality entry
for (unsigned int i = 0; i < no; i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
return getString(tempLoc);
}
uint32_t AFRT::getFragmentRunCount() {
int tempLoc = 9;
for (unsigned int i = 0; i < getQualityEntryCount(); ++i) {
tempLoc += getStringLen(tempLoc) + 1;
}
return getInt32(tempLoc);
}
void AFRT::setFragmentRun(afrt_runtable newRun, uint32_t no) {
int tempLoc = 9;
for (unsigned int i = 0; i < getQualityEntryCount(); ++i) {
tempLoc += getStringLen(tempLoc) + 1;
}
int countLoc = tempLoc;
unsigned int count = getInt32(countLoc);
tempLoc += 4;
for (unsigned int i = 0; i < no; i++) {
if (i + 1 > count) {
setInt32(0, tempLoc);
setInt64(0, tempLoc + 4);
setInt32(1, tempLoc + 12);
}
if (getInt32(tempLoc + 12) == 0) {
tempLoc += 17;
} else {
tempLoc += 16;
}
}
setInt32(newRun.firstFragment, tempLoc);
setInt64(newRun.firstTimestamp, tempLoc + 4);
setInt32(newRun.duration, tempLoc + 12);
if (newRun.duration == 0) {
setInt8(newRun.discontinuity, tempLoc + 16);
}
if (count < no + 1) {
setInt32(no + 1, countLoc);
}
}
afrt_runtable AFRT::getFragmentRun(uint32_t no) {
afrt_runtable res;
if (no > getFragmentRunCount()) {
return res;
}
int tempLoc = 9;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc += 4;
for (unsigned int i = 0; i < no; i++) {
if (getInt32(tempLoc + 12) == 0) {
tempLoc += 17;
} else {
tempLoc += 16;
}
}
res.firstFragment = getInt32(tempLoc);
res.firstTimestamp = getInt64(tempLoc + 4);
res.duration = getInt32(tempLoc + 12);
if (res.duration) {
res.discontinuity = getInt8(tempLoc + 16);
} else {
res.discontinuity = 0;
}
return res;
}
std::string AFRT::toPrettyString(uint32_t indent) {
std::stringstream r;
r << std::string(indent, ' ') << "[afrt] Fragment Run Table (" << boxedSize() << ")" << std::endl;
r << std::string(indent + 1, ' ') << "Version " << (int)getVersion() << std::endl;
if (getUpdate()) {
r << std::string(indent + 1, ' ') << "Update" << std::endl;
} else {
r << std::string(indent + 1, ' ') << "Replacement or new table" << std::endl;
}
r << std::string(indent + 1, ' ') << "Timescale " << getTimeScale() << std::endl;
r << std::string(indent + 1, ' ') << "QualitySegmentUrlModifiers (" << getQualityEntryCount() << ")" << std::endl;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
r << std::string(indent + 2, ' ') << i << ": " << getQualityEntry(i) << std::endl;
}
r << std::string(indent + 1, ' ') << "FragmentRunEntryTable (" << getFragmentRunCount() << ")" << std::endl;
for (unsigned int i = 0; i < getFragmentRunCount(); i++) {
afrt_runtable myRun = getFragmentRun(i);
if (myRun.duration) {
r << std::string(indent + 2, ' ') << i << ": " << myRun.firstFragment << " is at " << ((double)myRun.firstTimestamp / (double)getTimeScale())
<< "s, " << ((double)myRun.duration / (double)getTimeScale()) << "s per fragment." << std::endl;
} else {
r << std::string(indent + 2, ' ') << i << ": " << myRun.firstFragment << " is at " << ((double)myRun.firstTimestamp / (double)getTimeScale())
<< "s, discontinuity type " << myRun.discontinuity << std::endl;
}
}
return r.str();
}
ASRT::ASRT() {
memcpy(data + 4, "asrt", 4);
setVersion(0);
setUpdate(0);
}
void ASRT::setVersion(char newVersion) {
setInt8(newVersion, 0);
}
uint32_t ASRT::getVersion() {
return getInt8(0);
}
void ASRT::setUpdate(uint32_t newUpdate) {
setInt24(newUpdate, 1);
}
uint32_t ASRT::getUpdate() {
return getInt24(1);
}
uint32_t ASRT::getQualityEntryCount() {
return getInt8(4);
}
void ASRT::setQualityEntry(std::string & newEntry, uint32_t no) {
int countLoc = 4;
int tempLoc = countLoc + 1;
//attempt to reach the wanted position
unsigned int i;
for (i = 0; i < getQualityEntryCount() && i < no; ++i) {
tempLoc += getStringLen(tempLoc) + 1;
}
//we are now either at the end, or at the right position
//let's reserve any unreserved space...
if (no + 1 > getQualityEntryCount()) {
int amount = no + 1 - getQualityEntryCount();
if (!reserve(payloadOffset + tempLoc, 0, amount)) {
return;
};
memset(data + payloadOffset + tempLoc, 0, amount);
setInt8(no + 1, countLoc); //set new qualityEntryCount
tempLoc += no - i;
}
//now, tempLoc is at position for string number no, and we have at least 1 byte reserved.
setString(newEntry, tempLoc);
}
const char * ASRT::getQualityEntry(uint32_t no) {
if (no > getQualityEntryCount()) {
return "";
}
int tempLoc = 5; //position of qualityentry count;
for (unsigned int i = 0; i < no; i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
return getString(tempLoc);
}
uint32_t ASRT::getSegmentRunEntryCount() {
int tempLoc = 5; //position of qualityentry count;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
return getInt32(tempLoc);
}
void ASRT::setSegmentRun(uint32_t firstSegment, uint32_t fragmentsPerSegment, uint32_t no) {
int tempLoc = 5; //position of qualityentry count;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
tempLoc += getStringLen(tempLoc) + 1;
}
int countLoc = tempLoc;
tempLoc += 4 + no * 8;
if (no + 1 > getInt32(countLoc)) {
setInt32(no + 1, countLoc); //set new qualityEntryCount
}
setInt32(firstSegment, tempLoc);
setInt32(fragmentsPerSegment, tempLoc + 4);
}
asrt_runtable ASRT::getSegmentRun(uint32_t no) {
asrt_runtable res;
if (no >= getSegmentRunEntryCount()) {
return res;
}
int tempLoc = 5; //position of qualityentry count;
for (unsigned int i = 0; i < getQualityEntryCount(); ++i) {
tempLoc += getStringLen(tempLoc) + 1;
}
tempLoc += 4 + 8 * no;
res.firstSegment = getInt32(tempLoc);
res.fragmentsPerSegment = getInt32(tempLoc + 4);
return res;
}
std::string ASRT::toPrettyString(uint32_t indent) {
std::stringstream r;
r << std::string(indent, ' ') << "[asrt] Segment Run Table (" << boxedSize() << ")" << std::endl;
r << std::string(indent + 1, ' ') << "Version " << getVersion() << std::endl;
if (getUpdate()) {
r << std::string(indent + 1, ' ') << "Update" << std::endl;
} else {
r << std::string(indent + 1, ' ') << "Replacement or new table" << std::endl;
}
r << std::string(indent + 1, ' ') << "QualityEntryTable (" << getQualityEntryCount() << ")" << std::endl;
for (unsigned int i = 0; i < getQualityEntryCount(); i++) {
r << std::string(indent + 2, ' ') << i << ": " << getQualityEntry(i) << std::endl;
}
r << std::string(indent + 1, ' ') << "SegmentRunEntryTable (" << getSegmentRunEntryCount() << ")" << std::endl;
for (unsigned int i = 0; i < getSegmentRunEntryCount(); i++) {
r << std::string(indent + 2, ' ') << i << ": First=" << getSegmentRun(i).firstSegment << ", FragmentsPerSegment="
<< getSegmentRun(i).fragmentsPerSegment << std::endl;
}
return r.str();
}
AFRA::AFRA() {
memcpy(data + 4, "afra", 4);
setInt32(0, 9); //entrycount = 0
setFlags(0);
}
void AFRA::setVersion(uint32_t newVersion) {
setInt8(newVersion, 0);
}
uint32_t AFRA::getVersion() {
return getInt8(0);
}
void AFRA::setFlags(uint32_t newFlags) {
setInt24(newFlags, 1);
}
uint32_t AFRA::getFlags() {
return getInt24(1);
}
void AFRA::setLongIDs(bool newVal) {
if (newVal) {
setInt8((getInt8(4) & 0x7F) + 0x80, 4);
} else {
setInt8((getInt8(4) & 0x7F), 4);
}
}
bool AFRA::getLongIDs() {
return getInt8(4) & 0x80;
}
void AFRA::setLongOffsets(bool newVal) {
if (newVal) {
setInt8((getInt8(4) & 0xBF) + 0x40, 4);
} else {
setInt8((getInt8(4) & 0xBF), 4);
}
}
bool AFRA::getLongOffsets() {
return getInt8(4) & 0x40;
}
void AFRA::setGlobalEntries(bool newVal) {
if (newVal) {
setInt8((getInt8(4) & 0xDF) + 0x20, 4);
} else {
setInt8((getInt8(4) & 0xDF), 4);
}
}
bool AFRA::getGlobalEntries() {
return getInt8(4) & 0x20;
}
void AFRA::setTimeScale(uint32_t newVal) {
setInt32(newVal, 5);
}
uint32_t AFRA::getTimeScale() {
return getInt32(5);
}
uint32_t AFRA::getEntryCount() {
return getInt32(9);
}
void AFRA::setEntry(afraentry newEntry, uint32_t no) {
int entrysize = 12;
if (getLongOffsets()) {
entrysize = 16;
}
setInt64(newEntry.time, 13 + entrysize * no);
if (getLongOffsets()) {
setInt64(newEntry.offset, 21 + entrysize * no);
} else {
setInt32(newEntry.offset, 21 + entrysize * no);
}
if (no + 1 > getEntryCount()) {
setInt32(no + 1, 9);
}
}
afraentry AFRA::getEntry(uint32_t no) {
afraentry ret;
int entrysize = 12;
if (getLongOffsets()) {
entrysize = 16;
}
ret.time = getInt64(13 + entrysize * no);
if (getLongOffsets()) {
ret.offset = getInt64(21 + entrysize * no);
} else {
ret.offset = getInt32(21 + entrysize * no);
}
return ret;
}
uint32_t AFRA::getGlobalEntryCount() {
if (!getGlobalEntries()) {
return 0;
}
int entrysize = 12;
if (getLongOffsets()) {
entrysize = 16;
}
return getInt32(13 + entrysize * getEntryCount());
}
void AFRA::setGlobalEntry(globalafraentry newEntry, uint32_t no) {
int offset = 13 + 12 * getEntryCount() + 4;
if (getLongOffsets()) {
offset = 13 + 16 * getEntryCount() + 4;
}
int entrysize = 20;
if (getLongIDs()) {
entrysize += 4;
}
if (getLongOffsets()) {
entrysize += 8;
}
setInt64(newEntry.time, offset + entrysize * no);
if (getLongIDs()) {
setInt32(newEntry.segment, offset + entrysize * no + 8);
setInt32(newEntry.fragment, offset + entrysize * no + 12);
} else {
setInt16(newEntry.segment, offset + entrysize * no + 8);
setInt16(newEntry.fragment, offset + entrysize * no + 10);
}
if (getLongOffsets()) {
setInt64(newEntry.afraoffset, offset + entrysize * no + entrysize - 16);
setInt64(newEntry.offsetfromafra, offset + entrysize * no + entrysize - 8);
} else {
setInt32(newEntry.afraoffset, offset + entrysize * no + entrysize - 8);
setInt32(newEntry.offsetfromafra, offset + entrysize * no + entrysize - 4);
}
if (getInt32(offset - 4) < no + 1) {
setInt32(no + 1, offset - 4);
}
}
globalafraentry AFRA::getGlobalEntry(uint32_t no) {
globalafraentry ret;
int offset = 13 + 12 * getEntryCount() + 4;
if (getLongOffsets()) {
offset = 13 + 16 * getEntryCount() + 4;
}
int entrysize = 20;
if (getLongIDs()) {
entrysize += 4;
}
if (getLongOffsets()) {
entrysize += 8;
}
ret.time = getInt64(offset + entrysize * no);
if (getLongIDs()) {
ret.segment = getInt32(offset + entrysize * no + 8);
ret.fragment = getInt32(offset + entrysize * no + 12);
} else {
ret.segment = getInt16(offset + entrysize * no + 8);
ret.fragment = getInt16(offset + entrysize * no + 10);
}
if (getLongOffsets()) {
ret.afraoffset = getInt64(offset + entrysize * no + entrysize - 16);
ret.offsetfromafra = getInt64(offset + entrysize * no + entrysize - 8);
} else {
ret.afraoffset = getInt32(offset + entrysize * no + entrysize - 8);
ret.offsetfromafra = getInt32(offset + entrysize * no + entrysize - 4);
}
return ret;
}
std::string AFRA::toPrettyString(uint32_t indent) {
std::stringstream r;
r << std::string(indent, ' ') << "[afra] Fragment Random Access (" << boxedSize() << ")" << std::endl;
r << std::string(indent + 1, ' ') << "Version " << getVersion() << std::endl;
r << std::string(indent + 1, ' ') << "Flags " << getFlags() << std::endl;
r << std::string(indent + 1, ' ') << "Long IDs " << getLongIDs() << std::endl;
r << std::string(indent + 1, ' ') << "Long Offsets " << getLongOffsets() << std::endl;
r << std::string(indent + 1, ' ') << "Global Entries " << getGlobalEntries() << std::endl;
r << std::string(indent + 1, ' ') << "TimeScale " << getTimeScale() << std::endl;
uint32_t count = getEntryCount();
r << std::string(indent + 1, ' ') << "Entries (" << count << ") " << std::endl;
for (uint32_t i = 0; i < count; ++i) {
afraentry tmpent = getEntry(i);
r << std::string(indent + 1, ' ') << i << ": Time " << tmpent.time << ", Offset " << tmpent.offset << std::endl;
}
if (getGlobalEntries()) {
count = getGlobalEntryCount();
r << std::string(indent + 1, ' ') << "Global Entries (" << count << ") " << std::endl;
for (uint32_t i = 0; i < count; ++i) {
globalafraentry tmpent = getGlobalEntry(i);
r << std::string(indent + 1, ' ') << i << ": T " << tmpent.time << ", S" << tmpent.segment << "F" << tmpent.fragment << ", "
<< tmpent.afraoffset << "/" << tmpent.offsetfromafra << std::endl;
}
}
return r.str();
}
}