#include "CWallWalker.hpp"
#include "CPatternedInfo.hpp"
#include "CStateManager.hpp"
#include "TCastTo.hpp"
#include "CScriptWaypoint.hpp"

namespace urde
{
CWallWalker::CWallWalker(ECharacter chr, TUniqueId uid, std::string_view name, EFlavorType flavType,
                         const CEntityInfo& eInfo, const zeus::CTransform& xf,
                         CModelData&& mData, const CPatternedInfo& pInfo, EMovementType mType,
                         EColliderType colType, EBodyType bType, const CActorParameters& aParms,
                         float f1, float f2, EKnockBackVariant kbVariant, float f3, EWalkerType wType, float f4, bool b1)
: CPatterned(chr, uid, name, flavType, eInfo, xf, std::move(mData), pInfo, mType, colType, bType, aParms, kbVariant)
, x590_colSphere(zeus::CSphere(zeus::CVector3f::skZero, pInfo.GetHalfExtent()), x68_material)
, x5b0_collisionCloseMargin(f1)
, x5b4_(f2)
, x5c0_advanceWpRadius(f3)
, x5c4_(f4)
, x5cc_bendingHackAnim(GetModelData()->GetAnimationData()->GetCharacterInfo().GetAnimationIndex("BendingAnimationHack"sv))
, x5d0_walkerType(wType)
, x5d6_24_(false)
, x5d6_25_(false)
, x5d6_26_(false)
, x5d6_27_(b1)
, x5d6_28_addBendingWeight(true)
, x5d6_29_applyBendingHack(false)
{
}

void CWallWalker::OrientToSurfaceNormal(const zeus::CVector3f& normal, float clampAngle)
{
    float dot = x34_transform.basis[2].dot(normal);
    if (zeus::close_enough(dot, 1.f) || dot < -0.999f)
        return;
    zeus::CQuaternion q =
    zeus::CQuaternion::clampedRotateTo(x34_transform.basis[2], normal, zeus::degToRad(clampAngle));
    q.setImaginary(x34_transform.transposeRotate(q.getImaginary()));
    SetTransform((zeus::CQuaternion(x34_transform.basis) * q).normalized().toTransform(GetTranslation()));
}

void CWallWalker::GotoNextWaypoint(CStateManager& mgr)
{
    if (TCastToPtr<CScriptWaypoint> wp = mgr.ObjectById(x2dc_destObj))
    {
        zeus::CVector3f wpPos = wp->GetTranslation();
        if ((wpPos - GetTranslation()).magSquared() < x5c0_advanceWpRadius * x5c0_advanceWpRadius)
        {
            x2dc_destObj = wp->NextWaypoint(mgr);
            if (!zeus::close_enough(wp->GetPause(), 0.f))
            {
                x5bc_ = wp->GetPause();
                if (x5d0_walkerType == EWalkerType::Parasite)
                    x450_bodyController->SetLocomotionType(pas::ELocomotionType::Relaxed);
            }
            mgr.SendScriptMsg(wp, GetUniqueId(), EScriptObjectMessage::Arrived);
        }
        SetDestPos(wpPos);
    }
}

void CWallWalker::PreThink(float dt, CStateManager& mgr)
{
    CPatterned::PreThink(dt, mgr);
    if (!GetActive() || x5d6_26_ || x5bc_ > 0.f || x5d6_27_ || x450_bodyController->IsFrozen() || !x5d6_24_)
        return;

    zeus::CQuaternion quat(GetTransform().buildMatrix3f());
    AddMotionState(PredictMotion(dt));
    zeus::CQuaternion quat2(GetTransform().buildMatrix3f());
    ClearForcesAndTorques();
    if (x5d6_25_)
    {
        zeus::CPlane plane = x568_.GetPlane();
        const float futureDt = (10.f * dt);
        SetTranslation(GetTranslation() * (1.f - futureDt) +
                       (((GetTranslation() - ((plane.vec.dot(GetTranslation())) - plane.d) -
                          x590_colSphere.GetSphere().radius - 0.1f) * plane.vec) * futureDt));
    }
    else
        MoveCollisionPrimitive(zeus::CVector3f::skZero);
}

void CWallWalker::Think(float dt, CStateManager& mgr)
{
    if (!x450_bodyController->GetActive())
        x450_bodyController->Activate(mgr);
    CPatterned::Think(dt, mgr);

    if (x5cc_bendingHackAnim == -1)
        return;

    if (x5d6_28_addBendingWeight)
    {
        if (x5c8_bendingHackWeight < 1.f)
        {
            x5c8_bendingHackWeight += (dt * x138_velocity.magnitude()) / 0.6f;
            if (x5c8_bendingHackWeight >= 1.f)
                x5c8_bendingHackWeight = 1.f;
        }
    }
    else if (x5c8_bendingHackWeight > 0.f)
    {
        x5c8_bendingHackWeight -= (dt * x138_velocity.magnitude()) / 1.5f;
        if (x5c8_bendingHackWeight < 0.f)
            x5c8_bendingHackWeight = 0.f;
    }

    if (x5c8_bendingHackWeight <= 0.f && !x5d6_29_applyBendingHack)
        return;

    if (x5c8_bendingHackWeight > 0.0001f)
    {
        ModelData()->AnimationData()->AddAdditiveAnimation(x5cc_bendingHackAnim, x5c8_bendingHackWeight, true, false);
        x5d6_29_applyBendingHack = true;
    }
    else
    {
        ModelData()->AnimationData()->DelAdditiveAnimation(x5cc_bendingHackAnim);
        x5d6_29_applyBendingHack = false;
    }
}

void CWallWalker::Render(const CStateManager& mgr) const
{
    CPatterned::Render(mgr);
}

void CWallWalker::UpdateWPDestination(CStateManager& mgr)
{
    if (TCastToPtr<CScriptWaypoint> wp = mgr.ObjectById(x2dc_destObj))
    {
        zeus::CVector3f wpPos = wp->GetTranslation();
        if ((wpPos - GetTranslation()).magSquared() < x5c0_advanceWpRadius * x5c0_advanceWpRadius)
        {
            x2dc_destObj = wp->NextWaypoint(mgr);
            if (std::fabs(wp->GetPause()) > 0.00001f)
            {
                x5bc_ = wp->GetPause();
                if (x5d0_walkerType == EWalkerType::Parasite)
                    x450_bodyController->SetLocomotionType(pas::ELocomotionType::Relaxed);
                mgr.SendScriptMsg(wp, GetUniqueId(), EScriptObjectMessage::Arrived);
            }
        }

        SetDestPos(wpPos);
    }
}
}