Feature(compiler): PrefixUnaryExpressionCodeGenerator - initial

ovr · ovr · commit 42c4e900b2bd · 2019-01-27T02:44:43.000+09:00
diff --git a/src/backend/llvm/code-generation/prefix-unary-expression.ts b/src/backend/llvm/code-generation/prefix-unary-expression.ts
@@ -0,0 +1,52 @@
+
+import * as ts from "typescript";
+import * as llvm from 'llvm-node';
+import {NodeGenerateInterface} from "../node-generate.interface";
+import {Context} from "../context";
+import {Value} from "../value";
+import UnsupportedError from "../../error";
+import {buildFromExpression, loadIfNeeded} from "../index";
+
+export class PrefixUnaryExpressionCodeGenerator implements NodeGenerateInterface<ts.PrefixUnaryExpression, Value> {
+    generate(node: ts.PrefixUnaryExpression, ctx: Context, builder: llvm.IRBuilder): Value {
+        switch (node.operator) {
+            case ts.SyntaxKind.PlusPlusToken: {
+                const left = buildFromExpression(node.operand, ctx, builder);
+
+                const next = builder.createFAdd(
+                    loadIfNeeded(left, builder),
+                    llvm.ConstantFP.get(ctx.llvmContext, 1)
+                );
+
+                builder.createStore(
+                    next,
+                    left.getValue(),
+                    false
+                );
+
+                return left;
+            }
+            case ts.SyntaxKind.MinusMinusToken: {
+                const left = buildFromExpression(node.operand, ctx, builder);
+
+                const next = builder.createFSub(
+                    loadIfNeeded(left, builder),
+                    llvm.ConstantFP.get(ctx.llvmContext, 1)
+                );
+
+                builder.createStore(
+                    next,
+                    left.getValue(),
+                    false
+                );
+
+                return left;
+            }
+            default:
+                throw new UnsupportedError(
+                    node,
+                    `Unsupported PostfixUnaryExpression.operator: "${node.operator}"`
+                );
+        }
+    }
+}
diff --git a/src/backend/llvm/index.ts b/src/backend/llvm/index.ts
@@ -36,6 +36,7 @@ import {TryStatementGenerator} from "./code-generation/try-statement";
 import {PostfixUnaryExpressionCodeGenerator} from "./code-generation/postfix-unary-expression";
 import {FunctionDeclarationCodeGenerator} from "./code-generation/function-declaration";
 import {TypeOfExpressionCodeGenerator} from "./code-generation/typeof-expression";
+import {PrefixUnaryExpressionCodeGenerator} from "./code-generation/prefix-unary-expression";
 
 export function emitCondition(
     condition: ts.Expression,
@@ -304,6 +305,8 @@ export function buildFromExpression(block: ts.Expression, ctx: Context, builder:
             return buildFromFalseKeyword(<any>block, ctx, builder);
         case ts.SyntaxKind.BinaryExpression:
             return new BinaryExpressionCodeGenerator().generate(<any>block, ctx, builder);
+        case ts.SyntaxKind.PrefixUnaryExpression:
+            return new PrefixUnaryExpressionCodeGenerator().generate(block as ts.PrefixUnaryExpression, ctx, builder);
         case ts.SyntaxKind.PostfixUnaryExpression:
             return new PostfixUnaryExpressionCodeGenerator().generate(block as ts.PostfixUnaryExpression, ctx, builder);
         case ts.SyntaxKind.CallExpression:
