96 lines
3.0 KiB
C++
96 lines
3.0 KiB
C++
/* SPDX-FileCopyrightText: 2024 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
/** \file
|
|
* \ingroup gpu
|
|
*/
|
|
|
|
#include "vk_descriptor_set_layouts.hh"
|
|
#include "vk_backend.hh"
|
|
#include "vk_memory.hh"
|
|
|
|
namespace blender::gpu {
|
|
VKDescriptorSetLayouts::VKDescriptorSetLayouts()
|
|
{
|
|
vk_descriptor_set_layout_create_info_ = {};
|
|
vk_descriptor_set_layout_create_info_.sType =
|
|
VK_STRUCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO;
|
|
}
|
|
|
|
VKDescriptorSetLayouts::~VKDescriptorSetLayouts()
|
|
{
|
|
deinit();
|
|
}
|
|
|
|
VkDescriptorSetLayout VKDescriptorSetLayouts::get_or_create(const VKDescriptorSetLayoutInfo &info,
|
|
bool &r_created,
|
|
bool &r_needed)
|
|
{
|
|
r_created = false;
|
|
r_needed = !info.bindings.is_empty();
|
|
if (r_needed == false) {
|
|
return VK_NULL_HANDLE;
|
|
}
|
|
|
|
std::scoped_lock mutex(mutex_);
|
|
|
|
VkDescriptorSetLayout *layout = vk_descriptor_set_layouts_.lookup_ptr(info);
|
|
if (layout) {
|
|
return *layout;
|
|
}
|
|
|
|
update_layout_bindings(info);
|
|
|
|
vk_descriptor_set_layout_create_info_.bindingCount = vk_descriptor_set_layout_bindings_.size();
|
|
vk_descriptor_set_layout_create_info_.pBindings = vk_descriptor_set_layout_bindings_.data();
|
|
|
|
const VKDevice &device = VKBackend::get().device_get();
|
|
VkDescriptorSetLayout vk_descriptor_set_layout = VK_NULL_HANDLE;
|
|
VK_ALLOCATION_CALLBACKS;
|
|
vkCreateDescriptorSetLayout(device.device_get(),
|
|
&vk_descriptor_set_layout_create_info_,
|
|
vk_allocation_callbacks,
|
|
&vk_descriptor_set_layout);
|
|
BLI_assert(vk_descriptor_set_layout != VK_NULL_HANDLE);
|
|
|
|
vk_descriptor_set_layout_create_info_.bindingCount = 0;
|
|
vk_descriptor_set_layout_create_info_.pBindings = nullptr;
|
|
vk_descriptor_set_layout_bindings_.clear();
|
|
vk_descriptor_set_layouts_.add(info, vk_descriptor_set_layout);
|
|
r_created = true;
|
|
|
|
return vk_descriptor_set_layout;
|
|
}
|
|
|
|
void VKDescriptorSetLayouts::update_layout_bindings(const VKDescriptorSetLayoutInfo &info)
|
|
{
|
|
BLI_assert(vk_descriptor_set_layout_bindings_.is_empty());
|
|
vk_descriptor_set_layout_bindings_.reserve(info.bindings.size());
|
|
|
|
uint32_t index = 0;
|
|
for (const VkDescriptorType &vk_descriptor_type : info.bindings) {
|
|
VkDescriptorSetLayoutBinding binding = {};
|
|
binding.binding = index++;
|
|
binding.descriptorCount = 1;
|
|
binding.descriptorType = vk_descriptor_type;
|
|
binding.pImmutableSamplers = VK_NULL_HANDLE;
|
|
binding.stageFlags = info.vk_shader_stage_flags;
|
|
vk_descriptor_set_layout_bindings_.append(binding);
|
|
}
|
|
}
|
|
|
|
void VKDescriptorSetLayouts::deinit()
|
|
{
|
|
std::scoped_lock mutex(mutex_);
|
|
VK_ALLOCATION_CALLBACKS;
|
|
const VKDevice &device = VKBackend::get().device_get();
|
|
for (VkDescriptorSetLayout &vk_descriptor_set_layout : vk_descriptor_set_layouts_.values()) {
|
|
vkDestroyDescriptorSetLayout(
|
|
device.device_get(), vk_descriptor_set_layout, vk_allocation_callbacks);
|
|
}
|
|
vk_descriptor_set_layouts_.clear();
|
|
}
|
|
|
|
} // namespace blender::gpu
|