VulkanShaderProgram.hpp

Go to the documentation of this file.

#pragma once
 
#include "../Interface/ShaderProgram.hpp"
 
#include <initializer_list>
#include <vector>
#include <vulkan/vulkan.h>
 
namespace Video {
 
class Shader;
class VulkanShader;
class VulkanRenderer;
 
class VulkanShaderProgram : public ShaderProgram {
  public:
    struct StorageBufferInfo {
        VkPipelineStageFlags pipelineStages;
 
        bool readWrite;
    };
 
 
    VulkanShaderProgram(VulkanRenderer* vulkanRenderer, std::initializer_list<const Shader*> shaders);
 
    ~VulkanShaderProgram() final;
 
 
    const std::vector<const VulkanShader*>& GetShaders() const;
 
 
    const VkDescriptorSetLayout* GetDescriptorSetLayouts() const;
 
 
    const VkPushConstantRange* GetPushConstantRange() const;
 
 
    const StorageBufferInfo& GetStorageBufferInfo(uint32_t binding) const;
 
  private:
    VulkanShaderProgram(const VulkanShaderProgram& other) = delete;
 
    void AddUniformBuffer(BindingType bindingType);
    void AddMaterial(unsigned int textures);
    void AddPushConstants(unsigned int pushConstantCount, ShaderSource::ReflectionInfo::PushConstant* pushConstants, bool isComputeShader);
 
    VulkanRenderer* vulkanRenderer;
 
    std::vector<const VulkanShader*> shaders;
    VkDescriptorSetLayout descriptorSetLayouts[ShaderProgram::BindingType::BINDING_TYPES];
 
    bool usesPushConstants = false;
    VkPushConstantRange pushConstantRange;
 
    std::vector<StorageBufferInfo> storageBufferInfo;
};
 
} // namespace Video