mirror of
https://github.com/seekrs/MacroLibX.git
synced 2026-01-11 14:43:34 +00:00
112 lines
3.9 KiB
C++
112 lines
3.9 KiB
C++
/* ************************************************************************** */
|
|
/* */
|
|
/* ::: :::::::: */
|
|
/* DescriptorSet.cpp :+: :+: :+: */
|
|
/* +:+ +:+ +:+ */
|
|
/* By: maldavid <kbz_8.dev@akel-engine.com> +#+ +:+ +#+ */
|
|
/* +#+#+#+#+#+ +#+ */
|
|
/* Created: 2023/01/23 18:40:44 by maldavid #+# #+# */
|
|
/* Updated: 2024/04/23 19:50:06 by maldavid ### ########.fr */
|
|
/* */
|
|
/* ************************************************************************** */
|
|
|
|
#include <PreCompiled.h>
|
|
|
|
#include <Renderer/Descriptors/DescriptorSet.h>
|
|
#include <Renderer/Core/RenderCore.h>
|
|
#include <Renderer/Descriptors/DescriptorPool.h>
|
|
#include <Renderer/Buffers/UniformBuffer.h>
|
|
#include <Renderer/Renderer.h>
|
|
#include <Renderer/Images/Image.h>
|
|
|
|
namespace mlx
|
|
{
|
|
void DescriptorSet::Init(NonOwningPtr<Renderer> renderer, NonOwningPtr<DescriptorPool> pool, DescriptorSetLayout layout)
|
|
{
|
|
MLX_PROFILE_FUNCTION();
|
|
m_renderer = renderer;
|
|
m_layout = layout;
|
|
m_pool = pool;
|
|
|
|
for(int i = 0; i < MAX_FRAMES_IN_FLIGHT; i++)
|
|
m_desc_set[i] = pool->AllocateDescriptorSet(layout);
|
|
}
|
|
|
|
void DescriptorSet::WriteDescriptor(int binding, NonOwningPtr<UniformBuffer> ubo) const noexcept
|
|
{
|
|
MLX_PROFILE_FUNCTION();
|
|
auto device = RenderCore::Get().GetDevice().Get();
|
|
|
|
for(int i = 0; i < MAX_FRAMES_IN_FLIGHT; i++)
|
|
{
|
|
VkDescriptorBufferInfo buffer_info{};
|
|
buffer_info.buffer = ubo->Get(i);
|
|
buffer_info.offset = ubo->GetOffset(i);
|
|
buffer_info.range = ubo->GetSize(i);
|
|
|
|
VkWriteDescriptorSet descriptor_write{};
|
|
descriptor_write.sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET;
|
|
descriptor_write.dstSet = m_desc_set[i];
|
|
descriptor_write.dstBinding = binding;
|
|
descriptor_write.dstArrayElement = 0;
|
|
descriptor_write.descriptorType = VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER;
|
|
descriptor_write.descriptorCount = 1;
|
|
descriptor_write.pBufferInfo = &buffer_info;
|
|
|
|
vkUpdateDescriptorSets(device, 1, &descriptor_write, 0, nullptr);
|
|
}
|
|
}
|
|
|
|
void DescriptorSet::WriteDescriptor(int binding, const Image& image) const noexcept
|
|
{
|
|
MLX_PROFILE_FUNCTION();
|
|
auto device = RenderCore::Get().GetDevice().Get();
|
|
|
|
VkDescriptorImageInfo image_info{};
|
|
image_info.imageLayout = image.GetLayout();
|
|
image_info.imageView = image.GetImageView();
|
|
image_info.sampler = image.GetSampler();
|
|
|
|
VkWriteDescriptorSet descriptor_write{};
|
|
descriptor_write.sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET;
|
|
descriptor_write.dstSet = m_desc_set[m_renderer->GetActiveImageIndex()];
|
|
descriptor_write.dstBinding = binding;
|
|
descriptor_write.dstArrayElement = 0;
|
|
descriptor_write.descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER;
|
|
descriptor_write.descriptorCount = 1;
|
|
descriptor_write.pImageInfo = &image_info;
|
|
|
|
vkUpdateDescriptorSets(device, 1, &descriptor_write, 0, nullptr);
|
|
}
|
|
|
|
DescriptorSet DescriptorSet::Duplicate()
|
|
{
|
|
MLX_PROFILE_FUNCTION();
|
|
DescriptorSet set;
|
|
set.Init(m_renderer, &RenderCore::Get().GetDescriptorPool(), m_layout);
|
|
return set;
|
|
}
|
|
|
|
VkDescriptorSet& DescriptorSet::operator()() noexcept
|
|
{
|
|
return m_desc_set[m_renderer->GetActiveImageIndex()];
|
|
}
|
|
|
|
VkDescriptorSet& DescriptorSet::Get() noexcept
|
|
{
|
|
return m_desc_set[m_renderer->GetActiveImageIndex()];
|
|
}
|
|
|
|
void DescriptorSet::Destroy() noexcept
|
|
{
|
|
MLX_PROFILE_FUNCTION();
|
|
if(m_pool != nullptr && RenderCore::Get().IsInit()) // checks if the render core is still init (it should always be init but just in case)
|
|
m_pool->FreeDescriptor(*this);
|
|
for(auto& set : m_desc_set)
|
|
{
|
|
if(set != VK_NULL_HANDLE)
|
|
set = VK_NULL_HANDLE;
|
|
}
|
|
}
|
|
}
|