org.coode.oppl.search.OPPLAssertedSingleOWLAxiomSearchTree Maven / Gradle / Ivy
package org.coode.oppl.search;
import static org.coode.oppl.utils.ArgCheck.checkNotNull;
import java.util.*;
import org.coode.oppl.ConstraintSystem;
import org.coode.oppl.PartialOWLObjectInstantiator;
import org.coode.oppl.Variable;
import org.coode.oppl.VariableScope;
import org.coode.oppl.bindingtree.Assignment;
import org.coode.oppl.bindingtree.BindingNode;
import org.coode.oppl.exceptions.RuntimeExceptionHandler;
import org.coode.oppl.function.SimpleValueComputationParameters;
import org.coode.oppl.function.ValueComputationParameters;
import org.coode.oppl.utils.AbstractVariableVisitorExAdapter;
import org.coode.oppl.utils.OWLObjectExtractor;
import org.coode.oppl.variabletypes.*;
import org.semanticweb.owlapi.model.*;
import org.semanticweb.owlapi.util.OWLAxiomVisitorAdapter;
/**
* @author Luigi Iannone
*/
public class OPPLAssertedSingleOWLAxiomSearchTree extends
SearchTree {
protected final ConstraintSystem constraintSystem;
protected final RuntimeExceptionHandler runtimeExceptionHandler;
protected final OWLAxiom targetAxiom;
protected final Set allClasses = new HashSet();
protected final Set allObjectProperties = new HashSet();
protected final Set allDataProperties = new HashSet();
protected final Set allIndividuals = new HashSet();
protected final Set allConstants = new HashSet();
protected final Set allAnnotationProperties = new HashSet();
/**
* @param targetAxiom
* targetAxiom
* @param constraintSystem
* constraintSystem
* @param runtimeExceptionHandler
* runtimeExceptionHandler
*/
public OPPLAssertedSingleOWLAxiomSearchTree(OWLAxiom targetAxiom,
ConstraintSystem constraintSystem,
RuntimeExceptionHandler runtimeExceptionHandler) {
this.constraintSystem = checkNotNull(constraintSystem, "constraintSystem");
this.runtimeExceptionHandler = checkNotNull(runtimeExceptionHandler,
"runtimeExceptionHandler");
this.targetAxiom = targetAxiom;
initAssignableValues();
}
@Override
protected List getChildren(OPPLOWLAxiomSearchNode node) {
List toReturn = new ArrayList();
Set> variables = node.getBinding().getUnassignedVariables();
BindingNode binding = node.getBinding();
ValueComputationParameters parameters = new SimpleValueComputationParameters(
getConstraintSystem(), binding, getRuntimeExceptionHandler());
if (!variables.isEmpty()) {
Variable variable = variables.iterator().next();
Collection values = new HashSet(getAssignableValues(
variable, parameters));
for (OWLObject value : values) {
Assignment assignment = new Assignment(variable, value);
BindingNode childBinding = new BindingNode(binding);
childBinding.addAssignment(assignment);
ValueComputationParameters childParameters = new SimpleValueComputationParameters(
getConstraintSystem(), childBinding, getRuntimeExceptionHandler());
PartialOWLObjectInstantiator instantiator = new PartialOWLObjectInstantiator(
childParameters);
OWLAxiom instantiatedAxiom = (OWLAxiom) node.getAxiom().accept(
instantiator);
OPPLOWLAxiomSearchNode child = new OPPLOWLAxiomSearchNode(
instantiatedAxiom, childBinding);
toReturn.add(child);
}
}
return toReturn;
}
@Override
protected boolean goalReached(OPPLOWLAxiomSearchNode start) {
return targetAxiom.getAxiomWithoutAnnotations().equals(start.getAxiom());
}
private final VariableTypeVisitorEx> assignableValuesVisitor = new VariableTypeVisitorEx>() {
@Override
public Set visitCLASSVariableType(
CLASSVariableType classVariableType) {
return allClasses;
}
@Override
public Set visitOBJECTPROPERTYVariableType(
OBJECTPROPERTYVariableType objectpropertyVariableType) {
return allObjectProperties;
}
@Override
public Set visitDATAPROPERTYVariableType(
DATAPROPERTYVariableType datapropertyVariableType) {
return allDataProperties;
}
@Override
public Set visitINDIVIDUALVariableType(
INDIVIDUALVariableType individualVariableType) {
return allIndividuals;
}
@Override
public Set visitCONSTANTVariableType(
CONSTANTVariableType constantVariableType) {
return allConstants;
}
@Override
public Set visitANNOTATIONPROPERTYVariableType(
ANNOTATIONPROPERTYVariableType annotationpropertyVariableType) {
return allAnnotationProperties;
}
};
private Collection getAssignableValues(Variable variable,
ValueComputationParameters parameters) {
Set toReturn = new HashSet();
toReturn.addAll(variable.accept(new AssignableValueExtractor(
assignableValuesVisitor, parameters)));
Iterator iterator = toReturn.iterator();
while (iterator.hasNext()) {
final OWLObject owlObject = iterator.next();
boolean inScope = variable
.accept(new AbstractVariableVisitorExAdapter(true) {
@Override
public Boolean visit(InputVariable
v) {
VariableScope variableScope = v.getVariableScope();
try {
return variableScope == null
|| variableScope.check(owlObject);
} catch (OWLRuntimeException e) {
OPPLAssertedSingleOWLAxiomSearchTree.this
.getRuntimeExceptionHandler()
.handleOWLRuntimeException(e);
return false;
}
}
});
if (!inScope) {
iterator.remove();
}
}
return toReturn;
}
private void initAssignableValues() {
extractFromLogicAxiom();
getTargetAxiom().accept(new OWLAxiomVisitorAdapter() {
@Override
public void visit(OWLAsymmetricObjectPropertyAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLClassAssertionAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLDataPropertyAssertionAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLDataPropertyDomainAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLDataPropertyRangeAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLDeclarationAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLDifferentIndividualsAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLDisjointClassesAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLDisjointDataPropertiesAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLDisjointObjectPropertiesAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLDisjointUnionAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLEquivalentClassesAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLEquivalentDataPropertiesAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLEquivalentObjectPropertiesAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLFunctionalDataPropertyAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLFunctionalObjectPropertyAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLHasKeyAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLInverseFunctionalObjectPropertyAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLInverseObjectPropertiesAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLIrreflexiveObjectPropertyAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLNegativeDataPropertyAssertionAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLNegativeObjectPropertyAssertionAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLObjectPropertyAssertionAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLSubPropertyChainOfAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLObjectPropertyDomainAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLObjectPropertyRangeAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLReflexiveObjectPropertyAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLSameIndividualAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLSubClassOfAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLSubDataPropertyOfAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLSubObjectPropertyOfAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLSymmetricObjectPropertyAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLTransitiveObjectPropertyAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(SWRLRule rule) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLAnnotationPropertyDomainAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLAnnotationPropertyRangeAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLSubAnnotationPropertyOfAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLDatatypeDefinitionAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
}
@Override
public void visit(OWLAnnotationAssertionAxiom axiom) {
OPPLAssertedSingleOWLAxiomSearchTree.this.extractFromLogicAxiom();
OWLAnnotationSubject subject = axiom.getSubject();
subject.accept(new OWLAnnotationSubjectVisitor() {
@Override
public void visit(OWLAnonymousIndividual individual) {}
@Override
public void visit(IRI iri) {
Set ontologies = OPPLAssertedSingleOWLAxiomSearchTree.this
.getConstraintSystem().getOntologyManager()
.getOntologies();
for (OWLOntology ontology : ontologies) {
Set entitiesInSignature = ontology
.getEntitiesInSignature(iri);
for (OWLEntity entity : entitiesInSignature) {
entity.accept(new OWLEntityVisitor() {
@Override
public void visit(OWLAnnotationProperty property) {
// TODO Auto-generated method stub
}
@Override
public void visit(OWLDatatype datatype) {}
@Override
public void visit(OWLNamedIndividual individual) {
allIndividuals.add(individual);
}
@Override
public void visit(OWLDataProperty property) {
allDataProperties.add(property);
}
@Override
public void visit(OWLObjectProperty property) {
allObjectProperties.add(property);
}
@Override
public void visit(OWLClass cls) {
allClasses.add(cls);
}
});
}
}
}
});
}
});
}
protected void extractFromLogicAxiom() {
allClasses.addAll(OWLObjectExtractor.getAllClasses(getTargetAxiom()));
allDataProperties.addAll(OWLObjectExtractor
.getAllOWLDataProperties(getTargetAxiom()));
allObjectProperties.addAll(OWLObjectExtractor
.getAllOWLObjectProperties(getTargetAxiom()));
allIndividuals.addAll(OWLObjectExtractor.getAllOWLIndividuals(getTargetAxiom()));
allConstants.addAll(OWLObjectExtractor.getAllOWLLiterals(getTargetAxiom()));
allAnnotationProperties.addAll(OWLObjectExtractor
.getAllAnnotationProperties(getTargetAxiom()));
}
/**
* @return the constraintSystem
*/
public ConstraintSystem getConstraintSystem() {
return constraintSystem;
}
@Override
public boolean exhaustiveSearchTree(OPPLOWLAxiomSearchNode start,
List> solutions) {
checkNotNull(start, "start");
checkNotNull(solutions, "solutions");
boolean found = false;
if (getTargetAxiom().getAxiomType().equals(start.getAxiom().getAxiomType())) {
found = super.exhaustiveSearchTree(start, solutions);
}
return found;
}
/**
* @return the targetAxiom
*/
public OWLAxiom getTargetAxiom() {
return targetAxiom;
}
/**
* @return the runtimeExceptionHandler
*/
public RuntimeExceptionHandler getRuntimeExceptionHandler() {
return runtimeExceptionHandler;
}
}