From b2abd40811239e56099a38b54e380fd18daa64e5 Mon Sep 17 00:00:00 2001
From: cetra3 <cetra3@hotmail.com>
Date: Wed, 30 Jul 2025 11:14:20 +0930
Subject: [PATCH] Add try_acquire_frame_method

---
 blade-graphics/src/vulkan/surface.rs | 46 ++++++++++++++++------------
 1 file changed, 26 insertions(+), 20 deletions(-)

diff --git a/blade-graphics/src/vulkan/surface.rs b/blade-graphics/src/vulkan/surface.rs
index c53d1090..0863741c 100644
--- a/blade-graphics/src/vulkan/surface.rs
+++ b/blade-graphics/src/vulkan/surface.rs
@@ -19,26 +19,8 @@ impl super::Surface {
     }
 
     pub fn acquire_frame(&mut self) -> super::Frame {
-        let acquire_semaphore = self.next_semaphore;
-        match unsafe {
-            self.device.acquire_next_image(
-                self.swapchain.raw,
-                !0,
-                acquire_semaphore,
-                vk::Fence::null(),
-            )
-        } {
-            Ok((index, _suboptimal)) => {
-                self.next_semaphore = mem::replace(
-                    &mut self.frames[index as usize].acquire_semaphore,
-                    acquire_semaphore,
-                );
-                super::Frame {
-                    internal: self.frames[index as usize],
-                    swapchain: self.swapchain,
-                    image_index: Some(index),
-                }
-            }
+        match self.vulkan_try_acquire_frame() {
+            Ok(frame) => frame,
             Err(vk::Result::ERROR_OUT_OF_DATE_KHR) => {
                 log::warn!("Acquire failed because the surface is out of date");
                 super::Frame {
@@ -50,6 +32,30 @@ impl super::Surface {
             Err(other) => panic!("Aquire image error {}", other),
         }
     }
+
+    pub fn vulkan_try_acquire_frame(&mut self) -> Result<super::Frame, vk::Result> {
+        let acquire_semaphore = self.next_semaphore;
+
+        let (index, _suboptimal) = unsafe {
+            self.device.acquire_next_image(
+                self.swapchain.raw,
+                !0,
+                acquire_semaphore,
+                vk::Fence::null(),
+            )
+        }?;
+
+        self.next_semaphore = mem::replace(
+            &mut self.frames[index as usize].acquire_semaphore,
+            acquire_semaphore,
+        );
+
+        Ok(super::Frame {
+            internal: self.frames[index as usize],
+            swapchain: self.swapchain,
+            image_index: Some(index),
+        })
+    }
 }
 
 impl super::Context {