edu.stanford.protege.webprotege.hierarchy.DataPropertyHierarchyProviderImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of webprotege-backend-service Show documentation
The newest version!
package edu.stanford.protege.webprotege.hierarchy;


import edu.stanford.protege.webprotege.index.*;
import edu.stanford.protege.webprotege.inject.ProjectSingleton;
import edu.stanford.protege.webprotege.common.ProjectId;
import org.semanticweb.owlapi.model.*;

import javax.annotation.Nonnull;
import javax.inject.Inject;
import java.util.Collection;

import static com.google.common.base.Preconditions.checkNotNull;
import static java.util.stream.Collectors.toSet;
import static org.semanticweb.owlapi.model.AxiomType.SUB_DATA_PROPERTY;


/**
 * Author: Matthew Horridge

 * The University Of Manchester

 * Bio-Health Informatics Group

 * Date: 23-Jan-2007


 */
@ProjectSingleton
public class DataPropertyHierarchyProviderImpl extends AbstractOWLPropertyHierarchyProvider implements DataPropertyHierarchyProvider {

    @Nonnull
    private final ProjectOntologiesIndex projectOntologiesIndex;

    @Nonnull
    private final AxiomsByTypeIndex axiomsByTypeIndex;

    @Nonnull
    private final SubDataPropertyAxiomsBySubPropertyIndex subDataPropertyAxiomsBySubPropertyIndex;

    @Inject
    public DataPropertyHierarchyProviderImpl(ProjectId projectId,
                                             @DataPropertyHierarchyRoot OWLDataProperty root,
                                             @Nonnull ProjectOntologiesIndex projectOntologiesIndex,
                                             @Nonnull AxiomsByTypeIndex axiomsByTypeIndex,
                                             @Nonnull OntologySignatureByTypeIndex ontologySignatureByTypeIndex,
                                             @Nonnull SubDataPropertyAxiomsBySubPropertyIndex subDataPropertyAxiomsBySubPropertyIndex,
                                             @Nonnull EntitiesInProjectSignatureIndex entitiesInProjectSignatureIndex) {
        super(projectId, root, entitiesInProjectSignatureIndex, EntityType.DATA_PROPERTY, projectOntologiesIndex,
              ontologySignatureByTypeIndex);
        this.projectOntologiesIndex = checkNotNull(projectOntologiesIndex);
        this.axiomsByTypeIndex = checkNotNull(axiomsByTypeIndex);
        this.subDataPropertyAxiomsBySubPropertyIndex = checkNotNull(subDataPropertyAxiomsBySubPropertyIndex);
    }

    @Override
    protected String getHierarchyName() {
        return "data property";
    }



    @Override
    public Collection getChildren(OWLDataProperty property) {
        rebuildIfNecessary();
        if(getRoot().equals(property)) {
            return getChildrenOfRoot();
        }
        return projectOntologiesIndex.getOntologyIds()
                                     .flatMap(ontId -> axiomsByTypeIndex.getAxiomsByType(SUB_DATA_PROPERTY, ontId))
                                     .filter(ax -> ax.getSuperProperty().equals(property))
                                     .map(OWLSubDataPropertyOfAxiom::getSubProperty)
                                     .filter(OWLDataPropertyExpression::isNamed)
                                     .map(OWLDataPropertyExpression::asOWLDataProperty)
                                     .collect(toSet());
    }

    @Override
    public Collection getParents(OWLDataProperty property) {
        rebuildIfNecessary();
        return projectOntologiesIndex.getOntologyIds()
                                     .flatMap(ontId -> subDataPropertyAxiomsBySubPropertyIndex.getSubPropertyOfAxioms(
                                             property,
                                             ontId))
                                     .map(OWLSubPropertyAxiom::getSuperProperty)
                                     .filter(OWLDataPropertyExpression::isNamed)
                                     .map(OWLDataPropertyExpression::asOWLDataProperty)
                                     .collect(toSet());
    }

}