DescriptorSetUpdater.h

Go to the documentation of this file.

#pragma once
#include <vector>
#include <deque>
#include <vulkan/vulkan_core.h>
 
namespace EngineCore {

    class DescriptorSetUpdater {
    public:
        DescriptorSetUpdater(VkDescriptorSet dstSet);

        [[nodiscard]] DescriptorSetUpdater& addStorageBuffer(uint32_t binding, const VkDescriptorBufferInfo* pBufferInfo);
        [[nodiscard]] DescriptorSetUpdater& addStorageBuffer(uint32_t binding, const VkDescriptorBufferInfo& bufferInfo);
        [[nodiscard]] DescriptorSetUpdater& addUniformBuffer(uint32_t binding, const VkDescriptorBufferInfo* pBufferInfo);
        [[nodiscard]] DescriptorSetUpdater& addUniformBuffer(uint32_t binding, const VkDescriptorBufferInfo& bufferInfo);
        [[nodiscard]] DescriptorSetUpdater& addTextureArrayIf(const bool condition, uint32_t binding, std::vector<VkDescriptorImageInfo>* imageInfos);
        [[nodiscard]] DescriptorSetUpdater& addCombinedImageSampler(uint32_t binding, const VkDescriptorImageInfo* pImageInfo);
        [[nodiscard]] DescriptorSetUpdater& addStorageImage(uint32_t binding, const VkDescriptorImageInfo* pImageInfo);

        void update(VkDevice device) const;
 
    private:
        VkDescriptorSet dstSet = VK_NULL_HANDLE;
        std::vector<VkWriteDescriptorSet> writeDescriptorSets;
        // Storage for buffer/image infos to ensure they outlive the update() call
        // VkWriteDescriptorSet stores pointers, so we need stable storage
        // Using deque because it doesn't invalidate references when elements are added
        std::deque<VkDescriptorBufferInfo> bufferInfoStorage;
        std::deque<VkDescriptorImageInfo> imageInfoStorage;
    };
 
    class DescriptorSetLayoutBuilder {
    public:
        DescriptorSetLayoutBuilder() {};
 
        [[nodiscard]] DescriptorSetLayoutBuilder& addStorageBuffer(uint32_t binding, VkShaderStageFlags stageFlags);
        [[nodiscard]] DescriptorSetLayoutBuilder& addUniformBuffer(uint32_t binding, VkShaderStageFlags stageFlags);
        [[nodiscard]] DescriptorSetLayoutBuilder& addCombinedImageSampler(uint32_t binding, VkShaderStageFlags stageFlags);
        [[nodiscard]] DescriptorSetLayoutBuilder& addStorageImage(uint32_t binding, VkShaderStageFlags stageFlags);
        [[nodiscard]] DescriptorSetLayoutBuilder& addStorageImageArray(uint32_t binding, uint32_t count, VkShaderStageFlags stageFlags);
 
        [[nodiscard]] std::vector<VkDescriptorSetLayoutBinding> build();
    private:
        std::vector<VkDescriptorSetLayoutBinding> layoutBindings;
    };
 
 
}