mirror of
				https://github.com/OpenMW/openmw.git
				synced 2025-11-04 15:56:42 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			83 lines
		
	
	
	
		
			1.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			83 lines
		
	
	
	
		
			1.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
#ifndef OPENMW_COMPONENTS_MISC_OBJECTPOOL_H
 | 
						|
#define OPENMW_COMPONENTS_MISC_OBJECTPOOL_H
 | 
						|
 | 
						|
#include <deque>
 | 
						|
#include <memory>
 | 
						|
#include <vector>
 | 
						|
 | 
						|
namespace Misc
 | 
						|
{
 | 
						|
    template <class T>
 | 
						|
    class ObjectPool;
 | 
						|
 | 
						|
    template <class T>
 | 
						|
    class ObjectPtrDeleter
 | 
						|
    {
 | 
						|
        public:
 | 
						|
            ObjectPtrDeleter(std::nullptr_t)
 | 
						|
                : mPool(nullptr) {}
 | 
						|
 | 
						|
            ObjectPtrDeleter(ObjectPool<T>& pool)
 | 
						|
                : mPool(&pool) {}
 | 
						|
 | 
						|
            void operator()(T* object) const
 | 
						|
            {
 | 
						|
                mPool->recycle(object);
 | 
						|
            }
 | 
						|
 | 
						|
        private:
 | 
						|
            ObjectPool<T>* mPool;
 | 
						|
    };
 | 
						|
 | 
						|
    template <class T>
 | 
						|
    struct ObjectPtr final : std::unique_ptr<T, ObjectPtrDeleter<T>>
 | 
						|
    {
 | 
						|
        using std::unique_ptr<T, ObjectPtrDeleter<T>>::unique_ptr;
 | 
						|
        using std::unique_ptr<T, ObjectPtrDeleter<T>>::operator=;
 | 
						|
 | 
						|
        ObjectPtr()
 | 
						|
            : ObjectPtr(nullptr) {}
 | 
						|
 | 
						|
        ObjectPtr(std::nullptr_t)
 | 
						|
            : std::unique_ptr<T, ObjectPtrDeleter<T>>(nullptr, nullptr) {}
 | 
						|
    };
 | 
						|
 | 
						|
    template <class T>
 | 
						|
    class ObjectPool
 | 
						|
    {
 | 
						|
        friend class ObjectPtrDeleter<T>;
 | 
						|
 | 
						|
        public:
 | 
						|
            ObjectPool()
 | 
						|
                : mObjects(std::make_unique<std::deque<T>>()) {}
 | 
						|
 | 
						|
            ObjectPtr<T> get()
 | 
						|
            {
 | 
						|
                T* object;
 | 
						|
 | 
						|
                if (!mUnused.empty())
 | 
						|
                {
 | 
						|
                    object = mUnused.back();
 | 
						|
                    mUnused.pop_back();
 | 
						|
                }
 | 
						|
                else
 | 
						|
                {
 | 
						|
                    mObjects->emplace_back();
 | 
						|
                    object = &mObjects->back();
 | 
						|
                }
 | 
						|
 | 
						|
                return ObjectPtr<T>(object, ObjectPtrDeleter<T>(*this));
 | 
						|
            }
 | 
						|
 | 
						|
        private:
 | 
						|
            std::unique_ptr<std::deque<T>> mObjects;
 | 
						|
            std::vector<T*> mUnused;
 | 
						|
 | 
						|
            void recycle(T* object)
 | 
						|
            {
 | 
						|
                mUnused.push_back(object);
 | 
						|
            }
 | 
						|
    };
 | 
						|
}
 | 
						|
 | 
						|
#endif
 |