Graphics: Use memory pools for render elements

include/Nazara/Graphics/ElementRendererRegistry.inl

@@ -0,0 +1,84 @@
+// Copyright (C) 2022 Jérôme "Lynix" Leclercq (lynix680@gmail.com)
+// This file is part of the "Nazara Engine - Graphics module"
+// For conditions of distribution and use, see copyright notice in Config.hpp
+
+#include <Nazara/Graphics/ElementRendererRegistry.hpp>
+#include <Nazara/Utils/Algorithm.hpp>
+#include <Nazara/Graphics/Debug.hpp>
+
+namespace Nz
+{
+	template<typename T, typename ...Args>
+	RenderElementOwner ElementRendererRegistry::AllocateElement(Args&&... args)
+	{
+		ElementRenderer& elementRenderer = GetElementRenderer(SafeCast<std::size_t>(T::ElementType));
+		RenderElementPool<T>& pool = SafeCast<RenderElementPool<T>&>(elementRenderer.GetPool());
+
+		return pool.Allocate(std::forward<Args>(args)...);
+	}
+
+	inline ElementRenderer& ElementRendererRegistry::GetElementRenderer(std::size_t elementIndex)
+	{
+		assert(elementIndex < m_elementRenderers.size());
+		return *m_elementRenderers[elementIndex];
+	}
+
+	inline std::size_t ElementRendererRegistry::GetElementRendererCount() const
+	{
+		return m_elementRenderers.size();
+	}
+
+	template<typename F>
+	void ElementRendererRegistry::ForEachElementRenderer(F&& callback)
+	{
+		for (std::size_t i = 0; i < m_elementRenderers.size(); ++i)
+		{
+			if (m_elementRenderers[i])
+				callback(i, *m_elementRenderers[i]);
+		}
+	}
+
+	template<typename F>
+	void ElementRendererRegistry::ProcessRenderQueue(const RenderQueue<const RenderElement*>& renderQueue, F&& callback)
+	{
+		if (renderQueue.empty())
+			return;
+
+		auto it = renderQueue.begin();
+		auto itEnd = renderQueue.end();
+		while (it != itEnd)
+		{
+			const RenderElement* element = *it;
+			UInt8 elementType = element->GetElementType();
+
+			const Pointer<const RenderElement>* first = it;
+
+			++it;
+			while (it != itEnd && (*it)->GetElementType() == elementType)
+				++it;
+
+			std::size_t count = it - first;
+			if (elementType >= m_elementRenderers.size() || !m_elementRenderers[elementType])
+				continue;
+
+			callback(elementType, first, count);
+		}
+	}
+
+	template<typename T>
+	inline void ElementRendererRegistry::RegisterElementRenderer(std::unique_ptr<ElementRenderer> renderer)
+	{
+		return RegisterElementRenderer(SafeCast<std::size_t>(T::ElementType), std::move(renderer));
+	}
+
+	inline void ElementRendererRegistry::RegisterElementRenderer(std::size_t elementIndex, std::unique_ptr<ElementRenderer> renderer)
+	{
+		if (elementIndex >= m_elementRenderers.size())
+			m_elementRenderers.resize(elementIndex + 1);
+
+		assert(!m_elementRenderers[elementIndex]);
+		m_elementRenderers[elementIndex] = std::move(renderer);
+	}
+}
+
+#include <Nazara/Graphics/DebugOff.hpp>