From 634681ce2bb36cd686943102a0f071cd6a09049f Mon Sep 17 00:00:00 2001 From: Melodi <62381002+Melodi17@users.noreply.github.com> Date: Wed, 27 Aug 2025 21:22:55 +1000 Subject: fixed bugs and revolutionized the universe --- Compiler/CCompilerVisitor.cs | 906 ------------------------------------------- 1 file changed, 906 deletions(-) delete mode 100644 Compiler/CCompilerVisitor.cs (limited to 'Compiler/CCompilerVisitor.cs') diff --git a/Compiler/CCompilerVisitor.cs b/Compiler/CCompilerVisitor.cs deleted file mode 100644 index 9ffc639..0000000 --- a/Compiler/CCompilerVisitor.cs +++ /dev/null @@ -1,906 +0,0 @@ -namespace Q1.Compiler; - -using System.Runtime.CompilerServices; -using Antlr4.Runtime.Tree; -using Emulator; - -public class CCompilerVisitor : CGrammarBaseVisitor -{ - private int _branchCount = 1; - private List _instructions = new(); - private Stack> _variableStack = new(); - private List _methods = new(); - private static int _memLowest = Q1Layout.FreeMemoryStart; - private static int _memPointer = CCompilerVisitor._memLowest; - - private List<(CGrammarParser.FunctionDefinitionContext context, - FunctionReference function, FunctionOverload overload, - List paramNames, List paramTypes)> _functionBodies = new(); - - private const int ColumnWidth = 30; - - public readonly CommentCompilationMode CommentMode; - - public CCompilerVisitor(CommentCompilationMode commentMode) - { - this.CommentMode = commentMode; - } - - private VariableReference FindVariable(string varName) - { - foreach (Dictionary variables in this._variableStack) - { - if (variables.TryGetValue(varName, out VariableReference? reference)) - return reference; - } - - throw new CompilerException($"Variable {varName} could not be found in current scope."); - } - - private int Alloc(int size) - { - int loc = CCompilerVisitor._memPointer; - CCompilerVisitor._memPointer += size; - - return loc; - } - - private string GenerateBranchLabel(string type) - { - return $"_{type}_{_branchCount++}"; - } - private void Instruction(string instruction, string? generatedComment = null) - { - this._instructions.Add($" {instruction}"); - if (generatedComment != null) - this.CommentLastInstruction(generatedComment); - } - private void Blank() - { - this._instructions.Add(""); - } - private void Branch(string name, string? generatedComment = null) - { - this._instructions.Add($"{name}:"); - if (generatedComment != null) - this.CommentLastInstruction(generatedComment); - } - private void UserComment(string comment) - { - if (this.CommentMode.HasFlag(CommentCompilationMode.UserDefined)) - this._instructions.Add($" ; {comment}"); - } - - private void GeneratedComment(string comment) - { - if (this.CommentMode.HasFlag(CommentCompilationMode.Generated)) - this._instructions.Add($"{"".PadRight(CCompilerVisitor.ColumnWidth)}; {comment}"); - } - - private void CommentLastInstruction(string comment) - { - if (!this.CommentMode.HasFlag(CommentCompilationMode.Generated)) - return; - - if (this._instructions.Count == 0) - throw new Exception("Cannot add comment to last instruction, there are none"); - - string lastInstruction = this._instructions[^1]; - lastInstruction = $"{lastInstruction.PadRight(CCompilerVisitor.ColumnWidth)}; {comment}"; - - this._instructions[^1] = lastInstruction; - } - - - public IEnumerable Compile(CGrammarParser.ProgramContext program) - { - Instruction("jmp _main"); - this.Visit(program); - - foreach ((CGrammarParser.FunctionDefinitionContext context, - FunctionReference function, FunctionOverload overload, - List paramNames, List paramTypes) in this._functionBodies) - this.VisitFunctionBody(context, function, overload, paramNames, paramTypes); - - return this._instructions; - } - - private FunctionReference CreateOrGetFunctionReference(string name, string type) - { - FunctionReference? function = this._methods.FirstOrDefault(x => x.Name == name); - if (function != null) - return function; - - function = new FunctionReference(name, type); - this._methods.Add(function); - - return function; - } - - private FunctionReference GetFunctionReference(string name) - { - FunctionReference? function = this._methods.FirstOrDefault(x => x.Name == name); - if (function != null) - return function; - - throw new CompilerException($"Function '{name}' has not been defined or prototyped."); - } - - public override string? VisitFunctionPrototype(CGrammarParser.FunctionPrototypeContext context) - { - string functionName = context.name.Text; - string type = "int"; - - List parameterTypes = context._types.Select(x => x.Text).ToList(); - List parameterNames = context._params.Select(x => x.Text).ToList(); - - FunctionReference function = this.CreateOrGetFunctionReference(functionName, type); - FunctionOverload overload = function.CreateOverloadWithoutBody(parameterNames, parameterTypes); - - function.Overloads.Add(overload); - - return null; - } - public override string? VisitFunctionDefinition(CGrammarParser.FunctionDefinitionContext context) - { - string functionName = context.name.Text; - string type = "int"; - - List paramTypes = context._types.Select(x => x.Text).ToList(); - List paramNames = context._params.Select(x => x.Text).ToList(); - - FunctionReference function = this.CreateOrGetFunctionReference(functionName, type); - FunctionOverload overload = function.CreateOverloadWithBody(paramNames, paramTypes); - - this._functionBodies.Add((context, function, overload, paramNames, paramTypes)); - - return null; - } - private void VisitFunctionBody(CGrammarParser.FunctionDefinitionContext context, FunctionReference function, FunctionOverload overload, List paramNames, List paramTypes) - { - this.Branch(function.GetBranchName(overload), function.GetBranchFriendlyName(overload)); - - if (paramNames.Count > 0) - this.Instruction("pop V0", "preserve function return address"); - - Dictionary scope = this.EnterScope(); - foreach ((string paramName, string paramType) in paramNames.Zip(paramTypes)) - { - int paramSize = 2; - int addr = this.Alloc(paramSize); - scope[paramName] = new(addr, paramSize); - this.Instruction($"pop [${addr:X4}]", $"param {paramType} {paramName}"); - } - - if (paramNames.Count > 0) - this.Instruction("push V0", "restore function return address"); - - this.VisitChildren(context); - this.ExitScope(); - - this.Instruction("mov 0, V0", "fallback return 0"); - this.Instruction("ret"); - this.Blank(); - } - - public override string? VisitCallExpression(CGrammarParser.CallExpressionContext context) - { - string functionName = context.ID().GetText(); - - FunctionReference function = this.GetFunctionReference(functionName); - FunctionOverload overload = function.GetOverload(context._params.Count); - - // We reverse arugments so when we perform pop, its in the right order - CGrammarParser.ExpressionContext[] argumentsReversed = context._params.Reverse().ToArray(); - - for (int i = 0; i < argumentsReversed.Length; i++) - { - CGrammarParser.ExpressionContext argument = argumentsReversed[i]; - string result = this.Visit(argument) - ?? throw new CompilerException("Function call argument was not an expression"); - - this.Instruction($"push {result}", $"arg {overload.ParameterTypes[i]} {overload.ParameterNames[i]}"); - } - - this.Instruction($"call {function.GetBranchName(overload)}", function.GetBranchFriendlyName(overload)); - - return "V0"; - } - - public override string? VisitBlock(CGrammarParser.BlockContext context) - { - this.EnterScope(); - - this.VisitChildren(context); - - this.ExitScope(); - - return null; - } - private Dictionary EnterScope() - { - Dictionary scope = new(); - this._variableStack.Push(scope); - - return scope; - } - private void ExitScope() - { - foreach ((string key, VariableReference value) in this._variableStack.Peek()) - { - CCompilerVisitor._memPointer -= value.Size; - if (CCompilerVisitor._memPointer < CCompilerVisitor._memLowest) - throw new CompilerException("Memory pointer is lower than mem lowest!"); - } - - // Delete scope - this._variableStack.Pop(); - } - - public override string? VisitParenthesizedExpression(CGrammarParser.ParenthesizedExpressionContext context) - { - return this.Visit(context.expression()); - } - public override string? VisitReturnStatement(CGrammarParser.ReturnStatementContext context) - { - if (context.expression() != null) - { - string? expression = this.Visit(context.expression()); - if (expression != null) - { - this.Instruction($"mov {expression}, V0", "return value"); - this.Instruction($"ret"); - } - else - throw new CompilerException("Return value was not an expression."); - } - else - this.Instruction("ret"); - - return null; - } - public override string? VisitIntConstant(CGrammarParser.IntConstantContext context) - { - string valueString = context.INT().GetText(); - if (int.TryParse(valueString, out int value)) - return $"{value}"; - else - throw new CompilerException($"Invalid integer value: {valueString}"); - } - - public override string? VisitHexConstant(CGrammarParser.HexConstantContext context) - { - string valueString = context.HEX().GetText(); - valueString = valueString[2..]; // Remove 0x - return $"${valueString}"; - } - - public override string? VisitNotExpression(CGrammarParser.NotExpressionContext context) - { - string? value = this.Visit(context.expression()); - if (value == null) - throw new CompilerException("Value for negating was not an expression."); - - this.Instruction($"not {value}"); - return "LX"; - } - - public override string? VisitInvertExpression(CGrammarParser.InvertExpressionContext context) - { - string? value = this.Visit(context.expression()); - if (value == null) - throw new CompilerException("Value for inversion was not an expression."); - - this.Instruction($"inv {value}"); - return "DX"; - } - - private void BaseBinaryOperation(CGrammarParser.ExpressionContext leftExpr, CGrammarParser.ExpressionContext rightExpr, string operation, string operationFriendlyName) - { - string? left = this.Visit(leftExpr); - if (left == null) - throw new CompilerException($"Left operand for {operationFriendlyName} was not an expression."); - - this.Instruction($"push {left}", "left operand"); - - string? right = this.Visit(rightExpr); - if (right == null) - throw new CompilerException($"Right operand for {operationFriendlyName} was not an expression."); - - this.Instruction($"mov {right}, V1", "right operand"); - this.Instruction($"pop V0", "get left operand back"); - this.Instruction($"{operation} V0, V1", $"compute {operationFriendlyName}"); - } - private string BaseBinaryCompoundOperation(CGrammarParser.TargetContext leftTarget, CGrammarParser.ExpressionContext rightExpr, string operation, string operationFriendlyName, string operationOutput) - { - string? left = this.Visit(leftTarget); - if (left == null) - throw new CompilerException($"Left operand for {operationFriendlyName} was not an expression."); - - string? right = this.Visit(rightExpr); - if (right == null) - throw new CompilerException($"Right operand for {operationFriendlyName} was not an expression."); - - // No need for stack maniupulation because left is a target and is assumed to stay constant - this.Instruction($"{operation} {left}, {right}", $"compute {operationFriendlyName}"); - this.Instruction($"mov {operationOutput}, {left}", "store back into left param"); - return left; - } - public override string? VisitAddExpression(CGrammarParser.AddExpressionContext context) - { - BaseBinaryOperation(context.left, context.right, "add", "addition"); - return "AX"; - } - public override string? VisitSubtractExpression(CGrammarParser.SubtractExpressionContext context) - { - BaseBinaryOperation(context.left, context.right, "sub", "subtraction"); - return "AX"; - } - public override string? VisitMultiplyExpression(CGrammarParser.MultiplyExpressionContext context) - { - BaseBinaryOperation(context.left, context.right, "mul", "multiplication"); - return "AX"; - } - public override string? VisitDivideExpression(CGrammarParser.DivideExpressionContext context) - { - BaseBinaryOperation(context.left, context.right, "div", "division"); - return "AX"; - } - public override string? VisitModulusExpression(CGrammarParser.ModulusExpressionContext context) - { - BaseBinaryOperation(context.left, context.right, "mod", "modulus"); - return "AX"; - } - - public override string? VisitEqualExpression(CGrammarParser.EqualExpressionContext context) - { - BaseBinaryOperation(context.left, context.right, "cmp", "equality check"); - return "LX"; - } - - public override string? VisitNotEqualExpression(CGrammarParser.NotEqualExpressionContext context) - { - BaseBinaryOperation(context.left, context.right, "cmp", "inequality check"); - this.Instruction($"not"); // lx is implicit - return "LX"; - } - - public override string? VisitGreaterThanExpression(CGrammarParser.GreaterThanExpressionContext context) - { - BaseBinaryOperation(context.left, context.right, "gt", "greater than check"); - return "LX"; - } - - public override string? VisitLessThanExpression(CGrammarParser.LessThanExpressionContext context) - { - BaseBinaryOperation(context.left, context.right, "lt", "less than check"); - return "LX"; - } - - public override string? VisitLogicalOrExpression(CGrammarParser.LogicalOrExpressionContext context) - { - string? left = this.Visit(context.left); - if (left == null) - throw new CompilerException("Left operand for logical OR was not an expression."); - - string clauseBranch = this.GenerateBranchLabel("clause"); - string endBranch = this.GenerateBranchLabel("end"); - - this.Instruction($"cmp {left}, 0", "logical or"); - this.Instruction("bz"); // implicit LX - this.Instruction($"jmp {clauseBranch}"); // only if bz is not 0 - this.Instruction("mov 1, V0", "first operand succeeded, store 1 and skip second operand"); - this.Instruction($"jmp {endBranch}"); - this.Branch(clauseBranch); - - string? right = this.Visit(context.right); - if (right == null) - throw new CompilerException("Right operand for logical OR was not an expression."); - - this.Instruction($"cmp {right}, 0", "compare result from second operand to 0"); - this.Instruction("not"); - this.Instruction("mov LX, V0", "store second operand result"); - this.Branch(endBranch); - - return "V0"; - } - public override string? VisitLogicalAndExpression(CGrammarParser.LogicalAndExpressionContext context) - { - string? left = this.Visit(context.left); - if (left == null) - throw new CompilerException("Left operand for logical OR was not an expression."); - - string clauseBranch = this.GenerateBranchLabel("clause"); - string endBranch = this.GenerateBranchLabel("end"); - - this.Instruction($"cmp {left}, 0", "logical and"); - this.Instruction("not"); - this.Instruction("bz"); // implicit LX - this.Instruction($"jmp {clauseBranch}"); // only if bz is not 0 - this.Instruction("mov 0, V0", "first operand failed, store 0 and skip second"); - this.Instruction($"jmp {endBranch}"); - this.Branch(clauseBranch); - - string? right = this.Visit(context.right); - if (right == null) - throw new CompilerException("Right operand for logical OR was not an expression."); - - this.Instruction($"cmp {right}, 0", "compare result from second operand to 0"); - this.Instruction("not"); - this.Instruction("mov LX, V0", "store second operand result"); - this.Branch(endBranch); - - return "V0"; - } - - public override string? VisitGreaterThanOrEqualExpression(CGrammarParser.GreaterThanOrEqualExpressionContext context) - { - BaseBinaryOperation(context.left, context.right, "cmp", "equality check"); - string endBranch = this.GenerateBranchLabel("end"); - - this.Instruction("bz"); - this.Instruction($"jmp {endBranch}"); - this.Instruction($"gt V0, V1"); - this.Branch(endBranch); - - return "LX"; - } - public override string? VisitLessThanOrEqualExpression(CGrammarParser.LessThanOrEqualExpressionContext context) - { - BaseBinaryOperation(context.left, context.right, "cmp", "equality check"); - string endBranch = this.GenerateBranchLabel("end"); - - this.Instruction("bz"); - this.Instruction($"jmp {endBranch}"); - this.Instruction($"lt V0, V1"); - this.Branch(endBranch); - - return "LX"; - } - public override string? VisitBitwiseXorExpression(CGrammarParser.BitwiseXorExpressionContext context) - { - this.BaseBinaryOperation(context.left, context.right, "xor", "exclusive or"); - return "DX"; - } - public override string? VisitBitwiseOrExpression(CGrammarParser.BitwiseOrExpressionContext context) - { - this.BaseBinaryOperation(context.left, context.right, "or", "or"); - return "DX"; - } - public override string? VisitBitwiseAndExpression(CGrammarParser.BitwiseAndExpressionContext context) - { - this.BaseBinaryOperation(context.left, context.right, "and", "and"); - return "DX"; - } - public override string? VisitBitwiseLeftShiftExpression(CGrammarParser.BitwiseLeftShiftExpressionContext context) - { - this.BaseBinaryOperation(context.left, context.right, "shl", "left shift"); - return "DX"; - } - public override string? VisitBitwiseRightShiftExpression(CGrammarParser.BitwiseRightShiftExpressionContext context) - { - this.BaseBinaryOperation(context.left, context.right, "shr", "right shift"); - return "DX"; - } - - public override string? VisitVariableDeclaration(CGrammarParser.VariableDeclarationContext context) - { - int size = 2; - int addr = this.Alloc(size); - string name = context.name.Text; - - Dictionary scope = this._variableStack.Peek(); - if (scope.ContainsKey(name)) - throw new CompilerException($"Variable {name} already exists in current scope."); - - scope[name] = new VariableReference(addr, size); - if (context.value != null) - { - string initialValue = this.Visit(context.value) - ?? throw new CompilerException("Initial variable definition value was not an expression"); - this.Instruction($"mov {initialValue}, [${addr:X4}]", $"int {name} = {initialValue}"); - } - else - this.Instruction($"mov 0, [${addr:X4}]", $"int {name}"); - - return null; - } - public override string? VisitVariableExpression(CGrammarParser.VariableExpressionContext context) - { - string varName = context.ID().GetText(); - VariableReference reference = this.FindVariable(varName); - - return $"[${reference.Address:X4}]"; - } - - public override string? VisitVariableTarget(CGrammarParser.VariableTargetContext context) - { - string varName = context.ID().GetText(); - VariableReference reference = this.FindVariable(varName); - - return $"[${reference.Address:X4}]"; - } - public override string? VisitAssignmentExpression(CGrammarParser.AssignmentExpressionContext context) - { - string? target = this.Visit(context.target()); - string? value = this.Visit(context.expression()); - - this.Instruction($"mov {value}, {target}", $"{target} = {value}"); - return target; - } - - public override string? VisitCompoundAddExpression(CGrammarParser.CompoundAddExpressionContext context) - { - return BaseBinaryCompoundOperation(context.target(), - context.expression(), - "add", - "addition", - "AX"); - } - public override string? VisitCompoundSubtractExpression(CGrammarParser.CompoundSubtractExpressionContext context) - { - return BaseBinaryCompoundOperation(context.target(), - context.expression(), - "sub", - "subtraction", - "AX"); - } - public override string? VisitCompoundMultiplyExpression(CGrammarParser.CompoundMultiplyExpressionContext context) - { - return BaseBinaryCompoundOperation(context.target(), - context.expression(), - "mult", - "multiplication", - "AX"); - } - public override string? VisitCompoundDivideExpression(CGrammarParser.CompoundDivideExpressionContext context) - { - return BaseBinaryCompoundOperation(context.target(), - context.expression(), - "div", - "division", - "AX"); - } - public override string? VisitCompoundModulusExpression(CGrammarParser.CompoundModulusExpressionContext context) - { - return BaseBinaryCompoundOperation(context.target(), - context.expression(), - "mod", - "modulus", - "AX"); - } - - public override string? VisitCompoundBitwiseOrExpression(CGrammarParser.CompoundBitwiseOrExpressionContext context) - { - return BaseBinaryCompoundOperation(context.target(), - context.expression(), - "or", - "or", - "DX"); - } - public override string? VisitCompoundBitwiseXorExpression(CGrammarParser.CompoundBitwiseXorExpressionContext context) - { - return BaseBinaryCompoundOperation(context.target(), - context.expression(), - "xor", - "exclusive or", - "DX"); - } - public override string? VisitCompoundBitwiseAndExpression(CGrammarParser.CompoundBitwiseAndExpressionContext context) - { - return BaseBinaryCompoundOperation(context.target(), - context.expression(), - "and", - "and", - "DX"); - } - - public override string? VisitCompoundBitwiseLeftShiftExpression(CGrammarParser.CompoundBitwiseLeftShiftExpressionContext context) - { - return BaseBinaryCompoundOperation(context.target(), - context.expression(), - "shl", - "left shift", - "DX"); - } - public override string? VisitCompoundBitwiseRightShiftExpression(CGrammarParser.CompoundBitwiseRightShiftExpressionContext context) - { - return BaseBinaryCompoundOperation(context.target(), - context.expression(), - "shr", - "right shift", - "DX"); - } - - public override string? VisitIncrementPrefixExpression(CGrammarParser.IncrementPrefixExpressionContext context) - { - string target = this.Visit(context.target()) - ?? throw new CompilerException("Increment target was not an expression"); - this.Instruction($"inc {target}", "++i"); - return target; - } - public override string? VisitDecrementPrefixExpression(CGrammarParser.DecrementPrefixExpressionContext context) - { - string target = this.Visit(context.target()) - ?? throw new CompilerException("Decrement target was not an expression"); - this.Instruction($"dec {target}", "--i"); - return target; - } - public override string? VisitIncrementPostfixExpression(CGrammarParser.IncrementPostfixExpressionContext context) - { - string target = this.Visit(context.target()) - ?? throw new CompilerException("Increment target was not an expression"); - this.Instruction($"push {target}", "i++"); - this.Instruction($"inc {target}"); - this.Instruction("pop V0"); - return "V0"; - } - public override string? VisitDecrementPostfixExpression(CGrammarParser.DecrementPostfixExpressionContext context) - { - string target = this.Visit(context.target()) - ?? throw new CompilerException("Decrement target was not an expression"); - this.Instruction($"push {target}", "i--"); - this.Instruction($"dec {target}"); - this.Instruction($"pop V0"); - return "V0"; - } - // - // public override string? VisitCommaExpression(CGrammarParser.CommaExpressionContext context) - // { - // this.Visit(context.left); - // return this.Visit(context.right); - // } - - public override string? VisitTernaryExpression(CGrammarParser.TernaryExpressionContext context) - { - string cond = this.Visit(context.cond) - ?? throw new CompilerException("Ternary condition was not an expression"); - - string thenBranch = this.GenerateBranchLabel("then"); - string endBranch = this.GenerateBranchLabel("end"); - string result; - - // Else branch is first since it saves us from having to perform a NOT operation. - // If cond, jump to then, else fall through to else, then jump to end - - this.Instruction($"bz {cond}", "check ternary condition"); - this.Instruction($"jmp {thenBranch}"); - - result = this.Visit(context.@else) - ?? throw new CompilerException("Else branch was not an expression"); - this.Instruction($"mov {result}, V0", "store ternary fail result"); - this.Instruction($"jmp {endBranch}"); - - this.Branch(thenBranch); - result = this.Visit(context.then) - ?? throw new CompilerException("Else branch was not an expression"); - this.Instruction($"mov {result}, V0", "store ternary success result"); - - this.Branch(endBranch); - - return "V0"; - } - - public override string? VisitIfStatement(CGrammarParser.IfStatementContext context) - { - string cond = this.Visit(context.expression()) - ?? throw new CompilerException("If condition was not an expression"); - string endBranch = this.GenerateBranchLabel("end"); - - this.Instruction($"bz {cond}", "check if condition"); - - if (context.@else != null) - { - // Else branch is first since it saves us from having to perform a NOT operation. - // If cond, jump to then, else fall through to else, then jump to end - - string thenBranch = this.GenerateBranchLabel("then"); - - this.Instruction($"jmp {thenBranch}"); - - this.GeneratedComment("if failed"); - if (context.@else != null) - this.Visit(context.@else); - this.Instruction($"jmp {endBranch}"); - - this.Branch(thenBranch, "if succeeded"); - this.Visit(context.then); - } - else - { - this.Instruction($"not"); - this.Instruction($"jmp {endBranch}", "if succeeds, continue, else jump to end"); - this.Visit(context.then); - } - - this.Branch(endBranch); - - return null; - } - - public override string? VisitForExprStatement(CGrammarParser.ForExprStatementContext context) - { - string bodyBranch = this.GenerateBranchLabel("body"); - string endBranch = this.GenerateBranchLabel("end"); - - context.LoopBodyBranch = bodyBranch; - context.LoopEndBranch = endBranch; - - this.EnterScope(); - this.Visit(context.initializer); - - this.Branch(bodyBranch); - string cond = this.Visit(context.cond) - ?? throw new CompilerException("For condition was not an expression"); - - this.Instruction($"not {cond}"); - this.Instruction($"bz"); - this.Instruction($"jmp {endBranch}", "if for condition fails, jump to end"); - - this.Visit(context.body); - this.Visit(context.post); - - this.Branch(endBranch); - this.ExitScope(); - - return null; - } - public override string? VisitForDeclStatement(CGrammarParser.ForDeclStatementContext context) - { - string bodyBranch = this.GenerateBranchLabel("body"); - string endBranch = this.GenerateBranchLabel("end"); - - context.LoopBodyBranch = bodyBranch; - context.LoopEndBranch = endBranch; - - this.EnterScope(); - this.Visit(context.initializer); - - this.Branch(bodyBranch); - string cond = this.Visit(context.cond) - ?? throw new CompilerException("For condition was not an expression"); - - this.Instruction($"not {cond}"); - this.Instruction($"bz"); - this.Instruction($"jmp {endBranch}", "if for condition fails, jump to end"); - - this.Visit(context.body); - this.Visit(context.post); - - this.Branch(endBranch); - this.ExitScope(); - - return null; - } - public override string? VisitWhileStatement(CGrammarParser.WhileStatementContext context) - { - string bodyBranch = this.GenerateBranchLabel("body"); - string endBranch = this.GenerateBranchLabel("end"); - - context.LoopBodyBranch = bodyBranch; - context.LoopEndBranch = endBranch; - - this.Branch(bodyBranch); - string cond = this.Visit(context.cond) - ?? throw new CompilerException("While condition was not an expression"); - - this.Instruction($"not {cond}"); - this.Instruction($"bz"); - this.Instruction($"jmp {endBranch}", "if while condition fails, jump to end"); - - this.Visit(context.body); - - this.Branch(endBranch); - - return null; - } - - public override string? VisitDoWhileStatement(CGrammarParser.DoWhileStatementContext context) - { - string bodyBranch = this.GenerateBranchLabel("body"); - string endBranch = this.GenerateBranchLabel("end"); - - context.LoopBodyBranch = bodyBranch; - context.LoopEndBranch = endBranch; - - this.Branch(bodyBranch); - this.Visit(context.body); - - string cond = this.Visit(context.cond) - ?? throw new CompilerException("While condition was not an expression"); - - this.Instruction($"bz {cond}"); - this.Instruction($"jmp {bodyBranch}", "if while condition succeeds, jump back to start"); - - this.Branch(endBranch); - - return null; - } - - public override string? VisitBreakStatement(CGrammarParser.BreakStatementContext context) - { - if (context.LoopEndBranch == null) - throw new CompilerException("Cannot break, not inside loop with end"); - - this.Instruction($"jmp {context.LoopEndBranch}", "break"); - - return null; - } - public override string? VisitContinueStatement(CGrammarParser.ContinueStatementContext context) - { - if (context.LoopBodyBranch == null) - throw new CompilerException("Cannot skip body (continue), not inside loop with body"); - - this.Instruction($"jmp {context.LoopBodyBranch}", "continue"); - - return null; - } - - public override string? VisitAddressOfExpression(CGrammarParser.AddressOfExpressionContext context) - { - string target = this.Visit(context.target()) - ?? throw new CompilerException("Address of target was not a targetable expression"); - - if (target.StartsWith("@")) - return target[1..]; - - if (target.StartsWith("[") && target.EndsWith("]")) - return target[1..^1]; - - throw new CompilerException($"Could not get address of target '{target}'"); - } - public override string? VisitDereferenceExpression(CGrammarParser.DereferenceExpressionContext context) - { - string value = this.Visit(context.expression()) - ?? throw new CompilerException("Dereference pointer was not an expression"); - - this.Instruction($"mov {value}, V0", "move pointer into register for dereference"); - return "[V0]"; - } - - public override string? VisitErrorNode(IErrorNode node) - { - if (node.Symbol.Type == CGrammarParser.COMMENT) - { - // strip // and \n and whitespace - string comment = node - .Symbol.Text - .TrimStart("//") - .TrimEnd('\n') - .Trim(); - - this.UserComment(comment); - } - - return base.VisitErrorNode(node); - } - - public override string? Visit(IParseTree tree) - { - if (tree is Context childCtx && tree.Parent is Context ctx) - ctx.SyncProps(childCtx); - - return base.Visit(tree); - } - - /// - /// Basically identical to base version, except calls Visit on each child, making Visit not redundant. - /// - public override string? VisitChildren(IRuleNode node) - { - string? result = this.DefaultResult; - int childCount = node.ChildCount; - for (int i = 0; i < childCount && this.ShouldVisitNextChild(node, result); ++i) - { - string? nextResult = this.Visit(node.GetChild(i)); - result = this.AggregateResult(result, nextResult); - } - return result; - } -} - -public class CType -{ - public string - public int Size; -} \ No newline at end of file -- cgit v1.3.1