setElementOnFire

Author: FileEX

Client/game_sa/CObjectSA.cpp

@@ -15,6 +15,7 @@
 #include "CPoolsSA.h"
 #include "CRopesSA.h"
 #include "CWorldSA.h"
+#include "CFireManagerSA.h"
 
 extern CGameSA* pGame;
 
@@ -304,3 +305,32 @@ void CObjectSA::ResetScale()
 {
     SetScale(1.0f, 1.0f, 1.0f);
 }
+
+void CObjectSA::SetOnFire(bool onFire)
+{
+    CObjectSAInterface* objectInterface = GetObjectInterface();
+    if ((onFire && objectInterface->pFire) || (!onFire && !objectInterface->pFire))
+        return;
+
+    auto* fireManager = static_cast<CFireManagerSA*>(pGame->GetFireManager());
+
+    if (onFire)
+    {
+        CFire* fire = fireManager->StartFire(this, nullptr, static_cast<float>(DEFAULT_FIRE_PARTICLE_SIZE));
+        if (!fire)
+            return;
+
+        fire->SetTarget(this);
+        fire->SetStrength(1.0f);
+        fire->Ignite();
+        fire->SetNumGenerationsAllowed(0);
+
+        objectInterface->pFire = fire->GetInterface();
+    }
+    else
+    {
+        CFire* fire = fireManager->GetFire(objectInterface->pFire);
+        if (fire)
+            fire->Extinguish();
+    }
+}

Client/game_sa/CObjectSA.h

@@ -153,6 +153,9 @@ class CObjectSA : public virtual CObject, public virtual CPhysicalSA
     CVector* GetScale();
     void     ResetScale();
 
+    bool IsOnFire() { return GetObjectInterface()->pFire != nullptr; }
+    void SetOnFire(bool onFire);
+
 private:
     void CheckForGangTag();
-};
+};

Client/game_sa/CVehicleSA.cpp

@@ -25,6 +25,7 @@
 #include "CVisibilityPluginsSA.h"
 #include "CWorldSA.h"
 #include "gamesa_renderware.h"
+#include "CFireManagerSA.h"
 
 extern CGameSA* pGame;
 
@@ -1833,6 +1834,35 @@ void CVehicleSA::OnChangingPosition(const CVector& vecNewPosition)
     }
 }
 
+void CVehicleSA::SetOnFire(bool onFire)
+{
+    CVehicleSAInterface* vehicleInterface = GetVehicleInterface();
+    if ((onFire && vehicleInterface->m_pFire) || (!onFire && !vehicleInterface->m_pFire))
+        return;
+
+    auto* fireManager = static_cast<CFireManagerSA*>(pGame->GetFireManager());
+
+    if (onFire)
+    {
+        CFire* fire = fireManager->StartFire(this, nullptr, static_cast<float>(DEFAULT_FIRE_PARTICLE_SIZE));
+        if (!fire)
+            return;
+
+        fire->SetTarget(this);
+        fire->SetStrength(1.0f);
+        fire->Ignite();
+        fire->SetNumGenerationsAllowed(0);
+
+        vehicleInterface->m_pFire = fire->GetInterface();
+    }
+    else
+    {
+        CFire* fire = fireManager->GetFire(vehicleInterface->m_pFire);
+        if (fire)
+            fire->Extinguish();
+    }
+}
+
 void CVehicleSA::StaticSetHooks()
 {
     // Setup vehicle sun glare hook

Client/game_sa/CVehicleSA.h

@@ -309,7 +309,7 @@ class CVehicleSAInterface : public CPhysicalSAInterface
 
     unsigned char m_nSpecialColModel;
     CEntity*      pEntityWeAreOnForVisibilityCheck;
-    CFire*        m_pFire;
+    CFireSAInterface*        m_pFire;
 
     float m_fSteerAngle;               // +1172
     float m_f2ndSteerAngle;            // used for steering 2nd set of wheels or elevators etc..
@@ -690,6 +690,9 @@ class CVehicleSA : public virtual CVehicle, public virtual CPhysicalSA
     CVector*       GetDummyPositions() { return m_dummyPositions.data(); }
     const CVector* GetDummyPositions() const override { return m_dummyPositions.data(); }
 
+    bool IsOnFire() { return GetVehicleInterface()->m_pFire != nullptr; }
+    void SetOnFire(bool onFire);
+
     static void StaticSetHooks();
     static void SetVehiclesSunGlareEnabled(bool bEnabled);
     static bool GetVehiclesSunGlareEnabled();

Client/mods/deathmatch/logic/CClientObject.cpp

@@ -429,6 +429,20 @@ void CClientObject::SetHealth(float fHealth)
     m_fHealth = fHealth;
 }
 
+bool CClientObject::IsOnFire() const noexcept
+{
+    if (!m_pObject)
+        return false;
+
+    return m_pObject->IsOnFire();
+}
+
+void CClientObject::SetOnFire(bool onFire) noexcept
+{
+    if (m_pObject)
+        m_pObject->SetOnFire(onFire);
+}
+
 void CClientObject::StreamIn(bool bInstantly)
 {
     // Don't stream the object in, if respawn is disabled and the object is broken

Client/mods/deathmatch/logic/CClientObject.h

@@ -119,6 +119,9 @@ class CClientObject : public CClientStreamElement
     bool IsBeingRespawned() { return m_bBeingRespawned; };
     void SetBeingRespawned(bool bBeingRespawned) { m_bBeingRespawned = bBeingRespawned; };
 
+    bool IsOnFire() const noexcept;
+    void SetOnFire(bool onFire) noexcept;
+
 protected:
     void StreamIn(bool bInstantly);
     void StreamOut();

Client/mods/deathmatch/logic/CClientVehicle.cpp

@@ -5058,3 +5058,17 @@ CVector CClientVehicle::GetEntryPoint(std::uint32_t entryPointIndex)
 
     return entryPoint;
 }
+
+bool CClientVehicle::IsOnFire() const noexcept
+{
+    if (!m_pVehicle)
+        return false;
+
+    return m_pVehicle->IsOnFire();
+}
+
+void CClientVehicle::SetOnFire(bool onFire) noexcept
+{
+    if (m_pVehicle)
+        m_pVehicle->SetOnFire(onFire);
+}

Client/mods/deathmatch/logic/CClientVehicle.h

@@ -547,6 +547,9 @@ class CClientVehicle : public CClientStreamElement
 
     CVector GetEntryPoint(std::uint32_t entryPointIndex);
 
+    bool IsOnFire() const noexcept;
+    void SetOnFire(bool onFire) noexcept;
+
 protected:
     void ConvertComponentRotationBase(const SString& vehicleComponent, CVector& vecInOutRotation, EComponentBaseType inputBase, EComponentBaseType outputBase);
     void ConvertComponentPositionBase(const SString& vehicleComponent, CVector& vecInOutPosition, EComponentBaseType inputBase, EComponentBaseType outputBase);

Client/mods/deathmatch/logic/CNetAPI.cpp

@@ -1541,6 +1541,9 @@ void CNetAPI::ReadVehiclePuresync(CClientPlayer* pPlayer, CClientVehicle* pVehic
 
     pPlayer->SetControllerState(ControllerState);
 
+    if (BitStream.Can(eBitStreamVersion::SetElementOnFire))
+        pVehicle->SetOnFire(BitStream.ReadBit());
+
     // Remember now as the last puresync time
     CVector vecPosition;
     pVehicle->GetPosition(vecPosition);
@@ -1766,6 +1769,9 @@ void CNetAPI::WriteVehiclePuresync(CClientPed* pPlayerModel, CClientVehicle* pVe
         BitStream.WriteBit(ControllerState.RightShoulder2 != 0);
     }
 
+    if (BitStream.Can(eBitStreamVersion::SetElementOnFire))
+        BitStream.WriteBit(pVehicle->IsOnFire());
+
     // Write the sent position to the interpolator
     AddInterpolation(vecPosition);
 }

Client/mods/deathmatch/logic/luadefs/CLuaElementDefs.cpp

@@ -71,6 +71,7 @@ void CLuaElementDefs::LoadFunctions()
         {"isElementLowLOD", IsElementLowLod},
         {"isElementCallPropagationEnabled", IsElementCallPropagationEnabled},
         {"isElementWaitingForGroundToLoad", IsElementWaitingForGroundToLoad},
+        {"isElementOnFire", ArgumentParser<IsElementOnFire>},
 
         // Element set funcs
         {"createElement", CreateElement},
@@ -99,6 +100,7 @@ void CLuaElementDefs::LoadFunctions()
         {"setElementFrozen", SetElementFrozen},
         {"setLowLODElement", ArgumentParser<SetLowLodElement>},
         {"setElementCallPropagationEnabled", SetElementCallPropagationEnabled},
+        {"setElementOnFire", ArgumentParser<SetElementOnFire>},
     };
 
     // Add functions
@@ -169,6 +171,7 @@ void CLuaElementDefs::AddClass(lua_State* luaVM)
     lua_classfunction(luaVM, "getAttachedOffsets", "getElementAttachedOffsets");
     lua_classfunction(luaVM, "getData", "getElementData");
     lua_classfunction(luaVM, "getAllData", "getAllElementData");
+    lua_classfunction(luaVM, "isOnFire", "isElementOnFire");
 
     lua_classfunction(luaVM, "setAttachedOffsets", "setElementAttachedOffsets");
     lua_classfunction(luaVM, "setData", "setElementData");
@@ -191,6 +194,7 @@ void CLuaElementDefs::AddClass(lua_State* luaVM)
     lua_classfunction(luaVM, "setLowLOD", "setLowLODElement");
     lua_classfunction(luaVM, "setCallPropagationEnabled", "setElementCallPropagationEnabled");
     lua_classfunction(luaVM, "setStreamable", "setElementStreamable");
+    lua_classfunction(luaVM, "setOnFire", "setElementOnFire");
 
     lua_classvariable(luaVM, "callPropagationEnabled", "setElementCallPropagationEnabled", "isElementCallPropagationEnabled");
     lua_classvariable(luaVM, "waitingForGroundToLoad", NULL, "isElementWaitingForGroundToLoad");
@@ -225,6 +229,7 @@ void CLuaElementDefs::AddClass(lua_State* luaVM)
     lua_classvariable(luaVM, "velocity", SetElementVelocity, OOP_GetElementVelocity);
     lua_classvariable(luaVM, "angularVelocity", SetElementAngularVelocity, OOP_GetElementTurnVelocity);
     lua_classvariable(luaVM, "isElement", NULL, "isElement");
+    lua_classvariable(luaVM, "onFire", "setElementOnFire", "isElementOnFire");
     // TODO: Support element data: player.data["age"] = 1337; <=> setElementData(player, "age", 1337)
 
     lua_registerclass(luaVM, "Element");
@@ -2509,6 +2514,32 @@ bool CLuaElementDefs::SetLowLodElement(lua_State* luaVM, CClientEntity* pEntity,
     return CStaticFunctionDefinitions::SetLowLodElement(*pEntity, pLowLodEntity.value_or(nullptr));
 }
 
+bool CLuaElementDefs::SetElementOnFire(CClientEntity* entity, bool onFire)
+{
+    switch (entity->GetType())
+    {
+        case CCLIENTPED:
+        case CCLIENTPLAYER:
+        {
+            static_cast<CClientPed*>(entity)->SetOnFire(onFire);
+            return true;
+        }
+        case CCLIENTVEHICLE:
+        {
+            static_cast<CClientVehicle*>(entity)->SetOnFire(onFire);
+            return true;
+        }
+        case CCLIENTOBJECT:
+        case CCLIENTWEAPON:
+        {
+            static_cast<CClientObject*>(entity)->SetOnFire(onFire);
+            return true;
+        }
+    }
+
+    return false;
+}
+
 int CLuaElementDefs::IsElementLowLod(lua_State* luaVM)
 {
     //  bool isElementLowLOD ( element theElement )
@@ -2604,3 +2635,18 @@ int CLuaElementDefs::IsElementWaitingForGroundToLoad(lua_State* luaVM)
     lua_pushboolean(luaVM, false);
     return 1;
 }
+
+bool CLuaElementDefs::IsElementOnFire(CClientEntity* entity)
+{
+    switch (entity->GetType())
+    {
+        case CCLIENTPED:
+            return static_cast<CClientPed*>(entity)->IsOnFire();
+        case CCLIENTVEHICLE:
+            return static_cast<CClientVehicle*>(entity)->IsOnFire();
+        case CCLIENTOBJECT:
+            return static_cast<CClientObject*>(entity)->IsOnFire();
+    }
+
+    return false;
+}