mirror of
https://github.com/Kbz-8/42_vox.git
synced 2026-01-12 07:03:34 +00:00
92 lines
3.0 KiB
C++
92 lines
3.0 KiB
C++
#include <Renderer/Pipelines/Shader.h>
|
|
#include <Renderer/RenderCore.h>
|
|
#include <Core/Logs.h>
|
|
#include <fstream>
|
|
|
|
namespace Scop
|
|
{
|
|
Shader::Shader(const std::vector<std::uint32_t>& bytecode, ShaderType type, ShaderLayout layout) : m_bytecode(bytecode), m_layout(std::move(layout))
|
|
{
|
|
switch(type)
|
|
{
|
|
case ShaderType::Vertex : m_stage = VK_SHADER_STAGE_VERTEX_BIT; break;
|
|
case ShaderType::Fragment : m_stage = VK_SHADER_STAGE_FRAGMENT_BIT; break;
|
|
case ShaderType::Compute : m_stage = VK_SHADER_STAGE_COMPUTE_BIT; break;
|
|
default : FatalError("wtf"); break;
|
|
}
|
|
m_module = kvfCreateShaderModule(RenderCore::Get().GetDevice(), m_bytecode.data(), m_bytecode.size());
|
|
Message("Vulkan: shader module created");
|
|
|
|
GeneratePipelineLayout(m_layout);
|
|
}
|
|
|
|
void Shader::GeneratePipelineLayout(ShaderLayout layout)
|
|
{
|
|
for(auto& [n, set] : layout.set_layouts)
|
|
{
|
|
std::vector<VkDescriptorSetLayoutBinding> bindings(set.binds.size());
|
|
for(std::size_t i = 0; i < set.binds.size(); i++)
|
|
{
|
|
bindings[i].binding = set.binds[i].first;
|
|
bindings[i].descriptorCount = 1;
|
|
bindings[i].descriptorType = set.binds[i].second;
|
|
bindings[i].pImmutableSamplers = nullptr;
|
|
bindings[i].stageFlags = m_stage;
|
|
}
|
|
m_set_layouts.emplace_back(kvfCreateDescriptorSetLayout(RenderCore::Get().GetDevice(), bindings.data(), bindings.size()));
|
|
Message("Vulkan: descriptor set layout created");
|
|
m_pipeline_layout_part.set_layouts.push_back(m_set_layouts.back());
|
|
}
|
|
|
|
std::size_t i = 0;
|
|
std::vector<VkPushConstantRange> push_constants(layout.push_constants.size());
|
|
m_pipeline_layout_part.push_constants.resize(layout.push_constants.size());
|
|
for(auto& pc : layout.push_constants)
|
|
{
|
|
VkPushConstantRange push_constant_range = {};
|
|
push_constant_range.offset = pc.offset;
|
|
push_constant_range.size = pc.size;
|
|
push_constant_range.stageFlags = m_stage;
|
|
push_constants[i] = push_constant_range;
|
|
m_pipeline_layout_part.push_constants[i] = push_constant_range;
|
|
i++;
|
|
}
|
|
}
|
|
|
|
void Shader::Destroy()
|
|
{
|
|
if(m_module == VK_NULL_HANDLE)
|
|
return;
|
|
kvfDestroyShaderModule(RenderCore::Get().GetDevice(), m_module);
|
|
m_module = VK_NULL_HANDLE;
|
|
Message("Vulkan: shader module destroyed");
|
|
for(auto& layout : m_set_layouts)
|
|
{
|
|
kvfDestroyDescriptorSetLayout(RenderCore::Get().GetDevice(), layout);
|
|
Message("Vulkan: descriptor set layout destroyed");
|
|
}
|
|
}
|
|
|
|
Shader::~Shader()
|
|
{
|
|
Destroy();
|
|
}
|
|
|
|
std::shared_ptr<Shader> LoadShaderFromFile(const std::filesystem::path& filepath, ShaderType type, ShaderLayout layout)
|
|
{
|
|
std::ifstream stream(filepath, std::ios::binary);
|
|
if(!stream.is_open())
|
|
FatalError("Renderer : unable to open a spirv shader file, %", filepath);
|
|
std::vector<std::uint32_t> data;
|
|
stream.seekg(0);
|
|
std::uint32_t part = 0;
|
|
while(stream.read(reinterpret_cast<char*>(&part), sizeof(part)))
|
|
data.push_back(part);
|
|
stream.close();
|
|
|
|
std::shared_ptr<Shader> shader = std::make_shared<Shader>(data, type, layout);
|
|
Message("Vulkan: shader loaded %", filepath);
|
|
return shader;
|
|
}
|
|
}
|