foray_generalsbt.hpp

Go to the documentation of this file.

#pragma once
#include "foray_basesbt.hpp"
#include "foray_rtshadertypes.hpp"
#include "../core/foray_shadermodule.hpp"
#include <vector>
 
namespace foray::rtpipe {
    class GeneralShaderBindingTable : public ShaderBindingTableBase
    {
      public:
        struct ShaderGroup
        {
            core::ShaderModule* Module = nullptr;
        };
 
        explicit GeneralShaderBindingTable(RtShaderGroupType groupType, VkDeviceSize entryDataSize = 0);
 
        void SetGroup(GroupIndex groupIndex, core::ShaderModule* shader);
        void SetGroup(GroupIndex groupIndex, core::ShaderModule* shader, const void* data);
 
        FORAY_GETTER_CR(Groups)
        FORAY_GETTER_V(ShaderGroupType)
 
        virtual void        WriteToShaderCollection(RtShaderCollection& collection) const override;
        virtual VectorRange WriteToShaderGroupCiVector(std::vector<VkRayTracingShaderGroupCreateInfoKHR>& groupCis, const RtShaderCollection& shaderCollection) const override;
 
        virtual void Destroy() override;
 
      protected:
        RtShaderGroupType              mShaderGroupType = RtShaderGroupType::Undefined;
        std::vector<ShaderGroup> mGroups;
 
        inline virtual size_t GetGroupArrayCount() const override { return mGroups.size(); }
    };
}  // namespace foray::rtpipe