package org.graalvm.compiler.nodes.calc;
import static org.graalvm.compiler.nodeinfo.NodeCycles.CYCLES_2;
import static org.graalvm.compiler.nodeinfo.NodeSize.SIZE_1;
import org.graalvm.compiler.core.common.type.ArithmeticOpTable;
import org.graalvm.compiler.core.common.type.ArithmeticOpTable.UnaryOp.Neg;
import org.graalvm.compiler.core.common.type.FloatStamp;
import org.graalvm.compiler.core.common.type.Stamp;
import org.graalvm.compiler.graph.NodeClass;
import org.graalvm.compiler.graph.spi.CanonicalizerTool;
import org.graalvm.compiler.lir.gen.ArithmeticLIRGeneratorTool;
import org.graalvm.compiler.nodeinfo.NodeInfo;
import org.graalvm.compiler.nodes.NodeView;
import org.graalvm.compiler.nodes.ValueNode;
import org.graalvm.compiler.nodes.spi.NodeLIRBuilderTool;
import org.graalvm.compiler.nodes.spi.StampInverter;
@NodeInfo(cycles = CYCLES_2, size = SIZE_1)
public final class NegateNode extends UnaryArithmeticNode<Neg> implements NarrowableArithmeticNode, StampInverter {
public static final NodeClass<NegateNode> TYPE = NodeClass.create(NegateNode.class);
public NegateNode(ValueNode value) {
super(TYPE, ArithmeticOpTable::getNeg, value);
}
public static ValueNode create(ValueNode value, NodeView view) {
ValueNode synonym = findSynonym(value, view);
if (synonym != null) {
return synonym;
}
return new NegateNode(value);
}
@Override
public ValueNode canonical(CanonicalizerTool tool, ValueNode forValue) {
ValueNode synonym = findSynonym(forValue, getOp(forValue));
if (synonym != null) {
return synonym;
}
return this;
}
protected static ValueNode findSynonym(ValueNode forValue, NodeView view) {
ArithmeticOpTable.UnaryOp<Neg> negOp = ArithmeticOpTable.forStamp(forValue.stamp(view)).getNeg();
ValueNode synonym = UnaryArithmeticNode.findSynonym(forValue, negOp);
if (synonym != null) {
return synonym;
}
if (forValue instanceof NegateNode) {
return ((NegateNode) forValue).getValue();
}
if (forValue instanceof SubNode && !(forValue.stamp(view) instanceof FloatStamp)) {
SubNode sub = (SubNode) forValue;
return SubNode.create(sub.getY(), sub.getX(), view);
}
return null;
}
@Override
public void generate(NodeLIRBuilderTool nodeValueMap, ArithmeticLIRGeneratorTool gen) {
nodeValueMap.setResult(this, gen.emitNegate(nodeValueMap.operand(getValue())));
}
@Override
public Stamp invertStamp(Stamp outStamp) {
return getArithmeticOp().foldStamp(outStamp);
}
}