package org.graalvm.compiler.core.match;
import static org.graalvm.compiler.debug.DebugOptions.LogVerbose;
import java.util.ArrayList;
import java.util.List;
import jdk.internal.vm.compiler.collections.EconomicMap;
import jdk.internal.vm.compiler.collections.Equivalence;
import jdk.internal.vm.compiler.collections.MapCursor;
import org.graalvm.compiler.core.gen.NodeMatchRules;
import org.graalvm.compiler.debug.DebugContext;
import org.graalvm.compiler.debug.GraalError;
import org.graalvm.compiler.graph.Edges;
import org.graalvm.compiler.graph.Node;
import org.graalvm.compiler.graph.NodeClass;
import org.graalvm.compiler.graph.Position;
import org.graalvm.compiler.options.OptionValues;
import org.graalvm.compiler.serviceprovider.GraalServices;
public class MatchRuleRegistry {
public static Position[] findPositions(NodeClass<? extends Node> nodeClass, String[] names) {
Position[] result = new Position[names.length];
for (int i = 0; i < names.length; i++) {
Edges edges = nodeClass.getInputEdges();
for (int e = 0; e < edges.getDirectCount(); e++) {
if (names[i].equals(edges.getName(e))) {
result[i] = new Position(edges, e, Node.NOT_ITERABLE);
}
}
if (result[i] == null) {
throw new GraalError("unknown field \"%s\" in class %s", names[i], nodeClass);
}
}
return result;
}
private static final EconomicMap<Class<? extends NodeMatchRules>, EconomicMap<Class<? extends Node>, List<MatchStatement>>> registry = EconomicMap.create(Equivalence.IDENTITY);
@SuppressWarnings("try")
public static synchronized EconomicMap<Class<? extends Node>, List<MatchStatement>> lookup(Class<? extends NodeMatchRules> theClass, OptionValues options, DebugContext debug) {
EconomicMap<Class<? extends Node>, List<MatchStatement>> result = registry.get(theClass);
if (result == null) {
EconomicMap<Class<? extends Node>, List<MatchStatement>> rules = createRules(theClass);
registry.put(theClass, rules);
assert registry.get(theClass) == rules;
result = rules;
if (LogVerbose.getValue(options)) {
try (DebugContext.Scope s = debug.scope("MatchComplexExpressions")) {
debug.log("Match rules for %s", theClass.getSimpleName());
MapCursor<Class<? extends Node>, List<MatchStatement>> cursor = result.getEntries();
while (cursor.advance()) {
debug.log(" For node class: %s", cursor.getKey());
for (MatchStatement statement : cursor.getValue()) {
debug.log(" %s", statement.getPattern());
}
}
}
}
}
if (result.size() == 0) {
return null;
}
return result;
}
@SuppressWarnings("unchecked")
public static EconomicMap<Class<? extends Node>, List<MatchStatement>> createRules(Class<? extends NodeMatchRules> theClass) {
EconomicMap<Class<? extends NodeMatchRules>, MatchStatementSet> matchSets = EconomicMap.create(Equivalence.IDENTITY);
Iterable<MatchStatementSet> sl = GraalServices.load(MatchStatementSet.class);
for (MatchStatementSet rules : sl) {
matchSets.put(rules.forClass(), rules);
}
EconomicMap<Class<? extends Node>, List<MatchStatement>> rules = EconomicMap.create(Equivalence.IDENTITY);
Class<? extends NodeMatchRules> currentClass = theClass;
do {
MatchStatementSet matchSet = matchSets.get(currentClass);
if (matchSet != null) {
List<MatchStatement> statements = matchSet.statements();
for (MatchStatement statement : statements) {
Class<? extends Node> nodeClass = statement.getPattern().nodeClass();
List<MatchStatement> current = rules.get(nodeClass);
if (current == null) {
current = new ArrayList<>();
rules.put(nodeClass, current);
}
current.add(statement);
}
}
currentClass = (Class<? extends NodeMatchRules>) currentClass.getSuperclass();
} while (currentClass != NodeMatchRules.class);
return rules;
}
}