vk_descriptor_set_layouts.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2024 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */

 
#pragma once
 
#include "BLI_map.hh"
#include "BLI_mutex.hh"
#include "BLI_utility_mixins.hh"
#include "BLI_vector.hh"
 
#include "vk_common.hh"
 
namespace blender::gpu {
 
class VKDevice;

struct VKDescriptorSetLayoutInfo {
  using Bindings = Vector<VkDescriptorType>;
 
  Bindings bindings;
  VkShaderStageFlags vk_shader_stage_flags;
 
  bool operator==(const VKDescriptorSetLayoutInfo &other) const
  {
    return vk_shader_stage_flags == other.vk_shader_stage_flags && bindings == other.bindings;
  };
};
 
};  // namespace blender::gpu
 
namespace blender {
template<> struct DefaultHash<gpu::VKDescriptorSetLayoutInfo> {
  uint64_t operator()(const gpu::VKDescriptorSetLayoutInfo &key) const
  {
    uint64_t hash = uint64_t(key.vk_shader_stage_flags);
    for (VkDescriptorType vk_descriptor_type : key.bindings) {
      hash = hash * 33 ^ uint64_t(vk_descriptor_type);
    }
    return hash;
  }
};
}  // namespace blender
 
namespace blender::gpu {
 
struct VKDescriptorBufferLayout {
  VkDeviceSize size;

  Vector<VkDeviceSize> binding_offsets;
};

class VKDescriptorSetLayouts : NonCopyable {
  friend class VKDevice;
 
 private:
  Map<VKDescriptorSetLayoutInfo, VkDescriptorSetLayout> vk_descriptor_set_layouts_;
  Map<VkDescriptorSetLayout, VKDescriptorBufferLayout> descriptor_buffer_layouts_;

  VkDescriptorSetLayoutCreateInfo vk_descriptor_set_layout_create_info_;
  Vector<VkDescriptorSetLayoutBinding> vk_descriptor_set_layout_bindings_;
  Mutex mutex_;
 
 public:
  VKDescriptorSetLayouts();
  virtual ~VKDescriptorSetLayouts();

  VkDescriptorSetLayout get_or_create(const VKDescriptorSetLayoutInfo &info,
                                      bool &r_created,
                                      bool &r_needed);

  VKDescriptorBufferLayout descriptor_buffer_layout_get(
      VkDescriptorSetLayout vk_descriptor_set_layout)
  {
    std::scoped_lock lock(mutex_);
    return descriptor_buffer_layouts_.lookup(vk_descriptor_set_layout);
  }

  void deinit();

  int64_t size() const
  {
    return vk_descriptor_set_layouts_.size();
  }
 
 private:
  void update_layout_bindings(const VKDescriptorSetLayoutInfo &info);
  VKDescriptorBufferLayout create_descriptor_buffer_layout(
      const VKDevice &device,
      const VKDescriptorSetLayoutInfo &info,
      VkDescriptorSetLayout vk_descriptor_set_layout) const;
};
 
}  // namespace blender::gpu