mirror of https://github.com/doitsujin/dxvk
103 lines
3.8 KiB
C++
103 lines
3.8 KiB
C++
#include "dxvk_compute.h"
|
|
|
|
namespace dxvk {
|
|
|
|
DxvkComputePipeline::DxvkComputePipeline(
|
|
const Rc<vk::DeviceFn>& vkd,
|
|
const Rc<DxvkShader>& shader)
|
|
: m_vkd(vkd) {
|
|
TRACE(this, shader);
|
|
|
|
std::vector<VkDescriptorSetLayoutBinding> bindings;
|
|
|
|
for (uint32_t i = 0; i < shader->slotCount(); i++)
|
|
bindings.push_back(shader->slotBinding(0, i));
|
|
|
|
VkDescriptorSetLayoutCreateInfo dlayout;
|
|
dlayout.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO;
|
|
dlayout.pNext = nullptr;
|
|
dlayout.flags = 0;
|
|
dlayout.bindingCount = bindings.size();
|
|
dlayout.pBindings = bindings.data();
|
|
|
|
if (m_vkd->vkCreateDescriptorSetLayout(m_vkd->device(),
|
|
&dlayout, nullptr, &m_descriptorSetLayout) != VK_SUCCESS)
|
|
throw DxvkError("DxvkComputePipeline::DxvkComputePipeline: Failed to create descriptor set layout");
|
|
|
|
VkPipelineLayoutCreateInfo playout;
|
|
playout.sType = VK_STRUCTURE_TYPE_PIPELINE_LAYOUT_CREATE_INFO;
|
|
playout.pNext = nullptr;
|
|
playout.flags = 0;
|
|
playout.setLayoutCount = 1;
|
|
playout.pSetLayouts = &m_descriptorSetLayout;
|
|
playout.pushConstantRangeCount = 0;
|
|
playout.pPushConstantRanges = nullptr;
|
|
|
|
if (m_vkd->vkCreatePipelineLayout(m_vkd->device(),
|
|
&playout, nullptr, &m_pipelineLayout) != VK_SUCCESS) {
|
|
this->destroyObjects();
|
|
throw DxvkError("DxvkComputePipeline::DxvkComputePipeline: Failed to create pipeline layout");
|
|
}
|
|
|
|
SpirvCodeBuffer code = shader->code(0);
|
|
|
|
VkShaderModuleCreateInfo minfo;
|
|
minfo.sType = VK_STRUCTURE_TYPE_SHADER_MODULE_CREATE_INFO;
|
|
minfo.pNext = nullptr;
|
|
minfo.flags = 0;
|
|
minfo.codeSize = code.size();
|
|
minfo.pCode = code.data();
|
|
|
|
if (m_vkd->vkCreateShaderModule(m_vkd->device(),
|
|
&minfo, nullptr, &m_module) != VK_SUCCESS) {
|
|
this->destroyObjects();
|
|
throw DxvkError("DxvkComputePipeline::DxvkComputePipeline: Failed to create shader module");
|
|
}
|
|
|
|
VkPipelineShaderStageCreateInfo sinfo;
|
|
sinfo.sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO;
|
|
sinfo.pNext = nullptr;
|
|
sinfo.flags = 0;
|
|
sinfo.stage = VK_SHADER_STAGE_COMPUTE_BIT;
|
|
sinfo.module = m_module;
|
|
sinfo.pName = "main";
|
|
sinfo.pSpecializationInfo = nullptr;
|
|
|
|
VkComputePipelineCreateInfo info;
|
|
info.sType = VK_STRUCTURE_TYPE_COMPUTE_PIPELINE_CREATE_INFO;
|
|
info.pNext = nullptr;
|
|
info.flags = 0;
|
|
info.stage = sinfo;
|
|
info.layout = m_pipelineLayout;
|
|
info.basePipelineHandle = VK_NULL_HANDLE;
|
|
info.basePipelineIndex = 0;
|
|
|
|
if (m_vkd->vkCreateComputePipelines(m_vkd->device(),
|
|
VK_NULL_HANDLE, 1, &info, nullptr, &m_pipeline) != VK_SUCCESS) {
|
|
this->destroyObjects();
|
|
throw DxvkError("DxvkComputePipeline::DxvkComputePipeline: Failed to compipe pipeline");
|
|
}
|
|
}
|
|
|
|
|
|
DxvkComputePipeline::~DxvkComputePipeline() {
|
|
TRACE(this);
|
|
this->destroyObjects();
|
|
}
|
|
|
|
|
|
void DxvkComputePipeline::destroyObjects() {
|
|
if (m_pipeline != VK_NULL_HANDLE)
|
|
m_vkd->vkDestroyPipeline(m_vkd->device(), m_pipeline, nullptr);
|
|
|
|
if (m_module != VK_NULL_HANDLE)
|
|
m_vkd->vkDestroyShaderModule(m_vkd->device(), m_module, nullptr);
|
|
|
|
if (m_pipelineLayout != VK_NULL_HANDLE)
|
|
m_vkd->vkDestroyPipelineLayout(m_vkd->device(), m_pipelineLayout, nullptr);
|
|
|
|
if (m_descriptorSetLayout != VK_NULL_HANDLE)
|
|
m_vkd->vkDestroyDescriptorSetLayout(m_vkd->device(), m_descriptorSetLayout, nullptr);
|
|
}
|
|
|
|
} |