asdf-overlay-vulkan-layer 1.2.2

Asdf Overlay vulkan layer
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

use ash::vk::{self, Handle};
use once_cell::sync::Lazy;
use tracing::{debug, trace};

use crate::{instance::DISPATCH_TABLE, map::IntDashMap};

/// vkSurfaceKHR to HWND mapping table
static SURFACE_MAP: Lazy<IntDashMap<u64, u32>> = Lazy::new(IntDashMap::default);

/// Get the HWND associated with a given [`vk::SurfaceKHR`].`
pub fn get_surface_hwnd(surface: vk::SurfaceKHR) -> Option<u32> {
    SURFACE_MAP.get(&surface.as_raw()).map(|hwnd| *hwnd)
}

/// Layer `vkCreateWin32SurfaceKHR` implementation
pub(super) extern "system" fn create_win32_surface(
    instance: vk::Instance,
    create_info: *const vk::Win32SurfaceCreateInfoKHR,
    callback: *const vk::AllocationCallbacks,
    surface: *mut vk::SurfaceKHR,
) -> vk::Result {
    trace!("vkCreateWin32SurfaceKHR called");

    let res = unsafe {
        (DISPATCH_TABLE
            .get(&instance.as_raw())
            .unwrap()
            .create_win32_surface
            .unwrap())(instance, create_info, callback, surface)
    };
    if res != vk::Result::SUCCESS {
        return res;
    }

    let surface = unsafe { *surface }.as_raw();
    let hwnd = unsafe { *create_info }.hwnd as u32;
    debug!("registering surface: {surface} -> hwnd: {hwnd}");
    SURFACE_MAP.insert(surface, hwnd);

    vk::Result::SUCCESS
}

/// Layer `vkDestroySurfaceKHR` implementation
pub(super) extern "system" fn destroy_surface(
    instance: vk::Instance,
    surface: vk::SurfaceKHR,
    callback: *const vk::AllocationCallbacks,
) {
    trace!("vkDestroySurfaceKHR called");
    unsafe {
        (DISPATCH_TABLE
            .get(&instance.as_raw())
            .unwrap()
            .destroy_surface
            .unwrap())(instance, surface, callback);
    }

    SURFACE_MAP.remove(&surface.as_raw());
}