56 lines
1.4 KiB
C++
56 lines
1.4 KiB
C++
// Copyright (C) 2021 Jérôme "Lynix" Leclercq (lynix680@gmail.com)
|
|
// This file is part of the "Nazara Engine - Core module"
|
|
// For conditions of distribution and use, see copyright notice in Config.hpp
|
|
|
|
#pragma once
|
|
|
|
#ifndef NAZARA_MEMORYPOOL_HPP
|
|
#define NAZARA_MEMORYPOOL_HPP
|
|
|
|
#include <Nazara/Prerequisites.hpp>
|
|
#include <atomic>
|
|
#include <memory>
|
|
|
|
namespace Nz
|
|
{
|
|
class MemoryPool
|
|
{
|
|
public:
|
|
MemoryPool(unsigned int blockSize, unsigned int size = 1024, bool canGrow = true);
|
|
MemoryPool(const MemoryPool&) = delete;
|
|
MemoryPool(MemoryPool&& pool) noexcept;
|
|
~MemoryPool() = default;
|
|
|
|
void* Allocate(unsigned int size);
|
|
|
|
template<typename T> void Delete(T* ptr);
|
|
|
|
void Free(void* ptr);
|
|
|
|
inline unsigned int GetBlockSize() const;
|
|
inline unsigned int GetFreeBlocks() const;
|
|
inline unsigned int GetSize() const;
|
|
|
|
template<typename T, typename... Args> T* New(Args&&... args);
|
|
|
|
MemoryPool& operator=(const MemoryPool&) = delete;
|
|
MemoryPool& operator=(MemoryPool&& pool) noexcept;
|
|
|
|
private:
|
|
MemoryPool(MemoryPool* pool);
|
|
|
|
std::unique_ptr<void* []> m_freeList;
|
|
std::unique_ptr<UInt8[]> m_pool;
|
|
std::unique_ptr<MemoryPool> m_next;
|
|
std::atomic_uint m_freeCount;
|
|
MemoryPool* m_previous;
|
|
bool m_canGrow;
|
|
unsigned int m_blockSize;
|
|
unsigned int m_size;
|
|
};
|
|
}
|
|
|
|
#include <Nazara/Core/MemoryPool.inl>
|
|
|
|
#endif // NAZARA_MEMORYPOOL_HPP
|