Feature: CPPMangler - make it real

ovr · ovr · commit 3f3f6745e72f · 2018-11-07T23:17:48.000+09:00
diff --git a/src/backend/llvm/cpp.mangler.ts b/src/backend/llvm/cpp.mangler.ts
@@ -3,10 +3,21 @@ import * as ts from 'typescript';
 
 export class CPPMangler {
     static getFunctionName(name: string, parameters: ts.NodeArray<ts.ParameterDeclaration>): string {
-        if (name === 'puts') {
-            return name;
-        }
+        return '_Z' + name.length + name + parameters.map(
+            (parameter) => {
+                if (parameter.type) {
+                    switch (parameter.type.kind) {
+                        case ts.SyntaxKind.NumberKeyword:
+                            return 'd';
+                        default:
+                            throw new Error(
+                                `Unsupported mangling parameter type: ${parameter.type.kind}`
+                            );
+                    }
+                }
 
-        return '_Z13number2stringd';
+                throw new Error('Unsupported mangling without type');
+            }
+        );
     }
 }
diff --git a/src/backend/llvm/index.ts b/src/backend/llvm/index.ts
@@ -324,7 +324,11 @@ export function generateModuleFromProgram(program: ts.Program): llvm.Module {
     llvm.Function.create(
         number2stringFnType,
         llvm.LinkageTypes.ExternalLinkage,
-        CPPMangler.getFunctionName("number2string", <any>[]),
+        CPPMangler.getFunctionName("number2string", <any>[
+            {
+                type: { kind: ts.SyntaxKind.NumberKeyword }
+            }
+        ]),
         ctx.llvmModule
     );
 
