#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});
    }
  }
}
} // namespace urde