Add command buffers (WIP)

src/Nazara/VulkanRenderer/VulkanRenderImage.cpp

@@ -0,0 +1,93 @@
+// Copyright (C) 2020 Jérôme Leclercq
+// This file is part of the "Nazara Engine - Vulkan Renderer"
+// For conditions of distribution and use, see copyright notice in Config.hpp
+
+#include <Nazara/VulkanRenderer/VulkanRenderImage.hpp>
+#include <Nazara/VulkanRenderer/VkRenderWindow.hpp>
+#include <Nazara/VulkanRenderer/VulkanCommandBuffer.hpp>
+#include <Nazara/VulkanRenderer/VulkanCommandBufferBuilder.hpp>
+#include <stdexcept>
+#include <Nazara/VulkanRenderer/Debug.hpp>
+
+namespace Nz
+{
+	VulkanRenderImage::VulkanRenderImage(VkRenderWindow& owner) :
+	m_owner(owner),
+	m_uploadPool(m_owner.GetDevice(), 2 * 1024 * 1024)
+	{
+		Vk::QueueHandle& graphicsQueue = m_owner.GetGraphicsQueue();
+		if (!m_commandPool.Create(m_owner.GetDevice(), graphicsQueue.GetQueueFamilyIndex(), VK_COMMAND_POOL_CREATE_TRANSIENT_BIT | VK_COMMAND_POOL_CREATE_RESET_COMMAND_BUFFER_BIT))
+			throw std::runtime_error("failed to create command pool: " + TranslateVulkanError(m_commandPool.GetLastErrorCode()));
+
+		if (!m_imageAvailableSemaphore.Create(m_owner.GetDevice()))
+			throw std::runtime_error("failed to create image available semaphore: " + TranslateVulkanError(m_imageAvailableSemaphore.GetLastErrorCode()));
+
+		if (!m_renderFinishedSemaphore.Create(m_owner.GetDevice()))
+			throw std::runtime_error("failed to create image finished semaphore: " + TranslateVulkanError(m_imageAvailableSemaphore.GetLastErrorCode()));
+
+		if (!m_inFlightFence.Create(m_owner.GetDevice(), VK_FENCE_CREATE_SIGNALED_BIT))
+			throw std::runtime_error("failed to create in-flight fence: " + TranslateVulkanError(m_inFlightFence.GetLastErrorCode()));
+	}
+
+	VulkanRenderImage::~VulkanRenderImage()
+	{
+		m_inFlightCommandBuffers.clear();
+	}
+
+	void VulkanRenderImage::Execute(const std::function<void(CommandBufferBuilder& builder)>& callback, bool isGraphical)
+	{
+		Vk::CommandBuffer* commandBuffer;
+		if (m_currentCommandBuffer >= m_inFlightCommandBuffers.size())
+		{
+			Vk::AutoCommandBuffer& newlyAllocatedBuffer = m_inFlightCommandBuffers.emplace_back(m_commandPool.AllocateCommandBuffer(VK_COMMAND_BUFFER_LEVEL_PRIMARY));
+			commandBuffer = &newlyAllocatedBuffer.Get();
+			m_currentCommandBuffer++;
+		}
+		else
+			commandBuffer = &m_inFlightCommandBuffers[m_currentCommandBuffer++].Get();
+
+		if (!commandBuffer->Begin(VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT))
+			throw std::runtime_error("failed to begin command buffer: " + TranslateVulkanError(commandBuffer->GetLastErrorCode()));
+
+		VulkanCommandBufferBuilder builder(*commandBuffer);
+		callback(builder);
+
+		if (!commandBuffer->End())
+			throw std::runtime_error("failed to build command buffer: " + TranslateVulkanError(commandBuffer->GetLastErrorCode()));
+
+		SubmitCommandBuffer(*commandBuffer, isGraphical);
+	}
+
+	VulkanUploadPool& VulkanRenderImage::GetUploadPool()
+	{
+		return m_uploadPool;
+	}
+
+	void VulkanRenderImage::SubmitCommandBuffer(CommandBuffer* commandBuffer, bool isGraphical)
+	{
+		VulkanCommandBuffer& vkCommandBuffer = *static_cast<VulkanCommandBuffer*>(commandBuffer);
+
+		return SubmitCommandBuffer(vkCommandBuffer.GetCommandBuffer(), isGraphical);
+	}
+
+	void VulkanRenderImage::SubmitCommandBuffer(VkCommandBuffer commandBuffer, bool isGraphical)
+	{
+		if (isGraphical)
+			m_graphicalCommandsBuffers.push_back(commandBuffer);
+		else
+		{
+			Vk::QueueHandle& graphicsQueue = m_owner.GetGraphicsQueue();
+			if (!graphicsQueue.Submit(commandBuffer))
+				throw std::runtime_error("Failed to submit command buffer: " + TranslateVulkanError(graphicsQueue.GetLastErrorCode()));
+		}
+	}
+
+	void VulkanRenderImage::Present()
+	{
+		Vk::QueueHandle& graphicsQueue = m_owner.GetGraphicsQueue();
+		if (!graphicsQueue.Submit(UInt32(m_graphicalCommandsBuffers.size()), m_graphicalCommandsBuffers.data(), m_imageAvailableSemaphore, VK_PIPELINE_STAGE_COLOR_ATTACHMENT_OUTPUT_BIT, m_renderFinishedSemaphore, m_inFlightFence))
+			throw std::runtime_error("Failed to submit command buffers: " + TranslateVulkanError(graphicsQueue.GetLastErrorCode()));
+
+		m_owner.Present(m_imageIndex, m_renderFinishedSemaphore);
+	}
+}