#include "CScriptDistanceFog.hpp"
#include "CStateManager.hpp"
#include "CWorld.hpp"

namespace urde
{
bool close_enough(const zeus::CVector2f& a, const zeus::CVector2f& b, float epsilon = 0.000099999997f)
{
    if (std::fabs(a.x - b.x) < epsilon && std::fabs(a.y - b.y) < epsilon)
        return true;
    return false;
}

bool close_enough(float a, float b, double epsilon = 0.000009999999747378752)
{
    return std::fabs(a - b) < epsilon;
}

CScriptDistanceFog::CScriptDistanceFog(TUniqueId uid, const std::string& name, const CEntityInfo& info,
                                       const ERglFogMode& mode, const zeus::CColor& color,
                                       const zeus::CVector2f& range, float colorDelta,
                                       const zeus::CVector2f& rangeDelta, bool expl, bool active,
                                       float thermalTarget, float thermalSpeed, float xrayTarget, float xraySpeed)
    : CEntity(uid, info, active, name),
      x34_mode(mode),
      x38_color(color),
      x3c_range(range),
      x44_colorDelta(colorDelta),
      x48_rangeDelta(rangeDelta),
      x50_thermalTarget(thermalTarget),
      x54_thermalSpeed(thermalSpeed),
      x58_xrayTarget(xrayTarget),
      x5c_xraySpeed(xraySpeed),
      x60_explicit(expl)

{
    if (close_enough(rangeDelta, zeus::CVector2f::skZero) && close_enough(colorDelta, 0.f))
        x61_nonZero = false;
    else
        x61_nonZero = true;
}

void CScriptDistanceFog::AcceptScriptMsg(EScriptObjectMessage msg, TUniqueId objId, CStateManager& stateMgr)
{
    CEntity::AcceptScriptMsg(msg, objId, stateMgr);

    if (GetAreaId() == kInvalidAreaId || !x30_24_active)
        return;

    if (msg == EScriptObjectMessage::InternalMessage13)
    {
        if (!x60_explicit)
            return;
        CGameArea::CAreaFog* fog = stateMgr.GetWorld()->GetArea(GetAreaId())->AreaFog();
        if (x34_mode == ERglFogMode::None)
            fog->DisableFog();
        else
            fog->SetFogExplicit(x34_mode, x38_color, x3c_range);
    }
    else if (msg == EScriptObjectMessage::Action)
    {
        if (!x61_nonZero)
            return;

        CGameArea::CAreaFog* fog = stateMgr.GetWorld()->GetArea(GetAreaId())->AreaFog();
        if (x34_mode == ERglFogMode::None)
            fog->RollFogOut(x48_rangeDelta.x, x44_colorDelta, x38_color);
        else
            fog->FadeFog(x34_mode, x38_color, x3c_range, x44_colorDelta, x48_rangeDelta);

        if (close_enough(x54_thermalSpeed, 0.f) && !close_enough(x5c_xraySpeed, 0.f))
        {
            CWorld* world = stateMgr.GetWorld();
            CGameArea* area = world->GetArea(GetAreaId());
            area->SetXRaySpeedAndTarget(x5c_xraySpeed, x58_xrayTarget);
        }
        else
        {
            CWorld* world = stateMgr.GetWorld();
            CGameArea* area = world->GetArea(GetAreaId());
            area->SetThermalSpeedAndTarget(x54_thermalSpeed, x50_thermalTarget);
        }
    }
}
}