IBufferSuballocation: added SetUserData()/GetUserData() methods

2 files changed, 27 insertions, 0 deletions

diff --git a/Graphics/GraphicsTools/interface/BufferSuballocator.h b/Graphics/GraphicsTools/interface/BufferSuballocator.h
index 93c99db1..23fbdbbf 100644
--- a/Graphics/GraphicsTools/interface/BufferSuballocator.h
+++ b/Graphics/GraphicsTools/interface/BufferSuballocator.h
@@ -60,6 +60,21 @@ struct IBufferSuballocation : public IObject
 
     /// Returns the pointer to the parent allocator.
     virtual IBufferSuballocator* GetAllocator() = 0;
+
+    /// Stores a pointer to the user-provided data object, which
+    /// may later be retreived through GetUserData().
+    ///
+    /// \param [in] pUserData - Pointer to the user data object to store.
+    ///
+    /// \note   The method is not thread-safe and an application
+    ///         must externally synchronize the access.
+    virtual void SetUserData(IObject* pUserData) = 0;
+
+    /// Returns the pointer to the user data object previously
+    /// set with SetUserData() method.
+    ///
+    /// \return     Pointer to the user data object
+    virtual IObject* GetUserData() const = 0;
 };
 
 
diff --git a/Graphics/GraphicsTools/src/BufferSuballocator.cpp b/Graphics/GraphicsTools/src/BufferSuballocator.cpp
index 8dcb2879..2c5d60cd 100644
--- a/Graphics/GraphicsTools/src/BufferSuballocator.cpp
+++ b/Graphics/GraphicsTools/src/BufferSuballocator.cpp
@@ -92,6 +92,16 @@ public:
 
     virtual IBufferSuballocator* GetAllocator() override final;
 
+    virtual void SetUserData(IObject* pUserData) override final
+    {
+        m_pUserData = pUserData;
+    }
+
+    virtual IObject* GetUserData() const override final
+    {
+        return m_pUserData.RawPtr<IObject>();
+    }
+
 private:
     RefCntAutoPtr<BufferSuballocatorImpl> m_pParentAllocator;
 
@@ -99,6 +109,8 @@ private:
 
     const Uint32 m_Offset;
     const Uint32 m_Size;
+
+    RefCntAutoPtr<IObject> m_pUserData;
 };
 
 class BufferSuballocatorImpl final : public ObjectBase<IBufferSuballocator>