-
Notifications
You must be signed in to change notification settings - Fork 14
/
TypesHandler.java
105 lines (87 loc) · 4.6 KB
/
TypesHandler.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
/**
* Copyright (C) 2020 Czech Technical University in Prague
* <p>
* This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public
* License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later
* version.
* <p>
* This program 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 General Public License for more
* details. You should have received a copy of the GNU General Public License along with this program. If not, see
* <http://www.gnu.org/licenses/>.
*/
package cz.cvut.kbss.ontodriver.owlapi;
import cz.cvut.kbss.ontodriver.model.*;
import cz.cvut.kbss.ontodriver.owlapi.connector.OntologySnapshot;
import cz.cvut.kbss.ontodriver.owlapi.util.MutableAddAxiom;
import cz.cvut.kbss.ontodriver.owlapi.util.MutableRemoveAxiom;
import org.semanticweb.owlapi.model.*;
import org.semanticweb.owlapi.reasoner.OWLReasoner;
import org.semanticweb.owlapi.search.EntitySearcher;
import java.net.URI;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;
class TypesHandler {
private final OWLOntology ontology;
private final OWLDataFactory dataFactory;
private final OWLReasoner reasoner;
private final OntologySnapshot snapshot;
private final OwlapiAdapter adapter;
TypesHandler(OwlapiAdapter adapter, OntologySnapshot snapshot) {
this.adapter = adapter;
this.snapshot = snapshot;
this.ontology = snapshot.getOntology();
this.dataFactory = snapshot.getDataFactory();
this.reasoner = snapshot.getReasoner();
}
Set<Axiom<URI>> getTypes(NamedResource subject, Collection<URI> contexts, boolean includeInferred) {
final Collection<? extends OWLClassExpression> owlClasses;
if (!includeInferred) {
owlClasses = loadExplicitClasses(subject);
} else {
owlClasses = inferClasses(subject);
}
return owlClassesToAxioms(subject, includeInferred, owlClasses);
}
private Collection<OWLClassExpression> loadExplicitClasses(NamedResource subject) {
return EntitySearcher.getTypes(getIndividual(subject), ontology.importsClosure()).collect(Collectors.toSet());
}
private OWLNamedIndividual getIndividual(NamedResource subject) {
return dataFactory.getOWLNamedIndividual(IRI.create(subject.getIdentifier()));
}
private Set<Axiom<URI>> owlClassesToAxioms(NamedResource subject, boolean inferred,
Collection<? extends OWLClassExpression> owlClasses) {
return owlClasses.stream().map(expr -> new AxiomImpl<>(subject,
Assertion.createClassAssertion(inferred), new Value<>(expr.asOWLClass().getIRI().toURI())))
.collect(Collectors.toSet());
}
private Collection<? extends OWLClassExpression> inferClasses(NamedResource subject) {
final OWLNamedIndividual individual = getIndividual(subject);
return reasoner.getTypes(individual, false).entities().collect(Collectors.toSet());
}
void addTypes(NamedResource subject, URI context, Set<URI> types) {
assert !types.isEmpty();
final List<OWLAxiom> axioms = getOwlAxiomsForTypes(subject, types);
final List<OWLOntologyChange> changes = axioms.stream().map(axiom -> new MutableAddAxiom(ontology, axiom))
.collect(Collectors.toList());
adapter.addTransactionalChanges(snapshot.applyChanges(changes));
}
private List<OWLAxiom> getOwlAxiomsForTypes(NamedResource subject, Set<URI> types) {
final List<OWLAxiom> axioms = new ArrayList<>(types.size());
final OWLNamedIndividual individual = getIndividual(subject);
axioms.addAll(types.stream().map(type -> dataFactory
.getOWLClassAssertionAxiom(dataFactory.getOWLClass(IRI.create(type)), individual))
.collect(Collectors.toList()));
return axioms;
}
void removeTypes(NamedResource subject, URI context, Set<URI> types) {
assert !types.isEmpty();
final List<OWLAxiom> axioms = getOwlAxiomsForTypes(subject, types);
final List<OWLOntologyChange> changes = axioms.stream().map(axiom -> new MutableRemoveAxiom(ontology, axiom))
.collect(Collectors.toList());
adapter.addTransactionalChanges(snapshot.applyChanges(changes));
}
}