From 9f8cce3e4bc2041452fac402a547f9b90a9677a2 Mon Sep 17 00:00:00 2001
From: hyzboy <hyzboy@gmail.com>
Date: Thu, 15 Oct 2020 17:33:05 +0800
Subject: [PATCH] to improve CreateMemory(VkImage) function, auto BindImage

---
 src/RenderDevice/Vulkan/VKDeviceImage.cpp | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/RenderDevice/Vulkan/VKDeviceImage.cpp b/src/RenderDevice/Vulkan/VKDeviceImage.cpp
index abe0ba1e..4421631b 100644
--- 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