#include "CCollisionActorManager.hpp"
#include "TCastTo.hpp"
#include "World/CActor.hpp"
#include "CStateManager.hpp"
#include "CMaterialList.hpp"
#include "Collision/CCollisionActor.hpp"

namespace urde
{

CCollisionActorManager::CCollisionActorManager(CStateManager& mgr, TUniqueId owner, TAreaId area,
                                               const std::vector<CJointCollisionDescription>& descs, bool active)
: x10_ownerId(owner)
, x12_active(active)
{
    if (TCastToConstPtr<CActor> act = mgr.GetObjectById(x10_ownerId))
    {
        zeus::CTransform xf = act->GetTransform();
        const CAnimData* animData = act->GetModelData()->GetAnimationData();
        zeus::CVector3f scale = act->GetModelData()->GetScale();
        zeus::CTransform scaleXf = zeus::CTransform::Scale(scale);
        x0_jointDescriptions.reserve(descs.size());
        for (const CJointCollisionDescription& desc : descs)
        {
            CJointCollisionDescription modDesc = desc;
            modDesc.ScaleAllBounds(scale);
            zeus::CTransform locXf = GetWRLocatorTransform(*animData, modDesc.GetPivotId(), xf, scaleXf);
            if (modDesc.GetNextId() != 0xff)
            {
                zeus::CTransform locXf2 = GetWRLocatorTransform(*animData, modDesc.GetNextId(), xf, scaleXf);
                float dist = (locXf2.origin - locXf.origin).magnitude();
                if (modDesc.GetType() == CJointCollisionDescription::ECollisionType::OBBAutoSize)
                {
                    if (dist <= FLT_EPSILON)
                        continue;
                    zeus::CVector3f bounds = modDesc.GetBounds();
                    bounds.y += dist;
                    CCollisionActor* newAct =
                        new CCollisionActor(mgr.AllocateUniqueId(), area, x10_ownerId, bounds,
                                            zeus::CVector3f(0.f, 0.5f * dist, 0.f), active, modDesc.GetMass());
                    if (modDesc.GetOrientationType() == CJointCollisionDescription::EOrientationType::Zero)
                    {
                        newAct->SetTransform(locXf);
                    }
                    else
                    {
                        zeus::CVector3f delta = (locXf2.origin - locXf.origin).normalized();
                        zeus::CVector3f upVector = locXf.basis[2];
                        if (zeus::close_enough(std::fabs(delta.dot(upVector)), 1.f))
                            upVector = locXf.basis[1];
                        newAct->SetTransform(zeus::lookAt(locXf.origin, locXf.origin + delta, upVector));
                    }
                    mgr.AddObject(newAct);
                    x0_jointDescriptions.push_back(desc);
                }
                else
                {
                    TUniqueId newId = mgr.AllocateUniqueId();
                    CCollisionActor* newAct =
                        new CCollisionActor(newId, area, x10_ownerId,
                                            active, modDesc.GetRadius(), modDesc.GetMass());
                    newAct->SetTransform(locXf);
                    mgr.AddObject(newAct);
                    x0_jointDescriptions.push_back(CJointCollisionDescription::
                        SphereCollision(modDesc.GetPivotId(), modDesc.GetRadius(), modDesc.GetName(), 0.001f));
                    x0_jointDescriptions.back().SetCollisionActorId(newId);
                    u32 numSeps = u32(dist / modDesc.GetMaxSeparation());
                    if (numSeps)
                    {
                        x0_jointDescriptions.reserve(x0_jointDescriptions.capacity() + numSeps);
                        float pitch = dist / float(numSeps + 1);
                        for (u32 i = 0; i < numSeps; ++i)
                        {
                            float separation = pitch * float(i + 1);
                            x0_jointDescriptions.push_back(CJointCollisionDescription::
                                SphereSubdivideCollision(modDesc.GetPivotId(), modDesc.GetNextId(), modDesc.GetRadius(),
                                                         separation, CJointCollisionDescription::EOrientationType::One,
                                                         modDesc.GetName(), 0.001f));
                            TUniqueId newId2 = mgr.AllocateUniqueId();
                            CCollisionActor* newAct2 =
                                new CCollisionActor(newId2, area, x10_ownerId,
                                                    active, modDesc.GetRadius(), modDesc.GetMass());
                            if (modDesc.GetOrientationType() == CJointCollisionDescription::EOrientationType::Zero)
                            {
                                newAct2->SetTransform(
                                    zeus::CTransform::Translate(locXf.origin + separation * locXf.basis[1]));
                            }
                            else
                            {
                                zeus::CVector3f delta = (locXf2.origin - locXf.origin).normalized();
                                zeus::CVector3f upVector = locXf.basis[2];
                                if (zeus::close_enough(
                                    std::fabs(delta.dot(upVector)), 1.f))
                                    upVector = locXf.basis[1];
                                zeus::CTransform lookAt = zeus::lookAt(zeus::CVector3f::skZero, delta, upVector);
                                newAct2->SetTransform(
                                    zeus::CTransform::Translate(lookAt.basis[1] * separation + locXf.origin));
                            }
                            mgr.AddObject(newAct2);
                            x0_jointDescriptions.back().SetCollisionActorId(newId2);
                        }
                    }
                }
            }
            else
            {
                TUniqueId newId = mgr.AllocateUniqueId();
                CCollisionActor* newAct;
                if (modDesc.GetType() == CJointCollisionDescription::ECollisionType::Sphere)
                    newAct = new CCollisionActor(newId, area, x10_ownerId, active,
                                                 modDesc.GetRadius(), modDesc.GetMass());
                else if (modDesc.GetType() == CJointCollisionDescription::ECollisionType::OBB)
                    newAct = new CCollisionActor(newId, area, x10_ownerId, modDesc.GetBounds(),
                                                 modDesc.GetPivotPoint(), active, modDesc.GetMass());
                else
                    newAct = new CCollisionActor(newId, area, x10_ownerId, modDesc.GetBounds(),
                                                 active, modDesc.GetMass());
                newAct->SetTransform(locXf);
                mgr.AddObject(newAct);
                x0_jointDescriptions.push_back(desc);
                x0_jointDescriptions.back().SetCollisionActorId(newId);
            }
        }
    }
}

void CCollisionActorManager::Destroy(CStateManager& mgr) const
{
    for (const CJointCollisionDescription& desc : x0_jointDescriptions)
        mgr.FreeScriptObject(desc.GetCollisionActorId());

    const_cast<CCollisionActorManager&>(*this).x13_destroyed = true;
}

void CCollisionActorManager::SetActive(CStateManager& mgr, bool active)
{
    for (const CJointCollisionDescription& jDesc : x0_jointDescriptions)
    {
        if (TCastToPtr<CActor> act = mgr.ObjectById(jDesc.GetCollisionActorId()))
        {
            bool curActive = act->GetActive();
            if (curActive != active)
                act->SetActive(active);

            if (!curActive)
                Update(0.f, mgr, EUpdateOptions::WorldSpace);
        }
    }
}

void CCollisionActorManager::AddMaterial(CStateManager& mgr, const CMaterialList& list)
{
    for (const CJointCollisionDescription& jDesc : x0_jointDescriptions)
        if (TCastToPtr<CActor> act = mgr.ObjectById(jDesc.GetCollisionActorId()))
            act->AddMaterial(list);
}

void CCollisionActorManager::SetMovable(CStateManager& mgr, bool movable)
{
    if (x14_movable == movable)
        return;
    x14_movable = movable;
    for (const CJointCollisionDescription& jDesc : x0_jointDescriptions)
    {
        if (TCastToPtr<CCollisionActor> act = mgr.ObjectById(jDesc.GetCollisionActorId()))
        {
            act->SetMovable(x14_movable);
            act->SetUseInSortedLists(x14_movable);
        }
    }
}

void CCollisionActorManager::Update(float dt, CStateManager& mgr, EUpdateOptions opts)
{
    if (!x14_movable)
        SetMovable(mgr, true);
    if (x12_active)
    {
        if (TCastToPtr<CActor> act = mgr.ObjectById(x10_ownerId))
        {
            const CAnimData& animData = *act->GetModelData()->GetAnimationData();
            zeus::CTransform actXf = act->GetTransform();
            zeus::CTransform scaleXf = zeus::CTransform::Scale(act->GetModelData()->GetScale());
            for (const CJointCollisionDescription& jDesc : x0_jointDescriptions)
            {
                if (TCastToPtr<CCollisionActor> cAct = mgr.ObjectById(jDesc.GetCollisionActorId()))
                {
                    zeus::CTransform pivotXf = GetWRLocatorTransform(animData, jDesc.GetPivotId(), actXf, scaleXf);
                    zeus::CVector3f origin = pivotXf.origin;
                    if (jDesc.GetType() == CJointCollisionDescription::ECollisionType::OBB ||
                        jDesc.GetType() == CJointCollisionDescription::ECollisionType::OBBAutoSize)
                    {
                        if (jDesc.GetOrientationType() == CJointCollisionDescription::EOrientationType::Zero)
                        {
                            cAct->SetTransform(zeus::CQuaternion(pivotXf.basis).toTransform(cAct->GetTranslation()));
                        }
                        else
                        {
                            zeus::CTransform nextXf = GetWRLocatorTransform(animData, jDesc.GetNextId(), actXf, scaleXf);
                            cAct->SetTransform(
                                zeus::CQuaternion(zeus::lookAt(pivotXf.origin, nextXf.origin, pivotXf.basis[2]).basis).
                                toTransform(cAct->GetTranslation()));
                        }
                    }
                    else if (jDesc.GetType() == CJointCollisionDescription::ECollisionType::SphereSubdivide)
                    {
                        if (jDesc.GetOrientationType() == CJointCollisionDescription::EOrientationType::Zero)
                        {
                            origin += jDesc.GetMaxSeparation() * pivotXf.basis[1];
                        }
                        else
                        {
                            zeus::CTransform nextXf = GetWRLocatorTransform(animData, jDesc.GetNextId(), actXf, scaleXf);
                            origin += zeus::lookAt(origin, nextXf.origin, pivotXf.basis[2]).basis[1] *
                                      jDesc.GetMaxSeparation();
                        }
                    }
                    if (opts == EUpdateOptions::ObjectSpace)
                        cAct->MoveToOR(cAct->GetTransform().transposeRotate(origin - cAct->GetTranslation()), dt);
                    else
                        cAct->SetTranslation(origin);
                }
            }
        }
    }
}

zeus::CTransform CCollisionActorManager::GetWRLocatorTransform(const CAnimData& animData, CSegId id,
                                                               const zeus::CTransform& worldXf,
                                                               const zeus::CTransform& localXf)
{
    zeus::CTransform locXf = animData.GetLocatorTransform(id, nullptr);
    zeus::CVector3f origin = worldXf * (localXf * locXf.origin);
    locXf = worldXf.multiplyIgnoreTranslation(locXf);
    locXf.origin = origin;
    return locXf;
}

}