Feature(compiler): Support BinaryExpression - equals

ovr · ovr · commit abc86c8e24c2 · 2019-01-03T20:53:12.000+09:00
diff --git a/src/backend/llvm/code-generation/binary-expression.ts b/src/backend/llvm/code-generation/binary-expression.ts
@@ -89,6 +89,18 @@ export class BinaryExpressionCodeGenerator implements NodeGenerateInterface<ts.B
                     )
                 );
             }
+            // a == b
+            case ts.SyntaxKind.EqualsEqualsToken: {
+                const left = buildFromExpression(node.left, ctx, builder);
+                const right = buildFromExpression(node.right, ctx, builder);
+
+                return new Primitive(
+                    builder.createFCmpOEQ(
+                        loadIfNeeded(left, builder),
+                        loadIfNeeded(right, builder)
+                    )
+                );
+            }
             // a ^ b
             case ts.SyntaxKind.CaretEqualsToken:
             case ts.SyntaxKind.CaretToken: {
@@ -212,4 +224,4 @@ export class BinaryExpressionCodeGenerator implements NodeGenerateInterface<ts.B
                 );
         }
     }
-}
+}
diff --git a/src/backend/llvm/index.ts b/src/backend/llvm/index.ts
@@ -75,7 +75,6 @@ export function passFunctionDeclaration(parent: ts.FunctionDeclaration, ctx: Con
     );
     let fn = llvm.Function.create(fnType, llvm.LinkageTypes.ExternalLinkage, <string>parent.name.escapedText, ctx.llvmModule);
 
-
     let block = llvm.BasicBlock.create(ctx.llvmContext, 'Entry', fn);
     let irBuilder = new llvm.IRBuilder(block);
 
diff --git a/tests/snapshots/math/binary-expression.ts b/tests/snapshots/math/binary-expression.ts
@@ -1,5 +1,19 @@
 
 {
+    function equalTwoNumbers(left: number, right: number): void {
+        console_log("equalTwoNumbers");
+        console_log(left);
+        console_log(right);
+
+        if (left == right) {
+            console_log("true");
+        } else {
+            console_log("false");
+        }
+
+        console_log("");
+    }
+
     function compareNumberLeftLessRight(left: number, right: number): void {
         console_log("compareNumberLeftLessRight");
         console_log(left);
@@ -28,6 +42,9 @@
         console_log("");
     }
 
+    equalTwoNumbers(1, 1);
+    equalTwoNumbers(1, 0);
+
     compareNumberLeftLessRight(1, 100);
     compareNumberLeftLessRight(101, 100);
 
diff --git a/tests/snapshots/math/binary-expression.ts.stdout b/tests/snapshots/math/binary-expression.ts.stdout
@@ -1,3 +1,13 @@
+equalTwoNumbers
+1.000000
+1.000000
+true
+
+equalTwoNumbers
+1.000000
+0.000000
+false
+
 compareNumberLeftLessRight
 1.000000
 100.000000

Original file line number	Diff line number	Diff line change
`@@ -89,6 +89,18 @@ export class BinaryExpressionCodeGenerator implements NodeGenerateInterface<ts.B`
`89`	`89`	`)`
`90`	`90`	`);`
`91`	`91`	`}`
	`92`	`+ // a == b`
	`93`	`+ case ts.SyntaxKind.EqualsEqualsToken: {`
	`94`	`+ const left = buildFromExpression(node.left, ctx, builder);`
	`95`	`+ const right = buildFromExpression(node.right, ctx, builder);`
	`96`	`+`
	`97`	`+ return new Primitive(`
	`98`	`+ builder.createFCmpOEQ(`
	`99`	`+ loadIfNeeded(left, builder),`
	`100`	`+ loadIfNeeded(right, builder)`
	`101`	`+ )`
	`102`	`+ );`
	`103`	`+ }`
`92`	`104`	`// a ^ b`
`93`	`105`	`case ts.SyntaxKind.CaretEqualsToken:`
`94`	`106`	`case ts.SyntaxKind.CaretToken: {`
`@@ -212,4 +224,4 @@ export class BinaryExpressionCodeGenerator implements NodeGenerateInterface<ts.B`
`212`	`224`	`);`
`213`	`225`	`}`
`214`	`226`	`}`
`215`		`-}`
	`227`	`+}`