From 31ee062ad12adcf6dae6716cb566fcbcde9f6959 Mon Sep 17 00:00:00 2001
From: geno <gabriele.genovese2@studio.unibo.it>
Date: Thu, 18 Jul 2024 12:16:40 +0200
Subject: codegen while implemented, test added

---
 src/ast/nodes/WhileStmtNode.java | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

(limited to 'src/ast/nodes/WhileStmtNode.java')

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
-- 
cgit v1.2.3-18-g5258