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

...

Browse Source
This commit is contained in:
nub31
2025-05-08 18:57:38 +02:00
parent 1429c3a993
commit 626b2fef0c
5 changed files with 124 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
example/program.nub
View File

@@ -1,12 +1,20 @@
import "c";
struct Human {
age: int64;
age: int64 = 0;
name: string;
}
global func main() {
printName();
while true {
let dad = new Human
{
name = "John";
};
printName(dad);
}
}
func printName(human: Human) {

98
src/compiler/Nub.Lang/Backend/Generator.cs
View File

@@ -44,17 +44,22 @@ public class Generator
private string QbeTypeName(NubType type)
{
if (type.Equals(NubType.Int64) || type.Equals(NubType.String))
{
return "l";
}
if (type.Equals(NubType.Int32) || type.Equals(NubType.Bool))
{
return "w";
}
return $":{type.Name}";
return "l";
}
private int QbeTypeSize(NubType type)
{
if (type.Equals(NubType.Int32) || type.Equals(NubType.Bool))
{
return 4;
}
return 8;
}
private void GenerateFuncDefinition(LocalFuncDefinitionNode node)
@@ -261,25 +266,49 @@ public class Generator
return GenerateLiteral(literal);
case StructInitializerNode structInitializer:
return GenerateStructInitializer(structInitializer);
case StructMemberAccessorNode structMemberAccessor:
return GenerateStructMemberAccessor(structMemberAccessor);
case StructFieldAccessorNode structMemberAccessor:
return GenerateStructFieldAccessor(structMemberAccessor);
default:
throw new ArgumentOutOfRangeException(nameof(expression));
}
}
private string GenerateStructMemberAccessor(StructMemberAccessorNode structMemberAccessor)
private string GenerateStructFieldAccessor(StructFieldAccessorNode structFieldAccessor)
{
var type = _variables.First(x => x.Key == structMemberAccessor.Fields[0]).Value.Type;
var def = _definitions.OfType<StructDefinitionNode>().FirstOrDefault(s => s.Name == type.Name);
if (def == null)
var structType = structFieldAccessor.Struct.Type;
var structDefinition = _definitions
.OfType<StructDefinitionNode>()
.FirstOrDefault(s => s.Name == structType.Name);
if (structDefinition == null)
{
throw new Exception("def is null");
throw new Exception($"Struct {structType.Name} is not defined");
}
throw new NotImplementedException("Not finished yet");
var @struct = GenerateExpression(structFieldAccessor.Struct);
return $"load{QbeTypeName(structMemberAccessor.Type)} ";
var fieldIndex = -1;
for (var i = 0; i < structDefinition.Fields.Count; i++)
{
if (structDefinition.Fields[i].Name == structFieldAccessor.Field)
{
fieldIndex = i;
break;
}
}
if (fieldIndex == -1)
{
throw new Exception($"Field {structFieldAccessor.Field} is not defined in struct {structType.Name}");
}
var offsetLabel = GenName("offset");
_builder.AppendLine($" %{offsetLabel} ={QbeTypeName(structFieldAccessor.Type)} add {@struct}, {fieldIndex * QbeTypeSize(structFieldAccessor.Type)}");
var outputLabel = GenName("field");
_builder.AppendLine($" %{outputLabel} ={QbeTypeName(structFieldAccessor.Type)} load{QbeTypeName(structFieldAccessor.Type)} %{offsetLabel}");
return $"%{outputLabel}";
}
private string GenerateBinaryExpression(BinaryExpressionNode binaryExpression)
@@ -317,7 +346,44 @@ public class Generator
private string GenerateStructInitializer(StructInitializerNode structInitializer)
{
throw new NotImplementedException();
var structDefinition = _definitions.OfType<StructDefinitionNode>()
.FirstOrDefault(s => s.Name == structInitializer.StructType.Name);
if (structDefinition == null)
{
throw new Exception($"Struct {structInitializer.StructType.Name} is not defined");
}
var structVar = GenName("struct");
var size = structDefinition.Fields.Sum(x => QbeTypeSize(x.Type));
_builder.AppendLine($" %{structVar} =l alloc8 {size}");
for (var i = 0; i < structDefinition.Fields.Count; i++)
{
var field = structDefinition.Fields[i];
if (structInitializer.Initializers.TryGetValue(field.Name, out var fieldValue))
{
var var = GenerateExpression(fieldValue);
var offsetLabel = GenName("offset");
_builder.AppendLine($" %{offsetLabel} =l add %{structVar}, {i * QbeTypeSize(field.Type)}");
_builder.AppendLine($" store{QbeTypeName(field.Type)} {var}, %{offsetLabel}");
}
else if (field.Value.HasValue)
{
var var = GenerateExpression(field.Value.Value);
var offsetLabel = GenName("offset");
_builder.AppendLine($" %{offsetLabel} =l add %{structVar}, {i * QbeTypeSize(field.Type)}");
_builder.AppendLine($" store{QbeTypeName(field.Type)} {var}, %{offsetLabel}");
}
else
{
throw new Exception($"Field {field.Name} on struct {structInitializer.StructType.Name} is not initialized");
}
}
return $"%{structVar}";
}
private string GenerateExpressionFuncCall(FuncCallExpressionNode funcCall)

17
src/compiler/Nub.Lang/Frontend/Parsing/Parser.cs
View File

@@ -396,7 +396,9 @@ public class Parser
return new StructInitializerNode(type, initializers);
}
default:
{
throw new Exception($"Unknown symbol: {symbolToken.Symbol}");
}
}
}
default:
@@ -421,18 +423,15 @@ public class Parser
case Symbol.Period:
{
Next();
List<string> members =
[
identifier.Value,
ExpectIdentifier().Value
];
ExpressionNode result = new IdentifierNode(identifier.Value);
while (TryExpectSymbol(Symbol.Period))
do
{
members.Add(ExpectIdentifier().Value);
}
var field = ExpectIdentifier();
result = new StructFieldAccessorNode(result, field.Value);
} while (TryExpectSymbol(Symbol.Period));
return new StructMemberAccessorNode(members);
return result;
}
case Symbol.OpenParen:
{

5
src/compiler/Nub.Lang/Frontend/Parsing/StructFieldAccessorNode.cs
View File

@@ -1,6 +1,7 @@
namespace Nub.Lang.Frontend.Parsing;
public class StructFieldAccessorNode(List<string> fields) : ExpressionNode
public class StructFieldAccessorNode(ExpressionNode @struct, string field) : ExpressionNode
{
public List<string> Fields { get; } = fields;
public ExpressionNode Struct { get; } = @struct;
public string Field { get; } = field;
}

44
src/compiler/Nub.Lang/Frontend/Typing/ExpressionTyper.cs
View File

@@ -179,8 +179,8 @@ public class ExpressionTyper
case StructInitializerNode structInitializer:
PopulateStructInitializer(structInitializer);
break;
case StructMemberAccessorNode structMemberAccessor:
GenerateStructMemberAccessorNode(structMemberAccessor);
case StructFieldAccessorNode structMemberAccessor:
PopulateStructMemberAccessorNode(structMemberAccessor);
break;
default:
throw new ArgumentOutOfRangeException(nameof(expression));
@@ -262,43 +262,29 @@ public class ExpressionTyper
structInitializer.Type = structInitializer.StructType;
}
// TODO: Fix this ugly ass code
private void GenerateStructMemberAccessorNode(StructMemberAccessorNode structMemberAccessor)
private void PopulateStructMemberAccessorNode(StructFieldAccessorNode structFieldAccessor)
{
var variable = _variables.FirstOrDefault(v => v.Name == structMemberAccessor.Fields[0]);
if (variable == null)
PopulateExpression(structFieldAccessor.Struct);
var structType = structFieldAccessor.Struct.Type;
if (structType == null)
{
throw new Exception($"Variable {structMemberAccessor.Fields[0]} is not defined");
throw new Exception($"Cannot access field on non-struct type: {structFieldAccessor.Struct}");
}
var definition = _structDefinitions.FirstOrDefault(sd => sd.Name == variable.Type.Name);
if (definition == null)
var structDefinition = _structDefinitions.FirstOrDefault(s => s.Name == structType.Name);
if (structDefinition == null)
{
throw new Exception($"Struct {structMemberAccessor.Fields[0]} is not defined");
throw new Exception($"Struct {structType.Name} is not defined");
}
for (var i = 1; i < structMemberAccessor.Fields.Count - 1; i++)
var field = structDefinition.Fields.FirstOrDefault(f => f.Name == structFieldAccessor.Field);
if (field == null)
{
var member = definition.Fields.FirstOrDefault(m => m.Name == structMemberAccessor.Fields[i]);
if (member == null)
{
throw new Exception($"Member {structMemberAccessor.Fields[i]} does not exist on struct {definition.Name}");
}
definition = _structDefinitions.FirstOrDefault(sd => sd.Name == member.Type.Name);
if (definition == null)
{
throw new Exception($"Struct {structMemberAccessor.Fields[i]} is not defined");
}
throw new Exception($"Field {structFieldAccessor.Field} is not defined in struct {structType.Name}");
}
var tmp = definition.Fields.FirstOrDefault(m => m.Name == structMemberAccessor.Fields.Last());
if (tmp == null)
{
throw new Exception($"Member {structMemberAccessor.Fields.Last()} does not exist on struct {definition.Name}");
}
structMemberAccessor.Type = tmp.Type;
structFieldAccessor.Type = field.Type;
}
private class Variable(string name, NubType type)