implement getBufferDeviceAddress

Author: Erfan-Ahmadi

include/nbl/video/ILogicalDevice.h

@@ -182,6 +182,8 @@ class NBL_API ILogicalDevice : public core::IReferenceCounted, public IDeviceMem
 
         virtual core::smart_refctd_ptr<IGPUBuffer> createBuffer(IGPUBuffer::SCreationParams&& creationParams) { return nullptr; }
 
+        virtual uint64_t getBufferDeviceAddress(IGPUBuffer* buffer) { return ~0ull; }
+
         //! Binds memory allocation to provide the backing for the resource.
         /** Available only on Vulkan, in OpenGL all resources create their own memory implicitly,
         so pooling or aliasing memory for different resources is not possible.

src/nbl/video/CVulkanLogicalDevice.h

@@ -510,7 +510,38 @@ class CVulkanLogicalDevice final : public ILogicalDevice
             return nullptr;
         }
     }
-        
+
+    uint64_t getBufferDeviceAddress(IGPUBuffer* buffer) override
+    {
+        constexpr uint64_t invalid_address = ~0ull;
+        CVulkanBuffer* vulkanBuffer = IBackendObject::device_compatibility_cast<CVulkanBuffer*>(buffer, this);
+        if (!vulkanBuffer)
+        {
+            // TODO: log error
+            assert(false);
+            return invalid_address;
+        }
+
+        if (!buffer->getCreationParams().usage.hasFlags(asset::IBuffer::EUF_SHADER_DEVICE_ADDRESS_BIT))
+        {
+            // TODO: log error: Buffer should've been created with EUF_SHADER_DEVICE_ADDRESS_BIT
+            assert(false);
+            return invalid_address;
+        }
+
+        if (!m_enabledFeatures.bufferDeviceAddress)
+        {
+            // TODO: log error: bufferDeviceAddress extension is not enbaled
+            assert(false);
+            return invalid_address;
+        }
+
+        VkBufferDeviceAddressInfo info = {};
+        info.sType = VK_STRUCTURE_TYPE_BUFFER_DEVICE_ADDRESS_INFO;
+        info.buffer = vulkanBuffer->getInternalObject();
+        return m_devf.vk.vkGetBufferDeviceAddress(m_vkdev, &info);
+    }
+
     core::smart_refctd_ptr<IGPUShader> createShader(core::smart_refctd_ptr<asset::ICPUShader>&& cpushader) override
     {
         const char* entryPoint = "main";