From 00b3e48622b4685014c5382b50c4f0bd6bd7b126 Mon Sep 17 00:00:00 2001
From: Marijn Suijten <marijn@traverseresearch.nl>
Date: Wed, 16 Feb 2022 20:51:39 +0100
Subject: [PATCH] extensions/khr: Add VK_KHR_external_semaphore_win32

---
 Changelog.md                                  |  1 +
 .../khr/external_semaphore_win32.rs           | 56 +++++++++++++++++++
 ash/src/extensions/khr/mod.rs                 |  2 +
 3 files changed, 59 insertions(+)
 create mode 100644 ash/src/extensions/khr/external_semaphore_win32.rs

diff --git a/Changelog.md b/Changelog.md
index 45d070f7d..1ee65cdb5 100644
--- a/Changelog.md
+++ b/Changelog.md
@@ -13,6 +13,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 
+- Added `VK_KHR_external_semaphore_win32` device extension (#581)
 - Added `VK_KHR_external_memory_win32` device extension (#579)
 - Added `VK_EXT_extended_dynamic_state2` device extension (#572)
 - Added `VK_KHR_copy_commands2` device extension (#571)
diff --git a/ash/src/extensions/khr/external_semaphore_win32.rs b/ash/src/extensions/khr/external_semaphore_win32.rs
new file mode 100644
index 000000000..0925a8175
--- /dev/null
+++ b/ash/src/extensions/khr/external_semaphore_win32.rs
@@ -0,0 +1,56 @@
+use crate::prelude::*;
+use crate::vk;
+use crate::{Device, Instance};
+use std::ffi::CStr;
+use std::mem;
+use std::ptr;
+
+/// <https://www.khronos.org/registry/vulkan/specs/1.3-extensions/man/html/VK_KHR_external_semaphore_win32.html>
+#[derive(Clone)]
+pub struct ExternalSemaphoreWin32 {
+    handle: vk::Device,
+    fp: vk::KhrExternalSemaphoreWin32Fn,
+}
+
+impl ExternalSemaphoreWin32 {
+    pub fn new(instance: &Instance, device: &Device) -> Self {
+        let handle = device.handle();
+        let fp = vk::KhrExternalSemaphoreWin32Fn::load(|name| unsafe {
+            mem::transmute(instance.get_device_proc_addr(handle, name.as_ptr()))
+        });
+        Self { handle, fp }
+    }
+
+    /// <https://www.khronos.org/registry/vulkan/specs/1.3-extensions/man/html/vkImportSemaphoreWin32HandleKHR.html>
+    pub unsafe fn import_semaphore_win32_handle(
+        &self,
+        import_info: &vk::ImportSemaphoreWin32HandleInfoKHR,
+    ) -> VkResult<()> {
+        self.fp
+            .import_semaphore_win32_handle_khr(self.handle, import_info)
+            .result()
+    }
+
+    /// <https://www.khronos.org/registry/vulkan/specs/1.3-extensions/man/html/vkGetSemaphoreWin32HandleKHR.html>
+    pub unsafe fn get_semaphore_win32_handle(
+        &self,
+        get_info: &vk::SemaphoreGetWin32HandleInfoKHR,
+    ) -> VkResult<vk::HANDLE> {
+        let mut handle = ptr::null_mut();
+        self.fp
+            .get_semaphore_win32_handle_khr(self.handle, get_info, &mut handle)
+            .result_with_success(handle)
+    }
+
+    pub fn name() -> &'static CStr {
+        vk::KhrExternalSemaphoreWin32Fn::name()
+    }
+
+    pub fn fp(&self) -> &vk::KhrExternalSemaphoreWin32Fn {
+        &self.fp
+    }
+
+    pub fn device(&self) -> vk::Device {
+        self.handle
+    }
+}
diff --git a/ash/src/extensions/khr/mod.rs b/ash/src/extensions/khr/mod.rs
index ecba12258..64963ecce 100644
--- a/ash/src/extensions/khr/mod.rs
+++ b/ash/src/extensions/khr/mod.rs
@@ -12,6 +12,7 @@ pub use self::external_fence_fd::ExternalFenceFd;
 pub use self::external_memory_fd::ExternalMemoryFd;
 pub use self::external_memory_win32::ExternalMemoryWin32;
 pub use self::external_semaphore_fd::ExternalSemaphoreFd;
+pub use self::external_semaphore_win32::ExternalSemaphoreWin32;
 pub use self::get_memory_requirements2::GetMemoryRequirements2;
 pub use self::get_physical_device_properties2::GetPhysicalDeviceProperties2;
 pub use self::get_surface_capabilities2::GetSurfaceCapabilities2;
@@ -45,6 +46,7 @@ mod external_fence_fd;
 mod external_memory_fd;
 mod external_memory_win32;
 mod external_semaphore_fd;
+mod external_semaphore_win32;
 mod get_memory_requirements2;
 mod get_physical_device_properties2;
 mod get_surface_capabilities2;