Feature(compiler): DoStatement - initial support

ovr · ovr · commit 3aaa66aaee19 · 2018-11-17T18:18:51.000+03:00
diff --git a/src/backend/llvm/index.ts b/src/backend/llvm/index.ts
@@ -107,6 +107,35 @@ export function passForStatement(parent: ts.ForStatement, ctx: Context, builder:
     builder.setInsertionPoint(next);
 }
 
+export function passDoStatement(parent: ts.DoStatement, ctx: Context, builder: llvm.IRBuilder) {
+    const conditionBlock = llvm.BasicBlock.create(ctx.llvmContext, "for.condition");
+    ctx.scope.currentFunction.addBasicBlock(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);
+
+    builder.createBr(positiveBlock);
+    builder.setInsertionPoint(positiveBlock);
+
+    passStatement(<any>parent.statement, ctx, builder);
+
+    builder.createBr(conditionBlock);
+    builder.setInsertionPoint(conditionBlock);
+
+    emitCondition(
+        parent.expression,
+        ctx,
+        builder,
+        positiveBlock,
+        next
+    );
+
+    builder.setInsertionPoint(next);
+}
+
 export function emitCondition(
     condition: ts.Expression,
     ctx: Context,
@@ -527,6 +556,9 @@ export function passStatement(stmt: ts.Statement, ctx: Context, builder: llvm.IR
         case ts.SyntaxKind.ForStatement:
             passForStatement(<any>stmt, ctx, builder);
             break;
+        case ts.SyntaxKind.DoStatement:
+            passDoStatement(<any>stmt, ctx, builder);
+            break;
         case ts.SyntaxKind.BinaryExpression:
             buildFromBinaryExpression(<any>stmt, ctx, builder);
             break;
