foray_rtshadercollection.hpp

Go to the documentation of this file.

#pragma once
#include "../core/foray_shadermodule.hpp"
#include "foray_rtshadertypes.hpp"
#include <unordered_map>
#include <vector>
 
namespace foray::rtpipe {
 
    class RtShaderCollection
    {
      public:
        void Add(core::ShaderModule* module, RtShaderType type, const char* entryPointName = "main");
        void Clear();
        uint32_t IndexOf(core::ShaderModule* module) const;
 
        FORAY_GETTER_R(ShaderStageCis)
 
      protected:
        struct Entry
        {
            core::ShaderModule* Module         = nullptr;
            RtShaderType        Type           = RtShaderType::Undefined;
            const char*         EntryPointName = nullptr;
            uint32_t            StageCiIndex   = VK_SHADER_UNUSED_KHR;
        };
 
        std::unordered_map<core::ShaderModule*, Entry> mEntries;
        std::vector<VkPipelineShaderStageCreateInfo>   mShaderStageCis;
    };
}  // namespace foray::rtpipe