to improve CreateMemory(VkImage) function, auto BindImage

2020-10-15 17:33:05 +08:00 · 2020-10-15 17:33:05 +08:00 · 9f8cce3e4b
commit 9f8cce3e4b
parent c8f6017641
1 changed files with 11 additions and 1 deletions
--- a/src/RenderDevice/Vulkan/VKDeviceImage.cpp
+++ b/src/RenderDevice/Vulkan/VKDeviceImage.cpp
@ -29,6 +29,16 @@ Memory *Device::CreateMemory(VkImage image,const uint32_t flag)

    vkGetImageMemoryRequirements(attr->device,image,&memReqs);

-    return CreateMemory(memReqs,flag);
+    Memory *mem=CreateMemory(memReqs,flag);
+    
+    if(!mem)return(nullptr);
+
+    if(!mem->BindImage(image))
+    {
+        delete mem;
+        return(nullptr);
+    }
+
+    return(mem);
 }
 VK_NAMESPACE_END