#include "CAnimationParameters.h"
#include "CAnimSet.h"
#include "CResCache.h"
#include "CResourceInfo.h"
#include <Common/Log.h>
#include <iostream>

CAnimationParameters::CAnimationParameters()
{
    mGame = ePrime;
    mNodeIndex = 0;
    mUnknown1 = 0;
    mUnknown2 = 0;
    mUnknown3 = 0;
}

CAnimationParameters::CAnimationParameters(EGame Game)
{
    mGame = Game;
    mNodeIndex = 0;
    mUnknown1 = 0;
    mUnknown2 = 0;
    mUnknown3 = 0;
}

CAnimationParameters::CAnimationParameters(IInputStream& SCLY, EGame Game)
{
    mGame = Game;
    mNodeIndex = 0;
    mUnknown1 = 0;
    mUnknown2 = 0;
    mUnknown3 = 0;

    if (Game <= eEchoes)
    {
        mCharacter = CResourceInfo(SCLY.ReadLong(), "ANCS");
        mNodeIndex = SCLY.ReadLong();
        mUnknown1 = SCLY.ReadLong();
    }

    else if (Game <= eCorruption)
    {
        mCharacter = CResourceInfo(SCLY.ReadLongLong(), "CHAR");
        mUnknown1 = SCLY.ReadLong();
    }

    else if (Game == eReturns)
    {
        u8 Flags = SCLY.ReadByte();

        // 0x80 - CharacterAnimationSet is empty.
        if (Flags & 0x80)
        {
            mUnknown1 = -1;
            mUnknown2 = 0;
            mUnknown3 = 0;
            return;
        }

        mCharacter = CResourceInfo(SCLY.ReadLongLong(), "CHAR");

        // 0x20 - Default Anim is present
        if (Flags & 0x20)
            mUnknown1 = SCLY.ReadLong();
        else
            mUnknown1 = -1;

        // 0x40 - Two-value struct is present
        if (Flags & 0x40)
        {
            mUnknown2 = SCLY.ReadLong();
            mUnknown3 = SCLY.ReadLong();
        }
        else
        {
            mUnknown2 = 0;
            mUnknown3 = 0;
        }
    }
}

void CAnimationParameters::Write(IOutputStream& rSCLY)
{
    if (mGame <= eEchoes)
    {
        if (mCharacter.IsValid())
        {
            rSCLY.WriteLong(mCharacter.ID().ToLong());
            rSCLY.WriteLong(mNodeIndex);
            rSCLY.WriteLong(mUnknown1);
        }
        else
        {
            rSCLY.WriteLong(0xFFFFFFFF);
            rSCLY.WriteLong(0);
            rSCLY.WriteLong(0xFFFFFFFF);
        }
    }

    else if (mGame <= eCorruption)
    {
        if (mCharacter.IsValid())
        {
            rSCLY.WriteLongLong(mCharacter.ID().ToLongLong());
            rSCLY.WriteLong(mUnknown1);
        }

        else
        {
            rSCLY.WriteLongLong(CUniqueID::skInvalidID64.ToLongLong());
            rSCLY.WriteLong(0xFFFFFFFF);
        }
    }

    else
    {
        if (!mCharacter.IsValid())
            rSCLY.WriteByte((u8) 0x80);

        else
        {
            u8 Flag = 0;
            if (mUnknown1 != -1) Flag |= 0x20;
            if (mUnknown2 != 0 || mUnknown3 != 0) Flag |= 0x40;

            rSCLY.WriteByte(Flag);
            rSCLY.WriteLongLong(mCharacter.ID().ToLongLong());

            if (Flag & 0x20)
                rSCLY.WriteLong(mUnknown1);

            if (Flag & 0x40)
            {
                rSCLY.WriteLong(mUnknown2);
                rSCLY.WriteLong(mUnknown3);
            }
        }
    }
}

CModel* CAnimationParameters::GetCurrentModel(s32 NodeIndex /*= -1*/)
{
    if (!mCharacter.IsValid()) return nullptr;

    CAnimSet *pSet = (CAnimSet*) mCharacter.Load();
    if (!pSet) return nullptr;
    if (pSet->Type() != eAnimSet) return nullptr;
    if (NodeIndex == -1) NodeIndex = mNodeIndex;

    if (pSet->getNodeCount() <= (u32) NodeIndex) return nullptr;
    return pSet->getNodeModel(NodeIndex);
}

TString CAnimationParameters::GetCurrentCharacterName(s32 NodeIndex /*= -1*/)
{
    if (!mCharacter.IsValid()) return "";

    CAnimSet *pSet = (CAnimSet*) mCharacter.Load();
    if (!pSet) return "";
    if (pSet->Type() != eAnimSet) return "";
    if (NodeIndex == -1) NodeIndex = mNodeIndex;

    if (pSet->getNodeCount() <= (u32) NodeIndex) return "";
    return pSet->getNodeName((u32) NodeIndex);
}

// ************ GETTERS ************
EGame CAnimationParameters::Version()
{
    return mGame;
}

CAnimSet* CAnimationParameters::AnimSet()
{
    return (CAnimSet*) mCharacter.Load();
}

u32 CAnimationParameters::CharacterIndex()
{
    return mNodeIndex;
}

u32 CAnimationParameters::Unknown(u32 Index)
{
    switch (Index)
    {
    case 0: return mUnknown1;
    case 1: return mUnknown2;
    case 2: return mUnknown3;
    default: return 0;
    }
}

// ************ SETTERS ************
void CAnimationParameters::SetResource(CResourceInfo Res)
{
    if (Res.Type() == "ANCS" || Res.Type() == "CHAR")
    {
        mCharacter = Res;
        mNodeIndex = 0;
    }
    else
        Log::Error("Resource with invalid type passed to CAnimationParameters: " + Res.ToString());
}

void CAnimationParameters::SetNodeIndex(u32 Index)
{
    mNodeIndex = Index;
}

void CAnimationParameters::SetUnknown(u32 Index, u32 Value)
{
    switch (Index)
    {
    case 0: mUnknown1 = Value;
    case 1: mUnknown2 = Value;
    case 2: mUnknown3 = Value;
    }
}