mirror of
https://github.com/seekrs/MacroLibX.git
synced 2026-01-11 22:53:34 +00:00
103 lines
3.7 KiB
C++
103 lines
3.7 KiB
C++
/* ************************************************************************** */
|
|
/* */
|
|
/* ::: :::::::: */
|
|
/* vk_descriptor_set.cpp :+: :+: :+: */
|
|
/* +:+ +:+ +:+ */
|
|
/* By: maldavid <kbz_8.dev@akel-engine.com> +#+ +:+ +#+ */
|
|
/* +#+#+#+#+#+ +#+ */
|
|
/* Created: 2023/01/23 18:40:44 by maldavid #+# #+# */
|
|
/* Updated: 2023/04/22 19:52:08 by maldavid ### ########.fr */
|
|
/* */
|
|
/* ************************************************************************** */
|
|
|
|
#include "vk_descriptor_set.h"
|
|
#include "vk_descriptor_pool.h"
|
|
#include "vk_descriptor_set_layout.h"
|
|
#include <renderer/buffers/vk_ubo.h>
|
|
#include <renderer/renderer.h>
|
|
|
|
namespace mlx
|
|
{
|
|
void DescriptorSet::init(Renderer* renderer, DescriptorPool* pool, DescriptorSetLayout* layout)
|
|
{
|
|
_renderer = renderer;
|
|
_layout = layout;
|
|
_pool = pool;
|
|
|
|
auto device = Render_Core::get().getDevice().get();
|
|
|
|
std::array<VkDescriptorSetLayout, MAX_FRAMES_IN_FLIGHT> layouts;
|
|
layouts.fill(layout->get());
|
|
|
|
VkDescriptorSetAllocateInfo allocInfo{};
|
|
allocInfo.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_ALLOCATE_INFO;
|
|
allocInfo.descriptorPool = _pool->get();
|
|
allocInfo.descriptorSetCount = static_cast<uint32_t>(MAX_FRAMES_IN_FLIGHT);
|
|
allocInfo.pSetLayouts = layouts.data();
|
|
|
|
if(vkAllocateDescriptorSets(device, &allocInfo, _desc_set.data()) != VK_SUCCESS)
|
|
core::error::report(e_kind::fatal_error, "Vulkan : failed to allocate descriptor set");
|
|
}
|
|
|
|
void DescriptorSet::writeDescriptor(int binding, UBO* ubo) noexcept
|
|
{
|
|
auto device = Render_Core::get().getDevice().get();
|
|
|
|
for(int i = 0; i < MAX_FRAMES_IN_FLIGHT; i++)
|
|
{
|
|
VkDescriptorBufferInfo bufferInfo{};
|
|
bufferInfo.buffer = ubo->get(i);
|
|
bufferInfo.offset = ubo->getOffset(i);
|
|
bufferInfo.range = ubo->getSize(i);
|
|
|
|
VkWriteDescriptorSet descriptorWrite{};
|
|
descriptorWrite.sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET;
|
|
descriptorWrite.dstSet = _desc_set[i];
|
|
descriptorWrite.dstBinding = binding;
|
|
descriptorWrite.dstArrayElement = 0;
|
|
descriptorWrite.descriptorType = VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER;
|
|
descriptorWrite.descriptorCount = 1;
|
|
descriptorWrite.pBufferInfo = &bufferInfo;
|
|
|
|
vkUpdateDescriptorSets(device, 1, &descriptorWrite, 0, nullptr);
|
|
}
|
|
}
|
|
|
|
void DescriptorSet::writeDescriptor(int binding, VkImageView view, VkSampler sampler) noexcept
|
|
{
|
|
auto device = Render_Core::get().getDevice().get();
|
|
|
|
VkDescriptorImageInfo imageInfo{};
|
|
imageInfo.imageLayout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL;
|
|
imageInfo.imageView = view;
|
|
imageInfo.sampler = sampler;
|
|
|
|
VkWriteDescriptorSet descriptorWrite{};
|
|
descriptorWrite.sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET;
|
|
descriptorWrite.dstSet = _desc_set[_renderer->getActiveImageIndex()];
|
|
descriptorWrite.dstBinding = binding;
|
|
descriptorWrite.dstArrayElement = 0;
|
|
descriptorWrite.descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER;
|
|
descriptorWrite.descriptorCount = 1;
|
|
descriptorWrite.pImageInfo = &imageInfo;
|
|
|
|
vkUpdateDescriptorSets(device, 1, &descriptorWrite, 0, nullptr);
|
|
}
|
|
|
|
DescriptorSet DescriptorSet::duplicate()
|
|
{
|
|
DescriptorSet set;
|
|
set.init(_renderer, _pool, _layout);
|
|
return set;
|
|
}
|
|
|
|
VkDescriptorSet& DescriptorSet::operator()() noexcept
|
|
{
|
|
return _desc_set[_renderer->getActiveImageIndex()];
|
|
}
|
|
VkDescriptorSet& DescriptorSet::get() noexcept
|
|
{
|
|
return _desc_set[_renderer->getActiveImageIndex()];
|
|
}
|
|
}
|