-
Notifications
You must be signed in to change notification settings - Fork 14
/
StatementLoader.java
123 lines (110 loc) · 5.21 KB
/
StatementLoader.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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
/**
* Copyright (C) 2019 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.sesame;
import cz.cvut.kbss.ontodriver.descriptor.AxiomDescriptor;
import cz.cvut.kbss.ontodriver.model.Assertion;
import cz.cvut.kbss.ontodriver.model.Axiom;
import cz.cvut.kbss.ontodriver.sesame.config.RuntimeConfiguration;
import cz.cvut.kbss.ontodriver.sesame.connector.Connector;
import cz.cvut.kbss.ontodriver.sesame.exceptions.SesameDriverException;
import cz.cvut.kbss.ontodriver.sesame.util.AxiomBuilder;
import cz.cvut.kbss.ontodriver.sesame.util.SesameUtils;
import org.eclipse.rdf4j.model.IRI;
import org.eclipse.rdf4j.model.Resource;
import org.eclipse.rdf4j.model.Statement;
import org.eclipse.rdf4j.model.ValueFactory;
import java.util.Collection;
import java.util.HashSet;
import java.util.Map;
class StatementLoader {
private final AxiomDescriptor descriptor;
private final Connector connector;
private final Resource subject;
private final ValueFactory vf;
private final AxiomBuilder axiomBuilder;
private final int loadAllThreshold;
private boolean loadAll;
private boolean includeInferred;
StatementLoader(RuntimeConfiguration config, AxiomDescriptor descriptor, Connector connector, Resource subject,
AxiomBuilder axiomBuilder) {
this.loadAllThreshold = config.getLoadAllThreshold();
this.descriptor = descriptor;
this.connector = connector;
this.vf = connector.getValueFactory();
this.subject = subject;
this.axiomBuilder = axiomBuilder;
}
void setIncludeInferred(boolean includeInferred) {
this.includeInferred = includeInferred;
}
Collection<Axiom<?>> loadAxioms(Map<IRI, Assertion> properties)
throws SesameDriverException {
this.loadAll = properties.values().contains(Assertion.createUnspecifiedPropertyAssertion(includeInferred));
if (properties.size() < loadAllThreshold && !loadAll) {
return loadOneByOne(properties.values());
} else {
return loadAll(properties);
}
}
private Collection<Axiom<?>> loadOneByOne(Collection<Assertion> assertions) throws SesameDriverException {
final Collection<Axiom<?>> result = new HashSet<>();
for (Assertion a : assertions) {
final IRI context = SesameUtils.toSesameIri(descriptor.getAssertionContext(a), vf);
final IRI property = SesameUtils.toSesameIri(a.getIdentifier(), vf);
final Collection<Statement> statements;
statements = connector.findStatements(subject, property, null, includeInferred, context);
for (Statement s : statements) {
final Axiom<?> axiom = axiomBuilder.statementToAxiom(s, a);
if (axiom != null) {
result.add(axiom);
}
}
}
return result;
}
private Collection<Axiom<?>> loadAll(Map<IRI, Assertion> properties) throws SesameDriverException {
final Collection<Statement> statements = connector.findStatements(subject, null, null, includeInferred);
final Collection<Axiom<?>> result = new HashSet<>(statements.size());
final Assertion unspecified = Assertion.createUnspecifiedPropertyAssertion(includeInferred);
for (Statement s : statements) {
if (!properties.containsKey(s.getPredicate()) && !loadAll) {
continue;
}
final Assertion a = getAssertion(properties, s);
if (!contextMatches(a, s) && !(loadAll && contextMatches(unspecified, s))) {
continue;
}
final Axiom<?> axiom = axiomBuilder.statementToAxiom(s);
if (axiom != null) {
result.add(axiom);
}
}
return result;
}
private Assertion getAssertion(Map<IRI, Assertion> properties, Statement s) {
if (properties.containsKey(s.getPredicate())) {
return properties.get(s.getPredicate());
}
return Assertion.createUnspecifiedPropertyAssertion(includeInferred);
}
private boolean contextMatches(Assertion a, Statement s) {
final java.net.URI assertionCtx = descriptor.getAssertionContext(a);
final Resource statementContext = s.getContext();
if (assertionCtx == null) {
// If the assertion should be in default, we don't care about the context of the statement, because
// the default is a union of all the contexts
return true;
}
return statementContext != null && assertionCtx.toString().equals(statementContext.stringValue());
}
}