#include "CScriptPickup.hpp"
#include "Particle/CGenDescription.hpp"
#include "GameGlobalObjects.hpp"
#include "MP1/CArtifactDoll.hpp"
#include "CExplosion.hpp"
#include "CSimplePool.hpp"
#include "TCastTo.hpp"
#include "CPlayer.hpp"
#include "CGameState.hpp"
#include "CGameOptions.hpp"
#include "MP1/CSamusHud.hpp"
#include "GuiSys/CStringTable.hpp"
#include "Camera/CFirstPersonCamera.hpp"

namespace urde
{
CScriptPickup::CScriptPickup(TUniqueId uid, std::string_view name, const CEntityInfo& info,
                             const zeus::CTransform& xf, CModelData&& mData, const CActorParameters& aParams,
                             const zeus::CAABox& aabb, CPlayerState::EItemType itemType, s32 amount, s32 capacity,
                             CAssetId explosionEffect, float possibility, float f2, float f3, float f4, bool active)
: CPhysicsActor(uid, active, name, info, xf, std::move(mData), CMaterialList(), aabb, SMoverData(1.f), aParams, 0.3f,
                0.1f)
, x258_itemType(itemType)
, x25c_amount(amount)
, x260_capacity(capacity)
, x264_possibility(possibility)
, x268_(f3)
, x26c_(f2)
, x278_(f4)
, x28c_24_(false)
, x28c_25_(false)
, x28c_26_(false)
{
    if (explosionEffect.IsValid())
        x27c_explosionDesc = g_SimplePool->GetObj({SBIG('PART'), explosionEffect});

    if (x64_modelData && x64_modelData->AnimationData())
        x64_modelData->AnimationData()->SetAnimation(CAnimPlaybackParms(0, -1, 1.f, true), false);

    if (x278_ != 0.f)
    {
        xb4_drawFlags = CModelFlags(5, 0, 3, zeus::CColor(1.f, 1.f, 1.f, 0.f));
        xb4_drawFlags.x2_flags &= 0xFFFC;
        xb4_drawFlags.x2_flags |= 1;
    }
}

void CScriptPickup::Accept(IVisitor& visitor) { visitor.Visit(this); }

void CScriptPickup::Think(float dt, CStateManager& mgr)
{
    if (!GetActive())
        return;

    if (x278_ >= 0.f)
    {
        CPhysicsActor::Stop();
        x278_ -= dt;
        return;
    }

    x270_ += dt;
    if (x28c_25_ && (x270_ - x26c_) < 2.f)
        x270_ = zeus::max(2.f * dt - x270_, (2.f - x26c_) - FLT_EPSILON);

    CModelFlags drawFlags{0, 0, 3, zeus::CColor(1.f, 1.f, 1.f, 1.f)};

    if (x268_ != 0.f)
    {
        if (x270_ < x268_)
        {
            drawFlags = CModelFlags(5, 0, 3, zeus::CColor(1.f, 1.f, x270_ / x268_));
            drawFlags.x2_flags &= 0xFFFC;
            drawFlags.x2_flags |= 1;
        }
        else
            x268_ = 0.f;
    }
    else if (x26c_ != 0.f)
    {
        float alpha = 1.f;
        if (x26c_ < 2.f)
            alpha = 1.f - (x26c_ / x270_);
        else if ((x270_ - x26c_) < 2.f)
            alpha = (x270_ - x26c_) * 0.5f;

        drawFlags = CModelFlags(5, 0, 3, zeus::CColor(1.f, 1.f, 1.f, alpha));
        drawFlags.x2_flags &= 0xFFFC;
        drawFlags.x2_flags |= 1;
    }

    xb4_drawFlags = drawFlags;

    if (x64_modelData)
    {
        if (x64_modelData->HasAnimData())
        {
            SAdvancementDeltas deltas = UpdateAnimation(dt, mgr, true);
            MoveToOR(deltas.x0_posDelta, dt);
            RotateToOR(deltas.xc_rotDelta, dt);
        }

        if (x28c_25_)
        {
            zeus::CVector3f posVec =
                GetTranslation() - mgr.GetPlayer().GetTranslation() + (2.f * zeus::CVector3f::skUp);
            posVec = (20.f * (0.5f * zeus::max(2.f, x274_ + dt))) * posVec.normalized();

            float chargeFactor = 0.f;
            if (x28c_26_ && mgr.GetPlayer().GetPlayerGun()->IsCharging())
                chargeFactor = mgr.GetPlayer().GetPlayerGun()->GetChargeBeamFactor();

            if (chargeFactor < CPlayerGun::skTractorBeamFactor)
            {
                x28c_26_ = false;
                x28c_25_ = false;
                posVec.zeroOut();
            }
            SetVelocityOR(posVec);
        }
        else if (x28c_24_)
        {
            float chargeFactor = mgr.GetPlayer().GetPlayerGun()->IsCharging()
                                     ? mgr.GetPlayer().GetPlayerGun()->GetChargeBeamFactor()
                                     : 0.f;

            if (chargeFactor > CPlayerGun::skTractorBeamFactor)
            {
                zeus::CVector3f posVec =
                    (GetTranslation() - mgr.GetCameraManager()->GetFirstPersonCamera()->GetTranslation()).normalized();
                float relFov = zeus::CRelAngle(zeus::degToRad(g_tweakGame->GetFirstPersonFOV()));
                if (mgr.GetCameraManager()->GetFirstPersonCamera()->GetTransform().upVector().dot(posVec) >
                        std::cos(relFov) && posVec.magSquared() < (30.f * 30.f))
                {
                    x28c_25_ = true;
                    x28c_26_ = true;
                    x274_ = 0.f;
                }
            }
        }
    }

    if (x26c_ != 0.f && x270_ > x26c_)
        mgr.FreeScriptObject(GetUniqueId());

}

void CScriptPickup::Touch(CActor& act, CStateManager& mgr)
{
    if (GetActive() && x278_ < 0.f && TCastToPtr<CPlayer>(act))
    {
        if (x258_itemType >= CPlayerState::EItemType::Truth && x258_itemType <= CPlayerState::EItemType::Newborn)
        {
            CAssetId id = MP1::CArtifactDoll::GetArtifactHeadScanFromItemType(x258_itemType);
            if (id.IsValid())
                mgr.GetPlayerState()->SetScanTime(id, 0.5f);
        }

        if (x27c_explosionDesc)
        {
            if (mgr.GetPlayerState()->GetActiveVisor(mgr) != CPlayerState::EPlayerVisor::Thermal)
            {
                mgr.AddObject(new CExplosion(
                    x27c_explosionDesc, mgr.AllocateUniqueId(), true,
                    CEntityInfo(GetAreaIdAlways(), CEntity::NullConnectionList, kInvalidEditorId),
                    "Explosion - Pickup Effect", x34_transform, 0, zeus::CVector3f::skOne, zeus::CColor::skWhite));
            }
        }

        mgr.GetPlayerState()->InitializePowerUp(x258_itemType, x260_capacity);
        mgr.GetPlayerState()->IncrPickup(x258_itemType, x25c_amount);
        mgr.FreeScriptObject(GetUniqueId());
        SendScriptMsgs(EScriptObjectState::Arrived, mgr, EScriptObjectMessage::None);

        if (x260_capacity > 0)
        {
            u32 total = mgr.GetPlayerState()->GetPickupTotal();
            u32 colRate = mgr.GetPlayerState()->CalculateItemCollectionRate();
            if (total == colRate)
            {
                CPersistentOptions& opts = g_GameState->SystemOptions();
                mgr.QueueMessage(mgr.GetHUDMessageFrameCount() + 1,
                                 g_ResFactory
                                     ->GetResourceIdByName(opts.GetAllItemsCollected() ? "STRG_AllPickupsFound_2"
                                                                                       : "STRG_AllPickupsFound_1")
                                     ->id,
                                 0.f);
                opts.SetAllItemsCollected(true);
            }
        }

        if (x258_itemType == CPlayerState::EItemType::PowerBombs &&
            g_GameState->SystemOptions().GetShowPowerBombAmmoMessage())
        {
            g_GameState->SystemOptions().IncrementPowerBombAmmoCount();
            MP1::CSamusHud::DisplayHudMemo(g_MainStringTable->GetString(109), {0.5f, true, false, false});
        }
    }
}
}