323 lines
14 KiB
C++
323 lines
14 KiB
C++
// Copyright (C) 2022 Jérôme "Lynix" Leclercq (lynix680@gmail.com)
|
|
// 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/VulkanRenderPipeline.hpp>
|
|
#include <Nazara/Core/ErrorFlags.hpp>
|
|
#include <Nazara/VulkanRenderer/Utils.hpp>
|
|
#include <Nazara/VulkanRenderer/VulkanDevice.hpp>
|
|
#include <Nazara/VulkanRenderer/VulkanRenderPipelineLayout.hpp>
|
|
#include <Nazara/VulkanRenderer/VulkanShaderModule.hpp>
|
|
#include <cassert>
|
|
#include <Nazara/VulkanRenderer/Debug.hpp>
|
|
|
|
namespace Nz
|
|
{
|
|
VulkanRenderPipeline::VulkanRenderPipeline(VulkanDevice& device, RenderPipelineInfo pipelineInfo) :
|
|
m_device(&device),
|
|
m_pipelineInfo(std::move(pipelineInfo))
|
|
{
|
|
ValidatePipelineInfo(device, m_pipelineInfo);
|
|
|
|
m_pipelineCreateInfo = BuildCreateInfo(m_pipelineInfo);
|
|
}
|
|
|
|
VkPipeline VulkanRenderPipeline::Get(const VulkanRenderPass& renderPass, std::size_t subpassIndex) const
|
|
{
|
|
const Vk::RenderPass& renderPassHandle = renderPass.GetRenderPass();
|
|
|
|
// Use color attachment count as a key
|
|
const auto& subpasses = renderPass.GetSubpassDescriptions();
|
|
assert(subpassIndex < subpasses.size());
|
|
|
|
std::size_t colorAttachmentCount = subpasses[subpassIndex].colorAttachment.size();
|
|
|
|
std::pair<VkRenderPass, std::size_t> key = { renderPassHandle, colorAttachmentCount };
|
|
|
|
if (auto it = m_pipelines.find(key); it != m_pipelines.end())
|
|
return it->second.pipeline;
|
|
|
|
UpdateCreateInfo(colorAttachmentCount);
|
|
|
|
VkGraphicsPipelineCreateInfo pipelineCreateInfo = m_pipelineCreateInfo.pipelineInfo;
|
|
pipelineCreateInfo.renderPass = renderPassHandle;
|
|
|
|
PipelineData pipelineData;
|
|
pipelineData.onRenderPassRelease.Connect(renderPass.OnRenderPassRelease, [this, key](const VulkanRenderPass*)
|
|
{
|
|
m_pipelines.erase(key);
|
|
});
|
|
|
|
if (!pipelineData.pipeline.CreateGraphics(*m_device, pipelineCreateInfo))
|
|
return VK_NULL_HANDLE;
|
|
|
|
auto it = m_pipelines.emplace(key, std::move(pipelineData)).first;
|
|
return it->second.pipeline;
|
|
}
|
|
|
|
std::vector<VkPipelineColorBlendAttachmentState> VulkanRenderPipeline::BuildColorBlendAttachmentStateList(const RenderPipelineInfo& pipelineInfo)
|
|
{
|
|
std::vector<VkPipelineColorBlendAttachmentState> colorBlendStates;
|
|
|
|
VkPipelineColorBlendAttachmentState& colorBlendState = colorBlendStates.emplace_back();
|
|
colorBlendState.blendEnable = pipelineInfo.blending;
|
|
if (pipelineInfo.colorWrite)
|
|
colorBlendState.colorWriteMask = VK_COLOR_COMPONENT_R_BIT | VK_COLOR_COMPONENT_G_BIT | VK_COLOR_COMPONENT_B_BIT | VK_COLOR_COMPONENT_A_BIT; //< TODO
|
|
else
|
|
colorBlendState.colorWriteMask = 0;
|
|
|
|
if (pipelineInfo.blending)
|
|
{
|
|
colorBlendState.srcColorBlendFactor = ToVulkan(pipelineInfo.blend.srcColor);
|
|
colorBlendState.dstColorBlendFactor = ToVulkan(pipelineInfo.blend.dstColor);
|
|
colorBlendState.colorBlendOp = ToVulkan(pipelineInfo.blend.modeColor);
|
|
colorBlendState.srcAlphaBlendFactor = ToVulkan(pipelineInfo.blend.srcAlpha);
|
|
colorBlendState.dstAlphaBlendFactor = ToVulkan(pipelineInfo.blend.dstAlpha);
|
|
colorBlendState.alphaBlendOp = ToVulkan(pipelineInfo.blend.modeAlpha);
|
|
}
|
|
else
|
|
{
|
|
colorBlendState.srcColorBlendFactor = VK_BLEND_FACTOR_ONE;
|
|
colorBlendState.dstColorBlendFactor = VK_BLEND_FACTOR_ZERO;
|
|
colorBlendState.colorBlendOp = VK_BLEND_OP_ADD;
|
|
colorBlendState.srcAlphaBlendFactor = VK_BLEND_FACTOR_ONE;
|
|
colorBlendState.dstAlphaBlendFactor = VK_BLEND_FACTOR_ZERO;
|
|
colorBlendState.alphaBlendOp = VK_BLEND_OP_ADD;
|
|
}
|
|
|
|
return colorBlendStates;
|
|
}
|
|
|
|
VkPipelineColorBlendStateCreateInfo VulkanRenderPipeline::BuildColorBlendInfo(const RenderPipelineInfo& /*pipelineInfo*/, const std::vector<VkPipelineColorBlendAttachmentState>& attachmentState)
|
|
{
|
|
VkPipelineColorBlendStateCreateInfo createInfo = {};
|
|
createInfo.sType = VK_STRUCTURE_TYPE_PIPELINE_COLOR_BLEND_STATE_CREATE_INFO;
|
|
createInfo.attachmentCount = std::uint32_t(attachmentState.size());
|
|
createInfo.pAttachments = attachmentState.data();
|
|
|
|
return createInfo;
|
|
}
|
|
|
|
VkPipelineDepthStencilStateCreateInfo VulkanRenderPipeline::BuildDepthStencilInfo(const RenderPipelineInfo& pipelineInfo)
|
|
{
|
|
VkPipelineDepthStencilStateCreateInfo createInfo = {};
|
|
createInfo.sType = VK_STRUCTURE_TYPE_PIPELINE_DEPTH_STENCIL_STATE_CREATE_INFO;
|
|
createInfo.depthTestEnable = pipelineInfo.depthBuffer;
|
|
createInfo.depthWriteEnable = pipelineInfo.depthWrite;
|
|
createInfo.depthCompareOp = ToVulkan(pipelineInfo.depthCompare);
|
|
createInfo.stencilTestEnable = pipelineInfo.stencilTest;
|
|
createInfo.front = BuildStencilOp(pipelineInfo, true);
|
|
createInfo.back = BuildStencilOp(pipelineInfo, false);
|
|
|
|
return createInfo;
|
|
}
|
|
|
|
VkPipelineDynamicStateCreateInfo VulkanRenderPipeline::BuildDynamicStateInfo(const RenderPipelineInfo& /*pipelineInfo*/, const std::vector<VkDynamicState>& dynamicStates)
|
|
{
|
|
VkPipelineDynamicStateCreateInfo createInfo = {};
|
|
createInfo.sType = VK_STRUCTURE_TYPE_PIPELINE_DYNAMIC_STATE_CREATE_INFO;
|
|
createInfo.dynamicStateCount = std::uint32_t(dynamicStates.size());
|
|
createInfo.pDynamicStates = dynamicStates.data();
|
|
|
|
return createInfo;
|
|
}
|
|
|
|
std::vector<VkDynamicState> VulkanRenderPipeline::BuildDynamicStateList(const RenderPipelineInfo& pipelineInfo)
|
|
{
|
|
std::vector<VkDynamicState> dynamicStates;
|
|
dynamicStates.push_back(VK_DYNAMIC_STATE_VIEWPORT);
|
|
dynamicStates.push_back(VK_DYNAMIC_STATE_SCISSOR);
|
|
|
|
return dynamicStates;
|
|
}
|
|
|
|
VkPipelineInputAssemblyStateCreateInfo VulkanRenderPipeline::BuildInputAssemblyInfo(const RenderPipelineInfo& pipelineInfo)
|
|
{
|
|
VkPipelineInputAssemblyStateCreateInfo createInfo = {};
|
|
createInfo.sType = VK_STRUCTURE_TYPE_PIPELINE_INPUT_ASSEMBLY_STATE_CREATE_INFO;
|
|
createInfo.topology = ToVulkan(pipelineInfo.primitiveMode);
|
|
|
|
return createInfo;
|
|
}
|
|
|
|
VkPipelineMultisampleStateCreateInfo VulkanRenderPipeline::BuildMultisampleInfo(const RenderPipelineInfo& /*pipelineInfo*/)
|
|
{
|
|
VkPipelineMultisampleStateCreateInfo createInfo = {};
|
|
createInfo.sType = VK_STRUCTURE_TYPE_PIPELINE_MULTISAMPLE_STATE_CREATE_INFO;
|
|
createInfo.rasterizationSamples = VK_SAMPLE_COUNT_1_BIT;
|
|
createInfo.minSampleShading = 1.0f; //< TODO: Remove
|
|
|
|
return createInfo;
|
|
}
|
|
|
|
VkPipelineRasterizationStateCreateInfo VulkanRenderPipeline::BuildRasterizationInfo(const RenderPipelineInfo& pipelineInfo)
|
|
{
|
|
VkPipelineRasterizationStateCreateInfo createInfo = {};
|
|
createInfo.sType = VK_STRUCTURE_TYPE_PIPELINE_RASTERIZATION_STATE_CREATE_INFO;
|
|
createInfo.cullMode = (pipelineInfo.faceCulling) ? ToVulkan(pipelineInfo.cullingSide) : VK_CULL_MODE_NONE;
|
|
createInfo.depthClampEnable = pipelineInfo.depthClamp;
|
|
createInfo.frontFace = ToVulkan(pipelineInfo.frontFace);
|
|
createInfo.lineWidth = pipelineInfo.lineWidth;
|
|
createInfo.polygonMode = ToVulkan(pipelineInfo.faceFilling);
|
|
|
|
return createInfo;
|
|
}
|
|
|
|
VkPipelineViewportStateCreateInfo VulkanRenderPipeline::BuildViewportInfo(const RenderPipelineInfo& pipelineInfo)
|
|
{
|
|
VkPipelineViewportStateCreateInfo createInfo = {};
|
|
createInfo.sType = VK_STRUCTURE_TYPE_PIPELINE_VIEWPORT_STATE_CREATE_INFO;
|
|
createInfo.viewportCount = 1; //< TODO: Handle multiple viewport regions
|
|
createInfo.scissorCount = 1; //< TODO: Handle multiple scissor regions
|
|
|
|
return createInfo;
|
|
}
|
|
|
|
VkStencilOpState VulkanRenderPipeline::BuildStencilOp(const RenderPipelineInfo& pipelineInfo, bool front)
|
|
{
|
|
const auto& pipelineStencil = (front) ? pipelineInfo.stencilFront : pipelineInfo.stencilBack;
|
|
|
|
VkStencilOpState stencilStates;
|
|
stencilStates.compareMask = pipelineStencil.compareMask;
|
|
stencilStates.compareOp = ToVulkan(pipelineStencil.compare);
|
|
stencilStates.depthFailOp = ToVulkan(pipelineStencil.depthFail);
|
|
stencilStates.failOp = ToVulkan(pipelineStencil.fail);
|
|
stencilStates.passOp = ToVulkan(pipelineStencil.pass);
|
|
stencilStates.reference = pipelineStencil.reference;
|
|
stencilStates.writeMask = pipelineStencil.writeMask;
|
|
|
|
return stencilStates;
|
|
}
|
|
|
|
std::vector<VkPipelineShaderStageCreateInfo> VulkanRenderPipeline::BuildShaderStageInfo(const RenderPipelineInfo& pipelineInfo)
|
|
{
|
|
std::vector<VkPipelineShaderStageCreateInfo> shaderStageCreateInfos;
|
|
|
|
for (auto&& stagePtr : pipelineInfo.shaderModules)
|
|
{
|
|
assert(stagePtr);
|
|
|
|
VulkanShaderModule& vulkanModule = static_cast<VulkanShaderModule&>(*stagePtr);
|
|
for (auto& stage : vulkanModule.GetStages())
|
|
{
|
|
VkPipelineShaderStageCreateInfo& createInfo = shaderStageCreateInfos.emplace_back();
|
|
createInfo.sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO;
|
|
createInfo.module = vulkanModule.GetHandle();
|
|
createInfo.pName = stage.name.data();
|
|
createInfo.stage = ToVulkan(stage.stage);
|
|
}
|
|
}
|
|
|
|
return shaderStageCreateInfos;
|
|
}
|
|
|
|
std::vector<VkVertexInputAttributeDescription> VulkanRenderPipeline::BuildVertexAttributeDescription(const RenderPipelineInfo& pipelineInfo)
|
|
{
|
|
std::vector<VkVertexInputAttributeDescription> vertexAttributes;
|
|
|
|
std::uint32_t locationIndex = 0;
|
|
|
|
for (const auto& bufferData : pipelineInfo.vertexBuffers)
|
|
{
|
|
std::uint32_t binding = std::uint32_t(bufferData.binding);
|
|
|
|
for (const auto& componentInfo : bufferData.declaration->GetComponents())
|
|
{
|
|
if (componentInfo.component == VertexComponent::Unused)
|
|
continue;
|
|
|
|
auto& bufferAttribute = vertexAttributes.emplace_back();
|
|
bufferAttribute.binding = binding;
|
|
bufferAttribute.location = locationIndex++;
|
|
bufferAttribute.offset = std::uint32_t(componentInfo.offset);
|
|
bufferAttribute.format = ToVulkan(componentInfo.type);
|
|
}
|
|
}
|
|
|
|
return vertexAttributes;
|
|
}
|
|
|
|
std::vector<VkVertexInputBindingDescription> VulkanRenderPipeline::BuildVertexBindingDescription(const RenderPipelineInfo& pipelineInfo)
|
|
{
|
|
std::vector<VkVertexInputBindingDescription> vertexBindings;
|
|
|
|
for (const auto& bufferData : pipelineInfo.vertexBuffers)
|
|
{
|
|
auto& bufferBinding = vertexBindings.emplace_back();
|
|
bufferBinding.binding = std::uint32_t(bufferData.binding);
|
|
bufferBinding.stride = std::uint32_t(bufferData.declaration->GetStride());
|
|
bufferBinding.inputRate = ToVulkan(bufferData.declaration->GetInputRate());
|
|
}
|
|
|
|
return vertexBindings;
|
|
}
|
|
|
|
VkPipelineVertexInputStateCreateInfo VulkanRenderPipeline::BuildVertexInputInfo(const RenderPipelineInfo& /*pipelineInfo*/, const std::vector<VkVertexInputAttributeDescription>& vertexAttributes, const std::vector<VkVertexInputBindingDescription>& bindingDescriptions)
|
|
{
|
|
VkPipelineVertexInputStateCreateInfo createInfo = {};
|
|
createInfo.sType = VK_STRUCTURE_TYPE_PIPELINE_VERTEX_INPUT_STATE_CREATE_INFO;
|
|
|
|
createInfo.vertexAttributeDescriptionCount = std::uint32_t(vertexAttributes.size());
|
|
createInfo.pVertexAttributeDescriptions = vertexAttributes.data();
|
|
|
|
createInfo.vertexBindingDescriptionCount = std::uint32_t(bindingDescriptions.size());
|
|
createInfo.pVertexBindingDescriptions = bindingDescriptions.data();
|
|
|
|
return createInfo;
|
|
}
|
|
|
|
auto VulkanRenderPipeline::BuildCreateInfo(const RenderPipelineInfo& pipelineInfo) -> CreateInfo
|
|
{
|
|
CreateInfo createInfo = {};
|
|
createInfo.stateData = std::make_unique<CreateInfo::StateData>();
|
|
|
|
createInfo.colorBlendAttachmentState = BuildColorBlendAttachmentStateList(pipelineInfo);
|
|
createInfo.dynamicStates = BuildDynamicStateList(pipelineInfo);
|
|
createInfo.shaderStages = BuildShaderStageInfo(pipelineInfo);
|
|
createInfo.vertexAttributesDescription = BuildVertexAttributeDescription(pipelineInfo);
|
|
createInfo.vertexBindingDescription = BuildVertexBindingDescription(pipelineInfo);
|
|
|
|
createInfo.stateData->colorBlendState = BuildColorBlendInfo(pipelineInfo, createInfo.colorBlendAttachmentState);
|
|
createInfo.stateData->depthStencilState = BuildDepthStencilInfo(pipelineInfo);
|
|
createInfo.stateData->dynamicState = BuildDynamicStateInfo(pipelineInfo, createInfo.dynamicStates);
|
|
createInfo.stateData->inputAssemblyState = BuildInputAssemblyInfo(pipelineInfo);
|
|
createInfo.stateData->multiSampleState = BuildMultisampleInfo(pipelineInfo);
|
|
createInfo.stateData->rasterizationState = BuildRasterizationInfo(pipelineInfo);
|
|
createInfo.stateData->viewportState = BuildViewportInfo(pipelineInfo);
|
|
createInfo.stateData->vertexInputState = BuildVertexInputInfo(pipelineInfo, createInfo.vertexAttributesDescription, createInfo.vertexBindingDescription);
|
|
|
|
createInfo.pipelineInfo.sType = VK_STRUCTURE_TYPE_GRAPHICS_PIPELINE_CREATE_INFO;
|
|
createInfo.pipelineInfo.stageCount = std::uint32_t(createInfo.shaderStages.size());
|
|
createInfo.pipelineInfo.pStages = createInfo.shaderStages.data();
|
|
createInfo.pipelineInfo.pColorBlendState = &createInfo.stateData->colorBlendState;
|
|
createInfo.pipelineInfo.pDepthStencilState = &createInfo.stateData->depthStencilState;
|
|
createInfo.pipelineInfo.pDynamicState = &createInfo.stateData->dynamicState;
|
|
createInfo.pipelineInfo.pInputAssemblyState = &createInfo.stateData->inputAssemblyState;
|
|
createInfo.pipelineInfo.pMultisampleState = &createInfo.stateData->multiSampleState;
|
|
createInfo.pipelineInfo.pRasterizationState = &createInfo.stateData->rasterizationState;
|
|
createInfo.pipelineInfo.pVertexInputState = &createInfo.stateData->vertexInputState;
|
|
createInfo.pipelineInfo.pViewportState = &createInfo.stateData->viewportState;
|
|
|
|
VulkanRenderPipelineLayout& pipelineLayout = *static_cast<VulkanRenderPipelineLayout*>(pipelineInfo.pipelineLayout.get());
|
|
createInfo.pipelineInfo.layout = pipelineLayout.GetPipelineLayout();
|
|
|
|
return createInfo;
|
|
}
|
|
|
|
void VulkanRenderPipeline::UpdateCreateInfo(std::size_t colorBufferCount) const
|
|
{
|
|
// TODO: Add support for independent blend
|
|
std::size_t previousSize = m_pipelineCreateInfo.colorBlendAttachmentState.size();
|
|
if (previousSize < colorBufferCount)
|
|
{
|
|
assert(!m_pipelineCreateInfo.colorBlendAttachmentState.empty());
|
|
|
|
m_pipelineCreateInfo.colorBlendAttachmentState.resize(colorBufferCount);
|
|
for (std::size_t i = previousSize; i < colorBufferCount; ++i)
|
|
m_pipelineCreateInfo.colorBlendAttachmentState[i] = m_pipelineCreateInfo.colorBlendAttachmentState.front();
|
|
}
|
|
|
|
m_pipelineCreateInfo.stateData->colorBlendState = BuildColorBlendInfo(m_pipelineInfo, m_pipelineCreateInfo.colorBlendAttachmentState);
|
|
}
|
|
}
|