foray_hitsbt.hpp

Go to the documentation of this file.

#pragma once
#include "foray_basesbt.hpp"
#include "foray_rtshadertypes.hpp"
#include <vector>
 
namespace foray::rtpipe {
 
    class HitShaderBindingTable : public ShaderBindingTableBase
    {
      public:
        struct ShaderGroup
        {
            core::ShaderModule* ClosestHitModule = nullptr;
            core::ShaderModule* AnyHitModule     = nullptr;
            core::ShaderModule* IntersectModule  = nullptr;
        };
 
        inline explicit HitShaderBindingTable(VkDeviceSize entryDataSize = 0) : ShaderBindingTableBase(entryDataSize) {}
 
        void SetGroup(GroupIndex groupIndex, core::ShaderModule* closestHit, core::ShaderModule* anyHit, core::ShaderModule* intersect);
        void SetGroup(GroupIndex groupIndex, core::ShaderModule* closestHit, core::ShaderModule* anyHit, core::ShaderModule* intersect, const void* data);
 
        virtual void        WriteToShaderCollection(RtShaderCollection& collection) const override;
        virtual VectorRange WriteToShaderGroupCiVector(std::vector<VkRayTracingShaderGroupCreateInfoKHR>& groupCis, const RtShaderCollection& shaderCollection) const override;
 
 
        FORAY_GETTER_CR(Groups)
 
        virtual void Destroy() override;
 
      protected:
        std::vector<ShaderGroup> mGroups;
 
        inline virtual size_t GetGroupArrayCount() const override { return mGroups.size(); }
    };
 
}  // namespace foray