#include "Runtime/World/CFire.hpp"

#include "Runtime/CStateManager.hpp"
#include "Runtime/GameGlobalObjects.hpp"
#include "Runtime/Graphics/CCubeRenderer.hpp"
#include "Runtime/Particle/CElementGen.hpp"
#include "Runtime/World/CActorParameters.hpp"
#include "Runtime/World/CPlayer.hpp"

#include "TCastTo.hpp" // Generated file, do not modify include path

namespace metaforce {
CFire::CFire(TToken<CGenDescription> effect, TUniqueId uid, TAreaId aId, bool active, TUniqueId owner,
             const zeus::CTransform& xf, const CDamageInfo& dInfo, const zeus::CAABox& aabox,
             const zeus::CVector3f& vec, bool b1, CAssetId visorEffect, bool b2, bool b3, bool b4, float f1, float f2,
             float f3, float f4)
: CActor(uid, active, "Fire"sv, CEntityInfo(aId, NullConnectionList), xf, CModelData::CModelDataNull(),
         CMaterialList(EMaterialTypes::Projectile), CActorParameters::None(), owner)
, xe8_(std::make_unique<CElementGen>(effect))
, xec_ownerId(owner)
, xf0_damageInfo(dInfo)
, x10c_damageInfo(dInfo)
, x128_(aabox)
, x144_(f1)
, x148_24_(b2)
, x148_25_(b3)
, x148_26_(b4)
, x148_27_(b4 && b3 && b2)
, x148_29_(b1)
, x14c_(f2)
, x150_(visorEffect)
, x154_(f3)
, x158_(f4) {
  xe8_->SetGlobalScale(vec);
  xe8_->SetTranslation(xf.origin);
}

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

void CFire::AcceptScriptMsg(EScriptObjectMessage msg, TUniqueId uid, CStateManager& mgr) {
  CActor::AcceptScriptMsg(msg, uid, mgr);

  if (msg == EScriptObjectMessage::Registered) {
    xe8_->SetParticleEmission(true);
    SetActive(true);
  }
}

void CFire::Think(float dt, CStateManager& mgr) {
  float particleCount = xe8_->GetParticleCount() / xe8_->GetMaxParticles();
  if (GetActive()) {
    xe8_->Update(dt * x144_);
    x10c_damageInfo = CDamageInfo(xf0_damageInfo, dt * std::max(0.5f, particleCount));
  }

  bool doFree = false;
  if (xe8_->IsSystemDeletable())
    doFree = true;

  if (x148_29_) {
    auto playerBounds = mgr.GetPlayer().GetTouchBounds();
    auto bounds = GetTouchBounds();
    if (playerBounds->intersects(*bounds) && doFree && particleCount > 0.5f)
      mgr.GetPlayer().SetVisorSteam(particleCount * x14c_, x154_, x158_, x150_, true);
    else
      mgr.GetPlayer().SetVisorSteam(0.f, 1.f, 1.f, {}, true);
  }

  x15c_ += dt;

  if (x15c_ > 45.f)
    doFree = true;

  if (doFree)
    mgr.FreeScriptObject(GetUniqueId());
}

void CFire::Touch(CActor& act, CStateManager& mgr) {
  if (act.GetUniqueId() == xec_ownerId)
    return;

  mgr.ApplyDamage(GetUniqueId(), act.GetUniqueId(), GetUniqueId(), x10c_damageInfo,
                  CMaterialFilter::MakeIncludeExclude({EMaterialTypes::Solid}, {}), {});
}

void CFire::AddToRenderer(const zeus::CFrustum& frustum, CStateManager& mgr) {
  bool drawParticles = true;
  if (!x148_27_) {
    using EPlayerVisor = CPlayerState::EPlayerVisor;
    CPlayerState::EPlayerVisor visor = mgr.GetPlayerState()->GetActiveVisor(mgr);
    if (visor == EPlayerVisor::Combat || visor == EPlayerVisor::Scan)
      drawParticles = x148_24_;
    else if (visor == EPlayerVisor::XRay)
      drawParticles = x148_26_;
    else if (visor == EPlayerVisor::Thermal)
      drawParticles = x148_25_;
  }

  if (drawParticles)
    g_Renderer->AddParticleGen(*xe8_);
  CActor::AddToRenderer(frustum, mgr);
}

} // namespace metaforce