vk_descriptor_pools.hh

Go to the documentation of this file.

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

 
#pragma once
 
#include "BLI_vector.hh"
 
#include "vk_descriptor_set.hh"
 
namespace blender::gpu {
class VKDevice;

class VKDescriptorPools {
  static constexpr uint32_t POOL_SIZE_STORAGE_BUFFER = 1000;
  static constexpr uint32_t POOL_SIZE_DESCRIPTOR_SETS = 250;
  static constexpr uint32_t POOL_SIZE_STORAGE_IMAGE = 250;
  static constexpr uint32_t POOL_SIZE_COMBINED_IMAGE_SAMPLER = 250;
  static constexpr uint32_t POOL_SIZE_UNIFORM_BUFFER = 500;
  static constexpr uint32_t POOL_SIZE_UNIFORM_TEXEL_BUFFER = 100;
  static constexpr uint32_t POOL_SIZE_INPUT_ATTACHMENT = 100;

  Vector<VkDescriptorPool> recycled_pools_;
  VkDescriptorPool vk_descriptor_pool_ = VK_NULL_HANDLE;
  Mutex mutex_;
 
 public:
  ~VKDescriptorPools();
 
  void init(const VKDevice &vk_device);

  VkDescriptorSet allocate(const VkDescriptorSetLayout descriptor_set_layout);

  void recycle(VkDescriptorPool vk_descriptor_pool);
 
 private:
  void add_new_pool(const VKDevice &device);
  void discard_active_pool(VKContext &vk_context);
  void ensure_pool(const VKDevice &device);
};
}  // namespace blender::gpu