#include "mp4_ms.h"
#include "mp4_encryption.h" /*LTS*/

namespace MP4 {

  static char c2hex(int c) {
    if (c >= '0' && c <= '9') return c - '0';
    if (c >= 'a' && c <= 'f') return c - 'a' + 10;
    if (c >= 'A' && c <= 'F') return c - 'A' + 10;
    return 0;
  }


  SDTP::SDTP() {
    memcpy(data + 4, "sdtp", 4);
  }

  void SDTP::setVersion(uint32_t newVersion) {
    setInt8(newVersion, 0);
  }

  uint32_t SDTP::getVersion() {
    return getInt8(0);
  }

  void SDTP::setValue(uint32_t newValue, size_t index) {
    setInt8(newValue, index);
  }

  uint32_t SDTP::getValue(size_t index) {
    return getInt8(index);
  }

  std::string SDTP::toPrettyString(uint32_t indent) {
    std::stringstream r;
    r << std::string(indent, ' ') << "[sdtp] Sample Dependancy Type (" << boxedSize() << ")" << std::endl;
    r << std::string(indent + 1, ' ') << "Samples: " << (boxedSize() - 12) << std::endl;
    for (size_t i = 1; i <= boxedSize() - 12; ++i) {
      uint32_t val = getValue(i + 3);
      r << std::string(indent + 2, ' ') << "[" << i << "] = ";
      switch (val & 3) {
        case 0:
          r << "               ";
          break;
        case 1:
          r << "Redundant,     ";
          break;
        case 2:
          r << "Not redundant, ";
          break;
        case 3:
          r << "Error,         ";
          break;
      }
      switch (val & 12) {
        case 0:
          r << "                ";
          break;
        case 4:
          r << "Not disposable, ";
          break;
        case 8:
          r << "Disposable,     ";
          break;
        case 12:
          r << "Error,          ";
          break;
      }
      switch (val & 48) {
        case 0:
          r << "            ";
          break;
        case 16:
          r << "IFrame,     ";
          break;
        case 32:
          r << "Not IFrame, ";
          break;
        case 48:
          r << "Error,      ";
          break;
      }
      r << "(" << val << ")" << std::endl;
    }
    return r.str();
  }

  UUID::UUID() {
    memcpy(data + 4, "uuid", 4);
    setInt64(0, 0);
    setInt64(0, 8);
  }

  std::string UUID::getUUID() {
    std::stringstream r;
    r << std::hex;
    for (int i = 0; i < 16; ++i) {
      if (i == 4 || i == 6 || i == 8 || i == 10) {
        r << "-";
      }
      r << std::setfill('0') << std::setw(2) << std::right << (int)(data[8 + i]);
    }
    return r.str();
  }

  void UUID::setUUID(const std::string & uuid_string) {
    //reset UUID to zero
    for (int i = 0; i < 4; ++i) {
      ((uint32_t *)(data + 8))[i] = 0;
    }
    //set the UUID from the string, char by char
    int i = 0;
    for (size_t j = 0; j < uuid_string.size(); ++j) {
      if (uuid_string[j] == '-') {
        continue;
      }
      data[8 + i / 2] |= (c2hex(uuid_string[j]) << ((~i & 1) << 2));
      ++i;
    }
  }

  void UUID::setUUID(const char * raw_uuid) {
    memcpy(data + 8, raw_uuid, 16);
  }

  std::string UUID::toPrettyString(uint32_t indent) {
    std::string UUID = getUUID();
    if (UUID == "d4807ef2-ca39-4695-8e54-26cb9e46a79f") {
      return ((UUID_TrackFragmentReference *)this)->toPrettyString(indent);
    }
    /*LTS-START*/
    if (UUID == "a2394f52-5a9b-4f14-a244-6c427c648df4") {
      return ((UUID_SampleEncryption *)this)->toPrettyString(indent);
    }
    if (UUID == "8974dbce-7be7-4c51-84f9-7148f9882554") {
      return ((UUID_TrackEncryption *)this)->toPrettyString(indent);
    }
    if (UUID == "d08a4f18-10f3-4a82-b6c8-32d8aba183d3") {
      return ((UUID_ProtectionSystemSpecificHeader *)this)->toPrettyString(indent);
    }
    /*LTS-END*/
    std::stringstream r;
    r << std::string(indent, ' ') << "[uuid] Extension box (" << boxedSize() << ")" << std::endl;
    r << std::string(indent + 1, ' ') << "UUID: " << UUID << std::endl;
    r << std::string(indent + 1, ' ') << "Unknown UUID - ignoring contents." << std::endl;
    return r.str();
  }

  UUID_TrackFragmentReference::UUID_TrackFragmentReference() {
    setUUID((std::string)"d4807ef2-ca39-4695-8e54-26cb9e46a79f");
  }

  void UUID_TrackFragmentReference::setVersion(uint32_t newVersion) {
    setInt8(newVersion, 16);
  }

  uint32_t UUID_TrackFragmentReference::getVersion() {
    return getInt8(16);
  }

  void UUID_TrackFragmentReference::setFlags(uint32_t newFlags) {
    setInt24(newFlags, 17);
  }

  uint32_t UUID_TrackFragmentReference::getFlags() {
    return getInt24(17);
  }

  void UUID_TrackFragmentReference::setFragmentCount(uint32_t newCount) {
    setInt8(newCount, 20);
  }

  uint32_t UUID_TrackFragmentReference::getFragmentCount() {
    return getInt8(20);
  }

  void UUID_TrackFragmentReference::setTime(size_t num, uint64_t newTime) {
    if (getVersion() == 0) {
      setInt32(newTime, 21 + (num * 8));
    } else {
      setInt64(newTime, 21 + (num * 16));
    }
  }

  uint64_t UUID_TrackFragmentReference::getTime(size_t num) {
    if (getVersion() == 0) {
      return getInt32(21 + (num * 8));
    } else {
      return getInt64(21 + (num * 16));
    }
  }

  void UUID_TrackFragmentReference::setDuration(size_t num, uint64_t newDuration) {
    if (getVersion() == 0) {
      setInt32(newDuration, 21 + (num * 8) + 4);
    } else {
      setInt64(newDuration, 21 + (num * 16) + 8);
    }
  }

  uint64_t UUID_TrackFragmentReference::getDuration(size_t num) {
    if (getVersion() == 0) {
      return getInt32(21 + (num * 8) + 4);
    } else {
      return getInt64(21 + (num * 16) + 8);
    }
  }

  std::string UUID_TrackFragmentReference::toPrettyString(uint32_t indent) {
    std::stringstream r;
    r << std::string(indent, ' ') << "[d4807ef2-ca39-4695-8e54-26cb9e46a79f] Track Fragment Reference (" << boxedSize() << ")" << std::endl;
    r << std::string(indent + 1, ' ') << "Version: " << getVersion() << std::endl;
    r << std::string(indent + 1, ' ') << "Fragments: " << getFragmentCount() << std::endl;
    int j = getFragmentCount();
    for (int i = 0; i < j; ++i) {
      r << std::string(indent + 2, ' ') << "[" << i << "] Time = " << getTime(i) << ", Duration = " << getDuration(i) << std::endl;
    }
    return r.str();
  }
}