Add engineSetPoolCapacity support

TheNormalnij · Jan 29, 2025 · 36f88bc · 36f88bc
1 parent 53ad82a
commit 36f88bc
Show file tree

Hide file tree

Showing 6 changed files with 80 additions and 4 deletions.
diff --git a/Client/game_sa/CDynamicPool.h b/Client/game_sa/CDynamicPool.h
@@ -133,6 +133,53 @@ class CDynamicPool
         return size;
     }
 
+    bool SetCapacity(std::size_t newSize) {
+        if (newSize == 0)
+            return false;
+
+        std::size_t currentSize = GetCapacity();
+
+        if (currentSize == newSize)
+            return false;
+        else if (currentSize < newSize)
+        {
+            // Grown
+            while (currentSize < newSize)
+            {
+                try
+                {
+                    auto* nextPart = AllocateNewPart();
+                    currentSize += nextPart->GetCapacity();
+                }
+                catch (const std::bad_alloc& ex)
+                {
+                    return false;
+                }
+            }
+        }
+        else
+        {
+            // Shrink
+            while (true)
+            {
+                auto* part = m_poolParts.back().get();
+                if (part->GetUsedSize() != 0)
+                    return false;
+
+                currentSize -= part->GetCapacity();
+                if (currentSize < newSize)
+                    return false;
+
+                m_poolParts.pop_back();
+
+                if (currentSize == newSize)
+                    return true;
+            }
+        }
+
+        return true;
+    }
+
 private:
     CDynamicPoolPart<PoolObjT>* AllocateNewPart()
     {

diff --git a/Client/game_sa/CPoolsSA.h b/Client/game_sa/CPoolsSA.h
@@ -98,7 +98,7 @@ class CPoolsSA : public CPools
     CBuildingsPool& GetBuildingsPool() noexcept override { return m_BuildingsPool; };
     CDummyPool&     GetDummyPool() noexcept { return m_DummyPool; };
     CTxdPool&       GetTxdPool() noexcept { return m_TxdPool; };
-    CPtrNodeSingleLinkPoolSA& GetPtrNodeSingleLinkPool() noexcept { return m_PtrNodeSingleLinkPool; };
+    CPtrNodeSingleLinkPool& GetPtrNodeSingleLinkPool() noexcept override { return m_PtrNodeSingleLinkPool; };
 
 private:
     // Pools

diff --git a/Client/game_sa/CPtrNodeSingleLinkPoolSA.h b/Client/game_sa/CPtrNodeSingleLinkPoolSA.h
@@ -14,17 +14,20 @@
 #include "CPoolSAInterface.h"
 #include "CDynamicPool.h"
 #include "CPtrNodeSingleListSA.h"
+#include <game/CPtrNodeSingleLinkPool.h>
 
-class CPtrNodeSingleLinkPoolSA
+class CPtrNodeSingleLinkPoolSA final : public CPtrNodeSingleLinkPool
 {
 public:
     using pool_item_t = CPtrNodeSingleLink<void*>;
     using pool_t = CDynamicPool<pool_item_t, PoolGrownByHalfStrategy<MAX_POINTER_SINGLE_LINKS>>;
 
     CPtrNodeSingleLinkPoolSA();
 
-    int GetCapacity() const { return m_customPool->GetCapacity(); }
-    int GetUsedSize() const { return m_customPool->GetUsedSize(); }
+    std::size_t GetCapacity() const override { return m_customPool->GetCapacity(); }
+    std::size_t GetUsedSize() const override { return m_customPool->GetUsedSize(); }
+
+    bool Resize(std::size_t newSize) override { return m_customPool->SetCapacity(newSize); };
 
     static auto* GetPoolInstance() { return m_customPool; } 
     static void StaticSetHooks();

diff --git a/Client/mods/deathmatch/logic/luadefs/CLuaEngineDefs.cpp b/Client/mods/deathmatch/logic/luadefs/CLuaEngineDefs.cpp
@@ -13,6 +13,7 @@
 #include <game/CColPoint.h>
 #include <game/CObjectGroupPhysicalProperties.h>
 #include <game/CStreaming.h>
+#include <game/CPtrNodeSingleLinkPool.h>
 #include <lua/CLuaFunctionParser.h>
 #include "CLuaEngineDefs.h"
 
@@ -2505,6 +2506,10 @@ bool CLuaEngineDefs::EngineSetPoolCapacity(lua_State* luaVM, ePools pool, size_t
         {
             return m_pBuildingManager->SetPoolCapacity(newSize);
         }
+        case ePools::POINTER_SINGLE_LINK_POOL:
+        {
+            return g_pGame->GetPools()->GetPtrNodeSingleLinkPool().Resize(newSize);
+        }
         default:
             throw std::invalid_argument("Can not change this pool capacity");
     }

diff --git a/Client/sdk/game/CPools.h b/Client/sdk/game/CPools.h
@@ -15,6 +15,7 @@
 #include "CBuildingsPool.h"
 #include "CDummyPool.h"
 #include "CTxdPool.h"
+#include "CPtrNodeSingleLinkPool.h"
 
 class CClientEntity;
 class CEntity;
@@ -111,4 +112,5 @@ class CPools
     virtual CBuildingsPool& GetBuildingsPool() noexcept = 0;
     virtual CDummyPool&     GetDummyPool() noexcept = 0;
     virtual CTxdPool&       GetTxdPool() noexcept = 0;
+    virtual CPtrNodeSingleLinkPool& GetPtrNodeSingleLinkPool() noexcept = 0;
 };
diff --git a/Client/sdk/game/CPtrNodeSingleLinkPool.h b/Client/sdk/game/CPtrNodeSingleLinkPool.h
@@ -0,0 +1,19 @@
+/*****************************************************************************
+ *
+ *  PROJECT:     Multi Theft Auto
+ *  LICENSE:     See LICENSE in the top level directory
+ *  FILE:        sdk/game/CPtrNodeSingleLinkPool.h
+ *
+ *  Multi Theft Auto is available from http://www.multitheftauto.com/
+ *
+ *****************************************************************************/
+
+#pragma once
+
+class CPtrNodeSingleLinkPool
+{
+public:
+    virtual bool Resize(std::size_t size) = 0;
+    virtual std::size_t GetCapacity() const = 0;
+    virtual std::size_t GetUsedSize() const = 0;
+};