/*
* Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
package org.graalvm.compiler.replacements.amd64;
import static org.graalvm.compiler.nodeinfo.NodeCycles.CYCLES_1;
import org.graalvm.compiler.core.common.type.FloatStamp;
import org.graalvm.compiler.core.common.type.Stamp;
import org.graalvm.compiler.debug.GraalError;
import org.graalvm.compiler.graph.NodeClass;
import org.graalvm.compiler.graph.spi.CanonicalizerTool;
import org.graalvm.compiler.lir.amd64.AMD64ArithmeticLIRGeneratorTool;
import org.graalvm.compiler.lir.amd64.AMD64ArithmeticLIRGeneratorTool.RoundingMode;
import org.graalvm.compiler.lir.gen.ArithmeticLIRGeneratorTool;
import org.graalvm.compiler.nodeinfo.NodeInfo;
import org.graalvm.compiler.nodes.ConstantNode;
import org.graalvm.compiler.nodes.ValueNode;
import org.graalvm.compiler.nodes.calc.UnaryNode;
import org.graalvm.compiler.nodes.spi.ArithmeticLIRLowerable;
import org.graalvm.compiler.nodes.spi.NodeLIRBuilderTool;
import jdk.vm.ci.meta.JavaConstant;
import jdk.vm.ci.meta.JavaKind;
/**
* Round floating-point value.
*/
@NodeInfo(cycles = CYCLES_1)
public final class AMD64RoundNode extends UnaryNode implements ArithmeticLIRLowerable {
public static final NodeClass<AMD64RoundNode> TYPE = NodeClass.create(AMD64RoundNode.class);
private final RoundingMode mode;
public AMD64RoundNode(ValueNode value, RoundingMode mode) {
super(TYPE, roundStamp((FloatStamp) value.stamp(), mode), value);
this.mode = mode;
}
private static double round(RoundingMode mode, double input) {
switch (mode) {
case DOWN:
return Math.floor(input);
case NEAREST:
return Math.rint(input);
case UP:
return Math.ceil(input);
case TRUNCATE:
return (long) input;
default:
throw GraalError.unimplemented("unimplemented RoundingMode " + mode);
}
}
private static FloatStamp roundStamp(FloatStamp stamp, RoundingMode mode) {
double min = stamp.lowerBound();
min = Math.min(min, round(mode, min));
double max = stamp.upperBound();
max = Math.max(max, round(mode, max));
return new FloatStamp(stamp.getBits(), min, max, stamp.isNonNaN());
}
@Override
public Stamp foldStamp(Stamp newStamp) {
assert newStamp.isCompatible(getValue().stamp());
return roundStamp((FloatStamp) newStamp, mode);
}
public ValueNode tryFold(ValueNode input) {
if (input.isConstant()) {
JavaConstant c = input.asJavaConstant();
if (c.getJavaKind() == JavaKind.Double) {
return ConstantNode.forDouble(round(mode, c.asDouble()));
} else if (c.getJavaKind() == JavaKind.Float) {
return ConstantNode.forFloat((float) round(mode, c.asFloat()));
}
}
return null;
}
@Override
public ValueNode canonical(CanonicalizerTool tool, ValueNode forValue) {
ValueNode folded = tryFold(forValue);
return folded != null ? folded : this;
}
@Override
public void generate(NodeLIRBuilderTool builder, ArithmeticLIRGeneratorTool gen) {
builder.setResult(this, ((AMD64ArithmeticLIRGeneratorTool) gen).emitRound(builder.operand(getValue()), mode));
}
}