VK_RasterizationPipeline.cpp

浏览该文件的文档.

//
// Created by Administrator on 2025/3/24.
//
#include "./VK_Pass.h"
#include "./VK_PassGroup.h"
#include "./VK_RasterizationPipeline.h"
 
namespace FCT::RHI
{
    VK_RasterizationPipeline::VK_RasterizationPipeline(VK_Context* ctx) : m_createInfo{}
    {
        m_ctx = ctx;
        m_blendState = nullptr;
        m_rasterizationState = nullptr;
        m_viewportState = nullptr;
        m_inputLayout = nullptr;
        m_depthStencilState = nullptr;
    }
 
    VK_RasterizationPipeline::~VK_RasterizationPipeline()
    {
        FCT_SAFE_RELEASE(m_blendState);
        FCT_SAFE_RELEASE(m_rasterizationState);
        FCT_SAFE_RELEASE(m_viewportState);
        FCT_SAFE_RELEASE(m_inputLayout);
        FCT_SAFE_RELEASE(m_depthStencilState);
    }
 
    void VK_RasterizationPipeline::addResources(IPipelineResource* resource)
    {
        if (!resource)
            return;
        switch (resource->getType())
        {
        case PipelineResourceType::BlendState:
            m_blendState = static_cast<VK_BlendState*>(resource);
            m_blendState->addRef();
            break;
        case PipelineResourceType::RasterizationState:
            m_rasterizationState = static_cast<VK_RasterizationState*>(resource);
            m_rasterizationState->addRef();
            break;
        case PipelineResourceType::ViewportState:
            m_viewportState = static_cast<VK_ViewportState*>(resource);
            m_viewportState->addRef();
            break;
        case PipelineResourceType::InputLayout:
            m_inputLayout = static_cast<VK_InputLayout*>(resource);
            m_inputLayout->addRef();
            break;
        default:
            ferr << "Unsupported pipeline resource type: " << (unsigned int)(resource->getType()) << std::endl;
            break;
        }
    }
 
    void VK_RasterizationPipeline::create()
    {
        generateDefaultResources();
 
        //fill resource
        std::vector<vk::DynamicState> dynamicStates = {
            vk::DynamicState::eViewport,
            vk::DynamicState::eScissor
        };
        m_dynamicState = vk::PipelineDynamicStateCreateInfo();
        m_dynamicState.setDynamicStates(dynamicStates);
        m_createInfo.pDynamicState = &m_dynamicState;
        m_createInfo.renderPass = static_cast<VK_PassGroup*>(m_pass->passGroup())->getRenderPass();
        m_createInfo.subpass = m_pass->index();
        m_createInfo.layout = m_pipelineLayout;
        m_createInfo.pVertexInputState = m_inputLayout->pInputStateInfo();
        m_createInfo.pRasterizationState = &m_rasterizationState->rasterizationStateCreateInfo();
        m_createInfo.pMultisampleState = &m_rasterizationState->multisampleStateCreateInfo();
        m_inputAssemblyState = vk::PipelineInputAssemblyStateCreateInfo{};
        m_inputAssemblyState.topology = vk::PrimitiveTopology::eTriangleList;
        m_inputAssemblyState.primitiveRestartEnable = VK_FALSE;
        m_createInfo.pInputAssemblyState = &m_inputAssemblyState;
        m_createInfo.pViewportState = &m_viewportState->viewportStateCreateInfo();
        m_createInfo.pDepthStencilState = &m_depthStencilState->getCreateInfo();
        m_createInfo.pColorBlendState = &m_blendState->colorBlendStateCreateInfo();
 
        m_createInfo.stageCount = m_shaderStages.size();
        m_createInfo.pStages = m_shaderStages.data();
        m_pipeline = m_ctx->getDevice().createGraphicsPipeline(nullptr,m_createInfo).value;
    }
 
    void VK_RasterizationPipeline::bindPass(Pass* pass)
    {
        m_pass = static_cast<VK_Pass*>(pass);
    }
 
    void VK_RasterizationPipeline::generateDefaultResources()
    {
        if (!m_blendState)
        {
            m_blendState = new VK_BlendState(m_ctx);
            if (m_pass->renderTargets().size())
                m_blendState->targets(m_pass->renderTargets().rbegin()->first + 1);
            m_blendState->create();
        }
        if (!m_rasterizationState)
        {
            m_rasterizationState = new VK_RasterizationState(m_ctx);
            m_rasterizationState->create();
        }
        if (!m_viewportState)
        {
            m_viewportState = new VK_ViewportState(m_ctx);
            m_viewportState->create();
        }
        if (!m_inputLayout)
        {
            m_inputLayout = new VK_InputLayout(m_ctx);
            m_inputLayout->create();
        }
        if (!m_depthStencilState)
        {
            m_depthStencilState = new VK_DepthStencilState(m_ctx);
            m_depthStencilState->create();
        }
    }
}