Feature(compiler): ContinueStatement - initial support

ovr · ovr · commit 7ad996f18f3c · 2018-12-08T05:28:41.000+09:00
diff --git a/src/backend/llvm/code-generation/continue-statement.ts b/src/backend/llvm/code-generation/continue-statement.ts
@@ -0,0 +1,26 @@
+
+import * as ts from "typescript";
+import * as llvm from 'llvm-node';
+import {NodeGenerateInterface} from "../node-generate.interface";
+import {Context} from "../context";
+import UnsupportedError from "../../error/unsupported.error";
+
+export class ContinueStatementGenerator implements NodeGenerateInterface<ts.ContinueStatement, void> {
+    generate(node: ts.ContinueStatement, ctx: Context, builder: llvm.IRBuilder): void {
+        if (node.label) {
+            throw new UnsupportedError(
+                node,
+                'Continue with label is unsupported'
+            );
+        }
+
+        if (!ctx.scope.continueBlock) {
+            throw new UnsupportedError(
+                node,
+                'There is no pointer to jump outside this block to continue'
+            );
+        }
+
+        builder.createBr(ctx.scope.continueBlock);
+    }
+}
diff --git a/src/backend/llvm/index.ts b/src/backend/llvm/index.ts
@@ -16,6 +16,7 @@ import {ForStatementGenerator} from "./code-generation/for-statement";
 import {DoStatementGenerator} from "./code-generation/do-statement";
 import {WhileStatementGenerator} from "./code-generation/while-statement";
 import {BreakStatementGenerator} from "./code-generation/break-statement";
+import {ContinueStatementGenerator} from "./code-generation/continue-statement";
 
 export function passIfStatement(parent: ts.IfStatement, ctx: Context, builder: llvm.IRBuilder) {
     const positiveBlock = llvm.BasicBlock.create(ctx.llvmContext, "if.true");
@@ -499,6 +500,9 @@ export function passStatement(stmt: ts.Statement, ctx: Context, builder: llvm.IR
         case ts.SyntaxKind.BreakStatement:
             new BreakStatementGenerator().generate(<any>stmt, ctx, builder);
             break;
+        case ts.SyntaxKind.ContinueStatement:
+            new ContinueStatementGenerator().generate(<any>stmt, ctx, builder);
+            break;
         case ts.SyntaxKind.IfStatement:
             passIfStatement(<any>stmt, ctx, builder);
             break;
diff --git a/tests/snapshots/math/for.ts b/tests/snapshots/math/for.ts
@@ -24,7 +24,22 @@
         return total;
     }
 
+    function simpleContinue(): number {
+        let total: number = 0;
+
+        for (let i = 0; i++; i++) {
+            if (i > 50) {
+                continue;
+            }
+
+            total = total + i;
+        }
+
+        return total;
+    }
+
     calculateTotalFromRange(1, 100);
     calculateTotalFromRange(50, 100);
     simpleBreak();
+    simpleContinue();
 }
