#include "CCharacterFactory.hpp"
#include "CAnimCharacterSet.hpp"
#include "CSimplePool.hpp"
#include "CTransitionDatabaseGame.hpp"
#include "CAnimationDatabaseGame.hpp"
#include "CAnimationManager.hpp"
#include "CTransitionManager.hpp"
#include "CRandom16.hpp"
#include "CPrimitive.hpp"
#include "CAnimData.hpp"
#include "CAdditiveAnimPlayback.hpp"
#include "GameGlobalObjects.hpp"
#include "CParticleGenInfo.hpp"
#include "Graphics/CSkinnedModel.hpp"
#include "Character/CCharLayoutInfo.hpp"

namespace urde
{

CFactoryFnReturn CCharacterFactory::CDummyFactory::Build(const SObjectTag& tag,
                                                         const CVParamTransfer& params,
                                                         CObjectReference* selfRef)
{

    const CCharacterInfo& charInfo = *params.GetOwnedObj<const CCharacterInfo*>();

    switch (tag.type.toUint32() & 0x1)
    {
    case 0:
        return TToken<CSkinnedModel>::GetIObjObjectFor(
            std::make_unique<CSkinnedModel>(*g_SimplePool,
                                            charInfo.GetModelId(),
                                            charInfo.GetSkinRulesId(),
                                            charInfo.GetCharLayoutInfoId(),
                                            0, tag.type.toUint32() >> 16));
    case 1:
        return TToken<CSkinnedModel>::GetIObjObjectFor(
            std::make_unique<CMorphableSkinnedModel>(*g_SimplePool,
                                                     charInfo.GetIceModelId(),
                                                     charInfo.GetIceSkinRulesId(),
                                                     charInfo.GetCharLayoutInfoId(),
                                                     0, tag.type.toUint32() >> 16));
    default:
        break;
    }

    return {};
}

void CCharacterFactory::CDummyFactory::BuildAsync(const SObjectTag& tag,
                                                  const CVParamTransfer& parms,
                                                  IObj** objOut,
                                                  CObjectReference* selfRef)
{
    *objOut = Build(tag, parms, selfRef).release();
}

void CCharacterFactory::CDummyFactory::CancelBuild(const SObjectTag&)
{
}

bool CCharacterFactory::CDummyFactory::CanBuild(const SObjectTag&)
{
    return true;
}

const SObjectTag* CCharacterFactory::CDummyFactory::GetResourceIdByName(const char*) const
{
    return nullptr;
}

FourCC CCharacterFactory::CDummyFactory::GetResourceTypeById(CAssetId id) const
{
    return {};
}

void CCharacterFactory::CDummyFactory::EnumerateResources(
        const std::function<bool(const SObjectTag&)>& lambda) const
{
}
void CCharacterFactory::CDummyFactory::EnumerateNamedResources(
        const std::function<bool(const std::string&, const SObjectTag&)>& lambda) const
{
}

u32 CCharacterFactory::CDummyFactory::ResourceSize(const urde::SObjectTag& tag)
{
    return 0;
}

bool CCharacterFactory::CDummyFactory::LoadResourceAsync(const urde::SObjectTag& tag,
                                                         std::unique_ptr<u8[]>& target)
{
    return false;
}

bool CCharacterFactory::CDummyFactory::LoadResourcePartAsync(const urde::SObjectTag& tag,
                                                             u32 size, u32 off,
                                                             std::unique_ptr<u8[]>& target)
{
    return false;
}

std::unique_ptr<u8[]> CCharacterFactory::CDummyFactory::LoadResourceSync(const urde::SObjectTag& tag)
{
    return {};
}

std::unique_ptr<u8[]> CCharacterFactory::CDummyFactory::LoadResourcePartSync(const urde::SObjectTag& tag,
                                                                             u32 size, u32 off)
{
    return {};
}

std::unique_ptr<CAnimData>
CCharacterFactory::CreateCharacter(int charIdx, bool loop,
                                   const TLockedToken<CCharacterFactory>& factory,
                                   int defaultAnim, int drawInsts) const
{
    const CCharacterInfo& charInfo = x4_charInfoDB[charIdx];
    CVParamTransfer charParm(new TObjOwnerParam<const CCharacterInfo*>(&charInfo));

    TToken<CSkinnedModel> skinnedModel =
        const_cast<CCharacterFactory*>(this)->x70_cacheResPool.GetObj
            ({FourCC(drawInsts << 16), charInfo.GetModelId()}, charParm);

    rstl::optional_object<TToken<CMorphableSkinnedModel>> iceModel;
    if (charInfo.GetIceModelId().IsValid() &&
        charInfo.GetIceSkinRulesId().IsValid())
        iceModel.emplace(const_cast<CCharacterFactory*>(this)->x70_cacheResPool.GetObj
            ({FourCC((drawInsts << 16) | 1), charInfo.GetIceModelId()}, charParm));

    return std::make_unique<CAnimData>(x68_selfId, charInfo, defaultAnim, charIdx, loop,
                                       x14_charLayoutInfoDB[charIdx], skinnedModel,
                                       iceModel, x24_sysContext, x28_animMgr, x2c_transMgr,
                                       factory, drawInsts);
}

CAssetId CCharacterFactory::GetEventResourceIdForAnimResourceId(CAssetId id) const
{
    auto search = std::find_if(x58_animResources.cbegin(), x58_animResources.cend(),
    [&](const std::pair<CAssetId, CAssetId>& elem) -> bool
    {
        return id == elem.first;
    });
    if (search == x58_animResources.cend())
        return CAssetId();
    return search->second;
}

const CAdditiveAnimationInfo& CCharacterFactory::FindAdditiveInfo(u32 idx) const
{
    auto search = rstl::binary_find(x40_additiveInfo.cbegin(), x40_additiveInfo.cend(), idx,
    [](const auto& anim) { return anim.first; });

    if (search == x40_additiveInfo.cend())
        return x50_defaultAdditiveInfo;
    return search->second;
}

bool CCharacterFactory::HasAdditiveInfo(u32 idx) const
{
    auto search = rstl::binary_find(x40_additiveInfo.cbegin(), x40_additiveInfo.cend(), idx,
    [](const auto& anim) { return anim.first; });
    return search != x40_additiveInfo.cend();
}

std::vector<CCharacterInfo>
CCharacterFactory::GetCharacterInfoDB(const CAnimCharacterSet& ancs)
{
    std::vector<CCharacterInfo> ret;
    const std::map<u32, CCharacterInfo>& charInfoMap = ancs.GetCharacterSet().GetCharacterInfoMap();
    ret.reserve(charInfoMap.size());
    for (const auto& charInfo : charInfoMap)
        ret.push_back(charInfo.second);
    return ret;
}

std::vector<TLockedToken<CCharLayoutInfo>>
CCharacterFactory::GetCharLayoutInfoDB(CSimplePool& store,
                                       const std::vector<CCharacterInfo>& chars)
{
    std::vector<TLockedToken<CCharLayoutInfo>> ret;
    ret.reserve(chars.size());
    for (const CCharacterInfo& charInfo : chars)
        ret.push_back(store.GetObj({SBIG('CINF'), charInfo.GetCharLayoutInfoId()}));
    return ret;
}

CCharacterFactory::CCharacterFactory(CSimplePool& store,
                                     const CAnimCharacterSet& ancs,
                                     CAssetId selfId)
: x4_charInfoDB(GetCharacterInfoDB(ancs)),
  x14_charLayoutInfoDB(GetCharLayoutInfoDB(store, x4_charInfoDB)),
  x24_sysContext(std::make_shared<CAnimSysContext>(
                 TToken<CTransitionDatabaseGame>(
                 std::make_unique<CTransitionDatabaseGame>(ancs.GetAnimationSet().GetTransitions(),
                                                           ancs.GetAnimationSet().GetHalfTransitions(),
                                                           ancs.GetAnimationSet().GetDefaultTransition())),
                     2334, store)),
  x28_animMgr(std::make_shared<CAnimationManager>(
              TToken<CAnimationDatabaseGame>(
              std::make_unique<CAnimationDatabaseGame>(
              ancs.GetAnimationSet().GetAnimations())), *x24_sysContext)),
  x2c_transMgr(std::make_shared<CTransitionManager>(*x24_sysContext)),
  x40_additiveInfo(ancs.GetAnimationSet().GetAdditiveInfo()),
  x50_defaultAdditiveInfo(ancs.GetAnimationSet().GetDefaultAdditiveInfo()),
  x58_animResources(ancs.GetAnimationSet().GetAnimResIds()),
  x68_selfId(selfId),
  x70_cacheResPool(x6c_dummyFactory)
{
    std::vector<CPrimitive> primitives;
    x28_animMgr->GetAnimationDatabase()->GetAllUniquePrimitives(primitives);
    x30_animSourceDB.reserve(primitives.size());
    for (const CPrimitive& prim : primitives)
        x30_animSourceDB.push_back(store.GetObj({SBIG('ANIM'), prim.GetAnimResId()}));
}

}