foray_blasinstance.hpp

Go to the documentation of this file.

#pragma once
#include "../foray_basics.hpp"
#include "../foray_glm.hpp"
#include "../foray_vulkan.hpp"
#include "foray_as_declares.hpp"
#include <functional>
 
namespace foray::as {
    class BlasInstance
    {
      public:
        using TransformUpdateFunc = std::function<void(glm::mat4&)>;
 
        inline BlasInstance() {}
        BlasInstance(uint64_t instanceId, const Blas* blas, TransformUpdateFunc getUpdatedGlobalTransformFunc);
        BlasInstance(uint64_t instanceId, const Blas* blas, const glm::mat4& globalTransform);
 
        FORAY_GETTER_V(InstanceId)
        FORAY_GETTER_CR(AsInstance)
        FORAY_GETTER_V(Blas)
        FORAY_PROPERTY_V(GetUpdatedGlobalTransformFunc)
 
        bool IsAnimated() const { return !!mGetUpdatedGlobalTransformFunc; }
 
        void SetGeometryMetaOffset(uint32_t offset);
 
        void SetShaderBindingTableOffset(uint32_t offset);
 
        inline VkGeometryInstanceFlagsKHR GetGeometryInstanceFlags() const { return mAsInstance.flags; }
        inline BlasInstance&              SetGeometryInstanceFlags(VkGeometryFlagsKHR flags)
        {
            mAsInstance.flags = flags;
            return *this;
        }
        inline BlasInstance& AddGeometryInstanceFlag(VkGeometryFlagBitsKHR flag)
        {
            mAsInstance.flags |= flag;
            return *this;
        }
 
 
        void Update();
 
        static void TranslateTransformMatrix(const glm::mat4& in, VkTransformMatrixKHR& out);
 
      protected:
        uint64_t mInstanceId = 0UL;
        const Blas* mBlas;
        TransformUpdateFunc mGetUpdatedGlobalTransformFunc = nullptr;
        VkAccelerationStructureInstanceKHR mAsInstance = {};
    };
}  // namespace foray::as