nub31/nub-lang
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
nub31
2025-07-03 22:25:18 +02:00
parent 4ca1c6812d
commit 6fd905a921
2 changed files with 111 additions and 132 deletions
Download Patch File Download Diff File Expand all files Collapse all files

195
src/compiler/Generation/QBE/QBEGenerator.cs
View File

@@ -15,7 +15,7 @@ public static class QBEGenerator
private static BoundSyntaxTree _syntaxTree = null!; private static BoundSyntaxTree _syntaxTree = null!;
private static BoundDefinitionTable _definitionTable = null!; private static BoundDefinitionTable _definitionTable = null!;
private static QBEWriter _writer = new(); private static QBEWriter _writer = null!;
private static List<CStringLiteral> _cStringLiterals = []; private static List<CStringLiteral> _cStringLiterals = [];
private static List<StringLiteral> _stringLiterals = []; private static List<StringLiteral> _stringLiterals = [];
private static Stack<string> _breakLabels = []; private static Stack<string> _breakLabels = [];
@@ -35,7 +35,7 @@ public static class QBEGenerator
_syntaxTree = syntaxTree; _syntaxTree = syntaxTree;
_definitionTable = definitionTable; _definitionTable = definitionTable;
_writer = new QBEWriter(); _writer = new QBEWriter(file);
_cStringLiterals = []; _cStringLiterals = [];
_stringLiterals = []; _stringLiterals = [];
@@ -51,25 +51,19 @@ public static class QBEGenerator
_stringLiteralIndex = 0; _stringLiteralIndex = 0;
_codeIsReachable = true; _codeIsReachable = true;
_writer.WriteLine($"dbgfile \"{file}\"");
_writer.WriteLine();
foreach (var structDef in _definitionTable.GetStructs()) foreach (var structDef in _definitionTable.GetStructs())
{ {
EmitStructDefinition(structDef); EmitStructDefinition(structDef);
_writer.WriteLine();
} }
foreach (var funcDef in _syntaxTree.Definitions.OfType<BoundLocalFuncDefinitionNode>()) foreach (var funcDef in _syntaxTree.Definitions.OfType<BoundLocalFuncDefinitionNode>())
{ {
EmitFuncDefinition(funcDef, FuncName(funcDef), funcDef.Parameters, funcDef.ReturnType, funcDef.Body, funcDef.Exported); EmitFuncDefinition(funcDef, FuncName(funcDef), funcDef.Parameters, funcDef.ReturnType, funcDef.Body, funcDef.Exported);
_writer.WriteLine();
} }
while (_anonymousFunctions.TryDequeue(out var anon)) while (_anonymousFunctions.TryDequeue(out var anon))
{ {
EmitFuncDefinition(anon.Func, anon.Name, anon.Func.Parameters, anon.Func.ReturnType, anon.Func.Body, false); EmitFuncDefinition(anon.Func, anon.Name, anon.Func.Parameters, anon.Func.ReturnType, anon.Func.Body, false);
_writer.WriteLine();
} }
foreach (var cStringLiteral in _cStringLiterals) foreach (var cStringLiteral in _cStringLiterals)
@@ -141,7 +135,7 @@ public static class QBEGenerator
_ => throw new NotSupportedException($"'{type}' type cannot be used in store instructions") _ => throw new NotSupportedException($"'{type}' type cannot be used in store instructions")
}; };
_writer.WriteLine($" {store} {value}, {destination}"); _writer.Code($"{store} {value}, {destination}");
} }
private static Val EmitLoad(NubType type, string from) private static Val EmitLoad(NubType type, string from)
@@ -165,38 +159,38 @@ public static class QBEGenerator
_ => throw new NotSupportedException($"'{type}' type cannot be used in load instructions") _ => throw new NotSupportedException($"'{type}' type cannot be used in load instructions")
}; };
_writer.WriteLine($" {into} {QBEAssign(type)} {load} {from}"); _writer.Code($"{into} {QBEAssign(type)} {load} {from}");
return new Val(into, type, ValKind.Direct); return new Val(into, type, ValKind.Direct);
} }
private static void EmitMemcpy(string source, string destination, string length) private static void EmitMemcpy(string source, string destination, string length)
{ {
_writer.WriteLine($" call $nub_memcpy(l {source}, l {destination}, l {length})"); _writer.Code($"call $nub_memcpy(l {source}, l {destination}, l {length})");
} }
private static string EmitArraySizeInBytes(NubArrayType type, string array) private static string EmitArraySizeInBytes(NubArrayType type, string array)
{ {
var size = VarName(); var size = VarName();
_writer.WriteLine($" {size} =l loadl {array}"); _writer.Code($"{size} =l loadl {array}");
_writer.WriteLine($" {size} =l mul {size}, {SizeOf(type.ElementType)}"); _writer.Code($"{size} =l mul {size}, {SizeOf(type.ElementType)}");
_writer.WriteLine($" {size} =l add {size}, 8"); _writer.Code($"{size} =l add {size}, 8");
return size; return size;
} }
private static string EmitCStringSizeInBytes(string cstring) private static string EmitCStringSizeInBytes(string cstring)
{ {
var size = VarName(); var size = VarName();
_writer.WriteLine($" {size} =l call $nub_cstring_length(l {cstring})"); _writer.Code($"{size} =l call $nub_cstring_length(l {cstring})");
_writer.WriteLine($" {size} =l add {size}, 1"); _writer.Code($"{size} =l add {size}, 1");
return size; return size;
} }
private static string EmitStringSizeInBytes(string nubstring) private static string EmitStringSizeInBytes(string nubstring)
{ {
var size = VarName(); var size = VarName();
_writer.WriteLine($" {size} =l loadl {nubstring}"); _writer.Code($"{size} =l loadl {nubstring}");
_writer.WriteLine($" {size} =l add {size}, 8"); _writer.Code($"{size} =l add {size}, 8");
return size; return size;
} }
@@ -254,7 +248,7 @@ public static class QBEGenerator
case NubStringType: case NubStringType:
{ {
var buffer = VarName(); var buffer = VarName();
_writer.WriteLine($" {buffer} =l alloc8 {size}"); _writer.Code($"{buffer} =l alloc8 {size}");
EmitMemcpy(value, buffer, size); EmitMemcpy(value, buffer, size);
EmitStore(source.Type, buffer, destinationPointer); EmitStore(source.Type, buffer, destinationPointer);
return; return;
@@ -324,7 +318,7 @@ public static class QBEGenerator
_ => throw new ArgumentOutOfRangeException(nameof(complexType)) _ => throw new ArgumentOutOfRangeException(nameof(complexType))
}; };
_writer.WriteLine($" {destination} =l alloc8 {size}"); _writer.Code($"{destination} =l alloc8 {size}");
EmitMemcpy(value, destination, size); EmitMemcpy(value, destination, size);
return destination; return destination;
} }
@@ -462,8 +456,6 @@ public static class QBEGenerator
private static void EmitFuncDefinition(BoundNode debugNode, string name, List<BoundFuncParameter> parameters, NubType returnType, BoundBlockNode body, bool exported) private static void EmitFuncDefinition(BoundNode debugNode, string name, List<BoundFuncParameter> parameters, NubType returnType, BoundBlockNode body, bool exported)
{ {
_writer.ResetLineTracker();
_variables.Clear(); _variables.Clear();
_variableScopes.Clear(); _variableScopes.Clear();
@@ -524,42 +516,11 @@ public static class QBEGenerator
return $"{qbeType} %{parameter.Name}"; return $"{qbeType} %{parameter.Name}";
}); });
builder.Append($"({string.Join(", ", parameterStrings)}) {{"); builder.Append($"({string.Join(", ", parameterStrings)})");
_writer.WriteLine(builder.ToString()); _writer.StartFunction(builder.ToString());
_writer.WriteLine("@start");
List<Variable> parameterVars = []; var parameterVars = parameters.Select(parameter => new Variable(parameter.Name, new Val("%" + parameter.Name, parameter.Type, ValKind.Direct))).ToList();
foreach (var parameter in parameters)
{
var parameterName = "%" + parameter.Name;
if (parameter.Type is NubPrimitiveType primitiveType)
{
switch (primitiveType.Kind)
{
case PrimitiveTypeKind.I16:
parameterName = VarName();
_writer.WriteLine($" {parameterName} =w extsh %{parameter.Name}");
break;
case PrimitiveTypeKind.I8:
parameterName = VarName();
_writer.WriteLine($" {parameterName} =w extsb %{parameter.Name}");
break;
case PrimitiveTypeKind.U16:
parameterName = VarName();
_writer.WriteLine($" {parameterName} =w extuh %{parameter.Name}");
break;
case PrimitiveTypeKind.U8:
parameterName = VarName();
_writer.WriteLine($" {parameterName} =w extub %{parameter.Name}");
break;
}
}
parameterVars.Add(new Variable(parameter.Name, new Val(parameterName, parameter.Type, ValKind.Direct)));
}
EmitBlock(body, parameterVars); EmitBlock(body, parameterVars);
@@ -567,18 +528,16 @@ public static class QBEGenerator
{ {
if (returnType is NubVoidType) if (returnType is NubVoidType)
{ {
_writer.WriteLine(" ret"); _writer.Code("ret");
} }
} }
_writer.WriteLine("}"); _writer.EndFunction();
} }
private static void EmitStructDefinition(BoundStructDefinitionNode structDefinition) private static void EmitStructDefinition(BoundStructDefinitionNode structDefinition)
{ {
var builder = new StringBuilder(); _writer.Write($"type {StructName(structDefinition)} = {{ ");
builder.Append($"type {StructName(structDefinition)} = {{ ");
foreach (var structDefinitionField in structDefinition.Fields) foreach (var structDefinitionField in structDefinition.Fields)
{ {
var qbeType = structDefinitionField.Type switch var qbeType = structDefinitionField.Type switch
@@ -597,12 +556,10 @@ public static class QBEGenerator
NubArrayType or NubPointerType or NubFuncType or NubCStringType or NubStringType => "l", NubArrayType or NubPointerType or NubFuncType or NubCStringType or NubStringType => "l",
_ => throw new NotSupportedException($"'{structDefinitionField.Type}' type cannot be used in structs") _ => throw new NotSupportedException($"'{structDefinitionField.Type}' type cannot be used in structs")
}; };
builder.Append(qbeType + ", "); _writer.Write(qbeType + ", ");
} }
builder.Append("}"); _writer.WriteLine("}");
_writer.WriteLine(builder.ToString());
} }
private static void EmitStatement(BoundStatementNode statement) private static void EmitStatement(BoundStatementNode statement)
@@ -672,13 +629,13 @@ public static class QBEGenerator
private static void EmitBreak(BoundBreakNode @break) private static void EmitBreak(BoundBreakNode @break)
{ {
_writer.WriteLine($" jmp {_breakLabels.Peek()}"); _writer.Code($"jmp {_breakLabels.Peek()}");
_codeIsReachable = false; _codeIsReachable = false;
} }
private static void EmitContinue(BoundContinueNode @continue) private static void EmitContinue(BoundContinueNode @continue)
{ {
_writer.WriteLine($" jmp {_continueLabels.Peek()}"); _writer.Code($"jmp {_continueLabels.Peek()}");
_codeIsReachable = false; _codeIsReachable = false;
} }
@@ -689,11 +646,11 @@ public static class QBEGenerator
var endLabel = LabelName(); var endLabel = LabelName();
var result = EmitUnwrap(EmitExpression(ifStatement.Condition)); var result = EmitUnwrap(EmitExpression(ifStatement.Condition));
_writer.WriteLine($" jnz {result}, {trueLabel}, {falseLabel}"); _writer.Code($"jnz {result}, {trueLabel}, {falseLabel}");
_writer.WriteLine(trueLabel); _writer.Code(trueLabel);
EmitBlock(ifStatement.Body); EmitBlock(ifStatement.Body);
_writer.WriteLine($" jmp {endLabel}"); _writer.Code($"jmp {endLabel}");
_writer.WriteLine(falseLabel); _writer.Code(falseLabel);
if (ifStatement.Else.HasValue) if (ifStatement.Else.HasValue)
{ {
ifStatement.Else.Value.Match ifStatement.Else.Value.Match
@@ -703,7 +660,7 @@ public static class QBEGenerator
); );
} }
_writer.WriteLine(endLabel); _writer.Code(endLabel);
} }
private static void EmitReturn(BoundReturnNode @return) private static void EmitReturn(BoundReturnNode @return)
@@ -711,18 +668,18 @@ public static class QBEGenerator
if (@return.Value.HasValue) if (@return.Value.HasValue)
{ {
var result = EmitUnwrap(EmitExpression(@return.Value.Value)); var result = EmitUnwrap(EmitExpression(@return.Value.Value));
_writer.WriteLine($" ret {result}"); _writer.Code($"ret {result}");
} }
else else
{ {
_writer.WriteLine(" ret"); _writer.Code("ret");
} }
} }
private static void EmitVariableDeclaration(BoundVariableDeclarationNode variableDeclaration) private static void EmitVariableDeclaration(BoundVariableDeclarationNode variableDeclaration)
{ {
var variable = VarName(); var variable = VarName();
_writer.WriteLine($" {variable} =l alloc8 {SizeOf(variableDeclaration.Type)}"); _writer.Code($"{variable} =l alloc8 {SizeOf(variableDeclaration.Type)}");
if (variableDeclaration.Assignment.HasValue) if (variableDeclaration.Assignment.HasValue)
{ {
@@ -741,13 +698,13 @@ public static class QBEGenerator
_breakLabels.Push(endLabel); _breakLabels.Push(endLabel);
_continueLabels.Push(conditionLabel); _continueLabels.Push(conditionLabel);
_writer.WriteLine($" jmp {conditionLabel}"); _writer.Code($"jmp {conditionLabel}");
_writer.WriteLine(iterationLabel); _writer.Code(iterationLabel);
EmitBlock(whileStatement.Body); EmitBlock(whileStatement.Body);
_writer.WriteLine(conditionLabel); _writer.Code(conditionLabel);
var result = EmitUnwrap(EmitExpression(whileStatement.Condition)); var result = EmitUnwrap(EmitExpression(whileStatement.Condition));
_writer.WriteLine($" jnz {result}, {iterationLabel}, {endLabel}"); _writer.Code($"jnz {result}, {iterationLabel}, {endLabel}");
_writer.WriteLine(endLabel); _writer.Code(endLabel);
_continueLabels.Pop(); _continueLabels.Pop();
_breakLabels.Pop(); _breakLabels.Pop();
@@ -788,34 +745,34 @@ public static class QBEGenerator
var elementType = ((NubArrayType)arrayIndexAccess.Array.Type).ElementType; var elementType = ((NubArrayType)arrayIndexAccess.Array.Type).ElementType;
var pointer = VarName(); var pointer = VarName();
_writer.WriteLine($" {pointer} =l mul {index}, {SizeOf(elementType)}"); _writer.Code($"{pointer} =l mul {index}, {SizeOf(elementType)}");
_writer.WriteLine($" {pointer} =l add {pointer}, 8"); _writer.Code($"{pointer} =l add {pointer}, 8");
_writer.WriteLine($" {pointer} =l add {array}, {pointer}"); _writer.Code($"{pointer} =l add {array}, {pointer}");
return new Val(pointer, arrayIndexAccess.Type, ValKind.Pointer); return new Val(pointer, arrayIndexAccess.Type, ValKind.Pointer);
} }
private static void EmitArrayBoundsCheck(string array, string index) private static void EmitArrayBoundsCheck(string array, string index)
{ {
var count = VarName(); var count = VarName();
_writer.WriteLine($" {count} =l loadl {array}"); _writer.Code($"{count} =l loadl {array}");
var isNegative = VarName(); var isNegative = VarName();
_writer.WriteLine($" {isNegative} =w csltl {index}, 0"); _writer.Code($"{isNegative} =w csltl {index}, 0");
var isOob = VarName(); var isOob = VarName();
_writer.WriteLine($" {isOob} =w csgel {index}, {count}"); _writer.Code($"{isOob} =w csgel {index}, {count}");
var anyOob = VarName(); var anyOob = VarName();
_writer.WriteLine($" {anyOob} =w or {isNegative}, {isOob}"); _writer.Code($"{anyOob} =w or {isNegative}, {isOob}");
var oobLabel = LabelName(); var oobLabel = LabelName();
var notOobLabel = LabelName(); var notOobLabel = LabelName();
_writer.WriteLine($" jnz {anyOob}, {oobLabel}, {notOobLabel}"); _writer.Code($"jnz {anyOob}, {oobLabel}, {notOobLabel}");
_writer.WriteLine(oobLabel); _writer.Code(oobLabel);
_writer.WriteLine($" call $nub_panic_array_oob()"); _writer.Code($"call $nub_panic_array_oob()");
_writer.WriteLine(notOobLabel); _writer.Code(notOobLabel);
} }
private static Val EmitArrayInitializer(BoundArrayInitializerNode arrayInitializer) private static Val EmitArrayInitializer(BoundArrayInitializerNode arrayInitializer)
@@ -824,17 +781,17 @@ public static class QBEGenerator
var elementSize = SizeOf(arrayInitializer.ElementType); var elementSize = SizeOf(arrayInitializer.ElementType);
var capacityInBytes = VarName(); var capacityInBytes = VarName();
_writer.WriteLine($" {capacityInBytes} =l mul {capacity}, {elementSize}"); _writer.Code($"{capacityInBytes} =l mul {capacity}, {elementSize}");
var totalSize = VarName(); var totalSize = VarName();
_writer.WriteLine($" {totalSize} =l add {capacityInBytes}, 8"); _writer.Code($"{totalSize} =l add {capacityInBytes}, 8");
var arrayPointer = VarName(); var arrayPointer = VarName();
_writer.WriteLine($" {arrayPointer} =l alloc8 {totalSize}"); _writer.Code($"{arrayPointer} =l alloc8 {totalSize}");
_writer.WriteLine($" storel {capacity}, {arrayPointer}"); _writer.Code($"storel {capacity}, {arrayPointer}");
var dataPointer = VarName(); var dataPointer = VarName();
_writer.WriteLine($" {dataPointer} =l add {arrayPointer}, 8"); _writer.Code($"{dataPointer} =l add {arrayPointer}, 8");
_writer.WriteLine($" call $nub_memset(l {dataPointer}, w 0, l {capacityInBytes})"); _writer.Code($"call $nub_memset(l {dataPointer}, w 0, l {capacityInBytes})");
return new Val(arrayPointer, arrayInitializer.Type, ValKind.Direct); return new Val(arrayPointer, arrayInitializer.Type, ValKind.Direct);
} }
@@ -870,7 +827,7 @@ public static class QBEGenerator
var instruction = EmitBinaryInstructionFor(binaryExpression.Operator, binaryExpression.Left.Type, left, right); var instruction = EmitBinaryInstructionFor(binaryExpression.Operator, binaryExpression.Left.Type, left, right);
_writer.WriteLine($" {outputName} {QBEAssign(binaryExpression.Left.Type)} {instruction} {left}, {right}"); _writer.Code($"{outputName} {QBEAssign(binaryExpression.Left.Type)} {instruction} {left}, {right}");
return new Val(outputName, binaryExpression.Type, ValKind.Direct); return new Val(outputName, binaryExpression.Type, ValKind.Direct);
} }
@@ -890,13 +847,13 @@ public static class QBEGenerator
{ {
if (type.IsSignedInteger) if (type.IsSignedInteger)
{ {
_writer.WriteLine($" {left} =w extsb {left}"); _writer.Code($"{left} =w extsb {left}");
_writer.WriteLine($" {right} =w extsb {right}"); _writer.Code($"{right} =w extsb {right}");
} }
else else
{ {
_writer.WriteLine($" {left} =w extub {left}"); _writer.Code($"{left} =w extub {left}");
_writer.WriteLine($" {right} =w extub {right}"); _writer.Code($"{right} =w extub {right}");
} }
suffix = 'w'; suffix = 'w';
@@ -905,13 +862,13 @@ public static class QBEGenerator
{ {
if (type.IsSignedInteger) if (type.IsSignedInteger)
{ {
_writer.WriteLine($" {left} =w extsh {left}"); _writer.Code($"{left} =w extsh {left}");
_writer.WriteLine($" {right} =w extsh {right}"); _writer.Code($"{right} =w extsh {right}");
} }
else else
{ {
_writer.WriteLine($" {left} =w extuh {left}"); _writer.Code($"{left} =w extuh {left}");
_writer.WriteLine($" {right} =w extuh {right}"); _writer.Code($"{right} =w extuh {right}");
} }
suffix = 'w'; suffix = 'w';
@@ -1083,7 +1040,7 @@ public static class QBEGenerator
{ {
destination = VarName(); destination = VarName();
var size = SizeOf(structInitializer.StructType); var size = SizeOf(structInitializer.StructType);
_writer.WriteLine($" {destination} =l alloc8 {size}"); _writer.Code($"{destination} =l alloc8 {size}");
} }
foreach (var field in structDefinition.Fields) foreach (var field in structDefinition.Fields)
@@ -1096,7 +1053,7 @@ public static class QBEGenerator
Debug.Assert(valueExpression != null); Debug.Assert(valueExpression != null);
var offset = VarName(); var offset = VarName();
_writer.WriteLine($" {offset} =l add {destination}, {OffsetOf(structDefinition, field.Name)}"); _writer.Code($"{offset} =l add {destination}, {OffsetOf(structDefinition, field.Name)}");
EmitCopyIntoOrInitialize(valueExpression, offset); EmitCopyIntoOrInitialize(valueExpression, offset);
} }
@@ -1115,16 +1072,16 @@ public static class QBEGenerator
switch (unaryExpression.Operand.Type) switch (unaryExpression.Operand.Type)
{ {
case NubPrimitiveType { Kind: PrimitiveTypeKind.I64 }: case NubPrimitiveType { Kind: PrimitiveTypeKind.I64 }:
_writer.WriteLine($" {outputName} =l neg {operand}"); _writer.Code($"{outputName} =l neg {operand}");
return new Val(outputName, unaryExpression.Type, ValKind.Direct); return new Val(outputName, unaryExpression.Type, ValKind.Direct);
case NubPrimitiveType { Kind: PrimitiveTypeKind.I32 or PrimitiveTypeKind.I16 or PrimitiveTypeKind.I8 }: case NubPrimitiveType { Kind: PrimitiveTypeKind.I32 or PrimitiveTypeKind.I16 or PrimitiveTypeKind.I8 }:
_writer.WriteLine($" {outputName} =w neg {operand}"); _writer.Code($"{outputName} =w neg {operand}");
return new Val(outputName, unaryExpression.Type, ValKind.Direct); return new Val(outputName, unaryExpression.Type, ValKind.Direct);
case NubPrimitiveType { Kind: PrimitiveTypeKind.F64 }: case NubPrimitiveType { Kind: PrimitiveTypeKind.F64 }:
_writer.WriteLine($" {outputName} =d neg {operand}"); _writer.Code($"{outputName} =d neg {operand}");
return new Val(outputName, unaryExpression.Type, ValKind.Direct); return new Val(outputName, unaryExpression.Type, ValKind.Direct);
case NubPrimitiveType { Kind: PrimitiveTypeKind.F32 }: case NubPrimitiveType { Kind: PrimitiveTypeKind.F32 }:
_writer.WriteLine($" {outputName} =s neg {operand}"); _writer.Code($"{outputName} =s neg {operand}");
return new Val(outputName, unaryExpression.Type, ValKind.Direct); return new Val(outputName, unaryExpression.Type, ValKind.Direct);
} }
@@ -1135,7 +1092,7 @@ public static class QBEGenerator
switch (unaryExpression.Operand.Type) switch (unaryExpression.Operand.Type)
{ {
case NubPrimitiveType { Kind: PrimitiveTypeKind.Bool }: case NubPrimitiveType { Kind: PrimitiveTypeKind.Bool }:
_writer.WriteLine($" {outputName} =w xor {operand}, 1"); _writer.Code($"{outputName} =w xor {operand}, 1");
return new Val(outputName, unaryExpression.Type, ValKind.Direct); return new Val(outputName, unaryExpression.Type, ValKind.Direct);
} }
@@ -1161,7 +1118,7 @@ public static class QBEGenerator
{ {
if (memberAccess.Member == "count") if (memberAccess.Member == "count")
{ {
_writer.WriteLine($" {output} =l loadl {item}"); _writer.Code($"{output} =l loadl {item}");
return new Val(output, memberAccess.Type, ValKind.Direct); return new Val(output, memberAccess.Type, ValKind.Direct);
} }
@@ -1171,7 +1128,7 @@ public static class QBEGenerator
{ {
if (memberAccess.Member == "count") if (memberAccess.Member == "count")
{ {
_writer.WriteLine($" {output} =l call $nub_string_length(l {item})"); _writer.Code($"{output} =l call $nub_string_length(l {item})");
return new Val(output, memberAccess.Type, ValKind.Direct); return new Val(output, memberAccess.Type, ValKind.Direct);
} }
@@ -1181,7 +1138,7 @@ public static class QBEGenerator
{ {
if (memberAccess.Member == "count") if (memberAccess.Member == "count")
{ {
_writer.WriteLine($" {output} =l call $nub_cstring_length(l {item})"); _writer.Code($"{output} =l call $nub_cstring_length(l {item})");
return new Val(output, memberAccess.Type, ValKind.Direct); return new Val(output, memberAccess.Type, ValKind.Direct);
} }
@@ -1192,7 +1149,7 @@ public static class QBEGenerator
var structDefinition = _definitionTable.LookupStruct(structType.Namespace, structType.Name).GetValue(); var structDefinition = _definitionTable.LookupStruct(structType.Namespace, structType.Name).GetValue();
var offset = OffsetOf(structDefinition, memberAccess.Member); var offset = OffsetOf(structDefinition, memberAccess.Member);
_writer.WriteLine($" {output} =l add {item}, {offset}"); _writer.Code($"{output} =l add {item}, {offset}");
return new Val(output, memberAccess.Type, ValKind.Pointer); return new Val(output, memberAccess.Type, ValKind.Pointer);
} }
} }
@@ -1237,12 +1194,12 @@ public static class QBEGenerator
{ {
var outputName = VarName(); var outputName = VarName();
_writer.WriteLine($" {outputName} {QBEAssign(funcCall.Type)} call {funcPointer}({string.Join(", ", parameterStrings)})"); _writer.Code($"{outputName} {QBEAssign(funcCall.Type)} call {funcPointer}({string.Join(", ", parameterStrings)})");
return new Val(outputName, funcCall.Type, ValKind.Direct); return new Val(outputName, funcCall.Type, ValKind.Direct);
} }
else else
{ {
_writer.WriteLine($" call {funcPointer}({string.Join(", ", parameterStrings)})"); _writer.Code($"call {funcPointer}({string.Join(", ", parameterStrings)})");
return new Val(string.Empty, funcCall.Type, ValKind.Direct); return new Val(string.Empty, funcCall.Type, ValKind.Direct);
} }
} }

48
src/compiler/Generation/QBE/QBEWriter.cs
View File

@@ -6,12 +6,27 @@ namespace Generation.QBE;
public class QBEWriter public class QBEWriter
{ {
private int _currentLine = -1;
private readonly StringBuilder _builder = new(); private readonly StringBuilder _builder = new();
private int _currentLine = -1;
public void ResetLineTracker() public QBEWriter(string debugFile)
{
_builder.AppendLine($"dbgfile \"{debugFile}\"");
_builder.AppendLine();
}
public void StartFunction(string signature)
{ {
_currentLine = -1; _currentLine = -1;
_builder.AppendLine();
_builder.Append(signature);
_builder.AppendLine(" {");
_builder.AppendLine("@start");
}
public void EndFunction()
{
_builder.AppendLine("}");
} }
private void WriteDebugLocation(SourceSpan span) private void WriteDebugLocation(SourceSpan span)
@@ -37,23 +52,30 @@ public class QBEWriter
} }
} }
public QBEWriter WriteLine(BoundNode node, string value) public void Code(string value)
{
WriteDebugLocation(node);
WriteLine(value);
return this;
}
public QBEWriter WriteLine(string value)
{ {
_builder.Append('\t');
_builder.AppendLine(value); _builder.AppendLine(value);
return this;
} }
public QBEWriter WriteLine() public void Comment(string comment)
{
_builder.AppendLine("# " + comment);
}
public void WriteLine(string text)
{
_builder.AppendLine(text);
}
public void Write(string text)
{
_builder.Append(text);
}
public void NewLine()
{ {
_builder.AppendLine(); _builder.AppendLine();
return this;
} }
public override string ToString() public override string ToString()