foray_shaderstagecreateinfos.hpp

Go to the documentation of this file.

#pragma once
 
#include "../foray_basics.hpp"
#include "../foray_vulkan.hpp"
#include <vector>
 
namespace foray::util {
    class ShaderStageCreateInfos
    {
      public:
        ShaderStageCreateInfos()                                         = default;
        ShaderStageCreateInfos& operator=(const ShaderStageCreateInfos&) = delete;
        ShaderStageCreateInfos(const ShaderStageCreateInfos&)            = delete;
 
        std::vector<VkPipelineShaderStageCreateInfo>* Get() { return &mShaderStageCreateInfos; };
 
        ShaderStageCreateInfos& Add(VkShaderStageFlagBits flagBits, VkShaderModule shaderModule)
        {
            VkPipelineShaderStageCreateInfo shaderStageInfo{};
            shaderStageInfo.sType  = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO;
            shaderStageInfo.stage  = flagBits;
            shaderStageInfo.module = shaderModule;
            shaderStageInfo.pName  = "main";
            mShaderStageCreateInfos.push_back(shaderStageInfo);
            return *this;
        }
 
      protected:
        std::vector<VkPipelineShaderStageCreateInfo> mShaderStageCreateInfos;
    };
}  // namespace foray::util