From 7308efee5329de867cd794db263495f5468529b9 Mon Sep 17 00:00:00 2001
From: Lily Haverlag <flannyha@gmail.com>
Date: Mon, 8 Jan 2024 13:34:44 +0100
Subject: [PATCH] Add functions to IRObject and IRCompiler

---
 src/lib.rs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index 1f9e992..e972db7 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -57,6 +57,8 @@ struct IRCompilerFn<'lib> {
     set_vertex_viewport_index_id: libloading::Symbol<'lib, unsafe extern "C" fn() -> ()>,
     set_int_rt_mask: libloading::Symbol<'lib, unsafe extern "C" fn() -> ()>,
     ignore_root_signature: libloading::Symbol<'lib, unsafe extern "C" fn() -> ()>,
+    set_ray_tracing_pipeline_arguments: libloading::Symbol<'lib, unsafe extern "C" fn() -> ()>,
+    set_hitgroup_type: libloading::Symbol<'lib, unsafe extern "C" fn() -> ()>,
 }
 
 #[repr(u32)]
@@ -247,6 +249,7 @@ struct IRObjectFn<'lib> {
     >,
     get_type: libloading::Symbol<'lib, unsafe extern "C" fn(*mut IRObjectOpaque) -> IRObjectType>,
     serialize: libloading::Symbol<'lib, unsafe extern "C" fn() -> ()>,
+    gather_raytracing_intrinsics: libloading::Symbol<'_, _>,
 }
 
 pub struct IRObject<'lib> {
@@ -269,6 +272,7 @@ impl<'lib> IRObject<'lib> {
             let funcs = IRObjectFn {
                 create_from_dxil: lib.get(b"IRObjectCreateFromDXIL")?,
                 destroy: lib.get(b"IRObjectDestroy")?,
+                gather_raytracing_intrinsics: lib.get(b"IRObjectGatherRaytracingIntrinsics")?,
                 get_metal_ir_shader_stage: lib.get(b"IRObjectGetMetalIRShaderStage")?,
                 get_metal_lib_binary: lib.get(b"IRObjectGetMetalLibBinary")?,
                 get_reflection: lib.get(b"IRObjectGetReflection")?,
@@ -686,10 +690,13 @@ impl<'lib> IRCompiler<'lib> {
                 set_global_root_signature: lib.get(b"IRCompilerSetGlobalRootSignature")?,
                 set_has_geometry_shader: lib.get(b"IRCompilerSetHasGeometryShader")?,
                 set_hitgroup_arguments: lib.get(b"IRCompilerSetHitgroupArguments")?,
+                set_hitgroup_type: lib.get(b"IRCompilerSetHitgroupType")?,
                 set_input_topology: lib.get(b"IRCompilerSetInputTopology")?,
                 set_local_root_signature: lib.get(b"IRCompilerSetLocalRootSignature")?,
                 set_minimum_deployment_target: lib.get(b"IRCompilerSetMinimumDeploymentTarget")?,
                 set_minimum_gpu_family: lib.get(b"IRCompilerSetMinimumGPUFamily")?,
+                set_ray_tracing_pipeline_arguments: lib
+                    .get(b"IRCompilerSetRayTracingPipelineArguments")?,
                 set_shared_rt_arguments: lib.get(b"IRCompilerSetSharedRTArguments")?,
                 set_stage_in_generation_mode: lib.get(b"IRCompilerSetStageInGenerationMode")?,
                 set_stream_out_enabled: lib.get(b"IRCompilerSetStreamOutEnabled")?,