diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/ast/Python3VisitorImpl.java | 2 | ||||
| -rw-r--r-- | src/ast/nodes/WhileStmtNode.java | 18 | ||||
| -rw-r--r-- | src/codegen/Label.java | 2 |
3 files changed, 15 insertions, 7 deletions
diff --git a/src/ast/Python3VisitorImpl.java b/src/ast/Python3VisitorImpl.java index e6728d5..7e8881f 100644 --- a/src/ast/Python3VisitorImpl.java +++ b/src/ast/Python3VisitorImpl.java @@ -504,7 +504,6 @@ public class Python3VisitorImpl extends Python3ParserBaseVisitor<Node> { } private void optimizeWithThird(BlockNode block, int lineStart, int lineStop, int index) { - int counter = 0; ArrayList<Node> stms = block.getChilds(); for (var e : stms) { if (e instanceof SimpleStmtsNode) { @@ -543,7 +542,6 @@ public class Python3VisitorImpl extends Python3ParserBaseVisitor<Node> { rewriter.replace(firstToken, lastToken, newVar); } } - counter++; } } } diff --git a/src/ast/nodes/WhileStmtNode.java b/src/ast/nodes/WhileStmtNode.java index 43bf5ff..64b0b71 100644 --- a/src/ast/nodes/WhileStmtNode.java +++ b/src/ast/nodes/WhileStmtNode.java @@ -4,6 +4,7 @@ import ast.types.*; import java.util.ArrayList; import semanticanalysis.SemanticError; import semanticanalysis.SymbolTable; +import codegen.Label; /** * Node for the `while_stmt` statement of the grammar. @@ -33,12 +34,21 @@ public class WhileStmtNode implements Node { return new VoidType(); } - /** - * NOTE: It is not a part for this project. - */ @Override public String codeGeneration() { - return ""; + String startLabel = Label.newBasic("start"); + String endLabel = Label.newBasic("end"); + + String exprS = expr.codeGeneration(); + String blockS = block.codeGeneration(); + + return startLabel + ":\n" + + exprS + + "storei T1 0\n" + + "beq A0 T1 " + endLabel + "\n" + + blockS + + "b " + startLabel + "\n" + + endLabel + ":\n"; } @Override diff --git a/src/codegen/Label.java b/src/codegen/Label.java index a301ede..61403d8 100644 --- a/src/codegen/Label.java +++ b/src/codegen/Label.java @@ -41,6 +41,6 @@ public class Label { public static String newVar() { - return "_tmp" + (varDefCount++); + return "tmp" + (varDefCount++); } } |