package org.graalvm.compiler.core.test.ea;
import java.util.List;
import org.graalvm.compiler.core.test.GraalCompilerTest;
import org.graalvm.compiler.debug.DebugContext;
import org.graalvm.compiler.nodes.ReturnNode;
import org.graalvm.compiler.nodes.StructuredGraph;
import org.graalvm.compiler.nodes.StructuredGraph.AllowAssumptions;
import org.graalvm.compiler.nodes.java.NewArrayNode;
import org.graalvm.compiler.nodes.java.NewInstanceNode;
import org.graalvm.compiler.nodes.virtual.CommitAllocationNode;
import org.graalvm.compiler.phases.common.CanonicalizerPhase;
import org.graalvm.compiler.phases.common.DeadCodeEliminationPhase;
import org.graalvm.compiler.phases.common.inlining.InliningPhase;
import org.graalvm.compiler.phases.tiers.HighTierContext;
import org.graalvm.compiler.virtual.phases.ea.PartialEscapePhase;
import org.junit.Assert;
import jdk.vm.ci.meta.JavaConstant;
import jdk.vm.ci.meta.ResolvedJavaMethod;
public class EATestBase extends GraalCompilerTest {
public static class TestClassInt {
public int x;
public int y;
public int z;
public TestClassInt() {
this(0, 0);
}
public TestClassInt(int x) {
this(x, 0);
}
public TestClassInt(int x, int y) {
this.x = x;
this.y = y;
}
@Override
public boolean equals(Object obj) {
TestClassInt other = (TestClassInt) obj;
return x == other.x && y == other.y && z == other.z;
}
@Override
public String toString() {
return "{" + x + "," + y + "," + z + "}";
}
@Override
public int hashCode() {
return x + 13 * y;
}
}
public static class TestClassObject {
public Object x;
public Object y;
public TestClassObject() {
this(null, null);
}
public TestClassObject(Object x) {
this(x, null);
}
public TestClassObject(Object x, Object y) {
this.x = x;
this.y = y;
}
@Override
public boolean equals(Object obj) {
TestClassObject other = (TestClassObject) obj;
return x == other.x && y == other.y;
}
@Override
public String toString() {
return "{" + x + "," + y + "}";
}
@Override
public int hashCode() {
return (x == null ? 0 : x.hashCode()) + 13 * (y == null ? 0 : y.hashCode());
}
}
protected static native void notInlineable();
protected StructuredGraph graph;
protected HighTierContext context;
protected List<ReturnNode> returnNodes;
protected void testEscapeAnalysis(String snippet, JavaConstant expectedConstantResult, boolean iterativeEscapeAnalysis) {
prepareGraph(snippet, iterativeEscapeAnalysis);
if (expectedConstantResult != null) {
for (ReturnNode returnNode : returnNodes) {
Assert.assertTrue(returnNode.result().toString(), returnNode.result().isConstant());
Assert.assertEquals(expectedConstantResult, returnNode.result().asConstant());
}
}
int newInstanceCount = graph.getNodes().filter(NewInstanceNode.class).count() + graph.getNodes().filter(NewArrayNode.class).count() +
graph.getNodes().filter(CommitAllocationNode.class).count();
Assert.assertEquals(0, newInstanceCount);
}
@SuppressWarnings("try")
protected void prepareGraph(String snippet, boolean iterativeEscapeAnalysis) {
ResolvedJavaMethod method = getResolvedJavaMethod(snippet);
DebugContext debug = getDebugContext();
try (DebugContext.Scope s = debug.scope(getClass(), method, getCodeCache())) {
graph = parseEager(method, AllowAssumptions.YES, debug);
context = getDefaultHighTierContext();
new InliningPhase(new CanonicalizerPhase()).apply(graph, context);
new DeadCodeEliminationPhase().apply(graph);
canonicalizeGraph();
new PartialEscapePhase(iterativeEscapeAnalysis, false, new CanonicalizerPhase(), null, graph.getOptions()).apply(graph, context);
postEACanonicalizeGraph();
returnNodes = graph.getNodes(ReturnNode.TYPE).snapshot();
} catch (Throwable e) {
throw debug.handle(e);
}
}
protected void postEACanonicalizeGraph() {
}
protected void canonicalizeGraph() {
new CanonicalizerPhase().apply(graph, context);
}
}