Restore old initial/final layout requirement bahavior (#2288)

vulkano-rs · Aug 16, 2023 · db3df4e · db3df4e
1 parent 54d20c0
commit db3df4e
Showing 1 changed file with 25 additions and 1 deletion.
diff --git a/vulkano/src/image/sys.rs b/vulkano/src/image/sys.rs
@@ -1156,10 +1156,34 @@ impl RawImage {
             return Err((VulkanError::from(err), self, allocations.into_iter()));
         }
 
+        let usage = self
+            .usage
+            .difference(ImageUsage::TRANSFER_SRC | ImageUsage::TRANSFER_DST);
+
+        let layout = if usage.intersects(ImageUsage::SAMPLED | ImageUsage::INPUT_ATTACHMENT)
+            && usage
+                .difference(ImageUsage::SAMPLED | ImageUsage::INPUT_ATTACHMENT)
+                .is_empty()
+        {
+            ImageLayout::ShaderReadOnlyOptimal
+        } else if usage.intersects(ImageUsage::COLOR_ATTACHMENT)
+            && usage.difference(ImageUsage::COLOR_ATTACHMENT).is_empty()
+        {
+            ImageLayout::ColorAttachmentOptimal
+        } else if usage.intersects(ImageUsage::DEPTH_STENCIL_ATTACHMENT)
+            && usage
+                .difference(ImageUsage::DEPTH_STENCIL_ATTACHMENT)
+                .is_empty()
+        {
+            ImageLayout::DepthStencilAttachmentOptimal
+        } else {
+            ImageLayout::General
+        };
+
         Ok(Image::from_raw(
             self,
             ImageMemory::Normal(allocations),
-            ImageLayout::General,
+            layout,
         ))
     }