#include "CAnimSourceReader.hpp"
#include "CBoolPOINode.hpp"
#include "CInt32POINode.hpp"
#include "CParticlePOINode.hpp"
#include "CSoundPOINode.hpp"
#include "CFBStreamedAnimReader.hpp"

namespace urde
{

CAnimSourceInfo::CAnimSourceInfo(const TSubAnimTypeToken<CAnimSource>& token)
: x4_token(token) {}

bool CAnimSourceInfo::HasPOIData() const
{
    return x4_token->x58_evntData;
}

const std::vector<CBoolPOINode>& CAnimSourceInfo::GetBoolPOIStream() const
{
    return x4_token->GetBoolPOIStream();
}

const std::vector<CInt32POINode>& CAnimSourceInfo::GetInt32POIStream() const
{
    return x4_token->GetInt32POIStream();
}

const std::vector<CParticlePOINode>& CAnimSourceInfo::GetParticlePOIStream() const
{
    return x4_token->GetParticlePOIStream();
}

const std::vector<CSoundPOINode>& CAnimSourceInfo::GetSoundPOIStream() const
{
    return x4_token->GetSoundPOIStream();
}

CCharAnimTime CAnimSourceInfo::GetAnimationDuration() const
{
    return x4_token->GetDuration();
}

std::map<std::string, CParticleData::EParentedMode>
CAnimSourceReaderBase::GetUniqueParticlePOIs() const
{
    const std::vector<CParticlePOINode>& particleNodes = x4_sourceInfo->GetParticlePOIStream();
    std::map<std::string, CParticleData::EParentedMode> ret;
    for (const CParticlePOINode& node : particleNodes)
        ret[std::string(node.GetString())] = node.GetParticleData().GetParentedMode();
    return ret;
}

std::map<std::string, s32>
CAnimSourceReaderBase::GetUniqueInt32POIs() const
{
    const std::vector<CInt32POINode>& int32Nodes = x4_sourceInfo->GetInt32POIStream();
    std::map<std::string, s32> ret;
    for (const CInt32POINode& node : int32Nodes)
        ret[std::string(node.GetString())] = node.GetValue();
    return ret;
}

std::map<std::string, bool>
CAnimSourceReaderBase::GetUniqueBoolPOIs() const
{
    const std::vector<CBoolPOINode>& boolNodes = x4_sourceInfo->GetBoolPOIStream();
    std::map<std::string, bool> ret;
    for (const CBoolPOINode& node : boolNodes)
        ret[std::string(node.GetString())] = node.GetValue();
    return ret;
}

void CAnimSourceReaderBase::PostConstruct(const CCharAnimTime& time)
{
    x14_passedBoolCount = 0;
    x18_passedIntCount = 0;
    x1c_passedParticleCount = 0;
    x20_passedSoundCount = 0;

    if (x4_sourceInfo->HasPOIData())
    {
        std::map<std::string, bool> boolPOIs = GetUniqueBoolPOIs();
        std::map<std::string, s32> int32POIs = GetUniqueInt32POIs();
        std::map<std::string, CParticleData::EParentedMode> particlePOIs = GetUniqueParticlePOIs();

        x24_boolStates.reserve(boolPOIs.size());
        for (const auto& poi : boolPOIs)
            x24_boolStates.push_back(poi);

        x34_int32States.reserve(int32POIs.size());
        for (const auto& poi : int32POIs)
            x34_int32States.push_back(poi);

        x44_particleStates.reserve(particlePOIs.size());
        for (const auto& poi : particlePOIs)
            x44_particleStates.push_back(poi);
    }

    CCharAnimTime tmpTime = time;
    if (tmpTime.GreaterThanZero())
    {
        while (tmpTime.GreaterThanZero())
        {
            SAdvancementResults res = VAdvanceView(tmpTime);
            tmpTime = res.x0_remTime;
        }
    }
    else if (x4_sourceInfo->HasPOIData())
    {
        UpdatePOIStates();
        if (!time.GreaterThanZero())
        {
            x14_passedBoolCount = 0;
            x18_passedIntCount = 0;
            x1c_passedParticleCount = 0;
            x20_passedSoundCount = 0;
        }
    }
}

void CAnimSourceReaderBase::UpdatePOIStates()
{
    const std::vector<CBoolPOINode>& boolNodes = x4_sourceInfo->GetBoolPOIStream();
    const std::vector<CInt32POINode>& int32Nodes = x4_sourceInfo->GetInt32POIStream();
    const std::vector<CParticlePOINode>& particleNodes = x4_sourceInfo->GetParticlePOIStream();
    const std::vector<CSoundPOINode>& soundNodes = x4_sourceInfo->GetSoundPOIStream();

    for (const CBoolPOINode& node : boolNodes)
    {
        if (node.GetTime() > xc_curTime)
            break;
        if (node.GetIndex() != -1)
            x24_boolStates[node.GetIndex()].second = node.GetValue();
        ++x14_passedBoolCount;
    }

    for (const CInt32POINode& node : int32Nodes)
    {
        if (node.GetTime() > xc_curTime)
            break;
        if (node.GetIndex() != -1)
            x34_int32States[node.GetIndex()].second = node.GetValue();
        ++x18_passedIntCount;
    }

    for (const CParticlePOINode& node : particleNodes)
    {
        if (node.GetTime() > xc_curTime)
            break;
        if (node.GetIndex() != -1)
            x44_particleStates[node.GetIndex()].second = node.GetParticleData().GetParentedMode();
        ++x1c_passedParticleCount;
    }

    for (const CSoundPOINode& node : soundNodes)
    {
        if (node.GetTime() > xc_curTime)
            break;
        ++x20_passedSoundCount;
    }
}

u32 CAnimSourceReaderBase::VGetBoolPOIList(const CCharAnimTime& time,
                                           CBoolPOINode* listOut,
                                           u32 capacity, u32 iterator, u32 unk) const
{
    if (x4_sourceInfo->HasPOIData())
    {
        const std::vector<CBoolPOINode>& boolNodes = x4_sourceInfo->GetBoolPOIStream();
        return _getPOIList(time, listOut, capacity, iterator, unk, boolNodes,
                           xc_curTime, *x4_sourceInfo, x14_passedBoolCount);
    }
    return 0;
}

u32 CAnimSourceReaderBase::VGetInt32POIList(const CCharAnimTime& time,
                                            CInt32POINode* listOut,
                                            u32 capacity, u32 iterator, u32 unk) const
{
    if (x4_sourceInfo->HasPOIData())
    {
        const std::vector<CInt32POINode>& int32Nodes = x4_sourceInfo->GetInt32POIStream();
        return _getPOIList(time, listOut, capacity, iterator, unk, int32Nodes,
                           xc_curTime, *x4_sourceInfo, x18_passedIntCount);
    }
    return 0;
}

u32 CAnimSourceReaderBase::VGetParticlePOIList(const CCharAnimTime& time,
                                               CParticlePOINode* listOut,
                                               u32 capacity, u32 iterator, u32 unk) const
{
    if (x4_sourceInfo->HasPOIData())
    {
        const std::vector<CParticlePOINode>& particleNodes = x4_sourceInfo->GetParticlePOIStream();
        return _getPOIList(time, listOut, capacity, iterator, unk, particleNodes,
                           xc_curTime, *x4_sourceInfo, x1c_passedParticleCount);
    }
    return 0;
}

u32 CAnimSourceReaderBase::VGetSoundPOIList(const CCharAnimTime& time,
                                            CSoundPOINode* listOut,
                                            u32 capacity, u32 iterator, u32 unk) const
{
    if (x4_sourceInfo->HasPOIData())
    {
        const std::vector<CSoundPOINode>& soundNodes = x4_sourceInfo->GetSoundPOIStream();
        return _getPOIList(time, listOut, capacity, iterator, unk, soundNodes,
                           xc_curTime, *x4_sourceInfo, x20_passedSoundCount);
    }
    return 0;
}

bool CAnimSourceReaderBase::VGetBoolPOIState(const char* name) const
{
    for (const auto& node : x24_boolStates)
        if (!node.first.compare(name))
            return node.second;
    return false;
}

s32 CAnimSourceReaderBase::VGetInt32POIState(const char* name) const
{
    for (const auto& node : x34_int32States)
        if (!node.first.compare(name))
            return node.second;
    return 0;
}

CParticleData::EParentedMode
CAnimSourceReaderBase::VGetParticlePOIState(const char* name) const
{
    for (const auto& node : x44_particleStates)
        if (!node.first.compare(name))
            return node.second;
    return CParticleData::EParentedMode::Initial;
}

CAnimSourceReaderBase::CAnimSourceReaderBase(std::unique_ptr<IAnimSourceInfo>&& sourceInfo,
                                             const CCharAnimTime& time)
: x4_sourceInfo(std::move(sourceInfo)), xc_curTime(time) {}

CAnimSourceReaderBase::CAnimSourceReaderBase(std::unique_ptr<IAnimSourceInfo>&& sourceInfo,
                                             const CAnimSourceReaderBase& other)
: x4_sourceInfo(std::move(sourceInfo)),
  xc_curTime(other.xc_curTime),
  x14_passedBoolCount(other.x14_passedBoolCount),
  x18_passedIntCount(other.x18_passedIntCount),
  x1c_passedParticleCount(other.x1c_passedParticleCount),
  x20_passedSoundCount(other.x20_passedSoundCount),
  x24_boolStates(other.x24_boolStates),
  x34_int32States(other.x34_int32States),
  x44_particleStates(other.x44_particleStates)
{}

SAdvancementResults CAnimSourceReader::VGetAdvancementResults(const CCharAnimTime& dt,
                                                              const CCharAnimTime& startOff) const
{
    SAdvancementResults ret;
    CCharAnimTime accum = xc_curTime + startOff;

    if (xc_curTime + startOff >= x54_source->GetDuration())
    {
        ret.x0_remTime = dt;
        return ret;
    }
    else if (dt.EqualsZero())
    {
        return ret;
    }
    else
    {
        CCharAnimTime prevTime = accum;
        accum += dt;
        CCharAnimTime remTime;
        if (accum > x54_source->GetDuration())
        {
            remTime = accum - x54_source->GetDuration();
            accum = x54_source->GetDuration();
        }

        zeus::CQuaternion ra = x54_source->GetRotation(3, prevTime).inverse();
        zeus::CQuaternion rb = x54_source->GetRotation(3, accum);
        ret.x0_remTime = remTime;
        ret.x8_deltas.xc_rotDelta = rb * ra;

        if (x54_source->HasOffset(3))
        {
            zeus::CVector3f ta = x54_source->GetOffset(3, prevTime);
            zeus::CVector3f tb = x54_source->GetOffset(3, accum);
            ret.x8_deltas.x0_posDelta = zeus::CMatrix3f(rb) * (tb - ta);
        }

        return ret;
    }
}

void CAnimSourceReader::VSetPhase(float phase)
{
    xc_curTime = phase * x54_source->GetDuration().GetSeconds();
    if (x54_source->GetPOIData())
    {
        UpdatePOIStates();
        if (!xc_curTime.GreaterThanZero())
        {
            x14_passedBoolCount = 0;
            x18_passedIntCount = 0;
            x1c_passedParticleCount = 0;
            x20_passedSoundCount = 0;
        }
    }
}

SAdvancementResults CAnimSourceReader::VReverseView(const CCharAnimTime& dt)
{
    SAdvancementResults ret;

    if (xc_curTime.EqualsZero())
    {
        ret.x0_remTime = dt;
        return ret;
    }
    else if (dt.EqualsZero())
    {
        return ret;
    }
    else
    {
        CCharAnimTime prevTime = xc_curTime;
        xc_curTime -= dt;
        CCharAnimTime remTime;
        if (xc_curTime < CCharAnimTime())
        {
            remTime = CCharAnimTime() - xc_curTime;
            xc_curTime = CCharAnimTime();
        }

        if (x54_source->GetPOIData())
            UpdatePOIStates();

        zeus::CQuaternion ra = x54_source->GetRotation(3, prevTime).inverse();
        zeus::CQuaternion rb = x54_source->GetRotation(3, xc_curTime);
        ret.x0_remTime = remTime;
        ret.x8_deltas.xc_rotDelta = rb * ra;

        if (x54_source->HasOffset(3))
        {
            zeus::CVector3f ta = x54_source->GetOffset(3, prevTime);
            zeus::CVector3f tb = x54_source->GetOffset(3, xc_curTime);
            ret.x8_deltas.x0_posDelta = zeus::CMatrix3f(rb) * (tb - ta);
        }

        return ret;
    }
}

std::unique_ptr<IAnimReader> CAnimSourceReader::VClone() const
{
    return std::make_unique<CAnimSourceReader>(*this);
}

void CAnimSourceReader::VGetSegStatementSet(const CSegIdList& list,
                                            CSegStatementSet& setOut) const
{
    x54_source->GetSegStatementSet(list, setOut, xc_curTime);
}

void CAnimSourceReader::VGetSegStatementSet(const CSegIdList& list,
                                            CSegStatementSet& setOut,
                                            const CCharAnimTime& time) const
{
    x54_source->GetSegStatementSet(list, setOut, time);
}

SAdvancementResults CAnimSourceReader::VAdvanceView(const CCharAnimTime& dt)
{
    SAdvancementResults ret;

    if (xc_curTime >= x54_source->GetDuration())
    {
        ret.x0_remTime = dt;
        return ret;
    }
    else if (dt.EqualsZero())
    {
        return ret;
    }
    else
    {
        CCharAnimTime prevTime = xc_curTime;
        xc_curTime += dt;
        CCharAnimTime remTime;
        if (xc_curTime > x54_source->GetDuration())
        {
            remTime = xc_curTime - x54_source->GetDuration();
            xc_curTime = x54_source->GetDuration();
        }

        if (x54_source->GetPOIData())
            UpdatePOIStates();

        zeus::CQuaternion ra = x54_source->GetRotation(3, prevTime).inverse();
        zeus::CQuaternion rb = x54_source->GetRotation(3, xc_curTime);
        ret.x0_remTime = remTime;
        ret.x8_deltas.xc_rotDelta = rb * ra;

        if (x54_source->HasOffset(3))
        {
            zeus::CVector3f ta = x54_source->GetOffset(3, prevTime);
            zeus::CVector3f tb = x54_source->GetOffset(3, xc_curTime);
            ret.x8_deltas.x0_posDelta = zeus::CMatrix3f(rb) * (tb - ta);
        }

        return ret;
    }
}

CCharAnimTime CAnimSourceReader::VGetTimeRemaining() const
{
    return x54_source->GetDuration() - xc_curTime;
}

CSteadyStateAnimInfo CAnimSourceReader::VGetSteadyStateAnimInfo() const
{
    return x64_steadyStateInfo;
}

bool CAnimSourceReader::VHasOffset(const CSegId& seg) const
{
    return x54_source->HasOffset(seg);
}

zeus::CVector3f CAnimSourceReader::VGetOffset(const CSegId& seg) const
{
    return x54_source->GetOffset(seg, xc_curTime);
}

zeus::CVector3f CAnimSourceReader::VGetOffset(const CSegId& seg,
                                              const CCharAnimTime& time) const
{
    return x54_source->GetOffset(seg, time);
}

zeus::CQuaternion CAnimSourceReader::VGetRotation(const CSegId& seg) const
{
    return x54_source->GetRotation(seg, xc_curTime);
}

CAnimSourceReader::CAnimSourceReader(const TSubAnimTypeToken<CAnimSource>& source,
                                     const CCharAnimTime& time)
: CAnimSourceReaderBase(std::make_unique<CAnimSourceInfo>(source), {}),
  x54_source(source), x64_steadyStateInfo(false, source->GetDuration(),
                                          source->GetOffset(source->GetRootBoneId(), time))
{
    PostConstruct(time);
}

CAnimSourceReader::CAnimSourceReader(const CAnimSourceReader& other)
: CAnimSourceReaderBase(std::make_unique<CAnimSourceInfo>(other.x54_source), other),
  x54_source(other.x54_source), x64_steadyStateInfo(other.x64_steadyStateInfo)
{}


}