Support emitting OpLine directive.

Facilitates easier mapping from source language to cross-compiled output
in tooling.

1 files changed, 26 insertions, 2 deletions

diff --git a/spirv_parser.cpp b/spirv_parser.cpp
index 1c0a830f..c75fa1b6 100644
--- a/spirv_parser.cpp
+++ b/spirv_parser.cpp
@@ -162,12 +162,16 @@ void Parser::parse(const Instruction &instruction)
 	case OpSourceContinued:
 	case OpSourceExtension:
 	case OpNop:
-	case OpLine:
 	case OpNoLine:
-	case OpString:
 	case OpModuleProcessed:
 		break;
 
+	case OpString:
+	{
+		set<SPIRString>(ops[0], extract_string(ir.spirv, instruction.offset + 1));
+		break;
+	}
+
 	case OpMemoryModel:
 		ir.addressing_model = static_cast<AddressingModel>(ops[0]);
 		ir.memory_model = static_cast<MemoryModel>(ops[1]);
@@ -1030,6 +1034,26 @@ void Parser::parse(const Instruction &instruction)
 		break;
 	}
 
+	case OpLine:
+	{
+		// OpLine might come at global scope, but we don't care about those since they will not be declared in any
+		// meaningful correct order.
+		// For OpLine statements which
+		if (current_block)
+			current_block->ops.push_back(instruction);
+
+		if (current_function) // Line directives may arrive before first OpLabel.
+		{
+			// Store the first one we find and emit it before creating the function prototype.
+			if (current_function->entry_line.file_id == 0)
+			{
+				current_function->entry_line.file_id = ops[0];
+				current_function->entry_line.line_literal = ops[1];
+			}
+		}
+		break;
+	}
+
 	// Actual opcodes.
 	default:
 	{