add new object streamer functions

more tools for developers

Author: Xenius97

Client/mods/deathmatch/logic/CClientGame.cpp

@@ -431,6 +431,11 @@ CClientGame::~CClientGame()
     g_pClientGame->GetManager()->GetObjectStreamer()->ResetStreamerMaxSwaps();
     g_pClientGame->GetManager()->GetObjectStreamer()->ResetStreamerFurthestInLimit();
 
+    // Reset streamer limits
+    CClientObjectManager* pObjectManager = g_pClientGame->GetObjectManager();
+    if (pObjectManager)
+        pObjectManager->ResetMaxObjectStreamCount();
+
     // Reset camera shaking
     g_pGame->GetCamera()->SetShakeForce(0.0f);
 

Client/mods/deathmatch/logic/CClientObjectManager.cpp

@@ -245,8 +245,12 @@ bool CClientObjectManager::StaticIsLowLodObjectLimitReached()
 
 bool CClientObjectManager::IsObjectLimitReached()
 {
-    if (IsHardObjectLimitReached() || m_uiStreamedInCount >= m_uiMaxStreamedInCount)
+    if (IsHardObjectLimitReached() || (m_uiCustomMaxStreamedInCount > 0 && m_uiStreamedInCount >= m_uiCustomMaxStreamedInCount) ||
+        m_uiStreamedInCount >= m_uiMaxStreamedInCount)
+    {
         return true;
+    }
+
     return false;
 }
 
@@ -333,3 +337,24 @@ bool CClientObjectManager::Exists(CClientObject* pObject)
 {
     return ListContains(m_Objects, pObject);
 }
+
+void CClientObjectManager::SetMaxObjectStreamCount(int cValue)
+{
+    if (cValue < m_uiMaxStreamedInCount)
+        throw std::invalid_argument("Custom limit must be greater than or equal to default limit");
+
+    m_uiCustomMaxStreamedInCount = cValue;
+}
+
+void CClientObjectManager::ResetMaxObjectStreamCount()
+{
+    m_uiCustomMaxStreamedInCount = 0;
+}
+
+int CClientObjectManager::GetMaxObjectStreamCount()
+{
+    if (m_uiCustomMaxStreamedInCount > 0)
+        return m_uiCustomMaxStreamedInCount;
+
+    return m_uiMaxStreamedInCount;
+}

Client/mods/deathmatch/logic/CClientObjectManager.h

@@ -42,6 +42,9 @@ class CClientObjectManager
     bool        IsObjectLimitReached();
     bool        IsLowLodObjectLimitReached();
     bool        IsHardObjectLimitReached();
+    int         GetMaxObjectStreamCount();
+    void        SetMaxObjectStreamCount(int customCount);
+    void        ResetMaxObjectStreamCount();
 
     void RestreamObjects(unsigned short usModel);
     void RestreamAllObjects();
@@ -58,6 +61,7 @@ class CClientObjectManager
     int  m_iEntryInfoNodeEntries;
     int  m_iPointerNodeDoubleLinkEntries;
     uint m_uiMaxStreamedInCount;
+    uint m_uiCustomMaxStreamedInCount;
     uint m_uiMaxLowLodStreamedInCount;
     uint m_uiStreamedInCount;
     uint m_uiLowLodStreamedInCount;

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

@@ -159,6 +159,9 @@ void CLuaEngineDefs::LoadFunctions()
         {"engineStreamingResetMaxSwaps", ArgumentParser<EngineStreamingResetMaxSwaps>},
         {"engineStreamingSetFurthestInLimit", ArgumentParser<EngineStreamingSetFurthestInLimit>},
         {"engineStreamingResetFurthestInLimit", ArgumentParser<EngineStreamingResetFurthestInLimit>},
+        {"engineSetMaxObjectStreamCount", ArgumentParser<EngineSetMaxObjectStreamCount>},
+        {"engineResetMaxObjectStreamCount", ArgumentParser<EngineResetMaxObjectStreamCount>},
+        {"engineGetMaxObjectStreamCount", ArgumentParser<EngineGetMaxObjectStreamCount>},
 
 
         // CLuaCFunctions::AddFunction ( "engineReplaceMatchingAtomics", EngineReplaceMatchingAtomics );
@@ -230,6 +233,9 @@ void CLuaEngineDefs::AddClass(lua_State* luaVM)
         lua_classfunction(luaVM, "resetMaxSwaps", "engineStreamingResetMaxSwaps");
         lua_classfunction(luaVM, "setFurthestInLimit", "engineStreamingSetFurthestInLimit");
         lua_classfunction(luaVM, "resetFurthestInLimit", "engineStreamingResetFurthestInLimit");
+        lua_classfunction(luaVM, "setMaxObjectStreamCount", "engineSetMaxObjectStreamCount");
+        lua_classfunction(luaVM, "getMaxObjectStreamCount", "engineGetMaxObjectStreamCount");
+        lua_classfunction(luaVM, "resetMaxObjectStreamCount", "engineResetMaxObjectStreamCount");
 
         lua_classvariable(luaVM, "memorySize", "engineStreamingSetMemorySize", "engineStreamingGetMemorySize");
         lua_classvariable(luaVM, "bufferSize", "engineStreamingSetBufferSize", "engineStreamingGetBufferSize");
@@ -2668,3 +2674,26 @@ void CLuaEngineDefs::EngineStreamingResetFurthestInLimit()
 {
     g_pClientGame->GetManager()->GetObjectStreamer()->ResetStreamerFurthestInLimit();
 }
+
+void CLuaEngineDefs::EngineSetMaxObjectStreamCount(int limit)
+{
+    CClientObjectManager* pObjectManager = g_pClientGame->GetObjectManager();
+    if (pObjectManager)
+        pObjectManager->SetMaxObjectStreamCount(limit);
+}
+
+void CLuaEngineDefs::EngineResetMaxObjectStreamCount()
+{
+    CClientObjectManager* pObjectManager = g_pClientGame->GetObjectManager();
+    if (pObjectManager)
+        pObjectManager->ResetMaxObjectStreamCount();
+}
+
+int CLuaEngineDefs::EngineGetMaxObjectStreamCount()
+{
+    CClientObjectManager* pObjectManager = g_pClientGame->GetObjectManager();
+    if (pObjectManager)
+        return pObjectManager->GetMaxObjectStreamCount();
+
+    return 0;
+}

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

@@ -107,6 +107,10 @@ class CLuaEngineDefs : public CLuaDefs
     static void EngineStreamingSetFurthestInLimit(int limit);
     static void EngineStreamingResetFurthestInLimit();
 
+    static void EngineSetMaxObjectStreamCount(int limit);
+    static void EngineResetMaxObjectStreamCount();
+    static int EngineGetMaxObjectStreamCount();
+
 private:
     static void AddEngineColClass(lua_State* luaVM);
     static void AddEngineTxdClass(lua_State* luaVM);