fr.lirmm.graphik.integraal.homomorphism.BacktrackIteratorData Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of integraal-graal-ruleset-analysis Show documentation
Show all versions of integraal-graal-ruleset-analysis Show documentation
Rule base analysis for InteGraal. This is imported from Graal
The newest version!
/*
* Copyright (C) Inria Sophia Antipolis - Méditerranée / LIRMM
* (Université de Montpellier & CNRS) (2014 - 2017)
*
* Contributors :
*
* Clément SIPIETER
* Mélanie KÖNIG
* Swan ROCHER
* Jean-François BAGET
* Michel LECLÈRE
* Marie-Laure MUGNIER
*
*
* This file is part of Graal index;
int levelMax;
List ans;
Collection negParts;
Profiler profiler;
boolean isOpen = false;
public BacktrackIteratorData(InMemoryAtomSet query, Set variablesToParameterize,
Collection negParts, AtomSet data, List ans, Scheduler scheduler,
Bootstrapper bootstrapper, ForwardChecking fc, BackJumping bj, RulesCompilation compilation,
Profiler profiler) throws HomomorphismException {
this.query = query;
this.negParts = negParts;
this.data = data;
this.ans = ans;
this.scheduler = scheduler;
this.bootstrapper = bootstrapper;
this.fc = fc;
this.bj = bj;
this.fc.setBackJumping(bj);
this.compilation = compilation;
this.profiler = profiler;
this.bootstrapper.setProfiler(profiler);
this.scheduler.setProfiler(profiler);
this.fc.setProfiler(profiler);
this.bj.setProfiler(profiler);
this.preprocessing(variablesToParameterize, profiler);
}
private void preprocessing(Set variablesToParameterize, Profiler profiler) throws HomomorphismException {
profiler.start("preprocessingTime");
// Compute order on query variables and atoms
this.varsOrder = this.scheduler.execute(this.query, variablesToParameterize, ans, this.data, this.compilation);
this.levelMax = varsOrder.length - 2;
// PROFILING
if (profiler.isProfilingEnabled()) {
this.profilingVarOrder(this.varsOrder, profiler);
}
// Index Var structures by original variable object
this.index = new TreeMap();
for (VarSharedData v : this.varsOrder) {
if (v.value != null) { //
this.index.put(v.value, v.level);
}
}
if (this.ans.isEmpty()) {
this.varsOrder[this.levelMax + 1].previousLevel = -1;
}
computeAtomOrder(this.query, this.varsOrder, this.index);
this.fc.init(this.varsOrder, this.index);
this.bj.init(this.varsOrder);
Set allVarsFromH = query.getVariables();
allVarsFromH.removeAll(variablesToParameterize);
for (InMemoryAtomSet negPart : this.negParts) {
Set allVarsFromNegPart = negPart.getVariables();
allVarsFromNegPart.removeAll(variablesToParameterize);
Set frontierNotParametrized = SetUtils.intersection(allVarsFromH, allVarsFromNegPart);
this.varsOrder[maxLevel(frontierNotParametrized)].negatedPartsToCheck.add(BacktrackHomomorphismPattern.instance()
.prepareHomomorphism(new DefaultConjunctiveQuery(negPart, Collections.emptyList()), SetUtils.union(variablesToParameterize, frontierNotParametrized),
this.data, this.compilation));
}
profiler.stop("preprocessingTime");
}
public void clear() {
this.scheduler.clear();
this.fc.clear();
this.bj.clear();
}
/**
* The index 0 contains the fully instantiated atoms.
*
* @param atomset
* @param vars
*/
private static void computeAtomOrder(CloseableIterableWithoutException atomset, VarSharedData[] vars,
Map index) {
int tmp, rank;
// initialisation preAtoms and postAtoms Collections
for (int i = 0; i < vars.length; ++i) {
vars[i].preAtoms = new HashSet();
vars[i].postAtoms = new HashSet();
vars[i].postVars = new HashSet();
vars[i].preVars = new TreeSet();
}
//
CloseableIteratorWithoutException it = atomset.iterator();
while (it.hasNext()) {
Atom a = it.next();
rank = 0;
for (Variable t : a.getVariables()) {
Integer idx = index.get(t);
if (idx != null) {
tmp = vars[idx].level;
vars[tmp].postAtoms.add(a);
if (rank < tmp)
rank = tmp;
}
}
vars[rank].postAtoms.remove(a);
vars[rank].preAtoms.add(a);
}
for (int i = 0; i < vars.length; ++i) {
for (Atom a : vars[i].postAtoms) {
for (Variable t : a.getVariables()) {
Integer idx = index.get(t);
if (idx != null) {
if (vars[idx].level > i) {
vars[i].postVars.add(vars[idx]);
vars[idx].preVars.add(vars[i]);
}
}
}
}
}
}
/**
*
* @param frontier
* @return the biggest level of specified variables
*/
private int maxLevel(Set frontier) {
int max = 0;
for (Variable v : frontier) {
int level = this.index.get(v);
if (level > max) {
max = level;
}
}
return max;
}
// /////////////////////////////////////////////////////////////////////////
// PROFLING METHODS
// /////////////////////////////////////////////////////////////////////////
private void profilingVarOrder(VarSharedData[] varsOrder, Profiler profiler) {
StringBuilder sb = new StringBuilder();
for (int i = 1; i < varsOrder.length - 1; ++i) {
sb.append(varsOrder[i].value.toString());
sb.append(' ');
}
profiler.incr("__#cqs", 1);
profiler.put("SchedulingSubQuery" + profiler.get("__#cqs"), sb.toString());
}
}