Feature/assembly label handling

src/MimaSim/MimaSim/Core/Parsing/Emiting/ByteArrayBuilder.cs

@@ -137,6 +137,52 @@ public void Dispose()
         _store.Dispose();
     }
 
+    public void ReplaceAt(int index, byte value)
+    {
+        ValidateIndex(index);
+        var buffer = _store.GetBuffer();
+        buffer[index] = value;
+    }
+
+    public void ReplaceAt(int index, bool value)
+    {
+        ValidateIndex(index);
+        ReplaceAt(index, value ? streamTrue : streamFalse);
+    }
+
+    public void ReplaceAt(int index, char value)
+    {
+        ValidateIndex(index);
+        ReplaceAt(index, (byte)value);
+    }
+
+    public void ReplaceAt(int index, short value)
+    {
+        ValidateIndex(index);
+        var bytes = BitConverter.GetBytes(value);
+        Array.Copy(bytes, 0, _store.GetBuffer(), index, bytes.Length);
+    }
+
+    public void ReplaceAt(int index, int value)
+    {
+        ValidateIndex(index);
+        var bytes = BitConverter.GetBytes(value);
+        Array.Copy(bytes, 0, _store.GetBuffer(), index, bytes.Length);
+    }
+
+    public void ReplaceAt(int index, long value)
+    {
+        ValidateIndex(index);
+        var bytes = BitConverter.GetBytes(value);
+        Array.Copy(bytes, 0, _store.GetBuffer(), index, bytes.Length);
+    }
+
+    private void ValidateIndex(int index)
+    {
+        if (index < 0 || index >= Length)
+            throw new ArgumentOutOfRangeException(nameof(index), "Index is out of range.");
+    }
+
     public bool GetBool()
     {
         return _store.ReadByte() == streamTrue;

src/MimaSim/MimaSim/Core/Parsing/Emiting/ByteCodeEmitter.cs

@@ -6,9 +6,8 @@ namespace MimaSim.Core.Parsing.Emiting;
 public class ByteCodeEmitter
 {
     private readonly ByteArrayBuilder _builder = new();
-
-    private readonly Dictionary<string, byte> _labels = new();
-    private int _labelCount = 0;
+    private readonly Dictionary<string, short> _labels = new();
+    private readonly List<short> _labelReferences = new();
     public int Position => _builder.Length;
 
     public void Append(byte[] raw)
@@ -18,7 +17,7 @@ public void Append(byte[] raw)
 
     public Label DefineLabel()
     {
-        return new Label(_labelCount++);
+        return new Label(_labels.Count);
     }
 
     public void EmitInstruction(OpCodes opcode)
@@ -61,19 +60,37 @@ public void EmitRegister(Registers reg)
         _builder.Append((byte)reg);
     }
 
-    public byte GetLabel(string name)
+    public void AddLabelReference()
     {
-        return _labels[name];
+        _labelReferences.Add((short)_builder.Length);
+        _builder.Append((short)0);
     }
 
-    public void MarkLabel(Label label)
+    private void ReplaceLabelReferences()
     {
-        _labels.Add("L" + label.LabelNum, (byte)_builder.Length);
+        foreach (var kvp in _labels)
+        {
+            int index = 0;
+            while (index < _labelReferences.Count)
+            {
+                if (_labelReferences[index] == kvp.Value)
+                {
+                    _builder.ReplaceAt(index, kvp.Value);
+                }
+                index++;
+            }
+        }
+
+        _labelReferences.Clear();
     }
 
-    public byte[] ToArray() => _builder.ToArray();
+    public byte[] ToArray()
+    {
+        ReplaceLabelReferences(); // Ensure labels are replaced before converting to array
+        return _builder.ToArray();
+    }
 
-    internal void CreateLabel(string name)
+    public void CreateLabel(string name)
     {
         _labels.Add(name, (byte)_builder.Length);
     }

src/MimaSim/MimaSim/MIMA/Instructions/LoadInstruction.cs

@@ -18,6 +18,6 @@ public bool Invoke(CPU cpu)
 
     public void Dissassemble(StringBuilder builder, Disassembler disassembler)
     {
-        builder.AppendLine($"load {disassembler.Fetch16()}");
+        builder.AppendLine($"load 0x{disassembler.Fetch16():X2}");
     }
 }

src/MimaSim/MimaSim/MIMA/Instructions/Move/MovMemMemInstruction.cs

@@ -25,6 +25,6 @@ public bool Invoke(CPU cpu)
 
     public void Dissassemble(StringBuilder builder, Disassembler disassembler)
     {
-        builder.AppendLine($"mov {disassembler.Fetch16()}, {disassembler.Fetch16()}");
+        builder.AppendLine($"mov 0x{disassembler.Fetch16():X2}, 0x{disassembler.Fetch16():X2}");
     }
 }

src/MimaSim/MimaSim/MIMA/Instructions/Move/MovMemRegInstruction.cs

@@ -24,6 +24,6 @@ public bool Invoke(CPU cpu)
 
     public void Dissassemble(StringBuilder builder, Disassembler disassembler)
     {
-        builder.AppendLine($"mov {disassembler.Fetch16()}, {disassembler.FetchRegister()}");
+        builder.AppendLine($"mov 0x{disassembler.Fetch16():X2}, {disassembler.FetchRegister()}");
     }
 }

src/MimaSim/MimaSim/MIMA/Instructions/Move/MovRegMemInstruction.cs

@@ -23,6 +23,6 @@ public bool Invoke(CPU cpu)
 
     public void Dissassemble(StringBuilder builder, Disassembler disassembler)
     {
-        builder.AppendLine($"mov {disassembler.FetchRegister()}, {disassembler.Fetch16()}");
+        builder.AppendLine($"mov {disassembler.FetchRegister()}, 0x{disassembler.Fetch16():X2}");
     }
 }

src/MimaSim/MimaSim/MIMA/Parsing/Parsers/Assembler/AssemblyVisitor.cs

@@ -19,7 +19,25 @@ public AssemblyVisitor()
         For<InstructionNode>(Visit);
         For<LiteralNode>(VisitLiteral, node => node.Value is ulong);
         For<LiteralNode>(VisitRegister, node => node.Value is Registers);
-        For<PrefixOperatorNode>(VisitAddress, op => op.Operator == "&");
+        For<PrefixOperatorNode>(VisitAddress, op => op.Tag == "address");
+        For<PrefixOperatorNode>(VisitLabelRef, op => op.Tag == "labelref");
+        For<PostfixOperatorNode>(VisitLabel, op => op.Tag == "label");
+    }
+
+    private void VisitLabel(PostfixOperatorNode labelDef)
+    {
+        if (labelDef.Expr is NameNode nameNode)
+        {
+            _emitter.CreateLabel(nameNode.Token.Text.ToString());
+        }
+    }
+
+    private void VisitLabelRef(PrefixOperatorNode labelRef)
+    {
+        if (labelRef.Expr is NameNode nameNode)
+        {
+            //todo: emit 0 address and adjust later in GetRaw()
+        }
     }
 
     private void VisitAddress(PrefixOperatorNode obj)

src/MimaSim/MimaSim/MIMA/Visitors/HighParserVisitor.cs

@@ -261,7 +261,7 @@ private void VisitIfStatement(CallNode call)
     private void VisitLoopStatement(CallNode call)
     {
         var label = _emitter.DefineLabel();
-        _emitter.MarkLabel(label);
+       // _emitter.MarkLabel(label);
 
         Visit((CallNode)call.Args.First());