package com.oracle.truffle.js.nodes.function;
import com.oracle.truffle.api.frame.VirtualFrame;
import com.oracle.truffle.api.instrumentation.Tag;
import com.oracle.truffle.api.nodes.ExplodeLoop;
import com.oracle.truffle.api.profiles.BranchProfile;
import com.oracle.truffle.js.nodes.JavaScriptNode;
import com.oracle.truffle.js.nodes.instrumentation.JSInputGeneratingNodeWrapper;
import com.oracle.truffle.js.runtime.JSConfig;
import com.oracle.truffle.js.runtime.JSContext;
import com.oracle.truffle.js.runtime.util.SimpleArrayList;
import java.util.Set;
class JSFunctionArgumentsNode extends AbstractFunctionArgumentsNode {
@Children protected final JavaScriptNode[] args;
public static AbstractFunctionArgumentsNode create(JSContext context, JavaScriptNode[] args) {
assert args.length <= context.getFunctionArgumentsLimit();
for (JavaScriptNode arg : args) {
if (arg instanceof SpreadArgumentNode) {
return new SpreadFunctionArgumentsNode(args);
}
}
if (args.length == 0) {
return new JSFunctionZeroArgumentsNode();
} else if (args.length == 1) {
return JSFunctionOneArgumentNode.create(args[0]);
}
return new JSFunctionArgumentsNode(args);
}
protected JSFunctionArgumentsNode(JavaScriptNode[] args) {
this.args = args;
}
@Override
public int getCount(VirtualFrame frame) {
return args.length;
}
@Override
@ExplodeLoop
public Object[] executeFillObjectArray(VirtualFrame frame, Object[] arguments, int delta) {
for (int i = 0; i < args.length; i++) {
arguments[i + delta] = args[i].execute(frame);
}
return arguments;
}
@Override
protected AbstractFunctionArgumentsNode copyUninitialized(Set<Class<? extends Tag>> materializedTags) {
return new JSFunctionArgumentsNode(JavaScriptNode.cloneUninitialized(args, materializedTags));
}
@Override
public void materializeInstrumentableArguments() {
for (int i = 0; i < args.length; i++) {
if (!(args[i] instanceof SpreadArgumentNode) && !args[i].isInstrumentable()) {
args[i] = insert(JSInputGeneratingNodeWrapper.create(args[i]));
}
}
}
}
class JSFunctionZeroArgumentsNode extends AbstractFunctionArgumentsNode {
protected JSFunctionZeroArgumentsNode() {
}
@Override
public int getCount(VirtualFrame frame) {
return 0;
}
@Override
public Object[] executeFillObjectArray(VirtualFrame frame, Object[] arguments, int delta) {
return arguments;
}
@Override
protected AbstractFunctionArgumentsNode copyUninitialized(Set<Class<? extends Tag>> materializedTags) {
return new JSFunctionZeroArgumentsNode();
}
@Override
public void materializeInstrumentableArguments() {
}
}
class JSFunctionOneArgumentNode extends AbstractFunctionArgumentsNode {
@Child private JavaScriptNode child;
protected JSFunctionOneArgumentNode(JavaScriptNode child) {
this.child = child;
}
public static AbstractFunctionArgumentsNode create(JavaScriptNode child) {
return new JSFunctionOneArgumentNode(child);
}
public static AbstractFunctionArgumentsNode create(JavaScriptNode child, boolean optimizeConstantArguments) {
if (optimizeConstantArguments) {
return create(child);
} else {
return new JSFunctionOneArgumentNode(child);
}
}
@Override
public int getCount(VirtualFrame frame) {
return 1;
}
@Override
public Object[] executeFillObjectArray(VirtualFrame frame, Object[] arguments, int delta) {
arguments[delta] = child.execute(frame);
return arguments;
}
@Override
protected AbstractFunctionArgumentsNode copyUninitialized(Set<Class<? extends Tag>> materializedTags) {
return new JSFunctionOneArgumentNode(JavaScriptNode.cloneUninitialized(child, materializedTags));
}
@Override
public void materializeInstrumentableArguments() {
if (!child.isInstrumentable()) {
child = insert(JSInputGeneratingNodeWrapper.create(child));
}
}
}
class SpreadFunctionArgumentsNode extends JSFunctionArgumentsNode {
private final BranchProfile growProfile = BranchProfile.create();
protected SpreadFunctionArgumentsNode(JavaScriptNode[] args) {
super(args);
}
@Override
public int getCount(VirtualFrame frame) {
return args.length;
}
@Override
@ExplodeLoop
public Object[] executeFillObjectArray(VirtualFrame frame, Object[] arguments, int fixedArgumentsLength) {
SimpleArrayList<Object> argList = SimpleArrayList.create((long) fixedArgumentsLength + args.length + JSConfig.SpreadArgumentPlaceholderCount);
for (int i = 0; i < fixedArgumentsLength; i++) {
argList.addUnchecked(arguments[i]);
}
for (int i = 0; i < args.length; i++) {
if (args[i] instanceof SpreadArgumentNode) {
((SpreadArgumentNode) args[i]).executeToList(frame, argList, growProfile);
} else {
argList.add(args[i].execute(frame), growProfile);
}
}
return argList.toArray();
}
@Override
protected AbstractFunctionArgumentsNode copyUninitialized(Set<Class<? extends Tag>> materializedTags) {
return new SpreadFunctionArgumentsNode(JavaScriptNode.cloneUninitialized(args, materializedTags));
}
}