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> pools_;
  int64_t active_pool_index_ = 0;
 
 public:
  VKDescriptorPools();
  ~VKDescriptorPools();
 
  void init(const VKDevice &vk_device);
 
  VkDescriptorSet allocate(const VkDescriptorSetLayout descriptor_set_layout);

  void discard(VKContext &vk_context);
 
 private:
  VkDescriptorPool active_pool_get();
  void activate_next_pool();
  void activate_last_pool();
  bool is_last_pool_active();
  void add_new_pool(const VKDevice &device);
};
}  // namespace blender::gpu