Pool.hpp

Go to the documentation of this file.

#pragma once
#include "Core.hpp"
#include <mutex>
namespace Foundation::Core {
    template<typename K, typename V>
    class Pool {
        Allocator* mAlloc;
        Vector<K> mKeys;
        mutable Vector<UniquePtr<V>> mValues;        
        mutable std::mutex mMutex;
        K mTop{};
        void Push(K key) {            
            mKeys.push_back(key);
            if (key >= mValues.size())
                mValues.resize(key + 1);
        }     
        void CheckContains(K key) const {
            CHECK_MSG(key < mValues.size() && mValues[key], "Key not allocated"); 
        }
        K PopKey() {
            K key{};
            if (mKeys.empty())
                key = mTop++;
            else
                key = mKeys.back(), mKeys.pop_back();
            if (key >= mValues.size())
                mValues.resize(key + 1);
            return key;
        }
    public:
        Pool(Allocator* alloc) : mAlloc(alloc), mKeys(alloc), mValues(alloc) {}
        bool Contains(K key) const {
            std::unique_lock lock(mMutex);
            return key < mValues.size() && mValues[key];
        }
        V* At(K key) const {
            std::unique_lock lock(mMutex);
            CheckContains(key);
            return mValues[key].get();
        }
        template <typename... Args>
        K Pop(Args&&... args)
        {
            std::unique_lock lock(mMutex);
            K key = PopKey();
            mValues[key] = ConstructUnique<V>(mAlloc, std::forward<Args>(args)...);
            return key;
        }
        template <typename Derived, typename... Args>
        K PopBase(Args&&... args)
        {
            std::unique_lock lock(mMutex);
            K key = PopKey();
            mValues[key] = ConstructUniqueBase<V, Derived>(mAlloc, std::forward<Args>(args)...);
            return key;
        }
        void Free(K key) {
            std::unique_lock lock(mMutex);
            CheckContains(key);
            mValues[key].reset();            
            Push(key);
        }
    };
}