#include "CPakFile.hpp"

namespace urde
{
static logvisor::Module Log("urde::CPakFile");

CPakFile::CPakFile(std::string_view filename, bool buildDepList, bool worldPak)
: CDvdFile(filename)
{
    if (!CDvdFile::operator bool())
        Log.report(logvisor::Fatal, "%s: Unable to open", GetPath().data());
    x28_24_buildDepList = buildDepList;
    x28_24_buildDepList = true; // Always do this so URDE can rapidly pre-warm shaders
    x28_26_worldPak = worldPak;
    x28_27_stashedInARAM = false;
}

const SObjectTag* CPakFile::GetResIdByName(std::string_view name) const
{
    for (const std::pair<std::string, SObjectTag>& p : x54_nameList)
        if (!CStringExtras::CompareCaseInsensitive(p.first.c_str(), name))
            return &p.second;
    return nullptr;
}

void CPakFile::LoadResourceTable(athena::io::MemoryReader& r)
{
    x74_resList.reserve(std::max(size_t(64), size_t(ROUND_UP_32(x4c_resTableCount * sizeof(SResInfo)) +
        sizeof(SResInfo) - 1)) / sizeof(SResInfo));
    if (x28_24_buildDepList)
        x64_depList.reserve(x4c_resTableCount);
    for (u32 i=0 ; i<x4c_resTableCount ; ++i)
    {
        u32 flags = r.readUint32Big();
        FourCC fcc;
        r.readBytesToBuf(&fcc, 4);
        CAssetId id = r.readUint32Big();
        u32 size = r.readUint32Big();
        u32 offset = r.readUint32Big();
        x74_resList.emplace_back(id, fcc, offset, size, flags);
        if (x28_24_buildDepList)
            x64_depList.push_back(id);
    }
    std::sort(x74_resList.begin(), x74_resList.end(), [](const auto& a, const auto& b) { return a.x0_id < b.x0_id; });
}

void CPakFile::DataLoad()
{
    x30_dvdReq.reset();
    athena::io::MemoryReader r(x38_headerData.data() + x48_resTableOffset, x38_headerData.size() - x48_resTableOffset);
    LoadResourceTable(r);
    x2c_asyncLoadPhase = EAsyncPhase::Loaded;
    if (x28_26_worldPak)
    {
        // Allocate ARAM space DMA x74_resList to ARAM
    }
    x38_headerData.clear();
}

void CPakFile::InitialHeaderLoad()
{
    athena::io::MemoryReader r(x38_headerData.data(), x38_headerData.size());
    x30_dvdReq.reset();
    u32 version = r.readUint32Big();
    if (version != 0x80030005)
    {
        Log.report(logvisor::Fatal, "%s: Incompatible pak file version -- Current version is %x, you're using %x",
                   GetPath().data(), 0x80030005, version);
        return;
    }

    r.readUint32Big();
    u32 nameCount = r.readUint32Big();
    x54_nameList.reserve(nameCount);
    for (u32 i=0 ; i<nameCount ; ++i)
    {
        SObjectTag tag(r);
        u32 nameLen = r.readUint32Big();
        auto name = r.readString(nameLen);
        x54_nameList.emplace_back(name, tag);
    }

    x4c_resTableCount = r.readUint32Big();
    x48_resTableOffset = u32(r.position());
    x2c_asyncLoadPhase = EAsyncPhase::DataLoad;
    u32 newSize = ROUND_UP_32(x4c_resTableCount * 20 + x48_resTableOffset);
    u32 origSize = u32(x38_headerData.size());
    if (newSize > origSize)
    {
        x38_headerData.resize(newSize);
        x30_dvdReq = AsyncSeekRead(x38_headerData.data() + origSize, u32(x38_headerData.size() - origSize),
                                   ESeekOrigin::Begin, origSize);
    }
    else
    {
        DataLoad();
    }
};

void CPakFile::Warmup()
{
    u32 length = std::min(u32(Length()), u32(8192));
    x38_headerData.resize(length);
    x30_dvdReq = AsyncSeekRead(x38_headerData.data(), length, ESeekOrigin::Cur, 0);
    x2c_asyncLoadPhase = EAsyncPhase::InitialHeader;
}

const CPakFile::SResInfo* CPakFile::GetResInfoForLoadPreferForward(CAssetId id) const
{
    if (x28_27_stashedInARAM)
        return nullptr;
    auto search = rstl::binary_find(x74_resList.begin(), x74_resList.end(), id,
                  [](const SResInfo& test) { return test.x0_id; });
    if (search == x74_resList.end())
        return nullptr;
    const SResInfo* bestInfo = &*search;
    s32 bestDelta = x84_currentSeek - bestInfo->GetOffset();
    while (++search != x74_resList.end())
    {
        const SResInfo* thisInfo = &*search;
        if (thisInfo->x0_id != id)
            break;
        s32 thisDelta = x84_currentSeek - bestInfo->GetOffset();
        if ((bestDelta < 0 && (thisDelta > 0 || thisDelta > bestDelta)) ||
            (bestDelta >= 0 && thisDelta > 0 && thisDelta < bestDelta))
        {
            bestDelta = thisDelta;
            bestInfo = thisInfo;
        }
    }
    x84_currentSeek = bestInfo->GetOffset() + bestInfo->GetSize();
    return bestInfo;
}

const CPakFile::SResInfo* CPakFile::GetResInfoForLoadDirectionless(CAssetId id) const
{
    if (x28_27_stashedInARAM)
        return nullptr;
    auto search = rstl::binary_find(x74_resList.begin(), x74_resList.end(), id,
                  [](const SResInfo& test) { return test.x0_id; });
    if (search == x74_resList.end())
        return nullptr;
    const SResInfo* bestInfo = &*search;
    s32 bestDelta = std::abs(s32(x84_currentSeek - bestInfo->GetOffset()));
    while (++search != x74_resList.end())
    {
        const SResInfo* thisInfo = &*search;
        if (thisInfo->x0_id != id)
            break;
        s32 thisDelta = std::abs(s32(x84_currentSeek - bestInfo->GetOffset()));
        if (thisDelta < bestDelta)
        {
            bestDelta = thisDelta;
            bestInfo = thisInfo;
        }
    }
    x84_currentSeek = bestInfo->GetOffset() + bestInfo->GetSize();
    return bestInfo;
}

const CPakFile::SResInfo* CPakFile::GetResInfo(CAssetId id) const
{
    if (x2c_asyncLoadPhase != EAsyncPhase::Loaded)
        return nullptr;
    if (x28_27_stashedInARAM)
        return nullptr;
    auto search = rstl::binary_find(x74_resList.begin(), x74_resList.end(), id,
                                    [](const SResInfo& i) { return i.x0_id; });
    if (search == x74_resList.end())
        return nullptr;
    return &*search;
}

void CPakFile::AsyncIdle()
{
    if (x2c_asyncLoadPhase == EAsyncPhase::Loaded)
        return;
    if (x30_dvdReq && !x30_dvdReq->IsComplete())
        return;
    switch (x2c_asyncLoadPhase)
    {
    case EAsyncPhase::Warmup:
        Warmup();
        break;
    case EAsyncPhase::InitialHeader:
        InitialHeaderLoad();
        break;
    case EAsyncPhase::DataLoad:
        DataLoad();
        break;
    default: break;
    }
}

}