package com.oracle.truffle.js.nodes.binary;
import com.oracle.truffle.api.dsl.Cached;
import com.oracle.truffle.api.dsl.Specialization;
import com.oracle.truffle.api.instrumentation.Tag;
import com.oracle.truffle.api.nodes.NodeInfo;
import com.oracle.truffle.api.profiles.BranchProfile;
import com.oracle.truffle.js.nodes.JavaScriptNode;
import com.oracle.truffle.js.nodes.Truncatable;
import com.oracle.truffle.js.nodes.access.JSConstantNode.JSConstantIntegerNode;
import com.oracle.truffle.js.nodes.cast.JSToInt32Node;
import com.oracle.truffle.js.nodes.cast.JSToNumericNode;
import com.oracle.truffle.js.nodes.cast.JSToUInt32Node;
import com.oracle.truffle.js.runtime.BigInt;
import com.oracle.truffle.js.runtime.JSConfig;
import java.util.Set;
@NodeInfo(shortName = ">>")
public abstract class JSRightShiftNode extends JSBinaryNode {
protected JSRightShiftNode(JavaScriptNode left, JavaScriptNode right) {
super(left, right);
}
public static JavaScriptNode create(JavaScriptNode left, JavaScriptNode right) {
Truncatable.truncate(left);
Truncatable.truncate(right);
if (JSConfig.UseSuperOperations && (right instanceof JSConstantIntegerNode)) {
return JSRightShiftConstantNode.create(left, right);
}
return JSRightShiftNodeGen.create(left, right);
}
public abstract Object execute(Object a, Object b);
@Specialization
protected int doInteger(int a, int b) {
return a >> b;
}
@Specialization
protected BigInt doBigInt(BigInt a, BigInt b,
@Cached("create()") JSLeftShiftNode leftShift) {
return leftShift.doBigInt(a, b.negate());
}
@Specialization(guards = "!largerThan2e32(b)")
protected int doIntDouble(int a, double b) {
return a >> (int) ((long) b);
}
@Specialization
protected Object doDouble(double a, double b,
@Cached("create()") JSRightShiftNode rightShift,
@Cached("create()") JSToInt32Node leftInt32,
@Cached("create()") JSToUInt32Node rightUInt32) {
return rightShift.execute(leftInt32.executeInt(a), rightUInt32.execute(b));
}
@Specialization(replaces = {"doInteger", "doIntDouble", "doDouble", "doBigInt"})
protected Object doGeneric(Object a, Object b,
@Cached("create()") JSRightShiftNode rightShift,
@Cached("create()") JSToNumericNode leftToNumeric,
@Cached("create()") JSToNumericNode rightToNumeric,
@Cached("create()") BranchProfile mixedNumericTypes) {
Object operandA = leftToNumeric.execute(a);
Object operandB = rightToNumeric.execute(b);
ensureBothSameNumericType(operandA, operandB, mixedNumericTypes);
return rightShift.execute(operandA, operandB);
}
public static JSRightShiftNode create() {
return JSRightShiftNodeGen.create(null, null);
}
@Override
public boolean isResultAlwaysOfType(Class<?> clazz) {
return clazz == int.class;
}
@Override
protected JavaScriptNode copyUninitialized(Set<Class<? extends Tag>> materializedTags) {
return JSRightShiftNodeGen.create(cloneUninitialized(getLeft(), materializedTags), cloneUninitialized(getRight(), materializedTags));
}
}