fluencelabs/assemblyscript
1
0
Fork 0
mirror of https://github.com/fluencelabs/assemblyscript synced 2025-07-06 18:11:46 +00:00
Code Issues Projects Releases Wiki Activity

Export the trampoline and provide a helper for functions that take optional arguments

Browse Source
This commit is contained in:
dcodeIO
2018-04-17 16:04:29 +02:00
parent 4929fca363
commit 4026c087fd
22 changed files with 296 additions and 206 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
src/compiler.ts
View File

@ -227,7 +227,9 @@ export class Compiler extends DiagnosticEmitter {
/** Function table being compiled. */
functionTable: Function[] = new Array();
/** Argument count helper global. */
argumentCountRef: GlobalRef = 0;
argcVar: GlobalRef = 0;
/** Argument count helper setter. */
argcSet: FunctionRef = 0;
/** Compiles a {@link Program} to a {@link Module} using the specified options. */
static compile(program: Program, options: Options | null = null): Module {
@ -923,6 +925,22 @@ export class Compiler extends DiagnosticEmitter {
// check module-level export
if (instance.is(CommonFlags.MODULE_EXPORT)) {
if (signature.requiredParameters < signature.parameterTypes.length) {
// export the trampoline if the function takes optional parameters
instance = this.ensureTrampoline(instance);
if (!this.argcSet) {
this.ensureArgumentCount(0);
this.argcSet = module.addFunction("~setargc",
this.ensureFunctionType([ Type.u32 ], Type.void),
null,
module.createSetGlobal("~argc",
module.createGetLocal(0, NativeType.I32)
)
);
// export a helper to set argc prior to calling it
module.addFunctionExport("~setargc", "_setargc");
}
}
module.addFunctionExport(instance.internalName, mangleExportName(instance));
}
@ -5035,10 +5053,10 @@ export class Compiler extends DiagnosticEmitter {
minArguments
? module.createBinary(
BinaryOp.SubI32,
module.createGetGlobal("argumentCount", NativeType.I32),
module.createGetGlobal("~argc", NativeType.I32),
module.createI32(minArguments)
)
: module.createGetGlobal("argumentCount", NativeType.I32)
: module.createGetGlobal("~argc", NativeType.I32)
)
]),
module.createUnreachable()
@ -5211,18 +5229,18 @@ export class Compiler extends DiagnosticEmitter {
], returnType.toNativeType());
}
/** Makes sure that the `argumentCount` helper global is present and returns an expression that sets it. */
/** Makes sure that the argument count helper global is present and returns an expression that sets it. */
private ensureArgumentCount(argumentCount: i32): ExpressionRef {
var module = this.module;
if (!this.argumentCountRef) {
this.argumentCountRef = module.addGlobal(
"argumentCount",
if (!this.argcVar) {
this.argcVar = module.addGlobal(
"~argc",
NativeType.I32,
true,
module.createI32(0)
);
}
return module.createSetGlobal("argumentCount", module.createI32(argumentCount));
return module.createSetGlobal("~argc", module.createI32(argumentCount));
}
compileCommaExpression(expression: CommaExpression, contextualType: Type): ExpressionRef {