From 9130e0d8dfdf7ea550c04960c1fadd38aabfa6fb Mon Sep 17 00:00:00 2001
From: hodasemi <michaelh.95@t-online.de>
Date: Mon, 4 Dec 2023 12:41:10 +0100
Subject: [PATCH] Improve submit info

---
 vulkan-rs/src/queue.rs | 27 +++++++++++++--------------
 1 file changed, 13 insertions(+), 14 deletions(-)

diff --git a/vulkan-rs/src/queue.rs b/vulkan-rs/src/queue.rs
index 3efaf7f..e309c08 100644
--- a/vulkan-rs/src/queue.rs
+++ b/vulkan-rs/src/queue.rs
@@ -207,19 +207,20 @@ pub struct SubmitInfo {
 }
 
 impl SubmitInfo {
-    pub fn add_wait_semaphore(mut self, wait_semaphore: impl VkHandle<VkSemaphore>) -> Self {
-        self.wait_semaphores.push(wait_semaphore.vk_handle());
-
-        self
+    pub fn add_wait_semaphore(
+        self,
+        wait_semaphore: impl VkHandle<VkSemaphore>,
+        wait_stage: impl Into<VkPipelineStageFlagBits>,
+    ) -> Self {
+        self.add_wait_semaphore_vk(wait_semaphore.vk_handle(), wait_stage)
     }
 
-    pub fn add_wait_semaphore_vk(mut self, wait_semaphore: VkSemaphore) -> Self {
+    pub fn add_wait_semaphore_vk(
+        mut self,
+        wait_semaphore: VkSemaphore,
+        wait_stage: impl Into<VkPipelineStageFlagBits>,
+    ) -> Self {
         self.wait_semaphores.push(wait_semaphore);
-
-        self
-    }
-
-    pub fn add_wait_stage(mut self, wait_stage: impl Into<VkPipelineStageFlagBits>) -> Self {
         self.wait_stages.push(wait_stage.into());
 
         self
@@ -231,10 +232,8 @@ impl SubmitInfo {
         self
     }
 
-    pub fn add_signal_semaphore(mut self, signal_semaphore: impl VkHandle<VkSemaphore>) -> Self {
-        self.signal_semaphores.push(signal_semaphore.vk_handle());
-
-        self
+    pub fn add_signal_semaphore(self, signal_semaphore: impl VkHandle<VkSemaphore>) -> Self {
+        self.add_signal_semaphore_vk(signal_semaphore.vk_handle())
     }
 
     pub fn add_signal_semaphore_vk(mut self, signal_semaphore: VkSemaphore) -> Self {