foray_basesbt.hpp

Go to the documentation of this file.

#pragma once
#include "../core/foray_managedbuffer.hpp"
#include "../core/foray_shadermodule.hpp"
#include "foray_rtpipeline_declares.hpp"
#include "foray_rtshadertypes.hpp"
#include <unordered_map>
#include <vector>
 
namespace foray::rtpipe {
 
    class ShaderBindingTableBase
    {
      public:
        explicit ShaderBindingTableBase(VkDeviceSize entryDataSize = 0);
 
        struct VectorRange
        {
            int32_t Start = -1;
            int32_t Count = 0;
        };
 
        virtual void Build(core::Context* context, const VkPhysicalDeviceRayTracingPipelinePropertiesKHR& pipelineProperties, const std::vector<const uint8_t*>& handles);
 
        FORAY_GETTER_V(EntryDataSize)
        FORAY_GETTER_V(AddressRegion)
        FORAY_GETTER_CR(Buffer)
 
        
        std::vector<uint8_t>& GroupDataAt(GroupIndex groupIndex);
        const std::vector<uint8_t>& GroupDataAt(GroupIndex groupIndex) const;
 
        template <typename TData>
        TData* GroupDataAt(GroupIndex groupIndex)
        {
            std::vector<uint8_t>& data = GroupDataAt(groupIndex);
            if(data.size() > 0)
            {
                Assert(sizeof(TData) == data.size(), "Custom data and sizeof(TData) size mismatch!");
                return reinterpret_cast<TData*>(data.data());
            }
            else
            {
                return nullptr;
            }
        }
 
        template <typename TData>
        const TData* GroupDataAt(GroupIndex groupIndex) const
        {
            const std::vector<uint8_t>& data = GroupDataAt(groupIndex);
            if(data.size() > 0)
            {
                Assert(sizeof(TData) == data.size(), "Custom data and sizeof(TData) size mismatch!");
                return reinterpret_cast<const TData*>(data.data());
            }
            else
            {
                return nullptr;
            }
        }
 
        void SetData(GroupIndex groupIndex, const void* data);
 
        template <typename TData>
        void SetData(GroupIndex groupIndex, const TData& data)
        {
            Assert(sizeof(TData) == mEntryDataSize);
            SetData(groupIndex, &data);
        }
 
        virtual ShaderBindingTableBase& SetEntryDataSize(VkDeviceSize newSize);
 
        virtual void WriteToShaderCollection(RtShaderCollection& collection) const = 0;
        virtual VectorRange WriteToShaderGroupCiVector(std::vector<VkRayTracingShaderGroupCreateInfoKHR>& groupCis, const RtShaderCollection& shaderCollection) const = 0;
 
        virtual void Destroy();
 
        inline virtual ~ShaderBindingTableBase() { Destroy(); }
 
 
      protected:
        core::ManagedBuffer             mBuffer;
        VkStridedDeviceAddressRegionKHR mAddressRegion{};
 
        VkDeviceSize mEntryDataSize{};
 
        std::vector<std::vector<uint8_t>> mGroupData{};
 
        virtual size_t GetGroupArrayCount() const = 0;
        void           ArrayResized(size_t newSize);
    };
}  // namespace foray::rtpipe