Create object section encoder

2026-07-27 13:11:59 -07:00 · 2024-02-06 13:15:05 -06:00
parent 3928966f81
commit 0ecca8cbbf
2 changed files with 81 additions and 13 deletions
@@ -0,0 +1,75 @@
+using Microsoft.Iris.Asm.Models;
+using Microsoft.Iris.Markup;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+
+namespace Microsoft.Iris.Asm;
+
+public class ObjectSection
+{
+    readonly IEnumerable<Instruction> _instructions;
+    readonly MarkupLoadResult _loadResult;
+
+    public ObjectSection(IEnumerable<Instruction> instructions, MarkupLoadResult loadResult)
+    {
+        _instructions = instructions;
+        _loadResult = loadResult;
+    }
+
+    public ObjectSection(IEnumerable<IBodyItem> body, MarkupLoadResult loadResult)
+        : this(body.OfType<Instruction>(), loadResult)
+    {
+    }
+
+    public ObjectSection(Program program, MarkupLoadResult loadResult)
+        : this(program.Body, loadResult)
+    {
+    }
+
+    public ByteCodeReader Encode()
+    {
+        ByteCodeWriter writer = new();
+
+        foreach (var instruction in _instructions)
+        {
+            var opCode = instruction.OpCode;
+            writer.WriteByte(opCode);
+
+            foreach (var operand in instruction.Operands)
+            {
+                switch (operand.Value)
+                {
+                    case OperationType opType:
+                        writer.WriteByte((byte)opType);
+                        break;
+                    case byte b:
+                        writer.WriteByte(b);
+                        break;
+                    case ushort uint16:
+                        writer.WriteUInt16(uint16);
+                        break;
+                    case uint uint32:
+                        writer.WriteUInt32(uint32);
+                        break;
+                    case int int32:
+                        writer.WriteInt32(int32);
+                        break;
+
+                    default:
+                        if (opCode == OpCode.ConstructFromBinary)
+                        {
+                            ushort cbinTypeIndex = (UInt16)instruction.Operands.First().Value;
+                            TypeSchema cbinTypeSchema = _loadResult.ImportTables.TypeImports[cbinTypeIndex];
+                            cbinTypeSchema.EncodeBinary(writer, operand.Value);
+                            break;
+                        }
+
+                        throw new InvalidOperationException($"Unexpected operand '{operand.Value}' of type '{operand.Value.GetType()}'");
+                }
+            }
+        }
+
+        return writer.CreateReader();
+    }
+}