Feature(compiler): Initial support for TypedArray(s) inside backend

ovr · ovr · commit a2fbb0d48f6c · 2018-12-16T15:23:04.000+09:00
diff --git a/src/backend/llvm/code-generation/array-literal-expression.ts b/src/backend/llvm/code-generation/array-literal-expression.ts
@@ -11,21 +11,29 @@ export class ArrayLiteralExpressionCodeGenerator implements NodeGenerateInterfac
         // const elementType = NativeTypeResolver.getType(type, ctx);
 
         // @todo Fix this hardcore...
-        const elementType = llvm.Type.getDoublePtrTy(ctx.llvmContext);
-
-        const structType = llvm.StructType.create(ctx.llvmContext, 'array<int>');
-        structType.setBody([
-            elementType,
-            llvm.Type.getInt32Ty(ctx.llvmContext),
-        ]);
+        const structType = ArrayLiteralExpressionCodeGenerator.buildTypedArrayStructLLVMType(
+            llvm.Type.getDoubleTy(ctx.llvmContext),
+            ctx,
+            'array<double>'
+        );
 
         const allocate = builder.createAlloca(
             structType
         );
 
         return new ArrayReference(
-            elementType,
+            structType.getElementType(0),
             allocate
         );
     }
+
+    static buildTypedArrayStructLLVMType(elementType: llvm.Type, ctx: Context, name: string): llvm.StructType {
+        const structType = llvm.StructType.create(ctx.llvmContext, name);
+        structType.setBody([
+            elementType,
+            llvm.Type.getInt32Ty(ctx.llvmContext),
+        ]);
+
+        return structType;
+    }
 }
diff --git a/src/backend/llvm/context.ts b/src/backend/llvm/context.ts
@@ -2,6 +2,7 @@
 import * as ts from "typescript";
 import * as llvm from "llvm-node";
 import {Scope} from "./scope";
+import {ArrayLiteralExpressionCodeGenerator} from "./code-generation/array-literal-expression";
 
 export class SignatureToFunctionTable extends Map<ts.Signature, llvm.Function> {
 
@@ -20,5 +21,53 @@ export class Context {
         this.typeChecker = typeChecker;
         this.llvmContext = new llvm.LLVMContext();
         this.llvmModule = new llvm.Module("test", this.llvmContext);
+
+        this.scope.classes.set('Uint8Array', ArrayLiteralExpressionCodeGenerator.buildTypedArrayStructLLVMType(
+            llvm.Type.getInt8Ty(this.llvmContext),
+            this,
+            'array<uint8>'
+        ));
+
+        this.scope.classes.set('Int8Array', ArrayLiteralExpressionCodeGenerator.buildTypedArrayStructLLVMType(
+            llvm.Type.getInt8Ty(this.llvmContext),
+            this,
+            'array<int8>'
+        ));
+
+        this.scope.classes.set('Int16Array', ArrayLiteralExpressionCodeGenerator.buildTypedArrayStructLLVMType(
+            llvm.Type.getInt16Ty(this.llvmContext),
+            this,
+            'array<int16>'
+        ));
+
+        this.scope.classes.set('Uint16Array', ArrayLiteralExpressionCodeGenerator.buildTypedArrayStructLLVMType(
+            llvm.Type.getInt16Ty(this.llvmContext),
+            this,
+            'array<uint16>'
+        ));
+
+        this.scope.classes.set('Int32Array', ArrayLiteralExpressionCodeGenerator.buildTypedArrayStructLLVMType(
+            llvm.Type.getInt32Ty(this.llvmContext),
+            this,
+            'array<int32>'
+        ));
+
+        this.scope.classes.set('Uint32Array', ArrayLiteralExpressionCodeGenerator.buildTypedArrayStructLLVMType(
+            llvm.Type.getInt32Ty(this.llvmContext),
+            this,
+            'array<uint32>'
+        ));
+
+        this.scope.classes.set('Float32Array', ArrayLiteralExpressionCodeGenerator.buildTypedArrayStructLLVMType(
+            llvm.Type.getFloatTy(this.llvmContext),
+            this,
+            'array<float32>'
+        ));
+
+        this.scope.classes.set('Float64Array', ArrayLiteralExpressionCodeGenerator.buildTypedArrayStructLLVMType(
+            llvm.Type.getDoubleTy(this.llvmContext),
+            this,
+            'array<float64>'
+        ));
     }
 }
diff --git a/tests/snapshots/general/typed.array.ts b/tests/snapshots/general/typed.array.ts
@@ -0,0 +1,14 @@
+
+{
+    let int8Array = new Int8Array(5);
+    let uint8Array = new Uint8Array(5);
+
+    let int16Array = new Int16Array(5);
+    let uint16Array = new Uint16Array(5);
+
+    let int32Array = new Int32Array(5);
+    let uint32Array = new Uint32Array(5);
+
+    let float32Array = new Float32Array(5);
+    let float64Array = new Float64Array(5);
+}
