This commit is contained in:
nub31
2026-03-05 21:45:04 +01:00
parent 693b119781
commit 7dd635fa91
2 changed files with 30 additions and 19 deletions

View File

@@ -167,6 +167,9 @@ public class Generator
static inline void string_rc_dec(struct nub_core_string *string) static inline void string_rc_dec(struct nub_core_string *string)
{ {
if (string == NULL)
return;
if (string->flags & FLAG_STRING_LITERAL) if (string->flags & FLAG_STRING_LITERAL)
return; return;
@@ -362,10 +365,8 @@ public class Generator
if (statement.Value != null) if (statement.Value != null)
{ {
var value = EmitExpression(statement.Value); var value = EmitExpression(statement.Value);
var variableName = TmpName();
writer.WriteLine($"{CType(statement.Value.Type, variableName)} = {value};");
EmitCleanupAllScopes(); EmitCleanupAllScopes();
writer.WriteLine($"return {variableName};"); writer.WriteLine($"return {value};");
} }
else else
{ {
@@ -481,7 +482,7 @@ public class Generator
private string EmitExpression(TypedNodeExpression node) private string EmitExpression(TypedNodeExpression node)
{ {
var value = node switch return node switch
{ {
TypedNodeExpressionBinary expression => EmitExpressionBinary(expression), TypedNodeExpressionBinary expression => EmitExpressionBinary(expression),
TypedNodeExpressionUnary expression => EmitExpressionUnary(expression), TypedNodeExpressionUnary expression => EmitExpressionUnary(expression),
@@ -498,10 +499,6 @@ public class Generator
TypedNodeExpressionFuncCall expression => EmitExpressionFuncCall(expression), TypedNodeExpressionFuncCall expression => EmitExpressionFuncCall(expression),
_ => throw new ArgumentOutOfRangeException(nameof(node), node, null) _ => throw new ArgumentOutOfRangeException(nameof(node), node, null)
}; };
var tmp = TmpName();
writer.WriteLine($"{CType(node.Type, tmp)} = {value};");
return tmp;
} }
private string EmitExpressionBinary(TypedNodeExpressionBinary expression) private string EmitExpressionBinary(TypedNodeExpressionBinary expression)
@@ -509,15 +506,16 @@ public class Generator
var left = EmitExpression(expression.Left); var left = EmitExpression(expression.Left);
var right = EmitExpression(expression.Right); var right = EmitExpression(expression.Right);
var name = Tmp();
if (expression.Operation == TypedNodeExpressionBinary.Op.Add && expression.Left.Type is NubTypeString && expression.Right.Type is NubTypeString) if (expression.Operation == TypedNodeExpressionBinary.Op.Add && expression.Left.Type is NubTypeString && expression.Right.Type is NubTypeString)
{ {
var name = TmpName();
scopes.Peek().DeconstructableNames.Add((name, expression.Type)); scopes.Peek().DeconstructableNames.Add((name, expression.Type));
writer.WriteLine($"{CType(NubTypeString.Instance, name)} = string_concat({left}, {right});"); writer.WriteLine($"{CType(NubTypeString.Instance, name)} = string_concat({left}, {right});");
return name; return name;
} }
return expression.Operation switch var op = expression.Operation switch
{ {
TypedNodeExpressionBinary.Op.Add => $"({left} + {right})", TypedNodeExpressionBinary.Op.Add => $"({left} + {right})",
TypedNodeExpressionBinary.Op.Subtract => $"({left} - {right})", TypedNodeExpressionBinary.Op.Subtract => $"({left} - {right})",
@@ -536,30 +534,40 @@ public class Generator
TypedNodeExpressionBinary.Op.LogicalOr => $"({left} || {right})", TypedNodeExpressionBinary.Op.LogicalOr => $"({left} || {right})",
_ => throw new ArgumentOutOfRangeException() _ => throw new ArgumentOutOfRangeException()
}; };
writer.WriteLine($"{CType(expression.Type, name)} = {op};");
return name;
} }
private string EmitExpressionUnary(TypedNodeExpressionUnary expression) private string EmitExpressionUnary(TypedNodeExpressionUnary expression)
{ {
var target = EmitExpression(expression.Target); var target = EmitExpression(expression.Target);
return expression.Operation switch var name = Tmp();
var op = expression.Operation switch
{ {
TypedNodeExpressionUnary.Op.Negate => $"(-{target})", TypedNodeExpressionUnary.Op.Negate => $"(-{target})",
TypedNodeExpressionUnary.Op.Invert => $"(!{target})", TypedNodeExpressionUnary.Op.Invert => $"(!{target})",
_ => throw new ArgumentOutOfRangeException() _ => throw new ArgumentOutOfRangeException()
}; };
writer.WriteLine($"{CType(expression.Type, name)} = {op};");
return name;
} }
private string EmitExpressionStringLiteral(TypedNodeExpressionStringLiteral expression) private string EmitExpressionStringLiteral(TypedNodeExpressionStringLiteral expression)
{ {
var name = TmpName(); var name = Tmp();
referencedStringLiterals.Add(name, expression.Value.Value); referencedStringLiterals.Add(name, expression.Value.Value);
return $"&{name}"; return $"&{name}";
} }
private string EmitExpressionStructLiteral(TypedNodeExpressionStructLiteral expression) private string EmitExpressionStructLiteral(TypedNodeExpressionStructLiteral expression)
{ {
var name = TmpName(); var name = Tmp();
scopes.Peek().DeconstructableNames.Add((name, expression.Type)); scopes.Peek().DeconstructableNames.Add((name, expression.Type));
var initializerValues = new Dictionary<string, string>(); var initializerValues = new Dictionary<string, string>();
@@ -580,7 +588,7 @@ public class Generator
private string EmitExpressionEnumLiteral(TypedNodeExpressionEnumLiteral expression) private string EmitExpressionEnumLiteral(TypedNodeExpressionEnumLiteral expression)
{ {
var name = TmpName(); var name = Tmp();
scopes.Peek().DeconstructableNames.Add((name, expression.Type)); scopes.Peek().DeconstructableNames.Add((name, expression.Type));
var enumVariantType = (NubTypeEnumVariant)expression.Type; var enumVariantType = (NubTypeEnumVariant)expression.Type;
@@ -638,7 +646,10 @@ public class Generator
{ {
var name = EmitExpression(expression.Target); var name = EmitExpression(expression.Target);
var parameterValues = expression.Parameters.Select(EmitExpression).ToList(); var parameterValues = expression.Parameters.Select(EmitExpression).ToList();
return $"{name}({string.Join(", ", parameterValues)})";
var tmp = Tmp();
writer.WriteLine($"{CType(expression.Type, tmp)} = {name}({string.Join(", ", parameterValues)});");
return tmp;
} }
public string CType(NubType node, string? varName = null) public string CType(NubType node, string? varName = null)
@@ -667,7 +678,7 @@ public class Generator
return $"struct {NameMangler.Mangle("anonymous", "struct", type)}{(varName != null ? $" {varName}" : "")}"; return $"struct {NameMangler.Mangle("anonymous", "struct", type)}{(varName != null ? $" {varName}" : "")}";
} }
private string TmpName() private string Tmp()
{ {
return $"_tmp{tmpNameIndex++}"; return $"_tmp{tmpNameIndex++}";
} }

View File

@@ -21,9 +21,9 @@ func main(): i32 {
{ {
core::println("quit") core::println("quit")
} }
Say message Say msg
{ {
core::println(message) core::println(msg)
} }
} }