#include "DataSpec/DNACommon/WPSC.hpp"

#include "DataSpec/DNACommon/PAK.hpp"

#include <logvisor/logvisor.hpp>

namespace DataSpec::DNAParticle {

template <class IDType>
void WPSM<IDType>::_read(athena::io::YAMLDocReader& r) {
  for (const auto& elem : r.getCurNode()->m_mapChildren) {
    if (elem.first.size() < 4) {
      LogModule.report(logvisor::Warning, fmt("short FourCC in element '{}'"), elem.first);
      continue;
    }

    if (auto rec = r.enterSubRecord(elem.first.c_str())) {
      switch (*reinterpret_cast<const uint32_t*>(elem.first.data())) {
      case SBIG('IORN'):
        x0_IORN.read(r);
        break;
      case SBIG('IVEC'):
        x4_IVEC.read(r);
        break;
      case SBIG('PSOV'):
        x8_PSOV.read(r);
        break;
      case SBIG('PSVM'):
        xc_PSVM.read(r);
        break;
      case SBIG('VMD2'):
        x10_VMD2.read(r);
        break;
      case SBIG('PSLT'):
        x14_PSLT.read(r);
        break;
      case SBIG('PSCL'):
        x18_PSCL.read(r);
        break;
      case SBIG('PCOL'):
        x1c_PCOL.read(r);
        break;
      case SBIG('POFS'):
        x20_POFS.read(r);
        break;
      case SBIG('OFST'):
        x24_OFST.read(r);
        break;
      case SBIG('APSO'):
        x28_APSO.read(r);
        break;
      case SBIG('HOMG'):
        x29_HOMG.read(r);
        break;
      case SBIG('AP11'):
        x2a_AP11.read(r);
        break;
      case SBIG('AP21'):
        x2b_AP21.read(r);
        break;
      case SBIG('AS11'):
        x2c_AS11.read(r);
        break;
      case SBIG('AS12'):
        x2d_AS12.read(r);
        break;
      case SBIG('AS13'):
        x2e_AS13.read(r);
        break;
      case SBIG('TRAT'):
        x30_TRAT.read(r);
        break;
      case SBIG('APSM'):
        x34_APSM.read(r);
        break;
      case SBIG('APS2'):
        x44_APS2.read(r);
        break;
      case SBIG('ASW1'):
        x54_ASW1.read(r);
        break;
      case SBIG('ASW2'):
        x64_ASW2.read(r);
        break;
      case SBIG('ASW3'):
        x74_ASW3.read(r);
        break;
      case SBIG('OHEF'):
        x84_OHEF.read(r);
        break;
      case SBIG('COLR'):
        x94_COLR.read(r);
        break;
      case SBIG('EWTR'):
        xa4_EWTR.read(r);
        break;
      case SBIG('LWTR'):
        xa5_LWTR.read(r);
        break;
      case SBIG('SWTR'):
        xa6_SWTR.read(r);
        break;
      case SBIG('PJFX'):
        xa8_PJFX = r.readUint32(nullptr);
        break;
      case SBIG('RNGE'):
        xac_RNGE.read(r);
        break;
      case SBIG('FOFF'):
        xb0_FOFF.read(r);
        break;
      case SBIG('FC60'):
        xunk_FC60.read(r);
        break;
      case SBIG('SPS1'):
        xunk_SPS1.read(r);
        break;
      case SBIG('SPS2'):
        xunk_SPS2.read(r);
        break;
      }
    }
  }
}

template <class IDType>
void WPSM<IDType>::_write(athena::io::YAMLDocWriter& w) const {
  if (x0_IORN)
    if (auto rec = w.enterSubRecord("IORN"))
      x0_IORN.write(w);
  if (x4_IVEC)
    if (auto rec = w.enterSubRecord("IVEC"))
      x4_IVEC.write(w);
  if (x8_PSOV)
    if (auto rec = w.enterSubRecord("PSOV"))
      x8_PSOV.write(w);
  if (xc_PSVM)
    if (auto rec = w.enterSubRecord("PSVM"))
      xc_PSVM.write(w);
  if (x10_VMD2)
    if (auto rec = w.enterSubRecord("VMD2"))
      x10_VMD2.write(w);
  if (x14_PSLT)
    if (auto rec = w.enterSubRecord("PSLT"))
      x14_PSLT.write(w);
  if (x18_PSCL)
    if (auto rec = w.enterSubRecord("PSCL"))
      x18_PSCL.write(w);
  if (x1c_PCOL)
    if (auto rec = w.enterSubRecord("PCOL"))
      x1c_PCOL.write(w);
  if (x20_POFS)
    if (auto rec = w.enterSubRecord("POFS"))
      x20_POFS.write(w);
  if (x24_OFST)
    if (auto rec = w.enterSubRecord("OFST"))
      x24_OFST.write(w);
  if (x28_APSO)
    if (auto rec = w.enterSubRecord("APSO"))
      x28_APSO.write(w);
  if (x29_HOMG)
    if (auto rec = w.enterSubRecord("HOMG"))
      x29_HOMG.write(w);
  if (x2a_AP11)
    if (auto rec = w.enterSubRecord("AP11"))
      x2a_AP11.write(w);
  if (x2b_AP21)
    if (auto rec = w.enterSubRecord("AP21"))
      x2b_AP21.write(w);
  if (x2c_AS11)
    if (auto rec = w.enterSubRecord("AS11"))
      x2c_AS11.write(w);
  if (x2d_AS12)
    if (auto rec = w.enterSubRecord("AS12"))
      x2d_AS12.write(w);
  if (x2e_AS13)
    if (auto rec = w.enterSubRecord("AS13"))
      x2e_AS13.write(w);
  if (x30_TRAT)
    if (auto rec = w.enterSubRecord("TRAT"))
      x30_TRAT.write(w);
  if (x34_APSM)
    if (auto rec = w.enterSubRecord("APSM"))
      x34_APSM.write(w);
  if (x44_APS2)
    if (auto rec = w.enterSubRecord("APS2"))
      x44_APS2.write(w);
  if (x54_ASW1)
    if (auto rec = w.enterSubRecord("ASW1"))
      x54_ASW1.write(w);
  if (x64_ASW2)
    if (auto rec = w.enterSubRecord("ASW2"))
      x64_ASW2.write(w);
  if (x74_ASW3)
    if (auto rec = w.enterSubRecord("ASW3"))
      x74_ASW3.write(w);
  if (x84_OHEF)
    if (auto rec = w.enterSubRecord("OHEF"))
      x84_OHEF.write(w);
  if (x94_COLR)
    if (auto rec = w.enterSubRecord("COLR"))
      x94_COLR.write(w);
  if (!xa4_EWTR)
    if (auto rec = w.enterSubRecord("EWTR"))
      xa4_EWTR.write(w);
  if (!xa5_LWTR)
    if (auto rec = w.enterSubRecord("LWTR"))
      xa5_LWTR.write(w);
  if (!xa6_SWTR)
    if (auto rec = w.enterSubRecord("SWTR"))
      xa6_SWTR.write(w);
  if (xa8_PJFX != UINT32_MAX)
    w.writeUint32("PJFX", xa8_PJFX);
  if (xac_RNGE)
    if (auto rec = w.enterSubRecord("RNGE"))
      xac_RNGE.write(w);
  if (xb0_FOFF)
    if (auto rec = w.enterSubRecord("FOFF"))
      xb0_FOFF.write(w);
  if (xunk_FC60)
    if (auto rec = w.enterSubRecord("FC60"))
      xunk_FC60.write(w);
  if (xunk_SPS1)
    if (auto rec = w.enterSubRecord("SPS1"))
      xunk_SPS1.write(w);
  if (xunk_SPS1)
    if (auto rec = w.enterSubRecord("SPS2"))
      xunk_SPS2.write(w);
}

template <class IDType>
void WPSM<IDType>::_binarySize(size_t& __isz) const {
  __isz += 4;
  if (x0_IORN) {
    __isz += 4;
    x0_IORN.binarySize(__isz);
  }
  if (x4_IVEC) {
    __isz += 4;
    x4_IVEC.binarySize(__isz);
  }
  if (x8_PSOV) {
    __isz += 4;
    x8_PSOV.binarySize(__isz);
  }
  if (xc_PSVM) {
    __isz += 4;
    xc_PSVM.binarySize(__isz);
  }
  if (x10_VMD2) {
    __isz += 4;
    x10_VMD2.binarySize(__isz);
  }
  if (x14_PSLT) {
    __isz += 4;
    x14_PSLT.binarySize(__isz);
  }
  if (x18_PSCL) {
    __isz += 4;
    x18_PSCL.binarySize(__isz);
  }
  if (x1c_PCOL) {
    __isz += 4;
    x1c_PCOL.binarySize(__isz);
  }
  if (x20_POFS) {
    __isz += 4;
    x20_POFS.binarySize(__isz);
  }
  if (x24_OFST) {
    __isz += 4;
    x24_OFST.binarySize(__isz);
  }
  if (x28_APSO) {
    __isz += 4;
    x28_APSO.binarySize(__isz);
  }
  if (x29_HOMG) {
    __isz += 4;
    x29_HOMG.binarySize(__isz);
  }
  if (x2a_AP11) {
    __isz += 4;
    x2a_AP11.binarySize(__isz);
  }
  if (x2b_AP21) {
    __isz += 4;
    x2b_AP21.binarySize(__isz);
  }
  if (x2c_AS11) {
    __isz += 4;
    x2c_AS11.binarySize(__isz);
  }
  if (x2d_AS12) {
    __isz += 4;
    x2d_AS12.binarySize(__isz);
  }
  if (x2e_AS13) {
    __isz += 4;
    x2e_AS13.binarySize(__isz);
  }
  if (x30_TRAT) {
    __isz += 4;
    x30_TRAT.binarySize(__isz);
  }
  if (x34_APSM) {
    __isz += 4;
    x34_APSM.binarySize(__isz);
  }
  if (x44_APS2) {
    __isz += 4;
    x44_APS2.binarySize(__isz);
  }
  if (x54_ASW1) {
    __isz += 4;
    x54_ASW1.binarySize(__isz);
  }
  if (x64_ASW2) {
    __isz += 4;
    x64_ASW2.binarySize(__isz);
  }
  if (x74_ASW3) {
    __isz += 4;
    x74_ASW3.binarySize(__isz);
  }
  if (x84_OHEF) {
    __isz += 4;
    x84_OHEF.binarySize(__isz);
  }
  if (x94_COLR) {
    __isz += 4;
    x94_COLR.binarySize(__isz);
  }
  if (!xa4_EWTR) {
    __isz += 4;
    xa4_EWTR.binarySize(__isz);
  }
  if (!xa5_LWTR) {
    __isz += 4;
    xa5_LWTR.binarySize(__isz);
  }
  if (!xa6_SWTR) {
    __isz += 4;
    xa6_SWTR.binarySize(__isz);
  }
  if (xa8_PJFX != UINT32_MAX)
    __isz += 12;
  if (xac_RNGE) {
    __isz += 4;
    xac_RNGE.binarySize(__isz);
  }
  if (xb0_FOFF) {
    __isz += 4;
    xb0_FOFF.binarySize(__isz);
  }
  if (xunk_FC60) {
    __isz += 4;
    xunk_FC60.binarySize(__isz);
  }
  if (xunk_SPS1) {
    __isz += 4;
    xunk_SPS1.binarySize(__isz);
  }
  if (xunk_SPS2) {
    __isz += 4;
    xunk_SPS2.binarySize(__isz);
  }
}

template <class IDType>
void WPSM<IDType>::_read(athena::io::IStreamReader& r) {
  DNAFourCC clsId;
  clsId.read(r);
  if (clsId != SBIG('WPSM')) {
    LogModule.report(logvisor::Warning, fmt("non WPSM provided to WPSM parser"));
    return;
  }
  clsId.read(r);
  while (clsId != SBIG('_END')) {
    switch (clsId.toUint32()) {
    case SBIG('IORN'):
      x0_IORN.read(r);
      break;
    case SBIG('IVEC'):
      x4_IVEC.read(r);
      break;
    case SBIG('PSOV'):
      x8_PSOV.read(r);
      break;
    case SBIG('PSVM'):
      xc_PSVM.read(r);
      break;
    case SBIG('VMD2'):
      x10_VMD2.read(r);
      break;
    case SBIG('PSLT'):
      x14_PSLT.read(r);
      break;
    case SBIG('PSCL'):
      x18_PSCL.read(r);
      break;
    case SBIG('PCOL'):
      x1c_PCOL.read(r);
      break;
    case SBIG('POFS'):
      x20_POFS.read(r);
      break;
    case SBIG('OFST'):
      x24_OFST.read(r);
      break;
    case SBIG('APSO'):
      r.readUint32();
      x28_APSO = r.readBool();
      break;
    case SBIG('HOMG'):
      x29_HOMG.read(r);
      break;
    case SBIG('AP11'):
      x2a_AP11.read(r);
      break;
    case SBIG('AP21'):
      x2b_AP21.read(r);
      break;
    case SBIG('AS11'):
      x2c_AS11.read(r);
      break;
    case SBIG('AS12'):
      x2d_AS12.read(r);
      break;
    case SBIG('AS13'):
      x2e_AS13.read(r);
      break;
    case SBIG('TRAT'):
      x30_TRAT.read(r);
      break;
    case SBIG('APSM'):
      x34_APSM.read(r);
      break;
    case SBIG('APS2'):
      x44_APS2.read(r);
      break;
    case SBIG('ASW1'):
      x54_ASW1.read(r);
      break;
    case SBIG('ASW2'):
      x64_ASW2.read(r);
      break;
    case SBIG('ASW3'):
      x74_ASW3.read(r);
      break;
    case SBIG('OHEF'):
      x84_OHEF.read(r);
      break;
    case SBIG('COLR'):
      x94_COLR.read(r);
      break;
    case SBIG('EWTR'):
      r.readUint32();
      xa4_EWTR = r.readBool();
      break;
    case SBIG('LWTR'):
      r.readUint32();
      xa5_LWTR = r.readBool();
      break;
    case SBIG('SWTR'):
      r.readUint32();
      xa6_SWTR = r.readBool();
      break;
    case SBIG('PJFX'): {
      uint32_t fcc;
      r.readBytesToBuf(&fcc, 4);
      if (fcc != SBIG('NONE'))
        xa8_PJFX = r.readUint32Big();
    } break;
    case SBIG('RNGE'):
      xac_RNGE.read(r);
      break;
    case SBIG('FOFF'):
      xb0_FOFF.read(r);
      break;
    case SBIG('FC60'):
      xunk_FC60.read(r);
      break;
    case SBIG('SPS1'):
      xunk_SPS1.read(r);
      break;
    case SBIG('SPS2'):
      xunk_SPS2.read(r);
      break;
    default:
      LogModule.report(logvisor::Fatal, fmt("Unknown WPSM class {} @{}"), clsId, r.position());
      break;
    }
    clsId.read(r);
  }
}

template <class IDType>
void WPSM<IDType>::_write(athena::io::IStreamWriter& w) const {
  w.writeBytes("WPSM", 4);
  if (x0_IORN) {
    w.writeBytes("IORN", 4);
    x0_IORN.write(w);
  }
  if (x4_IVEC) {
    w.writeBytes("IVEC", 4);
    x4_IVEC.write(w);
  }
  if (x8_PSOV) {
    w.writeBytes("PSOV", 4);
    x8_PSOV.write(w);
  }
  if (xc_PSVM) {
    w.writeBytes("PSVM", 4);
    xc_PSVM.write(w);
  }
  if (x10_VMD2) {
    w.writeBytes("VMD2", 4);
    x10_VMD2.write(w);
  }
  if (x14_PSLT) {
    w.writeBytes("PSLT", 4);
    x14_PSLT.write(w);
  }
  if (x18_PSCL) {
    w.writeBytes("PSCL", 4);
    x18_PSCL.write(w);
  }
  if (x1c_PCOL) {
    w.writeBytes("PCOL", 4);
    x1c_PCOL.write(w);
  }
  if (x20_POFS) {
    w.writeBytes("POFS", 4);
    x20_POFS.write(w);
  }
  if (x24_OFST) {
    w.writeBytes("OFST", 4);
    x24_OFST.write(w);
  }
  if (x28_APSO) {
    w.writeBytes("APSO", 4);
    x28_APSO.write(w);
  }
  if (x29_HOMG) {
    w.writeBytes("HOMG", 4);
    x29_HOMG.write(w);
  }
  if (x2a_AP11) {
    w.writeBytes("AP11", 4);
    x2a_AP11.write(w);
  }
  if (x2b_AP21) {
    w.writeBytes("AP21", 4);
    x2b_AP21.write(w);
  }
  if (x2c_AS11) {
    w.writeBytes("AS11", 4);
    x2c_AS11.write(w);
  }
  if (x2d_AS12) {
    w.writeBytes("AS12", 4);
    x2d_AS12.write(w);
  }
  if (x2e_AS13) {
    w.writeBytes("AS13", 4);
    x2e_AS13.write(w);
  }
  if (x30_TRAT) {
    w.writeBytes("TRAT", 4);
    x30_TRAT.write(w);
  }
  if (x34_APSM) {
    w.writeBytes("APSM", 4);
    x34_APSM.write(w);
  }
  if (x44_APS2) {
    w.writeBytes("APS2", 4);
    x44_APS2.write(w);
  }
  if (x54_ASW1) {
    w.writeBytes("ASW1", 4);
    x54_ASW1.write(w);
  }
  if (x64_ASW2) {
    w.writeBytes("ASW2", 4);
    x64_ASW2.write(w);
  }
  if (x74_ASW3) {
    w.writeBytes("ASW3", 4);
    x74_ASW3.write(w);
  }
  if (x84_OHEF) {
    w.writeBytes("OHEF", 4);
    x84_OHEF.write(w);
  }
  if (x94_COLR) {
    w.writeBytes("COLR", 4);
    x94_COLR.write(w);
  }
  if (!xa4_EWTR) {
    w.writeBytes("EWTR", 4);
    xa4_EWTR.write(w);
  }
  if (!xa5_LWTR) {
    w.writeBytes("LWTR", 4);
    xa5_LWTR.write(w);
  }
  if (!xa6_SWTR) {
    w.writeBytes("SWTR", 4);
    xa6_SWTR.write(w);
  }
  if (xa8_PJFX != UINT32_MAX) {
    w.writeBytes("PJFXCNST", 8);
    w.writeUint32(xa8_PJFX);
  }
  if (xac_RNGE) {
    w.writeBytes("RNGE", 4);
    xac_RNGE.write(w);
  }
  if (xb0_FOFF) {
    w.writeBytes("FOFF", 4);
    xb0_FOFF.write(w);
  }
  if (xunk_FC60) {
    w.writeBytes("FC60", 4);
    xunk_FC60.write(w);
  }
  if (xunk_SPS1) {
    w.writeBytes("SPS1", 4);
    xunk_SPS1.write(w);
  }
  if (xunk_SPS2) {
    w.writeBytes("SPS2", 4);
    xunk_SPS2.write(w);
  }

  w.writeBytes("_END", 4);
}

AT_SUBSPECIALIZE_DNA_YAML(WPSM<UniqueID32>)
AT_SUBSPECIALIZE_DNA_YAML(WPSM<UniqueID64>)

template <>
const char* WPSM<UniqueID32>::DNAType() {
  return "WPSM<UniqueID32>";
}

template <>
const char* WPSM<UniqueID64>::DNAType() {
  return "WPSM<UniqueID64>";
}

template <class IDType>
void WPSM<IDType>::gatherDependencies(std::vector<hecl::ProjectPath>& pathsOut) const {
  g_curSpec->flattenDependencies(x34_APSM.id, pathsOut);
  g_curSpec->flattenDependencies(x44_APS2.id, pathsOut);
  g_curSpec->flattenDependencies(x54_ASW1.id, pathsOut);
  g_curSpec->flattenDependencies(x64_ASW2.id, pathsOut);
  g_curSpec->flattenDependencies(x74_ASW3.id, pathsOut);
  g_curSpec->flattenDependencies(x84_OHEF.id, pathsOut);
  g_curSpec->flattenDependencies(x94_COLR.id, pathsOut);
}

template struct WPSM<UniqueID32>;
template struct WPSM<UniqueID64>;

template <class IDType>
bool ExtractWPSM(PAKEntryReadStream& rs, const hecl::ProjectPath& outPath) {
  athena::io::FileWriter writer(outPath.getAbsolutePath());
  if (writer.isOpen()) {
    WPSM<IDType> wpsm;
    wpsm.read(rs);
    athena::io::ToYAMLStream(wpsm, writer);
    return true;
  }
  return false;
}
template bool ExtractWPSM<UniqueID32>(PAKEntryReadStream& rs, const hecl::ProjectPath& outPath);
template bool ExtractWPSM<UniqueID64>(PAKEntryReadStream& rs, const hecl::ProjectPath& outPath);

template <class IDType>
bool WriteWPSM(const WPSM<IDType>& wpsm, const hecl::ProjectPath& outPath) {
  athena::io::FileWriter w(outPath.getAbsolutePath(), true, false);
  if (w.hasError())
    return false;
  wpsm.write(w);
  int64_t rem = w.position() % 32;
  if (rem)
    for (int64_t i = 0; i < 32 - rem; ++i)
      w.writeUByte(0xff);
  return true;
}
template bool WriteWPSM<UniqueID32>(const WPSM<UniqueID32>& wpsm, const hecl::ProjectPath& outPath);
template bool WriteWPSM<UniqueID64>(const WPSM<UniqueID64>& wpsm, const hecl::ProjectPath& outPath);

} // namespace DataSpec::DNAParticle