From 44bf7cf539949a6f26862201729fce5000b493f0 Mon Sep 17 00:00:00 2001
From: assiduous <assiduous@diligentgraphics.com>
Date: Fri, 6 Nov 2020 20:25:14 -0800
Subject: Few minor updates to TLAS and BLAS implementations

---
 Graphics/GraphicsEngine/include/BottomLevelASBase.hpp | 8 ++++++++
 Graphics/GraphicsEngine/include/TopLevelASBase.hpp    | 8 ++++++++
 2 files changed, 16 insertions(+)

(limited to 'Graphics/GraphicsEngine')

diff --git a/Graphics/GraphicsEngine/include/BottomLevelASBase.hpp b/Graphics/GraphicsEngine/include/BottomLevelASBase.hpp
index 3ec0efd2..caee8620 100644
--- a/Graphics/GraphicsEngine/include/BottomLevelASBase.hpp
+++ b/Graphics/GraphicsEngine/include/BottomLevelASBase.hpp
@@ -117,6 +117,12 @@ public:
         return this->m_State;
     }
 
+    /// Implementation of IBottomLevelAS::GetScratchBufferSizes()
+    virtual ScratchBufferSizes DILIGENT_CALL_TYPE GetScratchBufferSizes() const override
+    {
+        return this->m_ScratchSize;
+    }
+
     bool IsInKnownState() const
     {
         return this->m_State != RESOURCE_STATE_UNKNOWN;
@@ -192,6 +198,8 @@ protected:
 
     void* m_pRawPtr = nullptr;
 
+    ScratchBufferSizes m_ScratchSize;
+
 #ifdef DILIGENT_DEVELOPMENT
     std::atomic<Uint32> m_Version{0};
 #endif
diff --git a/Graphics/GraphicsEngine/include/TopLevelASBase.hpp b/Graphics/GraphicsEngine/include/TopLevelASBase.hpp
index 83205cd2..efba30c7 100644
--- a/Graphics/GraphicsEngine/include/TopLevelASBase.hpp
+++ b/Graphics/GraphicsEngine/include/TopLevelASBase.hpp
@@ -185,6 +185,12 @@ public:
         return this->m_State;
     }
 
+    /// Implementation of ITopLevelAS::GetScratchBufferSizes().
+    virtual ScratchBufferSizes DILIGENT_CALL_TYPE GetScratchBufferSizes() const override
+    {
+        return m_ScratchSize;
+    }
+
     bool IsInKnownState() const
     {
         return this->m_State != RESOURCE_STATE_UNKNOWN;
@@ -248,6 +254,8 @@ protected:
     RESOURCE_STATE m_State                 = RESOURCE_STATE_UNKNOWN;
     Uint32         m_HitShadersPerInstance = 0;
 
+    ScratchBufferSizes m_ScratchSize;
+
     StringPool m_StringPool;
 
     struct InstanceDesc
-- 
cgit v1.2.3