package org.graalvm.compiler.core.aarch64;
import org.graalvm.compiler.asm.aarch64.AArch64Address;
import org.graalvm.compiler.asm.aarch64.AArch64Address.AddressingMode;
import org.graalvm.compiler.core.common.LIRKind;
import org.graalvm.compiler.graph.NodeClass;
import org.graalvm.compiler.lir.aarch64.AArch64AddressValue;
import org.graalvm.compiler.lir.gen.LIRGeneratorTool;
import org.graalvm.compiler.nodeinfo.NodeInfo;
import org.graalvm.compiler.nodes.NodeView;
import org.graalvm.compiler.nodes.ValueNode;
import org.graalvm.compiler.nodes.memory.address.AddressNode;
import org.graalvm.compiler.nodes.spi.LIRLowerable;
import org.graalvm.compiler.nodes.spi.NodeLIRBuilderTool;
import jdk.vm.ci.meta.AllocatableValue;
import jdk.vm.ci.meta.Value;
@NodeInfo
public class AArch64AddressNode extends AddressNode implements LIRLowerable {
public static final NodeClass<AArch64AddressNode> TYPE = NodeClass.create(AArch64AddressNode.class);
@OptionalInput private ValueNode base;
@OptionalInput private ValueNode index;
private AArch64Address.AddressingMode addressingMode;
private long displacement;
private int scaleFactor;
public AArch64AddressNode(ValueNode base) {
this(base, null);
}
public AArch64AddressNode(ValueNode base, ValueNode index) {
super(TYPE);
this.base = base;
this.index = index;
this.addressingMode = AddressingMode.REGISTER_OFFSET;
this.displacement = 0;
this.scaleFactor = 1;
}
@Override
public void generate(NodeLIRBuilderTool gen) {
LIRGeneratorTool tool = gen.getLIRGeneratorTool();
AllocatableValue baseValue = base == null ? Value.ILLEGAL : tool.asAllocatable(gen.operand(base));
AllocatableValue indexValue = index == null ? Value.ILLEGAL : tool.asAllocatable(gen.operand(index));
AllocatableValue baseReference = LIRKind.derivedBaseFromValue(baseValue);
AllocatableValue indexReference;
if (index == null) {
indexReference = null;
} else if (addressingMode.equals(AddressingMode.IMMEDIATE_UNSCALED)) {
indexReference = LIRKind.derivedBaseFromValue(indexValue);
} else {
if (LIRKind.isValue(indexValue.getValueKind())) {
indexReference = null;
} else {
indexReference = Value.ILLEGAL;
}
}
LIRKind kind = LIRKind.combineDerived(tool.getLIRKind(stamp(NodeView.DEFAULT)), baseReference, indexReference);
gen.setResult(this, new AArch64AddressValue(kind, baseValue, indexValue, (int) displacement, scaleFactor, addressingMode));
}
@Override
public ValueNode getBase() {
return base;
}
public void setBase(ValueNode base) {
if (isAlive()) {
updateUsages(this.base, base);
}
this.base = base;
}
@Override
public ValueNode getIndex() {
return index;
}
public void setIndex(ValueNode index) {
if (isAlive()) {
updateUsages(this.index, index);
}
this.index = index;
}
public long getDisplacement() {
return displacement;
}
public void setDisplacement(long displacement, int scaleFactor, AArch64Address.AddressingMode addressingMode) {
this.displacement = displacement;
this.scaleFactor = scaleFactor;
this.addressingMode = addressingMode;
}
@Override
public long getMaxConstantDisplacement() {
return displacement;
}
public AddressingMode getAddressingMode() {
return addressingMode;
}
}