vk: definition: Turn host/device AS handle into type safe union

Co-authored-by: Marijn Suijten <marijn@traverseresearch.nl>
ash-rs · Nov 24, 2020 · b216931 · b216931
1 parent 6bf3119
commit b216931
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/ash/src/vk/definitions.rs b/ash/src/vk/definitions.rs
@@ -45232,11 +45232,17 @@ pub struct TransformMatrixKHR {
 }
 #[repr(C)]
 #[derive(Copy, Clone)]
+pub union AccelerationStructureReferenceKHR {
+    pub device_handle: DeviceAddress,
+    pub host_handle: AccelerationStructureKHR,
+}
+#[repr(C)]
+#[derive(Copy, Clone)]
 pub struct AccelerationStructureInstanceKHR {
     pub transform: TransformMatrixKHR,
     pub instance_custom_index_and_mask: u32,
     pub instance_shader_binding_table_record_offset_and_flags: u32,
-    pub acceleration_structure_reference: u64,
+    pub acceleration_structure_reference: AccelerationStructureReferenceKHR,
 }
 #[repr(C)]
 #[derive(Copy, Clone, Debug)]

diff --git a/generator/src/lib.rs b/generator/src/lib.rs
@@ -1982,13 +1982,19 @@ pub fn generate_struct(
 
     if &_struct.name == "VkAccelerationStructureInstanceKHR" {
         return quote! {
+            #[repr(C)]
+            #[derive(Copy, Clone)]
+            pub union AccelerationStructureReferenceKHR {
+                pub device_handle: DeviceAddress,
+                pub host_handle: AccelerationStructureKHR,
+            }
             #[repr(C)]
             #[derive(Copy, Clone)]
             pub struct AccelerationStructureInstanceKHR {
                 pub transform: TransformMatrixKHR,
                 pub instance_custom_index_and_mask: u32,
                 pub instance_shader_binding_table_record_offset_and_flags: u32,
-                pub acceleration_structure_reference: u64,
+                pub acceleration_structure_reference: AccelerationStructureReferenceKHR,
             }
         };
     }