#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();
  }
}