Feature(compiler): ForStatement - initial

ovr · ovr · commit 1dc9e9d20b4e · 2018-11-14T23:17:03.000+03:00
diff --git a/src/backend/llvm/index.ts b/src/backend/llvm/index.ts
@@ -65,6 +65,42 @@ export function passIfStatement(parent: ts.IfStatement, ctx: Context, builder: l
     builder.setInsertionPoint(next);
 }
 
+export function passForStatement(parent: ts.ForStatement, ctx: Context, builder: llvm.IRBuilder) {
+    if (parent.initializer) {
+        passStatement(<any>parent.initializer, ctx, builder);
+    }
+
+    const conditionBlock = llvm.BasicBlock.create(ctx.llvmContext, "for.condition");
+    ctx.scope.currentFunction.addBasicBlock(conditionBlock);
+
+    builder.createBr(conditionBlock);
+
+    const positiveBlock = llvm.BasicBlock.create(ctx.llvmContext, "for.true");
+    ctx.scope.currentFunction.addBasicBlock(positiveBlock);
+
+    const next = llvm.BasicBlock.create(ctx.llvmContext, "for.end");
+    ctx.scope.currentFunction.addBasicBlock(next);
+
+    if (parent.condition) {
+        builder.createBr(conditionBlock);
+
+        emitCondition(
+            parent.condition,
+            ctx,
+            builder,
+            positiveBlock,
+            next
+        );
+    }
+
+    builder.setInsertionPoint(positiveBlock);
+    passNode(parent.statement, ctx, builder);
+
+    builder.createBr(conditionBlock);
+
+    builder.setInsertionPoint(next);
+}
+
 export function emitCondition(
     condition: ts.Expression,
     ctx: Context,
@@ -443,6 +479,9 @@ export function passStatement(stmt: ts.Statement, ctx: Context, builder: llvm.IR
         case ts.SyntaxKind.IfStatement:
             passIfStatement(<any>stmt, ctx, builder);
             break;
+        case ts.SyntaxKind.ForStatement:
+            passForStatement(<any>stmt, ctx, builder);
+            break;
         default:
             throw new UnsupportedError(
                 stmt,
