Add support for mapping reserved instructions

Author: XAMPPRocky

autogen/src/main.rs

@@ -35,6 +35,21 @@ 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" => 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 +67,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,17 @@ impl Builder {
             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

@@ -5803,7 +5803,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 +8576,7 @@ impl LiftContext {
             }),
             322u32 => Ok(Type::PipeStorage),
             327u32 => Ok(Type::NamedBarrier),
+            5341u32 => Ok(Type::AccelerationStructureKHR),
             _ => Err(InstructionError::WrongOpcode),
         }
     }

rspirv/sr/autogen_ops.rs

@@ -1481,7 +1481,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,5 @@ pub enum Type {
     },
     PipeStorage,
     NamedBarrier,
+    AccelerationStructureKHR,
 }