feat: implement basic while loop generation in asmGenerator

2025-02-24 04:47:21 +00:00 · 2025-02-24 04:47:21 +00:00 · 909114bdce
commit 909114bdce
parent 2bed722a4f
1 changed files with 12 additions and 0 deletions
--- a/src/main/wacc/backend/asmGenerator.scala
+++ b/src/main/wacc/backend/asmGenerator.scala
@ -105,6 +105,18 @@ object asmGenerator {
          elseBranch.flatMap(generateStmt) ++
          List(LabelDef(endLabel))
      }
+      case While(cond, body) => {
+        val startLabel = labelGenerator.getLabel()
+        val endLabel = labelGenerator.getLabel()
+        List(LabelDef(startLabel)) ++
+          evalExprIntoReg(cond, Register(RegSize.R64, RegName.AX)) ++
+          List(
+            Compare(Register(RegSize.R64, RegName.AX), ImmediateVal(0)),
+            Jump(LabelArg(endLabel), Cond.Equal)
+          ) ++
+          body.flatMap(generateStmt) ++
+          List(Jump(LabelArg(startLabel)), LabelDef(endLabel))
+      }
      case _ => List()
    }