org.ssssssss.script.parsing.ast.statement.MethodCall Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of magic-script Show documentation

magic script

The newest version!

package org.ssssssss.script.parsing.ast.statement;

import org.ssssssss.script.compile.MagicScriptCompiler;
import org.ssssssss.script.parsing.Span;
import org.ssssssss.script.parsing.ast.Expression;

import java.util.List;

public class MethodCall extends Expression {
	private final MemberAccess method;
	private final List arguments;
	private final boolean inLinq;

	public MethodCall(Span span, MemberAccess method, List arguments, boolean inLinq) {
		super(span);
		this.method = method;
		this.arguments = arguments;
		this.inLinq = inLinq;
	}

	public MemberAccess getMethod() {
		return method;
	}

	public List getArguments() {
		return arguments;
	}

	@Override
	public void visitMethod(MagicScriptCompiler compiler) {
		method.visitMethod(compiler);
		arguments.forEach(it -> it.visitMethod(compiler));
	}


	@Override
	public void compile(MagicScriptCompiler compiler) {
		int size = arguments.size();
		compiler.newRuntimeContext()
				.ldc(method.getName().getText())    // 方法名
				.insn(arguments.stream().anyMatch(it -> it instanceof Spread) ? ICONST_1 : ICONST_0)    // 是否是 (...xxx)
				.asBoolean()
				.insn(method.isOptional() ? ICONST_1 : ICONST_0)    // 是否允许可空调用
				.asBoolean()
				.visitInt(size)
				.typeInsn(ANEWARRAY, Object.class);
		for (int i = 0; i < size; i++) {
			Expression argument = arguments.get(i);
			compiler.insn(DUP).visitInt(i);
			if (inLinq && argument instanceof MemberAccess) {
				((MemberAccess) argument).compileLinq(compiler);
			} else {
				compiler.visit(argument);
			}
			compiler.insn(AASTORE);
		}
		compiler.visit(method.getObject())    // 访问目标对象
				.lineNumber(new Span(getSpan().getSource(), method.getName().getStart(), getSpan().getEnd()))
				.call("invoke_method", 6);    // 调用方法
	}
}