org.apache.jena.ontapi.common.ResourceVocabularyImpl Maven / Gradle / Ivy

Go to download
/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.jena.ontapi.common;

import org.apache.jena.ontapi.OntJenaException;
import org.apache.jena.ontapi.model.OntAnnotationProperty;
import org.apache.jena.ontapi.model.OntClass;
import org.apache.jena.ontapi.model.OntDataProperty;
import org.apache.jena.ontapi.model.OntDataRange;
import org.apache.jena.ontapi.model.OntIndividual;
import org.apache.jena.ontapi.model.OntObject;
import org.apache.jena.ontapi.model.OntObjectProperty;
import org.apache.jena.graph.Node;
import org.apache.jena.rdf.model.Property;
import org.apache.jena.rdf.model.Resource;

import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
 * A {@link ResourceVocabulary} abstraction and a holder for some particular vocabulary {@link OntPersonality}'s implementations.
 * Each of the impl has a simple in-memory cache to speed-up,
 * since it is important to have quick access to desired types.
 */
abstract class ResourceVocabularyImpl implements ResourceVocabulary {
    private final Map, Set> map;

    ResourceVocabularyImpl(Map, Set> map) {
        this.map = Objects.requireNonNull(map);
    }

    @Override
    public Set get(Class key) throws OntJenaException {
        return map.getOrDefault(OntJenaException.notNull(key, "Null key"), Set.of());
    }

    @Override
    public boolean supports(Class type) {
        return map.containsKey(type);
    }

    static class BuiltinsImpl extends EntitiesImpl implements OntPersonality.Builtins {
        BuiltinsImpl(Map, Set> map) {
            super(map);
        }
    }

    static class PunningsImpl extends EntitiesImpl implements OntPersonality.Punnings {
        PunningsImpl(Map, Set> map) {
            super(map);
        }
    }

    static abstract class EntitiesImpl extends ResourceVocabularyImpl {
        final Set classes;
        final Set datatypes;
        final Set objectProperties;
        final Set datatypeProperties;
        final Set annotationProperties;
        final Set ontProperties;
        final Set individuals;

        EntitiesImpl(Map, Set> map) {
            super(map);
            this.classes = get(OntClass.Named.class);
            this.datatypes = get(OntDataRange.Named.class);
            this.individuals = get(OntIndividual.Named.class);
            this.objectProperties = get(OntObjectProperty.Named.class);
            this.datatypeProperties = get(OntDataProperty.class);
            this.annotationProperties = get(OntAnnotationProperty.class);
            this.ontProperties = collect(OntObjectProperty.Named.class, OntDataProperty.class, OntAnnotationProperty.class);
        }

        public Set getNamedClasses() {
            return classes;
        }

        public Set getDatatypes() {
            return datatypes;
        }

        public Set getObjectProperties() {
            return objectProperties;
        }

        public Set getDatatypeProperties() {
            return datatypeProperties;
        }

        public Set getAnnotationProperties() {
            return annotationProperties;
        }

        public Set getNamedIndividuals() {
            return individuals;
        }

        public Set getOntProperties() {
            return ontProperties;
        }

        @SafeVarargs
        private Set collect(Class... types) {
            return Arrays.stream(types).map(this::get)
                    .flatMap(Collection::stream)
                    .collect(Collectors.toUnmodifiableSet());
        }
    }

    static class ReservedIml extends ResourceVocabularyImpl implements OntPersonality.Reserved {
        private final Map> nodes = new HashMap<>();
        private final Set resources;
        private final Set properties;
        private final Set allResources;

        ReservedIml(Map, Set> map) {
            super(map);
            this.properties = get(Property.class);
            this.resources = get(Resource.class);
            this.allResources = Stream.of(properties, resources)
                    .flatMap(Collection::stream).collect(Collectors.toUnmodifiableSet());
        }

        @Override
        public Set getResources() {
            return resources;
        }

        @Override
        public Set getProperties() {
            return properties;
        }

        @Override
        public Set getAllResources() {
            return allResources;
        }
    }
}