org.coode.oppl.querymatching.AssertedTreeSearchSingleAxiomQuery Maven / Gradle / Ivy
/**
* Copyright (C) 2008, University of Manchester
*
* Modifications to the initial code base are copyright of their
* respective authors, or their employers as appropriate. Authorship
* of the modifications may be determined from the ChangeLog placed at
* the end of this file.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
* This library 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
* Lesser General Public License for more details.
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
package org.coode.oppl.querymatching;
import static org.coode.oppl.utils.ArgCheck.checkNotNull;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.SortedSet;
import java.util.TreeSet;
import org.coode.oppl.ConstraintSystem;
import org.coode.oppl.Variable;
import org.coode.oppl.bindingtree.BindingNode;
import org.coode.oppl.exceptions.RuntimeExceptionHandler;
import org.coode.oppl.search.OPPLAssertedSingleOWLAxiomSearchTree;
import org.coode.oppl.search.OPPLOWLAxiomSearchNode;
import org.coode.oppl.search.SearchTree;
import org.coode.oppl.utils.OWLObjectExtractor;
import org.coode.oppl.utils.PositionBasedVariableComparator;
import org.coode.oppl.utils.VariableExtractor;
import org.semanticweb.owlapi.model.OWLAxiom;
import org.semanticweb.owlapi.model.OWLObject;
import org.semanticweb.owlapi.model.OWLOntology;
/** @author Luigi Iannone */
public class AssertedTreeSearchSingleAxiomQuery extends AbstractAxiomQuery {
private final Map> searchTrees = new HashMap>();
private final ConstraintSystem constraintSystem;
private final Set ontologies = new HashSet();
private final Map> instantiations = new HashMap>();
private final Map> cache = new HashMap>();
/** @param ontologies
* ontologies
* @param constraintSystem
* constraintSystem
* @param runtimeExceptionHandler
* runtimeExceptionHandler */
public AssertedTreeSearchSingleAxiomQuery(Set ontologies,
ConstraintSystem constraintSystem,
RuntimeExceptionHandler runtimeExceptionHandler) {
super(runtimeExceptionHandler);
this.constraintSystem = checkNotNull(constraintSystem, "constraintSystem");
this.ontologies.addAll(checkNotNull(ontologies, "ontologies"));
}
@Override
protected Set match(OWLAxiom axiom) {
clearInstantions();
List> solutions = new ArrayList>();
VariableExtractor variableExtractor = new VariableExtractor(
getConstraintSystem(), false);
Set> extractedVariables = variableExtractor.extractVariables(axiom);
SortedSet> sortedVariables = new TreeSet>(
new PositionBasedVariableComparator(axiom, getConstraintSystem()
.getOntologyManager().getOWLDataFactory()));
sortedVariables.addAll(extractedVariables);
OPPLOWLAxiomSearchNode start = new OPPLOWLAxiomSearchNode(axiom, new BindingNode(
sortedVariables));
solutions.addAll(doMatch(start));
return extractLeaves(solutions);
}
/** @param start
* start
* @return search nodes */
private List> doMatch(OPPLOWLAxiomSearchNode start) {
List> solutions = new ArrayList>();
for (OWLOntology ontology : ontologies) {
for (OWLAxiom targetAxiom : filterAxioms(start.getAxiom(),
ontology.getAxioms())) {
if (start.getAxiom().getAxiomType().equals(targetAxiom.getAxiomType())) {
solutions.addAll(matchTargetAxiom(start, targetAxiom));
}
}
}
return solutions;
}
private Set extractLeaves(List> solutions) {
Set toReturn = new HashSet();
for (List path : solutions) {
OPPLOWLAxiomSearchNode searchLeaf = path.get(path.size() - 1);
BindingNode leaf = searchLeaf.getBinding();
toReturn.add(leaf);
}
return toReturn;
}
/** @param targetAxiom
* targetAxiom
* @param start
* start
* @return search nodes */
private List> matchTargetAxiom(
OPPLOWLAxiomSearchNode start, OWLAxiom targetAxiom) {
SearchTree searchTree = getSearchTree(targetAxiom);
List> solutions = new ArrayList>();
searchTree.exhaustiveSearchTree(start, solutions);
return solutions;
}
/** @param targetAxiom
* targetAxiom
* @return search nodes */
private SearchTree getSearchTree(OWLAxiom targetAxiom) {
SearchTree toReturn = searchTrees.get(targetAxiom);
if (toReturn == null) {
toReturn = new OPPLAssertedSingleOWLAxiomSearchTree(targetAxiom,
getConstraintSystem(), getRuntimeExceptionHandler());
searchTrees.put(targetAxiom, toReturn);
}
return toReturn;
}
private void clearInstantions() {
instantiations.clear();
}
/** @return instantiations */
public Map> getInstantiations() {
return new HashMap>(instantiations);
}
/** @return the constraintSystem */
public ConstraintSystem getConstraintSystem() {
return constraintSystem;
}
private Collection filterAxioms(OWLAxiom toMatchAxiom,
Collection axioms) {
Set toReturn = new HashSet();
VariableExtractor variableExtractor = new VariableExtractor(
getConstraintSystem(), true);
Set> variables = variableExtractor.extractVariables(toMatchAxiom);
Collection toMatchAllOWLObjects = extractOWLObjects(toMatchAxiom);
for (OWLAxiom candidate : axioms) {
// if (candidate.getAxiomType().equals(toMatchAxiom.getAxiomType()))
// {
Collection candidateAllOWLObjects = extractOWLObjects(candidate);
if (candidate.getAxiomType().equals(toMatchAxiom.getAxiomType())
&& toMatchAllOWLObjects.containsAll(candidateAllOWLObjects)) {
toReturn.add(candidate);
} else {
Set difference = new HashSet(candidateAllOWLObjects);
difference.removeAll(toMatchAllOWLObjects);
Iterator iterator = difference.iterator();
boolean found = false;
while (!found && iterator.hasNext()) {
OWLObject leftOutOWLObject = iterator.next();
Iterator> variableIterator = variables
.iterator();
boolean compatible = false;
while (!compatible && variableIterator.hasNext()) {
Variable variable = variableIterator.next();
compatible = variable.getType()
.isCompatibleWith(leftOutOWLObject);
}
found = !compatible;
}
if (!found) {
toReturn.add(candidate);
}
}
}
return toReturn;
}
/** @param axiom
* axiom
* @return owl objects */
private Collection extractOWLObjects(OWLAxiom axiom) {
Collection toReturn = cache.get(axiom);
if (toReturn == null) {
toReturn = OWLObjectExtractor.getAllOWLPrimitives(axiom);
cache.put(axiom, toReturn);
}
return toReturn;
}
}