Add general folding infrastructure.

[platform/upstream/SPIRV-Tools.git] / source / opt / ir_builder.h

diff --git a/source/opt/ir_builder.h b/source/opt/ir_builder.h
index b376f6d..9c153de 100644 (file)
--- a/source/opt/ir_builder.h
+++ b/source/opt/ir_builder.h
@@ -152,6 +152,22 @@ class InstructionBuilder {
     return AddInstruction(std::move(construct));
   }
 
+  ir::Instruction* AddCompositeExtract(
+      uint32_t type, uint32_t id_of_composite,
+      const std::vector<uint32_t>& index_list) {
+    std::vector<ir::Operand> operands;
+    operands.push_back({SPV_OPERAND_TYPE_ID, {id_of_composite}});
+
+    for (uint32_t index : index_list) {
+      operands.push_back({SPV_OPERAND_TYPE_LITERAL_INTEGER, {index}});
+    }
+
+    std::unique_ptr<ir::Instruction> new_inst(
+        new ir::Instruction(GetContext(), SpvOpCompositeExtract, type,
+                            GetContext()->TakeNextId(), operands));
+    return AddInstruction(std::move(new_inst));
+  }
+
   // Inserts the new instruction before the insertion point.
   ir::Instruction* AddInstruction(std::unique_ptr<ir::Instruction>&& insn) {
     ir::Instruction* insn_ptr = &*insert_before_.InsertBefore(std::move(insn));