VulkanDescriptorSetLayout.h

#pragma once
 
#include "spark/render/DescriptorSet.h"
#include "spark/render/Resource.h"
#include "spark/render/ShaderStages.h"
#include "spark/render/vk/Export.h"
#include "spark/render/vk/Helpers.h"
#include "spark/render/vk/VulkanDescriptorLayout.h"
#include "spark/render/vk/VulkanDescriptorSet.h"
 
#include <memory>
 
SPARK_FWD_DECLARE_VK_HANDLE(VkDescriptorSetLayout)
 
namespace spark::render::vk
{
    class VulkanDevice;
    class VulkanDescriptorLayout;
    class VulkanDescriptorSet;
 
    class SPARK_RENDER_VK_EXPORT VulkanDescriptorSetLayout final : public DescriptorSetLayout<VulkanDescriptorLayout, VulkanDescriptorSet>, public Resource<VkDescriptorSetLayout>
    {
    public:
        explicit VulkanDescriptorSetLayout(const VulkanDevice& device,
                                           std::vector<std::unique_ptr<VulkanDescriptorLayout>>&& descriptor_layouts,
                                           unsigned int space,
                                           ShaderStage stages,
                                           unsigned int pool_size = 1024,
                                           unsigned int max_unbounded_array_size = 104857);
        ~VulkanDescriptorSetLayout() noexcept override;
 
        VulkanDescriptorSetLayout(const VulkanDescriptorSetLayout& other) = delete;
        VulkanDescriptorSetLayout(VulkanDescriptorSetLayout&& other) noexcept = delete;
        VulkanDescriptorSetLayout& operator=(const VulkanDescriptorSetLayout& other) = delete;
        VulkanDescriptorSetLayout& operator=(VulkanDescriptorSetLayout&& other) noexcept = delete;
 
        [[nodiscard]] unsigned poolSize() const noexcept;
 
        [[nodiscard]] std::size_t pools() const noexcept;
 
        [[nodiscard]] const VulkanDevice& device() const noexcept;
 
        [[nodiscard]] std::vector<const VulkanDescriptorLayout*> descriptors() const noexcept override;
 
        [[nodiscard]] const VulkanDescriptorLayout* descriptor(unsigned binding) const override;
 
        [[nodiscard]] unsigned space() const noexcept override;
 
        [[nodiscard]] ShaderStage shaderStage() const noexcept override;
 
        [[nodiscard]] unsigned uniforms() const noexcept override;
 
        [[nodiscard]] unsigned storages() const noexcept override;
 
        [[nodiscard]] unsigned images() const noexcept override;
 
        [[nodiscard]] unsigned samplers() const noexcept override;
 
        [[nodiscard]] unsigned buffers() const noexcept override;
 
        [[nodiscard]] unsigned staticSamplers() const noexcept override;
 
        [[nodiscard]] unsigned inputAttachments() const noexcept override;
 
        [[nodiscard]] std::unique_ptr<VulkanDescriptorSet> allocate(const std::vector<DescriptorBinding>& bindings = {}) const override;
 
        [[nodiscard]] std::unique_ptr<VulkanDescriptorSet> allocate(unsigned descriptors, const std::vector<DescriptorBinding>& bindings = {}) const override;
 
        [[nodiscard]] std::vector<std::unique_ptr<VulkanDescriptorSet>> allocateMultiple(unsigned descriptor_sets,
                                                                                         const std::vector<std::vector<DescriptorBinding>>& bindings = {}) const override;
 
        [[nodiscard]] std::vector<std::unique_ptr<VulkanDescriptorSet>> allocateMultiple(unsigned descriptor_sets,
                                                                                         const std::function<std::vector<DescriptorBinding>(
                                                                                             unsigned)>& binding_factory) const override;
 
        [[nodiscard]] std::vector<std::unique_ptr<VulkanDescriptorSet>> allocateMultiple(unsigned descriptor_sets,
                                                                                         unsigned descriptors,
                                                                                         const std::vector<std::vector<DescriptorBinding>>& bindings = {}) const override;
 
        [[nodiscard]] std::vector<std::unique_ptr<VulkanDescriptorSet>> allocateMultiple(unsigned descriptor_sets,
                                                                                         unsigned descriptors,
                                                                                         const std::function<std::vector<DescriptorBinding>(
                                                                                             unsigned)>& binding_factory) const override;
 
        void free(const VulkanDescriptorSet& descriptor_set) const noexcept override;
 
    private:
        struct Impl;
        std::unique_ptr<Impl> m_impl;
    };
}