// Copyright (C) 2015 Jérôme Leclercq
// This file is part of the "Nazara Engine - Graphics 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/Prerequesites.hpp>
#include <atomic>
#include <memory>

class NzMemoryPool
{
	public:
		NzMemoryPool(unsigned int blockSize, unsigned int size = 1024, bool canGrow = true);
		NzMemoryPool(const NzMemoryPool&) = delete;
		NzMemoryPool(NzMemoryPool&& pool) noexcept;
		~NzMemoryPool() = default;

		void* Allocate(unsigned int size);
		template<typename T> void Delete(T* ptr);
		void Free(void* ptr);

		unsigned int GetBlockSize() const;
		unsigned int GetFreeBlocks() const;
		unsigned int GetSize() const;

		template<typename T, typename... Args> T* New(Args&&... args);

		NzMemoryPool& operator=(const NzMemoryPool&) = delete;
		NzMemoryPool& operator=(NzMemoryPool&& pool) noexcept;

	private:
		NzMemoryPool(NzMemoryPool* pool);

		std::unique_ptr<void*[]> m_freeList;
		std::unique_ptr<nzUInt8[]> m_pool;
		std::unique_ptr<NzMemoryPool> m_next;
		std::atomic_uint m_freeCount;
		NzMemoryPool* m_previous;
		bool m_canGrow;
		unsigned int m_blockSize;
		unsigned int m_size;
};

#include <Nazara/Core/MemoryPool.inl>

#endif // NAZARA_MEMORYPOOL_HPP