Add LOD management functions

Client/mods/deathmatch/StdInc.h

@@ -130,6 +130,7 @@
 #include <luadefs/CLuaMarkerDefs.h>
 #include <luadefs/CLuaNetworkDefs.h>
 #include <luadefs/CLuaObjectDefs.h>
+#include <luadefs/CLuaModelDefs.h>
 #include <luadefs/CLuaPointLightDefs.h>
 #include <luadefs/CLuaPedDefs.h>
 #include <luadefs/CLuaPickupDefs.h>

Client/mods/deathmatch/logic/CClientModelManager.cpp

@@ -9,6 +9,8 @@
  *****************************************************************************/
 
 #include "StdInc.h"
+#include "CLodModels.h"
+
 CClientModelManager::CClientModelManager() : m_Models(std::make_unique<std::shared_ptr<CClientModel>[]>(g_pGame->GetBaseIDforCOL()))
 {
     const unsigned int uiMaxModelID = g_pGame->GetBaseIDforCOL();
@@ -21,6 +23,9 @@ CClientModelManager::CClientModelManager() : m_Models(std::make_unique<std::shar
 CClientModelManager::~CClientModelManager(void)
 {
     RemoveAll();
+
+    // Reset Level-Of-Detail system
+    CLodModels::ResetAllModelLOD();
 }
 
 void CClientModelManager::RemoveAll(void)
@@ -55,6 +60,11 @@ bool CClientModelManager::Remove(const std::shared_ptr<CClientModel>& pModel)
         m_Models[modelId]->RestoreEntitiesUsingThisModel();
         m_Models[modelId] = nullptr;
         m_modelCount--;
+
+        // Force reset the model in Level-Of-Detail system
+        CLodModels::ResetModelLODByHigh(modelId);
+        CLodModels::ResetModelLODByLow(modelId);
+
         return true;
     }
     return false;

Client/mods/deathmatch/logic/lua/CLuaManager.cpp

@@ -260,6 +260,7 @@ void CLuaManager::LoadCFunctions()
     CLuaMarkerDefs::LoadFunctions();
     CLuaNetworkDefs::LoadFunctions();
     CLuaObjectDefs::LoadFunctions();
+    CLuaModelDefs::LoadFunctions();
     CLuaPedDefs::LoadFunctions();
     CLuaPickupDefs::LoadFunctions();
     CLuaPlayerDefs::LoadFunctions();

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

@@ -0,0 +1,93 @@
+/*****************************************************************************
+ *
+ *  PROJECT:     Multi Theft Auto
+ *  LICENSE:     See LICENSE in the top level directory
+ *  FILE:        mods/shared_logic/luadefs/CLuaModelDefs.cpp
+ *  PURPOSE:     Lua model definitions class
+ *
+ *  Multi Theft Auto is available from http://www.multitheftauto.com/
+ *
+ *****************************************************************************/
+
+#include "StdInc.h"
+#include <lua/CLuaFunctionParser.h>
+#include "CLodModels.h"
+#include "CClientObjectManager.h"
+#include "CClientBuildingManager.h"
+
+void CLuaModelDefs::LoadFunctions()
+{
+    constexpr static const std::pair<const char*, lua_CFunction> functions[]{
+        // Util func
+        {"isValidModel", ArgumentParser<IsValidModel>},
+
+        // LOD funcs
+        {"getModelLowLOD", ArgumentParser<GetModelLowLOD>},
+        {"getModelHighLOD", ArgumentParser<GetModelHighLOD>},
+        {"setModelLOD", ArgumentParser<SetModelLOD>},
+        {"resetModelLODByHigh", ArgumentParser<ResetModelLODByHigh>},
+        {"resetModelLODByLow", ArgumentParser<ResetModelLODByLow>},
+        {"resetAllModelLOD", ArgumentParser<ResetAllModelLOD>},
+    };
+
+    // Add functions
+    for (const auto& [name, func] : functions)
+        CLuaCFunctions::AddFunction(name, func);
+}
+
+bool CLuaModelDefs::IsValidModel(std::string modelPurpose, std::uint32_t id)
+{
+    if (modelPurpose == "object")
+        return CClientObjectManager::IsValidModel(id);
+    else if (modelPurpose == "weapon")
+        return CClientPedManager::IsValidWeaponModel(id);
+    else if (modelPurpose == "upgrade")
+        return CVehicleUpgrades::IsUpgrade(id);
+    else if (modelPurpose == "building")
+        return CClientBuildingManager::IsValidModel(id);
+    else if (modelPurpose == "vehicle")
+        return CClientVehicleManager::IsValidModel(id);
+    else if (modelPurpose == "ped" || modelPurpose == "player")
+        return CClientPlayerManager::IsValidModel(id);
+
+    throw std::invalid_argument("Invalid model purpose passed, expected [object, weapon, upgrade, building, vehicle, ped, player]");
+}
+
+std::variant<bool, std::uint32_t> CLuaModelDefs::GetModelLowLOD(std::uint32_t hLODModel) noexcept
+{
+    return CLodModels::GetModelLowLOD(hLODModel);
+}
+
+std::variant<bool, std::uint32_t> CLuaModelDefs::GetModelHighLOD(std::uint32_t lLODModel) noexcept
+{
+    return CLodModels::GetModelHighLOD(lLODModel);
+}
+
+bool CLuaModelDefs::SetModelLOD(std::string modelPurpose, std::uint32_t hLODModel, std::uint32_t lLODModel)
+{
+    if (!(modelPurpose == "object" || modelPurpose == "building"))
+        throw std::invalid_argument("Invalid model purpose passed, the only options for LOD are 'object' or 'building'");
+
+    if (!IsValidModel(modelPurpose, hLODModel))
+        throw std::invalid_argument("Invalid High-LOD model ID passed");
+
+    if (!IsValidModel(modelPurpose, lLODModel))
+        throw std::invalid_argument("Invalid Low-LOD model ID passed");
+
+    return CLodModels::SetModelLOD(hLODModel, lLODModel);
+}
+
+bool CLuaModelDefs::ResetModelLODByHigh(std::uint32_t hLODModel) noexcept
+{
+    return CLodModels::ResetModelLODByHigh(hLODModel);
+}
+
+bool CLuaModelDefs::ResetModelLODByLow(std::uint32_t hLODModel) noexcept
+{
+    return CLodModels::ResetModelLODByLow(hLODModel);
+}
+
+void CLuaModelDefs::ResetAllModelLOD() noexcept
+{
+    CLodModels::ResetAllModelLOD();
+}

Client/mods/deathmatch/logic/luadefs/CLuaModelDefs.h

@@ -0,0 +1,30 @@
+/*****************************************************************************
+ *
+ *  PROJECT:     Multi Theft Auto
+ *  LICENSE:     See LICENSE in the top level directory
+ *  FILE:        mods/shared_logic/luadefs/CLuaModelDefs.h
+ *  PURPOSE:     Lua model definitions class header
+ *
+ *  Multi Theft Auto is available from http://www.multitheftauto.com/
+ *
+ *****************************************************************************/
+
+#pragma once
+#include "CLuaDefs.h"
+
+class CLuaModelDefs : public CLuaDefs
+{
+public:
+    static void LoadFunctions();
+
+    // Util func
+    static bool IsValidModel(std::string modelPurpose, std::uint32_t id);
+
+    // LOD funcs
+    static std::variant<bool, std::uint32_t> GetModelLowLOD(std::uint32_t hLODModel) noexcept;
+    static std::variant<bool, std::uint32_t> GetModelHighLOD(std::uint32_t lLODModel) noexcept;
+    static bool                              SetModelLOD(std::string modelPurpose, std::uint32_t hLODModel, std::uint32_t lLODModel);
+    static bool                              ResetModelLODByHigh(std::uint32_t hLODModel) noexcept;
+    static bool                              ResetModelLODByLow(std::uint32_t lLODModel) noexcept;
+    static void                              ResetAllModelLOD() noexcept;
+};