foray_tlas.hpp

Go to the documentation of this file.

#pragma once
#include "../core/foray_managedresource.hpp"
#include "../foray_basics.hpp"
#include "../scene/foray_component.hpp"
#include "../scene/foray_scene_declares.hpp"
#include "../util/foray_dualbuffer.hpp"
#include "foray_as_declares.hpp"
#include "foray_blasinstance.hpp"
#include "foray_geometrymetabuffer.hpp"
#include <map>
#include <vulkan/vulkan.h>
 
namespace foray::as {
 
 
    class Tlas : public core::ManagedResource
    {
      public:
        Tlas() = default;
        virtual ~Tlas() { Destroy(); }
 
        inline virtual std::string_view GetTypeName() const override { return "Top-Level AS"; }
 
        virtual void CreateOrUpdate(core::Context* context = nullptr);
        virtual void UpdateLean(VkCommandBuffer cmdBuffer, uint32_t frameIndex);
 
        inline virtual bool Exists() const override { return !!mAccelerationStructure; }
        virtual void        Destroy() override;
 
        FORAY_GETTER_V(AccelerationStructure)
        FORAY_GETTER_CR(TlasMemory)
        FORAY_GETTER_V(TlasAddress)
        FORAY_GETTER_CR(MetaBuffer)
        FORAY_GETTER_V(Dirty)
 
        operator VkAccelerationStructureKHR() { return mAccelerationStructure; }
 
        void RemoveBlasInstance(uint64_t id);
        const BlasInstance* GetBlasInstance(uint64_t id) const;
        uint64_t AddBlasInstanceAuto(scene::ncomp::MeshInstance* meshInstance);
        uint64_t AddBlasInstanceAnimated(const Blas& blas, BlasInstance::TransformUpdateFunc getUpdatedGlobalTransformFunc);
        uint64_t AddBlasInstanceStatic(const Blas& blas, const glm::mat4& transform);
 
        void ClearBlasInstances();
 
        FORAY_GETTER_CR(AnimatedBlasInstances)
        FORAY_GETTER_CR(StaticBlasInstances)
 
      protected:
        core::Context*                   mContext               = nullptr;
        bool                             mDirty                 = false;
        VkAccelerationStructureKHR       mAccelerationStructure = nullptr;
        core::ManagedBuffer              mTlasMemory;
        util::DualBuffer                 mInstanceBuffer;
        core::ManagedBuffer              mScratchBuffer;
        VkDeviceAddress                  mTlasAddress = 0;
        std::map<uint64_t, BlasInstance> mAnimatedBlasInstances;
        std::map<uint64_t, BlasInstance> mStaticBlasInstances;
        uint64_t                         mNextId = 0;
        GeometryMetaBuffer               mMetaBuffer;
    };
}  // namespace foray::as