StaticSet.h

Go to the documentation of this file.

#ifndef VULKAN_STATICSET
#define VULKAN_STATICSET
 
#include <glm/glm.hpp>
#include <tuple>
#include <concepts>
#include <type_traits>
 
#include "UniformBuffer.h"
#include "TextureImage.h"
#include "LogicalDevice.h"
#include "PhysicalDevice.h"
#include "Set.h"
 
 
namespace Vulkan {
 
    class StaticSet : public Set {
    public:
        //TODO comment
        struct StaticBufferBindingInfo : public Set::BindingInfo {
            StaticBufferBindingInfo(int size, const Buffers::UniformBuffer& buffer, int offset) : size{ size }, buffer{ buffer }, offset{ offset } {}
 
            DescriptorSetBindingCreationInfo generateDescriptorSetBindingInfo(const VkDescriptorSet& descriptorSet) const override {
                return DescriptorSetBindingCreationInfo{ binding, descriptorSet, size, buffer, offset };
            }
 
            int size; //the size of the binding
            const Buffers::UniformBuffer& buffer; //reference to the buffer of the binding
            int offset; //offset in his buffer of the binding
        };
 
 
        struct ImageBindingInfo : public Set::BindingInfo {
            ImageBindingInfo(const TextureImage& texture) : texture{ texture } {}
 
            DescriptorSetBindingCreationInfo generateDescriptorSetBindingInfo(const VkDescriptorSet& descriptorSet) const override {
                return DescriptorSetBindingCreationInfo{ binding, descriptorSet, texture };
            }
 
            const TextureImage& texture;
        };
 
 
 
        template<typename... Structs, typename... T> requires
            ((std::same_as<T, std::tuple<VkShaderStageFlagBits, Structs, Buffers::UniformBuffer*, int>> || std::same_as<T, std::tuple<VkShaderStageFlagBits, TextureImage*>>) && ...)
            StaticSet(const LogicalDevice& virtualGpu, const T&... bindingsInfo) : Set{ virtualGpu } {
            (this->bindingsInfo.push_back(createBindingInfo(bindingsInfo)), ...);
 
            createDescriptorSetLayout(std::pair{ std::get<0>(bindingsInfo), std::same_as<T, std::tuple<VkShaderStageFlagBits, TextureImage*>> ? VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER : VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER }...);
        }
 
 
 
        template<typename... Bindings>
        void fillBuffer(Buffers::UniformBuffer& buffer, const Bindings&... bindings) const {
            std::vector<StaticBufferBindingInfo*> bindingsInfo;
            try {
                bindingsInfo = bindingsPerBuffer.at(&buffer); //get the info of the bindings of the buffer
            }
            catch (const std::out_of_range&) {
                throw VulkanException{ "Failed to fill the uniform buffer", "The buffer to be filled isn't used in this set" };
            }
 
            //fill the buffer by placing the n-th binding at the position pointed by the n-th bindingInfo (for the buffer) offset
            int counter = 0;
            ([&bindingsInfo, &counter, &buffer](Bindings binding) {
                buffer.fillBuffer(binding, bindingsInfo[counter]->offset);
                counter++;
                }(bindings), ...);
 
        }
 
 
 
    private:
 
        template<typename Struct>
        std::unique_ptr<StaticBufferBindingInfo> createBindingInfo(const std::tuple<VkShaderStageFlagBits, Struct, Buffers::UniformBuffer*, int>& info) {
            auto tmp = std::make_unique<StaticBufferBindingInfo>(sizeof(std::get<1>(info)), *std::get<2>(info), std::get<3>(info));
            tmp->binding = bindingsInfo.size();
            bindingsPerBuffer[std::get<2>(info)].push_back(tmp.get());
            return tmp;
        }
 
 
        std::unique_ptr<ImageBindingInfo> createBindingInfo(const std::tuple<VkShaderStageFlagBits, TextureImage*>& info) {
            auto tmp = std::make_unique<ImageBindingInfo>(*std::get<1>(info));
            tmp->binding = bindingsInfo.size();
            return tmp;
        }
 
 
        std::map<const Buffers::UniformBuffer*, std::vector<StaticBufferBindingInfo*>> bindingsPerBuffer; //this map holds the bindings relative to each buffer used in the Set
 
 
    };
 
}
 
 
 
#endif