DescriptorSetBuilder.cpp

Go to the documentation of this file.

#include "Engine/Renderer/DescriptorSetUpdater.h"
#include <cassert>
 
namespace EngineCore {
 
    DescriptorSetUpdater::DescriptorSetUpdater(VkDescriptorSet dstSet) : dstSet(dstSet){
 
    }
 
    DescriptorSetUpdater& DescriptorSetUpdater::addStorageBuffer(uint32_t binding, const VkDescriptorBufferInfo *pBufferInfo)
    {
        // Store a copy of the buffer info to ensure it outlives this call
        bufferInfoStorage.push_back(*pBufferInfo);
        writeDescriptorSets.push_back({
            .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
            .dstSet = dstSet,
            .dstBinding = binding,
            .dstArrayElement = 0,
            .descriptorCount = 1u,
            .descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
            .pBufferInfo = &bufferInfoStorage.back(),
        });
        return *this;
    }
 
    DescriptorSetUpdater & DescriptorSetUpdater::addUniformBuffer(uint32_t binding, const VkDescriptorBufferInfo *pBufferInfo)
    {
        // Store a copy of the buffer info to ensure it outlives this call
        bufferInfoStorage.push_back(*pBufferInfo);
        writeDescriptorSets.push_back({
            .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
            .dstSet = dstSet,
            .dstBinding = binding,
            .dstArrayElement = 0,
            .descriptorCount = 1u,
            .descriptorType = VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER,
            .pBufferInfo = &bufferInfoStorage.back(),
        });
        return *this;
    }
 
    DescriptorSetUpdater& DescriptorSetUpdater::addStorageBuffer(uint32_t binding, const VkDescriptorBufferInfo& bufferInfo)
    {
        // Store a copy of the buffer info to ensure it outlives this call
        bufferInfoStorage.push_back(bufferInfo);
        writeDescriptorSets.push_back({
            .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
            .dstSet = dstSet,
            .dstBinding = binding,
            .dstArrayElement = 0,
            .descriptorCount = 1u,
            .descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
            .pBufferInfo = &bufferInfoStorage.back(),
        });
        return *this;
    }
 
    DescriptorSetUpdater & DescriptorSetUpdater::addUniformBuffer(uint32_t binding, const VkDescriptorBufferInfo& bufferInfo)
    {
        // Store a copy of the buffer info to ensure it outlives this call
        bufferInfoStorage.push_back(bufferInfo);
        writeDescriptorSets.push_back({
            .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
            .dstSet = dstSet,
            .dstBinding = binding,
            .dstArrayElement = 0,
            .descriptorCount = 1u,
            .descriptorType = VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER,
            .pBufferInfo = &bufferInfoStorage.back(),
        });
        return *this;
    }
 
    DescriptorSetUpdater & DescriptorSetUpdater::addTextureArrayIf(const bool condition, uint32_t binding, std::vector<VkDescriptorImageInfo>* imageInfos) {
        if (condition) {
            writeDescriptorSets.push_back(
            {
                .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
                .dstSet = dstSet,
                .dstBinding = binding,
                .dstArrayElement = 0,
                .descriptorCount = static_cast<uint32_t>(imageInfos->size()),
                .descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER,
                .pImageInfo = imageInfos->data(),
                .pBufferInfo = nullptr,
            });
        }
        return *this;
    }
 
    DescriptorSetUpdater & DescriptorSetUpdater::addCombinedImageSampler(uint32_t binding, const VkDescriptorImageInfo* pImageInfo) {
        // Store a copy of the image info to ensure it outlives this call
        imageInfoStorage.push_back(*pImageInfo);
        writeDescriptorSets.push_back({
            .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
            .dstSet = dstSet,
            .dstBinding = binding,
            .dstArrayElement = 0,
            .descriptorCount = 1u,
            .descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER,
            .pImageInfo = &imageInfoStorage.back(),
        });
        return *this;
    }
 
    DescriptorSetUpdater & DescriptorSetUpdater::addStorageImage(uint32_t binding, const VkDescriptorImageInfo* pImageInfo) {
        // Store a copy of the image info to ensure it outlives this call
        imageInfoStorage.push_back(*pImageInfo);
        writeDescriptorSets.push_back({
            .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
            .dstSet = dstSet,
            .dstBinding = binding,
            .dstArrayElement = 0,
            .descriptorCount = 1u,
            .descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE,
            .pImageInfo = &imageInfoStorage.back(),
        });
        return *this;
    }
 
    void DescriptorSetUpdater::update(VkDevice device) const {
        assert( device );
        vkUpdateDescriptorSets(device, static_cast<uint32_t>(writeDescriptorSets.size()), writeDescriptorSets.data(), 0, nullptr);
    }
 
    DescriptorSetLayoutBuilder & DescriptorSetLayoutBuilder::addStorageBuffer(uint32_t binding, VkShaderStageFlags stageFlags) {
        layoutBindings.push_back({
            .binding = binding,
            .descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
            .descriptorCount = 1u,
            .stageFlags = stageFlags,
        });
        return *this;
    }
 
    DescriptorSetLayoutBuilder & DescriptorSetLayoutBuilder::addUniformBuffer(uint32_t binding,
        VkShaderStageFlags stageFlags) {
        layoutBindings.push_back({
            .binding = binding,
            .descriptorType = VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER,
            .descriptorCount = 1u,
            .stageFlags = stageFlags,
        });
        return *this;
    }
 
    DescriptorSetLayoutBuilder & DescriptorSetLayoutBuilder::addCombinedImageSampler(uint32_t binding,
        VkShaderStageFlags stageFlags) {
        layoutBindings.push_back({
            .binding = binding,
            .descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER,
            .descriptorCount = 1u,
            .stageFlags = stageFlags,
        });
        return *this;
    }
 
    DescriptorSetLayoutBuilder & DescriptorSetLayoutBuilder::addStorageImage(uint32_t binding,
        VkShaderStageFlags stageFlags) {
        layoutBindings.push_back({
            .binding = binding,
            .descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE,
            .descriptorCount = 1u,
            .stageFlags = stageFlags,
        });
        return *this;
    }
 
    DescriptorSetLayoutBuilder & DescriptorSetLayoutBuilder::addStorageImageArray(uint32_t binding,
        uint32_t count, VkShaderStageFlags stageFlags) {
        layoutBindings.push_back({
            .binding = binding,
            .descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE,
            .descriptorCount = count,
            .stageFlags = stageFlags,
        });
        return *this;
    }
 
    std::vector<VkDescriptorSetLayoutBinding> DescriptorSetLayoutBuilder::build() {
        return layoutBindings;
    }
}