245 lines
8.5 KiB
C++
245 lines
8.5 KiB
C++
#include<hgl/graph/vulkan/VKPipeline.h>
|
||
#include<hgl/graph/vulkan/VKDevice.h>
|
||
#include<hgl/graph/vulkan/VKMaterial.h>
|
||
#include<hgl/graph/vulkan/VKRenderPass.h>
|
||
#include<hgl/graph/vulkan/VKRenderTarget.h>
|
||
#include<hgl/graph/vulkan/VKFramebuffer.h>
|
||
|
||
VK_NAMESPACE_BEGIN
|
||
Pipeline::~Pipeline()
|
||
{
|
||
vkDestroyPipeline(device,pipeline,nullptr);
|
||
}
|
||
|
||
void PipelineCreater::InitVertexInputState(const Material *material)
|
||
{
|
||
pipelineInfo.stageCount = material->GetStageCount();
|
||
pipelineInfo.pStages = material->GetStages();
|
||
|
||
{
|
||
vis_create_info.sType = VK_STRUCTURE_TYPE_PIPELINE_VERTEX_INPUT_STATE_CREATE_INFO;
|
||
vis_create_info.pNext = nullptr;
|
||
vis_create_info.flags = 0;
|
||
|
||
material->Write(vis_create_info);
|
||
|
||
pipelineInfo.pVertexInputState=&vis_create_info;
|
||
}
|
||
}
|
||
|
||
void PipelineCreater::InitViewportState()
|
||
{
|
||
viewport.x = 0.0f;
|
||
viewport.y = 0.0f;
|
||
viewport.width = extent.width;
|
||
viewport.height = extent.height;
|
||
viewport.minDepth = 0.0f;
|
||
viewport.maxDepth = 1.0f;
|
||
|
||
scissor.offset = {0, 0};
|
||
scissor.extent = extent;
|
||
|
||
viewportState.sType = VK_STRUCTURE_TYPE_PIPELINE_VIEWPORT_STATE_CREATE_INFO;
|
||
viewportState.pNext = nullptr;
|
||
viewportState.flags = 0;
|
||
viewportState.viewportCount = 1;
|
||
viewportState.pViewports = &viewport;
|
||
viewportState.scissorCount = 1;
|
||
viewportState.pScissors = &scissor;
|
||
|
||
pipelineInfo.pViewportState = &viewportState;
|
||
}
|
||
|
||
void PipelineCreater::InitDynamicState()
|
||
{
|
||
memset(dynamicStateEnables, 0, sizeof dynamicStateEnables);
|
||
|
||
dynamicState.sType = VK_STRUCTURE_TYPE_PIPELINE_DYNAMIC_STATE_CREATE_INFO;
|
||
dynamicState.pNext = nullptr;
|
||
dynamicState.flags = 0;
|
||
dynamicState.pDynamicStates = dynamicStateEnables;
|
||
dynamicState.dynamicStateCount = 0;
|
||
dynamicStateEnables[dynamicState.dynamicStateCount++] = VK_DYNAMIC_STATE_VIEWPORT;
|
||
dynamicStateEnables[dynamicState.dynamicStateCount++] = VK_DYNAMIC_STATE_SCISSOR;
|
||
dynamicStateEnables[dynamicState.dynamicStateCount++] = VK_DYNAMIC_STATE_LINE_WIDTH;
|
||
|
||
//如果窗口大小不变，可以不设置这两个。能不能提升效能未知
|
||
|
||
pipelineInfo.pDynamicState=&dynamicState;
|
||
}
|
||
|
||
//为什么一定要把ext放在这里，因为如果不放在这里，总是会让人遗忘它的重要性
|
||
|
||
PipelineCreater::PipelineCreater(Device *dev,const Material *material,const RenderTarget *rt)
|
||
{
|
||
device=dev->GetDevice();
|
||
extent=rt->GetExtent();
|
||
cache=dev->GetPipelineCache();
|
||
|
||
//未来这里需要增加是否有vs/fs的检测
|
||
|
||
hgl_zero(pipelineInfo);
|
||
pipelineInfo.sType = VK_STRUCTURE_TYPE_GRAPHICS_PIPELINE_CREATE_INFO;
|
||
InitVertexInputState(material);
|
||
|
||
tessellation.sType=VK_STRUCTURE_TYPE_PIPELINE_TESSELLATION_STATE_CREATE_INFO;
|
||
tessellation.pNext=nullptr;
|
||
tessellation.flags=0;
|
||
tessellation.patchControlPoints=0;
|
||
|
||
pipelineInfo.pTessellationState=&tessellation;
|
||
|
||
InitViewportState();
|
||
|
||
rasterizer.sType = VK_STRUCTURE_TYPE_PIPELINE_RASTERIZATION_STATE_CREATE_INFO;
|
||
rasterizer.pNext = nullptr;
|
||
rasterizer.flags = 0;
|
||
rasterizer.depthClampEnable = VK_FALSE;
|
||
rasterizer.rasterizerDiscardEnable = VK_FALSE;
|
||
rasterizer.polygonMode = VK_POLYGON_MODE_FILL;
|
||
rasterizer.cullMode = VK_CULL_MODE_BACK_BIT;
|
||
rasterizer.frontFace = VK_FRONT_FACE_COUNTER_CLOCKWISE; //逆时针，和opengl一样
|
||
rasterizer.depthBiasEnable = VK_FALSE;
|
||
rasterizer.depthBiasConstantFactor = 0;
|
||
rasterizer.depthBiasClamp = 0;
|
||
rasterizer.depthBiasSlopeFactor = 0;
|
||
rasterizer.lineWidth = 1.0f;
|
||
|
||
pipelineInfo.pRasterizationState = &rasterizer;
|
||
|
||
multisampling.sType = VK_STRUCTURE_TYPE_PIPELINE_MULTISAMPLE_STATE_CREATE_INFO;
|
||
multisampling.pNext = nullptr;
|
||
multisampling.flags = 0;
|
||
multisampling.rasterizationSamples = VK_SAMPLE_COUNT_1_BIT;
|
||
multisampling.sampleShadingEnable = VK_FALSE;
|
||
multisampling.minSampleShading = 0.0;
|
||
multisampling.pSampleMask = nullptr;
|
||
multisampling.alphaToCoverageEnable = VK_FALSE;
|
||
multisampling.alphaToOneEnable = VK_FALSE;
|
||
|
||
pipelineInfo.pMultisampleState = &multisampling;
|
||
|
||
depthStencilState.sType = VK_STRUCTURE_TYPE_PIPELINE_DEPTH_STENCIL_STATE_CREATE_INFO;
|
||
depthStencilState.pNext = nullptr;
|
||
depthStencilState.flags = 0;
|
||
depthStencilState.depthTestEnable = VK_TRUE;
|
||
depthStencilState.depthWriteEnable = VK_TRUE;
|
||
depthStencilState.depthCompareOp = VK_COMPARE_OP_LESS_OR_EQUAL;
|
||
depthStencilState.depthBoundsTestEnable = VK_FALSE;
|
||
depthStencilState.minDepthBounds = 0;
|
||
depthStencilState.maxDepthBounds = 0;
|
||
depthStencilState.stencilTestEnable = VK_FALSE;
|
||
depthStencilState.back.failOp = VK_STENCIL_OP_KEEP;
|
||
depthStencilState.back.passOp = VK_STENCIL_OP_KEEP;
|
||
depthStencilState.back.compareOp = VK_COMPARE_OP_ALWAYS;
|
||
depthStencilState.back.compareMask = 0;
|
||
depthStencilState.back.reference = 0;
|
||
depthStencilState.back.depthFailOp = VK_STENCIL_OP_KEEP;
|
||
depthStencilState.back.writeMask = 0;
|
||
depthStencilState.front = depthStencilState.back;
|
||
depthStencilState.front.compareOp=VK_COMPARE_OP_NEVER;
|
||
|
||
pipelineInfo.pDepthStencilState=&depthStencilState;
|
||
|
||
VkPipelineColorBlendAttachmentState cba;
|
||
cba.colorWriteMask = VK_COLOR_COMPONENT_R_BIT | VK_COLOR_COMPONENT_G_BIT | VK_COLOR_COMPONENT_B_BIT | VK_COLOR_COMPONENT_A_BIT;
|
||
cba.blendEnable = VK_FALSE;
|
||
cba.alphaBlendOp = VK_BLEND_OP_ADD;
|
||
cba.colorBlendOp = VK_BLEND_OP_ADD;
|
||
cba.srcColorBlendFactor = VK_BLEND_FACTOR_ZERO;
|
||
cba.dstColorBlendFactor = VK_BLEND_FACTOR_ZERO;
|
||
cba.srcAlphaBlendFactor = VK_BLEND_FACTOR_ZERO;
|
||
cba.dstAlphaBlendFactor = VK_BLEND_FACTOR_ZERO;
|
||
|
||
colorBlendAttachments.Add(cba,rt->GetColorCount()); //这个需要和subpass中的color attachment数量相等，所以添加多份
|
||
|
||
alpha_blend=false;
|
||
|
||
colorBlending.sType = VK_STRUCTURE_TYPE_PIPELINE_COLOR_BLEND_STATE_CREATE_INFO;
|
||
colorBlending.pNext = nullptr;
|
||
colorBlending.flags = 0;
|
||
colorBlending.logicOpEnable = VK_FALSE;
|
||
colorBlending.logicOp = VK_LOGIC_OP_CLEAR;
|
||
colorBlending.attachmentCount = colorBlendAttachments.GetCount();
|
||
colorBlending.pAttachments = colorBlendAttachments.GetData();
|
||
colorBlending.blendConstants[0] = 0.0f;
|
||
colorBlending.blendConstants[1] = 0.0f;
|
||
colorBlending.blendConstants[2] = 0.0f;
|
||
colorBlending.blendConstants[3] = 0.0f;
|
||
|
||
pipelineInfo.pColorBlendState = &colorBlending;
|
||
|
||
InitDynamicState();
|
||
|
||
pipelineInfo.layout = material->GetPipelineLayout();
|
||
{
|
||
pipelineInfo.renderPass = rt->GetRenderPass();
|
||
pipelineInfo.subpass = 0; //subpass由于还不知道有什么用，所以暂时写0，待知道功用后，需改进
|
||
}
|
||
|
||
{
|
||
pipelineInfo.basePipelineHandle = VK_NULL_HANDLE;
|
||
pipelineInfo.basePipelineIndex = -1;
|
||
}
|
||
}
|
||
|
||
PipelineCreater::PipelineCreater(Device *dev,const Material *material,const RenderTarget *rt,uchar *data,uint size)
|
||
{
|
||
LoadFromMemory(data,size);
|
||
|
||
device=dev->GetDevice();
|
||
extent=rt->GetExtent();
|
||
cache=dev->GetPipelineCache();
|
||
|
||
InitVertexInputState(material);
|
||
|
||
pipelineInfo.pInputAssemblyState=&inputAssembly;
|
||
pipelineInfo.pTessellationState =&tessellation;
|
||
|
||
InitViewportState();
|
||
|
||
pipelineInfo.pRasterizationState=&rasterizer;
|
||
pipelineInfo.pMultisampleState =&multisampling;
|
||
pipelineInfo.pDepthStencilState =&depthStencilState;
|
||
pipelineInfo.pColorBlendState =&colorBlending;
|
||
|
||
InitDynamicState();
|
||
|
||
pipelineInfo.layout = material->GetPipelineLayout();
|
||
{
|
||
pipelineInfo.renderPass = rt->GetRenderPass();
|
||
pipelineInfo.subpass = 0; //subpass由于还不知道有什么用，所以暂时写0，待知道功用后，需改进
|
||
}
|
||
|
||
{
|
||
pipelineInfo.basePipelineHandle = VK_NULL_HANDLE;
|
||
pipelineInfo.basePipelineIndex = -1;
|
||
}
|
||
}
|
||
|
||
bool PipelineCreater::Set(const uint topology,bool restart)
|
||
{
|
||
if(topology<PRIM_BEGIN||topology>PRIM_END)
|
||
if(topology!=PRIM_RECTANGLES)return(false);
|
||
|
||
inputAssembly.sType = VK_STRUCTURE_TYPE_PIPELINE_INPUT_ASSEMBLY_STATE_CREATE_INFO;
|
||
inputAssembly.pNext = nullptr;
|
||
inputAssembly.flags = 0;
|
||
inputAssembly.topology = VkPrimitiveTopology(topology==PRIM_RECTANGLES?VK_PRIMITIVE_TOPOLOGY_POINT_LIST:topology);
|
||
inputAssembly.primitiveRestartEnable = restart;
|
||
|
||
pipelineInfo.pInputAssemblyState = &inputAssembly;
|
||
return(true);
|
||
}
|
||
|
||
Pipeline *PipelineCreater::Create()
|
||
{
|
||
VkPipeline graphicsPipeline;
|
||
|
||
if (vkCreateGraphicsPipelines(device, cache, 1, &pipelineInfo, nullptr, &graphicsPipeline) != VK_SUCCESS)
|
||
return(nullptr);
|
||
|
||
return(new Pipeline(device,graphicsPipeline,alpha_test>0,alpha_blend));
|
||
}
|
||
VK_NAMESPACE_END
|