#include "CExplosion.hpp"
#include "CStateManager.hpp"
#include "CGameLight.hpp"
#include "TCastTo.hpp"
#include "Graphics/CBooRenderer.hpp"
#include "GameGlobalObjects.hpp"

namespace urde {

CExplosion::CExplosion(const TLockedToken<CGenDescription>& particle, TUniqueId uid, bool active,
                       const CEntityInfo& info, std::string_view name, const zeus::CTransform& xf, u32 flags,
                       const zeus::CVector3f& scale, const zeus::CColor& color)
: CEffect(uid, info, active, name, xf) {
  xe8_particleGen = std::make_unique<CElementGen>(particle, CElementGen::EModelOrientationType::Normal,
                                                  flags & 0x2 ? CElementGen::EOptionalSystemFlags::Two
                                                              : CElementGen::EOptionalSystemFlags::One);
  xf0_particleDesc = particle.GetObj();
  xf4_24_renderThermalHot = flags & 0x4;
  xf4_25_ = true;
  xf4_26_renderXray = flags & 0x8;
  xe6_27_thermalVisorFlags = flags & 0x1 ? 1 : 2;
  xe8_particleGen->SetGlobalTranslation(xf.origin);
  xe8_particleGen->SetOrientation(xf.getRotation());
  xe8_particleGen->SetGlobalScale(scale);
  xe8_particleGen->SetModulationColor(color);
}

CExplosion::CExplosion(const TLockedToken<CElectricDescription>& electric, TUniqueId uid, bool active,
                       const CEntityInfo& info, std::string_view name, const zeus::CTransform& xf, u32 flags,
                       const zeus::CVector3f& scale, const zeus::CColor& color)
: CEffect(uid, info, active, name, xf) {
  xe8_particleGen = std::make_unique<CParticleElectric>(electric);
  xf0_electricDesc = electric.GetObj();
  xf4_24_renderThermalHot = flags & 0x4;
  xf4_25_ = true;
  xf4_26_renderXray = flags & 0x8;
  xe6_27_thermalVisorFlags = flags & 0x1 ? 1 : 2;
  xe8_particleGen->SetGlobalTranslation(xf.origin);
  xe8_particleGen->SetOrientation(xf.getRotation());
  xe8_particleGen->SetGlobalScale(scale);
  xe8_particleGen->SetModulationColor(color);
}

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

void CExplosion::AcceptScriptMsg(EScriptObjectMessage msg, TUniqueId sender, CStateManager& mgr) {
  if (msg == EScriptObjectMessage::Deleted) {
    if (xec_explosionLight != kInvalidUniqueId)
      mgr.FreeScriptObject(xec_explosionLight);
  } else if (msg == EScriptObjectMessage::Registered) {
    if (xe8_particleGen->SystemHasLight()) {
      xec_explosionLight = mgr.AllocateUniqueId();
      mgr.AddObject(new CGameLight(xec_explosionLight, GetAreaIdAlways(), true, "ExplodePLight_" + x10_name,
                                   x34_transform, GetUniqueId(), xe8_particleGen->GetLight(), 1, /*xf0_particleDesc*/ 0,
                                   0.f));
    }
  }

  CActor::AcceptScriptMsg(msg, sender, mgr);

  if (xec_explosionLight != kInvalidUniqueId)
    mgr.SendScriptMsgAlways(sender, xec_explosionLight, msg);
}

void CExplosion::Think(float dt, CStateManager& mgr) {
  if (xe4_28_transformDirty) {
    xe8_particleGen->SetGlobalTranslation(GetTranslation());
    xe8_particleGen->SetOrientation(GetTransform().getRotation());
    xe4_28_transformDirty = false;
  }
  xe8_particleGen->Update(dt);

  if (xec_explosionLight != kInvalidUniqueId) {
    TCastToPtr<CGameLight> light = mgr.ObjectById(xec_explosionLight);
    if (light && x30_24_active)
      light->SetLight(xe8_particleGen->GetLight());
  }

  xf8_time += dt;

  if (xf8_time > 15.f || xe8_particleGen->IsSystemDeletable())
    mgr.FreeScriptObject(GetUniqueId());
}

void CExplosion::PreRender(CStateManager& mgr, const zeus::CFrustum& frustum) {
  CActor::PreRender(mgr, frustum);
  xe4_30_outOfFrustum = !xf4_25_ || !frustum.aabbFrustumTest(x9c_renderBounds);
}

void CExplosion::AddToRenderer(const zeus::CFrustum& frustum, const CStateManager& mgr) const {
  if (xe4_30_outOfFrustum)
    return;

  if (!(xf4_24_renderThermalHot && mgr.GetThermalDrawFlag() == EThermalDrawFlag::Hot) &&
      !(xf4_26_renderXray && mgr.GetPlayerState()->GetActiveVisor(mgr) == CPlayerState::EPlayerVisor::XRay)) {
    g_Renderer->AddParticleGen(*xe8_particleGen.get());
    return;
  }

  EnsureRendered(mgr);
}

void CExplosion::Render(const CStateManager& mgr) const {
  if (mgr.GetThermalDrawFlag() == EThermalDrawFlag::Hot && xf4_24_renderThermalHot) {
    CElementGen::SetSubtractBlend(true);
    CBooModel::SetRenderModelBlack(true);
    xe8_particleGen->Render();
    CBooModel::SetRenderModelBlack(false);
    CElementGen::SetSubtractBlend(false);
    return;
  }

  CElementGen::SetSubtractBlend(xf4_24_renderThermalHot);
  CGraphics::SetFog(ERglFogMode::PerspLin, 0.f, 74.f, zeus::skBlack);
  xe8_particleGen->Render();
  mgr.SetupFogForArea(GetAreaIdAlways());
  CElementGen::SetSubtractBlend(false);
}

bool CExplosion::CanRenderUnsorted(const CStateManager&) const { return false; }
} // namespace urde