org.coode.oppl.querymatching.AssertedTreeSearchAxiomQuery Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of oppl2-oppl2 Show documentation
There is a newer version: 5.0.0
/**
 * 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.HashMap;
import java.util.HashSet;
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.PartialOWLObjectInstantiator;
import org.coode.oppl.Variable;
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.search.OPPLAssertedOWLAxiomSearchTree;
import org.coode.oppl.search.OPPLOWLAxiomSearchNode;
import org.coode.oppl.utils.PositionBasedVariableComparator;
import org.coode.oppl.utils.VariableExtractor;
import org.semanticweb.owlapi.model.OWLAxiom;
import org.semanticweb.owlapi.model.OWLOntology;

/** @author Luigi Iannone */
// XXX unused
public class AssertedTreeSearchAxiomQuery extends AbstractAxiomQuery {
    private final ConstraintSystem constraintSystem;
    private final Set ontologies = new HashSet();
    private final Map> instantiations = new HashMap>();

    /** @param ontologies
     *            ontologies
     * @param constraintSystem
     *            constraintSystem
     * @param runtimeExceptionHandler
     *            runtimeExceptionHandler */
    public AssertedTreeSearchAxiomQuery(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();
        OPPLAssertedOWLAxiomSearchTree searchTree = new OPPLAssertedOWLAxiomSearchTree(
                getConstraintSystem(), getRuntimeExceptionHandler());
        VariableExtractor variableExtractor = new VariableExtractor(
                getConstraintSystem(), false);
        Set> extractedVariables = variableExtractor.extractVariables(axiom);
        SortedSet> sortedVariables = new TreeSet>(
                new PositionBasedVariableComparator(axiom, getConstraintSystem()
                        .getOntologyManager().getOWLDataFactory()));
        sortedVariables.addAll(extractedVariables);
        List> solutions = new ArrayList>();
        searchTree.exhaustiveSearchTree(new OPPLOWLAxiomSearchNode(axiom,
                new BindingNode(sortedVariables)), solutions);
        for (List path : solutions) {
            OPPLOWLAxiomSearchNode searchLeaf = path.get(path.size() - 1);
            BindingNode leaf = searchLeaf.getBinding();
            ValueComputationParameters parameters = new SimpleValueComputationParameters(
                    getConstraintSystem(), leaf, getRuntimeExceptionHandler());
            PartialOWLObjectInstantiator partialOWLObjectInstantiator = new PartialOWLObjectInstantiator(
                    parameters);
            Set leafInstantiations = instantiations.get(leaf);
            if (leafInstantiations == null) {
                leafInstantiations = new HashSet();
            }
            leafInstantiations.add((OWLAxiom) axiom.accept(partialOWLObjectInstantiator));
            instantiations.put(leaf, leafInstantiations);
        }
        return new HashSet(instantiations.keySet());
    }

    private void clearInstantions() {
        instantiations.clear();
    }

    /** @return instantiations */
    public Map> getInstantiations() {
        return new HashMap>(instantiations);
    }

    /** @return the constraintSystem */
    public ConstraintSystem getConstraintSystem() {
        return constraintSystem;
    }

    /** @return ontologies */
    public Set getOntologies() {
        return new HashSet(ontologies);
    }
}