Add support for generating OpTypeAccelerationStructureKHR

autogen/src/main.rs

@@ -35,6 +35,23 @@ fn write_formatted(path: &PathBuf, contents: impl ToString) {
     };
 }
 
+/// Maps some reserved instructions in the spec into their expected class in
+/// order to generate the proper methods for those instructions.
+fn map_reserved_instructions(grammar: &mut structs::Grammar) {
+    for instruction in grammar
+        .instructions
+        .iter_mut()
+        .filter(|i| i.class == Some(structs::Class::Reserved))
+    {
+        match &*instruction.opname {
+            | "OpTypeAccelerationStructureKHR"
+            | "OpTypeRayQueryKHR"
+                => instruction.class = Some(structs::Class::Type),
+            _ => {}
+        }
+    }
+}
+
 fn main() {
     // Path to the SPIR-V core grammar file.
     let env_var = env::var("CARGO_MANIFEST_DIR").unwrap();
@@ -52,7 +69,12 @@ fn main() {
         let mut file = fs::File::open(path).unwrap();
         file.read_to_string(&mut contents).unwrap();
     }
-    let grammar: structs::Grammar = serde_json::from_str(&contents).unwrap();
+
+    let grammar: structs::Grammar = {
+        let mut original = serde_json::from_str(&contents).unwrap();
+        map_reserved_instructions(&mut original);
+        original
+    };
 
     // For GLSLstd450 extended instruction set.
     {

rspirv/dr/build/autogen_type.rs

@@ -345,4 +345,29 @@ impl Builder {
             new_id
         }
     }
+    #[doc = "Appends an OpTypeRayQueryKHR instruction and returns the result id, or return the existing id if the instruction was already present."]
+    pub fn type_ray_query_khr(&mut self) -> spirv::Word {
+        let mut inst = dr::Instruction::new(spirv::Op::TypeRayQueryKHR, None, None, vec![]);
+        if let Some(id) = self.dedup_insert_type(&inst) {
+            id
+        } else {
+            let new_id = self.id();
+            inst.result_id = Some(new_id);
+            self.module.types_global_values.push(inst);
+            new_id
+        }
+    }
+    #[doc = "Appends an OpTypeAccelerationStructureKHR instruction and returns the result id, or return the existing id if the instruction was already present."]
+    pub fn type_acceleration_structure_khr(&mut self) -> spirv::Word {
+        let mut inst =
+            dr::Instruction::new(spirv::Op::TypeAccelerationStructureKHR, None, None, vec![]);
+        if let Some(id) = self.dedup_insert_type(&inst) {
+            id
+        } else {
+            let new_id = self.id();
+            inst.result_id = Some(new_id);
+            self.module.types_global_values.push(inst);
+            new_id
+        }
+    }
 }

rspirv/lift/autogen_context.rs

@@ -5338,7 +5338,6 @@ impl LiftContext {
                 })
                 .ok_or(OperandError::Missing)?,
             }),
-            4472u32 => Ok(ops::Op::TypeRayQueryKHR),
             4473u32 => Ok(ops::Op::RayQueryInitializeKHR {
                 ray_query: (match operands.next() {
                     Some(&dr::Operand::IdRef(ref value)) => Some(*value),
@@ -5803,7 +5802,6 @@ impl LiftContext {
                 .ok_or(OperandError::Missing)?,
             }),
             5341u32 => Ok(ops::Op::TypeAccelerationStructureNV),
-            5341u32 => Ok(ops::Op::TypeAccelerationStructureKHR),
             5344u32 => Ok(ops::Op::ExecuteCallableNV {
                 sbt_index: (match operands.next() {
                     Some(&dr::Operand::IdRef(ref value)) => Some(*value),
@@ -8577,6 +8575,8 @@ impl LiftContext {
             }),
             322u32 => Ok(Type::PipeStorage),
             327u32 => Ok(Type::NamedBarrier),
+            4472u32 => Ok(Type::RayQueryKHR),
+            5341u32 => Ok(Type::AccelerationStructureKHR),
             _ => Err(InstructionError::WrongOpcode),
         }
     }

rspirv/sr/autogen_ops.rs

@@ -1363,7 +1363,6 @@ pub enum Op {
     ConvertUToAccelerationStructureKHR {
         accel: spirv::Word,
     },
-    TypeRayQueryKHR,
     RayQueryInitializeKHR {
         ray_query: spirv::Word,
         accel: spirv::Word,
@@ -1481,7 +1480,6 @@ pub enum Op {
         payload_id: spirv::Word,
     },
     TypeAccelerationStructureNV,
-    TypeAccelerationStructureKHR,
     ExecuteCallableNV {
         sbt_index: spirv::Word,
         callable_data_id: spirv::Word,

rspirv/sr/autogen_types.rs

@@ -69,4 +69,6 @@ pub enum Type {
     },
     PipeStorage,
     NamedBarrier,
+    RayQueryKHR,
+    AccelerationStructureKHR,
 }